tfp.experimental.auto_batching.instructions.BranchOp

A conditional.

View aliases

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

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

Args
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.

Attributes
cond_var
true_block
false_block