Searched refs:trt_context_idx (Results 1 – 1 of 1) sorted by relevance
122 int trt_context_idx);702 int trt_context_idx = status.ValueOrDie().second; in ComputeAsync() local721 Status stat = ExecuteTrtEngine(ctx, engine_context, trt_context_idx); in ComputeAsync()743 int trt_context_idx) { in ExecuteTrtEngine() argument776 engine_context->GetExecutionContext(trt_context_idx, &execution_context)); in ExecuteTrtEngine()779 VLOG(2) << "Selected execution context: " << trt_context_idx; in ExecuteTrtEngine()785 trt_context_idx, buffers, in ExecuteTrtEngine()789 trt_context_idx, buffers, in ExecuteTrtEngine()