Home
last modified time | relevance | path

Searched refs:computeNumShards (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/distributed/autograd/context/
Dcontainer.h140 static uint32_t computeNumShards();
Dcontainer.cpp68 uint32_t DistAutogradContainer::computeNumShards() { in computeNumShards() function in torch::distributed::autograd::DistAutogradContainer
103 new DistAutogradContainer(computeNumShards()); in getInstanceInternal()