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

change ending tick mark style

Browse Source
This commit is contained in:
Erich Elsen 2020-06-29 18:13:36 +01:00
parent 491fcbb202
commit 77a023df48
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jax/lax/lax.py
View File

@ -1197,11 +1197,11 @@ def cumprod(operand: Array, axis: int) -> Array:
return cumprod_p.bind(operand, axis=int(axis))
def cummax(operand: Array, axis: int) -> Array:
"""Computes a cumulative maximum along `axis'."""
"""Computes a cumulative maximum along `axis`."""
return cummax_p.bind(operand, axis=int(axis))
def cummin(operand: Array, axis: int) -> Array:
"""Computes a cumulative minimum along `axis'."""
"""Computes a cumulative minimum along `axis`."""
return cummin_p.bind(operand, axis=int(axis))
def sort(operand: Union[Array, Sequence[Array]], dimension: int = -1,