Imported Upstream version 1.25.0

[platform/core/ml/nnfw.git] / compiler / luci-interpreter / src / loader / nodes / Gelu.cpp

diff --git a/onert-micro/luci-interpreter/src/loader/nodes/PRelu.cpp b/compiler/luci-interpreter/src/loader/nodes/Gelu.cpp
similarity index 52%
rename from onert-micro/luci-interpreter/src/loader/nodes/PRelu.cpp
rename to compiler/luci-interpreter/src/loader/nodes/Gelu.cpp
index 6adc549..fc77a58 100644 (file)
--- a/onert-micro/luci-interpreter/src/loader/nodes/PRelu.cpp
+++ b/compiler/luci-interpreter/src/loader/nodes/Gelu.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021 Samsung Electronics Co., Ltd. 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.
@@ -16,22 +16,23 @@
 
 #include "Builders.h"
 
-#include "kernels/PRelu.h"
+#include "kernels/Gelu.h"
 
 namespace luci_interpreter
 {
 
-std::unique_ptr<Kernel> build_kernel_CirclePRelu(std::vector<const Tensor *> &&inputs,
-                                                 std::vector<Tensor *> &&outputs,
-                                                 const uint32_t op_index, KernelBuilder &builder)
+std::unique_ptr<Kernel> build_kernel_CircleGelu(const luci::CircleNode *circle_node,
+                                                KernelBuilderHelper &helper)
 {
-  assert(inputs.size() == 2);
+  const auto *node = loco::must_cast<const luci::CircleGelu *>(circle_node);
+  assert(node->arity() == 1);
+  const Tensor *input = helper.getInputTensor(node->features());
+  Tensor *output = helper.getOutputTensor(node);
 
-  const Tensor *input = inputs.at(0);
-  const Tensor *alpha = inputs.at(1);
-  Tensor *output = outputs.at(0);
+  GeluParams params{};
+  params.approximate = node->approximate();
 
-  return std::make_unique<kernels::PRelu>(input, alpha, output);
+  return std::make_unique<kernels::Gelu>(input, output, params);
 }
 
 } // namespace luci_interpreter