Searched refs:num_boxes_to_generate (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | generate_box_proposals_op.cu.cc | 190 int num_boxes_to_generate, int box_dim) { in AllocateGenerationTempTensors() argument
212 TensorShape({num_images, box_dim * num_boxes_to_generate}), dev_boxes)); in AllocateGenerationTempTensors()
215 DataType::DT_INT8, TensorShape({num_images, num_boxes_to_generate}), in AllocateGenerationTempTensors()
226 Tensor* dev_prenms_nboxes, int num_images, int num_boxes_to_generate, in AllocatePreNMSTempTensors() argument
230 DataType::DT_FLOAT, TensorShape({box_dim * num_boxes_to_generate}), in AllocatePreNMSTempTensors()
235 DataType::DT_FLOAT, TensorShape({num_boxes_to_generate}), in AllocatePreNMSTempTensors()
240 DataType::DT_INT32, TensorShape({num_boxes_to_generate}), in AllocatePreNMSTempTensors()
244 const int max_postnms_nboxes = std::min(num_boxes_to_generate, post_nms_topn); in AllocatePreNMSTempTensors()
|