Trait

org.platanios.tensorflow.api.ops.training.distribute.strategies

DistributionContext

Related Doc: package strategies

Permalink

sealed trait DistributionContext extends AnyRef

Represents a distribution context (e.g., in-tower or cross-tower).

Distribution contexts are used to constrain what actions are allowed at every point in the code, and enforce those constraints at compile time, using implicits. More specifically, the current distribution context is always available implicitly and can be checked (e.g., some method require an implicit cross-tower context and will not be able to compile if there current context is an in-tower context).

For example, for the following execution steps:

  1. Start in the default (single-tower) tower context (i.e., implicit in-tower context is available). 2. Switch to a cross-tower context, when entering a distribution strategy scope (this will make available a cross-tower context implicit, for the all code within that scope). 3. Switch to a (non-default) tower context inside forEachTower(fn, ...) (i.e., the code in fn will have an in-tower implicit context available). 4. If fn calls currentTowerContext->mergeCall(mergeFn, ...), then inside mergeFn, a cross-tower context will again be implicitly available.

Note that you can also go directly from step 1 to 4 to switch to a cross-tower context for the default distribution strategy. You may also switch from the cross-tower context of 4 to an in-tower context by calling forEachTower(), jumping back to step 3.

Most distribution API methods may only be executed in cross-tower contexts.

Linear Supertypes
Known Subclasses
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. DistributionContext
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Abstract Value Members

  1. abstract def scope[R](block: ⇒ R): R

    Permalink
  2. abstract val strategy: DistributionStrategy

    Permalink

    Current distribution strategy.

Concrete Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  5. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  6. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  7. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  9. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  10. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  11. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  12. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  13. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  14. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  15. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  16. def toString(): String

    Permalink
    Definition Classes
    AnyRef → Any
  17. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  18. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  19. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from AnyRef

Inherited from Any

Ungrouped