tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <>
Subject [GitHub] [incubator-tvm] maheshambule opened a new pull request #5052: Relay to ONNX converter
Date Thu, 12 Mar 2020 13:44:48 GMT
maheshambule opened a new pull request #5052: Relay to ONNX converter
   This PR adds support to convert In-memory Relay Module to ONNX protobuf. Currently, only
a subset of operators is supported. The conversion can be done on optimized Relay modules
as well except for fused op passes.
   Operators supported:
   - [x] reshape 
   - [x] conv2d 
   - [x] add
   - [x] relu
   - [x] transpose
   - [x] dense
   - [x] max_pool2d
   - [x] batch_flatten
   - [x] multiply 
   - [x] bias_add 
   - [x] batch_norm
   - [x] global_avg_pool2d
   - [x] concatenate
   - [x] dropout
   - [x] avg_pool2d
   - [x] divide
   - [x] mean
   - [x] pad
   - [x] softmax
   - [x] squeeze 
   - [x] strided_slice
   - [x] greater
   - [x] less
   - [x] equal
   - [x] zeros_like
   - [x] ones_like
   - [x] subtract
   Models tested:
   - [x] ONNX zoo resent and squeezenet
   - [x] TF-Slim Resnet
   - [x] MXNet Resnet
   - Increase the operator coverage to add support for all the Relay operators
   - Add support for optimization pass - AlterOpLayout
   - Add support for Relay constructs such as functions, ADTs etc
   - Add support for different versions of ONNX opsets for different operators

This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:

With regards,
Apache Git Services

View raw message