From 72b01b5ca054f85fe042a9941576ea8edf347f1f Mon Sep 17 00:00:00 2001 From: Yuan Yu Date: Tue, 8 Nov 2016 12:44:49 -0800 Subject: Give a friendly error message if parallel_iterations is set to be less than 1. Change: 138551385 --- tensorflow/python/ops/control_flow_ops.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow/python/ops/control_flow_ops.py b/tensorflow/python/ops/control_flow_ops.py index ec66532afb..1efc2d5380 100644 --- a/tensorflow/python/ops/control_flow_ops.py +++ b/tensorflow/python/ops/control_flow_ops.py @@ -2572,6 +2572,7 @@ def while_loop(cond, body, loop_vars, shape_invariants=None, `Tensor`, and `TensorArray` objects. shape_invariants: The shape invariants for the loop variables. parallel_iterations: The number of iterations allowed to run in parallel. + It must be a positive integer. back_prop: Whether backprop is enabled for this while loop. swap_memory: Whether GPU-CPU memory swap is enabled for this loop. name: Optional name prefix for the returned tensors. @@ -2625,6 +2626,8 @@ def while_loop(cond, body, loop_vars, shape_invariants=None, raise TypeError("cond must be callable.") if not callable(body): raise TypeError("body must be callable.") + if parallel_iterations < 1: + raise TypeError("parallel_iterations must be a positive integer.") if shape_invariants is not None: nest.assert_same_structure(loop_vars, shape_invariants) -- cgit v1.2.3