fix: replace forbidden assert() with TF_LITE_ASSERT()

tensorflow · Aug 2, 2024 · 4784ada · 4784ada
1 parent df9852a
commit 4784ada
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tensorflow/lite/micro/BUILD b/tensorflow/lite/micro/BUILD
@@ -411,6 +411,9 @@ cc_library(
     name = "static_vector",
     hdrs = ["static_vector.h"],
     copts = micro_copts(),
+    deps = [
+        "//tensorflow/lite/kernels:op_macros",
+    ],
 )
 
 cc_library(

diff --git a/tensorflow/lite/micro/static_vector.h b/tensorflow/lite/micro/static_vector.h
@@ -19,6 +19,8 @@
 #include <cassert>
 #include <cstddef>
 
+#include "tensorflow/lite/kernels/op_macros.h"  // for TF_LITE_ASSERT
+
 namespace tflite {
 
 template <typename T, std::size_t MaxSize>
@@ -67,7 +69,7 @@ class StaticVector {
   }
 
   void push_back(const T& t) {
-    assert(!full());
+    TF_LITE_ASSERT(!full());
     *end() = t;
     ++size_;
   }