Ejemplos de Flatten.update_node_stat en Python

Ejemplo n.º 1

    def extract(cls, node):
        attrs = {
            'axis': 1,
            'end_axis': -1,
        }

        Flatten.update_node_stat(node, attrs)
        return cls.enabled

Ejemplo n.º 2

    def extract(node):
        attrs = {
            'axis': 1,
            'end_axis': -1,
        }

        Flatten.update_node_stat(node, attrs)
        return __class__.enabled

Ejemplo n.º 3

    def extract(cls, node):
        proto_layer = node.pb
        param = proto_layer.flatten_param

        attrs = {
            'axis': param.axis,
            'end_axis': param.end_axis,
        }

        Flatten.update_node_stat(node, attrs)
        return cls.enabled

Ejemplo n.º 4

 def extract(cls, node):
     attrs = {
         'axis': node.module.axis,
     }
     Flatten.update_node_stat(node, attrs)
     return cls.enabled