com.intel.analytics.bigdl.nn

Module

object Module

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. Module
  2. AnyRef
  3. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Value Members

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

    Definition Classes
    AnyRef
  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any
  3. final def ##(): Int

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

    Definition Classes
    AnyRef
  5. final def ==(arg0: Any): Boolean

    Definition Classes
    Any
  6. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  7. def clone(): AnyRef

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

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

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

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  11. def flatten[T](parameters: Array[Tensor[T]])(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Tensor[T]

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

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

    Definition Classes
    AnyRef → Any
  14. def isCompact[T](parameters: Array[Tensor[T]])(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Tensor[T]

  15. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  16. def loadCaffeModel[T](defPath: String, modelPath: String)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): AbstractModule[Activity, Activity, T]

    Loaf caffe trained model from prototxt and weight files

    Loaf caffe trained model from prototxt and weight files

    defPath

    caffe model definition file path

    modelPath

    caffe model binary file containing weight and bias

  17. def loadModule[T](path: String, weightPath: String = null)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): AbstractModule[Activity, Activity, T]

    Load model from path.

    Load model from path.

    T

    numeric type

    path

    path to save module, local file system, HDFS and Amazon S3 is supported. HDFS path should be like "hdfs://[host]:[port]/xxx" Amazon S3 path should be like "s3a://bucket/xxx"

    weightPath

    : where weight is stored

    returns

    model loaded from path

  18. def loadTF[T](graphFile: String, inputs: Seq[String], outputs: Seq[String], byteOrder: ByteOrder = ByteOrder.LITTLE_ENDIAN, binFile: Option[String] = None, generatedBackward: Boolean = true)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Module[T]

    Load tensorflow model from its saved protobuf file.

    Load tensorflow model from its saved protobuf file.

    graphFile

    where is the protobuf model file

    inputs

    input node names

    outputs

    output node names, the output tensor order is same with the node order

    byteOrder

    byte order in the tensorflow file. The default value is little endian

    binFile

    where is the model variable file

    generatedBackward

    if generate backward graph

    returns

    BigDL model

  19. def loadTorch[T](path: String)(implicit arg0: ClassTag[T]): AbstractModule[Activity, Activity, T]

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

    Definition Classes
    AnyRef
  21. final def notify(): Unit

    Definition Classes
    AnyRef
  22. final def notifyAll(): Unit

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

    Definition Classes
    AnyRef
  24. def tensorflowCheckpoints[T](graphFile: String, binFile: String, byteOrder: ByteOrder = ByteOrder.LITTLE_ENDIAN)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Session[T]

    Load tensorflow checkpoints

    Load tensorflow checkpoints

    T
    graphFile
    binFile
    returns

  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
    @throws( ... )

Deprecated Value Members

  1. def load[T](path: String)(implicit arg0: ClassTag[T]): AbstractModule[Activity, Activity, T]

    Load model from path.

    Load model from path.

    T

    numeric type

    path

    path to save module, local file system, HDFS and Amazon S3 is supported. HDFS path should be like "hdfs://[host]:[port]/xxx" Amazon S3 path should be like "s3a://bucket/xxx"

    returns

    model loaded from path

    Annotations
    @deprecated
    Deprecated

    (Since version 0.3) Java based serialization not recommended any more, please use loadModule instead

  2. def loadCaffe[T](model: AbstractModule[Activity, Activity, T], defPath: String, modelPath: String, matchAll: Boolean = true)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): AbstractModule[Activity, Activity, T]

    Annotations
    @deprecated
    Deprecated

    (Since version 0.2) Please try to use the loadCaffeModel API

Inherited from AnyRef

Inherited from Any

Ungrouped