Searched refs:EngineContext (Results 1 – 2 of 2) sorted by relevance
128 struct EngineContext { struct129 EngineContext() {} // Creates an empty context. in EngineContext() argument130 EngineContext( in EngineContext() argument179 LRUCache<std::vector<TensorShape>, std::unique_ptr<EngineContext>,
85 bool ExecuteTrtEngine(OpKernelContext* ctx, EngineContext* engine_context);92 EngineContext* GetEngine(const std::vector<TensorShape>& input_shapes,377 EngineContext* engine_context = GetEngine(input_shapes, ctx); in ComputeAsync()395 EngineContext* engine_context) { in ExecuteTrtEngine()519 EngineContext* TRTEngineOp::GetEngine( in GetEngine()521 static EngineContext empty_context; in GetEngine()575 absl::make_unique<EngineContext>( in GetEngine()633 cache.emplace(engine_input_shapes, absl::make_unique<EngineContext>()); in GetEngine()640 absl::make_unique<EngineContext>(std::move(engine), in GetEngine()