| cpu_layout_assignment_test.cc | 69 auto dot_lhs = builder.AddInstruction( local
74 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs));
87 dot_lhs->shape().layout()));
188 auto dot_lhs = builder.AddInstruction( local
193 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs));
205 for (HloInstruction* instruction : {dot_lhs, dot_rhs, dot_result}) {
228 auto dot_lhs = builder.AddInstruction( local
233 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs));
245 for (HloInstruction* instruction : {dot_lhs, dot_rhs, dot_result}) {
274 HloInstruction* dot_lhs = builder.AddInstruction local
[all...] |
