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`	A `namedtuple` alias for field number 0
`true_block`	A `namedtuple` alias for field number 1
`false_block`	A `namedtuple` alias for field number 2

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 2022-02-15 UTC.

Args