1 files changed, 14 insertions, 20 deletions
diff --git a/tensorflow/tensorboard/components/tf_graph_app/BUILD b/tensorflow/tensorboard/components/tf_graph_app/BUILD
index 415b20598e..af56889382 100644
--- a/tensorflow/tensorboard/components/tf_graph_app/BUILD
+++ b/tensorflow/tensorboard/components/tf_graph_app/BUILD
@@ -1,11 +1,11 @@
 package(default_visibility = ["//tensorflow:internal"])
 
-load("@io_bazel_rules_closure//closure:defs.bzl", "web_library")
 load("//tensorflow/tensorboard:defs.bzl", "tensorboard_webcomponent_library")
+load("//tensorflow/tensorboard:web.bzl", "ts_web_library")
 
 licenses(["notice"])  # Apache 2.0
 
-web_library(
+ts_web_library(
     name = "tf_graph_app",
     srcs = [
         "index.html",
@@ -16,34 +16,28 @@ web_library(
         "//tensorflow/tensorboard/components/tf_graph_board",
         "//tensorflow/tensorboard/components/tf_graph_controls",
         "//tensorflow/tensorboard/components/tf_graph_loader",
-        "@org_polymer",
+        "//tensorflow/tensorboard/components/tf_imports:polymer",
+        "//tensorflow/tensorboard/components/tf_imports:webcomponentsjs",
         "@org_polymer_iron_component_page",
-        "@org_polymer_webcomponentsjs",
     ],
 )
 
-filegroup(
-    name = "all_files",
-    srcs = glob(["**"]),
-    tags = ["notsan"],
-)
-
-################################################################################
-# MARKED FOR DELETION
-
 tensorboard_webcomponent_library(
     name = "legacy",
-    srcs = [
-        "index.html",
-        "tf-graph-app.html",
-    ],
+    srcs = [":tf_graph_app"],
     destdir = "tf-graph-app",
     deps = [
         "//tensorflow/tensorboard/components/tf_graph_board:legacy",
         "//tensorflow/tensorboard/components/tf_graph_controls:legacy",
         "//tensorflow/tensorboard/components/tf_graph_loader:legacy",
-        "//third_party/javascript/polymer/v1/iron-list:lib",
-        "//third_party/javascript/polymer/v1/paper-radio-group:lib",
-        "//third_party/javascript/polymer/v1/paper-tooltip:lib",
+        "//third_party/javascript/polymer/v1/iron-component-page:lib",
+        "//third_party/javascript/polymer/v1/polymer:lib",
+        "//third_party/javascript/polymer/v1/webcomponentsjs:lib",
     ],
 )
+
+filegroup(
+    name = "all_files",
+    srcs = glob(["**"]),
+    tags = ["notsan"],
+)