Trait

com.intel.analytics.bigdl.nn.mkldnn

MklDnnModule

Related Doc: package mkldnn

Permalink

trait MklDnnModule extends MklDnnModuleHelper

Helper utilities when integrating Module with MKL-DNN

Linear Supertypes
MklDnnModuleHelper, MemoryOwner, AnyRef, Any
Known Subclasses
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

    Permalink

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. implicit lazy val _this: MemoryOwner

    Permalink
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
  5. final def asInstanceOf[T0]: T0

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

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

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

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

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

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

    Permalink
    Definition Classes
    AnyRef → Any
  12. def initActivity(formats: Array[MemoryData]): Activity

    Permalink
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
  13. def initTensor(format: MemoryData): Tensor[_]

    Permalink
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
  14. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  15. def nativeData(formats: Array[MemoryData]): Array[MemoryData]

    Permalink
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
  16. final def ne(arg0: AnyRef): Boolean

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

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

    Permalink
    Definition Classes
    AnyRef
  19. def registerResource(m: Releasable): Unit

    Permalink
    Definition Classes
    MemoryOwner
  20. def releaseResources(): Unit

    Permalink
    Definition Classes
    MemoryOwner
  21. var runtime: MklDnnRuntime

    Permalink

    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
  22. def setRuntime(runtime: MklDnnRuntime): Unit

    Permalink
  23. def singleNativeData(formats: Array[MemoryData]): Array[MemoryData]

    Permalink
    Attributes
    protected
    Definition Classes
    MklDnnModuleHelper
  24. final def synchronized[T0](arg0: ⇒ T0): T0

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

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

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

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

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

Inherited from MklDnnModuleHelper

Inherited from MemoryOwner

Inherited from AnyRef

Inherited from Any

Ungrouped