Searched refs:src_indexing_axis_dim (Results 1 – 1 of 1) sorted by relevance
1261 auto src_indexing_axis_dim = self_sizes[1]; in index_select_out_cpu_dim1_() local1271 check_indexarray_range<index_t>(idxs, N, src_indexing_axis_dim); in index_select_out_cpu_dim1_()1330 auto src_indexing_axis_dim = self.size(dim); in index_select_out_cpu_() local1331 TORCH_CHECK(src_indexing_axis_dim > 0, in index_select_out_cpu_()1336 check_indexarray_range<index_t>(idxs, numel, src_indexing_axis_dim); in index_select_out_cpu_()