Searched refs:score_thresh (Results 1 – 3 of 3) sorted by relevance
444 const xla::XlaOp score_thresh = context->Input("score_threshold"); in Compile() local533 xla::Gt(scores, xla::Broadcast(score_thresh, {num_boxes})); in Compile()
1766 const float score_thresh = in prune_tx_2D() local1783 if (scores_2D[tx_idx] >= score_thresh && allow_tx_type) { in prune_tx_2D()
6020 float score_thresh = *(static_cast<float*>(score_threshold.GetValues())); in ConvertCombinedNMS() local6033 nvinfer1::PluginField{"scoreThreshold", &score_thresh, in ConvertCombinedNMS()