Searched refs:profiles_ (Results 1 – 6 of 6) sorted by relevance
68 profiles_.push_back(std::move(profConfig)); in ImplicitBatchModeCompatibleStrategy()80 profiles_.push_back(std::move(profConfig)); in OptimalStrategy()122 for (OptimizationProfileConfig& prof : profiles_) { in InitProfiles()140 for (int i = 0; i < profiles_.size(); i++) { in AddProfiles()142 Status status = profiles_[i].SetDimensions(network, optProfile); in AddProfiles()157 VLOG(1) << "Added optimization profile " << profiles_[i].DebugString() in AddProfiles()161 << profiles_[i].DebugString() in AddProfiles()165 if (!profiles_.empty() && config->getNbOptimizationProfiles() == 0) { in AddProfiles()223 for (int i = 0; i < profiles_.size(); i++) { in GetProfileNumber()224 if (profiles_[i].IncludesShapes(shapes)) { in GetProfileNumber()[all …]
123 if (profiles_.NeedProfiles() && profile_id >= profiles_.GetNumProfiles()) { in GetEngineContext()126 << profiles_.GetNumProfiles(); in GetEngineContext()
146 void clear() { profiles_.clear(); } in clear()187 std::vector<OptimizationProfileConfig> profiles_; variable
230 TrtShapeOptimizationProfile profiles_; variable
679 OP_REQUIRES(ctx, cache_res->profiles_.GetNumProfiles() == 0, in ComputeAsync()684 cache_res->profiles_.AddShape(input_concrete_shapes); in ComputeAsync()688 } else if (cache_res->profiles_.GetNumProfiles() == 0) { in ComputeAsync()690 if (!cache_res->profiles_.HasShape()) { in ComputeAsync()691 cache_res->profiles_.AddShape(input_concrete_shapes); in ComputeAsync()694 cache_res->profiles_.InitProfiles(input_partial_shapes_); in ComputeAsync()847 &cache_resource->profiles_, name()); in BuildEngine()961 profile_id = cache_res->profiles_.GetProfileNumber(input_concrete_shapes); in GetEngine()1000 TF_RETURN_IF_ERROR(cache_res->profiles_.CreateExecutionContexts( in GetEngine()1009 profile_id = cache_res->profiles_.GetProfileNumber(input_concrete_shapes); in GetEngine()
147 OP_REQUIRES_OK(ctx, resource->profiles_.RestoreProfiles(raw_engine)); in Compute()148 OP_REQUIRES_OK(ctx, resource->profiles_.CreateExecutionContexts( in Compute()