#10382: Update type 2 complex files

tenstorrent · Jul 20, 2024 · ac6b8c4 · ac6b8c4
1 parent 0ad8dda
commit ac6b8c4
Show file tree

Hide file tree

Showing 6 changed files with 629 additions and 452 deletions.
diff --git a/docs/source/ttnn/ttnn/api.rst b/docs/source/ttnn/ttnn/api.rst
@@ -263,7 +263,7 @@ Pointwise Unary
    ttnn/angle_bw
    ttnn/conj_bw
    ttnn/conj
-   ttmm/polar
+   ttnn/polar
 
 Pointwise Binary
 ================

diff --git a/docs/source/ttnn/ttnn/dependencies/tt_lib.rst b/docs/source/ttnn/ttnn/dependencies/tt_lib.rst
@@ -540,6 +540,31 @@ Complex Operations
 
 The following functions are available,
 
+
+Complex arithmetic can be carried out for multiply, divide, add and subtract as follows:
+
+.. autofunction:: tt_lib.tensor.complex_add
+
+.. autofunction:: tt_lib.tensor.complex_sub
+
+.. autofunction:: tt_lib.tensor.complex_mul
+
+.. autofunction:: tt_lib.tensor.complex_div
+
+and then unary operations for,
+
+.. autofunction:: tt_lib.tensor.real
+
+.. autofunction:: tt_lib.tensor.imag
+
+.. autofunction:: tt_lib.tensor.complex_abs
+
+.. autofunction:: tt_lib.tensor.conj
+
+.. autofunction:: tt_lib.tensor.complex_recip
+
+.. autofunction:: tt_lib.tensor.polar
+
 Complex Operations (Type 2)
 ===========================
 Type 2 Complex representation allows for more flexible storage than earlier one while providing same set of