diff --git a/tensorflow/lite/python/BUILD b/tensorflow/lite/python/BUILD
index 8333fa418c2..7248792523e 100644
--- a/tensorflow/lite/python/BUILD
+++ b/tensorflow/lite/python/BUILD
@@ -143,7 +143,6 @@ py_test(
     shard_count = 4,
     srcs_version = "PY2AND3",
     tags = [
-        "no_rocm",
         "no_windows",
     ],
     deps = [
@@ -160,7 +159,6 @@ py_test(
     python_version = "PY3",
     srcs_version = "PY2AND3",
     tags = [
-        "no_rocm",
         "no_windows",
     ],
     deps = [
diff --git a/tensorflow/tools/ci_build/linux/rocm/run_cc_core.sh b/tensorflow/tools/ci_build/linux/rocm/run_cc_core.sh
index f1912c512a7..b69431a938e 100755
--- a/tensorflow/tools/ci_build/linux/rocm/run_cc_core.sh
+++ b/tensorflow/tools/ci_build/linux/rocm/run_cc_core.sh
@@ -52,8 +52,7 @@ bazel test \
       -- \
       //tensorflow/... \
       -//tensorflow/compiler/... \
-      -//tensorflow/lite/delegates/gpu/gl/... \
-      -//tensorflow/lite/delegates/gpu/cl/... \
+      -//tensorflow/lite/... \
 && bazel test \
       --config=rocm \
       -k \
diff --git a/tensorflow/tools/ci_build/linux/rocm/run_py3_core.sh b/tensorflow/tools/ci_build/linux/rocm/run_py3_core.sh
index b255789b865..d55fa56f970 100755
--- a/tensorflow/tools/ci_build/linux/rocm/run_py3_core.sh
+++ b/tensorflow/tools/ci_build/linux/rocm/run_py3_core.sh
@@ -51,4 +51,5 @@ bazel test \
       --run_under=//tensorflow/tools/ci_build/gpu_build:parallel_gpu_execute \
       -- \
       //tensorflow/... \
-      -//tensorflow/compiler/...
+      -//tensorflow/compiler/... \
+      -//tensorflow/lite/...