From b6f0efd62cf62b80e1ee66873f5f3d104811f6df Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Fri, 20 Mar 2020 01:12:48 +0900 Subject: [PATCH] fix names --- python/tvm/relay/frontend/pytorch.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py index fe9acadfdf44..83436f25ca85 100644 --- a/python/tvm/relay/frontend/pytorch.py +++ b/python/tvm/relay/frontend/pytorch.py @@ -163,7 +163,7 @@ def _impl(inputs, input_types): return _op.nn.relu(data) return _impl -def _adaptive_avg_2d(): +def _adaptive_avg_pool_2d(): def _impl(inputs, input_types): data = inputs[0] output_size = _infer_shape(inputs[1]) @@ -178,7 +178,7 @@ def func(x): return _impl -def _adaptive_max_2d(): +def _adaptive_max_pool_2d(): def _impl(inputs, input_types): data = inputs[0] output_size = _infer_shape(inputs[1]) @@ -189,7 +189,7 @@ def _impl(inputs, input_types): output_size=output_size), None return _impl -def _adaptive_max_3d(): +def _adaptive_max_pool_3d(): def _impl(inputs, input_types): data = inputs[0] output_size = _infer_shape(inputs[1]) @@ -198,7 +198,7 @@ def _impl(inputs, input_types): return _impl -def _adaptive_avg_3d(): +def _adaptive_avg_pool_3d(): def _impl(inputs, input_types): data = inputs[0] output_size = _infer_shape(inputs[1]) @@ -859,8 +859,8 @@ def _wrap_const(c): "aten::select" : _select(), "aten::relu" : _relu(), "aten::relu_" : _relu(), - "aten::adaptive_avg_pool2d" : _adaptive_avg_2d(), - "aten::adaptive_max_pool2d" : _adaptive_max_2d(), + "aten::adaptive_avg_pool2d" : _adaptive_avg_pool_2d(), + "aten::adaptive_max_pool2d" : _adaptive_max_pool_2d(), "aten::max_pool2d" : _maxpool_2d(), "aten::max_pool2d_with_indices" : _maxpool_2d(), "aten::hardtanh" : _hardtanh(), @@ -910,8 +910,8 @@ def _wrap_const(c): "aten::Float" : _Float(), "aten::neg" : _neg(), "aten::tanh" : _tanh(), - "aten::adaptive_avg_pool3d" : _adaptive_avg_3d(), - "aten::adaptive_max_pool3d" : _adaptive_max_3d() + "aten::adaptive_avg_pool3d" : _adaptive_avg_pool_3d(), + "aten::adaptive_max_pool3d" : _adaptive_max_pool_3d() }