Searched defs:update_bn_stats (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/ | ||
| H A D | linear_fused.py | 82 def update_bn_stats(self): member in LinearBn1d |
| H A D | conv_fused.py | 121 def update_bn_stats(self): member in _ConvBnNd |
| /aosp_15_r20/external/pytorch/test/quantization/eager/ | ||
| H A D | test_quantize_eager_qat.py | 129 def update_bn_stats(self): member in _ReferenceConvBnNd |