| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/ |
| H A D | quantize_model.py | 30 from tensorflow.compiler.mlir.quantization.tensorflow.python import representative_dataset as repr_…
207 representative_dataset: repr_dataset.RepresentativeDatasetOrMapping,
233 if isinstance(representative_dataset, collections.abc.Mapping):
234 if set(signature_keys) != set(representative_dataset.keys()):
313 representative_dataset: repr_dataset.RepresentativeDataset) -> None:
334 for sample in map(sample_validator, representative_dataset):
397 sess, signature_def=sig_def, representative_dataset=repr_ds)
406 representative_dataset: repr_dataset.RepresentativeDataset) -> None:
421 for sample in map(sample_validator, representative_dataset):
457 func=root.signatures[signature_key], representative_dataset=repr_ds)
[all …]
|
| H A D | BUILD | 141 ":representative_dataset",
190 ":representative_dataset",
227 name = "representative_dataset",
229 "representative_dataset.py",
244 ":representative_dataset",
|
| H A D | representative_dataset_test.py | 20 from tensorflow.compiler.mlir.quantization.tensorflow.python import representative_dataset as repr_…
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/ |
| H A D | quantize_model_test.py | 27 from tensorflow.compiler.mlir.quantization.tensorflow.python import representative_dataset as repr_…
165 representative_dataset=self._simple_model_data_gen())
404 representative_dataset=gen_data())
506 representative_dataset=data_gen())
613 representative_dataset=data_gen())
664 representative_dataset=data_gen())
705 representative_dataset=data_gen)
729 representative_dataset: repr_dataset.RepresentativeDataset = [{
737 representative_dataset=representative_dataset)
760 representative_dataset = [{
[all …]
|
| H A D | concurrency_test.py | 81 representative_dataset=data_gen())
|
| H A D | quantize_model_test_base.py | 22 from tensorflow.compiler.mlir.quantization.tensorflow.python import representative_dataset as repr_…
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/g3doc/performance/ |
| H A D | post_training_quantization.md | 66 `representative_dataset()` function below.
72 def representative_dataset():
84 def representative_dataset():
107 def representative_dataset():
119 def representative_dataset():
135 converter.representative_dataset = representative_dataset</b>
162 converter.representative_dataset = representative_dataset
215 converter.representative_dataset = representative_dataset
227 converter.representative_dataset = representative_dataset
|
| H A D | quantization_debugger.ipynb | 187 "def representative_dataset(dataset):\n",
206 " for data in representative_dataset(dataset)():\n",
233 "converter.representative_dataset = representative_dataset(ds)\n",
294 "converter.representative_dataset = representative_dataset(ds)\n",
298 " converter=converter, debug_dataset=representative_dataset(ds))"
559 " debug_dataset=representative_dataset(ds),\n",
601 " debug_dataset=representative_dataset(ds),\n",
686 " debug_dataset=representative_dataset(ds),\n",
788 "converter.representative_dataset = representative_dataset(ds)\n",
809 " debug_dataset=representative_dataset(ds))"
|
| H A D | post_training_integer_quant.ipynb | 294 "converter.representative_dataset = representative_data_gen\n",
370 "converter.representative_dataset = representative_data_gen\n",
|
| H A D | post_training_integer_quant_16x8.ipynb | 310 "converter.representative_dataset = representative_data_gen"
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/ |
| H A D | mlir_convert.py | 84 def representative_dataset(input_tensors): function
95 yield representative_dataset(input_tensors)
107 converter.representative_dataset = representative_dataset_gen
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
| H A D | lite.py | 236 representative_dataset, argument
252 self._representative_dataset = representative_dataset
520 self.representative_dataset = None
603 if not isinstance(self.representative_dataset, RepresentativeDataset):
604 self.representative_dataset = RepresentativeDataset(
605 self.representative_dataset)
614 self.representative_dataset.input_gen)
629 self.representative_dataset.input_gen,
769 self.optimizations, self.target_spec, self.representative_dataset,
1014 self.optimizations, self.target_spec, self.representative_dataset,
[all …]
|
| H A D | lite_v2_test.py | 301 quantized_converter.representative_dataset = calibration_gen
396 converter.representative_dataset = calibration_gen
437 quantized_converter.representative_dataset = calibration_gen
492 quantized_converter.representative_dataset = calibration_gen
520 converter.representative_dataset = calibration_gen
710 quantized_converter.representative_dataset = calibration_gen
826 quantized_converter.representative_dataset = calibration_gen
925 quantized_converter.representative_dataset = calib_gen
1029 quantized_converter.representative_dataset = calib_gen
1089 quantized_converter.representative_dataset = calibration_gen
[all …]
|
| H A D | lite_test.py | 346 quantized_converter.representative_dataset = calibration_gen
402 representative_dataset=True): argument
414 if representative_dataset:
415 quantized_converter.representative_dataset = calibration_gen
912 quantized_converter.representative_dataset = calibration_gen
958 quantized_converter.representative_dataset = calibration_gen
1005 quantized_converter.representative_dataset = calibration_gen
1172 quantized_converter.representative_dataset = calibration_gen
1307 converter.representative_dataset = calibration_gen
1995 converter.representative_dataset = representative_dataset_gen
|
| H A D | util_test.py | 285 converter.representative_dataset = representative_dataset_gen
|
| H A D | metrics_nonportable_test.py | 211 quantized_converter.representative_dataset = calibration_gen
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/ |
| H A D | mlir_convert.py | 76 def representative_dataset(input_tensors): function
88 yield representative_dataset(input_tensors)
100 converter.representative_dataset = representative_dataset_gen
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/g3doc/examples/jax_conversion/ |
| H A D | overview.ipynb | 336 …"We will provide a `representative_dataset` to do post-training quantiztion to optimize the model.…
347 "def representative_dataset():\n",
356 "converter.representative_dataset = representative_dataset\n",
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/debugging/ |
| H A D | README.md | 44 converter.representative_dataset = calibration_gen
62 # `debug_dataset` accpets the same type as `converter.representative_dataset`.
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/python/ |
| H A D | modify_model_interface_lib_test.py | 44 converter.representative_dataset = representative_dataset_gen
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/debugging/python/ |
| H A D | debugger_test.py | 70 converter.representative_dataset = calibration_gen
|
| H A D | debugger.py | 319 if not converter.representative_dataset:
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/metrics/ |
| H A D | metrics_nonportable_test.py | 211 quantized_converter.representative_dataset = calibration_gen
|