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
def extract(node):
attrs = {}
Gather.update_node_stat(node, attrs)
return __class__.enabled
def extract(cls, node):
Gather.update_node_stat(node, {})
return cls.enabled
def extract(cls, node):
Gather.update_node_stat(node, {'batch_dims': node.pb.attr['batch_dims'].i})
return cls.enabled
