Searched refs:MatMulBCast (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | matmul_bcast_test.cc | 25 string MatMulBCastToStr(const MatMulBCast& b) { in MatMulBCastToStr()
38 MatMulBCast bcast({1, 5, 3}, {4, 3, 7}); in TEST()
51 MatMulBCast bcast({5, 3}, {3, 7}); in TEST()
64 MatMulBCast bcast({2, 4, 6, 5, 3}, {2, 4, 6, 3, 7}); in TEST()
77 MatMulBCast bcast1({5, 3}, {6, 3, 7}); in TEST()
87 MatMulBCast bcast2({2, 5, 3}, {3, 7}); in TEST()
98 MatMulBCast bcast1({3, 3}, {3}); in TEST()
101 MatMulBCast bcast2({3}, {3, 3}); in TEST()
104 MatMulBCast bcast3({3, 3}, {3, 3}); in TEST()
110 MatMulBCast bcast3({4, 5, 3}, {2, 3, 7}); in TEST()
[all …]
|
| D | matmul_bcast.h | 32 class MatMulBCast {
36 MatMulBCast(const Vec& x, const Vec& y) { in MatMulBCast() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_triangular_solve_op.cc | 45 MatMulBCast bcast(BCast::FromShape(lhs_shape), BCast::FromShape(rhs_shape)); in Compile()
76 const TensorShape& rhs_shape, const MatMulBCast& broadcast_helper);
84 const MatMulBCast& broadcast_helper) { in Broadcast()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_triangular_solve_op_impl.h | 85 bool adjoint, const MatMulBCast& bcast, Tensor* out, in Run()
122 const MatMulBCast& bcast, Tensor* out) {
166 MatMulBCast bcast(in0.shape().dim_sizes(), in1.shape().dim_sizes());
259 const MatMulBCast& bcast, Tensor* out) {
|
| D | banded_triangular_solve_op.cc | 63 bool adjoint, const MatMulBCast& bcast, Tensor* out, in Run()
170 const MatMulBCast& bcast, Tensor* out) { in Launch()
221 MatMulBCast bcast(in0.shape().dim_sizes(), in1.shape().dim_sizes()); in Compute()
|
| D | einsum_op_impl.h | 553 MatMulBCast bcast(inputs[0].shape().dim_sizes(), in ContractOperands()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.h | 48 const tensorflow::MatMulBCast& bcast,
|
| D | unroll_batch_matmul.cc | 157 const tensorflow::MatMulBCast& bcast, int rows, int cols, Type element_type, in createMatMulOps()
274 tensorflow::MatMulBCast bcast(absl::InlinedVector<tensorflow::int64, 4>( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 79 bool trans_y, const MatMulBCast& bcast, Tensor* out, in Run()
124 bool trans_y, const MatMulBCast& bcast, Tensor* out,
180 const MatMulBCast& bcast, Tensor* out, int start, int limit) {
230 bool trans_y, const MatMulBCast& bcast, Tensor* out) {
326 bool trans_y, const MatMulBCast& bcast, Tensor* out) {
497 bool trans_y, const MatMulBCast& bcast, Tensor* out) {
675 MatMulBCast bcast(in0.shape().dim_sizes(), in1.shape().dim_sizes());
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 97 MatMulBCast bcast(lhs.shape().dim_sizes(), rhs.shape().dim_sizes()); in Compute()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 174 ::tensorflow::MatMulBCast bcast( in Run()
|