Operation

Type Members

1. case class ATenEmbeddingBag(weight: Register, input: Register, offsets: Register, output: Register, mode: Mode) extends Operation with Product with Serializable

   

2. case class ATenLayerNorm(input: Register, weight: Register, bias: Register, output: Register, normalizedShape: List[Long], eps: Double) extends Operation with Product with Serializable

   

3. case class BatchNormalizationOp(data: Register, scale: Register, bias: Register, mean: Register, variance: Register, epsilon: Float, momentum: Float, output: Register) extends Operation with Product with Serializable

   

4. sealed abstract class BinOp extends Operation

   

   this is shared code for all the basic binary operators on tensors

5. case class CastOp(input: Register, castTo: DataType, output: Register) extends Operation with Product with Serializable

   

6. case class ConcatOp(inputs: NonEmptyList[Register], axis: Long, output: Register) extends Operation with Product with Serializable

   

7. case class ConstantOfShape(input: Register, value: Option[U], output: Register) extends Operation with Product with Serializable

   

8. case class ConstantOp(t: Unknown, output: Register) extends Operation with Product with Serializable

   

9. case class DropoutOp(input: Register, output: Register, maskOutput: Register, ratio: Float) extends Operation with Product with Serializable

   

10. sealed abstract class FloatMapOp extends Operation

    

11. case class GatherOp(data: Register, indices: Register, output: Register, axis: Long) extends Operation with Product with Serializable

    

12. case class GemmOp(a: Register, b: Register, c: Register, alpha: Float, beta: Float, transA: Boolean, transB: Boolean, output: Register) extends Operation with Product with Serializable

    

13. case class IdentityOp(input: Register, output: Register) extends Operation with Product with Serializable

    

14. case class MaxPoolOp(autoPad: AutoPad, ceilMode: Boolean, dilations: Option[List[Long]], kernelShape: List[Long], pads: Option[List[Long]], storageOrder: StorageOrder, strides: List[Long], input: Register, output: Register, indices: Option[Register]) extends Operation with Product with Serializable

    

15. case class NonZeroOp(input: Register, output: Register) extends Operation with Product with Serializable

    

    This is the NonZero operator which computes the indices of the non-zero elements in the input tensor.

    This is the NonZero operator which computes the indices of the non-zero elements in the input tensor.

    Important note: Since the ONNX spec and PyTorch differ here, we are returning the transposed result in order to match what PyTorch computes.

16. case class ReduceSumOp(axes: Option[List[Long]], keepDims: Boolean, input: Register, output: Register) extends Operation with Product with Serializable

    

17. case class ReshapeOp(input: Register, shape: Register, output: Register) extends Operation with Product with Serializable

    

18. case class ShapeOp(input: Register, output: Register) extends Operation with Product with Serializable

    

19. case class SliceOp(data: Register, starts: Register, ends: Register, axes: Option[Register], steps: Option[Register], output: Register) extends Operation with Product with Serializable

    

20. case class SoftmaxOp(input: Register, axis: Int, output: Register) extends Operation with Product with Serializable

    

21. case class SqueezeOp(input: Register, axes: List[Long], output: Register) extends Operation with Product with Serializable

    

22. case class TransposeOp(permOpt: Option[List[Long]], input: Register, output: Register) extends Operation with Product with Serializable

    

23. case class UnsqueezeOp(input: Register, axes: List[Long], output: Register) extends Operation with Product with Serializable