Home
last modified time | relevance | path

Searched refs:onnx_file (Results 1 – 10 of 10) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python-pytest/onnx/
H A Dtest_onnxruntime_nlp.py64 onnx_file = "%s.onnx" % prefix
118 onnx_file = "%s.onnx" % prefix
169 onnx_file = "%s.onnx" % prefix
219 onnx_file = "%s.onnx" % prefix
276 onnx_file = "%s.onnx" % prefix
340 onnx_file = "%s.onnx" % prefix
407 onnx_file = "%s.onnx" % prefix
472 onnx_file = "%s.onnx" % prefix
477 onnx_file = "%s.onnx" % prefix
481 def onnx_runtime_predict(onnx_file, onnx_inputs): argument
[all …]
H A Dtest_onnxruntime_cv.py52 onnx_file = self.modelpath + ".onnx"
55 return onnx_file
58 onnx_file = self.modelpath + ".onnx"
62 return onnx_file
65 onnx_file = self.modelpath + ".onnx"
86 return onnx_file
207 onnx_file = M.export_onnx()
325 onnx_file = M.export_onnx()
419 onnx_file = M.export_onnx()
483 onnx_file = M.export_onnx()
[all …]
H A Dtest_models.py136 onnx_file = os.path.join(dir_path, new_model_name)
139 … converted_model_path = onnx_mxnet.export_model(sym, params, [input_shape], np.float32, onnx_file)
H A Dtest_operators.py60 onnx_file = '{}/{}.onnx'.format(tmp_path, model_name)
62 [inp.dtype for inp in inputs], onnx_file)
63 return onnx_file
65 def onnx_rt(onnx_file, inputs): argument
66 sess = rt.InferenceSession(onnx_file)
77 onnx_file = export_to_onnx(model, model_name, inputs)
78 pred_onx = onnx_rt(onnx_file, inputs)
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python-pytest/onnx/
H A Dtest_onnxruntime_nlp.py64 onnx_file = "%s.onnx" % prefix
118 onnx_file = "%s.onnx" % prefix
169 onnx_file = "%s.onnx" % prefix
219 onnx_file = "%s.onnx" % prefix
276 onnx_file = "%s.onnx" % prefix
340 onnx_file = "%s.onnx" % prefix
407 onnx_file = "%s.onnx" % prefix
472 onnx_file = "%s.onnx" % prefix
477 onnx_file = "%s.onnx" % prefix
481 def onnx_runtime_predict(onnx_file, onnx_inputs): argument
[all …]
H A Dtest_onnxruntime_cv.py52 onnx_file = self.modelpath + ".onnx"
55 return onnx_file
58 onnx_file = self.modelpath + ".onnx"
62 return onnx_file
65 onnx_file = self.modelpath + ".onnx"
86 return onnx_file
207 onnx_file = M.export_onnx()
325 onnx_file = M.export_onnx()
419 onnx_file = M.export_onnx()
483 onnx_file = M.export_onnx()
[all …]
H A Dtest_models.py136 onnx_file = os.path.join(dir_path, new_model_name)
139 … converted_model_path = onnx_mxnet.export_model(sym, params, [input_shape], np.float32, onnx_file)
H A Dtest_operators.py60 onnx_file = '{}/{}.onnx'.format(tmp_path, model_name)
62 [inp.dtype for inp in inputs], onnx_file)
63 return onnx_file
65 def onnx_rt(onnx_file, inputs): argument
66 sess = rt.InferenceSession(onnx_file)
77 onnx_file = export_to_onnx(model, model_name, inputs)
78 pred_onx = onnx_rt(onnx_file, inputs)
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/onnx/
H A Dcv_model_inference.py63 onnx_file = prefix + '.onnx' variable
69 mx.onnx.export_model(mx_sym, mx_params, in_shapes, in_types, onnx_file)
87 session = onnxruntime.InferenceSession(onnx_file, ses_opt)
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/onnx/
H A Dcv_model_inference.py63 onnx_file = prefix + '.onnx' variable
69 mx.onnx.export_model(mx_sym, mx_params, in_shapes, in_types, onnx_file)
87 session = onnxruntime.InferenceSession(onnx_file, ses_opt)