Searched refs:TrtShapeOptimizationProfile (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.cc | 71 void TrtShapeOptimizationProfile::ImplicitBatchModeCompatibleStrategy( in ImplicitBatchModeCompatibleStrategy()
105 Status TrtShapeOptimizationProfile::RangeStrategy( in RangeStrategy()
127 void TrtShapeOptimizationProfile::OptimalStrategy( in OptimalStrategy()
142 Status TrtShapeOptimizationProfile::CollectShapeValues(OpKernelContext* ctx) { in CollectShapeValues()
195 Status TrtShapeOptimizationProfile::CollectShapeValues(const DataVec& input) { in CollectShapeValues()
255 void TrtShapeOptimizationProfile::InitProfiles( in InitProfiles()
325 void TrtShapeOptimizationProfile::InitCalibProfile( in InitCalibProfile()
344 Status TrtShapeOptimizationProfile::AddProfiles( in AddProfiles()
411 Status TrtShapeOptimizationProfile::ConfigureBuilder( in ConfigureBuilder()
419 void TrtShapeOptimizationProfile::SetShapeTensorMask( in SetShapeTensorMask()
[all …]
|
| D | trt_shape_optimization_profiles_test.cc | 111 TrtShapeOptimizationProfile* profile, bool has_prof, in CheckProfile()
165 TrtShapeOptimizationProfile profile; in TEST_P()
189 TrtShapeOptimizationProfile profile; in TEST_P()
|
| D | trt_engine_utils.h | 48 const TrtShapeOptimizationProfile& profiles,
|
| D | trt_shape_optimization_profiles.h | 204 class TrtShapeOptimizationProfile {
206 TrtShapeOptimizationProfile() {} in TrtShapeOptimizationProfile() function
|
| D | trt_lru_cache.h | 253 TrtShapeOptimizationProfile profiles_;
|
| D | trt_engine_utils.cc | 120 const TrtShapeOptimizationProfile& profiles, in SetTrtEngineInputs()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 61 TrtShapeOptimizationProfile* profile,
|
| D | convert_nodes.h | 176 TrtShapeOptimizationProfile* profiles, absl::string_view engine_name,
297 TrtShapeOptimizationProfile* profiles);
|
| D | convert_graph.cc | 695 TrtShapeOptimizationProfile* profile, in CreateStaticEngine()
|
| D | convert_nodes.cc | 1228 TRTInt8Calibrator* calibrator, TrtShapeOptimizationProfile* profiles) { in BuildCudaEngine()
5837 TrtShapeOptimizationProfile* profiles, absl::string_view engine_name, in ConvertGraphDefToEngine()
|
| D | convert_nodes_test.cc | 1282 TrtShapeOptimizationProfile profiles; in BuildAndRun()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 101 TrtShapeOptimizationProfile profile;
|
| D | trt_engine_resource_ops_test.cc | 162 TrtShapeOptimizationProfile profile; in CreateTRTEngine()
|
| D | trt_engine_op.cc | 171 const TrtShapeOptimizationProfile& profiles,
943 const TrtShapeOptimizationProfile& profiles, TRTBaseAllocator* allocator) { in ExecuteTrtEngine()
|