IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added log extractor for tf (#4090)

Browse Source
This commit is contained in:
Anna Likholat 2021-02-01 11:26:03 +03:00 committed by GitHub
parent 8cb25fdb98
commit a6a5635a59
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
model-optimizer/extensions/front/tf/activation_ext.py
View File

@ -14,7 +14,7 @@
limitations under the License.
"""
from extensions.ops.activation_ops import Abs, Elu, Erf, Exp, ReLU, LeakyReLU, LogicalNot, ReLU6, Sigmoid, \
Sin, Sinh, Cos, Cosh, Tan, Tanh, Ceiling, Atanh, Acosh, Asinh, Mish
Sin, Sinh, Cos, Cosh, Tan, Tanh, Ceiling, Atanh, Acosh, Asinh, Mish, Log
from mo.front.extractor import FrontExtractorOp
@ -220,3 +220,13 @@ class MishExtractor(FrontExtractorOp):
def extract(cls, node):
Mish.update_node_stat(node)
return cls.enabled
class LogExtractor(FrontExtractorOp):
op = 'Log'
enabled = True
@classmethod
def extract(cls, node):
Log.update_node_stat(node)
return cls.enabled