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