Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[MXNET-533] MXNet-ONNX export #11213

Merged
merged 83 commits into from
Jun 25, 2018
Merged

[MXNET-533] MXNet-ONNX export #11213

merged 83 commits into from
Jun 25, 2018

Commits on Jun 14, 2018

  1. Resolve conflicts

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    8d083f8 View commit details
    Browse the repository at this point in the history

  2. Export module Test Framework

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    4c04e19 View commit details
    Browse the repository at this point in the history

  3. refactoring export to work with pretrained models

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    e66ef3b View commit details
    Browse the repository at this point in the history

  4. comments added

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    aa4e7cb View commit details
    Browse the repository at this point in the history

  5. 1. Refactored export module. 

    2. Refactored test framework to support ONNX backened tests.
2. Added Operator support:
   - Convolution2D
   - BatchNorm
   - Add
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    f9e8cbf View commit details
    Browse the repository at this point in the history

  6. Added Arithmetic operators: 

    - Add, Sub, Mul, Div, Sum
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    de3632d View commit details
    Browse the repository at this point in the history

  7. Added operator support: 

    - sigmoid, relu, pad( constant, edge, reflect), tanh
- enabled corresponding ONNX backend tests.
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    46931d3 View commit details
    Browse the repository at this point in the history

  8. Enabled ONNX tests: test_conv, test_basic_conv 

    Added Operators :
Ceil, Floor
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    ca74b02 View commit details
    Browse the repository at this point in the history

  9. Added support for: 

    MaxPool, AvgPool, GlobalMaxPool, GlobalAvgPool, matmul
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    f653baf View commit details
    Browse the repository at this point in the history

  10. adding more operators

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    2cf7be0 View commit details
    Browse the repository at this point in the history

  11. Added Operator support: 

    ArgMax, ArgMin, maximum, minimum
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    8ecf9dd View commit details
    Browse the repository at this point in the history

  12. Enabled more BASIC_MODEL tests

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    af73173 View commit details
    Browse the repository at this point in the history

  13. Added power operator tests

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    e851cf3 View commit details
    Browse the repository at this point in the history

  14. Added support for reshape. ONNX only supports 0, -1 special values. A… 

    …dded only for these.

Fixed logic error with convert_string_to_list()
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    54edda1 View commit details
    Browse the repository at this point in the history

  15. some tests enabled

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    bb0b76f View commit details
    Browse the repository at this point in the history

  16. enabling squeezenet

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    917894b View commit details
    Browse the repository at this point in the history

  17. LRN Op support

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    b7e671b View commit details
    Browse the repository at this point in the history

  18. mul_scalar modified to take scalar input

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    e44960d View commit details
    Browse the repository at this point in the history

  19. cleaning some code

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    b918aa5 View commit details
    Browse the repository at this point in the history

  20. Resolving conlicts on rebase

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    7b1e0ab View commit details
    Browse the repository at this point in the history

  21. Resolving rebase conflicts

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    42c695c View commit details
    Browse the repository at this point in the history

  22. id mapping updated for all operators

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    03226b3 View commit details
    Browse the repository at this point in the history

  23. save onnx models added, some code cleanup

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    43dad9b View commit details
    Browse the repository at this point in the history

  24. enabled more tests

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    7cebe80 View commit details
    Browse the repository at this point in the history

  25. conv pad calc fixed

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    5ce301b View commit details
    Browse the repository at this point in the history

  26. reshape op fix

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    f8c0afc View commit details
    Browse the repository at this point in the history

  27. valueinfoproto fix, googlenet test added

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    cfc081b View commit details
    Browse the repository at this point in the history

  28. Added support for elu, leakyRelu, prelu

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    f070c22 View commit details
    Browse the repository at this point in the history

  29. Cleanup 

    - Removed run_node, not needed anymore.
- Used correct get_metadata api
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    bb51c7f View commit details
    Browse the repository at this point in the history

  30. dilation added

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    ae2f2c2 View commit details
    Browse the repository at this point in the history

  31. Removed redundant code. 

    - run_node
- Using correct get_metadata_api
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    94d8614 View commit details
    Browse the repository at this point in the history

  32. some fixes to make export work with onx1.2.1

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    f8943b3 View commit details
    Browse the repository at this point in the history

  33. Lint fixes

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    1794554 View commit details
    Browse the repository at this point in the history

  34. lint fixes

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    628b1cd View commit details
    Browse the repository at this point in the history

  35. enabled more tests

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    0d503e6 View commit details
    Browse the repository at this point in the history

  36. mxnet_export_test file added

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    d6e73dc View commit details
    Browse the repository at this point in the history

  37. duplicate file deleted

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    26128a0 View commit details
    Browse the repository at this point in the history

  38. reduce ops added

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    2de3d5a View commit details
    Browse the repository at this point in the history

  39. some small fixes

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    7e56744 View commit details
    Browse the repository at this point in the history

  40. some lint fixes

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    dc3d8e7 View commit details
    Browse the repository at this point in the history

  41. docstring added

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    4765b49 View commit details
    Browse the repository at this point in the history

  42. Add tests for inception_v1 and inception_v2

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    25566e0 View commit details
    Browse the repository at this point in the history

  43. Add CI runs for export module

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    322b63a View commit details
    Browse the repository at this point in the history

  44. lint fixes, pooling attr fix

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    3de8a25 View commit details
    Browse the repository at this point in the history

  45. fix

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    f95f022 View commit details
    Browse the repository at this point in the history

  46. fix global_pool

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    e1f324b View commit details
    Browse the repository at this point in the history

  47. code cleanup

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    dc45f06 View commit details
    Browse the repository at this point in the history

  48. CI run fix

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    17b7d85 View commit details
    Browse the repository at this point in the history

  49. lint fix

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    33e5ab6 View commit details
    Browse the repository at this point in the history

  50. some code cleanup

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    6de4403 View commit details
    Browse the repository at this point in the history

  51. Added required license comments

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    21c96c8 View commit details
    Browse the repository at this point in the history

  52. pad in pooling added

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    4aa962f View commit details
    Browse the repository at this point in the history

  53. slicechannel notimplementederror raised

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    31ba992 View commit details
    Browse the repository at this point in the history

  54. Lint fixes

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    6d150b4 View commit details
    Browse the repository at this point in the history

  55. lint fix

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    db70ce3 View commit details
    Browse the repository at this point in the history

  56. lint fix

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    69a3aa8 View commit details
    Browse the repository at this point in the history

  57. lint fix

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    4ecd786 View commit details
    Browse the repository at this point in the history

  58. lint fix

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    ab0bdc6 View commit details
    Browse the repository at this point in the history

  59. Correct license statement

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    14c1362 View commit details
    Browse the repository at this point in the history

  60. Adding onnx a runtime dependency

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    acd3c79 View commit details
    Browse the repository at this point in the history

  61. Fix import module error for string_types

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    ba2d6ba View commit details
    Browse the repository at this point in the history

  62. Making ONNX runtime dependency

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    0829e4f View commit details
    Browse the repository at this point in the history

  63. fixing some comments

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    8988165 View commit details
    Browse the repository at this point in the history

  64. addressing some comments

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    98d5dae View commit details
    Browse the repository at this point in the history

  65. params rename

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    1ff47e1 View commit details
    Browse the repository at this point in the history

  66. lint fixes

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    112c930 View commit details
    Browse the repository at this point in the history

  67. fixes

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    50ade3e View commit details
    Browse the repository at this point in the history

  68. spatial disabled, path fixed

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    7d334e1 View commit details
    Browse the repository at this point in the history

  69. fixing some comments

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    ff123ee View commit details
    Browse the repository at this point in the history

  70. Added support for remaining act_type(softsign, sigmoid, softrelu) in … 

    …Activation operator
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    7e626c7 View commit details
    Browse the repository at this point in the history

  71. changing import

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    4199ae8 View commit details
    Browse the repository at this point in the history

  72. adding some comments

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    3d5f8be View commit details
    Browse the repository at this point in the history

  73. Add squeeze op

    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    7412820 View commit details
    Browse the repository at this point in the history

  74. Refactored logic to handle extra node(output label node) for saved mx… 

    …net model

Added comments
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    a303eb1 View commit details
    Browse the repository at this point in the history

  75. identity operator added

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    777144a View commit details
    Browse the repository at this point in the history

  76. scalar ops added

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    4801fb3 View commit details
    Browse the repository at this point in the history

  77. minor fix for squeeze operator. 

    Also, added error handling
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    c668689 View commit details
    Browse the repository at this point in the history

  78. Renamed onnx support folders to mark it public folders 

    Changed underline files public or private as per usage

Resolved conflicts with the latest
    @rajanksin
    rajanksin committed Jun 14, 2018
    Configuration menu
    Copy the full SHA
    20bb2fd View commit details
    Browse the repository at this point in the history

Commits on Jun 15, 2018

  1. Added support L2Normalization op 

    Added some error checking
    @rajanksin
    rajanksin committed Jun 15, 2018
    Configuration menu
    Copy the full SHA
    f4902e1 View commit details
    Browse the repository at this point in the history

  2. added comments and warning

    @Roshrini
    Roshrini committed Jun 15, 2018
    Configuration menu
    Copy the full SHA
    84c9320 View commit details
    Browse the repository at this point in the history

Commits on Jun 16, 2018

  1. added comments and warning

    @Roshrini @rajanksin
    Roshrini authored and rajanksin committed Jun 16, 2018
    Configuration menu
    Copy the full SHA
    f36230d View commit details
    Browse the repository at this point in the history

Commits on Jun 18, 2018

  1. doc API ref added

    @Roshrini
    Roshrini committed Jun 18, 2018
    Configuration menu
    Copy the full SHA
    6327847 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'onnx_export' of https://github.com/Roshrini/mxnet into … 

    …onnx_export
    @Roshrini
    Roshrini committed Jun 18, 2018
    Configuration menu
    Copy the full SHA
    83d798c View commit details
    Browse the repository at this point in the history