Searched refs:GetEngineContext (Results 1 – 3 of 3) sorted by relevance
99 EngineContext* TRTEngineCacheResource::GetEngineContext( in GetEngineContext() function in tensorflow::tensorrt::TRTEngineCacheResource122 EngineContext* TRTEngineCacheResource::GetEngineContext(const int profile_id) { in GetEngineContext() function in tensorflow::tensorrt::TRTEngineCacheResource
207 EngineContext* GetEngineContext(const std::vector<TensorShape>& input_shapes);213 EngineContext* GetEngineContext(const int profile_id);
971 engine_contexts = cache_res->GetEngineContext(input_concrete_shapes); in GetEngine()973 engine_contexts = cache_res->GetEngineContext(profile_id); in GetEngine()