Searched refs:lhs_slice (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1176 llvm_ir::IrArray lhs_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 | 3673 auto lhs_slice = in TEST_F() local
3681 EXPECT_THAT(root, AllOf(op::Dot(lhs_slice, rhs_slice), in TEST_F()
5077 auto lhs_slice = AllOf(op::Shape("f32[24,100]"), op::DynamicSlice(lhs, _, _)); in TEST_F() local
5084 op::Dot(lhs_slice, partial_replicated_rhs))); in TEST_F()
|