Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc71 auto dot_rhs = builder.AddInstruction( in TEST_F() local
74 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs)); in TEST_F()
89 dot_rhs->shape().layout())); in TEST_F()
108 auto dot_rhs = builder.AddInstruction( in TEST_F() local
111 CreateCanonicalDot(result_shape, dot_a_lhs, dot_rhs)); in TEST_F()
113 CreateCanonicalDot(result_shape, dot_b_lhs, dot_rhs)); in TEST_F()
128 dot_rhs->shape().layout())); in TEST_F()
152 auto dot_rhs = builder.AddInstruction( in TEST_F() local
155 CreateCanonicalDot(result_a_shape, dot_a_lhs, dot_rhs)); in TEST_F()
157 CreateCanonicalDot(result_b_shape, dot_b_lhs, dot_rhs)); in TEST_F()
[all …]
Dcpu_instruction_fusion_test.cc651 auto* dot_rhs = 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.cc139 const HloInstruction* dot_rhs, int64 lhs_concat_dim, in GenNewWindow() argument
171 auto rhs_size = dot_rhs->shape().dimensions(rhs_concat_dim + 1); in GenNewWindow()
223 const HloInstruction* dot_rhs, int64 lhs_concat_dim, int64 rhs_concat_dim, in GenNewConvDNums() argument
278 kernel_spatial_dimensions.push_back(dot_rhs->shape().dimensions_size() - 1); in GenNewConvDNums()
760 auto dot_rhs = r; in PartitionBaseCase() local
871 dot_rhs = slice; in PartitionBaseCase()
911 dot_rhs = body_b.AddInstruction(HloInstruction::CreateConcatenate( in PartitionBaseCase()
913 original_dot_rhs = dot_rhs; in PartitionBaseCase()
916 std::vector<int64> reshaped_dims(dot_rhs->shape().dimensions().begin(), in PartitionBaseCase()
917 dot_rhs->shape().dimensions().end()); in PartitionBaseCase()
[all …]