Searched defs:keys_tmp (Results 1 – 2 of 2) sorted by relevance
95 auto keys_tmp = at::empty_like(keys); in randperm_out_cuda() local111 auto keys_tmp = at::empty_like(keys); in randperm_out_cuda() local
702 auto keys_tmp = std::make_unique<int64_t[]>(nnz); in cpu_scatter_reduce_expanded_index() local