Ejemplos de Squeeze.update_node_stat en Python

Ejemplo n.º 1

    def extract(cls, node):
        attrs = get_mxnet_layer_attrs(node.symbol_dict)

        Squeeze.update_node_stat(node, {
            'squeeze_dims': attrs.int("axis", None),
            'keep_at_least_1d': True
        })
        return cls.enabled

Ejemplo n.º 2

Archivo: squeeze_ext.py Proyecto: mikhailk62/openvino

    def extract(cls, node):
        axis = int64_array(onnx_attr(node, 'axes', 'ints', default=[]))

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

        # update the attributes of the node
        Squeeze.update_node_stat(node, attrs)
        return cls.enabled

Ejemplo n.º 3

Archivo: reshape_related_ext.py Proyecto: yury-intel/openvino

 def extract(cls, node: Node):
     Squeeze.update_node_stat(
         node,
         {'squeeze_dims': tf_int_list(node.pb.attr['squeeze_dims'].list)})
     return cls.enabled