diff options
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 19b00b31..6be9dc25 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -51,6 +51,7 @@ class NpuBlockType(Enum): ConvolutionDepthWise = 4 ElementWise = 5 ReduceSum = 6 + Dma = 7 class Kernel: @@ -174,6 +175,7 @@ class Op(Enum): ) Dequantize = OperatorInfo(indices=NNG_IFM_INDICES) Div = OperatorInfo() + Memcpy = OperatorInfo(block_type=NpuBlockType.Dma, indices=NNG_IFM_INDICES) Elu = OperatorInfo() EmbeddingLookup = OperatorInfo() EmbeddingLookupSparse = OperatorInfo() @@ -373,6 +375,9 @@ class Op(Enum): def is_resize_op(self): return self in (Op.ResizeBilinear, Op.ResizeNearestNeighbor) + def is_memcpy_op(self): + return self.info.block_type == NpuBlockType.Dma + def needs_bias(self): return bool(self.info.indices.biases) |