experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Decouple TFRT native ops from c_api_tfrt.

Browse Source 
Enable three python tests.

PiperOrigin-RevId: 335484903
Change-Id: Ie4c07cdaff53ccedd54e216cbf72312a60905487
This commit is contained in:
Chuanhao Zhuge 2020-10-05 13:17:12 -07:00 committed by TensorFlower Gardener
parent 9323ca984a
commit 0931bdf88f
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/BUILD
View File

@ -2194,7 +2194,7 @@ py_library(
)
# Including this as a dependency will result in tests to use TFRT.
# TODO(b/153582383): Move tf_ops_alwayslink dependency to c_api_tfrt instead.
# TODO(b/170139514): Add a curated list of TFRT native ops.
py_library(
name = "is_tfrt_test_true",
srcs = ["framework/is_tfrt_test_true.py"],
@ -5066,6 +5066,7 @@ cuda_py_test(
size = "medium",
srcs = ["ops/gradient_checker_v2_test.py"],
python_version = "PY3",
tfrt_enabled = True,
deps = [
":array_ops",
":client_testlib",

2
tensorflow/python/kernel_tests/BUILD
View File

@ -817,6 +817,7 @@ cuda_py_test(
name = "matrix_solve_ls_op_test",
size = "medium",
srcs = ["matrix_solve_ls_op_test.py"],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",
@ -3608,6 +3609,7 @@ cuda_py_test(
"no_oss", # b/117185141.
"nomsan", # TODO(b/117236102): Re-enable in msan build.
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",