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

remove core.axis_substitution_rules

Browse Source 
PiperOrigin-RevId: 509989925
This commit is contained in:
Roy Frostig 2023-02-15 18:41:37 -08:00 committed by jax authors
parent 768960b4e4
commit 1b2a318fd1
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jax/core.py
View File

@ -117,7 +117,6 @@ from jax._src.core import (
aval_method as aval_method,
aval_property as aval_property,
axis_frame as axis_frame,
axis_substitution_rules as axis_substitution_rules,
call as call,
call_bind_with_continuation as call_bind_with_continuation,
call_impl as call_impl,