def extract(node):
attrs = tf_create_attrs(node, 4, 3)
attrs.update({'op': __class__.op,
'get_weights_permute': PermuteAttrs.Permutation(perm=int64_array([4, 3, 0, 1, 2]),
inv=int64_array([2, 3, 4, 1, 0]))
})
# update the attributes of the node
Deconvolution.update_node_stat(node, attrs)
return __class__.enabled
def extract(cls, node):
attrs = tf_create_attrs(node, 4, 3)
attrs.update({'op': cls.op,
'get_weights_permute': PermuteAttrs.Permutation(perm=int64_array([4, 3, 0, 1, 2]),
inv=int64_array([2, 3, 4, 1, 0])),
'swap_0_and_2_inputs': True,
'shape_input': True,
})
# update the attributes of the node
Deconvolution.update_node_stat(node, attrs)
return cls.enabled