chromium/components/assist_ranker/nn_classifier.cc

// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/assist_ranker/nn_classifier.h"

#include "base/check_op.h"
#include "components/assist_ranker/proto/nn_classifier.pb.h"

namespace assist_ranker {
namespace nn_classifier {
namespace {

RepeatedPtrField;
vector;

vector<float> FeedForward(const NNLayer& layer, const vector<float>& input) { … }

// Apply ReLU activation function to a vector, which sets all values to
// max(0, value).
void Relu(vector<float>* const v) { … }

bool ValidateLayer(const NNLayer& layer) { … }

}  // namespace

bool Validate(const NNClassifierModel& model) { … }

vector<float> Inference(const NNClassifierModel& model,
                        const vector<float>& input) { … }

}  // namespace nn_classifier
}  // namespace assist_ranker