Searched refs:per_batch_size (Results 1 – 1 of 1) sorted by relevance
247 int per_batch_size = total_size_per_batch; in BatchedNonMaxSuppressionOp() local362 per_batch_size = total_size_per_batch; in BatchedNonMaxSuppressionOp()364 per_batch_size = in BatchedNonMaxSuppressionOp()367 std::min(per_batch_size, (int)result_candidate_vec.size()); in BatchedNonMaxSuppressionOp()395 nmsed_boxes[batch].resize(per_batch_size * 4, 0); in BatchedNonMaxSuppressionOp()396 nmsed_scores[batch].resize(per_batch_size, 0); in BatchedNonMaxSuppressionOp()397 nmsed_classes[batch].resize(per_batch_size, 0); in BatchedNonMaxSuppressionOp()401 TensorShape boxes_shape({num_batches, per_batch_size, 4}); in BatchedNonMaxSuppressionOp()407 TensorShape scores_shape({num_batches, per_batch_size}); in BatchedNonMaxSuppressionOp()425 for (int j = 0; j < per_batch_size; ++j) { in BatchedNonMaxSuppressionOp()[all …]