Searched refs:nmsed_classes_flat (Results 1 – 1 of 1) sorted by relevance
415 auto nmsed_classes_flat = nmsed_classes_t->template flat<float>(); in BatchedNonMaxSuppressionOp() local427 nmsed_classes_flat(i * per_batch_size + j) = nmsed_classes[i][j]; in BatchedNonMaxSuppressionOp()