A control flow graph (CFG).
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 Block s in the same CFG. Otherwise,
downstream passes or staging may fail.
|
Methods
block
View source
block(
index
)
Returns the Block
given by the input index
.
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".