Class/Object

org.platanios.tensorflow.api.learn.layers.rnn.cell

DeviceWrapper

Related Docs: object DeviceWrapper | package cell

Permalink

class DeviceWrapper[O, OS, S, SS] extends RNNCell[O, OS, S, SS]

RNN cell that ensures another RNN cell runs on a specific device.

Linear Supertypes
RNNCell[O, OS, S, SS], Layer[Tuple[O, S], Tuple[O, S]], AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. DeviceWrapper
  2. RNNCell
  3. Layer
  4. AnyRef
  5. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new DeviceWrapper(name: String, cell: RNNCell[O, OS, S, SS], device: String = "", deviceFunction: (OpSpecification) ⇒ String = _.device)(implicit evO: Aux[O, OS], evS: Aux[S, SS])

    Permalink

    name

    Name scope (also acting as variable scope) for this layer.

    cell

    RNN cell being wrapped.

    device

    Device to use.

    deviceFunction

    Device function to use.

Value Members

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

    Permalink
    Definition Classes
    AnyRef → Any

  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  3. def +(other: Layer[Tuple[O, S], Tuple[O, S]]): Concatenate[Tuple[O, S], Tuple[O, S]]

    Permalink
    Definition Classes
    Layer

  4. def ++(others: Seq[Layer[Tuple[O, S], Tuple[O, S]]]): Concatenate[Tuple[O, S], Tuple[O, S]]

    Permalink
    Definition Classes
    Layer

  5. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  6. def >>[S](other: Layer[Tuple[O, S], S]): Compose[Tuple[O, S], Tuple[O, S], S]

    Permalink
    Definition Classes
    Layer

  7. def apply(input: Tuple[O, S])(implicit mode: Mode): Tuple[O, S]

    Permalink
    Definition Classes
    Layer

  8. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any

  9. val cell: RNNCell[O, OS, S, SS]

    Permalink

    RNN cell being wrapped.

  10. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  11. def compose[S](other: Layer[Tuple[O, S], S]): Compose[Tuple[O, S], Tuple[O, S], S]

    Permalink
    Definition Classes
    Layer

  12. def concatenate(others: Layer[Tuple[O, S], Tuple[O, S]]*): Concatenate[Tuple[O, S], Tuple[O, S]]

    Permalink
    Definition Classes
    Layer

  13. final def createCell(mode: Mode, inputShape: OS): ops.rnn.cell.RNNCell[O, OS, S, SS]

    Permalink
    Definition Classes
    RNNCell

  14. def createCellWithoutContext(mode: Mode, inputShape: OS): ops.rnn.cell.RNNCell[O, OS, S, SS]

    Permalink
    Definition Classes
    DeviceWrapper → RNNCell

  15. final def currentStep: ops.Output

    Permalink
    Definition Classes
    Layer

  16. val device: String

    Permalink

    Device to use.

  17. val deviceFunction: (OpSpecification) ⇒ String

    Permalink

    Device function to use.

  18. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  19. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  20. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )

  21. def forward(input: Tuple[O, S])(implicit mode: Mode): Tuple[O, S]

    Permalink
    Definition Classes
    Layer

  22. final def forwardWithoutContext(input: Tuple[O, S])(implicit mode: Mode): Tuple[O, S]

    Permalink
    Definition Classes
    RNNCell → Layer

  23. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any

  24. final def getParameter(name: String, dataType: types.DataType, shape: core.Shape, initializer: Initializer = null, regularizer: Regularizer = null, trainable: Boolean = true, reuse: Reuse = ReuseOrCreateNew, collections: Set[Key[ops.variables.Variable]] = Set.empty, cachingDevice: (OpSpecification) ⇒ String = null): ops.Output

    Permalink
    Definition Classes
    Layer

  25. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  26. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any

  27. val layerType: String

    Permalink
    Definition Classes
    DeviceWrapper → Layer

  28. def map[MR](mapFn: (Tuple[O, S]) ⇒ MR): Layer[Tuple[O, S], MR]

    Permalink
    Definition Classes
    Layer

  29. val name: String

    Permalink

    Name scope (also acting as variable scope) for this layer.

    Name scope (also acting as variable scope) for this layer.

    Definition Classes
    DeviceWrapper → RNNCell → Layer

  30. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  31. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef

  32. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef

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

    Permalink
    Definition Classes
    AnyRef

  34. def toString(): String

    Permalink
    Definition Classes
    Layer → AnyRef → Any

  35. final def wait(): Unit

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

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

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

  37. final def wait(arg0: Long): Unit

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

Inherited from RNNCell[O, OS, S, SS]

Inherited from Layer[Tuple[O, S], Tuple[O, S]]

Inherited from AnyRef

Inherited from Any

Ungrouped