Searched refs:getInputSplitSizes (Results 1 – 2 of 2) sorted by relevance
70 const std::vector<int64_t>& getInputSplitSizes() const { in getInputSplitSizes() function in torch::ParamCommsDebugInfo
390 auto& inSplitSizes = debugInfo->getInputSplitSizes(); in saveNcclMeta()