fftconv2d_bhl_w_reshape#

fftconv2d_bhl_w_reshape(x, kernel, shortcut=None)#

2D FFT convolution via subq_ops for BLH inputs [B, X, Y, H].

Reshapes to BHL, runs fftconv2d_bhl(), reshapes back.

Parameters:
Return type:

Tensor