masahi opened a new pull request #4741: [Partitioning] Add test cases for fused 
ops with manual annotation
URL: https://github.com/apache/incubator-tvm/pull/4741
 
 
   * A custom annotator which detects conv + bias add + relu ops
   * An example of applying FoldScaleAxis and FoldConstant to layers of conv + 
bn + relu to get conv + bias add + relu ops which the annotator can detect 
(before partitioning)
   * Test cases on a simple network and mobilenet which verify that 
partitioning is working as expected.
   
   The result of partitioning mobilenet is dumped 
[here](https://gist.github.com/masahi/df0619d860b61b3cf9d7faad36e43fd7). 
   
   The actual execution of partitioned graphs is disabled due to the runtime 
issue described [in the 
forum](https://discuss.tvm.ai/t/external-codegen-how-the-runtime-determines-function-signatures-for-generated-functions/5455/).
 
   
   please review @zhiics @comaniac 
   

----------------------------------------------------------------
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:
us...@infra.apache.org


With regards,
Apache Git Services

Reply via email to