Searched refs:engine_context (Results 1 – 2 of 2) sorted by relevance
101 EngineContext* engine_context = nullptr; in GetEngineContext() local115 engine_context = pair.second.get(); in GetEngineContext()119 return engine_context; in GetEngineContext()
121 Status ExecuteTrtEngine(OpKernelContext* ctx, EngineContext* engine_context,701 EngineContext* engine_context = status.ValueOrDie().first; in ComputeAsync() local711 if (!engine_context->cuda_engine) { in ComputeAsync()721 Status stat = ExecuteTrtEngine(ctx, engine_context, trt_context_idx); in ComputeAsync()742 EngineContext* engine_context, in ExecuteTrtEngine() argument748 auto& cuda_engine = engine_context->cuda_engine; in ExecuteTrtEngine()773 mutex_lock lock(engine_context->mu); in ExecuteTrtEngine()776 engine_context->GetExecutionContext(trt_context_idx, &execution_context)); in ExecuteTrtEngine()