This is an automated email from the ASF dual-hosted git repository. skm pushed a commit to branch master in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push: new 100a4aa Addressed ONNX module documentation warnings and added notes for short-form representation (#13259) 100a4aa is described below commit 100a4aa6d42942df9bc191411553d72e11acf866 Author: vdantu <36211508+vda...@users.noreply.github.com> AuthorDate: Wed Nov 14 16:31:03 2018 -0800 Addressed ONNX module documentation warnings and added notes for short-form representation (#13259) --- python/mxnet/contrib/onnx/mx2onnx/export_model.py | 5 +++++ python/mxnet/contrib/onnx/onnx2mx/import_model.py | 9 +++++++++ python/mxnet/contrib/onnx/onnx2mx/import_to_gluon.py | 5 +++++ 3 files changed, 19 insertions(+) diff --git a/python/mxnet/contrib/onnx/mx2onnx/export_model.py b/python/mxnet/contrib/onnx/mx2onnx/export_model.py index e515805..f5e4c3b 100644 --- a/python/mxnet/contrib/onnx/mx2onnx/export_model.py +++ b/python/mxnet/contrib/onnx/mx2onnx/export_model.py @@ -58,6 +58,11 @@ def export_model(sym, params, input_shape, input_type=np.float32, ------- onnx_file_path : str Onnx file path + + Notes + ----- + This method is available when you ``import mxnet.contrib.onnx`` + """ try: diff --git a/python/mxnet/contrib/onnx/onnx2mx/import_model.py b/python/mxnet/contrib/onnx/onnx2mx/import_model.py index 0f1a7c4..0048c74 100644 --- a/python/mxnet/contrib/onnx/onnx2mx/import_model.py +++ b/python/mxnet/contrib/onnx/onnx2mx/import_model.py @@ -41,6 +41,11 @@ def import_model(model_file): aux_params : dict of ``str`` to :class:`~mxnet.ndarray.NDArray` Dict of converted parameters stored in ``mxnet.ndarray.NDArray`` format + + Notes + ----- + This method is available when you ``import mxnet.contrib.onnx`` + """ graph = GraphProto() @@ -58,6 +63,10 @@ def get_model_metadata(model_file): """ Returns the name and shape information of input and output tensors of the given ONNX model file. + Notes + ----- + This method is available when you ``import mxnet.contrib.onnx`` + Parameters ---------- model_file : str diff --git a/python/mxnet/contrib/onnx/onnx2mx/import_to_gluon.py b/python/mxnet/contrib/onnx/onnx2mx/import_to_gluon.py index 8e1b354..13ad5b9 100644 --- a/python/mxnet/contrib/onnx/onnx2mx/import_to_gluon.py +++ b/python/mxnet/contrib/onnx/onnx2mx/import_to_gluon.py @@ -36,6 +36,11 @@ def import_to_gluon(model_file, ctx): ------- sym_block : :class:`~mxnet.gluon.SymbolBlock` A SymbolBlock object representing the given model file. + + Notes + ----- + This method is available when you ``import mxnet.contrib.onnx`` + """ graph = GraphProto() try: