GitHub

tfp.experimental.auto_batching.instructions.BranchOp

A conditional.

Main aliases

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

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2021-06-19 UTC.