Searched refs:coo_rows_out (Results 1 – 1 of 1) sorted by relevance
150 int* coo_rows_out, in SparseTensorToCOOMatrixKernel() argument154 coo_rows_out[i] = static_cast<int>(ldg(indices + i * stride + offset)); in SparseTensorToCOOMatrixKernel()