Impl now requires an input_layer with a 'get_output_shape' method.