Home
last modified time | relevance | path

Searched defs:engine_context (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
H A Dtrt_engine_op.cc900 EngineContext* engine_context = status.ValueOrDie().first; in ComputeAsync() local
942 OpKernelContext* ctx, EngineContext* engine_context, int trt_context_idx, in ExecuteTrtEngine()
1154 EngineContext* engine_context = cache_res->GetEngineContext(profile_id); in GetEngine() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
H A Dtrt_lru_cache.cc101 EngineContext* engine_context = nullptr; in GetEngineContext() local