tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [incubator-tvm] comaniac opened a new pull request #5288: [BYOC] Refine DNNL Codegen
Date Thu, 09 Apr 2020 06:16:07 GMT
comaniac opened a new pull request #5288: [BYOC] Refine DNNL Codegen
URL: https://github.com/apache/incubator-tvm/pull/5288
 
 
   With multiple output and merge compiler region support, we can now refine the DNNL codegen
to support ops with multiple outputs such as `batch_norm`. This PR:
   
   - Support operators with multiple outputs.
   - Fix the issue that the second fuse run in VM will go into external functions and decompose
`batch_norm`.
   - Turn on DNNL `batch_norm`.
   - Fix `TupleGeItem` in DNNL codegen.
   - Skip MobileNet test because of the constant node issue for now.
   
   cc @zhiics, @masahi.
   

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


With regards,
Apache Git Services

Mime
View raw message