Python Unsqueeze.update_node_stat Examples

Example #1

    def extract(node):
        axis = np.array(onnx_attr(node, 'axes', 'ints', default=[]),
                        dtype=np.int64)

        attrs = {'unsqueeze_dims': axis if len(axis) != 0 else None}

        # update the attributes of the node
        Unsqueeze.update_node_stat(node, attrs)
        return __class__.enabled

Example #2

File: expand_dims_ext.py Project: zoeysgithub/openvino

 def extract(cls, node: Node):
     Unsqueeze.update_node_stat(node, {})
     return cls.enabled

Example #3

File: expand_dims_ext.py Project: tianxingyzxq/LLvisionCompile

 def extract(node: Node):
     Unsqueeze.update_node_stat(node, {})
     return __class__.enabled