def _fft(bottom, sequential, compute_size): if sequential: return sequential_batch_fft(bottom, compute_size) else: return tf.fft(bottom)