Searched refs:rhs_slice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1178 llvm_ir::IrArray rhs_slice = in EmitBatchDotOperation() local
1185 dot_info, dot.name(), target_slice, lhs_slice, rhs_slice, nullptr, in EmitBatchDotOperation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 3677 auto rhs_slice = in TEST_F() local
3681 EXPECT_THAT(root, AllOf(op::Dot(lhs_slice, rhs_slice), in TEST_F()
5197 auto rhs_slice = in TEST_F() local
5201 op::Dot(resharded_lhs, rhs_slice))); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 1922 HloInstruction* rhs_slice = in OptimizeDotOfConcatHelper() local
1934 new_dot_lhs = rhs_slice; in OptimizeDotOfConcatHelper()
1938 new_dot_rhs = rhs_slice; in OptimizeDotOfConcatHelper()
|