Searched refs:max_cached_engines_ (Results 1 – 2 of 2) sorted by relevance
87 ctx, ctx->GetAttr("max_cached_engines_count", &max_cached_engines_)); in InitializeTRTResource()98 ctx, this->max_cached_engines_); in Compute()170 int max_cached_engines_; member in tensorflow::tensorrt::InitializeTRTResource
245 int max_cached_engines_; member in tensorflow::tensorrt::TRTEngineOp473 &max_cached_engines_)); in TRTEngineOp()1038 *cr = new TRTEngineCacheResource(ctx, this->max_cached_engines_); in GetEngineCacheResource()