diff options
author | Tai Ly <tai.ly@arm.com> | 2024-01-16 19:52:58 +0000 |
---|---|---|
committer | Tai Ly <tai.ly@arm.com> | 2024-01-18 23:46:37 +0000 |
commit | 4d04238050a50081f35d7e615b590d00e702f501 (patch) | |
tree | 045fbbc352d9b988ca501c3557adf4124879111f /include | |
parent | 20f6941b21f84cd5f0152d42f343b0992dd5a6e5 (diff) | |
download | tosa_mlir_translator-4d04238050a50081f35d7e615b590d00e702f501.tar.gz |
[tosa_mlir_translator] Add Shape Ops Support
- Add serialization/deserialization of tosa shape ops.
- Changed TileOp's multiples from attribute to shape input.
- Change 'shape' attribute of RESHAPE to an input
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: I4329e621fd7637b1a3491c195fbda77d2a0ad23a
Diffstat (limited to 'include')
-rw-r--r-- | include/operator.def | 10 | ||||
-rw-r--r-- | include/schema_operator.def | 8 |
2 files changed, 16 insertions, 2 deletions
diff --git a/include/operator.def b/include/operator.def index 6198c0e..5b0cc7f 100644 --- a/include/operator.def +++ b/include/operator.def @@ -1,5 +1,5 @@ -// Copyright (c) 2020-2023, ARM Limited. +// Copyright (c) 2020-2024, ARM Limited. // // Licensed under the Apache License, Version 2.0 with LLVM Exceptions // (the "License"); you may not use this file except in compliance with @@ -119,3 +119,11 @@ DEF_OPERATOR(Custom) /* control flow operators */ DEF_OPERATOR(If) DEF_OPERATOR(While) + +/* shape operators */ +DEF_OPERATOR(ConstShape) +DEF_OPERATOR(ConcatShape) +DEF_OPERATOR(AddShape) +DEF_OPERATOR(SubShape) +DEF_OPERATOR(MulShape) +DEF_OPERATOR(DivShape) diff --git a/include/schema_operator.def b/include/schema_operator.def index 52c7ae4..675b4f2 100644 --- a/include/schema_operator.def +++ b/include/schema_operator.def @@ -1,4 +1,4 @@ -// Copyright (c) 2023, ARM Limited. +// Copyright (c) 2023-2024, ARM Limited. // // Licensed under the Apache License, Version 2.0 with LLVM Exceptions // (the "License"); you may not use this file except in compliance with @@ -93,3 +93,9 @@ DEF_SCHEMA_OPERATOR(COND_IF) DEF_SCHEMA_OPERATOR(WHILE_LOOP) DEF_SCHEMA_OPERATOR(FFT2D) DEF_SCHEMA_OPERATOR(RFFT2D) +DEF_SCHEMA_OPERATOR(CONST_SHAPE) +DEF_SCHEMA_OPERATOR(CONCAT_SHAPE) +DEF_SCHEMA_OPERATOR(ADD_SHAPE) +DEF_SCHEMA_OPERATOR(SUB_SHAPE) +DEF_SCHEMA_OPERATOR(MUL_SHAPE) +DEF_SCHEMA_OPERATOR(DIV_SHAPE) |