'Define all the inputs (with type, shape, name) that
the graph will need.'

def _get_inputs(self):

    return [InputDesc(tf.float32, (None, IMAGE_SIZE, IMAGE_SIZE), 'input'), InputDesc(tf.int32, (None,), 'label')]
