Home
last modified time | relevance | path

Searched refs:b_cols (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dquantized_matmul_op_test.cc94 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 …]
Dsparse_tensor_dense_matmul_op_gpu.cu.cc32 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()