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