object TensorflowSaver
- Alphabetic
- By Inheritance
- TensorflowSaver
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
-
def
clone(): AnyRef
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
equals(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
def
finalize(): Unit
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
def
hashCode(): Int
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
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
-
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
- model
graph model instance
- inputs
placeholder information
- path
where to save
- byteOrder
model byte order
- dataFormat
model data format
-
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
- model
graph model instance
- inputs
input node defs
- path
where to save
- byteOrder
model byte order
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
-
def
toString(): String
- Definition Classes
- AnyRef → Any
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )