Searched defs:indices_contig (Results 1 – 2 of 2) sorted by relevance
120 auto indices_contig = indices.contiguous(); in embedding_dense_backward_cpu() local188 auto indices_contig = indices.contiguous(); in embedding_renorm_cpu_() local
263 auto indices_contig = indices.contiguous(); in embedding_dense_backward_cuda() local359 auto indices_contig = std::get<0>(indices.sort()).contiguous(); in embedding_renorm_cuda_() local