Home
last modified time | relevance | path

Searched defs:b_rows (Results 1 – 2 of 2) 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 __global__ void SparseTensorDenseMatMulKernel(int nnz, int m, int b_rows, in SparseTensorDenseMatMulKernel()
77 int b_rows = b.dimension(0); in Compute() local