tfp.experimental.auto_batching.instructions.ControlFlowGraph

View source on GitHub

A control flow graph (CFG).

View aliases

Main aliases

tfp.experimental.auto_batching.frontend.instructions.ControlFlowGraph, tfp.experimental.auto_batching.frontend.st.inst.ControlFlowGraph, tfp.experimental.auto_batching.frontend.stack.inst.ControlFlowGraph, tfp.experimental.auto_batching.stack_optimization.inst.ControlFlowGraph, tfp.experimental.auto_batching.stackless.inst.ControlFlowGraph

tfp.experimental.auto_batching.instructions.ControlFlowGraph(
    blocks
)

Args
blocks	Python list of Block objects, the content of the CFG. Any terminator instructions of said Block objects should refer to other Blocks in the same CFG. Otherwise, downstream passes or staging may fail.

Attributes
blocks

Methods

block

View source

block(
    index
)

Returns the Block given by the input index.

Args
index	A Python int.

Returns
block	The Block at that location in the block list.

block_index

View source

block_index(
    block
)

Returns the int index of the given Block.

Args
block	The block to look up. If None, returns the exit index.

Returns
index	Python int, the index of the requested block.

enter_block

View source

enter_block()

Returns the entry Block.

exit_index

View source

exit_index()

Returns the int index denoting "exit this CFG".