Searched refs:sliced_lhs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 156 const std::vector<Value>& sliced_lhs, const std::vector<Value>& sliced_rhs, in createMatMulOps() argument
173 /*a=*/sliced_lhs[lhs_batch_idx], in createMatMulOps()
285 std::vector<Value> sliced_lhs = in matchAndRewrite() local
293 createMatMulOps(sliced_lhs, sliced_rhs, bcast, lhs_shape[dims_a - 2], in matchAndRewrite()
|
| D | unroll_batch_matmul.h | 46 static TF::PackOp createMatMulOps(const std::vector<Value>& sliced_lhs,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 875 auto sliced_lhs = in HandleAdd() local
882 sliced_lhs, rhs->mutable_operand(1))); in HandleAdd()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 1230 auto sliced_lhs = in TEST_F() local
1236 EXPECT_THAT(root, AllOf(op::Convolution(sliced_lhs, rhs), in TEST_F()
|