I am not sure what type to put for getInputPrimitiveType() and getOutputPrimitiveType(). Currently it is Binary, should it be Array?