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

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)

❌
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)

❌
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. — — → — —
ConvDepthWise1D
(aidge_core.ConvDepthWise1D)

❌
ConvDepthWise2D
(aidge_core.ConvDepthWise2D)

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

✔️
1. — nwc | — default | — — → — nwc
ConvTranspose2D
(aidge_core.ConvTranspose2D)

✔️
1. — nhwc | — default | — — → — 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. complex32 — → complex32 —
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. — — → — —
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 — → complex32 —
InstanceNorm
(aidge_core.InstanceNorm)

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

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

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

✅
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. — nwc → — nwc
MaxPooling2D
(aidge_core.MaxPooling2D)

✅
1. — 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)

❌
NonZero
(aidge_core.NonZero)

❌
Not
(aidge_core.Not)

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

❌
PadConv
(aidge_export_arm_cortexm._Aidge_Arm.operators.Conv.ArmPadConv)
1. — — → — —
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. — — → — —
PadConvDw
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvDw.ArmPadConvDw)
1. — — → — —
PadConvDwAct
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvDw.ArmPadConvDwAct)
1. — — → — —
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. — — → — —
PaddedConvDepthWise2D
(aidge_export_arm_cortexm._Aidge_Arm.operators.ConvDw.ArmPadConvDw)
1. — — → — —
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. — — → — —
QFC
(aidge_export_arm_cortexm._Aidge_Arm.operators.Fc.ArmQFC)
1. — default | — — | — — → — —
2. — nhwc | — — | — — → — —
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)

❌
Select
(aidge_core.Select)

❌
Shape
(aidge_core.Shape)

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

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

❌
Sin
(aidge_core.Sin)

❌
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. — — → — —
Unfold2D
(aidge_core.Unfold2D)

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

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

❌
Where
(aidge_core.Where)

✅
1. — — → — —

CMSIS-NN implementations#

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

Operator

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 —