Searched refs:box_check (Results 1 – 2 of 2) sorted by relevance
104 bool box_check = boxes.dim_size(2) == 1 || boxes.dim_size(2) == num_classes; in ParseAndCheckCombinedNMSBoxSizes() local105 OP_REQUIRES(context, box_check, in ParseAndCheckCombinedNMSBoxSizes()
5947 bool box_check = in ConvertCombinedNMS() local5949 if (!box_check) { in ConvertCombinedNMS()