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

Disable DLPack python test for TFRT.

Browse Source 
PiperOrigin-RevId: 351656527
Change-Id: Ifff1b67a025bdb881cce36512745fdafdffde333
This commit is contained in:
Chuanhao Zhuge 2021-01-13 13:42:29 -08:00 committed by TensorFlower Gardener
parent 1486708ff6
commit e513f24d41
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/dlpack/BUILD
View File

@ -19,6 +19,9 @@ cuda_py_test(
name = "dlpack_test", name = "dlpack_test",
srcs = ["dlpack_test.py"], srcs = ["dlpack_test.py"],
srcs_version = "PY2AND3", srcs_version = "PY2AND3",
tags = [
"no_tfrt", # TODO(b/177356372)
],
deps = [ deps = [
":dlpack", ":dlpack",
"//tensorflow/python/eager:test", "//tensorflow/python/eager:test",