Searched refs:rhs_slice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1449 llvm_ir::IrArray rhs_slice = in EmitBatchDotOperation() local
1456 dot_info, dot.name(), target_slice, lhs_slice, rhs_slice, nullptr, in EmitBatchDotOperation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 2230 HloInstruction* rhs_slice = rhs->AddInstruction(HloInstruction::CreateSlice( in OptimizeDotOfConcatHelper() local
2241 new_dot_lhs = rhs_slice; in OptimizeDotOfConcatHelper()
2245 new_dot_rhs = rhs_slice; in OptimizeDotOfConcatHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 4135 const auto rhs_slice = in TEST_F() local
4139 EXPECT_THAT(root, AllOf(op::Dot(lhs_slice, rhs_slice), in TEST_F()
7576 const auto rhs_slice = in TEST_F() local
7580 op::Dot(resharded_lhs, rhs_slice))); in TEST_F()
|