diff --git a/tensorflow/core/kernels/list_kernels.h b/tensorflow/core/kernels/list_kernels.h index 855506e9d8a..37fc1b3ae08 100644 --- a/tensorflow/core/kernels/list_kernels.h +++ b/tensorflow/core/kernels/list_kernels.h @@ -435,8 +435,10 @@ class TensorListConcat : public OpKernel { for (int i = 0; i < tensor_list->tensors().size(); i++) { const Tensor& element_tensor = tensor_list->tensors()[i]; if (element_tensor.dtype() != DT_INVALID) { - inputs_flat.emplace_back(new typename TTypes::ConstMatrix( - element_tensor.shaped({1, element_tensor.NumElements()}))); + if (element_tensor.NumElements() > 0) { + inputs_flat.emplace_back(new typename TTypes::ConstMatrix( + element_tensor.shaped({1, element_tensor.NumElements()}))); + } } else { AllocatorAttributes attr; if (element_dtype_ == DT_VARIANT) {