com.intel.analytics.bigdl.utils.tf

BigDLSessionImpl

class BigDLSessionImpl[T] extends Session[T]

Linear Supertypes
Session[T], AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. BigDLSessionImpl
  2. Session
  3. AnyRef
  4. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new BigDLSessionImpl(graph: Seq[NodeDef], context: Context[T], byteOrder: ByteOrder = java.nio.ByteOrder.LITTLE_ENDIAN)(implicit arg0: ClassTag[T], ev: TensorNumeric[T])

Type Members

  1. type DataCache = HashMap[String, Array[Seq[Table]]]

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. final def getClass(): Class[_]

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

    Definition Classes
    AnyRef → Any
  13. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  14. final def ne(arg0: AnyRef): Boolean

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

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

    Definition Classes
    AnyRef
  17. def predict(endPoints: Seq[String], isDataBatch: Boolean, batchSize: Int, sc: SparkContext): RDD[Activity]

    Predict data with tensorflow graph.

    Predict data with tensorflow graph. The data must hold in a queue

    endPoints
    isDataBatch

    if the model input is the batch

    batchSize

    batch size, which should be original batch size * total core number

    sc
    returns

    Definition Classes
    BigDLSessionImplSession
  18. def saveParameters(binFile: String): BigDLSessionImpl.this.type

    Dump varaible contents to a file

    Dump varaible contents to a file

    binFile
    returns

    Definition Classes
    BigDLSessionImplSession
  19. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  20. def toString(): String

    Definition Classes
    AnyRef → Any
  21. def train(endPoints: Seq[String], optMethod: OptimMethod[T], endWhen: Trigger, isDataBatch: Boolean, batchSize: Int, sc: SparkContext, loss: Option[String]): BigDLSessionImpl.this.type

    Train the tensorflow graph.

    Train the tensorflow graph. The model must be fed data with a queue

    endPoints
    optMethod
    endWhen
    isDataBatch

    if the model input is the batch

    batchSize

    batch size, which should be original batch size * total core number

    sc
    loss
    returns

    Definition Classes
    BigDLSessionImplSession
  22. def train(outputs: Seq[String], dataSet: DistributedDataSet[MiniBatch[T]], optMethod: OptimMethod[T], criterion: Criterion[T], endWhen: Trigger): Graph[T]

    Train the tensorflow graph

    Train the tensorflow graph

    outputs
    dataSet
    optMethod
    criterion
    endWhen
    returns

    Definition Classes
    BigDLSessionImplSession
  23. final def wait(): Unit

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from Session[T]

Inherited from AnyRef

Inherited from Any

Ungrouped