From 2146f3682d6bb294b6c1f50365c048599c5509f5 Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Tue, 1 Dec 2020 20:21:26 +0000 Subject: [PATCH 1/2] [TFLite] Bugfix - ensure pad calcalution to be in int32 --- python/tvm/relay/frontend/tflite.py | 2 +- src/printer/relay_text_printer.cc | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py index 3572d35c6e3b..ff987f802470 100644 --- a/python/tvm/relay/frontend/tflite.py +++ b/python/tvm/relay/frontend/tflite.py @@ -3210,7 +3210,7 @@ def get_pad_value(data, kernel, stride): """ out = int(math.ceil(float(data) / float(stride))) - pad = max(0, (out - 1) * stride + kernel - data) + pad = max(0, (out - 1) * int(stride) + int(kernel) - int(data)) pad_before = pad // 2 pad_after = pad - pad_before return pad_before, pad_after diff --git a/src/printer/relay_text_printer.cc b/src/printer/relay_text_printer.cc index da4f8cadfb3d..e1c85abcce96 100644 --- a/src/printer/relay_text_printer.cc +++ b/src/printer/relay_text_printer.cc @@ -305,6 +305,8 @@ Doc RelayTextPrinter::ScalarLiteral(DataType dtype, const T& value) { std::ostringstream os; if (dtype == DataType::Int(32)) { os << value; + } else if (dtype == DataType::Int(64)) { + os << value << "i64"; } else if (dtype == DataType::Float(32)) { os << value << 'f'; } else if (dtype == DataType::Float(64)) { From 676b8791e8d470b8b02b6c2c14df9a670a6dd28f Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Wed, 2 Dec 2020 03:36:41 +0000 Subject: [PATCH 2/2] Empty