Imported Upstream version 1.25.0
[platform/core/ml/nnfw.git] / compiler / luci-interpreter / pal / linux / PALGelu.h
@@ -1,6 +1,5 @@
 /*
- * 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