Refactored operators. (#532)

blueoil/converter/core/operators/__init__.py

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+# Copyright 2019 The Blueoil Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# =============================================================================
+from .base import Operator, Conv, Pool, MaxPool, AveragePool, Add, Gemm, Mul,\
+    Maximum, MatMul, Minimum, Identity, BatchNormalization, Dropout, Gather,\
+    Unique, Cast, Prod, BatchNormalizationOptimized, Softmax, Relu, LeakyRelu,\
+    Quantizer, QTZ_binary_mean_scaling, QTZ_linear_mid_tread_half,\
+    QTZ_binary_channel_wise_mean_scaling, Lookup, SpaceToDepth, Transpose,\
+    Reshape, Flatten, ConcatOnDepth, DepthToSpace, ResizeNearestNeighbor,\
+    Split, Pad, Shape, StridedSlice, Variable, Input, Constant, Output

blueoil/converter/core/operators.py → blueoil/converter/core/operators/base.py

@@ -25,7 +25,7 @@
 from blueoil.converter.core.view import View
 from blueoil.converter.util import classproperty
 
-from .data_types import *
+from blueoil.converter.core.data_types import *
 
 if TYPE_CHECKING:
     import blueoil.converter.core.operators as ops