trait MklDnnModule extends MklDnnModuleHelper

Helper utilities when integrating Module with MKL-DNN

Linear Supertypes
MklDnnModuleHelper, MemoryOwner, AnyRef, Any
Known Subclasses
AvgPooling, CAddTable, ConcatTable, DnnGraph, Dropout, Identity, Input, JoinTable, LRN, Linear, MaxPooling, MklDnnContainer, MklDnnLayer, Output, RNN, ReLU, ReorderMemory, SelectTable, Sequential, SoftMax, SpatialBatchNormalization, SpatialConvolution
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. MklDnnModule
  2. MklDnnModuleHelper
  3. MemoryOwner
  4. AnyRef
  5. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Abstract Value Members

  1. abstract def setQuantize(value: Boolean): MklDnnModule.this.type

Concrete Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. implicit lazy val _this: MemoryOwner
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
    Annotations
    @transient()
  5. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  6. def clone(): AnyRef
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )
  7. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  8. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  9. def finalize(): Unit
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  10. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  11. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  12. def initActivity(formats: Array[MemoryData]): Activity
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
  13. def initTensor(format: MemoryData): Tensor[_]
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
  14. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  15. def nativeData(formats: Array[MemoryData]): Array[MemoryData]
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
  16. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  17. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  18. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  19. def registerResource(m: Releasable): Unit
    Definition Classes
    MemoryOwner
  20. def releaseResources(): Unit
    Definition Classes
    MemoryOwner
  21. val runtime: MklDnnRuntime

    MklDnn runtime, which includes a MKL-DNN engine and a MKL-DNN stream.

    MklDnn runtime, which includes a MKL-DNN engine and a MKL-DNN stream. Note that this instance will be erased when send to remote worker, so you should recreate a MklDnnRuntime.

    Attributes
    protected
    Annotations
    @transient()
  22. def setRuntime(runtime: MklDnnRuntime): Unit
  23. def singleNativeData(formats: Array[MemoryData]): Array[MemoryData]
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
  24. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  25. def toString(): String
    Definition Classes
    AnyRef → Any
  26. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  27. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  28. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )

Inherited from MklDnnModuleHelper

Inherited from MemoryOwner

Inherited from AnyRef

Inherited from Any

Ungrouped