Home
last modified time | relevance | path

Searched refs:outer_dimension_partitions (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Ddynamic_update_slice_util.cc32 if (!instr->outer_dimension_partitions().empty()) { in MayBeImplementedAsInPlaceDynamicUpdateSlice()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo.proto279 // outer_dimension_partitions[0] pieces, the next-outermost dim gets split
280 // into outer_dimension_partitions[1] pieces, etc.
284 repeated int64 outer_dimension_partitions = 64; field
H A Dhlo_instruction.h1845 const std::vector<int64_t>& outer_dimension_partitions() const { in outer_dimension_partitions() function
1849 const std::vector<int64_t>& outer_dimension_partitions);
H A Dhlo_instruction.cc985 proto.outer_dimension_partitions().begin(), in CreateFromProto()
986 proto.outer_dimension_partitions().end()); in CreateFromProto()
4427 const std::vector<int64_t>& outer_dimension_partitions) { in set_outer_dimension_partitions() argument
4428 outer_dimension_partitions_ = outer_dimension_partitions; in set_outer_dimension_partitions()
H A Dhlo_parser.cc1170 optional<std::vector<int64_t>> outer_dimension_partitions; in ParseInstructionRhs() local
1173 &outer_dimension_partitions}; in ParseInstructionRhs()
1236 if (outer_dimension_partitions) { in ParseInstructionRhs()
1237 instruction->set_outer_dimension_partitions(*outer_dimension_partitions); in ParseInstructionRhs()
H A Dhlo_instruction_test.cc2114 EXPECT_THAT(clone->outer_dimension_partitions(), in TEST_F()
H A Dhlo_parser_test.cc2154 ROOT foo = f32[100]{0} parameter(0), outer_dimension_partitions={0,10,20} in CreateTestCases()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc176 if (!computation->root_instruction()->outer_dimension_partitions().empty()) { in EmitComputation()
178 computation->root_instruction()->outer_dimension_partitions().size(); in EmitComputation()
2236 if (!computation->root_instruction()->outer_dimension_partitions().empty()) { in HandleCall()
2257 call_args, root->shape(), root->outer_dimension_partitions(), &b_, in HandleCall()
H A Ddot_op_emitter.cc1513 CHECK(dot.parent()->root_instruction()->outer_dimension_partitions().empty()); in EmitDotOperation()