Searched refs:selected_boxes (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_NonMaxSuppressionWithOverlaps.pbtxt | 60 selected_boxes = tf.gather(boxes, selected_indices)
|
| D | api_def_NonMaxSuppression.pbtxt | 54 selected_boxes = tf.gather(boxes, selected_indices)
|
| D | api_def_NonMaxSuppressionV2.pbtxt | 56 selected_boxes = tf.gather(boxes, selected_indices)
|
| D | api_def_NonMaxSuppressionV3.pbtxt | 62 selected_boxes = tf.gather(boxes, selected_indices)
|
| D | api_def_NonMaxSuppressionV4.pbtxt | 76 selected_boxes = tf.gather(boxes, selected_indices)
|
| D | api_def_NonMaxSuppressionV5.pbtxt | 93 selected_boxes = tf.gather(boxes, selected_indices)
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 312 Tensor selected_boxes; in NmsGpu() local
314 DataType::DT_INT8, TensorShape({num_boxes}), &selected_boxes)); in NmsGpu()
323 char* selected = (char*)(selected_boxes.flat<int8>().data()); in NmsGpu()
|
| D | non_max_suppression_op.cc | 344 std::vector<float> selected_boxes; in DoNMSPerClass() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 5517 selected_boxes, _, output_size, _ = control_flow_ops.while_loop(
5536 selected_boxes > 0, [2]), dtypes.int32) *
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 1299 selected_boxes = tf.gather(boxes, selected_indices)
1349 selected_boxes = tf.gather(boxes, selected_indices)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 8887 selected_boxes = tf.gather(boxes, selected_indices)
8935 selected_boxes = tf.gather(boxes, selected_indices)
8985 selected_boxes = tf.gather(boxes, selected_indices)
|