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 D | dynamic_update_slice_util.cc | 32 if (!instr->outer_dimension_partitions().empty()) { in MayBeImplementedAsInPlaceDynamicUpdateSlice()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo.proto | 279 // 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 D | hlo_instruction.h | 1845 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 D | hlo_instruction.cc | 985 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 D | hlo_parser.cc | 1170 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 D | hlo_instruction_test.cc | 2114 EXPECT_THAT(clone->outer_dimension_partitions(), in TEST_F()
|
H A D | hlo_parser_test.cc | 2154 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 D | ir_emitter.cc | 176 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 D | dot_op_emitter.cc | 1513 CHECK(dot.parent()->root_instruction()->outer_dimension_partitions().empty()); in EmitDotOperation()
|