Searched refs:engine_context (Results 1 – 1 of 1) sorted by relevance
85 bool ExecuteTrtEngine(OpKernelContext* ctx, EngineContext* engine_context);377 EngineContext* engine_context = GetEngine(input_shapes, ctx); in ComputeAsync() local378 if (!engine_context->cuda_engine) { in ComputeAsync()385 const bool retry = ExecuteTrtEngine(ctx, engine_context); in ComputeAsync()395 EngineContext* engine_context) { in ExecuteTrtEngine() argument397 auto& cuda_engine = engine_context->cuda_engine; in ExecuteTrtEngine()507 mutex_lock lock(engine_context->mu); in ExecuteTrtEngine()509 auto ret = engine_context->execution_context->enqueue(num_batch, &buffers[0], in ExecuteTrtEngine()