// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef THIRD_PARTY_BLINK_RENDERER_PLATFORM_GRAPHICS_DARKMODE_DARKMODE_CLASSIFIER_H_
#define THIRD_PARTY_BLINK_RENDERER_PLATFORM_GRAPHICS_DARKMODE_DARKMODE_CLASSIFIER_H_

// This file is automatically generated using tfNative from a neural network,
// trained by TensorFlow. Please do not edit.

#include <cstdint>

namespace darkmode_tfnative_model {

struct alignas(16) FixedAllocations {
  float alloc0[10];
  int32_t alloc0_shape[2];
  float alloc1[10];
  int32_t alloc1_shape[2];
};

extern int32_t input0Shape[2];
extern int32_t logits_MatMul_merged_with_dnn_logits_BiasAdd0Shape[2];

void Inference(
    base::span<const float> input0 /* shape: 1,4 */,
    base::span<float> logits_MatMul_merged_with_dnn_logits_BiasAdd0 /* shape:
                                                                       1,1 */
    ,
    FixedAllocations* __restrict fixed);

}  // namespace darkmode_tfnative_model

#endif  // THIRD_PARTY_BLINK_RENDERER_PLATFORM_GRAPHICS_DARKMODE_DARKMODE_CLASSIFIER_H_