Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5931 const auto boxes_dims = boxes_tensor->getDimensions(); in ConvertCombinedNMS() local
5934 (!HasStaticShape(boxes_dims) || !HasStaticShape(scores_dims))) { in ConvertCombinedNMS()
5939 if (boxes_dims.nbDims != 3 + offset) { in ConvertCombinedNMS()
5946 const int num_boxes = boxes_dims.d[0 + offset]; in ConvertCombinedNMS()
5948 boxes_dims.d[class_idx] == 1 || boxes_dims.d[class_idx] == num_classes; in ConvertCombinedNMS()
6007 bool share_location = (boxes_dims.d[class_idx] == 1); in ConvertCombinedNMS()