Searched defs:_batch_norm_with_update (Results 1 – 2 of 2) sorted by relevance
/external/pytorch/torch/jit/ | ||
D | _shape_functions.py | 1150 def _batch_norm_with_update( function |
/external/pytorch/torch/_decomp/ | ||
D | decompositions.py | 2058 def _batch_norm_with_update( function |