Lines Matching refs:onnx_model

902     onnx_model: OnnxModel,
924     def create_onnx_input_binded_fn(onnx_model: OnnxModel, pt_inputs, example_outputs):
926         iobinding, outputs = onnx_model.create_iobinding(pt_inputs, example_outputs)
929             onnx_model.run_with_iobinding(iobinding, outputs)
935     def create_onnx_fn(onnx_model: OnnxModel, pt_inputs):
939         onnx_inputs = onnx_model.adapt_pt_inputs_to_onnx(pt_inputs)
942             return onnx_model.run_with_onnx_inputs(onnx_inputs)
946     def timed_onnx(model, onnx_model: OnnxModel, inputs):
947         if current_device == "cpu" or onnx_model.is_cpu():
948             onnxrt_model_iter_fn = create_onnx_fn(onnx_model, inputs)
952                 onnx_model, inputs, expected_output
979         timed_onnx(model, onnx_model, inputs)
1005         timings[rep, 1], actual_output = timed_onnx(model, onnx_model, inputs)
1822     onnx_model: Optional[OnnxModel] = None  variable in OnnxContext
1855             if context.onnx_model is None:
1856                 context.onnx_model = onnx_model_cls(
1863             onnx_model = context.onnx_model
1870                     outputs = onnx_model.run_and_serialize_inputs_outputs(inputs)
1873                     outputs = onnx_model.run(inputs)
1882             if context.onnx_model is not None:
2878                     experiment, optimized_model_iter_fn.context.onnx_model