Fix function signatures

frreiss · Nov 19, 2019 · d6dbc72 · d6dbc72
1 parent 9d5cbf4
commit d6dbc72
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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.