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)) {