tfp.experimental.auto_batching.instructions.BranchOp

A conditional.

Main aliases

tfp.experimental.auto_batching.frontend.instructions.BranchOp, tfp.experimental.auto_batching.frontend.st.inst.BranchOp, tfp.experimental.auto_batching.frontend.stack.inst.BranchOp, tfp.experimental.auto_batching.stack_optimization.inst.BranchOp, tfp.experimental.auto_batching.stackless.inst.BranchOp

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