Ejemplo n.º 1
0
    def extract(node):

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

        Gather.update_node_stat(node, attrs)
        return __class__.enabled
Ejemplo n.º 2
0
    def extract(node):
        attrs = {}

        Gather.update_node_stat(node, attrs)

        return __class__.enabled
Ejemplo n.º 3
0
 def extract(cls, node):
     Gather.update_node_stat(node, {})
     return cls.enabled
Ejemplo n.º 4
0
 def extract(cls, node):
     Gather.update_node_stat(node, {'batch_dims': node.pb.attr['batch_dims'].i})
     return cls.enabled