Lunderberg opened a new pull request #8529:
URL: https://github.com/apache/tvm/pull/8529
The goal of this PR is to add float16 test cases to topi schedules used for
ResNet50. This PR is split up into independent commits, as described below,
for ease of review.
[Topi][Testing] Minor cleanup for python reference implementations
- Use input dtype for dilate/conv2d accumulate in python
impl. Previously, the python implementations of dilation and conv2d
would use numpy default dtype in some cases, rather than the input
data's dtype.
- Added fallback for datatypes not supported by scipy.signal.convolve2d
(e.g. float16).
- Refactored to avoid duplication, use common get_pad_tuple functionality.
[Topi][UnitTests] Added float16 tests to test_topi_dense.py
[Topi][UnitTests] Added float16 to test_topi_conv2d_nchw.py
[Topi][Float16] Added float16 tests for depthwise conv2d.
--
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]
