Searched refs:rhs_rows (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 78 const int64 rhs_rows = rhs_shape.dimensions(num_batch_dims + 0); in DecomposeBatchDot() local
81 ShapeUtil::MakeShape(rhs_type, {batch_size, rhs_rows, rhs_cols}); in DecomposeBatchDot()
83 ShapeUtil::MakeShape(rhs_type, {1, rhs_rows, rhs_cols}); in DecomposeBatchDot()
85 ShapeUtil::MakeShape(rhs_type, {rhs_rows, rhs_cols}); in DecomposeBatchDot()
115 {i + 1, rhs_rows, rhs_cols}, {1, 1, 1})); in DecomposeBatchDot()
129 ShapeUtil::MakeShape(rhs_type, {rhs_cols, rhs_rows}); in DecomposeBatchDot()
|
| D | algebraic_simplifier_test.cc | 4662 int64 rhs_rows = (spec.rcd == 0) ? spec.k : spec.n; in TEST_P() local
4664 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P()
4667 /*from=*/10.0, /*to=*/10000.0, /*rows=*/rhs_rows, in TEST_P()
4721 int64 rhs_rows = (spec.rcd == 0) ? (spec.k + k_increase) : spec.n; in TEST_P() local
4723 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P()
4726 /*from=*/10.0, /*to=*/10000.0, /*rows=*/rhs_rows, in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 45 int64 rhs_rows = k; in MatMul() local
48 std::swap(rhs_rows, rhs_cols); in MatMul()
53 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> B(rhs, rhs_rows, in MatMul()
|
| D | runtime_matmul.cc | 51 int64 rhs_rows = k; in MatMul() local
54 std::swap(rhs_rows, rhs_cols); in MatMul()
59 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> B(rhs, rhs_rows, in MatMul()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_batch_matmul_op.cc | 80 auto rhs_rows = rhs.dim_size(ndims - 2); in Compute() local
83 if (adj_y_) std::swap(rhs_rows, rhs_cols); in Compute()
84 OP_REQUIRES(ctx, lhs_cols == rhs_rows, in Compute()
86 "lhs mismatch rhs shape: ", lhs_cols, " vs. ", rhs_rows, in Compute()
|