Searched refs:rhs_rows (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 108 auto rhs_rows = rhs.dim_size(ndims_rhs - 2); in Compute() local
112 if (adj_y_) std::swap(rhs_rows, rhs_cols); in Compute()
113 OP_REQUIRES(ctx, lhs_cols == rhs_rows, in Compute()
115 "lhs mismatch rhs shape: ", lhs_cols, " vs. ", rhs_rows, in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 42 tensorflow::int64 rhs_rows = k; in MatMul() local
45 std::swap(rhs_rows, rhs_cols); in MatMul()
50 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> B(rhs, rhs_rows, in MatMul()
|
| D | runtime_matmul.cc | 49 tensorflow::int64 rhs_rows = k; in MatMul() local
52 std::swap(rhs_rows, rhs_cols); in MatMul()
57 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> B(rhs, rhs_rows, in MatMul()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 5622 int64 rhs_rows = (spec.rcd == 0) ? spec.k : spec.n; in TEST_P() local
5624 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P()
5627 /*from=*/10.0, /*to=*/10000.0, /*rows=*/rhs_rows, in TEST_P()
5681 int64 rhs_rows = (spec.rcd == 0) ? (spec.k + k_increase) : spec.n; in TEST_P() local
5683 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P()
5686 /*from=*/10.0, /*to=*/10000.0, /*rows=*/rhs_rows, in TEST_P()
|