Searched refs:inp_boxes (Results 1 – 1 of 1) sorted by relevance
227 OpKernelContext* context, const Tensor& inp_boxes, const Tensor& inp_scores, in BatchedNonMaxSuppressionOp() argument231 int q = inp_boxes.dim_size(2); in BatchedNonMaxSuppressionOp()233 const int num_batches = inp_boxes.dim_size(0); in BatchedNonMaxSuppressionOp()252 Tensor per_batch_boxes = inp_boxes.Slice(batch, batch + 1); in BatchedNonMaxSuppressionOp()