Searched refs:inp_scores (Results 1 – 1 of 1) sorted by relevance
227 OpKernelContext* context, const Tensor& inp_boxes, const Tensor& inp_scores, in BatchedNonMaxSuppressionOp() argument232 int num_classes = inp_scores.dim_size(2); in BatchedNonMaxSuppressionOp()254 Tensor per_batch_scores = inp_scores.Slice(batch, batch + 1); in BatchedNonMaxSuppressionOp()