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

[TFRT] Register OpHandler for all CPU devices.

Browse Source 
This enables EagerOpHandlerSelector to select the correct requested CPU device, example:

with ops.device('CPU:1')
  some_op()

disable a test for TFRT which is falsely passing before.

PiperOrigin-RevId: 356854011
Change-Id: I5e8d7b98570b9ef00ab51644744b567a62ed1e84
This commit is contained in:
Chuanhao Zhuge 2021-02-10 16:27:17 -08:00 committed by TensorFlower Gardener
parent d99168f592
commit 84f074f68d
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/keras/mixed_precision/BUILD
View File

@ -146,6 +146,9 @@ tf_py_test(
size = "medium", size = "medium",
srcs = ["autocast_variable_test.py"], srcs = ["autocast_variable_test.py"],
python_version = "PY3", python_version = "PY3",
tags = [
"no_tfrt", # TODO(b/179863362)
],
deps = [ deps = [
":autocast_variable", ":autocast_variable",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",