Home
last modified time | relevance | path

Searched defs:inSplitSizes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DParamCommsUtils.hpp105 inSplitSizes, \ argument
146 inSplitSizes, \ argument
H A DParamCommsUtils.cpp17 std::vector<int64_t> inSplitSizes, in ParamCommsDebugInfo()
H A DProcessGroupNCCL.cpp4261 std::vector<int64_t> inSplitSizes; in checkForNCCLErrorsInternal() local
/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dutil.cpp390 auto& inSplitSizes = debugInfo->getInputSplitSizes(); in saveNcclMeta() local