From d6dbc7237fb0a0665911d7efa82fa60231a7bcb2 Mon Sep 17 00:00:00 2001
From: Pooya Davoodi <pdavoodi@nvidia.com>
Date: Tue, 19 Nov 2019 11:03:55 -0800
Subject: [PATCH] Fix function signatures

---
 tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc | 6 +++---
 tensorflow/compiler/tf2tensorrt/convert/convert_nodes.h  | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc
index ea608568db93cf..5e7468f48a7aa1 100644
--- a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc
+++ b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc
@@ -1221,19 +1221,19 @@ StatusOr<std::unique_ptr<Converter>> Converter::Create(
     nvinfer1::ILogger* trt_logger) {
   std::unique_ptr<Converter> converter = absl::WrapUnique(
       new Converter(precision_mode, use_calibration, trt_logger));
-  TF_RETURN_IF_ERROR(converter->Init());
+  TF_RETURN_IF_ERROR(converter->Init(trt_logger));
   return converter;
 }
 
 Converter::Converter(TrtPrecisionMode precision_mode, bool use_calibration,
                      nvinfer1::ILogger* trt_logger)
-      precision_mode_(precision_mode),
+    : precision_mode_(precision_mode),
       use_calibration_(use_calibration) {
   InitializeTrtPlugins(trt_logger);
   this->RegisterOpConverters();
 }
 
-Status Converter::Init() {
+Status Converter::Init(nvinfer1::ILogger* trt_logger) {
   VLOG(1) << "Creating TensorRT builder";
   trt_builder_.reset(nvinfer1::createInferBuilder(*trt_logger));
 
diff --git a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.h b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.h
index d4c24ef41d9382..7b2f25d7dc160a 100644
--- a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.h
+++ b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.h
@@ -446,7 +446,7 @@ class Converter {
   };
 
   static StatusOr<std::unique_ptr<Converter>> Create(
-      nvinfer1::IBuilder* trt_builder, TrtPrecisionMode precision_mode,
+      TrtPrecisionMode precision_mode,
       bool use_calibration, nvinfer1::ILogger* trt_logger);
 
   //////////////////////////////////////////////////////////////////////////////
@@ -529,10 +529,10 @@ class Converter {
                                          const nvinfer1::Dims& dims);
 
  private:
-  Converter(nvinfer1::IBuilder* trt_builder, TrtPrecisionMode precision_mode,
+  Converter(TrtPrecisionMode precision_mode,
             bool use_calibration, nvinfer1::ILogger* trt_logger);
 
-  Status Init();
+  Status Init(nvinfer1::ILogger* trt_logger);
 
   // Verify the provided batch_size is consistent with batch_size_ and update it
   // if necessary.