TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfp.experimental.auto_batching.instructions.BranchOp

View source on GitHub

Class BranchOp

A conditional.

Aliases:

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.

__new__

__new__(
    _cls,
    cond_var,
    true_block,
    false_block
)

Create new instance of BranchOp(cond_var, true_block, false_block)

Properties

cond_var

true_block

false_block