"""
- Copyright (c) 2018 Intel Corporation
+ Copyright (c) 2019 Intel Corporation
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
See the License for the specific language governing permissions and
limitations under the License.
"""
-import numpy as np
-from mo.front.tf.extractors.utils import tf_dtype_extractor
-from mo.ops.shape import Shape
+from mo.front.extractor import FrontExtractorOp
-def tf_shape_ext(pb):
- return {
- 'infer': Shape.infer,
- 'data_type': tf_dtype_extractor(pb.attr['out_type'].type, np.int32)
- }
+class GatherFrontExtractor(FrontExtractorOp):
+ op = 'Embedding'
+ enabled = True
+
+ @staticmethod
+ def extract(node):
+ return __class__.enabled