Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dquantized_matmul_op_test.cc93 const int b_rows = 1; in TEST_F() local
143 const int b_rows = 1; in TEST_F() local
183 const int b_rows = 3; in TEST_F() local
307 const int b_rows = 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()
75 int b_rows = b.dimension(0); 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