def extract(cls, node):
attrs = {
'op': __class__.op,
'dim': node.module.shape,
}
Reshape.update_node_stat(node, attrs)
return cls.enabled
def extract(cls, node):
dim = onnx_attr(node, 'shape', 'ints', None)
if dim is not None:
dim = int64_array(dim)
Reshape.update_node_stat(node, {'dim': dim})
else:
Reshape.update_node_stat(node)
return cls.enabled
def extract(cls, node):
attrs = get_mxnet_layer_attrs(node.symbol_dict)
dim = attrs.tuple("shape", int, None)
reverse = attrs.bool("reverse", False)
update_attrs = {
'dim': int64_array(dim),
'reverse': reverse
}
for d in dim:
if d in [-2, -3, -4] or reverse:
MXReshape.update_node_stat(node, update_attrs)
return cls.enabled
# update the attributes of the node
Reshape.update_node_stat(node, update_attrs)
return cls.enabled
def extract(cls, node):
param = node.pb.reshape_param
if param.axis != 0:
log.error(
'The operation "Reshape" has attribute "axis" with unsupported value "{}"'
.format(param['axis']))
return False
if param.num_axes != -1:
log.error(
'The operation "Reshape" has attribute "num_axes" with unsupported value "{}"'
.format(param['num_axes']))
return False
Reshape.update_node_stat(node, {
'dim': list(param.shape.dim),
})
return cls.enabled
def extract(cls, node: Node):
Reshape.update_node_stat(node, {'special_zero': False})
return cls.enabled
