Home
last modified time | relevance | path

Searched defs:weights_scales (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqconv.cpp1262 const ideep::scale_t& weights_scales = weights.get_scale(); in apply_impl() local
1523 …ideep::scale_t weights_scales(weight_scales.data_ptr<float>(), weight_scales.data_ptr<float>()+wei… in _quantized_convolution_onednn() local
1533 ideep::scale_t weights_scales(weight_scales.numel()); in _quantized_convolution_onednn() local
H A Dqlinear_dynamic.cpp555 const ideep::scale_t& weights_scales = w.get_scale(); in apply_dynamic_impl() local
H A Dqconv_prepack.cpp542 ideep::scale_t weights_scales(weight_scales.numel()); in _qconv_prepack_onednn() local
H A Dqlinear.cpp822 const ideep::scale_t& weights_scales = w.get_scale(); in apply_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model_test.cc647 const std::vector<float>& weights_scales = in TEST_F() local
759 const std::vector<float>& weights_scales = in TEST_P() local
863 const std::vector<float>& weights_scales = in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_model_test.cc452 const std::vector<float>& weights_scales = in TEST_P() local
557 const std::vector<float>& weights_scales = in TEST_P() local