tf.contrib.framework.smart_cond

Return either true_fn() if predicate pred is true else false_fn().

If pred is a bool or has a constant value, we return either true_fn() or false_fn(), otherwise we use tf.cond to dynamically route to both.

Arguments
pred A scalar determining whether to return the result of true_fn or false_fn.
true_fn The callable to be performed if pred is true.
false_fn The callable to be performed if pred is false.
name Optional name prefix when using tf.cond.
Returns
Tensors returned by the call to either true_fn or false_fn.
Raises
TypeError If true_fn or false_fn is not callable.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/framework/smart_cond