Searched refs:b_cols (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | quantized_matmul_op_test.cc | 94 const int b_cols = 1; in TEST_F() local 120 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0}); in TEST_F() 130 Tensor expected(allocator(), DT_QINT32, TensorShape({a_cols, b_cols})); in TEST_F() 144 const int b_cols = 1; in TEST_F() local 164 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0}); in TEST_F() 184 const int b_cols = 2; in TEST_F() local 228 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), { in TEST_F() 255 Tensor expected(allocator(), DT_QINT32, TensorShape({a_cols, b_cols})); in TEST_F() 308 const int b_cols = 8; in TEST_F() local 311 Tensor b_float(DT_FLOAT, {b_rows, b_cols}); in TEST_F() [all …]
|
D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 32 int b_cols, int p, in SparseTensorDenseMatMulKernel() argument 38 const int n = (ADJ_B) ? b_cols : b_rows; in SparseTensorDenseMatMulKernel() 58 const T b_value = ldg(b + ((ADJ_B) ? j * b_cols + k : k * b_cols + j)); in SparseTensorDenseMatMulKernel() 78 int b_cols = b.dimension(1); in Compute() local 86 nnz, m, b_rows, b_cols, p, a_indices.data(), a_values.data(), in Compute()
|