Files
ascend-tools/pt2pb/onnx-tensorflow/onnx_tf/handlers/backend/reduce_l1.py
T
2020-10-14 08:55:07 +08:00

24 lines
579 B
Python

import tensorflow as tf
from onnx_tf.handlers.backend_handler import BackendHandler
from onnx_tf.handlers.handler import onnx_op
from onnx_tf.handlers.handler import tf_func
from .math_mixin import ReductionMixin
@onnx_op("ReduceL1")
@tf_func(tf.norm)
class ReduceL1(ReductionMixin, BackendHandler):
@classmethod
def get_attrs_processor_param(cls):
return {"default": {"ord": 1}}
@classmethod
def version_1(cls, node, **kwargs):
return cls._common(node, **kwargs)
@classmethod
def version_11(cls, node, **kwargs):
return cls._common(node, **kwargs)