add Long dtype

Author: zewenli98

core/ir/Input.cpp

@@ -15,6 +15,14 @@ bool valid_dtype_format_combo(nvinfer1::DataType dtype, nvinfer1::TensorFormat f
         default:
           return false;
       }
+    case nvinfer1::DataType::kINT64: // Supports just Linear (NCHW)
+      switch (format) {
+        case nvinfer1::TensorFormat::kLINEAR:
+          return true;
+        case nvinfer1::TensorFormat::kHWC:
+        default:
+          return false;
+      }
     case nvinfer1::DataType::kINT32: // Supports just Linear (NCHW)
       switch (format) {
         case nvinfer1::TensorFormat::kLINEAR:
@@ -64,6 +72,8 @@ bool valid_input_dtype(nvinfer1::DataType dtype) {
       return true;
     case nvinfer1::DataType::kINT32:
       return true;
+    case nvinfer1::DataType::kINT64:
+      return true;
     default:
       return false;
   }