Searched defs:num_selected (Results 1 – 6 of 6) sorted by relevance
513 std::vector<int>& num_selected; member650 std::vector<int> num_selected(num_classes); in NonMaxSuppressionMultiClassRegularHelper() local
357 const auto num_selected = result_scores_after_nms.size(); in run() local
166 NumSelectedIteratorT num_selected, int64_t num_input_items) in unique_by_key()
818 auto num_selected = Reshape(scope, Sum(scope, indicators, reduction_indices), in MinOrMaxGrad() local1280 auto num_selected = UnsortedSegmentSum( in UnsortedSegmentMinOrMaxGrad() local
296 Tensor num_selected; in CountIf() local
422 size_t num_selected; member