Use a forward declaration to avoid tpu_system_interface depending on c_api.

tensorflower-gardener · tensorflower-gardener · commit c4840e9f15cc · 2022-05-02T13:02:19.000-07:00
(interface should not depend on implementation)

PiperOrigin-RevId: 446007965
diff --git a/tensorflow/dtensor/cc/BUILD b/tensorflow/dtensor/cc/BUILD
@@ -142,7 +142,6 @@ cc_library(
     srcs = ["tpu_system_interface.cc"],
     hdrs = ["tpu_system_interface.h"],
     deps = [
-        "//tensorflow/c/eager:c_api",
         "//tensorflow/core:framework",
         "//tensorflow/core:lib",
         "@com_google_absl//absl/time",
diff --git a/tensorflow/dtensor/cc/tpu_system_interface.h b/tensorflow/dtensor/cc/tpu_system_interface.h
@@ -19,12 +19,14 @@ limitations under the License.
 #include <vector>
 
 #include "absl/time/time.h"
-#include "tensorflow/c/eager/c_api.h"
 #include "tensorflow/core/framework/op_kernel.h"
 #include "tensorflow/core/framework/resource_mgr.h"
 #include "tensorflow/core/platform/logging.h"
 #include "tensorflow/core/platform/status.h"
 
+// Forward declare TFE_Context to avoid interface depending on c_api.
+typedef struct TFE_Context TFE_Context;
+
 namespace tensorflow {
 namespace dtensor {
 
