chromium/ui/events/ozone/evdev/touch_filter/palm_model/onedevice_train_palm_detection_filter_inference_beta.cc

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

#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/351564777): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif

#include "ui/events/ozone/evdev/touch_filter/palm_model/onedevice_train_palm_detection_filter_inference_beta.h"

#include <algorithm>
#include <cassert>
#include <cmath>
#include <cstdint>
#include <cstring>
#include <limits>
#include <tuple>

#ifndef USE_EIGEN
#define USE_EIGEN …
#endif

namespace ui::internal_onedevice::beta_model {
namespace …  // anonymous namespace

// -----------------------------------------------------------------------------
// INFERENCE
// -----------------------------------------------------------------------------

int32_t input_from_feature_columns_input_layer_concat_concat0Shape[2] = …;
int32_t logits_MatMul_merged_with_dnn_logits_BiasAdd0Shape[2] = …;

void Inference(
    const float* __restrict input_from_feature_columns_input_layer_concat_concat0 /* shape: 1,325 */
    ,
    float* __restrict logits_MatMul_merged_with_dnn_logits_BiasAdd0 /* shape:
                                                                       1,1 */
    ,
    FixedAllocations* __restrict fixed) { … }

}  // namespace ui::internal_onedevice::beta_model