Searched refs:TRTEngineOp (Results 1 – 2 of 2) sorted by relevance
88 class TRTEngineOp : public AsyncOpKernel { class90 explicit TRTEngineOp(OpKernelConstruction* context);276 StatusOr<FunctionLibraryRuntime::Handle> TRTEngineOp::ConstructFunctionHandle( in ConstructFunctionHandle()321 Status TRTEngineOp::ImportSegmentGraphDef(FunctionLibraryRuntime* lib, in ImportSegmentGraphDef()328 TRTEngineOp::TRTEngineOp(OpKernelConstruction* context) in TRTEngineOp() function in tensorflow::tensorrt::TRTEngineOp445 void TRTEngineOp::ExecuteNativeSegment(OpKernelContext* ctx, in ExecuteNativeSegment()483 void TRTEngineOp::ExecuteCalibration(OpKernelContext* ctx, in ExecuteCalibration()533 Status TRTEngineOp::VerifyInputShapes( in VerifyInputShapes()612 void TRTEngineOp::ComputeAsync(OpKernelContext* ctx, in ComputeAsync()741 Status TRTEngineOp::ExecuteTrtEngine(OpKernelContext* ctx, in ExecuteTrtEngine()[all …]
3 This module provides necessary bindings and introduces `TRTEngineOp` operator