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
def extract(cls, node): AttributedGather.update_node_stat(node, {'axis': 0}) return cls.enabled
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