def extract(cls, node):
# update the attributes of the node
block_size = node.pb.attr['block_size'].i
data_format = node.pb.attr['data_format'].s.decode('utf-8')
SpaceToDepth.update_node_stat(node, {
'block_size': block_size,
'data_format': data_format
})
return cls.enabled
def extract(cls, node):
# update the attributes of the node
block_size = onnx_attr(node, 'blocksize', 'i', default=None)
SpaceToDepth.update_node_stat(node, {'block_size': block_size})
return cls.enabled