Package

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

values

Permalink

package values

Visibility
  1. Public
  2. All

Type Members

  1. class DistributedValue[+T] extends AnyRef

    Permalink

    Holds a map from devices to values.

  2. trait DistributedVariable extends Serializable

    Permalink

    Holds a map from devices to variables.

  3. class MirroredValue[+T] extends DistributedValue[T]

    Permalink

    Represents a synchronous value mirrored among multiple devices.

  4. class MirroredVariable extends MirroredValue[variables.Variable] with DistributedVariable

    Permalink

    Holds a map from devices to synchronized variables.

    Holds a map from devices to synchronized variables.

    NOTE: We use MirroredVariable.getUpdateDevice for the assignment methods to enforce that we are in an update scope. The arguments to update() are automatically unwrapped and so the update() function would normally see regular variables and not mirrored variables. However, the update() function can still operate on wrapped mirrored variables through object members, captured arguments, etc. This is more likely in an updateNonSlot() scope, which can update several non-slot variables in one call.

  5. class PerDeviceValue[+T] extends DistributedValue[T]

    Permalink

    Represents a asynchronous value distributed among multiple devices.

  6. class PerDeviceVariable extends PerDeviceValue[variables.Variable] with DistributedVariable

    Permalink

    Holds a map from devices to variables whose values are reduced on save.

Ungrouped