intermediate

Type Members

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

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

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

4. case class IRDropout[T](initP: Double = 0.5, inplace: Boolean = false, scale: Boolean = true)(implicit evidence$8: ClassTag[T]) extends IROperator[T] with Product with Serializable

5. case class IRGeneralModule[T](model: AbstractModule[Activity, Activity, T])(implicit evidence$20: 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

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

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

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

9. 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$10: ClassTag[T]) extends IROperator[T] with Product with Serializable

10. sealed class IROperator[T] extends Serializable

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

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

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

14. 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

15. 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

16. 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

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

18. 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

19. 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

20. case class IRSqueeze[T](dims: Array[Int], batchMode: Boolean)(implicit evidence$11: ClassTag[T]) extends IROperator[T] with Product with Serializable