tvm icon indicating copy to clipboard operation
tvm copied to clipboard

[frontend][pytorch] Add a new test case for torch aten::fill_ operator implementation

Open oviazlo opened this issue 2 years ago • 2 comments

Add a new test case for torch aten::fill_ operator implementation.

This new test is testing a case of usage of a sequence of operators aten::div() and aten::fill_() that was causing a failure during the deployment of Mask R-CNN model (torchvision 0.13.0 version of the model).

Issue: #12844

oviazlo avatar Sep 21 '22 10:09 oviazlo

You need to include my change in frontend/pytorch.py from https://github.com/apache/tvm/issues/12844 as well.

masahi avatar Sep 21 '22 19:09 masahi

Thanks for the comment. I have included yours changes as well to the PR.

oviazlo avatar Sep 21 '22 19:09 oviazlo