Searched refs:lhs_slice (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1447 llvm_ir::IrArray lhs_slice = in EmitBatchDotOperation() local
1456 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 | 4131 const auto lhs_slice = in TEST_F() local
4139 EXPECT_THAT(root, AllOf(op::Dot(lhs_slice, rhs_slice), in TEST_F()
7457 const auto lhs_slice = in TEST_F() local
7465 op::Dot(lhs_slice, partial_replicated_rhs))); in TEST_F()
|