diff options
Diffstat (limited to 'tensorflow/contrib/tensorrt/kernels/trt_engine_op.h')
-rw-r--r-- | tensorflow/contrib/tensorrt/kernels/trt_engine_op.h | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/tensorflow/contrib/tensorrt/kernels/trt_engine_op.h b/tensorflow/contrib/tensorrt/kernels/trt_engine_op.h index 0d2f9e8a9d..6fe318be6a 100644 --- a/tensorflow/contrib/tensorrt/kernels/trt_engine_op.h +++ b/tensorflow/contrib/tensorrt/kernels/trt_engine_op.h @@ -52,19 +52,17 @@ class TRTEngineOp : public AsyncOpKernel { private: // Execute calibration - void ExecuteCalibration(OpKernelContext* ctx, - AsyncHelper* helper); + void ExecuteCalibration(OpKernelContext* ctx, AsyncHelper* helper); // Construct a function handle for executing native funcdef graph Status ConstructFunctionHandle(OpKernelContext* ctx); // Execute replaced native segment as function Op. - void ExecuteNativeSegment(OpKernelContext* ctx, - AsyncHelper* helper); + void ExecuteNativeSegment(OpKernelContext* ctx, AsyncHelper* helper); // Allocate necessary resources for calibration - Status AllocateCalibrationResources( - OpKernelContext* ctx, TRTCalibrationResource** cr); + Status AllocateCalibrationResources(OpKernelContext* ctx, + TRTCalibrationResource** cr); // TODO(samikama): context should go to a resource manager! typedef std::pair<TrtUniquePtrType<nvinfer1::ICudaEngine>, |