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

Return an error if we cannot create a tensor

Browse Source 
PiperOrigin-RevId: 304715417
Change-Id: If41b187af1b38311c6d5e462ae9a2e4d420dd15b
This commit is contained in:
Gaurav Jain 2020-04-03 16:28:35 -07:00 committed by TensorFlower Gardener
parent 280ddc4abf
commit 52edc1cc02
1 changed files with 6 additions and 0 deletions

6
tensorflow/python/lib/core/py_seq_tensor.cc
View File

@ -288,8 +288,14 @@ struct Converter {
Py_DECREF(scalar);
if (*error != nullptr) return errors::InvalidArgument(*error);
t = ConverterTraits<T>::CreateScalar(ctx, value);
if (t == nullptr) {
return errors::Internal("Cannot create tensor.");
}
} else {
t = ConverterTraits<T>::CreateTensor(ctx, state->inferred_shape);
if (t == nullptr) {
return errors::Internal("Cannot create tensor.");
}
if (t->NumElements() > 0) {
T* buf = static_cast<T*>(t->Data());
*error = Helper(obj, 0, state, &buf);