Attend the Women in ML Symposium on December 7 Register now

tfp.experimental.auto_batching.instructions.BranchOp

Stay organized with collections Save and categorize content based on your preferences.

A conditional.

cond_var The string name of the VM variable holding the condition. This variable must have boolean dtype and scalar data shape.
true_block The Block where to transfer logical threads whose condition is True.
false_block The Block where to transfer logical threads whose condition is False.

cond_var A namedtuple alias for field number 0
true_block A namedtuple alias for field number 1
false_block A namedtuple alias for field number 2