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

Add a bit of missing handle data type information to TensorList ops

Browse Source 
PiperOrigin-RevId: 337513633
Change-Id: I999c031de1092a3aa38909830db58dcb875971a4
This commit is contained in:
Allen Lavoie 2020-10-16 08:52:07 -07:00 committed by TensorFlower Gardener
parent 4c5d73cb63
commit 96ecf9cf33
1 changed files with 4 additions and 4 deletions

8
tensorflow/core/ops/list_ops.cc
View File

@ -69,7 +69,7 @@ REGISTER_OP("EmptyTensorList")
0, &element_shape));
c->set_output_handle_shapes_and_types(
0, std::vector<shape_inference::ShapeAndType>{
{element_shape, element_dtype}});
{element_shape, element_dtype, ST_TENSOR_LIST}});
return Status::OK();
});
@ -106,7 +106,7 @@ REGISTER_OP("TensorListPushBack")
}
c->set_output_handle_shapes_and_types(
0, std::vector<shape_inference::ShapeAndType>{
{element_shape, element_dtype}});
{element_shape, element_dtype, ST_TENSOR_LIST}});
return Status::OK();
});
@ -153,7 +153,7 @@ REGISTER_OP("TensorListPushBackBatch")
}
c->set_output_handle_shapes_and_types(
0, std::vector<shape_inference::ShapeAndType>{
{element_shape, element_dtype}});
{element_shape, element_dtype, ST_TENSOR_LIST}});
return Status::OK();
});
@ -345,7 +345,7 @@ REGISTER_OP("TensorListSplit")
&element_shape_from_tensor_shape));
c->set_output_handle_shapes_and_types(
0, std::vector<shape_inference::ShapeAndType>{
{element_shape, element_dtype}});
{element_shape, element_dtype, ST_TENSOR_LIST}});
return Status::OK();
});