Searched defs:SetShapeTensorMask (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ | ||
| D | trt_shape_optimization_profiles.cc | 419 void TrtShapeOptimizationProfile::SetShapeTensorMask( in SetShapeTensorMask() function in tensorflow::tensorrt::TrtShapeOptimizationProfile |