Searched refs:getOutputSplitSizes (Results 1 – 2 of 2) sorted by relevance
74 const std::vector<int64_t>& getOutputSplitSizes() const { in getOutputSplitSizes() function in torch::ParamCommsDebugInfo
393 auto& outSplitSizes = debugInfo->getOutputSplitSizes(); in saveNcclMeta()