From 55aefc22a6a24ad21953b887f171eac75a33b825 Mon Sep 17 00:00:00 2001 From: Samuel Date: Sun, 31 May 2020 02:55:36 +0530 Subject: [PATCH] [PYTORCH]floor_divide support for squeezenet (#5702) https://github.com/apache/incubator-tvm/issues/5133#issuecomment-636330705 --- python/tvm/relay/frontend/pytorch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py index f68affd..f2e24a1 100644 --- a/python/tvm/relay/frontend/pytorch.py +++ b/python/tvm/relay/frontend/pytorch.py @@ -1760,6 +1760,7 @@ def _get_convert_map(prelude): "aten::arange" : _arange(), "aten::div" : _elemwise("divide"), "aten::div_" : _elemwise("divide"), + "aten::floor_divide" : _elemwise("floor_divide"), "aten::addcdiv" : _addcdiv(), "aten::addcmul" : _addcmul(), "aten::ones" : _ones(), -- 2.7.4