Searched defs:dim_ind (Results 1 – 3 of 3) sorted by relevance
442 for (const auto dim_ind : c10::irange(ndim)) { in create_dim_backshift_permutation() local459 for (const auto dim_ind : c10::irange(ndim)) { in create_reverse_permutation() local
89 int dim_ind) -> std::vector<Shard> { in ExpandShardVector()109 for (int dim_ind = 0; dim_ind < new_num_shards_per_dim.size(); ++dim_ind) { in ExpandShardVector() local
366 for (auto dim_ind : dim_) { in norm_sparse() local