/aosp_15_r20/external/pytorch/test/quantization/eager/ |
H A D | test_numeric_suite_eager.py | 102 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 D | test_bias_correction_eager.py | 30 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 D | calibrator_test.py | 37 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 D | test_adaround_eager.py | 96 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 D | apot_fx_graph_mode_ptq.py | 15 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 D | apot_fx_graph_mode_qat.py | 16 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 D | debugger.py | 165 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 D | debugger_test.py | 141 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 D | operator_test.cc | 883 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.py | 404 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.py | 97 float_model, argument 119 float_model, quantized_model, _supported_modules, MeanShadowLogger
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/debugging/ |
H A D | README.md | 156 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 D | BUILD | 11 "@tflite_ovic_testdata//:float_model.lite",
|
/aosp_15_r20/external/pytorch/test/quantization/fx/ |
H A D | test_equalize_fx.py | 849 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 D | README.md | 111 "@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 D | quantize_weights_test.cc | 163 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 D | README.md | 4 float_model QConfigMapping BackendConfig 34 float_model:::nofs --> prepare_fx:::sub
|
/aosp_15_r20/external/pytorch/test/quantization/jit/ |
H A D | test_quantize_jit.py | 3031 float_model = torch.jit.script(model) 3032 prepared_model = prepare_jit(float_model, qconfig)
|
/aosp_15_r20/external/pytorch/docs/source/ |
H A D | quantization.rst | 526 float_model = M().eval()
|