chromium/third_party/webrtc/modules/audio_processing/agc2/rnn_vad/rnn.cc

/*
 *  Copyright (c) 2018 The WebRTC project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include "modules/audio_processing/agc2/rnn_vad/rnn.h"

#include "rtc_base/checks.h"
#include "third_party/rnnoise/src/rnn_vad_weights.h"

namespace webrtc {
namespace rnn_vad {
namespace {

kInputLayerInputSize;
static_assert …;
kInputDenseBias;
kInputDenseWeights;
kInputLayerOutputSize;
static_assert …;

kHiddenGruBias;
kHiddenGruRecurrentWeights;
kHiddenGruWeights;
kHiddenLayerOutputSize;
static_assert …;

kOutputDenseBias;
kOutputDenseWeights;
kOutputLayerOutputSize;
static_assert …;

}  // namespace

RnnVad::RnnVad(const AvailableCpuFeatures& cpu_features)
    : … { … }

RnnVad::~RnnVad() = default;

void RnnVad::Reset() { … }

float RnnVad::ComputeVadProbability(
    rtc::ArrayView<const float, kFeatureVectorSize> feature_vector,
    bool is_silence) { … }

}  // namespace rnn_vad
}  // namespace webrtc