classifier operators: Fix support for prediction mask input.