Home
last modified time | relevance | path

Searched defs:reset_running_stats (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/api/include/torch/nn/modules/
Dbatchnorm.h56 void reset_running_stats() { in reset_running_stats() function
/external/pytorch/torch/ao/nn/intrinsic/qat/modules/
Dlinear_fused.py71 def reset_running_stats(self): member in LinearBn1d
Dconv_fused.py105 def reset_running_stats(self): member in _ConvBnNd
/external/pytorch/torch/nn/modules/
Dbatchnorm.py85 def reset_running_stats(self) -> None: member in _NormBase
/external/pytorch/test/quantization/eager/
Dtest_quantize_eager_qat.py109 def reset_running_stats(self): member in _ReferenceConvBnNd