Searched defs:pool_indices (Results 1 – 2 of 2) sorted by relevance
336 auto pool_indices = pools[p]; in cpu_sparse_coo_softmax() local477 auto pool_indices = pools[p]; in cpu_sparse_coo_softmax_backward() local
115 int64_t* pool_indices = sorted_pool_indices + offset; in cuda_sparse_coo_softmax_kernel() local177 int64_t* pool_indices = sorted_pool_indices + offset; in cuda_sparse_coo_softmax_backward_kernel() local