com.intel.analytics.bigdl.utils

intermediate

package intermediate

Visibility
  1. Public
  2. All

Type Members

  1. abstract class ConvertBase[T, D] extends AnyRef

  2. case class IRCAddTable[T, D](inplace: Boolean = false)(implicit evidence$13: ClassTag[T], evidence$14: ClassTag[D]) extends IROperator[T] with Product with Serializable

  3. case class IRConcatTable[T]()(implicit evidence$16: ClassTag[T]) extends IROperator[T] with Product with Serializable

  4. case class IRGeneralModule[T](model: AbstractModule[Activity, Activity, T])(implicit evidence$18: ClassTag[T]) extends IROperator[T] with Product with Serializable

    if blas module has no corresponding IROperator, then we can use IRGeneralModule to wrap this layer to IROperator

  5. case class IRIdentity[T]()(implicit evidence$7: ClassTag[T]) extends IROperator[T] with Product with Serializable

  6. case class IRInput[T]()(implicit evidence$17: ClassTag[T]) extends IROperator[T] with Product with Serializable

  7. case class IRJoinTable[T](dimension: Int, nInputDims: Int = 0)(implicit evidence$15: ClassTag[T]) extends IROperator[T] with Product with Serializable

  8. case class IRLinear[T](inputSize: Int, outputSize: Int, withBias: Boolean = true, wRegularizer: Regularizer[T] = null, bRegularizer: Regularizer[T] = null, initWeight: Tensor[T] = null, initBias: Tensor[T] = null, initGradWeight: Tensor[T] = null, initGradBias: Tensor[T] = null)(implicit evidence$9: ClassTag[T]) extends IROperator[T] with Product with Serializable

  9. sealed class IROperator[T] extends Serializable

  10. case class IRReLU[T](ip: Boolean = false)(implicit evidence$8: ClassTag[T]) extends IROperator[T] with Product with Serializable

  11. case class IRSelectTable[T](dimension: Int)(implicit evidence$12: ClassTag[T]) extends IROperator[T] with Product with Serializable

  12. case class IRSoftMax[T]()(implicit evidence$11: ClassTag[T]) extends IROperator[T] with Product with Serializable

  13. case class IRSpatialAveragePooling[T](kW: Int, kH: Int, dW: Int = 1, dH: Int = 1, padW: Int = 0, padH: Int = 0, globalPooling: Boolean = false, ceilMode: Boolean = false, countIncludePad: Boolean = true, divide: Boolean = true, format: DataFormat = ...)(implicit evidence$3: ClassTag[T]) extends IROperator[T] with Product with Serializable

  14. case class IRSpatialBatchNormalization[T](nOutput: Int, eps: Double = 1.0E-5, momentum: Double = 0.1, affine: Boolean = true, initWeight: Tensor[T] = null, initBias: Tensor[T] = null, initGradWeight: Tensor[T] = null, initGradBias: Tensor[T] = null, dataFormat: DataFormat = ..., runningMean: Tensor[T] = null, runningVar: Tensor[T] = null)(implicit evidence$6: ClassTag[T]) extends IROperator[T] with Product with Serializable

  15. case class IRSpatialConvolution[T](nInputPlane: Int, nOutputPlane: Int, kernelW: Int, kernelH: Int, strideW: Int = 1, strideH: Int = 1, padW: Int = 0, padH: Int = 0, nGroup: Int = 1, propagateBack: Boolean = true, wRegularizer: Regularizer[T] = null, bRegularizer: Regularizer[T] = null, initWeight: Tensor[T] = null, initBias: Tensor[T] = null, initGradWeight: Tensor[T] = null, initGradBias: Tensor[T] = null, withBias: Boolean = true, format: DataFormat = ...)(implicit evidence$4: ClassTag[T]) extends IROperator[T] with Product with Serializable

  16. case class IRSpatialCrossMapLRN[T](size: Int = 5, alpha: Double = 1.0, beta: Double = 0.75, k: Double = 1.0, format: DataFormat = ...)(implicit evidence$10: ClassTag[T]) extends IROperator[T] with Product with Serializable

  17. case class IRSpatialMaxPooling[T](kW: Int, kH: Int, dW: Int = 1, dH: Int = 1, padW: Int = 0, padH: Int = 0, format: DataFormat = ..., ceilMode: Boolean = false)(implicit evidence$2: ClassTag[T]) extends IROperator[T] with Product with Serializable

  18. case class IRSpatialShareConvolution[T](nInputPlane: Int, nOutputPlane: Int, kernelW: Int, kernelH: Int, strideW: Int = 1, strideH: Int = 1, padW: Int = 0, padH: Int = 0, nGroup: Int = 1, propagateBack: Boolean = true, wRegularizer: Regularizer[T] = null, bRegularizer: Regularizer[T] = null, initWeight: Tensor[T] = null, initBias: Tensor[T] = null, initGradWeight: Tensor[T] = null, initGradBias: Tensor[T] = null, withBias: Boolean = true, format: DataFormat = ...)(implicit evidence$5: ClassTag[T]) extends IROperator[T] with Product with Serializable

Ungrouped