fused_fftconv2d_bhl_w_reshape#

fused_fftconv2d_bhl_w_reshape(x, kernel, shortcut=None, fft_size=None)#

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

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

Parameters:

x (Tensor)
kernel (Tensor)
shortcut (Tensor | None)
fft_size (int | None)

Return type: