/*
- * Copyright (c) 2021 Samsung Electronics Co., Ltd. All Rights Reserved
- * Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+ * Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* limitations under the License.
*/
-#ifndef LUCI_INTERPRETER_PAL_ELU_H
-#define LUCI_INTERPRETER_PAL_ELU_H
+#ifndef LUCI_INTERPRETER_PAL_GELU_H
+#define LUCI_INTERPRETER_PAL_GELU_H
-#include <tensorflow/lite/kernels/internal/optimized/optimized_ops.h>
+#include <tensorflow/lite/kernels/internal/reference/gelu.h>
namespace luci_interpreter_pal
{
-static inline void Elu(const tflite::RuntimeShape &input_shape, const float *input_data,
- const tflite::RuntimeShape &output_shape, float *output_data)
+static inline void Gelu(bool approximate, const tflite::RuntimeShape &input_shape,
+ const float *input_data, const tflite::RuntimeShape &output_shape,
+ float *output_data)
{
- tflite::optimized_ops::Elu(input_shape, input_data, output_shape, output_data);
+ tflite::reference_ops::Gelu(input_shape, input_data, approximate, output_shape, output_data);
}
} // namespace luci_interpreter_pal
-#endif // LUCI_INTERPRETER_PAL_ELU_H
+#endif // LUCI_INTERPRETER_PAL_GELU_H