diff options
| author |  Jie <jiej@nvidia.com> | 2018-04-16 17:47:00 -0700 |
|---|---|---|
| committer | Jie <jiej@nvidia.com> | 2018-04-17 13:49:47 -0700 |
| commit | df0ce53aee6f4e14b3f1c9e0e772a1f7bd1bb95a (patch) | |
| tree | a2f75f85834021fd7c8950e705dfefef54def0e8 /tensorflow/contrib/tensorrt/kernels | |
| parent | 9d1aa895adda8644ddbb55b5e1dbb0797ea6cbb0 (diff) | |
[PR comment addressed]
adding plugin test for registration
updating plugin API wrapper addressing comments in the PR
addressing coding style issues
removing commented code
Diffstat (limited to 'tensorflow/contrib/tensorrt/kernels')
| -rw-r--r-- | tensorflow/contrib/tensorrt/kernels/trt_engine_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensorrt/kernels/trt_engine_op.cc b/tensorflow/contrib/tensorrt/kernels/trt_engine_op.cc index 8881c48fe6..162301fb52 100644 --- a/tensorflow/contrib/tensorrt/kernels/trt_engine_op.cc +++ b/tensorflow/contrib/tensorrt/kernels/trt_engine_op.cc @@ -60,7 +60,7 @@ TRTEngineOp::TRTEngineOp(OpKernelConstruction* context) : OpKernel(context) { IRuntime* infer = nvinfer1::createInferRuntime(logger); trt_engine_ptr_.reset(infer->deserializeCudaEngine( serialized_engine.c_str(), serialized_engine.size(), - &PluginFactoryTensorRT::GetInstance())); + PluginFactoryTensorRT::GetInstance())); trt_execution_context_ptr_.reset(trt_engine_ptr_->createExecutionContext()); // Runtime is safe to delete after engine creation infer->destroy(); |