description: A conditional.
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
|
|