Home
last modified time | relevance | path

Searched refs:SetShapeTensorMask (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles.cc120 SetShapeTensorMask(input_partial_shapes); in InitProfiles()
171 SetShapeTensorMask(network); in AddProfiles()
187 void TrtShapeOptimizationProfile::SetShapeTensorMask( in SetShapeTensorMask() function in tensorflow::tensorrt::TrtShapeOptimizationProfile
207 void TrtShapeOptimizationProfile::SetShapeTensorMask( in SetShapeTensorMask() function in tensorflow::tensorrt::TrtShapeOptimizationProfile
Dtrt_shape_optimization_profiles.h208 void SetShapeTensorMask(const nvinfer1::INetworkDefinition* network);
209 void SetShapeTensorMask(