Searched defs:selected_dim_indices (Results 1 – 2 of 2) sorted by relevance
1650 ) -> Tensor { in index_select_sparse_cuda()1745 Tensor selected_dim_indices, res_dim_indices; in index_select_sparse_cuda() local1748 auto selected_dim_indices = at::empty_like(res_dim_indices); in index_select_sparse_cuda() local
2412 const Tensor& res_dim_indices) -> Tensor { in index_select_sparse_cpu()2470 Tensor selected_dim_indices; in index_select_sparse_cpu() local