def extract(cls, node):
attrs = {'batch_dims': onnx_attr(node, 'batch_dims', 'i', default=0)}
GatherND.update_node_stat(node, attrs)
return cls.enabled
def extract(cls, node):
attrs = {
'batch_dims': 0,
}
GatherND.update_node_stat(node, attrs)
return cls.enabled