Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:dot_rhs (Results 1 - 2 of 2) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc	71 auto dot_rhs = builder.AddInstruction( local 74 CreateCanonicalDot(result_shape, dot_lhs, dot_rhs)); 89 dot_rhs->shape().layout())); 108 auto dot_rhs = builder.AddInstruction( local 111 CreateCanonicalDot(result_shape, dot_a_lhs, dot_rhs)); 113 CreateCanonicalDot(result_shape, dot_b_lhs, dot_rhs)); 128 dot_rhs->shape().layout())); 152 auto dot_rhs = builder.AddInstruction( local 155 CreateCanonicalDot(result_a_shape, dot_a_lhs, dot_rhs)); 157 CreateCanonicalDot(result_b_shape, dot_b_lhs, dot_rhs)); 190 auto dot_rhs = builder.AddInstruction( local 230 auto dot_rhs = builder.AddInstruction( local 278 HloInstruction* dot_rhs = builder.AddInstruction( local     [all...]
cpu_instruction_fusion_test.cc	642 auto* dot_rhs = builder.AddInstruction( local 648 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs));     [all...]

Completed in 75 milliseconds