Home
last modified time | relevance | path

Searched refs:calibration_inputs (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/testing/
Dtoco_convert.py126 calibration_inputs = []
130 calibration_inputs.append(
133 return calibration_inputs
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Dmodel_handler.py424 calibration_inputs: Optional[Mapping[str, np.ndarray]] = None,
471 calibration_inputs: Optional[Mapping[str, np.ndarray]] = None,
479 if not calibration_inputs:
486 feed_dict_fn=lambda: calibration_inputs)
519 calibration_inputs: Optional[Sequence[
528 if not calibration_inputs:
534 yield calibration_inputs
600 def convert(self, calibration_inputs=None, num_runs=1) -> None: argument
609 trt_model.convert(calibration_inputs, num_runs)