Searched refs:TRTEngineOp (Results 1 – 2 of 2) sorted by relevance
66 class TRTEngineOp : public AsyncOpKernel { class68 explicit TRTEngineOp(OpKernelConstruction* context);161 Status TRTEngineOp::ConstructFunctionHandle(OpKernelContext* ctx) { in ConstructFunctionHandle()186 TRTEngineOp::TRTEngineOp(OpKernelConstruction* context) in TRTEngineOp() function in tensorflow::tensorrt::TRTEngineOp242 void TRTEngineOp::ExecuteNativeSegment(OpKernelContext* ctx, in ExecuteNativeSegment()289 void TRTEngineOp::ExecuteCalibration(OpKernelContext* ctx, in ExecuteCalibration()333 bool TRTEngineOp::GetCompatibleCachedEngine( in GetCompatibleCachedEngine()363 void TRTEngineOp::ComputeAsync(OpKernelContext* ctx, in ComputeAsync()394 bool TRTEngineOp::ExecuteTrtEngine(OpKernelContext* ctx, in ExecuteTrtEngine()519 EngineContext* TRTEngineOp::GetEngine( in GetEngine()[all …]
3 This module provides necessary bindings and introduces `TRTEngineOp` operator