Searched defs:CollectShapeValues (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ | ||
| D | trt_shape_optimization_profiles.cc | 140 Status TrtShapeOptimizationProfile::CollectShapeValues(OpKernelContext* ctx) { in CollectShapeValues() function in tensorflow::tensorrt::TrtShapeOptimizationProfile |