Searched refs:max_nms_mask_size (Results 1 – 1 of 1) sorted by relevance
266 int64 max_nms_mask_size = num_boxes * bit_mask_len; in NmsGpu() local268 DataType::DT_INT32, TensorShape({max_nms_mask_size}), &d_nms_mask)); in NmsGpu()