trim

Samsung · Jan 15, 2024 · 99ec326 · 99ec326
1 parent 3e4bd34
commit 99ec326
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 12 deletions.
diff --git a/compute/cker/include/cker/train/operation/ReLU6.h b/compute/cker/include/cker/train/operation/ReLU6.h
@@ -41,7 +41,7 @@ inline void ReLU6Grad(const Shape &output_shape, const float *output_data,
       incoming_map.array() *
       (0.0f < output_map.array() && output_map.array() < 6.0f).template cast<float>();
   else
-    throw std::runtime_error("cker::ReLUGrad: Unsupported shape");
+    throw std::runtime_error("cker::ReLU6Grad: Unsupported shape");
 }
 
 } // namespace train

diff --git a/compute/cker/src/train/Relu6.test.cc b/compute/cker/src/train/Relu6.test.cc
@@ -29,17 +29,6 @@ using namespace nnfw::cker;
 template <typename T> class Relu6OpVerifier
 {
 public:
-  void verifyForward(const std::vector<T> &input, const std::vector<T> &expected_output)
-  {
-    assert(input.size() == expected_output.size());
-
-    std::vector<T> calc_output(input.size()); // calcuated output
-    ReLU6(Shape{static_cast<int>(input.size())}, input.data(), calc_output.data());
-
-    for (size_t i = 0; i < calc_output.size(); ++i)
-      ASSERT_EQ(expected_output[i], calc_output[i]);
-  }
-
   void verifyBackward(const std::vector<T> &output, const std::vector<T> &input_bwd,
                       const std::vector<T> &expected_output_bwd, bool expect_eq = true)
   {