diff options
Diffstat (limited to 'tensorflow/core/kernels/random_op.cc')
-rw-r--r-- | tensorflow/core/kernels/random_op.cc | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tensorflow/core/kernels/random_op.cc b/tensorflow/core/kernels/random_op.cc index f3c7e0f26b..3063fedac8 100644 --- a/tensorflow/core/kernels/random_op.cc +++ b/tensorflow/core/kernels/random_op.cc @@ -303,10 +303,7 @@ class RandomGammaOp : public OpKernel { &samples_shape)); } const int64 num_samples = samples_shape.num_elements(); - OP_REQUIRES(ctx, num_samples > 0, - errors::InvalidArgument( - "Input shape should have non-zero element count, got: ", - num_samples)); + if (num_samples == 0) return; samples_shape.AppendShape(alpha_t.shape()); // Allocate output samples. |