From 1f05883799957300ea88d7ec314a5897e6c0ccb1 Mon Sep 17 00:00:00 2001
From: Eric Kunze <eric.kunze@arm.com>
Date: Tue, 13 Feb 2024 16:51:17 -0800
Subject: Add SIN/COS to elementwise operator set

Only defined for floating-point types.

Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: Ieae355076f9aac6882fab8530e6d7d9985a6e661
---
 tosa.xml | 58 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)

(limited to 'tosa.xml')

diff --git a/tosa.xml b/tosa.xml
index 19822f6..27b49b8 100644
--- a/tosa.xml
+++ b/tosa.xml
@@ -1427,6 +1427,35 @@
           <type name='in_out_t'/>
         </types>
         <typesupport mode="signed 32" in_out_t="i32_t"/>
+      </operator>
+            <operator>
+        <name>COS</name>
+        <arguments>
+          <argument category="input" name="input" type="tensor_t" shape="shape" tensor-element-type="in_out_t">
+            <description>Input tensor</description>
+            <levellimit value="rank(shape)" limit="MAX_RANK"/>
+            <rank min="0" max="MAX_RANK"/>
+          </argument>
+          <argument category="output" name="output" type="tensor_t" shape="shape" tensor-element-type="in_out_t">
+            <description>Output tensor of same type and shape as input</description>
+            <rank min="0" max="MAX_RANK"/>
+          </argument>
+        </arguments>
+        <types>
+          <type name='in_out_t'/>
+        </types>
+        <typesupport mode="fp16" in_out_t="fp16_t">
+          <profile name="MI"/>
+          <profile name="MT"/>
+        </typesupport>
+        <typesupport mode="bf16" in_out_t="bf16_t">
+          <profile name="MI"/>
+          <profile name="MT"/>
+        </typesupport>
+        <typesupport mode="fp32" in_out_t="fp32_t">
+          <profile name="MI"/>
+          <profile name="MT"/>
+        </typesupport>
       </operator>
       <operator>
         <name>EXP</name>
@@ -1674,6 +1703,35 @@
           <profile name="MI"/>
           <profile name="MT"/>
         </typesupport>
+      </operator>
+            <operator>
+        <name>SIN</name>
+        <arguments>
+          <argument category="input" name="input" type="tensor_t" shape="shape" tensor-element-type="in_out_t">
+            <description>Input tensor</description>
+            <levellimit value="rank(shape)" limit="MAX_RANK"/>
+            <rank min="0" max="MAX_RANK"/>
+          </argument>
+          <argument category="output" name="output" type="tensor_t" shape="shape" tensor-element-type="in_out_t">
+            <description>Output tensor of same type and shape as input</description>
+            <rank min="0" max="MAX_RANK"/>
+          </argument>
+        </arguments>
+        <types>
+          <type name='in_out_t'/>
+        </types>
+        <typesupport mode="fp16" in_out_t="fp16_t">
+          <profile name="MI"/>
+          <profile name="MT"/>
+        </typesupport>
+        <typesupport mode="bf16" in_out_t="bf16_t">
+          <profile name="MI"/>
+          <profile name="MT"/>
+        </typesupport>
+        <typesupport mode="fp32" in_out_t="fp32_t">
+          <profile name="MI"/>
+          <profile name="MT"/>
+        </typesupport>
       </operator>
     </operatorgroup>
     <operatorgroup name="comparison">
-- 
cgit v1.2.1