HomeSort by relevance Sort by last modified time
    Searched refs:outer_dimension_partitions (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_parallelization_preparation.cc 146 if (instruction->outer_dimension_partitions().empty()) {
152 instruction->outer_dimension_partitions();
183 return !instruction->outer_dimension_partitions().empty() ||
187 ->outer_dimension_partitions()
parallel_cpu_executable.cc 193 root->shape(), root->outer_dimension_partitions());
305 ->outer_dimension_partitions()
ir_emitter.cc 115 if (!computation->root_instruction()->outer_dimension_partitions().empty()) {
117 computation->root_instruction()->outer_dimension_partitions().size();
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_instruction.h 1260 const std::vector<int64>& outer_dimension_partitions() const { function in class:xla::HloInstruction
    [all...]
hlo_instruction.cc     [all...]

Completed in 100 milliseconds