Searched defs:kept_indices (Results 1 – 2 of 2) sorted by relevance
615 …View<int> kept_indices, View<int> kept_count, View<T> decoded_bboxes, View<T> scores, bool share_l… in consolidate_detections()827 void nms_collect(const Stream& stream, TensorSpan<int> kept_indices, TensorSpan<int> kept_count, in nms_collect()868 TensorView<int> kept_indices, TensorView<int> kept_count, in consolidate_detections()
235 csl::TensorSpan<int> kept_indices; in forward() local