Searched refs:computeNumShards (Results 1 – 2 of 2) sorted by relevance
140 static uint32_t computeNumShards();
68 uint32_t DistAutogradContainer::computeNumShards() { in computeNumShards() function in torch::distributed::autograd::DistAutogradContainer103 new DistAutogradContainer(computeNumShards()); in getInstanceInternal()