tfp.experimental.auto_batching.instructions.BranchOp

View source on GitHub

A conditional.

@staticmethod
tfp.experimental.auto_batching.instructions.BranchOp(
    _cls, 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