fused_fftconv2d_bhl_w_reshape_chunked

fused_fftconv2d_bhl_w_reshape_chunked(

x,

kernel,

shortcut=None,

chunk_size=None,

fft_size=None,

)

Channel-chunked fused 2D FFT convolution for BLH inputs [B, X, Y, H].

Parameters:

• x (Tensor)

• kernel (Tensor)

• shortcut (Tensor | None)

• chunk_size (int | None)

• fft_size (int | None)

Return type:

Tensor