diff --git a/layers/__init__.py b/layers/__init__.py index 60fda4e..911d400 100644 --- a/layers/__init__.py +++ b/layers/__init__.py @@ -1,4 +1,5 @@ -from utils import OPERATOR +from utils.op_registry import OPERATOR + from .conv_layers import * from .dimension_utils import * from .common_layers import * diff --git a/utils/__init__.py b/utils/__init__.py index 89af612..f65bcb1 100644 --- a/utils/__init__.py +++ b/utils/__init__.py @@ -1,6 +1,4 @@ from .onnx_loader import load_onnx_modelproto from .builder import keras_builder, tflite_builder -from .op_registry import OPERATOR -from layers import * -__all__ = ['OPERATOR', 'LOG', 'load_onnx_modelproto', 'keras_builder', 'tflite_builder'] \ No newline at end of file +__all__ = ['load_onnx_modelproto', 'keras_builder', 'tflite_builder'] \ No newline at end of file