Searched refs:score_thresh (Results 1 – 2 of 2) sorted by relevance
441 const xla::XlaOp score_thresh = context->Input("score_threshold"); in Compile() local527 xla::Gt(scores, xla::Broadcast(score_thresh, {num_boxes})); in Compile()
1878 const float score_thresh = in prune_tx_2D() local1883 if (scores_2D[i] < score_thresh && i != max_score_i) in prune_tx_2D()