tfp.experimental.auto_batching.instructions.BranchOp

View source on GitHub

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

true_block

false_block