Home
last modified time | relevance | path

Searched defs:dot_rhs (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment_test.cc70 auto dot_rhs = builder.AddInstruction( in TEST_F() local
107 auto dot_rhs = builder.AddInstruction( in TEST_F() local
151 auto dot_rhs = builder.AddInstruction( in TEST_F() local
189 auto dot_rhs = builder.AddInstruction( in TEST_F() local
229 auto dot_rhs = builder.AddInstruction( in TEST_F() local
284 HloInstruction* dot_rhs = builder.AddInstruction( in RunDotOutputFusion() local
H A Dcpu_instruction_fusion_test.cc633 auto* dot_rhs = builder.AddInstruction( in CreateComputationForDotAddOutputFusionTest() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddot_merger.cc180 HloInstruction* dot_rhs = lhs_same ? concat_op : shared_op; in TryMergeSameOperand() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc152 const HloInstruction* dot_rhs, int64_t lhs_concat_dim, in GenNewWindow()
236 const HloInstruction* dot_rhs, int64_t lhs_concat_dim, in GenNewConvDNums()
1085 auto dot_rhs = r; in PartitionBaseCase() local
1431 auto dot_rhs = r; in PartitionBaseCase() local