Function load_backend_weights

src/models/silero_vad/runtime.cpp:64–95

}

std::shared_ptr<const SileroBackendWeights> load_backend_weights(
    const SileroWeights & weights,
    ggml_backend_t backend,
    core::BackendType backend_type,
    assets::TensorStorageType storage_type) {
    auto out = std::make_shared<SileroBackendWeights>();
    out->store = std::make_shared<core::BackendWeightStore>(
        backend, backend_type, "silero_vad.weights", 16ull * 1024ull * 1024ull);
    auto & store = *out->store;
    if (weights.source == nullptr) {
        throw std::runtime_error("Silero VAD weights require a tensor source");
    }
    const auto & source = *weights.source;
    out->stft_conv_weight = store.load_tensor(source, "stft_conv.weight", storage_type, {258, 1, 256});
    out->conv1_weight = store.load_tensor(source, "conv1.weight", storage_type, {128, 129, 3});
    out->conv1_bias = load_f32_bias(store, source, "conv1.bias", {128});
    out->conv2_weight = store.load_tensor(source, "conv2.weight", storage_type, {64, 128, 3});
    out->conv2_bias = load_f32_bias(store, source, "conv2.bias", {64});
    out->conv3_weight = store.load_tensor(source, "conv3.weight", storage_type, {64, 64, 3});
    out->conv3_bias = load_f32_bias(store, source, "conv3.bias", {64});
    out->conv4_weight = store.load_tensor(source, "conv4.weight", storage_type, {128, 64, 3});
    out->conv4_bias = load_f32_bias(store, source, "conv4.bias", {128});
    out->lstm_weight_ih = store.load_tensor(source, "lstm_cell.weight_ih", storage_type, {4 * kHiddenSize, kHiddenSize});
    out->lstm_weight_hh = store.load_tensor(source, "lstm_cell.weight_hh", storage_type, {4 * kHiddenSize, kHiddenSize});
    out->lstm_bias_ih = load_f32_bias(store, source, "lstm_cell.bias_ih", {4 * kHiddenSize});
    out->lstm_bias_hh = load_f32_bias(store, source, "lstm_cell.bias_hh", {4 * kHiddenSize});
    out->final_conv_weight = store.load_tensor(source, "final_conv.weight", storage_type, {1, kHiddenSize, 1});
    out->final_conv_bias = load_f32_bias(store, source, "final_conv.bias", {1});
    store.upload();
    weights.source->release_storage();
    return out;
}

std::shared_ptr<const SileroWeights> require_weights(std::shared_ptr<const SileroWeights> weights) {
    if (!weights) {