Searched refs:onnx_model (Results 1 – 13 of 13) sorted by relevance
/aosp_15_r20/external/pytorch/test/onnx/ |
H A D | test_pytorch_onnx_no_runtime.py | 72 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 D | test_utility_funs.py | 724 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 D | error_reproduction.py | 112 onnx_model: onnx.ModelProto, 131 onnx_model_text = onnx.printer.to_text(onnx_model)
|
H A D | onnx_test_common.py | 342 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 D | test_operators.py | 366 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 D | test_fx_to_onnx.py | 740 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 D | test_fx_to_onnx_with_onnxruntime.py | 789 onnx_model = onnx_program.model_proto 813 onnx_model,
|
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
H A D | serialization.py | 121 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 D | onnxruntime.py | 971 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.py | 562 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 D | common.py | 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) [all …]
|
/aosp_15_r20/external/pytorch/test/onnx/dynamo/ |
H A D | test_dynamo_with_onnxruntime_backend.py | 766 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 D | common_utils.py | 4128 def assertAtenOp(self, onnx_model, operator, overload_name=""): argument 4129 all_aten_nodes = [p for p in onnx_model.graph.node
|