Searched refs:max_postnms_nboxes (Results 1 – 1 of 1) sorted by relevance
244 const int max_postnms_nboxes = std::min(num_boxes_to_generate, post_nms_topn); in AllocatePreNMSTempTensors() local247 TensorShape({box_dim * num_images * max_postnms_nboxes}), in AllocatePreNMSTempTensors()252 DataType::DT_FLOAT, TensorShape({num_images * max_postnms_nboxes}), in AllocatePreNMSTempTensors()