From ff1fc1d1cb6e7a833b998801f8038f80a475902b Mon Sep 17 00:00:00 2001
From: yuht6 <yuhaitao6@huawei.com>
Date: Thu, 7 Mar 2024 17:13:31 +0800
Subject: [PATCH] ut add headfile

---
 tf_adapter/tests/st/util/testcase/util_test.cc | 1 +
 tf_adapter/tests/ut/util/testcase/util_test.cc | 1 +
 2 files changed, 2 insertions(+)

diff --git a/tf_adapter/tests/st/util/testcase/util_test.cc b/tf_adapter/tests/st/util/testcase/util_test.cc
index aa046f141..c45ecc249 100644
--- a/tf_adapter/tests/st/util/testcase/util_test.cc
+++ b/tf_adapter/tests/st/util/testcase/util_test.cc
@@ -2,6 +2,7 @@
 #include <vector>
 #include "gtest/gtest.h"
 #include "graph/def_types.h"
+#include "external/graph/types.h"
 
 namespace tensorflow {
 namespace {
diff --git a/tf_adapter/tests/ut/util/testcase/util_test.cc b/tf_adapter/tests/ut/util/testcase/util_test.cc
index aa046f141..c45ecc249 100644
--- a/tf_adapter/tests/ut/util/testcase/util_test.cc
+++ b/tf_adapter/tests/ut/util/testcase/util_test.cc
@@ -2,6 +2,7 @@
 #include <vector>
 #include "gtest/gtest.h"
 #include "graph/def_types.h"
+#include "external/graph/types.h"
 
 namespace tensorflow {
 namespace {
-- 
Gitee