tfp.experimental.auto_batching.instructions.BranchOp

A conditional.

tfp.experimental.auto_batching.instructions.BranchOp(
    cond_var, true_block, false_block
)

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