Home
last modified time | relevance | path

Searched refs:float_model (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_numeric_suite_eager.py102 def compare_and_validate_results(float_model, q_model): argument
104 float_model.state_dict(), q_model.state_dict()
124 def compare_and_validate_results(float_model, q_model): argument
126 float_model.state_dict(), q_model.state_dict()
146 def compare_and_validate_results(float_model, q_model): argument
148 float_model.state_dict(), q_model.state_dict()
170 def compare_and_validate_results(float_model, q_model): argument
172 float_model.state_dict(), q_model.state_dict()
194 def compare_and_validate_results(float_model, q_model, module_swap_list, data): argument
195 ob_dict = compare_model_stub(float_model, q_model, module_swap_list, data)
[all …]
H A Dtest_bias_correction_eager.py30 def correct_artificial_bias_quantize(self, float_model, img_data): argument
34 artificial_model = copy.deepcopy(float_model)
49 …bias_correction(float_model, artificial_model, img_data, target_modules=_supported_modules_quantiz…
60 submodule = get_module(float_model, name)
81 float_model = QuantWrapper(LinearChain())
84 self.correct_artificial_bias_quantize(float_model, img_data)
100 float_model = QuantWrapper(ConvChain())
103 self.correct_artificial_bias_quantize(float_model, img_data)
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/optimize/
H A Dcalibrator_test.py37 float_model = open(model_path, 'rb').read()
38 quantizer = _calibrator.Calibrator(float_model)
60 float_model = open(model_path, 'rb').read()
61 quantizer = _calibrator.Calibrator(float_model)
78 float_model = open(model_path, 'rb').read()
79 quantizer = _calibrator.Calibrator(float_model)
116 float_model = open(model_path, 'rb').read()
117 quantizer = _calibrator.Calibrator(float_model)
132 float_model = b'\0' * 100
134 _calibrator.Calibrator(float_model)
[all …]
/aosp_15_r20/external/pytorch/test/quantization/core/experimental/
H A Dtest_adaround_eager.py96 float_model = LinearChain()
99 float_model, img_data, self.get_feed_forward_wrapper()
101 fq_model = self.get_fake_quant(float_model)
106 float_out = float_model(rand_input)
125 float_model = ConvChain()
127 adarounded_model = self.run_adaround(float_model, img_data)
128 fq_model = self.get_fake_quant(float_model)
133 float_out = float_model(rand_input)
H A Dapot_fx_graph_mode_ptq.py15 float_model = resnet18(pretrained=True) variable
16 float_model.eval()
20 model_to_quantize = copy.deepcopy(float_model)
49 full_precision_model = float_model
59 …prepared_model = prepare_qat_fx(copy.deepcopy(float_model), qconfig_dict) # fuse modules and inse…
H A Dapot_fx_graph_mode_qat.py16 float_model = resnet18(pretrained=True) variable
17 float_model.eval()
21 model_to_quantize = copy.deepcopy(float_model)
30 …prepared_model = prepare_fx(copy.deepcopy(float_model), qconfig_dict) # fuse modules and insert o…
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/debugging/python/
H A Ddebugger.py165 self.float_model = None
171 self.float_model = self.converter.convert()
181 float_model=self.float_model)
206 float_model=self.float_model)
279 float_model: Optional[bytes] = None) -> None:
302 if float_model is not None:
304 model_content=float_model)
H A Ddebugger_test.py141 cls.float_model = _convert_model(cls.tf_model_root, cls.tf_model)
241 float_model_content=QuantizationDebuggerTest.float_model,
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Doperator_test.cc883 Model float_model; in TEST_F() local
884 Array& float_array = float_model.GetOrCreateArray(op.inputs[0]); in TEST_F()
887 OperatorSignature float_signature = {.op = &op, .model = &float_model}; in TEST_F()
944 Model float_model; in TEST_F() local
945 Array& float_array = float_model.GetOrCreateArray(op.inputs[0]); in TEST_F()
948 OperatorSignature float_signature = {.op = &op, .model = &float_model}; in TEST_F()
1236 Model float_model; in TEST_F() local
1237 Array& input_float_array = float_model.GetOrCreateArray(dequant_op.inputs[0]); in TEST_F()
1240 .model = &float_model}; in TEST_F()
1271 Model float_model; in TEST_F() local
[all …]
/aosp_15_r20/external/pytorch/torch/ao/ns/
H A D_numeric_suite.py404 float_model: nn.Module,
442 prepare_model_with_stubs(float_model, q_model, module_swap_list, logger_cls)
517 float_model: nn.Module,
559 prepare_model_outputs(float_model, q_model, logger_cls, allow_list)
560 float_model(*data)
562 act_compare_dict = get_matching_activations(float_model, q_model)
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A D_correct_bias.py97 float_model, argument
119 float_model, quantized_model, _supported_modules, MeanShadowLogger
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/debugging/
H A DREADME.md156 float_model = converter.convert() # converted without any optimizations.
160 float_model_content=float_model, # can pass `float_model_path` instead.
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/ovic/src/testdata/
H A DBUILD11 "@tflite_ovic_testdata//:float_model.lite",
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_equalize_fx.py849 float_model = M().eval()
860 copy.deepcopy(float_model),
876 copy.deepcopy(float_model),
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/ovic/
H A DREADME.md111 "@tflite_ovic_testdata//:float_model.lite",
226 float_model.lite | 97 | 113 | 37
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_weights_test.cc163 std::unique_ptr<ModelT> float_model = in FindMatchingExpectedTensor() local
185 float_model->buffers[float_tensor->buffer()].get()->data; in FindMatchingExpectedTensor()
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A DREADME.md4 float_model QConfigMapping BackendConfig
34 float_model:::nofs --> prepare_fx:::sub
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py3031 float_model = torch.jit.script(model)
3032 prepared_model = prepare_jit(float_model, qconfig)
/aosp_15_r20/external/pytorch/docs/source/
H A Dquantization.rst526 float_model = M().eval()