Home
last modified time | relevance | path

Searched refs:dot_lhs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc69 auto dot_lhs = builder.AddInstruction( in TEST_F() local
74 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs)); in TEST_F()
87 dot_lhs->shape().layout())); in TEST_F()
188 auto dot_lhs = builder.AddInstruction( in TEST_F() local
193 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs)); in TEST_F()
205 for (HloInstruction* instruction : {dot_lhs, dot_rhs, dot_result}) { in TEST_F()
228 auto dot_lhs = builder.AddInstruction( in TEST_F() local
233 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs)); in TEST_F()
245 for (HloInstruction* instruction : {dot_lhs, dot_rhs, dot_result}) { in TEST_F()
281 HloInstruction* dot_lhs = builder.AddInstruction( in RunDotOutputFusion() local
[all …]
Dcpu_instruction_fusion_test.cc649 auto* dot_lhs = builder.AddInstruction( in CreateComputationForDotAddOutputFusionTest() local
657 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs)); in CreateComputationForDotAddOutputFusionTest()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc138 const HloInstruction* dot_lhs, in GenNewWindow() argument
149 auto lhs_size = dot_lhs->shape().dimensions(lhs_concat_dim + 1); in GenNewWindow()
222 const HloInstruction* original_dot, const HloInstruction* dot_lhs, in GenNewConvDNums() argument
253 input_spatial_dimensions.push_back(dot_lhs->shape().dimensions_size() - 1); in GenNewConvDNums()
759 auto dot_lhs = l; in PartitionBaseCase() local
865 dot_lhs = slice; in PartitionBaseCase()
888 dot_lhs = body_b.AddInstruction(HloInstruction::CreateConcatenate( in PartitionBaseCase()
890 original_dot_lhs = dot_lhs; in PartitionBaseCase()
893 std::vector<int64> reshaped_dims(dot_lhs->shape().dimensions().begin(), in PartitionBaseCase()
894 dot_lhs->shape().dimensions().end()); in PartitionBaseCase()
[all …]