Searched defs:dense_rows (Results 1 – 2 of 2) sorted by relevance
88 GpuLaunchConfig cfg, Tindex dense_rows, int rank, const Tindex* indices, in CountElementsPerRowKernel()130 GpuLaunchConfig cfg, Tindex dense_rows, int rank, in ScatterInputElementsKernel()183 const Tindex dense_rows = dense_shape(0); // Must be on the host in operator ()() local
61 const Tindex dense_rows = dense_shape(0); in operator ()() local