diff options
Diffstat (limited to 'tensorflow/core/framework/unique_tensor_references.cc')
-rw-r--r-- | tensorflow/core/framework/unique_tensor_references.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/framework/unique_tensor_references.cc b/tensorflow/core/framework/unique_tensor_references.cc index 2ac6431c54..ab33d9ede6 100644 --- a/tensorflow/core/framework/unique_tensor_references.cc +++ b/tensorflow/core/framework/unique_tensor_references.cc @@ -33,7 +33,7 @@ UniqueTensorReferences::~UniqueTensorReferences() { void UniqueTensorReferences::Add(const Tensor& tensor) { DCHECK(!frozen_); // Do nothing if the tensor has a null buffer. - if (tensor.IsInitialized()) { + if (tensor.IsInitialized() && tensor.NumElements() > 0) { if (referenced_tensors_set_ != nullptr) { // There are enough tensors that we are using a hash set to // de-duplicate. |