changes to cast the convolution input layers to float

Author: apbose

core/conversion/converters/impl/conv_deconv.cpp

@@ -13,12 +13,12 @@ namespace {
 bool add_conv_deconv(ConversionCtx* ctx, const torch::jit::Node* n, args& args) {
   // Input to conv/deconv
   auto in = args[0].ITensor();
-  // if (in->getType() == nvinfer1::DataType::kINT32) {
-  //   LOG_DEBUG(
-  //     "Found type  " << in->getType() << " in aten::convolution, casting to "
-  //     << nvinfer1::DataType::kFLOAT << " for compatibility.");
-  //   in = castITensor(ctx, in, nvinfer1::DataType::kFLOAT);
-  // }
+  if (in->getType() == nvinfer1::DataType::kINT32) {
+    LOG_DEBUG(
+      "Found type  " << in->getType() << " in aten::convolution, casting to "
+      << nvinfer1::DataType::kFLOAT << " for compatibility.");
+    in = castITensor(ctx, in, nvinfer1::DataType::kFLOAT);
+  }
   // Conv /deconv parameters
   auto stride = util::toDims(args[3].unwrapToIntList());
   auto padding = util::toDims(args[4].unwrapToIntList());