Home
last modified time | relevance | path

Searched refs:onnx_model (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_no_runtime.py72 onnx_model = onnx.load_from_string(f.getvalue())
73 onnx.checker.check_model(onnx_model)
74 return onnx_model
406 onnx_model = export_to_onnx(
412 self.assertAtenOp(onnx_model, "clamp", "Tensor")
430 onnx_model = export_to_onnx(
442 self.assertAtenOp(onnx_model, "clamp", "Tensor")
453 onnx_model = export_to_onnx(cast_fn, torch.zeros([1, 3, 32, 32]))
454 for n in onnx_model.graph.node:
500 onnx_model = io.BytesIO()
[all …]
H A Dtest_utility_funs.py724 onnx_model = onnx.load(io.BytesIO(f.getvalue()))
725 self.assertEqual(len(onnx_model.graph.initializer), 0)
787 onnx_model = onnx.load(io.BytesIO(f.getvalue()))
788 loop_output_value_info_proto = onnx_model.graph.output[0]
798 onnx_model = onnx.load(io.BytesIO(f.getvalue()))
799 loop_output_value_info_proto = onnx_model.graph.output[0]
915 onnx_model = onnx.load(io.BytesIO(f.getvalue()))
918 funcs = onnx_model.functions
929 nodes = onnx_model.graph.node
949 onnx_model = onnx.load(io.BytesIO(f.getvalue()))
[all …]
H A Derror_reproduction.py112 onnx_model: onnx.ModelProto,
131 onnx_model_text = onnx.printer.to_text(onnx_model)
H A Donnx_test_common.py342 onnx_model: Union[str, torch.onnx.ONNXProgram],
359 if isinstance(onnx_model, torch.onnx.ONNXProgram):
361 onnx_model.save(buffer)
364 ort_model = onnx_model
H A Dtest_operators.py366 onnx_model = onnx.load(model_proto_file.name)
367 onnx.checker.check_model(onnx_model)
371 onnx_model.graph.input[0].type.tensor_type.shape.dim[0].dim_param
375 onnx_model.graph.input[0].type.tensor_type.shape.dim[2].dim_param
379 onnx_model.graph.input[0].type.tensor_type.shape.dim[3].dim_param
385 onnx_model.graph.output[0].type.tensor_type.shape.dim[0].dim_param
389 onnx_model.graph.output[0].type.tensor_type.shape.dim[1].dim_param
H A Dtest_fx_to_onnx.py740 onnx_model = onnx.load(tmp_onnx_file.name)
742 (include_initializer and len(onnx_model.graph.initializer) > 0)
743 or (not include_initializer and len(onnx_model.graph.initializer) == 0),
H A Dtest_fx_to_onnx_with_onnxruntime.py789 onnx_model = onnx_program.model_proto
813 onnx_model,
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Dserialization.py121 onnx_model: onnx.ModelProto, # type: ignore[name-defined]
159 k.name: idx for idx, k in enumerate(onnx_model.graph.initializer)
161 onnx_input_names = {input.name for input in onnx_model.graph.input}
222 model_input_types = {k.name: k.type for k in onnx_model.graph.input}
235 onnx_model.graph.initializer.append(tensor_proto)
241 del onnx_model.graph.initializer[idx]
244 onnx.save(onnx_model, os.path.join(basepath, model_location)) # type: ignore[attr-defined]
/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A Donnxruntime.py971 onnx_model = exported.to_model_proto(
981 onnx_model = optimizer.optimize(onnx_model)
982 onnx_model = ort_rewriter.rewrite(onnx_model)
994 transform(onnx_model)
996 onnx_model_bytes = onnx_model.SerializeToString()
1025 input_names = tuple(input.name for input in onnx_model.graph.input)
1026 output_names = tuple(output.name for output in onnx_model.graph.output)
1036 input_value_infos = tuple(input for input in onnx_model.graph.input)
1037 output_value_infos = tuple(output for output in onnx_model.graph.output)
H A D_exporter_legacy.py562 onnx_model = os.path.join(tmpdir_path, "model.onnx")
568 onnx_model,
572 onnx_model = self.model_proto.SerializeToString() # type: ignore[assignment]
583 ort_session = onnxruntime.InferenceSession(onnx_model, providers=providers)
1006 onnx_model = onnxscript_graph.to_model_proto(
1013 onnx_model = optimizer.optimize(onnx_model)
1027 onnx_model,
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dcommon.py902 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)
[all …]
/aosp_15_r20/external/pytorch/test/onnx/dynamo/
H A Dtest_dynamo_with_onnxruntime_backend.py766 def record_onnx_model_transform(onnx_model): argument
768 recorded_models.append(onnx_model)
800 def record_onnx_model_transform(onnx_model): argument
802 recorded_models.append(onnx_model)
826 def replace_relu_with_sigmoid(onnx_model): argument
827 for node in onnx_model.graph.node:
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py4128 def assertAtenOp(self, onnx_model, operator, overload_name=""): argument
4129 all_aten_nodes = [p for p in onnx_model.graph.node