Searched refs:pre_nms_topn (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | generate_box_proposals_op.cu.cc | 227 int box_dim, int post_nms_topn, int pre_nms_topn) { in AllocatePreNMSTempTensors() argument
331 int pre_nms_topn; in Compute() local
339 OP_REQUIRES_OK(context, GetScalarValue(context, 5, &pre_nms_topn)); in Compute()
340 if (pre_nms_topn <= 0) { in Compute()
342 "pre_nms_topn should be greater than 0", pre_nms_topn)); in Compute()
374 const int nboxes_to_generate = std::min(conv_layer_nboxes, pre_nms_topn); in Compute()
440 nboxes_generated, box_dim, post_nms_topn_, pre_nms_topn)); in Compute()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | GenerateBoundingBoxProposals.pbtxt | 24 name: "pre_nms_topn"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | GenerateBoundingBoxProposals.pbtxt | 24 name: "pre_nms_topn"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_GenerateBoundingBoxProposals.pbtxt | 36 name: "pre_nms_topn"
69 The op selects top `pre_nms_topn` scoring boxes, decodes them with respect to anchors,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.image.pbtxt | 109 …', \'bbox_deltas\', \'image_info\', \'anchors\', \'nms_threshold\', \'pre_nms_topn\', \'min_size\'…
|
| D | tensorflow.raw_ops.pbtxt | 1769 …', \'bbox_deltas\', \'image_info\', \'anchors\', \'nms_threshold\', \'pre_nms_topn\', \'min_size\'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.image.pbtxt | 113 …', \'bbox_deltas\', \'image_info\', \'anchors\', \'nms_threshold\', \'pre_nms_topn\', \'min_size\'…
|
| D | tensorflow.raw_ops.pbtxt | 1769 …', \'bbox_deltas\', \'image_info\', \'anchors\', \'nms_threshold\', \'pre_nms_topn\', \'min_size\'…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 5730 pre_nms_topn=6000, argument
5768 pre_nms_topn=pre_nms_topn,
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 41752 …ut, image_info tf.Output, anchors tf.Output, nms_threshold tf.Output, pre_nms_topn tf.Output, min_…
41763 scores, bbox_deltas, image_info, anchors, nms_threshold, pre_nms_topn, min_size,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 18311 name: "pre_nms_topn"
|