Function head_dim

src/models/ace_step/text_encoder.cpp:56–64

};

int64_t head_dim(const AceStepTextEncoderConfig & config) {
    if (config.num_attention_heads <= 0 || config.num_key_value_heads <= 0 || config.head_dim <= 0) {
        throw std::runtime_error("ACE-Step text encoder attention config is invalid");
    }
    if (config.num_attention_heads % config.num_key_value_heads != 0) {
        throw std::runtime_error("ACE-Step text encoder num_attention_heads must be divisible by num_key_value_heads");
    }
    return config.head_dim;
}

core::TensorValue ensure_contiguous(core::ModuleBuildContext & ctx, const core::TensorValue & input) {
    return core::ensure_backend_addressable_layout(ctx, input);