chromium/components/assist_ranker/quantized_nn_classifier_unittest.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/quantized_nn_classifier.h"
#include "components/assist_ranker/nn_classifier.h"
#include "components/assist_ranker/nn_classifier_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace assist_ranker {
namespace quantized_nn_classifier {
namespace {

RepeatedFieldBackInserter;
RepeatedPtrField;
copy;
vector;

void CreateLayer(const vector<int>& biases,
                 const vector<vector<int>>& weights,
                 float low,
                 float high,
                 QuantizedNNLayer* layer) { … }

// Creates a QuantizedDNNClassifierModel proto using a trained set of biases and
// weights.
QuantizedNNClassifierModel CreateModel(
    const vector<int>& hidden_biases,
    const vector<vector<int>>& hidden_weights,
    const vector<int>& logits_biases,
    const vector<vector<int>>& logits_weights,
    float low,
    float high) { … }

TEST(QuantizedNNClassifierTest, Dequantize) { … }

TEST(QuantizedNNClassifierTest, XorTest) { … }

TEST(QuantizedNNClassifierTest, ValidateQuantizedNNClassifierModel) { … }

}  // namespace
}  // namespace quantized_nn_classifier
}  // namespace assist_ranker