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

Make the tf2xla "tensor list size not set" error message a bit more ergonomic

Browse Source 
PiperOrigin-RevId: 312916547
Change-Id: Idbbe406a35205a0fb6dc5e620e04cf3bccefa43d
This commit is contained in:
Sanjoy Das 2020-05-23 09:43:17 -07:00 committed by TensorFlower Gardener
parent 73b9acd143
commit df5e319d05
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tensorflow/compiler/tf2xla/kernels/tensor_list_ops.cc
View File

@ -136,8 +136,11 @@ class TensorListReserveOp : public XlaOpKernel {
OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &num_elements));
OP_REQUIRES(
ctx, num_elements >= 0,
errors::InvalidArgument("XLA compilation requires a fixed tensor list "
"size. Set the number of elements."));
errors::InvalidArgument(
"XLA compilation requires a fixed tensor list size. Set the number "
"of elements. This could also happen if you're using a TensorArray "
"in a while loop that does not have its maximum_iteration set, you "
"can fix this by setting maximum_iteration to a suitable value."));
// If element shape is compile time constant and it's not "unknown rank"
// shape (-1), create an initialized TensorList. Otherwise create an
@ -197,10 +200,13 @@ class EmptyTensorListOp : public XlaOpKernel {
void Compile(XlaOpKernelContext* ctx) override {
int64 max_num_elements;
OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &max_num_elements));
OP_REQUIRES(
ctx, max_num_elements >= 0,
errors::InvalidArgument("XLA compilation requires a fixed tensor list "
"size. Set the max number of elements."));
OP_REQUIRES(ctx, max_num_elements >= 0,
errors::InvalidArgument(
"XLA compilation requires a fixed tensor list size. Set "
"the max number of elements. This could also happen if "
"you're using a TensorArray in a while loop that does not "
"have its maximum_iteration set, you can fix this by "
"setting maximum_iteration to a suitable value."));
if (dtype_ != DT_VARIANT) {
// We are creating a non-nested TensorList.