zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

bindings for memory info (#761)

Browse Source 
* bindings for memory info

* update api

* keep cache low if requested

* fix default

* nit in ops error
This commit is contained in:
Awni Hannun
2024-03-01 19:51:58 -08:00
committed by GitHub
parent cf3eb87e52
commit d5964a2710
11 changed files with 300 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/src/python/ops.rst
View File

@@ -57,6 +57,7 @@ Operations
greater_equal
identity
inner
isclose
isnan
isposinf
isneginf
@@ -121,6 +122,7 @@ Operations
tan
tanh
tensordot
tile
transpose
tri
tril