Home
last modified time | relevance | path

Searched defs:b_cols (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dquantized_matmul_op_test.cc94 const int b_cols = 1; in TEST_F() local
144 const int b_cols = 1; in TEST_F() local
184 const int b_cols = 2; in TEST_F() local
308 const int b_cols = 8; in TEST_F() local
Dsparse_tensor_dense_matmul_op_gpu.cu.cc31 int nnz, int m, int b_rows, int b_cols, int p, in SparseTensorDenseMatMulKernel()
76 int b_cols = b.dimension(1); in Compute() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils_test.cc1094 const int b_rows = 29, b_cols = 1, batches = 2; in TEST() local