Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfold_conv_bn.h19 at::Tensor bn_rv; member
H A Dfold_linear_bn.h11 at::Tensor bn_rv; member
H A Dfrozen_linear_folding.cpp56 auto bn_rv = constant_as<Tensor>(bn->namedInput("running_var")).value(); in FoldFrozenLinearBatchnorm() local
H A Dfrozen_conv_folding.cpp75 auto bn_rv = constant_as<Tensor>(bn->namedInput("running_var")).value(); in FoldFrozenConvBatchnorm() local
/aosp_15_r20/external/pytorch/torch/ao/quantization/quantizer/
H A Dxnnpack_quantizer_utils.py482 def _conv_bn(x, conv_weight, conv_bias, bn_weight, bn_bias, bn_rm, bn_rv): argument