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. final def evaluateImage(imageFrame: ImageFrame, vMethods: Array[_ <: ValidationMethod[Float]], batchSize: Option[Int]): Array[(ValidationResult, ValidationMethod[Float])]

    Definition Classes
    AbstractModule
  21. def finalize(): Unit

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

    Definition Classes
    AbstractModule
  23. var forwardTime: Long

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

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

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

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

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

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

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

    Definition Classes
    InferShape
  31. def getParametersTable(): Table

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

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

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

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

    Definition Classes
    AbstractModule
  36. final def getTimesGroupByModuleType(): Array[(String, Long, Long)]

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

    Definition Classes
    AbstractModule
  38. var gradInput: Activity

    Definition Classes
    AbstractModule
  39. final def hasName: Boolean

    Definition Classes
    AbstractModule
  40. def hashCode(): Int

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

    the input tensor names of this subgraph

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

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

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

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

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

    Definition Classes
    AbstractModule
  47. var line: String

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

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

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

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

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

    Definition Classes
    AnyRef
  53. var output: Activity

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

    the output tensor names of this subgraph

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

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

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

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

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

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

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

    Definition Classes
    AbstractModule
  62. def release(): Unit

    Definition Classes
    AbstractModule
  63. def reset(): Unit

    Definition Classes
    AbstractModule
  64. def resetTimes(): Unit

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

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

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

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

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

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

    Definition Classes
    AbstractModule
  71. var scaleB: Double

    Attributes
    protected
    Definition Classes
    AbstractModule
  72. var scaleW: Double

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

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

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

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

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

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

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

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

    Definition Classes
    AbstractModule
  81. def toString(): String

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

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

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

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

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

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

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  90. 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