| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_matmul.cc | 43 int64 rhs_cols = n; local
45 std::swap(rhs_rows, rhs_cols);
51 rhs, rhs_rows, rhs_cols);
|
| runtime_single_threaded_matmul.cc | 37 int64 rhs_cols = n; local
39 std::swap(rhs_rows, rhs_cols);
45 rhs, rhs_rows, rhs_cols);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dot_decomposer.cc | 77 const int64 rhs_cols = rhs_shape.dimensions(num_batch_dims + 1); local
79 ShapeUtil::MakeShape(rhs_type, {batch_size, rhs_rows, rhs_cols});
81 ShapeUtil::MakeShape(rhs_type, {1, rhs_rows, rhs_cols});
83 ShapeUtil::MakeShape(rhs_type, {rhs_rows, rhs_cols});
113 {i + 1, rhs_rows, rhs_cols}, {1, 1, 1}));
127 ShapeUtil::MakeShape(rhs_type, {rhs_cols, rhs_rows});
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_batch_matmul_op.cc | 81 auto rhs_cols = rhs.dim_size(ndims - 1); variable
83 if (adj_y_) std::swap(rhs_rows, rhs_cols);
90 out_shape.AddDim(rhs_cols);
|
