Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/distributed/c10d/
DParamCommsUtils.hpp70 const std::vector<int64_t>& getInputSplitSizes() const { in getInputSplitSizes() function in torch::ParamCommsDebugInfo
/external/pytorch/torch/csrc/profiler/
Dutil.cpp390 auto& inSplitSizes = debugInfo->getInputSplitSizes(); in saveNcclMeta()