mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pass x into transpose in autodidax

Browse Source
This commit is contained in:
Sharad Vikram 2021-08-24 17:46:34 -07:00
parent d349086ca5
commit 0fa70084a3
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/autodidax.ipynb
View File

@ -123,7 +123,7 @@
"def reduce_sum(x, axis=None): return bind1(reduce_sum_p, x, axis=axis)\n",
"def greater(x, y): return bind1(greater_p, x, y)\n",
"def less(x, y): return bind1(less_p, x, y)\n",
"def transpose(x, perm): return bind1(transpose_p, perm=perm)\n",
"def transpose(x, perm): return bind1(transpose_p, x, perm=perm)\n",
"def broadcast(x, shape, axes): return bind1(broadcast_p, x, shape=shape, axes=axes)\n",
"\n",
"def bind1(prim, *args, **params):\n",

2
docs/autodidax.md
View File

@ -107,7 +107,7 @@ def cos(x): return bind1(cos_p, x)
def reduce_sum(x, axis=None): return bind1(reduce_sum_p, x, axis=axis)
def greater(x, y): return bind1(greater_p, x, y)
def less(x, y): return bind1(less_p, x, y)
def transpose(x, perm): return bind1(transpose_p, perm=perm)
def transpose(x, perm): return bind1(transpose_p, x, perm=perm)
def broadcast(x, shape, axes): return bind1(broadcast_p, x, shape=shape, axes=axes)
def bind1(prim, *args, **params):

2
docs/autodidax.py
View File

@ -97,7 +97,7 @@ def cos(x): return bind1(cos_p, x)
def reduce_sum(x, axis=None): return bind1(reduce_sum_p, x, axis=axis)
def greater(x, y): return bind1(greater_p, x, y)
def less(x, y): return bind1(less_p, x, y)
def transpose(x, perm): return bind1(transpose_p, perm=perm)
def transpose(x, perm): return bind1(transpose_p, x, perm=perm)
def broadcast(x, shape, axes): return bind1(broadcast_p, x, shape=shape, axes=axes)
def bind1(prim, *args, **params):