Searched refs:box_check (Results 1 – 1 of 1) sorted by relevance
97 bool box_check = boxes.dim_size(2) == 1 || boxes.dim_size(2) == num_classes; in ParseAndCheckCombinedNMSBoxSizes() local98 OP_REQUIRES(context, box_check, in ParseAndCheckCombinedNMSBoxSizes()