Searched refs:act_input_scale (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qconv.cpp | 635 const auto act_input_scale = act_nhwc.q_scale(); in apply_impl_xnnp() local
641 (!input_scale.has_value() || input_scale.value() != act_input_scale)) { in apply_impl_xnnp()
645 input_scale = act_input_scale; in apply_impl_xnnp()
699 … at::Tensor qbias = quant_utils::QuantizeBias(per_channel(), bias, weight_contig, act_input_scale); in apply_impl_xnnp()
718 act_input_scale, in apply_impl_xnnp()
878 double act_input_scale = act_ndhwc.q_scale(); in apply_impl() local
881 if (!input_scale.has_value() || input_scale.value() != act_input_scale) { in apply_impl()
898 w_scales, act_input_scale, output_scale, requantization_scales); in apply_impl()
915 … quant_utils::QuantizeBias(convolution_op->per_channel, bias_fp32, weight_contig, act_input_scale); in apply_impl()
918 input_scale = act_input_scale; in apply_impl()
|