Skip to content

Commit

Permalink
Update unary doc examples set3 (#15425)
Browse files Browse the repository at this point in the history
### Ticket
#14982 

### Problem description
Provided example tensors are inconsistent with the ops.
Currently we support device operations for ranks >= 2 as per doc for
most ops. However, the example tensors in the docs are 1D tensors.

### What's changed
- Updated and tested examples to create 2D tensors in documentation that
are consistent with unary ops
- Updated supported params
- Added unary_chain, dropout, pow under Pointwise Unary documentation

No. of Ops: 17

### Checklist
- [x] [Post commit CI
passes](https://github.com/tenstorrent/tt-metal/actions/runs/12017503786)
  • Loading branch information
mcw-anasuya authored Nov 26, 2024
1 parent 4edfd3f commit 88829c5
Show file tree
Hide file tree
Showing 2 changed files with 138 additions and 201 deletions.
4 changes: 3 additions & 1 deletion docs/source/ttnn/ttnn/api.rst
Original file line number Diff line number Diff line change
Expand Up @@ -114,6 +114,7 @@ Pointwise Unary
ttnn.cosh
ttnn.deg2rad
ttnn.digamma
ttnn.dropout
ttnn.elu
ttnn.eqz
ttnn.erf
Expand Down Expand Up @@ -161,6 +162,7 @@ Pointwise Unary
ttnn.normalize_global
ttnn.normalize_hw
ttnn.polygamma
ttnn.pow
ttnn.prelu
ttnn.rad2deg
ttnn.rdiv
Expand Down Expand Up @@ -197,6 +199,7 @@ Pointwise Unary
ttnn.tril
ttnn.triu
ttnn.trunc
ttnn.unary_chain
ttnn.clamp_bw
ttnn.clip_bw
ttnn.hardtanh_bw
Expand Down Expand Up @@ -305,7 +308,6 @@ Pointwise Binary
ttnn.logical_and_
ttnn.logical_or_
ttnn.logical_xor_
ttnn.pow
ttnn.rpow
ttnn.ldexp
ttnn.logical_and
Expand Down
Loading

0 comments on commit 88829c5

Please sign in to comment.