com.intel.analytics.zoo.pipeline.api.net

TFNet

class TFNet extends AbstractModule[Activity, Activity, Float]

TFNet wraps a tensorflow subgraph as a layer, and use tensorflow to calculate the layer's output.

This subgraph should not contain any tensorflow Variable and the input/output must be numeric types

When used with other layers for training, there should be no trainable layer before this one, as the gradInput of this layer is always zero.

Linear Supertypes
AbstractModule[Activity, Activity, Float], InferShape, Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. TFNet
  2. AbstractModule
  3. InferShape
  4. Serializable
  5. Serializable
  6. AnyRef
  7. 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. def accGradParameters(input: Activity, gradOutput: Activity): Unit

    Definition Classes
    AbstractModule
  7. def apply(name: String): Option[AbstractModule[Activity, Activity, Float]]

    Definition Classes
    AbstractModule
  8. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  9. def backward(input: Activity, gradOutput: Activity): Activity

    Definition Classes
    AbstractModule
  10. var backwardTime: Long

    Attributes
    protected
    Definition Classes
    AbstractModule
  11. def clearState(): TFNet.this.type

    Definition Classes
    AbstractModule
  12. final def clone(deepCopy: Boolean): AbstractModule[Activity, Activity, Float]

    Definition Classes
    AbstractModule
  13. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  14. final def cloneModule(): AbstractModule[Activity, Activity, Float]

    Definition Classes
    AbstractModule
  15. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  16. def equals(other: Any): Boolean

    Definition Classes
    AbstractModule → AnyRef → Any
  17. final def evaluate(dataSet: LocalDataSet[MiniBatch[Float]], vMethods: Array[_ <: ValidationMethod[Float]]): Array[(ValidationResult, ValidationMethod[Float])]

    Definition Classes
    AbstractModule
  18. final def evaluate(dataset: RDD[Sample[Float]], vMethods: Array[_ <: ValidationMethod[Float]], batchSize: Option[Int]): Array[(ValidationResult, ValidationMethod[Float])]

    Definition Classes
    AbstractModule
  19. def evaluate(): TFNet.this.type

    Definition Classes
    AbstractModule
  20. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  21. final def forward(input: Activity): Activity

    Definition Classes
    AbstractModule
  22. var forwardTime: Long

    Attributes
    protected
    Definition Classes
    AbstractModule
  23. def freeze(names: String*): TFNet.this.type

    Definition Classes
    AbstractModule
  24. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  25. def getExtraParameter(): Array[Tensor[Float]]

    Definition Classes
    AbstractModule
  26. final def getInputShape(): Shape

    Definition Classes
    InferShape
  27. final def getName(): String

    Definition Classes
    AbstractModule
  28. final def getNumericType(): TensorDataType

    Definition Classes
    AbstractModule
  29. final def getOutputShape(): Shape

    Definition Classes
    InferShape
  30. def getParametersTable(): Table

    Definition Classes
    AbstractModule
  31. final def getPrintName(): String

    Attributes
    protected
    Definition Classes
    AbstractModule
  32. final def getScaleB(): Double

    Definition Classes
    AbstractModule
  33. final def getScaleW(): Double

    Definition Classes
    AbstractModule
  34. def getTimes(): Array[(AbstractModule[_ <: Activity, _ <: Activity, Float], Long, Long)]

    Definition Classes
    AbstractModule
  35. final def getWeightsBias(): Array[Tensor[Float]]

    Definition Classes
    AbstractModule
  36. var gradInput: Activity

    Definition Classes
    AbstractModule
  37. final def hasName: Boolean

    Definition Classes
    AbstractModule
  38. def hashCode(): Int

    Definition Classes
    AbstractModule → AnyRef → Any
  39. val inputNames: Seq[String]

    the input tensor names of this subgraph

  40. def inputs(first: (ModuleNode[Float], Int), nodesWithIndex: (ModuleNode[Float], Int)*): ModuleNode[Float]

    Definition Classes
    AbstractModule
  41. def inputs(nodes: Array[ModuleNode[Float]]): ModuleNode[Float]

    Definition Classes
    AbstractModule
  42. def inputs(nodes: ModuleNode[Float]*): ModuleNode[Float]

    Definition Classes
    AbstractModule
  43. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  44. final def isTraining(): Boolean

    Definition Classes
    AbstractModule
  45. var line: String

    Attributes
    protected
    Definition Classes
    AbstractModule
  46. final def loadModelWeights(srcModel: Module[Float], matchAll: Boolean): TFNet.this.type

    Definition Classes
    AbstractModule
  47. final def loadWeights(weightPath: String, matchAll: Boolean): TFNet.this.type

    Definition Classes
    AbstractModule
  48. final def ne(arg0: AnyRef): Boolean

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

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

    Definition Classes
    AnyRef
  51. var output: Activity

    Definition Classes
    AbstractModule
  52. val outputNames: Seq[String]

    the output tensor names of this subgraph

  53. def parameters(): (Array[Tensor[Float]], Array[Tensor[Float]])

    Definition Classes
    TFNet → AbstractModule
  54. final def predict(dataset: RDD[Sample[Float]], batchSize: Int, shareBuffer: Boolean): RDD[Activity]

    Definition Classes
    AbstractModule
  55. final def predictClass(dataset: RDD[Sample[Float]], batchSize: Int): RDD[Int]

    Definition Classes
    AbstractModule
  56. final def predictImage(imageFrame: ImageFrame, outputLayer: String, shareBuffer: Boolean, batchPerPartition: Int, predictKey: String, featurePaddingParam: Option[PaddingParam[Float]]): ImageFrame

    Definition Classes
    AbstractModule
  57. def processInputs(first: (ModuleNode[Float], Int), nodesWithIndex: (ModuleNode[Float], Int)*): ModuleNode[Float]

    Attributes
    protected
    Definition Classes
    AbstractModule
  58. def processInputs(nodes: Seq[ModuleNode[Float]]): ModuleNode[Float]

    Attributes
    protected
    Definition Classes
    AbstractModule
  59. final def quantize(): Module[Float]

    Definition Classes
    AbstractModule
  60. def reset(): Unit

    Definition Classes
    AbstractModule
  61. def resetTimes(): Unit

    Definition Classes
    AbstractModule
  62. final def saveCaffe(prototxtPath: String, modelPath: String, useV2: Boolean, overwrite: Boolean): TFNet.this.type

    Definition Classes
    AbstractModule
  63. final def saveDefinition(path: String, overWrite: Boolean): TFNet.this.type

    Definition Classes
    AbstractModule
  64. final def saveModule(path: String, weightPath: String, overWrite: Boolean): TFNet.this.type

    Definition Classes
    AbstractModule
  65. final def saveTF(inputs: Seq[(String, Seq[Int])], path: String, byteOrder: ByteOrder, dataFormat: TensorflowDataFormat): TFNet.this.type

    Definition Classes
    AbstractModule
  66. final def saveTorch(path: String, overWrite: Boolean): TFNet.this.type

    Definition Classes
    AbstractModule
  67. final def saveWeights(path: String, overWrite: Boolean): Unit

    Definition Classes
    AbstractModule
  68. var scaleB: Double

    Attributes
    protected
    Definition Classes
    AbstractModule
  69. var scaleW: Double

    Attributes
    protected
    Definition Classes
    AbstractModule
  70. final def setExtraParameter(extraParam: Array[Tensor[Float]]): TFNet.this.type

    Definition Classes
    AbstractModule
  71. final def setLine(line: String): TFNet.this.type

    Definition Classes
    AbstractModule
  72. final def setName(name: String): TFNet.this.type

    Definition Classes
    AbstractModule
  73. def setScaleB(b: Double): TFNet.this.type

    Definition Classes
    AbstractModule
  74. def setScaleW(w: Double): TFNet.this.type

    Definition Classes
    AbstractModule
  75. final def setWeightsBias(newWeights: Array[Tensor[Float]]): TFNet.this.type

    Definition Classes
    AbstractModule
  76. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  77. def toGraph(startNodes: ModuleNode[Float]*): Graph[Float]

    Definition Classes
    AbstractModule
  78. def toString(): String

    Definition Classes
    AbstractModule → AnyRef → Any
  79. var train: Boolean

    Attributes
    protected
    Definition Classes
    AbstractModule
  80. def training(): TFNet.this.type

    Definition Classes
    AbstractModule
  81. def unFreeze(names: String*): TFNet.this.type

    Definition Classes
    AbstractModule
  82. def updateGradInput(input: Activity, gradOutput: Activity): Activity

    Definition Classes
    TFNet → AbstractModule
  83. def updateOutput(input: Activity): Activity

    Definition Classes
    TFNet → AbstractModule
  84. final def wait(): Unit

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  87. final def zeroGradParameters(): Unit

    Definition Classes
    AbstractModule

Deprecated Value Members

  1. final def save(path: String, overWrite: Boolean): TFNet.this.type

    Definition Classes
    AbstractModule
    Annotations
    @deprecated
    Deprecated

    (Since version 0.3.0) please use recommended saveModule(path, overWrite)

Inherited from AbstractModule[Activity, Activity, Float]

Inherited from InferShape

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped