ccjoechou commented on pull request #9235: URL: https://github.com/apache/tvm/pull/9235#issuecomment-946302949
@comaniac, @vinx13, @masahi: I have pushed changes based on feedback. Latest changes are now utilizing a new out_layout field of pooling op (similar to what conv op has) so that, when out_layout is specified by user for pooling, pooling op will use and stick to the specified out_layout format. Otherwise, the original infer layout will be used as originally. 4 new test cases have been added to the test_pass_convert_op_layout.py test suite to illustrate how to specify layout in pooling. -- 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]
