flashinfer.cake_causal_conv1d_train.cake_causal_conv1d_train_backward_workspace¶
- flashinfer.cake_causal_conv1d_train.cake_causal_conv1d_train_backward_workspace(S: int, C: int, device: device) Tensor¶
Allocate a backward workspace (see
..._workspace_bytes; no zeroing is needed).