Searched refs:TRTEngineCacheResource (Results 1 – 7 of 7) sorted by relevance
54 Logger& TRTEngineCacheResource::GetLogger() { in GetLogger()59 TRTEngineCacheResource::TRTEngineCacheResource(OpKernelContext* ctx, in TRTEngineCacheResource() function in tensorflow::tensorrt::TRTEngineCacheResource73 TRTEngineCacheResource::~TRTEngineCacheResource() { in ~TRTEngineCacheResource()77 string TRTEngineCacheResource::DebugString() const { in DebugString()99 EngineContext* TRTEngineCacheResource::GetEngineContext( in GetEngineContext()122 EngineContext* TRTEngineCacheResource::GetEngineContext(const int profile_id) { in GetEngineContext()
190 class TRTEngineCacheResource : public ResourceBase {199 TRTEngineCacheResource(OpKernelContext* ctx, size_t capacity);201 ~TRTEngineCacheResource() override;
61 MakeResourceHandle<TRTEngineCacheResource>( in Compute()92 core::RefCountPtr<TRTEngineCacheResource> resource; in Compute()94 ctx, LookupOrCreateResource<TRTEngineCacheResource>( in Compute()96 [this, ctx](TRTEngineCacheResource** resource) -> Status { in Compute()97 *resource = new TRTEngineCacheResource( in Compute()135 nvinfer1::createInferRuntime(TRTEngineCacheResource::GetLogger())); in Compute()192 TRTEngineCacheResource* resource = nullptr; in Compute()236 ctx->resource_manager()->Delete<TRTEngineCacheResource>( in Compute()
102 TRTEngineCacheResource* cache_res,126 TRTEngineCacheResource* cache_res);129 TRTEngineCacheResource** cache_res);139 OpKernelContext* ctx, TRTEngineCacheResource* cache_resource);147 TRTEngineCacheResource* cache_resource);484 TRTEngineCacheResource* cache_res, in ExecuteCalibration()620 TRTEngineCacheResource* cache_res = nullptr; in ComputeAsync()805 TRTEngineCacheResource** cache_res) { in GetEngineCacheResource()821 {[this, ctx](TRTEngineCacheResource** cr) -> Status { in GetEngineCacheResource()822 *cr = new TRTEngineCacheResource(ctx, this->max_cached_engines_); in GetEngineCacheResource()[all …]
162 TRTEngineCacheResource* cache_resource = nullptr; in TEST_F()215 TRTEngineCacheResource* cache_resource = nullptr; in TEST_F()243 TRTEngineCacheResource* cache_resource = nullptr; in TEST_F()271 TRTEngineCacheResource* cache_resource = nullptr; in TEST_F()
44 TRTEngineCacheResource* resource = nullptr; in Compute()
125 TRTEngineCacheResource* resource = nullptr; in TEST_F()