com.intel.analytics.bigdl.utils.tf

TensorflowSaver

object TensorflowSaver

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. TensorflowSaver
  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. 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 register(className: String, saver: BigDLToTensorflow): Unit

    Register a customized BigDL module saver.

    Register a customized BigDL module saver.

    className

    class name of the BigDL module

    saver

    customized saver

  18. def saveGraph[T](model: Graph[T], inputs: Seq[(String, Seq[Int])], path: String, byteOrder: ByteOrder = ByteOrder.LITTLE_ENDIAN, dataFormat: TensorflowDataFormat = TensorflowDataFormat.NHWC)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Unit

    Save a graph model to protobuf files so that it can be used in tensorflow inference.

    Save a graph model to protobuf files so that it can be used in tensorflow inference.

    When save the model, placeholders will be added to the tf model as input nodes. So you need to pass in the names and shape for the placeholders. BigDL model doesn't have such information. The order of the placeholder information should be same as the inputs of the graph model

    T

    model

    graph model instance

    inputs

    placeholder information

    path

    where to save

    byteOrder

    model byte order

    dataFormat

    model data format

  19. def saveGraphWithNodeDef[T](model: Graph[T], inputs: Seq[NodeDef], path: String, byteOrder: ByteOrder = ByteOrder.LITTLE_ENDIAN, extraNodes: Set[NodeDef] = Set()): Unit

    Save a graph model to protobuf files so that it can be used in tensorflow inference.

    Save a graph model to protobuf files so that it can be used in tensorflow inference.

    When save the model, placeholders will be added to the tf model as input nodes. So you need to pass in the names and shape for the placeholders. BigDL model doesn't have such information. The order of the placeholder information should be same as the inputs of the graph model

    T

    model

    graph model instance

    inputs

    input node defs

    path

    where to save

    byteOrder

    model byte order

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

    Definition Classes
    AnyRef
  21. def toString(): String

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

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from AnyRef

Inherited from Any

Ungrouped