Searched refs:rhs_cols (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 79 const int64 rhs_cols = rhs_shape.dimensions(num_batch_dims + 1); 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 | 4663 int64 rhs_cols = (spec.rcd == 0) ? spec.n : spec.k; in TEST_P() local
4664 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P()
4668 /*cols=*/rhs_cols))); in TEST_P()
4722 int64 rhs_cols = (spec.rcd == 0) ? spec.n : (spec.k + k_increase); in TEST_P() local
4723 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P()
4727 /*cols=*/rhs_cols))); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 46 int64 rhs_cols = n; in MatMul() local
48 std::swap(rhs_rows, rhs_cols); in MatMul()
54 rhs_cols); in MatMul()
|
| D | runtime_matmul.cc | 52 int64 rhs_cols = n; in MatMul() local
54 std::swap(rhs_rows, rhs_cols); in MatMul()
60 rhs_cols); in MatMul()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_batch_matmul_op.cc | 81 auto rhs_cols = rhs.dim_size(ndims - 1); in Compute() local
83 if (adj_y_) std::swap(rhs_rows, rhs_cols); in Compute()
90 out_shape.AddDim(rhs_cols); in Compute()
|