Aidge export ARM Cortex-M#

When imported automatically update availables implementations for aidge core objects.

Available implementations#

Supported operators: ✔️ Yes (∇ = grad), ✅ Generic impl., ☑️ MetaOp impl., ❌ No#

Operator (⧖ = custom prod. conso.)

Support
Impl specs
(— = any ; ✱ = additional requirements)
Abs
(aidge_core.Abs)

✅
1. — — → — —
Add
(aidge_core.Add)

✅
1. — —1 → — —1
And
(aidge_core.And)

✅
1. — — → boolean —
ArgMax
(aidge_core.ArgMax)

✅
1. float32 — → int64 —
ArgMin
(aidge_core.ArgMin)

❌
Asin
(aidge_core.Asin)

✅
1. float32 — → float32 —
Atan
(aidge_core.Atan)

❌
AvgPooling1D
(aidge_core.AvgPooling1D)

✅
1. — nwc → — nwc
AvgPooling2D
(aidge_core.AvgPooling2D)

✅
1. — nhwc → — nhwc
AvgPooling3D
(aidge_core.AvgPooling3D)

❌
BatchNorm2D
(aidge_core.BatchNorm2D)

✅
1. float32 nchw → float32 nchw
BitErrorRate
(aidge_core.BitErrorRate)

❌
BitShift
(aidge_core.BitShift)

✅
1. — — → — —
CastLike
(aidge_core.CastLike)

✅
1. — — → — —
Cast
(aidge_core.Cast)

✅
1. — — → — —
Ceil
(aidge_core.Ceil)

❌
Clip
(aidge_core.Clip)

✅
1. — — → — —
ComplexToInnerPair
(aidge_core.ComplexToInnerPair)

✅
1. — — → — —
Concat
(aidge_core.Concat)

✅
1. — — → — —
ConnectedComponentLabeling
(aidge_core.ConnectedComponentLabeling)

❌
ConstantOfShape
(aidge_core.ConstantOfShape)

❌
Conv1D
(aidge_core.Conv1D)

✔️
1. — nwc | — nwc | — — → — nwc
Conv1DAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.Conv.ArmConv1DAct)
1. — nwc | — nwc | — — → — nwc
Conv2D
(aidge_core.Conv2D)

✔️
1. — nhwc | — nhwc | — — → — nhwc
Conv3D
(aidge_core.Conv3D)

❌
ConvAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.Conv.ArmConvAct)
1. — nhwc | — nhwc | — — → — nhwc
ConvDepthWise1D
(aidge_core.ConvDepthWise1D)

❌
ConvDepthWise2D
(aidge_core.ConvDepthWise2D)

✔️
1. — nhwc | — nhwc | — — → — nhwc
ConvDwAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvDw.ArmConvDwAct)
1. — nhwc | — nhwc | — — → — nhwc
ConvTranspose1D
(aidge_core.ConvTranspose1D)

✔️
1. — nwc | — nwc | — — → — nwc
ConvTranspose1DAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvTranspose.ArmConvTranspose1DAct)
1. — nwc | — nwc | — — → — nwc
ConvTranspose2D
(aidge_core.ConvTranspose2D)

✔️
1. — nhwc | — nhwc | — — → — nhwc
ConvTranspose2DAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvTranspose.ArmConvTranspose2DAct)
1. — nhwc | — nhwc | — — → — nhwc
ConvTranspose3D
(aidge_core.ConvTranspose3D)

❌
Cosh
(aidge_core.Cosh)

❌
Cos
(aidge_core.Cos)

❌
CryptoHash
(aidge_core.CryptoHash)

❌
DepthToSpace
(aidge_core.DepthToSpace)

❌
DFT
(aidge_core.DFT)

✅
1. complex64 — → complex64 —
Div
(aidge_core.Div)

✅
1. — —1 → — —1
DropBlock
(aidge_core.DropBlock)

❌
Dropout
(aidge_core.Dropout)

❌
Equal
(aidge_core.Equal)

✅
1. — — → boolean —
2. — — → — —
Erf
(aidge_core.Erf)

✅
1. float32 — → float32 —
Expand
(aidge_core.Expand)

✅
1. — — → — —
Exp
(aidge_core.Exp)

❌
FC
(aidge_core.FC)

✔️
1. — default | — — | — — → — —
2. — nhwc | — — | — — → — —
FCAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.Fc.ArmFCAct)
1. — default | — — | — — → — —
2. — nhwc | — — | — — → — —
FixedNBitFlip
(aidge_core.FixedNBitFlip)

❌
Flatten
(aidge_core.Flatten)

❌
Floor
(aidge_core.Floor)

❌
Fold2D
(aidge_core.Fold2D)

❌
GatherElements
(aidge_core.GatherElements)

❌
GatherND
(aidge_core.GatherND)

❌
Gather
(aidge_core.Gather)

✅
1. — — → — —
GlobalAveragePooling
(aidge_core.GlobalAveragePooling)

✅
1. — nhwc → — nhwc
Greater
(aidge_core.Greater)

✅
1. — — → — —
GreaterOrEqual
(aidge_core.GreaterOrEqual)

✅
1. — — → — —
GridSample
(aidge_core.GridSample)

❌
Hardmax
(aidge_core.Hardmax)

✅
1. float32 — → float32 —
HardSigmoid
(aidge_core.HardSigmoid)

✅
1. float32 — → float32 —
Heaviside
(aidge_core.Heaviside)

❌
Identity
(aidge_core.Identity)

✅
1. — — → — —
InnerPairToComplex
(aidge_core.InnerPairToComplex)

✅
1. float32 — → complex64 —
InstanceNorm
(aidge_core.InstanceNorm)

✅
1. float32 nchw → float32 nchw
2. float32 ncw → float32 ncw
LeakyReLU
(aidge_core.LeakyReLU)

✅
1. — — → — —
Less
(aidge_core.Less)

✅
1. — — → — —
LessOrEqual
(aidge_core.LessOrEqual)

✅
1. — — → — —
Ln
(aidge_core.Ln)

✅
1. float32 — → float32 —
LogSoftmax
(aidge_core.LogSoftmax)

❌
LRN
(aidge_core.LRN)

❌
MatMul
(aidge_core.MatMul)

✅
1. — ncdhw | — ncdhw → — ncdhw
2. — nchw | — nchw → — nchw
3. — chw | — chw → — chw
4. — default | — default → — default
Max
(aidge_core.Max)

✅
1. —1 —1 → —1 —1
MaxPooling1D
(aidge_core.MaxPooling1D)

✅
1. —or_packed_last_1 nwc → —or_packed_last_1 nwc | — nwc
2. — nwc → — nwc
MaxPooling2D
(aidge_core.MaxPooling2D)

✅
1. —or_packed_last_1 nhwc → —or_packed_last_1 nhwc | — nhwc
2. — nhwc → — nhwc
MaxPooling3D
(aidge_core.MaxPooling3D)

❌
Memorize
(aidge_core.Memorize)

✅
1. —1 —1 | —1 —1 → —1 —1 | —1 —1
Min
(aidge_core.Min)

✅
1. —1 —1 → —1 —1
Mod
(aidge_core.Mod)

❌
Mul
(aidge_core.Mul)

✅
1. — —1 → — —1
NBitFlip
(aidge_core.NBitFlip)

❌
Neg
(aidge_core.Neg)

✅
1. — — → — —
NonMaxSuppression
(aidge_core.NonMaxSuppression)

❌
NonZero
(aidge_core.NonZero)

❌
Not
(aidge_core.Not)

✅
1. boolean — → boolean —
OneHot
(aidge_core.OneHot)

❌
Pack
(aidge_core.Pack)

✅
1. — — → —or_packed_last —
PadConv
(aidge_export_arm_cortexm._Aidge_Arm.operators.Conv.ArmPadConv)
1. — nhwc | — nhwc | — — → — nhwc
PadConv1D
(aidge_export_arm_cortexm._Aidge_Arm.operators.Conv.ArmPadConv1D)
1. — nwc | — nwc | — — → — nwc
PadConvAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.Conv.ArmPadConvAct)
1. — nhwc | — nhwc | — — → — nhwc
PadConvDw
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvDw.ArmPadConvDw)
1. — nhwc | — nhwc | — — → — nhwc
PadConvDwAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvDw.ArmPadConvDwAct)
1. — nhwc | — nhwc | — — → — nhwc
PadConvTranspose1D
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvTranspose.ArmPadConvTranspose1D)
1. — nwc | — nwc | — — → — nwc
PadConvTranspose1DAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvTranspose.ArmPadConvTranspose1DAct)
1. — nwc | — nwc | — — → — nwc
PadConvTranspose2D
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvTranspose.ArmPadConvTranspose2D)
1. — nhwc | — nhwc | — — → — nhwc
PadConvTranspose2DAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvTranspose.ArmPadConvTranspose2DAct)
1. — nhwc | — nhwc | — — → — nhwc
PaddedConv1D
(aidge_export_arm_cortexm._Aidge_Arm.operators.Conv.ArmPadConv1D)
1. — nwc | — nwc | — — → — nwc
PaddedConv2D
(aidge_export_arm_cortexm._Aidge_Arm.operators.Conv.ArmPadConv)
1. — nhwc | — nhwc | — — → — nhwc
PaddedConvDepthWise2D
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvDw.ArmPadConvDw)
1. — nhwc | — nhwc | — — → — nhwc
PaddedConvTranspose1D
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvTranspose.ArmPadConvTranspose1D)
1. — nwc | — nwc | — — → — nwc
PaddedConvTranspose2D
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvTranspose.ArmPadConvTranspose2D)
1. — nhwc | — nhwc | — — → — nhwc
Pad
(aidge_core.Pad)

✅
1. — nchw → — nchw
Pop
(aidge_core.Pop)

✅
1. —1 — → —1 —
Pow
(aidge_core.Pow)

✅
1. — —1 → — —1
Producer
(aidge_core.Producer)

✅
1. —or_packed_last — → —or_packed_last —
QFC
(aidge_export_arm_cortexm._Aidge_Arm.operators.Fc.ArmQFC)
1. — default | — — | — — → — —
2. — nhwc | — — | — — → — —
Quantizer
(aidge_export_arm_cortexm._Aidge_Arm.operators.Quantizer.QuantizerArm)
1. — — → — —
RandomNormalLike
(aidge_core.RandomNormalLike)

❌
Range
(aidge_core.Range)

❌
Reciprocal
(aidge_core.Reciprocal)

✅
1. float32 — → float32 —
ReduceMax
(aidge_core.ReduceMax)

✅
1. — — → — —
ReduceMean
(aidge_core.ReduceMean)

✅
1. — — → — —
ReduceMin
(aidge_core.ReduceMin)

✅
1. — — → — —
ReduceSum
(aidge_core.ReduceSum)

✅
1. — — → — —
ReLU
(aidge_core.ReLU)

✅
1. — — → — —
Reshape
(aidge_core.Reshape)

✅
1. — default → — default
Resize
(aidge_core.Resize)

✅
1. — nhwc | — — | — — | — — → — nhwc
Round
(aidge_core.Round)

✅
1. float32 — → float32 —
Scatter
(aidge_core.Scatter)

✅
1. — — → — —
SDLayerNorm
(aidge_core.SDLayerNorm)

✅
1. float32 — → float32 —
2. int4 — → int4 —
3. int8 — → int8 —
4. int16 — → int16 —
5. int32 — → int32 —
SDShiftGELU
(aidge_core.SDShiftGELU)

✅
1. float32 — → float32 —
2. int4 — → int4 —
3. int8 — → int8 —
4. int16 — → int16 —
5. int32 — → int32 —
SDShiftmax
(aidge_core.SDShiftmax)

✅
1. float32 — → float32 —
2. int4 — → int4 —
3. int8 — → int8 —
4. int16 — → int16 —
5. int32 — → int32 —
SDShiftSiLU
(aidge_core.SDShiftSiLU)

✅
1. float32 — → float32 —
2. int4 — → int4 —
3. int8 — → int8 —
4. int16 — → int16 —
5. int32 — → int32 —
Select
(aidge_core.Select)

✅
1. — — → — —
Shape
(aidge_core.Shape)

✅
1. — — → int64 —
Sigmoid
(aidge_core.Sigmoid)

✅
1. — — → — —
Sign
(aidge_core.Sign)

❌
Sinh
(aidge_core.Sinh)

❌
Sin
(aidge_core.Sin)

❌
Size
(aidge_core.Size)

❌
Slice
(aidge_core.Slice)

✅
1. — — → — —
Softmax
(aidge_core.Softmax)

✅
1. float32 — → float32 —
Split
(aidge_core.Split)

✅
1. — — → — —
Sqrt
(aidge_core.Sqrt)

✅
1. float32 — → float32 —
Squeeze
(aidge_core.Squeeze)

✅
1. — — → — —
Stack
(aidge_core.Stack)

✅
1. —1 — | int64 — → —1 —
STFT
(aidge_core.STFT)

❌
Sub
(aidge_core.Sub)

✅
1. — —1 → — —1
Sum
(aidge_core.Sum)

✅
1. —1 —1 → —1 —1
SVMRegressor
(aidge_core.SVMRegressor)

✅
1. float32 — → float32 —
Tanh
(aidge_core.Tanh)

✅
1. — — → — —
Tan
(aidge_core.Tan)

❌
Tile
(aidge_core.Tile)

✅
1. — — → — —
TopK
(aidge_core.TopK)

❌
Transpose
(aidge_core.Transpose)

✅
1. — — → — —
Trilu
(aidge_core.Trilu)

❌
Unfold2D
(aidge_core.Unfold2D)

✅
1. — nchw → — default
Unpack
(aidge_core.Unpack)

✅
1. —or_packed_last — → — —
Unsqueeze
(aidge_core.Unsqueeze)

✅
1. — — → — —
Where
(aidge_core.Where)

✅
1. — — → — —

CMSIS-NN implementations#

Supported operators: ✔️ Yes (∇ = grad), ✅ Generic impl., ☑️ MetaOp impl., ❌ No#

Operator (⧖ = custom prod. conso.)

Support
Impl specs
(— = any ; ✱ = additional requirements)
AvgPoolAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
AvgPooling2D
(aidge_core.AvgPooling2D)

✔️
1. int8 nhwc → int8 nhwc
ConvAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc | int8 nhwc | int32 — → int8 nhwc
ConvDwAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc | int8 chwn | int32 — → int8 nhwc
FCAct
(aidge_core.MetaOperator)

☑️
1. int8 — | int8 — | int32 — → int8 —
GlobalAveragePooling
(aidge_core.GlobalAveragePooling)

✔️
1. int8 nhwc → int8 nhwc
GlobalAvgPoolAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
MaxPoolAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
MaxPooling2D
(aidge_core.MaxPooling2D)

✔️
1. int8 nhwc → int8 nhwc
PadAvgPool
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
PadAvgPoolAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
PadConv
(aidge_core.MetaOperator)

☑️
1. int8 nhwc | int8 nhwc | int32 — → int8 nhwc
PadConvAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc | int8 nhwc | int32 — → int8 nhwc
PadConvDw
(aidge_core.MetaOperator)

☑️
1. int8 nhwc | int8 chwn | int32 — → int8 nhwc
PadConvDwAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc | int8 chwn | int32 — → int8 nhwc
PaddedAvgPooling2D
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
PaddedMaxPooling2D
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
PadGlobalAvgPool
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
PadGlobalAvgPoolAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
PadMaxPool
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
PadMaxPoolAct
(aidge_core.MetaOperator)

☑️
1. int8 nhwc → int8 nhwc
QAdd
(aidge_core.MetaOperator)

☑️
1. int8 — | int8 — → int8 —
QAddAct
(aidge_core.MetaOperator)

☑️
1. — — → — —
QConv
(aidge_core.MetaOperator)

☑️
1. int8 nhwc | int8 nhwc | int32 — → int8 nhwc
QConvDw
(aidge_core.MetaOperator)

☑️
1. int8 nhwc | int8 chwn | int32 — → int8 nhwc
QFC
(aidge_core.MetaOperator)

☑️
1. int8 — | int8 — | int32 — → int8 —
Quantizer
(aidge_core.MetaOperator)

☑️
1. — — → — —