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

Minor error message improvement for incompatible type conversion.

Browse Source 
PiperOrigin-RevId: 222282332
This commit is contained in:
Akshay Modi 2018-11-20 11:52:19 -08:00 committed by TensorFlower Gardener
parent bf8544fccf
commit a8ae0f47e5
3 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/eager/pywrap_tensor.cc
View File

@ -439,8 +439,8 @@ int EagerTensor_init(EagerTensor* self, PyObject* args, PyObject* kwds) {
PyErr_SetString( PyErr_SetString(
PyExc_TypeError, PyExc_TypeError,
tensorflow::strings::StrCat( tensorflow::strings::StrCat(
"Cannot convert value ", TFE_GetPythonString(value_str.get()), "Cannot convert provided value to EagerTensor. Provided value: ",
" to EagerTensor with requested dtype: ", TFE_GetPythonString(value_str.get()), " Requested dtype: ",
tensorflow::DataTypeString( tensorflow::DataTypeString(
static_cast<tensorflow::DataType>(desired_dtype))) static_cast<tensorflow::DataType>(desired_dtype)))
.c_str()); .c_str());

6
tensorflow/python/eager/pywrap_tfe_src.cc
View File

@ -2303,8 +2303,10 @@ bool ConvertToTensor(
PyErr_SetString( PyErr_SetString(
PyExc_TypeError, PyExc_TypeError,
tensorflow::strings::StrCat( tensorflow::strings::StrCat(
"Cannot convert value ", TFE_GetPythonString(input_str.get()), "Cannot convert provided value to EagerTensor. Provided value: ",
" to EagerTensor with requested dtype: ", desired_dtype) TFE_GetPythonString(input_str.get()), " Requested dtype: ",
tensorflow::DataTypeString(
static_cast<tensorflow::DataType>(desired_dtype)))
.c_str()); .c_str());
return false; return false;
} }

8
tensorflow/python/eager/tensor_test.py
View File

@ -323,6 +323,14 @@ class TFETensorTest(test_util.TensorFlowTestCase):
def testConvertToTensorAllowsOverflow(self): def testConvertToTensorAllowsOverflow(self):
_ = ops.convert_to_tensor(123456789, dtype=dtypes.uint8) _ = ops.convert_to_tensor(123456789, dtype=dtypes.uint8)
def testEagerTensorError(self):
with self.assertRaisesRegexp(
TypeError,
"Cannot convert provided value to EagerTensor. "
"Provided value.*Requested dtype.*"):
_ = ops.convert_to_tensor(1., dtype=dtypes.int32)
class TFETensorUtilTest(test_util.TensorFlowTestCase): class TFETensorUtilTest(test_util.TensorFlowTestCase):