Searched refs:getInMessageNelems (Results 1 – 2 of 2) sorted by relevance
58 int64_t getInMessageNelems() const { in getInMessageNelems() function in torch::ParamCommsDebugInfo
387 map.emplace(kInMsgNelems, std::to_string(debugInfo->getInMessageNelems())); in saveNcclMeta()