Searched defs:return_mean (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 5181 def var_mean_sum_(x, axis, correction, keepdim, return_mean): argument
5223 def var_mean_welford_(x, axis, *, correction, keepdim, return_mean): argument
5266 def var_mean_helper_(x, *, axis, correction, keepdim, return_mean): argument
|