Fabio
c36510c21e
Some checks failed
Linux Build / Linux cpu (push) Has been cancelled
Linux Build / Linux cpu-clang (push) Has been cancelled
Linux Build / Linux vulkan (push) Has been cancelled
mac-build / macOS CPU (push) Has been cancelled
Nix Build / Nix (cpu) (push) Has been cancelled
Nix Build / Nix (vulkan) (push) Has been cancelled
windows-build / Windows CPU (push) Has been cancelled
19 lines
712 B
C++
19 lines
712 B
C++
#pragma once
|
|
#include "engine/framework/modules/streaming_conv_modules.h"
|
|
#include "engine/framework/modules/linear_module.h"
|
|
#include "engine/framework/modules/norm_modules.h"
|
|
|
|
namespace engine::modules {
|
|
// Vocos/F5-style non-causal ConvNeXt, logical [batch, frames, channels].
|
|
struct ConvNeXt1dWeights {
|
|
DepthwiseConv1dWeights depthwise;
|
|
NormWeights norm;
|
|
LinearWeights expansion;
|
|
LinearWeights projection;
|
|
core::TensorValue gamma;
|
|
};
|
|
// Optional BF16 autocast preserves F32 normalization and residual arithmetic.
|
|
core::TensorValue build_convnext1d(
|
|
core::ModuleBuildContext & ctx, const core::TensorValue & input,
|
|
const ConvNeXt1dWeights & weights, bool bf16_autocast = false);
|
|
}
|