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

Rollback code introduced in #38848

Browse Source 
PiperOrigin-RevId: 309839045
Change-Id: I5d017c5200b6953ea267a1b77fa2a30a775af968
This commit is contained in:
Mihai Maruseac 2020-05-04 16:20:17 -07:00 committed by TensorFlower Gardener
parent d47e86444a
commit 57f6650fbb
2 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/variable_ops.cc
View File

@ -230,7 +230,7 @@ TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNEL);
.HostMemory("is_initialized"), \
IsVariableInitializedOp);
TF_CALL_GPU_ALL_TYPES(REGISTER_GPU_KERNELS);
TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNELS);
TF_CALL_int64(REGISTER_GPU_KERNELS);
TF_CALL_uint32(REGISTER_GPU_KERNELS);
#undef REGISTER_GPU_KERNELS

8
tensorflow/python/kernel_tests/variable_ops_test.py
View File

@ -33,13 +33,10 @@ from tensorflow.python.ops import variables
from tensorflow.python.platform import test
_NP_TO_TF = {
np.float16: dtypes.float16,
np.float32: dtypes.float32,
np.float64: dtypes.float64,
np.int32: dtypes.int32,
np.int64: dtypes.int64,
np.complex64: dtypes.complex64,
np.complex128: dtypes.complex128,
}
@ -53,10 +50,7 @@ class VariableOpTest(test.TestCase):
return self.evaluate(p)
def _testTypes(self, vals):
for dtype in [
np.float16, np.float32, np.float64, np.complex64, np.complex128,
np.int32, np.int64
]:
for dtype in [np.float32, np.float64, np.int32, np.int64]:
self.setUp()
x = vals.astype(dtype)
tftype = _NP_TO_TF[dtype]