Searched refs:per_batch_boxes (Results 1 – 1 of 1) sorted by relevance
252 Tensor per_batch_boxes = inp_boxes.Slice(batch, batch + 1); in BatchedNonMaxSuppressionOp() local266 float* boxes_data = per_batch_boxes.unaligned_flat<float>().data(); in BatchedNonMaxSuppressionOp()292 Tensor boxes(per_batch_boxes.dtype(), boxesShape); in BatchedNonMaxSuppressionOp()