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

Don't concatenate empty tensors

Browse Source 
PiperOrigin-RevId: 310468853
Change-Id: I30c5006d172c49d1203e92eda7c040f80ca4a1ed
This commit is contained in:
Mihai Maruseac 2020-05-07 17:28:51 -07:00 committed by TensorFlower Gardener
parent 1a03912767
commit cab22e9d46
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/kernels/list_kernels.h
View File

@ -435,8 +435,10 @@ class TensorListConcat : public OpKernel {
for (int i = 0; i < tensor_list->tensors().size(); i++) { for (int i = 0; i < tensor_list->tensors().size(); i++) {
const Tensor& element_tensor = tensor_list->tensors()[i]; const Tensor& element_tensor = tensor_list->tensors()[i];
if (element_tensor.dtype() != DT_INVALID) { if (element_tensor.dtype() != DT_INVALID) {
inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix( if (element_tensor.NumElements() > 0) {
element_tensor.shaped<T, 2>({1, element_tensor.NumElements()}))); inputs_flat.emplace_back(new typename TTypes<T, 2>::ConstMatrix(
element_tensor.shaped<T, 2>({1, element_tensor.NumElements()})));
}
} else { } else {
AllocatorAttributes attr; AllocatorAttributes attr;
if (element_dtype_ == DT_VARIANT) { if (element_dtype_ == DT_VARIANT) {