Searched refs:GetProfileNumber (Results 1 – 4 of 4) sorted by relevance
152 EXPECT_EQ(0, profile.GetProfileNumber(shape_vec)); in TEST_F()193 int idx = profile.GetProfileNumber(shape_vec); in TEST_F()
151 int GetProfileNumber(const std::vector<TensorShape>& shapes);
220 int TrtShapeOptimizationProfile::GetProfileNumber( in GetProfileNumber() function in tensorflow::tensorrt::TrtShapeOptimizationProfile
961 profile_id = cache_res->profiles_.GetProfileNumber(input_concrete_shapes); in GetEngine()1009 profile_id = cache_res->profiles_.GetProfileNumber(input_concrete_shapes); in GetEngine()