Home
last modified time | relevance | path

Searched refs:InitProfiles (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles.h169 void InitProfiles(
Dtrt_shape_optimization_profiles_test.cc175 profile.InitProfiles(input_partial_shapes); in TEST_F()
Dtrt_shape_optimization_profiles.cc100 void TrtShapeOptimizationProfile::InitProfiles( in InitProfiles() function in tensorflow::tensorrt::TrtShapeOptimizationProfile
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc694 cache_res->profiles_.InitProfiles(input_partial_shapes_); in ComputeAsync()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1488 profiles.InitProfiles(input_partial_shapes); in BuildAndRun()