# Path: deep_architect/contrib/misc/search_spaces/tensorflow/common.py
# D = hp.Discrete
# def siso_tensorflow_module(name, compile_fn, name_to_hyperp, scope=None):


import keras
import numpy as np
import deep_architect.modules as mo
import deep_architect.hyperparameters as hp
from deep_architect.contrib.misc.search_spaces.tensorflow.common import siso_tfm
# Using TensorFlow Backend

# Search Space


D = hp.Discrete  # Discrete Hyperparameter


def flatten():

    def compile_fn(di, dh):
        Flatten = keras.layers.Flatten()

        def fn(di):
            return {'out': Flatten(di['in'])}


# Path: deep_architect/contrib/misc/search_spaces/tensorflow/common.py
# D = hp.Discrete
# def siso_tensorflow_module(name, compile_fn, name_to_hyperp, scope=None):


# Path: dev/tutorials/multiworker/search_space.py
import keras
import numpy as np
import deep_architect.modules as mo
import deep_architect.hyperparameters as hp
from deep_architect.contrib.misc.search_spaces.tensorflow.common import siso_tfm

# Using TensorFlow Backend

# Search Space


D = hp.Discrete  # Discrete Hyperparameter


def flatten():

    def compile_fn(di, dh):
        Flatten = keras.layers.Flatten()

        def fn(di):
            return {'out': Flatten(di['in'])}


        return fn