Ejemplos de AttributedGather.update_node_stat en Python

Ejemplo n.º 1

    def extract(cls, node):
        attrs = {
            'axis': np.array(onnx_attr(node, 'axis', 'i', default=0), dtype=np.int64)
        }

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

Ejemplo n.º 2

Archivo: gather_ext.py Proyecto: zoeysgithub/openvino

 def extract(cls, node):
     AttributedGather.update_node_stat(node, {'axis': 0})
     return cls.enabled

Ejemplo n.º 3

Archivo: take_ext.py Proyecto: shyransystems/OpenVINO-toolkit-orig

 def extract(cls, node: Node):
     attrs = get_mxnet_layer_attrs(node.symbol_dict)
     AttributedGather.update_node_stat(node, {
         'axis': attrs.int('axis', 0),
     })
     return cls.enabled