Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfp.experimental.auto_batching.instructions.IndirectGotoOp

View source on GitHub

Jump to the address in the reserved "program counter" variable.

@staticmethod
tfp.experimental.auto_batching.instructions.IndirectGotoOp(
    _cls
)

Use to return from a function call.

Also restores the previous saved program counter (under the one jumped to), enforcing proper nesting with invocations of PushGotoOp.