Searched defs:outNelems (Results 1 – 3 of 3) sorted by relevance
103 outNelems, \ argument144 outNelems, \ argument
15 int64_t outNelems, in ParamCommsDebugInfo()
89 auto outNelems = (!outputTensors.empty()) ? outputTensors[0].numel() : 0; in recordComms() local