| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_matmul.cc | 51 int64 rhs_rows = k; local
54 std::swap(rhs_rows, rhs_cols);
59 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> B(rhs, rhs_rows,
|
| runtime_single_threaded_matmul.cc | 45 int64 rhs_rows = k; local
48 std::swap(rhs_rows, rhs_cols);
53 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> B(rhs, rhs_rows,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dot_decomposer.cc | 78 const int64 rhs_rows = rhs_shape.dimensions(num_batch_dims + 0); local
81 ShapeUtil::MakeShape(rhs_type, {batch_size, rhs_rows, rhs_cols});
83 ShapeUtil::MakeShape(rhs_type, {1, rhs_rows, rhs_cols});
85 ShapeUtil::MakeShape(rhs_type, {rhs_rows, rhs_cols});
115 {i + 1, rhs_rows, rhs_cols}, {1, 1, 1}));
129 ShapeUtil::MakeShape(rhs_type, {rhs_cols, rhs_rows});
|
| algebraic_simplifier_test.cc | 4662 int64 rhs_rows = (spec.rcd == 0) ? spec.k : spec.n; local
4721 int64 rhs_rows = (spec.rcd == 0) ? (spec.k + k_increase) : spec.n; local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_batch_matmul_op.cc | 80 auto rhs_rows = rhs.dim_size(ndims - 2); variable
83 if (adj_y_) std::swap(rhs_rows, rhs_cols);
84 OP_REQUIRES(ctx, lhs_cols == rhs_rows,
86 "lhs mismatch rhs shape: ", lhs_cols, " vs. ", rhs_rows,
|
