tfp.experimental.auto_batching.instructions.Block

A basic block.

View aliases

Main aliases

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

tfp.experimental.auto_batching.instructions.Block(
    instructions=None, terminator=None, name=None
)

Args
instructions	A list of PrimOp, PopOp, and FunctionCallOp instructions to execute in order. Control transfer instructions (that do not return) are not permitted in this list.
terminator	A single BranchOp, GotoOp, PushGotoOp or IndirectGotoOp, indicating how to transfer control out of this basic block.
name	An object serving as the name of this Block, for display.

Attributes
label_str	A string suitable for referring to this Block in printed output.

Methods

assign_instructions

View source

assign_instructions(
    instructions
)

Assigns the body instructions and the terminator at once.

This is a convenience method, to set a Block's program content in one invocation instead of having to assign the instructions and the terminator fields separately.

Args
instructions	A non-empty Python list of Op objects. The last one must be a BranchOp, GotoOp, PushGotoOp, or IndirectGotoOp, and becomes the terminator. The others, if any, must be PrimOp, PopOp, or FunctionCallOp, and become the instructions, in order.