def extract(cls, node): attrs = {'num_of_dimensions': 3, 'is_inverse': True} TFFFT.update_node_stat(node, attrs) return cls.enabled
def extract(cls, node): attrs = {'num_of_dimensions': 2, 'fft_kind': 'RDFT'} TFFFT.update_node_stat(node, attrs) return cls.enabled