Searched refs:post_nms_top_n (Results 1 – 1 of 1) sorted by relevance
1138 int post_nms_top_n; in __anon30b5031d1e02() local1139 TF_RETURN_IF_ERROR(c->GetAttr("post_nms_topn", &post_nms_top_n)); in __anon30b5031d1e02()1141 {c->Dim(scores, 0), post_nms_top_n, 4}); //(N,post_nms_top_n,4) in __anon30b5031d1e02()1143 {c->Dim(scores, 0), post_nms_top_n}); // (N,post_nms_top_n) in __anon30b5031d1e02()