XLAControlFlowContext

abstract def add(output: Output): Output

Adds output to the current context and its outer context recursively.

abstract def backPropagate: Boolean

Returns true if back-propagation is supported for this control flow context.

abstract def gradientLoopState: Option[GradientLoopState]

Gradient loop state for this context, used for back-propagation.

abstract val name: String

Name of this control flow context.

final def !=(arg0: Any): Boolean

final def ##(): Int

final def ==(arg0: Any): Boolean

def add(op: Op): Unit

Adds op to the current context.

final def asInstanceOf[T0]: T0

def clone(): AnyRef

def condContext: Option[CondContext]

Returns the cond context containing this context.

def controlPivot: Option[Op]

Returns the control pivot op output for this context.

def enter(): Unit

Enters this control flow context.

def enterGradientColocation(colocationOps: Set[Op], gradientUID: String): Unit

Enters a control flow context for building a gradient colocated with colocationOps.

final def eq(arg0: AnyRef): Boolean

def equals(arg0: Any): Boolean

def exit(): Unit

Exits this control flow context.

def exitGradientColocation(colocationOps: Set[Op], gradientUID: String): Unit

Exits a control flow context for building a gradient colocated with colocationOps.

def exitResult(result: Seq[OutputLike]): Unit

Makes a sequence of tensors available in the outer context.

def finalize(): Unit

final def getClass(): Class[_]

def hashCode(): Int

final def isInstanceOf[T0]: Boolean

final def ne(arg0: AnyRef): Boolean

final def notify(): Unit

final def notifyAll(): Unit

val outerContext: Option[Context]

Control flow context containing this context.

final def synchronized[T0](arg0: ⇒ T0): T0

def toProto(exportScope: String): GeneratedMessageV3

Alias for toValuesDef.

def toProto: GeneratedMessageV3

Converts this object to its corresponding ProtoBuf object.

def toString(): String

def toValuesDef(exportScope: String = null): ValuesDef

Constructs and returns a ValuesDef object that represents this control flow context.

final def wait(): Unit

final def wait(arg0: Long, arg1: Int): Unit

final def wait(arg0: Long): Unit

def whileLoopContext(stopContext: Option[Context] = None): Option[WhileLoopContext]

Returns the while context containing this context.

def xlaContext: Option[XLAControlFlowContext]

Returns the XLA context containing this context.