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