import tensorflow as tf
from extra_layers.Clip import Clip
from extra_layers.OutputSplit import OutputSplit
from extra_layers.Padding import ReflectPadding2D
from extra_layers.BinaryOp import Div
from extra_layers.UnaryOp import Sqrt
from extra_layers.UnaryOp import Swish
from extra_layers.Resize import Interp
from extra_layers.UnaryOp import DropConnect

extra_custom_objects = {'relu6': tf.nn.relu6,
                        'OutputSplit': OutputSplit, 'Clip': Clip, 'ReflectPadding2D': ReflectPadding2D,
                        'Div': Div, 'Sqrt': Sqrt, 'Interp': Interp, 'Swish': Swish, 'DropConnect': DropConnect}



import tensorflow as tf
from extra_layers.Clip import Clip
from extra_layers.OutputSplit import OutputSplit
from extra_layers.Padding import ReflectPadding2D
from extra_layers.BinaryOp import Div
from extra_layers.UnaryOp import Sqrt
from extra_layers.UnaryOp import Swish
from extra_layers.Resize import Interp
from extra_layers.UnaryOp import DropConnect

extra_custom_objects = {'relu6': tf.nn.relu6,
                        'OutputSplit': OutputSplit, 'Clip': Clip, 'ReflectPadding2D': ReflectPadding2D,
                        'Div': Div, 'Sqrt': Sqrt, 'Interp': Interp, 'Swish': Swish, 'DropConnect': DropConnect}


