intermediate

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 = DataFormat.NCHW)(implicit evidence$3: ClassTag[T]) extends IROperator[T] with Product with Serializable

    

14. case class IRSpatialBatchNormalization[T](nOutput: Int, eps: Double = 1e-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 = DataFormat.NCHW, 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 = DataFormat.NCHW)(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 = DataFormat.NCHW)(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 = DataFormat.NCHW, 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 = DataFormat.NCHW)(implicit evidence$5: ClassTag[T]) extends IROperator[T] with Product with Serializable