mbs-octoml commented on PR #11203:
URL: https://github.com/apache/tvm/pull/11203#issuecomment-1117360408
The background on the bits I fiddled with:
- Finish the transition from operator predicate based to pattern based for
the partition_for_tensorrt function. The current state in main is broken.
- Remove choice of operator vs predicate method since they are
indistinguishable from the outside.
- Though most operators can be translated looking only at the operator
call, nn.batch_norm can only be translated in the form of a sub-graph
`nn.batch_norm(...).0`. So switch the translation to follow the standard
composite style.
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
