| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_matmul.cc | 37 int64 lhs_cols = k; local
39 std::swap(lhs_rows, lhs_cols);
49 lhs, lhs_rows, lhs_cols);
|
| runtime_single_threaded_matmul.cc | 31 int64 lhs_cols = k; local
33 std::swap(lhs_rows, lhs_cols);
43 lhs, lhs_rows, lhs_cols);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dot_decomposer.cc | 66 const int64 lhs_cols = lhs_shape.dimensions(num_batch_dims + 1); local
68 ShapeUtil::MakeShape(lhs_type, {batch_size, lhs_rows, lhs_cols});
70 ShapeUtil::MakeShape(lhs_type, {1, lhs_rows, lhs_cols});
72 ShapeUtil::MakeShape(lhs_type, {lhs_rows, lhs_cols});
106 {i + 1, lhs_rows, lhs_cols}, {1, 1, 1}));
120 ShapeUtil::MakeShape(lhs_type, {lhs_cols, lhs_rows});
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_batch_matmul_op.cc | 79 auto lhs_cols = lhs.dim_size(ndims - 1); variable
82 if (adj_x_) std::swap(lhs_rows, lhs_cols);
84 OP_REQUIRES(ctx, lhs_cols == rhs_rows,
86 "lhs mismatch rhs shape: ", lhs_cols, " vs. ", rhs_rows,
|
