diff --git a/docs/build/html/.buildinfo b/docs/build/html/.buildinfo index 11664a76f..e5591e92d 100644 --- a/docs/build/html/.buildinfo +++ b/docs/build/html/.buildinfo @@ -1,4 +1,4 @@ # Sphinx build info version 1 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done. -config: b9afea4432e35a4ac5dbffb76ed39348 +config: 81aaeaba965ad2156af7b499f90bc0e9 tags: 645f666f9bcd5a90fca523b33c5a78b7 diff --git a/docs/build/html/_sources/usage/distributed.rst b/docs/build/html/_sources/usage/distributed.rst index ec3accab3..ee0952194 100644 --- a/docs/build/html/_sources/usage/distributed.rst +++ b/docs/build/html/_sources/usage/distributed.rst @@ -57,7 +57,7 @@ with the Anaconda package manager as follows: .. code:: shell - $ conda install openmpi + $ conda install conda-forge::openmpi Installing with Homebrew may require specifying the location of ``libmpi.dyld`` so that MLX can find it and load it at runtime. This can simply be achieved by diff --git a/docs/build/html/_static/documentation_options.js b/docs/build/html/_static/documentation_options.js index c4eb24305..bd3b55de9 100644 --- a/docs/build/html/_static/documentation_options.js +++ b/docs/build/html/_static/documentation_options.js @@ -1,5 +1,5 @@ const DOCUMENTATION_OPTIONS = { - VERSION: '0.22.0', + VERSION: '0.22.1', LANGUAGE: 'en', COLLAPSE_INDEX: false, BUILDER: 'html', diff --git a/docs/build/html/accelerate__fp16__simd_8h.html b/docs/build/html/accelerate__fp16__simd_8h.html new file mode 100644 index 000000000..af4a689fd --- /dev/null +++ b/docs/build/html/accelerate__fp16__simd_8h.html @@ -0,0 +1,253 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/accelerate_fp16_simd.h File Reference + + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
+Namespaces | +Macros | +Functions | +Variables
+
accelerate_fp16_simd.h File Reference
+
+
+
#include "mlx/backend/cpu/simd/base_simd.h"
+#include "mlx/backend/cpu/simd/neon_fp16_simd.h"
+
+

Go to the source code of this file.

+ + + + + + + + +

+Namespaces

namespace  mlx
 
namespace  mlx::core
 
namespace  mlx::core::simd
 
+ + + + + +

+Macros

#define SIMD_FP16_DEFAULT_UNARY(op)
 
#define SIMD_FP16_DEFAULT_BINARY(op)
 
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

+Functions

template<>
Simd< float16_t, N > mlx::core::simd::acos (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::acosh (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::asin (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::asinh (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::atan (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::atanh (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::cosh (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::expm1 (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::log (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::log2 (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::log10 (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::log1p (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::sinh (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::tan (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::tanh (Simd< float16_t, N > v)
 
template<>
Simd< float16_t, N > mlx::core::simd::atan2 (Simd< float16_t, N > x, Simd< float16_t, N > y)
 
template<>
Simd< float16_t, N > mlx::core::simd::remainder (Simd< float16_t, N > x, Simd< float16_t, N > y)
 
template<>
Simd< float16_t, N > mlx::core::simd::pow (Simd< float16_t, N > x, Simd< float16_t, N > y)
 
+ + + + +

+Variables

template<>
static constexpr int mlx::core::simd::max_size< float16_t > = N
 
+

Macro Definition Documentation

+ +

◆ SIMD_FP16_DEFAULT_BINARY

+ +
+
+ + + + + + + +
#define SIMD_FP16_DEFAULT_BINARY( op)
+
+Value:
template <> \
+
inline Simd<float16_t, N> op(Simd<float16_t, N> x, Simd<float16_t, N> y) { \
+
Simd<float, N> a = x; \
+
Simd<float, N> b = y; \
+
return op(a, b); \
+
}
+
+
+
+ +

◆ SIMD_FP16_DEFAULT_UNARY

+ +
+
+ + + + + + + +
#define SIMD_FP16_DEFAULT_UNARY( op)
+
+Value:
template <> \
+
inline Simd<float16_t, N> op(Simd<float16_t, N> v) { \
+
Simd<float, N> in = v; \
+
return op(in); \
+
}
+
+
+
+
+
+ + + + diff --git a/docs/build/html/accelerate__fp16__simd_8h.js b/docs/build/html/accelerate__fp16__simd_8h.js new file mode 100644 index 000000000..2565f11c2 --- /dev/null +++ b/docs/build/html/accelerate__fp16__simd_8h.js @@ -0,0 +1,24 @@ +var accelerate__fp16__simd_8h = +[ + [ "SIMD_FP16_DEFAULT_BINARY", "accelerate__fp16__simd_8h.html#a6437645d67e6a8b7c1349b0040aaa6bb", null ], + [ "SIMD_FP16_DEFAULT_UNARY", "accelerate__fp16__simd_8h.html#a60af883c0392ecab270ecc1146f5e8a2", null ], + [ "mlx::core::simd::acos", "namespacemlx_1_1core_1_1simd.html#ae623449dfa7aab3031aa2f14c1b10a2d", null ], + [ "mlx::core::simd::acosh", "namespacemlx_1_1core_1_1simd.html#a4f8a64e7742fcd8f759f723a36a7c826", null ], + [ "mlx::core::simd::asin", "namespacemlx_1_1core_1_1simd.html#a9c7723fc49137394fa817136a7ffb50f", null ], + [ "mlx::core::simd::asinh", "namespacemlx_1_1core_1_1simd.html#aaa76bdf1db09261d84da51d394837f5d", null ], + [ "mlx::core::simd::atan", "namespacemlx_1_1core_1_1simd.html#aa7550a1210e50c996d0db84034b8a22e", null ], + [ "mlx::core::simd::atan2", "namespacemlx_1_1core_1_1simd.html#a23dba4ee3f0811b41c381733a6e6ff16", null ], + [ "mlx::core::simd::atanh", "namespacemlx_1_1core_1_1simd.html#a18d330fd2c7360b2890a722232ba35b7", null ], + [ "mlx::core::simd::cosh", "namespacemlx_1_1core_1_1simd.html#aedc18b6fdb820cce9125c977c02833aa", null ], + [ "mlx::core::simd::expm1", "namespacemlx_1_1core_1_1simd.html#a9407980793ecff5d5eb19c9a2cbda1eb", null ], + [ "mlx::core::simd::log", "namespacemlx_1_1core_1_1simd.html#a8cec82f4fb15bfd31d7554c6c09ceed4", null ], + [ "mlx::core::simd::log10", "namespacemlx_1_1core_1_1simd.html#a3cb6ea94836e999c07329b34c501ed85", null ], + [ "mlx::core::simd::log1p", "namespacemlx_1_1core_1_1simd.html#ad06680bbc041e76efe2dbff4e11b9a13", null ], + [ "mlx::core::simd::log2", "namespacemlx_1_1core_1_1simd.html#a66426c28a4324b9f617b7018d9354ea1", null ], + [ "mlx::core::simd::pow", "namespacemlx_1_1core_1_1simd.html#a19d535de1fc179cc39ec9643c9863cbc", null ], + [ "mlx::core::simd::remainder", "namespacemlx_1_1core_1_1simd.html#ac66bdf1a8e86a4d350c85037bc764da5", null ], + [ "mlx::core::simd::sinh", "namespacemlx_1_1core_1_1simd.html#a09a2f3f2bc999c16babf3d8d90994d6e", null ], + [ "mlx::core::simd::tan", "namespacemlx_1_1core_1_1simd.html#a445ddc4ed928656df64d889942588cfd", null ], + [ "mlx::core::simd::tanh", "namespacemlx_1_1core_1_1simd.html#aa244fbe7456b653aa50a473108fd6a2b", null ], + [ "mlx::core::simd::max_size< float16_t >", "namespacemlx_1_1core_1_1simd.html#a155df1de3c26e1a3725b63e9e97c0b53", null ] +]; \ No newline at end of file diff --git a/docs/build/html/accelerate__fp16__simd_8h_source.html b/docs/build/html/accelerate__fp16__simd_8h_source.html new file mode 100644 index 000000000..41a4802ac --- /dev/null +++ b/docs/build/html/accelerate__fp16__simd_8h_source.html @@ -0,0 +1,208 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/accelerate_fp16_simd.h Source File + + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
accelerate_fp16_simd.h
+
+
+Go to the documentation of this file.
1#pragma once
+
2
+
3#include "mlx/backend/cpu/simd/base_simd.h"
+
4
+
5#if MLX_SIMD_LIBRARY_VERSION < 6
+
6#include "mlx/backend/cpu/simd/neon_fp16_simd.h"
+
7#endif
+
8
+
+
9namespace mlx::core::simd {
+
10
+
11#if MLX_SIMD_LIBRARY_VERSION >= 6
+
12constexpr int N = 8;
+
13template <int N>
+
14struct ScalarT<float16_t, N> {
+
15 using v = _Float16;
+
16};
+
17#endif
+
18
+
19template <>
+
20static constexpr int max_size<float16_t> = N;
+
21
+
+
22#define SIMD_FP16_DEFAULT_UNARY(op) \
+
23 template <> \
+
24 inline Simd<float16_t, N> op(Simd<float16_t, N> v) { \
+
25 Simd<float, N> in = v; \
+
26 return op(in); \
+
27 }
+
+
28
+
29SIMD_FP16_DEFAULT_UNARY(acos)
+
30SIMD_FP16_DEFAULT_UNARY(acosh)
+
31SIMD_FP16_DEFAULT_UNARY(asin)
+
32SIMD_FP16_DEFAULT_UNARY(asinh)
+
33SIMD_FP16_DEFAULT_UNARY(atan)
+
34SIMD_FP16_DEFAULT_UNARY(atanh)
+
35SIMD_FP16_DEFAULT_UNARY(cosh)
+
36SIMD_FP16_DEFAULT_UNARY(expm1)
+
37SIMD_FP16_DEFAULT_UNARY(log)
+
38SIMD_FP16_DEFAULT_UNARY(log2)
+
39SIMD_FP16_DEFAULT_UNARY(log10)
+
40SIMD_FP16_DEFAULT_UNARY(log1p)
+
41SIMD_FP16_DEFAULT_UNARY(sinh)
+
42SIMD_FP16_DEFAULT_UNARY(tan)
+
43SIMD_FP16_DEFAULT_UNARY(tanh)
+
44
+
+
45#define SIMD_FP16_DEFAULT_BINARY(op) \
+
46 template <> \
+
47 inline Simd<float16_t, N> op(Simd<float16_t, N> x, Simd<float16_t, N> y) { \
+
48 Simd<float, N> a = x; \
+
49 Simd<float, N> b = y; \
+
50 return op(a, b); \
+
51 }
+
+
52SIMD_FP16_DEFAULT_BINARY(atan2)
+
53SIMD_FP16_DEFAULT_BINARY(remainder)
+
54SIMD_FP16_DEFAULT_BINARY(pow)
+
55
+
56} // namespace mlx::core::simd
+
+
SIMD_FP16_DEFAULT_UNARY
#define SIMD_FP16_DEFAULT_UNARY(op)
Definition accelerate_fp16_simd.h:22
+
SIMD_FP16_DEFAULT_BINARY
#define SIMD_FP16_DEFAULT_BINARY(op)
Definition accelerate_fp16_simd.h:45
+
base_simd.h
+
mlx::core::simd
Definition accelerate_fp16_simd.h:9
+
mlx::core::simd::sinh
Simd< float16_t, N > sinh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:41
+
mlx::core::simd::N
constexpr int N
Definition neon_fp16_simd.h:9
+
mlx::core::simd::atanh
Simd< float16_t, N > atanh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:34
+
mlx::core::simd::pow
Simd< float16_t, N > pow(Simd< float16_t, N > x, Simd< float16_t, N > y)
Definition accelerate_fp16_simd.h:54
+
mlx::core::simd::atan2
Simd< float16_t, N > atan2(Simd< float16_t, N > x, Simd< float16_t, N > y)
Definition accelerate_fp16_simd.h:52
+
mlx::core::simd::log10
Simd< float16_t, N > log10(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:39
+
mlx::core::simd::tan
Simd< float16_t, N > tan(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:42
+
mlx::core::simd::acosh
Simd< float16_t, N > acosh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:30
+
mlx::core::simd::log2
Simd< float16_t, N > log2(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:38
+
mlx::core::simd::log
Simd< float16_t, N > log(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:37
+
mlx::core::simd::expm1
Simd< float16_t, N > expm1(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:36
+
mlx::core::simd::asin
Simd< float16_t, N > asin(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:31
+
mlx::core::simd::tanh
Simd< float16_t, N > tanh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:43
+
mlx::core::simd::atan
Simd< float16_t, N > atan(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:33
+
mlx::core::simd::asinh
Simd< float16_t, N > asinh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:32
+
mlx::core::simd::remainder
Simd< float16_t, N > remainder(Simd< float16_t, N > x, Simd< float16_t, N > y)
Definition accelerate_fp16_simd.h:53
+
mlx::core::simd::max_size
static constexpr int max_size
Definition base_simd.h:13
+
mlx::core::simd::log1p
Simd< float16_t, N > log1p(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:40
+
mlx::core::simd::acos
Simd< float16_t, N > acos(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:29
+
mlx::core::simd::cosh
Simd< float16_t, N > cosh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:35
+
mlx::core::float16_t
struct _MLX_Float16 float16_t
Definition half_types.h:17
+
neon_fp16_simd.h
+
mlx::core::simd::ScalarT
Definition accelerate_simd.h:34
+
mlx::core::simd::ScalarT::v
T v
Definition accelerate_simd.h:35
+
+
+ + + + diff --git a/docs/build/html/accelerate__simd_8h.html b/docs/build/html/accelerate__simd_8h.html new file mode 100644 index 000000000..0139cf13d --- /dev/null +++ b/docs/build/html/accelerate__simd_8h.html @@ -0,0 +1,572 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/accelerate_simd.h File Reference + + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
+Classes | +Namespaces | +Macros | +Functions | +Variables
+
accelerate_simd.h File Reference
+
+
+
#include <simd/math.h>
+#include <simd/vector.h>
+#include <stdint.h>
+#include <cmath>
+#include <complex>
+#include "mlx/backend/cpu/simd/base_simd.h"
+
+

Go to the source code of this file.

+ + + + + + + + + + + + + + +

+Classes

struct  mlx::core::simd::ScalarT< T, N >
 
struct  mlx::core::simd::ScalarT< bool, N >
 
struct  mlx::core::simd::ScalarT< int8_t, N >
 
struct  mlx::core::simd::ScalarT< uint64_t, N >
 
struct  mlx::core::simd::ScalarT< int64_t, N >
 
struct  mlx::core::simd::Simd< T, N >
 
+ + + + + + + +

+Namespaces

namespace  mlx
 
namespace  mlx::core
 
namespace  mlx::core::simd
 
+ + + + + + + + + +

+Macros

#define MLX_SIMD_LIBRARY_VERSION   5
 
#define SIMD_DEFAULT_UNARY(name, op)
 
#define SIMD_DEFAULT_BINARY(OP)
 
#define SIMD_DEFAULT_COMPARISONS(OP)
 
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

+Functions

template<typename T, int N>
Simd< T, N > mlx::core::simd::abs (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::floor (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::acos (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::acosh (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::asin (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::asinh (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::atan (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::atanh (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::ceil (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::cosh (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::expm1 (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::log (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::log2 (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::log10 (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::log1p (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::rint (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::sinh (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::sqrt (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::rsqrt (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::recip (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::tan (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::tanh (Simd< T, N > v)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::operator- (Simd< T, N > v)
 
template<typename T, int N>
Simd< bool, N > mlx::core::simd::isnan (Simd< T, N > v)
 
template<typename T, int N>
Simd< bool, N > mlx::core::simd::operator! (Simd< T, N > v)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator+ (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator+ (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator+ (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator- (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator- (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator- (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator/ (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator/ (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator/ (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator* (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator* (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator* (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator<< (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator<< (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator<< (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator>> (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator>> (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator>> (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator| (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator| (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator| (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator^ (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator^ (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator^ (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator& (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator& (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator& (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator&& (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator&& (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator&& (Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::operator|| (Simd< T, N > x, U y)
 
template<typename T1, typename T2, int N>
Simd< T2, N > mlx::core::simd::operator|| (T1 x, Simd< T2, N > y)
 
template<typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::operator|| (Simd< T1, N > x, Simd< T2, N > y)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator> (Simd< T, N > a, U b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator> (T a, Simd< U, N > b)
 
template<int N, typename T1, typename T2>
Simd< bool, N > mlx::core::simd::operator> (Simd< T1, N > a, Simd< T2, N > b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator< (Simd< T, N > a, U b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator< (T a, Simd< U, N > b)
 
template<int N, typename T1, typename T2>
Simd< bool, N > mlx::core::simd::operator< (Simd< T1, N > a, Simd< T2, N > b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator>= (Simd< T, N > a, U b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator>= (T a, Simd< U, N > b)
 
template<int N, typename T1, typename T2>
Simd< bool, N > mlx::core::simd::operator>= (Simd< T1, N > a, Simd< T2, N > b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator<= (Simd< T, N > a, U b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator<= (T a, Simd< U, N > b)
 
template<int N, typename T1, typename T2>
Simd< bool, N > mlx::core::simd::operator<= (Simd< T1, N > a, Simd< T2, N > b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator== (Simd< T, N > a, U b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator== (T a, Simd< U, N > b)
 
template<int N, typename T1, typename T2>
Simd< bool, N > mlx::core::simd::operator== (Simd< T1, N > a, Simd< T2, N > b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator!= (Simd< T, N > a, U b)
 
template<int N, typename T, typename U>
Simd< bool, N > mlx::core::simd::operator!= (T a, Simd< U, N > b)
 
template<int N, typename T1, typename T2>
Simd< bool, N > mlx::core::simd::operator!= (Simd< T1, N > a, Simd< T2, N > b)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::atan2 (Simd< T, N > a, Simd< T, N > b)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::maximum (Simd< T, N > a, Simd< T, N > b)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::minimum (Simd< T, N > a, Simd< T, N > b)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::remainder (Simd< T, N > a, Simd< T, N > b)
 
template<typename MaskT, typename T1, typename T2, int N>
Simd< T1, N > mlx::core::simd::select (Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::pow (Simd< T, N > base, Simd< T, N > exp)
 
template<typename T, int N>
Simd< T, N > mlx::core::simd::clamp (Simd< T, N > v, Simd< T, N > min, Simd< T, N > max)
 
template<typename T, typename U, int N>
Simd< T, N > mlx::core::simd::fma (Simd< T, N > x, Simd< T, N > y, U z)
 
template<typename T, int N>
bool mlx::core::simd::all (Simd< T, N > x)
 
template<typename T, int N>
bool mlx::core::simd::any (Simd< T, N > x)
 
template<typename T, int N>
T mlx::core::simd::sum (Simd< T, N > x)
 
template<typename T, int N>
T mlx::core::simd::max (Simd< T, N > x)
 
template<typename T, int N>
T mlx::core::simd::min (Simd< T, N > x)
 
template<typename T, int N>
T mlx::core::simd::prod (Simd< T, N > x)
 
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

+Variables

template<>
static constexpr int mlx::core::simd::max_size< int8_t > = 16
 
template<>
static constexpr int mlx::core::simd::max_size< int16_t > = 16
 
template<>
static constexpr int mlx::core::simd::max_size< int > = 8
 
template<>
static constexpr int mlx::core::simd::max_size< int64_t > = 4
 
template<>
static constexpr int mlx::core::simd::max_size< uint8_t > = 16
 
template<>
static constexpr int mlx::core::simd::max_size< uint16_t > = 16
 
template<>
static constexpr int mlx::core::simd::max_size< uint32_t > = 8
 
template<>
static constexpr int mlx::core::simd::max_size< uint64_t > = 4
 
template<>
static constexpr int mlx::core::simd::max_size< float > = 8
 
template<>
static constexpr int mlx::core::simd::max_size< double > = 4
 
+

Macro Definition Documentation

+ +

◆ MLX_SIMD_LIBRARY_VERSION

+ +
+
+ + + + +
#define MLX_SIMD_LIBRARY_VERSION   5
+
+ +
+
+ +

◆ SIMD_DEFAULT_BINARY

+ +
+
+ + + + + + + +
#define SIMD_DEFAULT_BINARY( OP)
+
+Value:
template <typename T, typename U, int N> \
+
Simd<T, N> operator OP(Simd<T, N> x, U y) { \
+
return asd::convert<typename Simd<T, N>::scalar_t>(x.value OP y); \
+
} \
+
template <typename T1, typename T2, int N> \
+
Simd<T2, N> operator OP(T1 x, Simd<T2, N> y) { \
+
return asd::convert<typename Simd<T2, N>::scalar_t>(x OP y.value); \
+
} \
+
template <typename T1, typename T2, int N> \
+
Simd<T1, N> operator OP(Simd<T1, N> x, Simd<T2, N> y) { \
+
return asd::convert<typename Simd<T1, N>::scalar_t>(x.value OP y.value); \
+
}
+
+
+
+ +

◆ SIMD_DEFAULT_COMPARISONS

+ +
+
+ + + + + + + +
#define SIMD_DEFAULT_COMPARISONS( OP)
+
+Value:
template <int N, typename T, typename U> \
+
Simd<bool, N> operator OP(Simd<T, N> a, U b) { \
+
return asd::convert<char>(a.value OP b); \
+
} \
+
template <int N, typename T, typename U> \
+
Simd<bool, N> operator OP(T a, Simd<U, N> b) { \
+
return asd::convert<char>(a OP b.value); \
+
} \
+
template <int N, typename T1, typename T2> \
+
Simd<bool, N> operator OP(Simd<T1, N> a, Simd<T2, N> b) { \
+
return asd::convert<char>(a.value OP b.value); \
+
}
+
+
+
+ +

◆ SIMD_DEFAULT_UNARY

+ +
+
+ + + + + + + + + + + +
#define SIMD_DEFAULT_UNARY( name,
op )
+
+Value:
template <typename T, int N> \
+
Simd<T, N> name(Simd<T, N> v) { \
+
return op(v.value); \
+
}
+
+
+
+
+
+ + + + diff --git a/docs/build/html/accelerate__simd_8h.js b/docs/build/html/accelerate__simd_8h.js new file mode 100644 index 000000000..0a2068f72 --- /dev/null +++ b/docs/build/html/accelerate__simd_8h.js @@ -0,0 +1,113 @@ +var accelerate__simd_8h = +[ + [ "mlx::core::simd::ScalarT< T, N >", "structmlx_1_1core_1_1simd_1_1_scalar_t.html", "structmlx_1_1core_1_1simd_1_1_scalar_t" ], + [ "mlx::core::simd::ScalarT< bool, N >", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4" ], + [ "mlx::core::simd::ScalarT< int8_t, N >", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4" ], + [ "mlx::core::simd::ScalarT< uint64_t, N >", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4" ], + [ "mlx::core::simd::ScalarT< int64_t, N >", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4" ], + [ "mlx::core::simd::Simd< T, N >", "structmlx_1_1core_1_1simd_1_1_simd.html", "structmlx_1_1core_1_1simd_1_1_simd" ], + [ "MLX_SIMD_LIBRARY_VERSION", "accelerate__simd_8h.html#a628420ebe2f37d352c8c4b0dd0db59d0", null ], + [ "SIMD_DEFAULT_BINARY", "accelerate__simd_8h.html#a9aa795d90ddc485f24cd4a5268fe0e82", null ], + [ "SIMD_DEFAULT_COMPARISONS", "accelerate__simd_8h.html#aca22aa431d399cea13c969926689b459", null ], + [ "SIMD_DEFAULT_UNARY", "accelerate__simd_8h.html#a3b449b9ce2c623ab4dce3f6fe349bb6d", null ], + [ "mlx::core::simd::abs", "namespacemlx_1_1core_1_1simd.html#a4f3cc8b2493586e83fd65640df3b60ad", null ], + [ "mlx::core::simd::acos", "namespacemlx_1_1core_1_1simd.html#a85999467c83b07e4fa5f093f7ddf19e1", null ], + [ "mlx::core::simd::acosh", "namespacemlx_1_1core_1_1simd.html#a90092f3826ad3be4b2b1785f7ff4a86b", null ], + [ "mlx::core::simd::all", "namespacemlx_1_1core_1_1simd.html#a5109118acb6766855878b9e8a56b156a", null ], + [ "mlx::core::simd::any", "namespacemlx_1_1core_1_1simd.html#a9d968537ad5ef18630f5afce8453b30e", null ], + [ "mlx::core::simd::asin", "namespacemlx_1_1core_1_1simd.html#a069963ffb15f06d1c48258054750dadf", null ], + [ "mlx::core::simd::asinh", "namespacemlx_1_1core_1_1simd.html#a7b47a5f370e8e59e1debfa5405e13266", null ], + [ "mlx::core::simd::atan", "namespacemlx_1_1core_1_1simd.html#a99e84cece5722fb764844a2badc5426b", null ], + [ "mlx::core::simd::atan2", "namespacemlx_1_1core_1_1simd.html#ac6104b5667e0eb379528bf7e2de23bee", null ], + [ "mlx::core::simd::atanh", "namespacemlx_1_1core_1_1simd.html#ae55fd26c3e18a6a27679d2b47566f8bc", null ], + [ "mlx::core::simd::ceil", "namespacemlx_1_1core_1_1simd.html#ae1d5460c58c507a0104d8dfa90343f12", null ], + [ "mlx::core::simd::clamp", "namespacemlx_1_1core_1_1simd.html#ae4be4d88cd8eba7a8c1784fd53b86edb", null ], + [ "mlx::core::simd::cosh", "namespacemlx_1_1core_1_1simd.html#aa5b4f7d3b776e8d16907e15a11800f01", null ], + [ "mlx::core::simd::expm1", "namespacemlx_1_1core_1_1simd.html#a464687a8809d0180035acc9af2943a94", null ], + [ "mlx::core::simd::floor", "namespacemlx_1_1core_1_1simd.html#a8e22c484298d9af10b6604c835e52052", null ], + [ "mlx::core::simd::fma", "namespacemlx_1_1core_1_1simd.html#a9ddc7f119cc1dc04372ec1adcaf55f70", null ], + [ "mlx::core::simd::isnan", "namespacemlx_1_1core_1_1simd.html#a05f4422a037c3bef343fb11f71363b65", null ], + [ "mlx::core::simd::log", "namespacemlx_1_1core_1_1simd.html#a97a8ca857fe0edd84c68dc0f3dc2c6c4", null ], + [ "mlx::core::simd::log10", "namespacemlx_1_1core_1_1simd.html#aba81b735e8f99cedf8b4846b2ab4e236", null ], + [ "mlx::core::simd::log1p", "namespacemlx_1_1core_1_1simd.html#a830591eb3007fef5d87dc296f5615108", null ], + [ "mlx::core::simd::log2", "namespacemlx_1_1core_1_1simd.html#acb1c49b90d029bc4a7eed257ec52791d", null ], + [ "mlx::core::simd::max", "namespacemlx_1_1core_1_1simd.html#a6fcea259041cecfd042d0c4e6afc4b8f", null ], + [ "mlx::core::simd::maximum", "namespacemlx_1_1core_1_1simd.html#a7f7a298284e71ddbd2ba0bb6d98b0d16", null ], + [ "mlx::core::simd::min", "namespacemlx_1_1core_1_1simd.html#acd4196d0c66204cfae70b064c305e146", null ], + [ "mlx::core::simd::minimum", "namespacemlx_1_1core_1_1simd.html#a1996e77a8c3c24b1ba706113ed9028c4", null ], + [ "mlx::core::simd::operator!", "namespacemlx_1_1core_1_1simd.html#a745e05627c77152ec13d8d90c19cc9bf", null ], + [ "mlx::core::simd::operator!=", "namespacemlx_1_1core_1_1simd.html#a4971bfe7f9f9319f859b3040c18f39ca", null ], + [ "mlx::core::simd::operator!=", "namespacemlx_1_1core_1_1simd.html#a125cbaa7c5dd0931b0abd11003ab584a", null ], + [ "mlx::core::simd::operator!=", "namespacemlx_1_1core_1_1simd.html#a5c49123bf2647a5ca4f0579a54f3e53a", null ], + [ "mlx::core::simd::operator&", "namespacemlx_1_1core_1_1simd.html#a0727c897502944659b3e32b3cde9ee9b", null ], + [ "mlx::core::simd::operator&", "namespacemlx_1_1core_1_1simd.html#ac790406f4cf51cbc40d750d377dd741b", null ], + [ "mlx::core::simd::operator&", "namespacemlx_1_1core_1_1simd.html#a832bbc02ed5589e70106c831c04500f1", null ], + [ "mlx::core::simd::operator&&", "namespacemlx_1_1core_1_1simd.html#a85c23e7ed6fe0ec6dfe4c61f7412a362", null ], + [ "mlx::core::simd::operator&&", "namespacemlx_1_1core_1_1simd.html#a070f1fa094cf2da5ab7d6baecbbf4f56", null ], + [ "mlx::core::simd::operator&&", "namespacemlx_1_1core_1_1simd.html#a8a2c8aea209236b06c594c8451017ecb", null ], + [ "mlx::core::simd::operator*", "namespacemlx_1_1core_1_1simd.html#a08c1e7a00b1b4bc60e30d1554f4f46f2", null ], + [ "mlx::core::simd::operator*", "namespacemlx_1_1core_1_1simd.html#a4555cd6a3b50af00700f97fdf00f63a7", null ], + [ "mlx::core::simd::operator*", "namespacemlx_1_1core_1_1simd.html#ae4ec5f1f081d20b46b13eb83eb1b6431", null ], + [ "mlx::core::simd::operator+", "namespacemlx_1_1core_1_1simd.html#aac6acd134f1498b4fb45fdbc882335bf", null ], + [ "mlx::core::simd::operator+", "namespacemlx_1_1core_1_1simd.html#aed655ffa017ade5e0f954f906d9f7ae6", null ], + [ "mlx::core::simd::operator+", "namespacemlx_1_1core_1_1simd.html#a8b622c47d07b171b2303ea744bf72284", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#af5be79b8dada8f8e91ae7c03c16606ec", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#ad5761065b4a655cd086d88846ae08d97", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#a678cddce777549a39474449d56fd1de6", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#acc490f7f5195acfa7b7c5df7afb39438", null ], + [ "mlx::core::simd::operator/", "namespacemlx_1_1core_1_1simd.html#ac86a54a5e2ccc79bc92739f143bc0bef", null ], + [ "mlx::core::simd::operator/", "namespacemlx_1_1core_1_1simd.html#a18a2689f4ae197c5b204fe9b3370da4c", null ], + [ "mlx::core::simd::operator/", "namespacemlx_1_1core_1_1simd.html#ac5d10f465c21ab259041042ff0159187", null ], + [ "mlx::core::simd::operator<", "namespacemlx_1_1core_1_1simd.html#a6cd6e41660608d17ca8d38658d5e385c", null ], + [ "mlx::core::simd::operator<", "namespacemlx_1_1core_1_1simd.html#ac962a14c88c87082fc70a9c0370f35b0", null ], + [ "mlx::core::simd::operator<", "namespacemlx_1_1core_1_1simd.html#ad9bebf95b37fa0c6517be82af5ccd4eb", null ], + [ "mlx::core::simd::operator<<", "namespacemlx_1_1core_1_1simd.html#ae21cbfd232edd7fe0f6f6c9fa11a354e", null ], + [ "mlx::core::simd::operator<<", "namespacemlx_1_1core_1_1simd.html#a4ecd782ffa497ac7dc2482a232b0dd00", null ], + [ "mlx::core::simd::operator<<", "namespacemlx_1_1core_1_1simd.html#a56fccba38270fe3ae9fa7b2ecdeb5e87", null ], + [ "mlx::core::simd::operator<=", "namespacemlx_1_1core_1_1simd.html#a4d5e4c31af23d2871e09b88c1f6e418c", null ], + [ "mlx::core::simd::operator<=", "namespacemlx_1_1core_1_1simd.html#aadd49786edc08f867e592d234327a031", null ], + [ "mlx::core::simd::operator<=", "namespacemlx_1_1core_1_1simd.html#ae0fcb84973e4762a543ad3843db4f153", null ], + [ "mlx::core::simd::operator==", "namespacemlx_1_1core_1_1simd.html#a273fcc5387c1c9878e658ba6bc32f00c", null ], + [ "mlx::core::simd::operator==", "namespacemlx_1_1core_1_1simd.html#a63768090c16e5dcffccadf550d169abc", null ], + [ "mlx::core::simd::operator==", "namespacemlx_1_1core_1_1simd.html#a46ede415296683771bb22246a813482a", null ], + [ "mlx::core::simd::operator>", "namespacemlx_1_1core_1_1simd.html#abd37e62eff936a64677b5aba787b4d18", null ], + [ "mlx::core::simd::operator>", "namespacemlx_1_1core_1_1simd.html#ab7b291b3559792e18208e17432d25342", null ], + [ "mlx::core::simd::operator>", "namespacemlx_1_1core_1_1simd.html#a71a6902e729e3facdc609e93cd12d485", null ], + [ "mlx::core::simd::operator>=", "namespacemlx_1_1core_1_1simd.html#a87e11ab36aae3328fe3d5230bdf31692", null ], + [ "mlx::core::simd::operator>=", "namespacemlx_1_1core_1_1simd.html#a673b4d8d228f35f06cf5b882335f04d5", null ], + [ "mlx::core::simd::operator>=", "namespacemlx_1_1core_1_1simd.html#a4e65febbfa8b4df2970c1d78801b3c66", null ], + [ "mlx::core::simd::operator>>", "namespacemlx_1_1core_1_1simd.html#a6e45c9c2f0591d9d5dd37a07ebcc3c2a", null ], + [ "mlx::core::simd::operator>>", "namespacemlx_1_1core_1_1simd.html#aebf93b8179621e83bb3f3c4a8816eca8", null ], + [ "mlx::core::simd::operator>>", "namespacemlx_1_1core_1_1simd.html#aa35a2aab733e4bfc80a9f4e3f508daee", null ], + [ "mlx::core::simd::operator^", "namespacemlx_1_1core_1_1simd.html#a25b3de1947dbab7c4864b41ec226453b", null ], + [ "mlx::core::simd::operator^", "namespacemlx_1_1core_1_1simd.html#a16c4a2c8fc59a2e2fcc05db243289706", null ], + [ "mlx::core::simd::operator^", "namespacemlx_1_1core_1_1simd.html#a93e69a8170b8fe14f0a3188b4e8ccd49", null ], + [ "mlx::core::simd::operator|", "namespacemlx_1_1core_1_1simd.html#ab2b540d7329491000e7722f9b3ef797d", null ], + [ "mlx::core::simd::operator|", "namespacemlx_1_1core_1_1simd.html#acd57dc91aa205d9d3f8804df4261a7fb", null ], + [ "mlx::core::simd::operator|", "namespacemlx_1_1core_1_1simd.html#a0cd57bba23daed624df5e2b06b676dca", null ], + [ "mlx::core::simd::operator||", "namespacemlx_1_1core_1_1simd.html#ab380b8f73672727a38ea0931e731fe4a", null ], + [ "mlx::core::simd::operator||", "namespacemlx_1_1core_1_1simd.html#ab2bc61c02b9096163e9db91a3f88788f", null ], + [ "mlx::core::simd::operator||", "namespacemlx_1_1core_1_1simd.html#ac34f6b278627949d2ee68cdbf3d2f50f", null ], + [ "mlx::core::simd::pow", "namespacemlx_1_1core_1_1simd.html#ab18b3a88a2439fd026b6551b38d1f14a", null ], + [ "mlx::core::simd::prod", "namespacemlx_1_1core_1_1simd.html#a271cedfc48efc69db43813e8c424bf7c", null ], + [ "mlx::core::simd::recip", "namespacemlx_1_1core_1_1simd.html#ae344abefc91c7d9c0a9506c868a84d61", null ], + [ "mlx::core::simd::remainder", "namespacemlx_1_1core_1_1simd.html#ab020d2c434fad0cdf79fd37b0f6c1676", null ], + [ "mlx::core::simd::rint", "namespacemlx_1_1core_1_1simd.html#a400d89d040f43d471b306a8e8bdb3974", null ], + [ "mlx::core::simd::rsqrt", "namespacemlx_1_1core_1_1simd.html#aea75ddf8c696efc2e5e924667ed48e70", null ], + [ "mlx::core::simd::select", "namespacemlx_1_1core_1_1simd.html#afb3bcbd8d8b34128cd0c8eb677a170ef", null ], + [ "mlx::core::simd::sinh", "namespacemlx_1_1core_1_1simd.html#a479ccddac341bd0760857b77e449e5e1", null ], + [ "mlx::core::simd::sqrt", "namespacemlx_1_1core_1_1simd.html#ae39b8e1d1fff94947406eeb8ec6e0414", null ], + [ "mlx::core::simd::sum", "namespacemlx_1_1core_1_1simd.html#a53b547b886918dc13d4da88eeb8811d2", null ], + [ "mlx::core::simd::tan", "namespacemlx_1_1core_1_1simd.html#a829842f854aecfae93b7d42f83aec9a7", null ], + [ "mlx::core::simd::tanh", "namespacemlx_1_1core_1_1simd.html#ad78f543dc5da87a14ca113a1dd9852fd", null ], + [ "mlx::core::simd::max_size< double >", "namespacemlx_1_1core_1_1simd.html#a3fa3d1f571027c5cdd1dce5d2cd041e3", null ], + [ "mlx::core::simd::max_size< float >", "namespacemlx_1_1core_1_1simd.html#ae745e117cacfe455df39aa4569c34c11", null ], + [ "mlx::core::simd::max_size< int >", "namespacemlx_1_1core_1_1simd.html#ab25fc96fa6f00d0a8c335b8da293fbbb", null ], + [ "mlx::core::simd::max_size< int16_t >", "namespacemlx_1_1core_1_1simd.html#a7e63a5eb08898b84fd4000dadc460fd9", null ], + [ "mlx::core::simd::max_size< int64_t >", "namespacemlx_1_1core_1_1simd.html#a7913cb2854ffc37efcf26635a097f0a9", null ], + [ "mlx::core::simd::max_size< int8_t >", "namespacemlx_1_1core_1_1simd.html#ac368e4701363cfece4935e57f3c709b1", null ], + [ "mlx::core::simd::max_size< uint16_t >", "namespacemlx_1_1core_1_1simd.html#a0cc9ca2925c25d2eb225af9125bd6bc4", null ], + [ "mlx::core::simd::max_size< uint32_t >", "namespacemlx_1_1core_1_1simd.html#a06cb29f91deeaec69471058044abd2aa", null ], + [ "mlx::core::simd::max_size< uint64_t >", "namespacemlx_1_1core_1_1simd.html#ab367b9b65be2fda4830a56fc9cc0cd2f", null ], + [ "mlx::core::simd::max_size< uint8_t >", "namespacemlx_1_1core_1_1simd.html#a8f731e5a287c714dfc92879fe37503d5", null ] +]; \ No newline at end of file diff --git a/docs/build/html/accelerate__simd_8h_source.html b/docs/build/html/accelerate__simd_8h_source.html new file mode 100644 index 000000000..ef1d9e513 --- /dev/null +++ b/docs/build/html/accelerate__simd_8h_source.html @@ -0,0 +1,545 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/accelerate_simd.h Source File + + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
accelerate_simd.h
+
+
+Go to the documentation of this file.
1#pragma once
+
2
+
3#include <simd/math.h>
+
4#include <simd/vector.h>
+
5
+
6#include <stdint.h>
+
7#include <cmath>
+
8#include <complex>
+
9
+
10#include "mlx/backend/cpu/simd/base_simd.h"
+
11
+
12// There seems to be a bug in sims/base.h
+
13// __XROS_2_0 is not defined, the expression evaluates
+
14// to true instead of false setting the SIMD library
+
15// higher than it should be even on macOS < 15
+
16#if __MAC_OS_X_VERSION_MIN_REQUIRED >= 150000 || \
+
17 __IPHONE_OS_VERSION_MIN_REQUIRED >= 180000 || \
+
18 __WATCH_OS_VERSION_MIN_REQUIRED >= 110000 || \
+
19 __WATCH_OS_VERSION_MIN_REQUIRED >= 110000 || \
+
20 __TV_OS_VERSION_MIN_REQUIRED >= 180000
+
21#define MLX_SIMD_LIBRARY_VERSION 6
+
22#else
+
23#define MLX_SIMD_LIBRARY_VERSION 5
+
24#endif
+
25
+
26namespace mlx::core::simd {
+
27
+
28// Apple simd namespace
+
29namespace asd = ::simd;
+
30
+
31// This indirection is needed to remap certain types to ones that accelerate
+
32// SIMD can handle
+
33template <typename T, int N>
+
+
34struct ScalarT {
+
35 using v = T;
+
36};
+
+
37template <int N>
+
+
38struct ScalarT<bool, N> {
+
39 using v = char;
+
40};
+
+
41template <int N>
+
+
42struct ScalarT<int8_t, N> {
+
43 using v = char;
+
44};
+
+
45template <int N>
+
+
46struct ScalarT<uint64_t, N> {
+
47 using v = unsigned long;
+
48};
+
+
49template <int N>
+
+
50struct ScalarT<int64_t, N> {
+
51 using v = long;
+
52};
+
+
53
+
54template <typename T, int N>
+
+
55struct Simd {
+
+
56 static constexpr int size = N;
+
57 using scalar_t = typename ScalarT<T, N>::v;
+
58
+
59 Simd<T, N>() {}
+
+
60
+
61 template <typename U>
+
62 Simd<T, N>(Simd<U, N> other) : value(asd::convert<scalar_t>(other.value)) {}
+
63
+
64 template <typename U>
+
65 Simd<T, N>(U v) : value(v){};
+
66
+
67 Simd<T, N>(Simd<T, N / 2> x, Simd<T, N / 2> y) {
+
68 value = asd::make<typename asd::Vector<scalar_t, N>::packed_t>(
+
69 x.value, y.value);
+
70 };
+
71
+
+
72 T operator[](int idx) const {
+
73 return reinterpret_cast<const T*>(&value)[idx];
+
74 }
+
+
75
+
+
76 T& operator[](int idx) {
+
77 return reinterpret_cast<T*>(&value)[idx];
+
78 }
+
+
79
+
80 typename asd::Vector<scalar_t, N>::packed_t value;
+
81};
+
+
82
+
83// Values chosen based on benchmarks on M3 Max
+
84// TODO: consider choosing these more optimally
+
85template <>
+
86static constexpr int max_size<int8_t> = 16;
+
87template <>
+
88static constexpr int max_size<int16_t> = 16;
+
89template <>
+
90static constexpr int max_size<int> = 8;
+
91template <>
+
92static constexpr int max_size<int64_t> = 4;
+
93template <>
+
94static constexpr int max_size<uint8_t> = 16;
+
95template <>
+
96static constexpr int max_size<uint16_t> = 16;
+
97template <>
+
98static constexpr int max_size<uint32_t> = 8;
+
99template <>
+
100static constexpr int max_size<uint64_t> = 4;
+
101template <>
+
102static constexpr int max_size<float> = 8;
+
103template <>
+
104static constexpr int max_size<double> = 4;
+
105
+
+
106#define SIMD_DEFAULT_UNARY(name, op) \
+
107 template <typename T, int N> \
+
108 Simd<T, N> name(Simd<T, N> v) { \
+
109 return op(v.value); \
+
110 }
+
+
111
+
112SIMD_DEFAULT_UNARY(abs, asd::abs)
+
113SIMD_DEFAULT_UNARY(floor, asd::floor)
+
114SIMD_DEFAULT_UNARY(acos, asd::acos)
+
115SIMD_DEFAULT_UNARY(acosh, asd::acosh)
+
116SIMD_DEFAULT_UNARY(asin, asd::asin)
+
117SIMD_DEFAULT_UNARY(asinh, asd::asinh)
+
118SIMD_DEFAULT_UNARY(atan, asd::atan)
+
119SIMD_DEFAULT_UNARY(atanh, asd::atanh)
+
120SIMD_DEFAULT_UNARY(ceil, asd::ceil)
+
121SIMD_DEFAULT_UNARY(cosh, asd::cosh)
+
122SIMD_DEFAULT_UNARY(expm1, asd::expm1)
+
123SIMD_DEFAULT_UNARY(log, asd::log)
+
124SIMD_DEFAULT_UNARY(log2, asd::log2)
+
125SIMD_DEFAULT_UNARY(log10, asd::log10)
+
126SIMD_DEFAULT_UNARY(log1p, asd::log1p)
+
127SIMD_DEFAULT_UNARY(rint, asd::rint)
+
128SIMD_DEFAULT_UNARY(sinh, asd::sinh)
+
129SIMD_DEFAULT_UNARY(sqrt, asd::sqrt)
+
130SIMD_DEFAULT_UNARY(rsqrt, asd::rsqrt)
+
131SIMD_DEFAULT_UNARY(recip, asd::recip)
+
132SIMD_DEFAULT_UNARY(tan, asd::tan)
+
133SIMD_DEFAULT_UNARY(tanh, asd::tanh)
+
134
+
135template <typename T, int N>
+
+
136Simd<T, N> operator-(Simd<T, N> v) {
+
137 return -v.value;
+
138}
+
+
139
+
140template <typename T, int N>
+
+
141Simd<bool, N> isnan(Simd<T, N> v) {
+
142 return asd::convert<char>(v.value != v.value);
+
143}
+
+
144
+
145// No simd_boolN in accelerate, use int8_t instead
+
146template <typename T, int N>
+
+
147Simd<bool, N> operator!(Simd<T, N> v) {
+
148 return asd::convert<char>(!v.value);
+
149}
+
+
150
+
+
151#define SIMD_DEFAULT_BINARY(OP) \
+
152 template <typename T, typename U, int N> \
+
153 Simd<T, N> operator OP(Simd<T, N> x, U y) { \
+
154 return asd::convert<typename Simd<T, N>::scalar_t>(x.value OP y); \
+
155 } \
+
156 template <typename T1, typename T2, int N> \
+
157 Simd<T2, N> operator OP(T1 x, Simd<T2, N> y) { \
+
158 return asd::convert<typename Simd<T2, N>::scalar_t>(x OP y.value); \
+
159 } \
+
160 template <typename T1, typename T2, int N> \
+
161 Simd<T1, N> operator OP(Simd<T1, N> x, Simd<T2, N> y) { \
+
162 return asd::convert<typename Simd<T1, N>::scalar_t>(x.value OP y.value); \
+
163 }
+
+
164
+
165SIMD_DEFAULT_BINARY(+)
+
166SIMD_DEFAULT_BINARY(-)
+
167SIMD_DEFAULT_BINARY(/)
+
168SIMD_DEFAULT_BINARY(*)
+
169SIMD_DEFAULT_BINARY(<<)
+
170SIMD_DEFAULT_BINARY(>>)
+
171SIMD_DEFAULT_BINARY(|)
+
172SIMD_DEFAULT_BINARY(^)
+
173SIMD_DEFAULT_BINARY(&)
+
174SIMD_DEFAULT_BINARY(&&)
+
175SIMD_DEFAULT_BINARY(||)
+
176
+
+
177#define SIMD_DEFAULT_COMPARISONS(OP) \
+
178 template <int N, typename T, typename U> \
+
179 Simd<bool, N> operator OP(Simd<T, N> a, U b) { \
+
180 return asd::convert<char>(a.value OP b); \
+
181 } \
+
182 template <int N, typename T, typename U> \
+
183 Simd<bool, N> operator OP(T a, Simd<U, N> b) { \
+
184 return asd::convert<char>(a OP b.value); \
+
185 } \
+
186 template <int N, typename T1, typename T2> \
+
187 Simd<bool, N> operator OP(Simd<T1, N> a, Simd<T2, N> b) { \
+
188 return asd::convert<char>(a.value OP b.value); \
+
189 }
+
+
190
+ + +
195SIMD_DEFAULT_COMPARISONS(==)
+
196SIMD_DEFAULT_COMPARISONS(!=)
+
197
+
198template <typename T, int N>
+
+
199Simd<T, N> atan2(Simd<T, N> a, Simd<T, N> b) {
+
200 return asd::atan2(a.value, b.value);
+
201}
+
+
202
+
203template <typename T, int N>
+
+
204Simd<T, N> maximum(Simd<T, N> a, Simd<T, N> b) {
+
205 // TODO add isnan
+
206 return asd::max(a.value, b.value);
+
207}
+
+
208
+
209template <typename T, int N>
+
+
210Simd<T, N> minimum(Simd<T, N> a, Simd<T, N> b) {
+
211 // TODO add isnan
+
212 return asd::min(a.value, b.value);
+
213}
+
+
214
+
215template <typename T, int N>
+
+
216Simd<T, N> remainder(Simd<T, N> a, Simd<T, N> b) {
+
217 Simd<T, N> r;
+
218 if constexpr (!std::is_integral_v<T>) {
+
219 r = asd::remainder(a.value, b.value);
+
220 } else {
+
221 r = a - b * (a / b);
+
222 }
+
223 if constexpr (std::is_signed_v<T>) {
+
224 auto mask = r != 0 && (r < 0 != b < 0);
+
225 r = select(mask, r + b, r);
+
226 }
+
227 return r;
+
228}
+
+
229
+
230template <typename MaskT, typename T1, typename T2, int N>
+
+
231Simd<T1, N> select(Simd<MaskT, N> mask, Simd<T1, N> x, Simd<T2, N> y) {
+
232 if constexpr (sizeof(T1) == 1) {
+
233 return asd::bitselect(y.value, x.value, asd::convert<char>(mask.value));
+
234 } else if constexpr (sizeof(T1) == 2) {
+
235 return asd::bitselect(y.value, x.value, asd::convert<short>(mask.value));
+
236 } else if constexpr (sizeof(T1) == 4) {
+
237 return asd::bitselect(y.value, x.value, asd::convert<int>(mask.value));
+
238 } else {
+
239 return asd::bitselect(y.value, x.value, asd::convert<long>(mask.value));
+
240 }
+
241}
+
+
242
+
243template <typename T, int N>
+
+
244Simd<T, N> pow(Simd<T, N> base, Simd<T, N> exp) {
+
245 if constexpr (!std::is_integral_v<T>) {
+
246 return asd::pow(base.value, exp.value);
+
247 } else {
+
248 Simd<T, N> res = 1;
+
249 while (any(exp)) {
+
250 res = select(exp & 1, res * base, res);
+
251 base = select(exp, base * base, base);
+
252 exp = exp >> 1;
+
253 }
+
254 return res;
+
255 }
+
256}
+
+
257
+
258template <typename T, int N>
+
+
259Simd<T, N> clamp(Simd<T, N> v, Simd<T, N> min, Simd<T, N> max) {
+
260 return asd::clamp(v.value, min.value, max.value);
+
261}
+
+
262
+
263template <typename T, typename U, int N>
+
+
264Simd<T, N> fma(Simd<T, N> x, Simd<T, N> y, U z) {
+
265 return asd::muladd(x.value, y.value, Simd<T, N>(z).value);
+
266}
+
+
267
+
268// Reductions
+
269
+
270template <typename T, int N>
+
+
271bool all(Simd<T, N> x) {
+
272 return asd::all(x.value);
+
273}
+
+
274template <typename T, int N>
+
+
275bool any(Simd<T, N> x) {
+
276 return asd::any(x.value);
+
277}
+
+
278template <typename T, int N>
+
+
279T sum(Simd<T, N> x) {
+
280 return asd::reduce_add(x.value);
+
281}
+
+
282template <typename T, int N>
+
+
283T max(Simd<T, N> x) {
+
284 return asd::reduce_max(x.value);
+
285}
+
+
286template <typename T, int N>
+
+
287T min(Simd<T, N> x) {
+
288 return asd::reduce_min(x.value);
+
289}
+
+
290
+
291template <typename T, int N>
+
+
292T prod(Simd<T, N> x) {
+
293 auto ptr = (T*)&x;
+
294 auto lhs = load<T, N / 2>(ptr);
+
295 auto rhs = load<T, N / 2>(ptr + N / 2);
+
296 return prod(lhs * rhs);
+
297}
+
+
298
+
299} // namespace mlx::core::simd
+
300
+
301#if __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
+
302#include "mlx/backend/cpu/simd/accelerate_fp16_simd.h"
+
303#endif
+
accelerate_fp16_simd.h
+
SIMD_DEFAULT_UNARY
#define SIMD_DEFAULT_UNARY(name, op)
Definition accelerate_simd.h:106
+
SIMD_DEFAULT_BINARY
#define SIMD_DEFAULT_BINARY(OP)
Definition accelerate_simd.h:151
+
SIMD_DEFAULT_COMPARISONS
#define SIMD_DEFAULT_COMPARISONS(OP)
Definition accelerate_simd.h:177
+
base_simd.h
+ +
mlx::core::simd
Definition accelerate_fp16_simd.h:9
+
mlx::core::simd::isnan
Simd< bool, N > isnan(Simd< T, N > v)
Definition accelerate_simd.h:141
+
mlx::core::simd::sinh
Simd< float16_t, N > sinh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:41
+
mlx::core::simd::N
constexpr int N
Definition neon_fp16_simd.h:9
+
mlx::core::simd::atanh
Simd< float16_t, N > atanh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:34
+
mlx::core::simd::minimum
Simd< T, N > minimum(Simd< T, N > a, Simd< T, N > b)
Definition accelerate_simd.h:210
+
mlx::core::simd::pow
Simd< float16_t, N > pow(Simd< float16_t, N > x, Simd< float16_t, N > y)
Definition accelerate_fp16_simd.h:54
+
mlx::core::simd::atan2
Simd< float16_t, N > atan2(Simd< float16_t, N > x, Simd< float16_t, N > y)
Definition accelerate_fp16_simd.h:52
+
mlx::core::simd::prod
T prod(Simd< T, N > x)
Definition accelerate_simd.h:292
+
mlx::core::simd::log10
Simd< float16_t, N > log10(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:39
+
mlx::core::simd::rint
Simd< T, N > rint(Simd< T, N > v)
Definition accelerate_simd.h:127
+
mlx::core::simd::load
Simd< T, N > load(const T *x)
Definition base_simd.h:27
+
mlx::core::simd::tan
Simd< float16_t, N > tan(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:42
+
mlx::core::simd::abs
Simd< T, N > abs(Simd< T, N > v)
Definition accelerate_simd.h:112
+
mlx::core::simd::acosh
Simd< float16_t, N > acosh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:30
+
mlx::core::simd::all
bool all(Simd< T, N > x)
Definition accelerate_simd.h:271
+
mlx::core::simd::sum
T sum(Simd< T, N > x)
Definition accelerate_simd.h:279
+
mlx::core::simd::log2
Simd< float16_t, N > log2(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:38
+
mlx::core::simd::max
T max(Simd< T, N > x)
Definition accelerate_simd.h:283
+
mlx::core::simd::operator!
Simd< bool, N > operator!(Simd< T, N > v)
Definition accelerate_simd.h:147
+
mlx::core::simd::maximum
Simd< T, N > maximum(Simd< T, N > a, Simd< T, N > b)
Definition accelerate_simd.h:204
+
mlx::core::simd::exp
Simd< T, N > exp(Simd< T, N > in)
Compute exp(x) in an optimizer friendly way as follows:
Definition math.h:28
+
mlx::core::simd::log
Simd< float16_t, N > log(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:37
+
mlx::core::simd::floor
Simd< T, N > floor(Simd< T, N > v)
Definition accelerate_simd.h:113
+
mlx::core::simd::expm1
Simd< float16_t, N > expm1(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:36
+
mlx::core::simd::asin
Simd< float16_t, N > asin(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:31
+
mlx::core::simd::any
bool any(Simd< T, N > x)
Definition accelerate_simd.h:275
+
mlx::core::simd::fma
Simd< T, N > fma(Simd< T, N > x, Simd< T, N > y, U z)
Definition accelerate_simd.h:264
+
mlx::core::simd::tanh
Simd< float16_t, N > tanh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:43
+
mlx::core::simd::atan
Simd< float16_t, N > atan(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:33
+
mlx::core::simd::asinh
Simd< float16_t, N > asinh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:32
+
mlx::core::simd::remainder
Simd< float16_t, N > remainder(Simd< float16_t, N > x, Simd< float16_t, N > y)
Definition accelerate_fp16_simd.h:53
+
mlx::core::simd::max_size
static constexpr int max_size
Definition base_simd.h:13
+
mlx::core::simd::min
T min(Simd< T, N > x)
Definition accelerate_simd.h:287
+
mlx::core::simd::log1p
Simd< float16_t, N > log1p(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:40
+
mlx::core::simd::ceil
Simd< T, N > ceil(Simd< T, N > v)
Definition accelerate_simd.h:120
+
mlx::core::simd::recip
Simd< T, N > recip(Simd< T, N > v)
Definition accelerate_simd.h:131
+
mlx::core::simd::sqrt
Simd< T, N > sqrt(Simd< T, N > v)
Definition accelerate_simd.h:129
+
mlx::core::simd::clamp
Simd< T, N > clamp(Simd< T, N > v, Simd< T, N > min, Simd< T, N > max)
Definition accelerate_simd.h:259
+
mlx::core::simd::acos
Simd< float16_t, N > acos(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:29
+
mlx::core::simd::rsqrt
Simd< T, N > rsqrt(Simd< T, N > v)
Definition accelerate_simd.h:130
+
mlx::core::simd::cosh
Simd< float16_t, N > cosh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:35
+
mlx::core::simd::select
Simd< T1, N > select(Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)
Definition accelerate_simd.h:231
+
mlx::core::simd::ScalarT< bool, N >::v
char v
Definition accelerate_simd.h:39
+
mlx::core::simd::ScalarT< int64_t, N >::v
long v
Definition accelerate_simd.h:51
+
mlx::core::simd::ScalarT< int8_t, N >::v
char v
Definition accelerate_simd.h:43
+
mlx::core::simd::ScalarT< uint64_t, N >::v
unsigned long v
Definition accelerate_simd.h:47
+
mlx::core::simd::ScalarT
Definition accelerate_simd.h:34
+
mlx::core::simd::ScalarT::v
T v
Definition accelerate_simd.h:35
+
mlx::core::simd::Simd
Definition accelerate_simd.h:55
+
mlx::core::simd::Simd::operator[]
T operator[](int idx) const
Definition accelerate_simd.h:72
+
mlx::core::simd::Simd::scalar_t
typename ScalarT< T, N >::v scalar_t
Definition accelerate_simd.h:57
+
mlx::core::simd::Simd::value
asd::Vector< scalar_t, N >::packed_t value
Definition accelerate_simd.h:80
+
mlx::core::simd::Simd::Simd
Simd()
Definition accelerate_simd.h:59
+
mlx::core::simd::Simd::size
static constexpr int size
Definition accelerate_simd.h:56
+
mlx::core::simd::Simd::operator[]
T & operator[](int idx)
Definition accelerate_simd.h:76
+
+
+ + + + diff --git a/docs/build/html/allocator_8h.html b/docs/build/html/allocator_8h.html index 77238d70c..d376ce7cb 100644 --- a/docs/build/html/allocator_8h.html +++ b/docs/build/html/allocator_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/allocator.h File Reference @@ -52,7 +52,7 @@ - + - - - - - - - - - - - - - - -
-
- - - - - - - -
-
MLX -
-
- -   - - - - -
-
-
- - - - -
-
- -
-
-
- -
- -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- -
-
utils.h
-
-
-Go to the documentation of this file.
1// Copyright © 2023-2024 Apple Inc.
-
2
-
3#pragma once
-
4
-
5#include <Accelerate/Accelerate.h>
-
6#include "mlx/dtype.h"
-
7
-
8namespace mlx::core {
-
9
-
-
10BNNSDataType to_bnns_dtype(Dtype mlx_dtype) {
-
11 uint32_t size_bits = size_of(mlx_dtype) * 8;
-
12 switch (kindof(mlx_dtype)) {
-
13 case Dtype::Kind::b:
-
14 return BNNSDataTypeBoolean;
-
15 case Dtype::Kind::u:
-
16 return BNNSDataType(BNNSDataTypeUIntBit | size_bits);
-
17 case Dtype::Kind::i:
-
18 return BNNSDataType(BNNSDataTypeIntBit | size_bits);
-
19 case Dtype::Kind::f:
-
20 return BNNSDataType(BNNSDataTypeFloatBit | size_bits);
-
21 case Dtype::Kind::V:
-
22 return BNNSDataTypeBFloat16;
-
23 case Dtype::Kind::c:
-
24 throw std::invalid_argument("BNNS does not support complex types");
-
25 }
-
26}
-
-
27
-
28} // namespace mlx::core
- -
mlx::core
Definition allocator.h:7
-
mlx::core::to_bnns_dtype
BNNSDataType to_bnns_dtype(Dtype mlx_dtype)
Definition utils.h:10
-
mlx::core::kindof
Dtype::Kind kindof(const Dtype &t)
-
mlx::core::size_of
uint8_t size_of(const Dtype &t)
Definition dtype.h:102
-
mlx::core::Dtype
Definition dtype.h:13
-
mlx::core::Dtype::Kind::c
@ c
Definition dtype.h:35
-
mlx::core::Dtype::Kind::V
@ V
Definition dtype.h:36
-
mlx::core::Dtype::Kind::u
@ u
Definition dtype.h:32
-
mlx::core::Dtype::Kind::i
@ i
Definition dtype.h:33
-
mlx::core::Dtype::Kind::f
@ f
Definition dtype.h:34
-
mlx::core::Dtype::Kind::b
@ b
Definition dtype.h:31
-
-
- - - - diff --git a/docs/build/html/backend_2common_2load_8h.html b/docs/build/html/backend_2common_2load_8h.html index 78fedd71a..83f8d0753 100644 --- a/docs/build/html/backend_2common_2load_8h.html +++ b/docs/build/html/backend_2common_2load_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/common/load.h File Reference @@ -52,7 +52,7 @@ - + - - - - - - - - - - - - - - -
-
- - - - - - - -
-
MLX -
-
- -   - - - - -
-
-
- - - - -
-
- -
-
-
- -
- -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- -
-
ops.h
-
-
-Go to the documentation of this file.
1// Copyright © 2023-2024 Apple Inc.
-
2
-
3#pragma once
-
4#include <stdint.h>
-
5#include <cmath>
-
6#include <complex>
-
7
-
-
8namespace mlx::core::detail {
-
9
-
10namespace {
-
11constexpr float inf = std::numeric_limits<float>::infinity();
-
12} // namespace
-
13
-
-
14typedef union {
-
15 int i;
-
16 float f;
-
17} IntOrFloat;
-
-
18
-
-
19inline float fast_exp(float x) {
-
20 if (x == -std::numeric_limits<float>::infinity()) {
-
21 return 0.0f;
-
22 } else if (x == std::numeric_limits<float>::infinity() || std::isnan(x)) {
-
23 return x;
-
24 }
-
25 x *= 1.442695; // multiply with log_2(e)
-
26 float ipart, fpart;
-
27 IntOrFloat epart;
-
28 x = std::max(-80.f, std::min(x, 80.f));
-
29 ipart = std::floor(x + 0.5);
-
30 fpart = x - ipart;
-
31
-
32 x = 1.535336188319500e-4f;
-
33 x = x * fpart + 1.339887440266574e-3f;
-
34 x = x * fpart + 9.618437357674640e-3f;
-
35 x = x * fpart + 5.550332471162809e-2f;
-
36 x = x * fpart + 2.402264791363012e-1f;
-
37 x = x * fpart + 6.931472028550421e-1f;
-
38 x = x * fpart + 1.000000000000000f;
-
39
-
40 // generate 2**ipart in the floating point representation using integer
-
41 // bitshifting
-
42 epart.i = (int(ipart) + 127) << 23;
-
43
-
44 return epart.f * x;
-
45}
-
-
46
-
-
47inline float fast_erf(float a) {
-
48 float r, s, t, u;
-
49 t = std::abs(a);
-
50 s = a * a;
-
51 if (t > 0.927734375f) {
-
52 // maximum error 0.99527 ulp
-
53 r = std::fma(
-
54 -1.72853470e-5f, t, 3.83197126e-4f); // -0x1.220000p-16,0x1.91cfb2p-12
-
55 u = std::fma(
-
56 -3.88396438e-3f, t, 2.42546219e-2f); // -0x1.fd1438p-9, 0x1.8d6342p-6
-
57 r = std::fma(r, s, u);
-
58 r = std::fma(r, t, -1.06777877e-1f); // -0x1.b55cb8p-4
-
59 r = std::fma(r, t, -6.34846687e-1f); // -0x1.450aa0p-1
-
60 r = std::fma(r, t, -1.28717512e-1f); // -0x1.079d0cp-3
-
61 r = std::fma(r, t, -t);
-
62 // TODO, replace with expm1 when implemented
-
63 r = 1.0f - std::exp(r);
-
64 r = std::copysign(r, a);
-
65 } else {
-
66 // maximum error 0.98929 ulp
-
67 r = -5.96761703e-4f; // -0x1.38e000p-11
-
68 r = std::fma(r, s, 4.99119423e-3f); // 0x1.471a58p-8
-
69 r = std::fma(r, s, -2.67681349e-2f); // -0x1.b691b2p-6
-
70 r = std::fma(r, s, 1.12819925e-1f); // 0x1.ce1c44p-4
-
71 r = std::fma(r, s, -3.76125336e-1f); // -0x1.812700p-2
-
72 r = std::fma(r, s, 1.28379166e-1f); // 0x1.06eba8p-3
-
73 r = std::fma(r, a, a);
-
74 }
-
75 return r;
-
76}
-
-
77
-
-
78inline float fast_erfinv(float a) {
-
79 auto t = std::fma(a, 0.0f - a, 1.0f);
-
80 t = std::log(t);
-
81 float p;
-
82 if (std::abs(t) > 6.125f) { // maximum ulp error = 2.35793
-
83 p = 3.03697567e-10f; // 0x1.4deb44p-32
-
84 p = std::fma(p, t, 2.93243101e-8f); // 0x1.f7c9aep-26
-
85 p = std::fma(p, t, 1.22150334e-6f); // 0x1.47e512p-20
-
86 p = std::fma(p, t, 2.84108955e-5f); // 0x1.dca7dep-16
-
87 p = std::fma(p, t, 3.93552968e-4f); // 0x1.9cab92p-12
-
88 p = std::fma(p, t, 3.02698812e-3f); // 0x1.8cc0dep-9
-
89 p = std::fma(p, t, 4.83185798e-3f); // 0x1.3ca920p-8
-
90 p = std::fma(p, t, -2.64646143e-1f); // -0x1.0eff66p-2
-
91 p = std::fma(p, t, 8.40016484e-1f); // 0x1.ae16a4p-1
-
92 } else { // maximum ulp error = 2.35002
-
93 p = 5.43877832e-9f; // 0x1.75c000p-28
-
94 p = std::fma(p, t, 1.43285448e-7f); // 0x1.33b402p-23
-
95 p = std::fma(p, t, 1.22774793e-6f); // 0x1.499232p-20
-
96 p = std::fma(p, t, 1.12963626e-7f); // 0x1.e52cd2p-24
-
97 p = std::fma(p, t, -5.61530760e-5f); // -0x1.d70bd0p-15
-
98 p = std::fma(p, t, -1.47697632e-4f); // -0x1.35be90p-13
-
99 p = std::fma(p, t, 2.31468678e-3f); // 0x1.2f6400p-9
-
100 p = std::fma(p, t, 1.15392581e-2f); // 0x1.7a1e50p-7
-
101 p = std::fma(p, t, -2.32015476e-1f); // -0x1.db2aeep-3
-
102 p = std::fma(p, t, 8.86226892e-1f); // 0x1.c5bf88p-1
-
103 }
-
104 return a * p;
-
105}
-
-
106
-
-
107struct Abs {
-
108 template <typename T>
-
-
109 T operator()(T x) {
-
110 return std::abs(x);
-
111 }
-
-
-
112 uint8_t operator()(uint8_t x) {
-
113 return x;
-
114 }
-
-
-
115 uint16_t operator()(uint16_t x) {
-
116 return x;
-
117 }
-
-
-
118 uint32_t operator()(uint32_t x) {
-
119 return x;
-
120 }
-
-
-
121 uint64_t operator()(uint64_t x) {
-
122 return x;
-
123 }
-
-
-
124 bool operator()(bool x) {
-
125 return x;
-
126 }
-
-
127};
-
-
128
-
-
129struct ArcCos {
-
130 template <typename T>
-
-
131 T operator()(T x) {
-
132 return std::acos(x);
-
133 }
-
-
134};
-
-
135
-
-
136struct ArcCosh {
-
137 template <typename T>
-
-
138 T operator()(T x) {
-
139 return std::acosh(x);
-
140 }
-
-
141};
-
-
142
-
-
143struct ArcSin {
-
144 template <typename T>
-
-
145 T operator()(T x) {
-
146 return std::asin(x);
-
147 }
-
-
148};
-
-
149
-
-
150struct ArcSinh {
-
151 template <typename T>
-
-
152 T operator()(T x) {
-
153 return std::asinh(x);
-
154 }
-
-
155};
-
-
156
-
-
157struct ArcTan {
-
158 template <typename T>
-
-
159 T operator()(T x) {
-
160 return std::atan(x);
-
161 }
-
-
162};
-
-
163
-
-
164struct ArcTan2 {
-
165 template <typename T>
-
-
166 T operator()(T y, T x) {
-
167 return std::atan2(y, x);
-
168 }
-
-
169};
-
-
170
-
-
171struct ArcTanh {
-
172 template <typename T>
-
-
173 T operator()(T x) {
-
174 return std::atanh(x);
-
175 }
-
-
176};
-
-
177
-
-
178struct Ceil {
-
179 template <typename T>
-
-
180 T operator()(T x) {
-
181 return std::ceil(x);
-
182 }
-
-
-
183 int8_t operator()(int8_t x) {
-
184 return x;
-
185 }
-
-
-
186 int16_t operator()(int16_t x) {
-
187 return x;
-
188 }
-
-
-
189 int32_t operator()(int32_t x) {
-
190 return x;
-
191 }
-
-
-
192 int64_t operator()(int64_t x) {
-
193 return x;
-
194 }
-
-
-
195 uint8_t operator()(uint8_t x) {
-
196 return x;
-
197 }
-
-
-
198 uint16_t operator()(uint16_t x) {
-
199 return x;
-
200 }
-
-
-
201 uint32_t operator()(uint32_t x) {
-
202 return x;
-
203 }
-
-
-
204 uint64_t operator()(uint64_t x) {
-
205 return x;
-
206 }
-
-
-
207 bool operator()(bool x) {
-
208 return x;
-
209 }
-
-
210};
-
-
211
-
-
212struct Conjugate {
-
-
213 complex64_t operator()(complex64_t x) {
-
214 return std::conj(x);
-
215 }
-
-
216};
-
-
217
-
-
218struct Cos {
-
219 template <typename T>
-
-
220 T operator()(T x) {
-
221 return std::cos(x);
-
222 }
-
-
223};
-
-
224
-
-
225struct Cosh {
-
226 template <typename T>
-
-
227 T operator()(T x) {
-
228 return std::cosh(x);
-
229 }
-
-
230};
-
-
231
-
-
232struct Erf {
-
233 template <typename T>
-
-
234 T operator()(T x) {
-
235 return static_cast<T>(fast_erf(static_cast<float>(x)));
-
236 }
-
-
237};
-
-
238
-
-
239struct ErfInv {
-
240 template <typename T>
-
-
241 T operator()(T x) {
-
242 return static_cast<T>(fast_erfinv(static_cast<float>(x)));
-
243 }
-
-
244};
-
-
245
-
-
246struct Exp {
-
247 template <typename T>
-
-
248 T operator()(T x) {
-
249 return fast_exp(x);
-
250 }
-
-
251
-
-
252 complex64_t operator()(complex64_t x) {
-
253 return std::exp(x);
-
254 }
-
-
255};
-
-
256
-
-
257struct Expm1 {
-
258 template <typename T>
-
-
259 T operator()(T x) {
-
260 return expm1(x);
-
261 }
-
-
262};
-
-
263
-
-
264struct Floor {
-
265 template <typename T>
-
-
266 T operator()(T x) {
-
267 return std::floor(x);
-
268 }
-
-
-
269 int8_t operator()(int8_t x) {
-
270 return x;
-
271 }
-
-
-
272 int16_t operator()(int16_t x) {
-
273 return x;
-
274 }
-
-
-
275 int32_t operator()(int32_t x) {
-
276 return x;
-
277 }
-
-
-
278 int64_t operator()(int64_t x) {
-
279 return x;
-
280 }
-
-
-
281 uint8_t operator()(uint8_t x) {
-
282 return x;
-
283 }
-
-
-
284 uint16_t operator()(uint16_t x) {
-
285 return x;
-
286 }
-
-
-
287 uint32_t operator()(uint32_t x) {
-
288 return x;
-
289 }
-
-
-
290 uint64_t operator()(uint64_t x) {
-
291 return x;
-
292 }
-
-
-
293 bool operator()(bool x) {
-
294 return x;
-
295 }
-
-
296};
-
-
297
-
-
298struct Imag {
-
299 template <typename T>
-
-
300 T operator()(T x) {
-
301 return std::imag(x);
-
302 }
-
-
303};
-
-
304
-
-
305struct Log {
-
306 template <typename T>
-
-
307 T operator()(T x) {
-
308 return std::log(x);
-
309 }
-
-
310};
-
-
311
-
-
312struct Log2 {
-
313 template <typename T>
-
-
314 T operator()(T x) {
-
315 return std::log2(x);
-
316 }
-
-
317};
-
-
318
-
-
319struct Log10 {
-
320 template <typename T>
-
-
321 T operator()(T x) {
-
322 return std::log10(x);
-
323 }
-
-
324};
-
-
325
-
-
326struct Log1p {
-
327 template <typename T>
-
-
328 T operator()(T x) {
-
329 return log1p(x);
-
330 }
-
-
331};
-
-
332
-
-
333struct LogicalNot {
-
334 template <typename T>
-
-
335 T operator()(T x) {
-
336 return !x;
-
337 }
-
-
338};
-
-
339
-
-
340struct Negative {
-
341 template <typename T>
-
-
342 T operator()(T x) {
-
343 return -x;
-
344 }
-
-
345};
-
-
346
-
-
347struct Real {
-
348 template <typename T>
-
-
349 T operator()(T x) {
-
350 return std::real(x);
-
351 }
-
-
352};
-
-
353
-
-
354struct Round {
-
355 template <typename T>
-
-
356 T operator()(T x) {
-
357 return std::rint(x);
-
358 }
-
-
359
-
-
360 complex64_t operator()(complex64_t x) {
-
361 return {std::rint(x.real()), std::rint(x.imag())};
-
362 }
-
-
363};
-
-
364
-
-
365struct Sigmoid {
-
366 template <typename T>
-
-
367 T operator()(T x) {
-
368 auto one = static_cast<decltype(x)>(1.0);
-
369 return one / (one + fast_exp(-x));
-
370 }
-
-
371};
-
-
372
-
-
373struct Sign {
-
374 template <typename T>
-
-
375 T operator()(T x) {
-
376 return (x > T(0)) - (x < T(0));
-
377 }
-
-
-
378 uint8_t operator()(uint8_t x) {
-
379 return x != 0;
-
380 }
-
-
-
381 uint16_t operator()(uint16_t x) {
-
382 return x != 0;
-
383 }
-
-
-
384 uint32_t operator()(uint32_t x) {
-
385 return x != 0;
-
386 }
-
-
-
387 uint64_t operator()(uint64_t x) {
-
388 return x != 0;
-
389 }
-
-
390
-
-
391 complex64_t operator()(complex64_t x) {
-
392 return x == complex64_t(0) ? x : x / std::abs(x);
-
393 }
-
-
394};
-
-
395
-
-
396struct Sin {
-
397 template <typename T>
-
-
398 T operator()(T x) {
-
399 return std::sin(x);
-
400 }
-
-
401};
-
-
402
-
-
403struct Sinh {
-
404 template <typename T>
-
-
405 T operator()(T x) {
-
406 return std::sinh(x);
-
407 }
-
-
408};
-
-
409
-
-
410struct Square {
-
411 template <typename T>
-
-
412 T operator()(T x) {
-
413 return x * x;
-
414 }
-
-
415};
-
-
416
-
-
417struct Sqrt {
-
418 template <typename T>
-
-
419 T operator()(T x) {
-
420 return std::sqrt(x);
-
421 }
-
-
422};
-
-
423
-
-
424struct Rsqrt {
-
425 template <typename T>
-
-
426 T operator()(T x) {
-
427 return static_cast<decltype(x)>(1.0) / std::sqrt(x);
-
428 }
-
-
429};
-
-
430
-
-
431struct Tan {
-
432 template <typename T>
-
-
433 T operator()(T x) {
-
434 return std::tan(x);
-
435 }
-
-
436};
-
-
437
-
-
438struct Tanh {
-
439 template <typename T>
-
-
440 T operator()(T x) {
-
441 return std::tanh(x);
-
442 }
-
-
443};
-
-
444
-
-
445struct Add {
-
446 template <typename T>
-
-
447 T operator()(T x, T y) {
-
448 return x + y;
-
449 }
-
-
450};
-
-
451
-
-
452struct Divide {
-
453 template <typename T>
-
-
454 T operator()(T x, T y) {
-
455 return x / y;
-
456 }
-
-
457};
-
-
458
-
-
459struct Remainder {
-
460 template <typename T>
-
-
461 std::enable_if_t<std::is_integral_v<T> & !std::is_signed_v<T>, T> operator()(
-
462 T numerator,
-
463 T denominator) {
-
464 return numerator % denominator;
-
465 }
-
-
466
-
467 template <typename T>
-
-
468 std::enable_if_t<std::is_integral_v<T> & std::is_signed_v<T>, T> operator()(
-
469 T numerator,
-
470 T denominator) {
-
471 auto r = numerator % denominator;
-
472 if (r != 0 && (r < 0 != denominator < 0))
-
473 r += denominator;
-
474 return r;
-
475 }
-
-
476
-
477 template <typename T>
-
-
478 std::enable_if_t<!std::is_integral_v<T>, T> operator()(
-
479 T numerator,
-
480 T denominator) {
-
481 auto r = std::fmod(numerator, denominator);
-
482 if (r != 0 && (r < 0 != denominator < 0)) {
-
483 r += denominator;
-
484 }
-
485 return r;
-
486 }
-
-
487
-
-
488 complex64_t operator()(complex64_t numerator, complex64_t denominator) {
-
489 return numerator % denominator;
-
490 }
-
-
491};
-
-
492
-
-
493struct Equal {
-
494 template <typename T>
-
-
495 bool operator()(T x, T y) {
-
496 return x == y;
-
497 }
-
-
498};
-
-
499
-
-
500struct NaNEqual {
-
501 template <typename T>
-
-
502 bool operator()(T x, T y) {
-
503 if constexpr (std::is_integral_v<T>) {
-
504 // isnan always returns false for integers, and MSVC refuses to compile.
-
505 return x == y;
-
506 } else {
-
507 return x == y || (std::isnan(x) && std::isnan(y));
-
508 }
-
509 }
-
-
510};
-
-
511
-
-
512struct Greater {
-
513 template <typename T>
-
-
514 bool operator()(T x, T y) {
-
515 return x > y;
-
516 }
-
-
517};
-
-
518
-
-
519struct GreaterEqual {
-
520 template <typename T>
-
-
521 bool operator()(T x, T y) {
-
522 return x >= y;
-
523 }
-
-
524};
-
-
525
-
-
526struct Less {
-
527 template <typename T>
-
-
528 bool operator()(T x, T y) {
-
529 return x < y;
-
530 }
-
-
531};
-
-
532
-
-
533struct LessEqual {
-
534 template <typename T>
-
-
535 bool operator()(T x, T y) {
-
536 return x <= y;
-
537 }
-
-
538};
-
-
539
-
-
540struct Maximum {
-
541 template <typename T>
-
-
542 std::enable_if_t<std::is_integral_v<T>, T> operator()(T x, T y) {
-
543 return (x > y) ? x : y;
-
544 }
-
-
545
-
546 template <typename T>
-
-
547 std::enable_if_t<!std::is_integral_v<T>, T> operator()(T x, T y) {
-
548 if (std::isnan(x)) {
-
549 return x;
-
550 }
-
551 return (x > y) ? x : y;
-
552 }
-
-
553};
-
-
554
-
-
555struct Minimum {
-
556 template <typename T>
-
-
557 std::enable_if_t<std::is_integral_v<T>, T> operator()(T x, T y) {
-
558 return x < y ? x : y;
-
559 }
-
-
560
-
561 template <typename T>
-
-
562 std::enable_if_t<!std::is_integral_v<T>, T> operator()(T x, T y) {
-
563 if (std::isnan(x)) {
-
564 return x;
-
565 }
-
566 return x < y ? x : y;
-
567 }
-
-
568};
-
-
569
-
-
570struct LogAddExp {
-
571 template <typename T>
-
-
572 T operator()(T x, T y) {
-
573 constexpr float inf = std::numeric_limits<float>::infinity();
-
574 auto maxval = Maximum()(x, y);
-
575 auto minval = Minimum()(x, y);
-
576 return (minval == -inf || maxval == inf)
-
577 ? maxval
-
578 : static_cast<decltype(x)>(
-
579 maxval + std::log1p(fast_exp(minval - maxval)));
-
580 }
-
-
581};
-
-
582
-
-
583struct Multiply {
-
584 template <typename T>
-
-
585 T operator()(T x, T y) {
-
586 return x * y;
-
587 }
-
-
588};
-
-
589
-
-
590struct NotEqual {
-
591 template <typename T>
-
-
592 bool operator()(T x, T y) {
-
593 return x != y;
-
594 }
-
-
595};
-
-
596
-
-
597struct Power {
-
598 template <typename T>
-
-
599 std::enable_if_t<!std::is_integral_v<T>, T> operator()(T base, T exp) {
-
600 return std::pow(base, exp);
-
601 }
-
-
602
-
603 template <typename T>
-
-
604 std::enable_if_t<std::is_integral_v<T>, T> operator()(T base, T exp) {
-
605 T res = 1;
-
606 while (exp) {
-
607 if (exp & 1) {
-
608 res *= base;
-
609 }
-
610 exp >>= 1;
-
611 base *= base;
-
612 }
-
613 return res;
-
614 }
-
-
615};
-
-
616
-
-
617struct Subtract {
-
618 template <typename T>
-
-
619 T operator()(T x, T y) {
-
620 return x - y;
-
621 }
-
-
622};
-
-
623
-
-
624struct LogicalAnd {
-
625 template <typename T>
-
-
626 T operator()(T x, T y) {
-
627 return x && y;
-
628 }
-
-
629};
-
-
630
-
-
631struct LogicalOr {
-
632 template <typename T>
-
-
633 T operator()(T x, T y) {
-
634 return x || y;
-
635 }
-
-
636};
-
-
637
-
-
638struct Select {
-
639 template <typename T>
-
-
640 T operator()(bool condition, T x, T y) {
-
641 return condition ? x : y;
-
642 }
-
-
643};
-
-
644
-
-
645struct BitwiseAnd {
-
646 template <typename T>
-
-
647 T operator()(T x, T y) {
-
648 return x & y;
-
649 }
-
-
650};
-
-
651
-
-
652struct BitwiseOr {
-
653 template <typename T>
-
-
654 T operator()(T x, T y) {
-
655 return x | y;
-
656 }
-
-
657};
-
-
658
-
-
659struct BitwiseXor {
-
660 template <typename T>
-
-
661 T operator()(T x, T y) {
-
662 return x ^ y;
-
663 }
-
-
664};
-
-
665
-
-
666struct LeftShift {
-
667 template <typename T>
-
-
668 T operator()(T x, T y) {
-
669 return x << y;
-
670 }
-
-
671};
-
-
672
-
-
673struct RightShift {
-
674 template <typename T>
-
-
675 T operator()(T x, T y) {
-
676 return x >> y;
-
677 }
-
-
678};
-
-
679
-
680} // namespace mlx::core::detail
-
-
mlx::core::log1p
array log1p(const array &a, StreamOrDevice s={})
Natural logarithm of one plus elements in the array: log(1 + a).
-
mlx::core::expm1
array expm1(const array &a, StreamOrDevice s={})
Computes the expm1 function of the elements of an array.
-
mlx::core::exp
array exp(const array &a, StreamOrDevice s={})
Exponential of the elements of an array.
-
mlx::core::detail
Definition ops.h:8
-
mlx::core::detail::fast_exp
float fast_exp(float x)
Definition ops.h:19
-
mlx::core::detail::fast_erf
float fast_erf(float a)
Definition ops.h:47
-
mlx::core::detail::fast_erfinv
float fast_erfinv(float a)
Definition ops.h:78
-
mlx::core::complex64_t
Definition complex.h:34
-
mlx::core::detail::Abs
Definition ops.h:107
-
mlx::core::detail::Abs::operator()
T operator()(T x)
Definition ops.h:109
-
mlx::core::detail::Abs::operator()
uint8_t operator()(uint8_t x)
Definition ops.h:112
-
mlx::core::detail::Abs::operator()
uint64_t operator()(uint64_t x)
Definition ops.h:121
-
mlx::core::detail::Abs::operator()
uint16_t operator()(uint16_t x)
Definition ops.h:115
-
mlx::core::detail::Abs::operator()
bool operator()(bool x)
Definition ops.h:124
-
mlx::core::detail::Abs::operator()
uint32_t operator()(uint32_t x)
Definition ops.h:118
-
mlx::core::detail::Add
Definition ops.h:445
-
mlx::core::detail::Add::operator()
T operator()(T x, T y)
Definition ops.h:447
-
mlx::core::detail::ArcCos
Definition ops.h:129
-
mlx::core::detail::ArcCos::operator()
T operator()(T x)
Definition ops.h:131
-
mlx::core::detail::ArcCosh
Definition ops.h:136
-
mlx::core::detail::ArcCosh::operator()
T operator()(T x)
Definition ops.h:138
-
mlx::core::detail::ArcSin
Definition ops.h:143
-
mlx::core::detail::ArcSin::operator()
T operator()(T x)
Definition ops.h:145
-
mlx::core::detail::ArcSinh
Definition ops.h:150
-
mlx::core::detail::ArcSinh::operator()
T operator()(T x)
Definition ops.h:152
-
mlx::core::detail::ArcTan2
Definition ops.h:164
-
mlx::core::detail::ArcTan2::operator()
T operator()(T y, T x)
Definition ops.h:166
-
mlx::core::detail::ArcTan
Definition ops.h:157
-
mlx::core::detail::ArcTan::operator()
T operator()(T x)
Definition ops.h:159
-
mlx::core::detail::ArcTanh
Definition ops.h:171
-
mlx::core::detail::ArcTanh::operator()
T operator()(T x)
Definition ops.h:173
-
mlx::core::detail::BitwiseAnd
Definition ops.h:645
-
mlx::core::detail::BitwiseAnd::operator()
T operator()(T x, T y)
Definition ops.h:647
-
mlx::core::detail::BitwiseOr
Definition ops.h:652
-
mlx::core::detail::BitwiseOr::operator()
T operator()(T x, T y)
Definition ops.h:654
-
mlx::core::detail::BitwiseXor
Definition ops.h:659
-
mlx::core::detail::BitwiseXor::operator()
T operator()(T x, T y)
Definition ops.h:661
-
mlx::core::detail::Ceil
Definition ops.h:178
-
mlx::core::detail::Ceil::operator()
uint8_t operator()(uint8_t x)
Definition ops.h:195
-
mlx::core::detail::Ceil::operator()
T operator()(T x)
Definition ops.h:180
-
mlx::core::detail::Ceil::operator()
uint32_t operator()(uint32_t x)
Definition ops.h:201
-
mlx::core::detail::Ceil::operator()
int8_t operator()(int8_t x)
Definition ops.h:183
-
mlx::core::detail::Ceil::operator()
int16_t operator()(int16_t x)
Definition ops.h:186
-
mlx::core::detail::Ceil::operator()
bool operator()(bool x)
Definition ops.h:207
-
mlx::core::detail::Ceil::operator()
uint16_t operator()(uint16_t x)
Definition ops.h:198
-
mlx::core::detail::Ceil::operator()
uint64_t operator()(uint64_t x)
Definition ops.h:204
-
mlx::core::detail::Ceil::operator()
int32_t operator()(int32_t x)
Definition ops.h:189
-
mlx::core::detail::Ceil::operator()
int64_t operator()(int64_t x)
Definition ops.h:192
-
mlx::core::detail::Conjugate
Definition ops.h:212
-
mlx::core::detail::Conjugate::operator()
complex64_t operator()(complex64_t x)
Definition ops.h:213
-
mlx::core::detail::Cos
Definition ops.h:218
-
mlx::core::detail::Cos::operator()
T operator()(T x)
Definition ops.h:220
-
mlx::core::detail::Cosh
Definition ops.h:225
-
mlx::core::detail::Cosh::operator()
T operator()(T x)
Definition ops.h:227
-
mlx::core::detail::Divide
Definition ops.h:452
-
mlx::core::detail::Divide::operator()
T operator()(T x, T y)
Definition ops.h:454
-
mlx::core::detail::Equal
Definition ops.h:493
-
mlx::core::detail::Equal::operator()
bool operator()(T x, T y)
Definition ops.h:495
-
mlx::core::detail::Erf
Definition ops.h:232
-
mlx::core::detail::Erf::operator()
T operator()(T x)
Definition ops.h:234
-
mlx::core::detail::ErfInv
Definition ops.h:239
-
mlx::core::detail::ErfInv::operator()
T operator()(T x)
Definition ops.h:241
-
mlx::core::detail::Exp
Definition ops.h:246
-
mlx::core::detail::Exp::operator()
T operator()(T x)
Definition ops.h:248
-
mlx::core::detail::Exp::operator()
complex64_t operator()(complex64_t x)
Definition ops.h:252
-
mlx::core::detail::Expm1
Definition ops.h:257
-
mlx::core::detail::Expm1::operator()
T operator()(T x)
Definition ops.h:259
-
mlx::core::detail::Floor
Definition ops.h:264
-
mlx::core::detail::Floor::operator()
T operator()(T x)
Definition ops.h:266
-
mlx::core::detail::Floor::operator()
uint32_t operator()(uint32_t x)
Definition ops.h:287
-
mlx::core::detail::Floor::operator()
uint16_t operator()(uint16_t x)
Definition ops.h:284
-
mlx::core::detail::Floor::operator()
uint8_t operator()(uint8_t x)
Definition ops.h:281
-
mlx::core::detail::Floor::operator()
int32_t operator()(int32_t x)
Definition ops.h:275
-
mlx::core::detail::Floor::operator()
int64_t operator()(int64_t x)
Definition ops.h:278
-
mlx::core::detail::Floor::operator()
bool operator()(bool x)
Definition ops.h:293
-
mlx::core::detail::Floor::operator()
int8_t operator()(int8_t x)
Definition ops.h:269
-
mlx::core::detail::Floor::operator()
uint64_t operator()(uint64_t x)
Definition ops.h:290
-
mlx::core::detail::Floor::operator()
int16_t operator()(int16_t x)
Definition ops.h:272
- -
mlx::core::detail::GreaterEqual::operator()
bool operator()(T x, T y)
Definition ops.h:521
-
mlx::core::detail::Greater
Definition ops.h:512
-
mlx::core::detail::Greater::operator()
bool operator()(T x, T y)
Definition ops.h:514
-
mlx::core::detail::Imag
Definition ops.h:298
-
mlx::core::detail::Imag::operator()
T operator()(T x)
Definition ops.h:300
-
mlx::core::detail::LeftShift
Definition ops.h:666
-
mlx::core::detail::LeftShift::operator()
T operator()(T x, T y)
Definition ops.h:668
-
mlx::core::detail::LessEqual
Definition ops.h:533
-
mlx::core::detail::LessEqual::operator()
bool operator()(T x, T y)
Definition ops.h:535
-
mlx::core::detail::Less
Definition ops.h:526
-
mlx::core::detail::Less::operator()
bool operator()(T x, T y)
Definition ops.h:528
-
mlx::core::detail::Log10
Definition ops.h:319
-
mlx::core::detail::Log10::operator()
T operator()(T x)
Definition ops.h:321
-
mlx::core::detail::Log1p
Definition ops.h:326
-
mlx::core::detail::Log1p::operator()
T operator()(T x)
Definition ops.h:328
-
mlx::core::detail::Log2
Definition ops.h:312
-
mlx::core::detail::Log2::operator()
T operator()(T x)
Definition ops.h:314
-
mlx::core::detail::LogAddExp
Definition ops.h:570
-
mlx::core::detail::LogAddExp::operator()
T operator()(T x, T y)
Definition ops.h:572
-
mlx::core::detail::Log
Definition ops.h:305
-
mlx::core::detail::Log::operator()
T operator()(T x)
Definition ops.h:307
-
mlx::core::detail::LogicalAnd
Definition ops.h:624
-
mlx::core::detail::LogicalAnd::operator()
T operator()(T x, T y)
Definition ops.h:626
-
mlx::core::detail::LogicalNot
Definition ops.h:333
-
mlx::core::detail::LogicalNot::operator()
T operator()(T x)
Definition ops.h:335
-
mlx::core::detail::LogicalOr
Definition ops.h:631
-
mlx::core::detail::LogicalOr::operator()
T operator()(T x, T y)
Definition ops.h:633
-
mlx::core::detail::Maximum
Definition ops.h:540
-
mlx::core::detail::Maximum::operator()
std::enable_if_t< std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:542
-
mlx::core::detail::Maximum::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:547
-
mlx::core::detail::Minimum
Definition ops.h:555
-
mlx::core::detail::Minimum::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:562
-
mlx::core::detail::Minimum::operator()
std::enable_if_t< std::is_integral_v< T >, T > operator()(T x, T y)
Definition ops.h:557
-
mlx::core::detail::Multiply
Definition ops.h:583
-
mlx::core::detail::Multiply::operator()
T operator()(T x, T y)
Definition ops.h:585
-
mlx::core::detail::NaNEqual
Definition ops.h:500
-
mlx::core::detail::NaNEqual::operator()
bool operator()(T x, T y)
Definition ops.h:502
-
mlx::core::detail::Negative
Definition ops.h:340
-
mlx::core::detail::Negative::operator()
T operator()(T x)
Definition ops.h:342
-
mlx::core::detail::NotEqual
Definition ops.h:590
-
mlx::core::detail::NotEqual::operator()
bool operator()(T x, T y)
Definition ops.h:592
-
mlx::core::detail::Power
Definition ops.h:597
-
mlx::core::detail::Power::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T base, T exp)
Definition ops.h:599
-
mlx::core::detail::Power::operator()
std::enable_if_t< std::is_integral_v< T >, T > operator()(T base, T exp)
Definition ops.h:604
-
mlx::core::detail::Real
Definition ops.h:347
-
mlx::core::detail::Real::operator()
T operator()(T x)
Definition ops.h:349
-
mlx::core::detail::Remainder
Definition ops.h:459
-
mlx::core::detail::Remainder::operator()
std::enable_if_t<!std::is_integral_v< T >, T > operator()(T numerator, T denominator)
Definition ops.h:478
-
mlx::core::detail::Remainder::operator()
std::enable_if_t< std::is_integral_v< T > &!std::is_signed_v< T >, T > operator()(T numerator, T denominator)
Definition ops.h:461
-
mlx::core::detail::Remainder::operator()
std::enable_if_t< std::is_integral_v< T > &std::is_signed_v< T >, T > operator()(T numerator, T denominator)
Definition ops.h:468
-
mlx::core::detail::Remainder::operator()
complex64_t operator()(complex64_t numerator, complex64_t denominator)
Definition ops.h:488
-
mlx::core::detail::RightShift
Definition ops.h:673
-
mlx::core::detail::RightShift::operator()
T operator()(T x, T y)
Definition ops.h:675
-
mlx::core::detail::Round
Definition ops.h:354
-
mlx::core::detail::Round::operator()
T operator()(T x)
Definition ops.h:356
-
mlx::core::detail::Round::operator()
complex64_t operator()(complex64_t x)
Definition ops.h:360
-
mlx::core::detail::Rsqrt
Definition ops.h:424
-
mlx::core::detail::Rsqrt::operator()
T operator()(T x)
Definition ops.h:426
-
mlx::core::detail::Select
Definition ops.h:638
-
mlx::core::detail::Select::operator()
T operator()(bool condition, T x, T y)
Definition ops.h:640
-
mlx::core::detail::Sigmoid
Definition ops.h:365
-
mlx::core::detail::Sigmoid::operator()
T operator()(T x)
Definition ops.h:367
-
mlx::core::detail::Sign
Definition ops.h:373
-
mlx::core::detail::Sign::operator()
uint64_t operator()(uint64_t x)
Definition ops.h:387
-
mlx::core::detail::Sign::operator()
T operator()(T x)
Definition ops.h:375
-
mlx::core::detail::Sign::operator()
uint8_t operator()(uint8_t x)
Definition ops.h:378
-
mlx::core::detail::Sign::operator()
uint16_t operator()(uint16_t x)
Definition ops.h:381
-
mlx::core::detail::Sign::operator()
complex64_t operator()(complex64_t x)
Definition ops.h:391
-
mlx::core::detail::Sign::operator()
uint32_t operator()(uint32_t x)
Definition ops.h:384
-
mlx::core::detail::Sin
Definition ops.h:396
-
mlx::core::detail::Sin::operator()
T operator()(T x)
Definition ops.h:398
-
mlx::core::detail::Sinh
Definition ops.h:403
-
mlx::core::detail::Sinh::operator()
T operator()(T x)
Definition ops.h:405
-
mlx::core::detail::Sqrt
Definition ops.h:417
-
mlx::core::detail::Sqrt::operator()
T operator()(T x)
Definition ops.h:419
-
mlx::core::detail::Square
Definition ops.h:410
-
mlx::core::detail::Square::operator()
T operator()(T x)
Definition ops.h:412
-
mlx::core::detail::Subtract
Definition ops.h:617
-
mlx::core::detail::Subtract::operator()
T operator()(T x, T y)
Definition ops.h:619
-
mlx::core::detail::Tan
Definition ops.h:431
-
mlx::core::detail::Tan::operator()
T operator()(T x)
Definition ops.h:433
-
mlx::core::detail::Tanh
Definition ops.h:438
-
mlx::core::detail::Tanh::operator()
T operator()(T x)
Definition ops.h:440
- -
mlx::core::detail::IntOrFloat::f
float f
Definition ops.h:16
-
mlx::core::detail::IntOrFloat::i
int i
Definition ops.h:15
-
-
- - - - diff --git a/docs/build/html/backend_2common_2utils_8h.html b/docs/build/html/backend_2common_2utils_8h.html index ca7b1bf4f..88924b13c 100644 --- a/docs/build/html/backend_2common_2utils_8h.html +++ b/docs/build/html/backend_2common_2utils_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/common/utils.h File Reference @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
+Namespaces | +Functions
+
event.h File Reference
+
+
+ +

Go to the source code of this file.

+ + + + + + +

+Namespaces

namespace  mlx
 
namespace  mlx::core
 
+ + + + + +

+Functions

void mlx::core::encode_wait (Event e)
 
void mlx::core::encode_signal (Event e)
 
+
+
+ + + + diff --git a/docs/build/html/backend_2metal_2event_8h.js b/docs/build/html/backend_2metal_2event_8h.js new file mode 100644 index 000000000..e2170ce94 --- /dev/null +++ b/docs/build/html/backend_2metal_2event_8h.js @@ -0,0 +1,5 @@ +var backend_2metal_2event_8h = +[ + [ "mlx::core::encode_signal", "namespacemlx_1_1core.html#a6d452306f0f046a7d021bd94f8713a89", null ], + [ "mlx::core::encode_wait", "namespacemlx_1_1core.html#a2874ba55b73057b76c23a7429fdd2d6e", null ] +]; \ No newline at end of file diff --git a/docs/build/html/backend_2metal_2event_8h_source.html b/docs/build/html/backend_2metal_2event_8h_source.html new file mode 100644 index 000000000..7fe5da87d --- /dev/null +++ b/docs/build/html/backend_2metal_2event_8h_source.html @@ -0,0 +1,132 @@ + + + + + + + +MLX: mlx/backend/metal/event.h Source File + + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
event.h
+
+
+Go to the documentation of this file.
1// Copyright © 2024 Apple Inc.
+
2#pragma once
+
3
+
4namespace mlx::core {
+
5
+
6void encode_wait(Event e);
+
7
+
8void encode_signal(Event e);
+
9
+
10} // namespace mlx::core
+
mlx::core::Event
Definition event.h:11
+
mlx::core
Definition allocator.h:7
+
mlx::core::encode_wait
void encode_wait(Event e)
+
mlx::core::encode_signal
void encode_signal(Event e)
+
+
+ + + + diff --git a/docs/build/html/backend_2metal_2kernels_2complex_8h.html b/docs/build/html/backend_2metal_2kernels_2complex_8h.html index c306930cc..765f17760 100644 --- a/docs/build/html/backend_2metal_2kernels_2complex_8h.html +++ b/docs/build/html/backend_2metal_2kernels_2complex_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/complex.h File Reference @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
+Classes | +Namespaces | +Macros | +Functions | +Variables
+
base_simd.h File Reference
+
+
+
#include <stdint.h>
+#include <algorithm>
+#include <cmath>
+#include <complex>
+
+

Go to the source code of this file.

+ + + + +

+Classes

struct  mlx::core::simd::Simd< T, 1 >
 
+ + + + + + + +

+Namespaces

namespace  mlx
 
namespace  mlx::core
 
namespace  mlx::core::simd
 
+ + + + + + + + + +

+Macros

#define DEFAULT_UNARY(name, op)
 
#define DEFAULT_BINARY(OP)
 
#define DEFAULT_COMPARISONS(OP)
 
#define DEFAULT_REDUCTION(name, type)
 
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

+Functions

template<typename T, int N>
Simd< T, N > mlx::core::simd::load (const T *x)
 
template<typename T, int N>
void mlx::core::simd::store (T *dst, Simd< T, N > x)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::rint (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::rsqrt (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::recip (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::operator- (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::operator! (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::abs (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::acos (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::acosh (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::asin (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::asinh (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::atan (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::atanh (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::ceil (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::conj (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::cosh (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::expm1 (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::floor (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::log (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::log2 (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::log10 (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::log1p (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::sinh (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::sqrt (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::tan (Simd< T, 1 > in)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::tanh (Simd< T, 1 > in)
 
template<typename T>
auto mlx::core::simd::real (Simd< T, 1 > in) -> Simd< decltype(std::real(in.value)), 1 >
 
template<typename T>
auto mlx::core::simd::imag (Simd< T, 1 > in) -> Simd< decltype(std::imag(in.value)), 1 >
 
template<typename T>
Simd< bool, 1 > mlx::core::simd::isnan (Simd< T, 1 > in)
 
template<typename T1, typename T2>
auto mlx::core::simd::operator+ (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value+b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator+ (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a+b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator+ (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value+b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator- (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value - b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator- (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a - b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator- (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value - b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator* (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value *b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator* (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a *b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator* (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value *b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator/ (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value/b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator/ (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a/b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator/ (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value/b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator<< (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value<< b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator<< (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a<< b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator<< (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value<< b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator>> (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value > > b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator>> (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a > > b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator>> (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value > > b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator| (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value|b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator| (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a|b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator| (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value|b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator^ (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value ^ b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator^ (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a ^ b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator^ (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value ^ b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator& (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value &b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator& (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a &b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator& (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value &b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator&& (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value &&b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator&& (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a &&b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator&& (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value &&b), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator|| (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value||b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator|| (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a||b.value), 1 >
 
template<typename T1, typename T2>
auto mlx::core::simd::operator|| (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value||b), 1 >
 
template<typename T>
Simd< T, 1 > mlx::core::simd::remainder (Simd< T, 1 > a_, Simd< T, 1 > b_)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::maximum (Simd< T, 1 > a_, Simd< T, 1 > b_)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::minimum (Simd< T, 1 > a_, Simd< T, 1 > b_)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::pow (Simd< T, 1 > a, Simd< T, 1 > b)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::atan2 (Simd< T, 1 > a, Simd< T, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator> (Simd< T1, 1 > a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator> (T1 a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator> (Simd< T1, 1 > a, T2 b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator< (Simd< T1, 1 > a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator< (T1 a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator< (Simd< T1, 1 > a, T2 b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator>= (Simd< T1, 1 > a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator>= (T1 a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator>= (Simd< T1, 1 > a, T2 b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator<= (Simd< T1, 1 > a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator<= (T1 a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator<= (Simd< T1, 1 > a, T2 b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator== (Simd< T1, 1 > a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator== (T1 a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator== (Simd< T1, 1 > a, T2 b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator!= (Simd< T1, 1 > a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator!= (T1 a, Simd< T2, 1 > b)
 
template<typename T1, typename T2>
Simd< bool, 1 > mlx::core::simd::operator!= (Simd< T1, 1 > a, T2 b)
 
template<typename MaskT, typename T>
Simd< T, 1 > mlx::core::simd::select (Simd< MaskT, 1 > mask, Simd< T, 1 > x, Simd< T, 1 > y)
 
template<typename T>
Simd< T, 1 > mlx::core::simd::clamp (Simd< T, 1 > v, Simd< T, 1 > min, Simd< T, 1 > max)
 
template<typename T, typename U>
Simd< T, 1 > mlx::core::simd::fma (Simd< T, 1 > x, Simd< T, 1 > y, U z)
 
template<typename T>
T mlx::core::simd::max (Simd< T, 1 > x)
 
template<typename T>
T mlx::core::simd::min (Simd< T, 1 > x)
 
template<typename T>
T mlx::core::simd::sum (Simd< T, 1 > x)
 
template<typename T>
T mlx::core::simd::prod (Simd< T, 1 > x)
 
template<typename T>
bool mlx::core::simd::any (Simd< T, 1 > x)
 
template<typename T>
bool mlx::core::simd::all (Simd< T, 1 > x)
 
+ + + + + + + + + + +

+Variables

template<typename T>
static constexpr int mlx::core::simd::max_size = 1
 
template<typename, typename = void>
constexpr bool mlx::core::simd::is_complex = false
 
template<typename T>
constexpr bool mlx::core::simd::is_complex< T, std::void_t< decltype(std::declval< T >().real())> >
 
+

Macro Definition Documentation

+ +

◆ DEFAULT_BINARY

+ +
+
+ + + + + + + +
#define DEFAULT_BINARY( OP)
+
+Value:
template <typename T1, typename T2> \
+
auto operator OP(Simd<T1, 1> a, Simd<T2, 1> b) \
+
->Simd<decltype(a.value OP b.value), 1> { \
+
return a.value OP b.value; \
+
} \
+
template <typename T1, typename T2> \
+
auto operator OP(T1 a, Simd<T2, 1> b)->Simd<decltype(a OP b.value), 1> { \
+
return a OP b.value; \
+
} \
+
template <typename T1, typename T2> \
+
auto operator OP(Simd<T1, 1> a, T2 b)->Simd<decltype(a.value OP b), 1> { \
+
return a.value OP b; \
+
}
+
+
+
+ +

◆ DEFAULT_COMPARISONS

+ +
+
+ + + + + + + +
#define DEFAULT_COMPARISONS( OP)
+
+Value:
template <typename T1, typename T2> \
+
Simd<bool, 1> operator OP(Simd<T1, 1> a, Simd<T2, 1> b) { \
+
return a.value OP b.value; \
+
} \
+
template <typename T1, typename T2> \
+
Simd<bool, 1> operator OP(T1 a, Simd<T2, 1> b) { \
+
return a OP b.value; \
+
} \
+
template <typename T1, typename T2> \
+
Simd<bool, 1> operator OP(Simd<T1, 1> a, T2 b) { \
+
return a.value OP b; \
+
}
+
+
+
+ +

◆ DEFAULT_REDUCTION

+ +
+
+ + + + + + + + + + + +
#define DEFAULT_REDUCTION( name,
type )
+
+Value:
template <typename T> \
+
type name(Simd<T, 1> x) { \
+
return x.value; \
+
}
+
+
+
+ +

◆ DEFAULT_UNARY

+ +
+
+ + + + + + + + + + + +
#define DEFAULT_UNARY( name,
op )
+
+Value:
template <typename T> \
+
Simd<T, 1> name(Simd<T, 1> in) { \
+
return op(in.value); \
+
}
+
+
+
+
+
+ + + + diff --git a/docs/build/html/base__simd_8h.js b/docs/build/html/base__simd_8h.js new file mode 100644 index 000000000..b68085333 --- /dev/null +++ b/docs/build/html/base__simd_8h.js @@ -0,0 +1,106 @@ +var base__simd_8h = +[ + [ "mlx::core::simd::Simd< T, 1 >", "structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html", "structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4" ], + [ "DEFAULT_BINARY", "base__simd_8h.html#a1d01fad7230543d56f4facf17dcc4047", null ], + [ "DEFAULT_COMPARISONS", "base__simd_8h.html#af5153f51cdd18d56d1de24b5da9f4366", null ], + [ "DEFAULT_REDUCTION", "base__simd_8h.html#a0adf6d25084019eff671abc59031573e", null ], + [ "DEFAULT_UNARY", "base__simd_8h.html#a122d6a2fa4dcfe78b221e52155419124", null ], + [ "mlx::core::simd::abs", "namespacemlx_1_1core_1_1simd.html#a1958f026f26f313d17155ac87ea6eca3", null ], + [ "mlx::core::simd::acos", "namespacemlx_1_1core_1_1simd.html#a3d4f9d08d1902e3d62c6f63d39329dbd", null ], + [ "mlx::core::simd::acosh", "namespacemlx_1_1core_1_1simd.html#a51071c8104494b5bd8097990da3bf943", null ], + [ "mlx::core::simd::all", "namespacemlx_1_1core_1_1simd.html#a4ba3690489c2bf861e22e1175255438c", null ], + [ "mlx::core::simd::any", "namespacemlx_1_1core_1_1simd.html#ae3b138b4bbcee0ca70b58a3e2ebd818c", null ], + [ "mlx::core::simd::asin", "namespacemlx_1_1core_1_1simd.html#a75349994f899aecb68553c2247580163", null ], + [ "mlx::core::simd::asinh", "namespacemlx_1_1core_1_1simd.html#a9323e370f6740651ebfd51367985d0e2", null ], + [ "mlx::core::simd::atan", "namespacemlx_1_1core_1_1simd.html#a421845a6f68f88c58f520d2c1fa15914", null ], + [ "mlx::core::simd::atan2", "namespacemlx_1_1core_1_1simd.html#af5e8e8230c7d7af8201a3aaa7f491a2d", null ], + [ "mlx::core::simd::atanh", "namespacemlx_1_1core_1_1simd.html#a5ebae2e6cce1889513f15be3adb265ea", null ], + [ "mlx::core::simd::ceil", "namespacemlx_1_1core_1_1simd.html#a4e54bd4ceb51ec41b0f95ebabe558713", null ], + [ "mlx::core::simd::clamp", "namespacemlx_1_1core_1_1simd.html#a567c06bf988af03988478679055a6c45", null ], + [ "mlx::core::simd::conj", "namespacemlx_1_1core_1_1simd.html#a660b79a51fb439f4aba91e2aea276300", null ], + [ "mlx::core::simd::cosh", "namespacemlx_1_1core_1_1simd.html#ae1265896d855818d20f2de2a9ebb684a", null ], + [ "mlx::core::simd::expm1", "namespacemlx_1_1core_1_1simd.html#a8f73d1dac82177e0aeadaeda349c4f96", null ], + [ "mlx::core::simd::floor", "namespacemlx_1_1core_1_1simd.html#aa396efa6e9c94f4ac1f8381d5e07f069", null ], + [ "mlx::core::simd::fma", "namespacemlx_1_1core_1_1simd.html#a8aa81ebff4c26f21cae2253d885fd87a", null ], + [ "mlx::core::simd::imag", "namespacemlx_1_1core_1_1simd.html#a97c69b04852ccba242f1348fda17ca20", null ], + [ "mlx::core::simd::isnan", "namespacemlx_1_1core_1_1simd.html#ac1c6c9b8bc7f3cd32ae39fa84975194d", null ], + [ "mlx::core::simd::load", "namespacemlx_1_1core_1_1simd.html#a4041676517d96870293e5448c7e2b5a4", null ], + [ "mlx::core::simd::log", "namespacemlx_1_1core_1_1simd.html#abd09d3f5989558ce5156549a94d0fb04", null ], + [ "mlx::core::simd::log10", "namespacemlx_1_1core_1_1simd.html#adf754ade6cc1dd0e0bae0e31c7b513a2", null ], + [ "mlx::core::simd::log1p", "namespacemlx_1_1core_1_1simd.html#a92fcc8037ddb767bff517814ab55c259", null ], + [ "mlx::core::simd::log2", "namespacemlx_1_1core_1_1simd.html#af9eafa15692dec783860ddae3dd8c072", null ], + [ "mlx::core::simd::max", "namespacemlx_1_1core_1_1simd.html#a312ecd0ae1c38d32147cee71fd8539d7", null ], + [ "mlx::core::simd::maximum", "namespacemlx_1_1core_1_1simd.html#ab54ff0f073be504e8428912f8e21effd", null ], + [ "mlx::core::simd::min", "namespacemlx_1_1core_1_1simd.html#a96db878d780a8da6abad19ac772d08ca", null ], + [ "mlx::core::simd::minimum", "namespacemlx_1_1core_1_1simd.html#ac836568622a3e5957c275e115e2fcaf3", null ], + [ "mlx::core::simd::operator!", "namespacemlx_1_1core_1_1simd.html#afaa6ce61de4d80a4b7e9b2ab7454fff4", null ], + [ "mlx::core::simd::operator!=", "namespacemlx_1_1core_1_1simd.html#a757838b9d56e132e797a381d3bb0dc86", null ], + [ "mlx::core::simd::operator!=", "namespacemlx_1_1core_1_1simd.html#a7f1cebaff9cb88df59b5ec7557b5d167", null ], + [ "mlx::core::simd::operator!=", "namespacemlx_1_1core_1_1simd.html#ae8ca6615d51866d876b5efb3425600ed", null ], + [ "mlx::core::simd::operator&", "namespacemlx_1_1core_1_1simd.html#a3c42ac1dc74f6c0bb934dfa45986875b", null ], + [ "mlx::core::simd::operator&", "namespacemlx_1_1core_1_1simd.html#a3a060a225b6ead483ca93247c9ad8e4d", null ], + [ "mlx::core::simd::operator&", "namespacemlx_1_1core_1_1simd.html#a8beb567724ab9735b616afb777b93abd", null ], + [ "mlx::core::simd::operator&&", "namespacemlx_1_1core_1_1simd.html#a82676bd32059d1172296f8074a841de6", null ], + [ "mlx::core::simd::operator&&", "namespacemlx_1_1core_1_1simd.html#a1eca7cf07b2a238307459c28204319fb", null ], + [ "mlx::core::simd::operator&&", "namespacemlx_1_1core_1_1simd.html#af97917ef704103c6ea1d0e44f22ec0d3", null ], + [ "mlx::core::simd::operator*", "namespacemlx_1_1core_1_1simd.html#ab6a73491bcb185cd91ae4db6b0f21e49", null ], + [ "mlx::core::simd::operator*", "namespacemlx_1_1core_1_1simd.html#acd5ac48dc7895f06daf55f0a7e0667fb", null ], + [ "mlx::core::simd::operator*", "namespacemlx_1_1core_1_1simd.html#a4030444ea38ce1529a8cbb8c183a28bd", null ], + [ "mlx::core::simd::operator+", "namespacemlx_1_1core_1_1simd.html#a27dfc3843dbefbbebed5b7137bacbb59", null ], + [ "mlx::core::simd::operator+", "namespacemlx_1_1core_1_1simd.html#a98b77f1ca24bff373f48ef62f0013a02", null ], + [ "mlx::core::simd::operator+", "namespacemlx_1_1core_1_1simd.html#aa78806bf6a3be64b44e9a1f04bad3862", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#a70563bcd6c28802d11199812ffef38c8", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#ab1f7f553d3a9176a70404a29cad06619", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#ab35a129d6e31b86c06b61252c7b26d4e", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#aa73282cb05b65b931b97ce35c46bae20", null ], + [ "mlx::core::simd::operator/", "namespacemlx_1_1core_1_1simd.html#a1d45c3b97cecfff86a2e43ae1f7fa185", null ], + [ "mlx::core::simd::operator/", "namespacemlx_1_1core_1_1simd.html#a1c61bd3ac3ec5d8d2da65b45d59f543e", null ], + [ "mlx::core::simd::operator/", "namespacemlx_1_1core_1_1simd.html#a89be64949908f19dd42aa7e38b320b0c", null ], + [ "mlx::core::simd::operator<", "namespacemlx_1_1core_1_1simd.html#a23b59272b0760326844fffe20db9b3e2", null ], + [ "mlx::core::simd::operator<", "namespacemlx_1_1core_1_1simd.html#acf35d81032bb9043804fd1de43540f60", null ], + [ "mlx::core::simd::operator<", "namespacemlx_1_1core_1_1simd.html#a01259c9188e6ecd48979cdc2fd766372", null ], + [ "mlx::core::simd::operator<<", "namespacemlx_1_1core_1_1simd.html#a33232e2342d5a3e542c9428924a25830", null ], + [ "mlx::core::simd::operator<<", "namespacemlx_1_1core_1_1simd.html#a68e7b952915e629d246d1ffac98b54ce", null ], + [ "mlx::core::simd::operator<<", "namespacemlx_1_1core_1_1simd.html#a50044315dc365f026830416f6b615c77", null ], + [ "mlx::core::simd::operator<=", "namespacemlx_1_1core_1_1simd.html#aec6783f79ca181d6782a810ffb267482", null ], + [ "mlx::core::simd::operator<=", "namespacemlx_1_1core_1_1simd.html#a914e821c358e05dfe8d0208888646793", null ], + [ "mlx::core::simd::operator<=", "namespacemlx_1_1core_1_1simd.html#a05240b8fd6f54632b676d4b66449f799", null ], + [ "mlx::core::simd::operator==", "namespacemlx_1_1core_1_1simd.html#a7928482ed5d25932be80413c7239125c", null ], + [ "mlx::core::simd::operator==", "namespacemlx_1_1core_1_1simd.html#a4877ae5406d081680b785a86ad656e03", null ], + [ "mlx::core::simd::operator==", "namespacemlx_1_1core_1_1simd.html#a51de2acf3dcd55c7c52e3ce7ed6ed9d7", null ], + [ "mlx::core::simd::operator>", "namespacemlx_1_1core_1_1simd.html#ad8b67f9ced9c7f3cb472b9c3df817f08", null ], + [ "mlx::core::simd::operator>", "namespacemlx_1_1core_1_1simd.html#ac971bfa5c7ec8abc432eab5f3c5646aa", null ], + [ "mlx::core::simd::operator>", "namespacemlx_1_1core_1_1simd.html#a4113a94fb8dcd0d88f14ec9d82089508", null ], + [ "mlx::core::simd::operator>=", "namespacemlx_1_1core_1_1simd.html#a530ac8728e4d7e7be2482d5b2467906c", null ], + [ "mlx::core::simd::operator>=", "namespacemlx_1_1core_1_1simd.html#a034d7b57cb3c6ca711c573515327d1a8", null ], + [ "mlx::core::simd::operator>=", "namespacemlx_1_1core_1_1simd.html#ac7f3848b48c8e23c71c85fcc9909b933", null ], + [ "mlx::core::simd::operator>>", "namespacemlx_1_1core_1_1simd.html#a1108d186d57c2010c743d3f9297befc7", null ], + [ "mlx::core::simd::operator>>", "namespacemlx_1_1core_1_1simd.html#a4bf8c887eb6943563ceb1e603d1325b1", null ], + [ "mlx::core::simd::operator>>", "namespacemlx_1_1core_1_1simd.html#a9ac36abfb7dffc7ad24b4d0c295452e5", null ], + [ "mlx::core::simd::operator^", "namespacemlx_1_1core_1_1simd.html#a369178519e0e91fa936c0fd4aa9ee109", null ], + [ "mlx::core::simd::operator^", "namespacemlx_1_1core_1_1simd.html#a16fa3c809e46b5cae3e8abfaf98199a4", null ], + [ "mlx::core::simd::operator^", "namespacemlx_1_1core_1_1simd.html#a5b877b5eb7044d9b2a42a9af4af21f01", null ], + [ "mlx::core::simd::operator|", "namespacemlx_1_1core_1_1simd.html#a60805b5f57ddbbf74f700b54cd3fc4f8", null ], + [ "mlx::core::simd::operator|", "namespacemlx_1_1core_1_1simd.html#a6449faa1666afe1186d55b61bb3e5b5a", null ], + [ "mlx::core::simd::operator|", "namespacemlx_1_1core_1_1simd.html#a495d15a18ee4a6dda22e37e8dc02e45b", null ], + [ "mlx::core::simd::operator||", "namespacemlx_1_1core_1_1simd.html#a7a1c3be1c37d41e450469f2e98cd9dde", null ], + [ "mlx::core::simd::operator||", "namespacemlx_1_1core_1_1simd.html#aad2d440fbb9e5478b5ed24400a859942", null ], + [ "mlx::core::simd::operator||", "namespacemlx_1_1core_1_1simd.html#a0c8bd67982681ecd53cd8d739be3a5a9", null ], + [ "mlx::core::simd::pow", "namespacemlx_1_1core_1_1simd.html#afe3d50bc4a11061898aa57377fa9536d", null ], + [ "mlx::core::simd::prod", "namespacemlx_1_1core_1_1simd.html#ac27cdc630e86b25ad607ca409de2b274", null ], + [ "mlx::core::simd::real", "namespacemlx_1_1core_1_1simd.html#acdf822b7626bbab6a495552aea3457b5", null ], + [ "mlx::core::simd::recip", "namespacemlx_1_1core_1_1simd.html#afc915aed256295475ac88fde3a736f1f", null ], + [ "mlx::core::simd::remainder", "namespacemlx_1_1core_1_1simd.html#a54c7f2f2b995eb767462b1228982967f", null ], + [ "mlx::core::simd::rint", "namespacemlx_1_1core_1_1simd.html#a797196eccc3690aac5c45e5f9c804ceb", null ], + [ "mlx::core::simd::rsqrt", "namespacemlx_1_1core_1_1simd.html#a74ac0fd799967b0f303bfd26fc6a17cf", null ], + [ "mlx::core::simd::select", "namespacemlx_1_1core_1_1simd.html#a9e3e7b35d564c70de8fa0b6150570ed8", null ], + [ "mlx::core::simd::sinh", "namespacemlx_1_1core_1_1simd.html#a40879bf874309c0a5abef783aea2057d", null ], + [ "mlx::core::simd::sqrt", "namespacemlx_1_1core_1_1simd.html#a2ba6c75c0821db3e9ac525a89b3ac859", null ], + [ "mlx::core::simd::store", "namespacemlx_1_1core_1_1simd.html#afa2236afddfdec312eb7e27b89a5316a", null ], + [ "mlx::core::simd::sum", "namespacemlx_1_1core_1_1simd.html#a96ce7d90b3b8b6dddab36ef5b49fffc2", null ], + [ "mlx::core::simd::tan", "namespacemlx_1_1core_1_1simd.html#a5abc381a85fe8b0e9cb472f874704652", null ], + [ "mlx::core::simd::tanh", "namespacemlx_1_1core_1_1simd.html#ab80a7db8d99e3f4032e761c60216027d", null ], + [ "mlx::core::simd::is_complex", "namespacemlx_1_1core_1_1simd.html#a64e80f096a8baf99ba8d396414473cc7", null ], + [ "mlx::core::simd::is_complex< T, std::void_t< decltype(std::declval< T >().real())> >", "namespacemlx_1_1core_1_1simd.html#a29fe8445e54a61f6bccc8d50f142ca54", null ], + [ "mlx::core::simd::max_size", "namespacemlx_1_1core_1_1simd.html#ac91bd36c7caafd3c7ff176e7e2f81887", null ] +]; \ No newline at end of file diff --git a/docs/build/html/base__simd_8h_source.html b/docs/build/html/base__simd_8h_source.html new file mode 100644 index 000000000..450652fc7 --- /dev/null +++ b/docs/build/html/base__simd_8h_source.html @@ -0,0 +1,476 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/base_simd.h Source File + + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
base_simd.h
+
+
+Go to the documentation of this file.
1#pragma once
+
2
+
3#include <stdint.h>
+
4#include <algorithm>
+
5#include <cmath>
+
6#include <complex>
+
7
+
8namespace mlx::core::simd {
+
9template <typename T, int N>
+
10struct Simd;
+
11
+
12template <typename T>
+
13static constexpr int max_size = 1;
+
14
+
15template <typename T>
+
+
16struct Simd<T, 1> {
+
17 static constexpr int size = 1;
+
18 T value;
+
19 Simd() {}
+
20 template <typename U>
+
21 Simd(Simd<U, 1> v) : value(v.value) {}
+
22 template <typename U>
+
23 Simd(U v) : value(v) {}
+
24};
+
+
25
+
26template <typename T, int N>
+
+
27Simd<T, N> load(const T* x) {
+
28 return *(Simd<T, N>*)x;
+
29}
+
+
30
+
31template <typename T, int N>
+
+
32void store(T* dst, Simd<T, N> x) {
+
33 // Maintain invariant that bool is either 0 or 1 as
+
34 // simd comparison ops set all bits in the result to 1
+
35 if constexpr (std::is_same_v<T, bool> && N > 1) {
+
36 x = x & 1;
+
37 }
+
38 *(Simd<T, N>*)dst = x;
+
39}
+
+
40
+
41template <typename, typename = void>
+
42constexpr bool is_complex = false;
+
43
+
44template <typename T>
+
45constexpr bool is_complex<T, std::void_t<decltype(std::declval<T>().real())>> =
+
46 true;
+
47
+
48template <typename T>
+
+
49Simd<T, 1> rint(Simd<T, 1> in) {
+
50 if constexpr (is_complex<T>) {
+
51 return Simd<T, 1>{
+
52 T{std::rint(in.value.real()), std::rint(in.value.imag())}};
+
53 } else {
+
54 return Simd<T, 1>{std::rint(in.value)};
+
55 }
+
56}
+
+
57
+
58template <typename T>
+
+
59Simd<T, 1> rsqrt(Simd<T, 1> in) {
+
60 return T(1.0) / sqrt(in);
+
61}
+
+
62
+
63template <typename T>
+
+
64Simd<T, 1> recip(Simd<T, 1> in) {
+
65 return T(1.0) / in;
+
66}
+
+
67
+
+
68#define DEFAULT_UNARY(name, op) \
+
69 template <typename T> \
+
70 Simd<T, 1> name(Simd<T, 1> in) { \
+
71 return op(in.value); \
+
72 }
+
+
73
+
74DEFAULT_UNARY(operator-, std::negate{})
+
75DEFAULT_UNARY(operator!, std::logical_not{})
+
76DEFAULT_UNARY(abs, std::abs)
+
77DEFAULT_UNARY(acos, std::acos)
+
78DEFAULT_UNARY(acosh, std::acosh)
+
79DEFAULT_UNARY(asin, std::asin)
+
80DEFAULT_UNARY(asinh, std::asinh)
+
81DEFAULT_UNARY(atan, std::atan)
+
82DEFAULT_UNARY(atanh, std::atanh)
+
83DEFAULT_UNARY(ceil, std::ceil)
+
84DEFAULT_UNARY(conj, std::conj)
+
85DEFAULT_UNARY(cosh, std::cosh)
+
86DEFAULT_UNARY(expm1, std::expm1)
+
87DEFAULT_UNARY(floor, std::floor)
+
88DEFAULT_UNARY(log, std::log)
+
89DEFAULT_UNARY(log2, std::log2)
+
90DEFAULT_UNARY(log10, std::log10)
+
91DEFAULT_UNARY(log1p, std::log1p)
+
92DEFAULT_UNARY(sinh, std::sinh)
+
93DEFAULT_UNARY(sqrt, std::sqrt)
+
94DEFAULT_UNARY(tan, std::tan)
+
95DEFAULT_UNARY(tanh, std::tanh)
+
96
+
97template <typename T>
+
+
98auto real(Simd<T, 1> in) -> Simd<decltype(std::real(in.value)), 1> {
+
99 return std::real(in.value);
+
100}
+
+
101template <typename T>
+
+
102auto imag(Simd<T, 1> in) -> Simd<decltype(std::imag(in.value)), 1> {
+
103 return std::imag(in.value);
+
104}
+
+
105template <typename T>
+
+
106Simd<bool, 1> isnan(Simd<T, 1> in) {
+
107 return std::isnan(in.value);
+
108}
+
+
109
+
+
110#define DEFAULT_BINARY(OP) \
+
111 template <typename T1, typename T2> \
+
112 auto operator OP(Simd<T1, 1> a, Simd<T2, 1> b) \
+
113 ->Simd<decltype(a.value OP b.value), 1> { \
+
114 return a.value OP b.value; \
+
115 } \
+
116 template <typename T1, typename T2> \
+
117 auto operator OP(T1 a, Simd<T2, 1> b)->Simd<decltype(a OP b.value), 1> { \
+
118 return a OP b.value; \
+
119 } \
+
120 template <typename T1, typename T2> \
+
121 auto operator OP(Simd<T1, 1> a, T2 b)->Simd<decltype(a.value OP b), 1> { \
+
122 return a.value OP b; \
+
123 }
+
+
124
+
125DEFAULT_BINARY(+)
+
126DEFAULT_BINARY(-)
+
127DEFAULT_BINARY(*)
+
128DEFAULT_BINARY(/)
+
129DEFAULT_BINARY(<<)
+
130DEFAULT_BINARY(>>)
+
131DEFAULT_BINARY(|)
+
132DEFAULT_BINARY(^)
+
133DEFAULT_BINARY(&)
+
134DEFAULT_BINARY(&&)
+
135DEFAULT_BINARY(||)
+
136
+
137template <typename T>
+
+
138Simd<T, 1> remainder(Simd<T, 1> a_, Simd<T, 1> b_) {
+
139 T a = a_.value;
+
140 T b = b_.value;
+
141 T r;
+
142 if constexpr (std::is_integral_v<T>) {
+
143 r = a % b;
+
144 } else {
+
145 r = std::remainder(a, b);
+
146 }
+
147 if constexpr (std::is_signed_v<T>) {
+
148 if (r != 0 && (r < 0 != b < 0)) {
+
149 r += b;
+
150 }
+
151 }
+
152 return r;
+
153}
+
+
154
+
155template <typename T>
+
+
156Simd<T, 1> maximum(Simd<T, 1> a_, Simd<T, 1> b_) {
+
157 T a = a_.value;
+
158 T b = b_.value;
+
159 if constexpr (!std::is_integral_v<T>) {
+
160 if (std::isnan(a)) {
+
161 return a;
+
162 }
+
163 }
+
164 return (a > b) ? a : b;
+
165}
+
+
166
+
167template <typename T>
+
+
168Simd<T, 1> minimum(Simd<T, 1> a_, Simd<T, 1> b_) {
+
169 T a = a_.value;
+
170 T b = b_.value;
+
171 if constexpr (!std::is_integral_v<T>) {
+
172 if (std::isnan(a)) {
+
173 return a;
+
174 }
+
175 }
+
176 return (a < b) ? a : b;
+
177}
+
+
178
+
179template <typename T>
+
+
180Simd<T, 1> pow(Simd<T, 1> a, Simd<T, 1> b) {
+
181 T base = a.value;
+
182 T exp = b.value;
+
183 if constexpr (!std::is_integral_v<T>) {
+
184 return std::pow(base, exp);
+
185 } else {
+
186 T res = 1;
+
187 while (exp) {
+
188 if (exp & 1) {
+
189 res *= base;
+
190 }
+
191 exp >>= 1;
+
192 base *= base;
+
193 }
+
194 return res;
+
195 }
+
196}
+
+
197
+
198template <typename T>
+
+
199Simd<T, 1> atan2(Simd<T, 1> a, Simd<T, 1> b) {
+
200 return std::atan2(a.value, b.value);
+
201}
+
+
202
+
+
203#define DEFAULT_COMPARISONS(OP) \
+
204 template <typename T1, typename T2> \
+
205 Simd<bool, 1> operator OP(Simd<T1, 1> a, Simd<T2, 1> b) { \
+
206 return a.value OP b.value; \
+
207 } \
+
208 template <typename T1, typename T2> \
+
209 Simd<bool, 1> operator OP(T1 a, Simd<T2, 1> b) { \
+
210 return a OP b.value; \
+
211 } \
+
212 template <typename T1, typename T2> \
+
213 Simd<bool, 1> operator OP(Simd<T1, 1> a, T2 b) { \
+
214 return a.value OP b; \
+
215 }
+
+
216
+ + +
221DEFAULT_COMPARISONS(==)
+
222DEFAULT_COMPARISONS(!=)
+
223
+
224template <typename MaskT, typename T>
+
+
225Simd<T, 1> select(Simd<MaskT, 1> mask, Simd<T, 1> x, Simd<T, 1> y) {
+
226 return mask.value ? x.value : y.value;
+
227}
+
+
228
+
229template <typename T>
+
+
230Simd<T, 1> clamp(Simd<T, 1> v, Simd<T, 1> min, Simd<T, 1> max) {
+
231 return std::clamp(v.value, min.value, max.value);
+
232}
+
+
233
+
234template <typename T, typename U>
+
+
235Simd<T, 1> fma(Simd<T, 1> x, Simd<T, 1> y, U z) {
+
236 return std::fma(x.value, y.value, Simd<T, 1>(z).value);
+
237}
+
+
238
+
239// Reductions
+
+
240#define DEFAULT_REDUCTION(name, type) \
+
241 template <typename T> \
+
242 type name(Simd<T, 1> x) { \
+
243 return x.value; \
+
244 }
+
+
245
+
246DEFAULT_REDUCTION(max, T)
+
247DEFAULT_REDUCTION(min, T)
+
248DEFAULT_REDUCTION(sum, T)
+
249DEFAULT_REDUCTION(prod, T)
+
250DEFAULT_REDUCTION(any, bool)
+
251DEFAULT_REDUCTION(all, bool)
+
252
+
253} // namespace mlx::core::simd
+
DEFAULT_REDUCTION
#define DEFAULT_REDUCTION(name, type)
Definition base_simd.h:240
+
DEFAULT_UNARY
#define DEFAULT_UNARY(name, op)
Definition base_simd.h:68
+
DEFAULT_BINARY
#define DEFAULT_BINARY(OP)
Definition base_simd.h:110
+
DEFAULT_COMPARISONS
#define DEFAULT_COMPARISONS(OP)
Definition base_simd.h:203
+
mlx::core::std
array std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})
Computes the standard deviation of the elements of an array.
+
mlx::core::simd
Definition accelerate_fp16_simd.h:9
+
mlx::core::simd::isnan
Simd< bool, N > isnan(Simd< T, N > v)
Definition accelerate_simd.h:141
+
mlx::core::simd::sinh
Simd< float16_t, N > sinh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:41
+
mlx::core::simd::N
constexpr int N
Definition neon_fp16_simd.h:9
+
mlx::core::simd::atanh
Simd< float16_t, N > atanh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:34
+
mlx::core::simd::minimum
Simd< T, N > minimum(Simd< T, N > a, Simd< T, N > b)
Definition accelerate_simd.h:210
+
mlx::core::simd::pow
Simd< float16_t, N > pow(Simd< float16_t, N > x, Simd< float16_t, N > y)
Definition accelerate_fp16_simd.h:54
+
mlx::core::simd::atan2
Simd< float16_t, N > atan2(Simd< float16_t, N > x, Simd< float16_t, N > y)
Definition accelerate_fp16_simd.h:52
+
mlx::core::simd::prod
T prod(Simd< T, N > x)
Definition accelerate_simd.h:292
+
mlx::core::simd::log10
Simd< float16_t, N > log10(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:39
+
mlx::core::simd::rint
Simd< T, N > rint(Simd< T, N > v)
Definition accelerate_simd.h:127
+
mlx::core::simd::load
Simd< T, N > load(const T *x)
Definition base_simd.h:27
+
mlx::core::simd::tan
Simd< float16_t, N > tan(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:42
+
mlx::core::simd::abs
Simd< T, N > abs(Simd< T, N > v)
Definition accelerate_simd.h:112
+
mlx::core::simd::acosh
Simd< float16_t, N > acosh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:30
+
mlx::core::simd::all
bool all(Simd< T, N > x)
Definition accelerate_simd.h:271
+
mlx::core::simd::sum
T sum(Simd< T, N > x)
Definition accelerate_simd.h:279
+
mlx::core::simd::is_complex
constexpr bool is_complex
Definition base_simd.h:42
+
mlx::core::simd::conj
Simd< T, 1 > conj(Simd< T, 1 > in)
Definition base_simd.h:84
+
mlx::core::simd::log2
Simd< float16_t, N > log2(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:38
+
mlx::core::simd::max
T max(Simd< T, N > x)
Definition accelerate_simd.h:283
+
mlx::core::simd::maximum
Simd< T, N > maximum(Simd< T, N > a, Simd< T, N > b)
Definition accelerate_simd.h:204
+
mlx::core::simd::exp
Simd< T, N > exp(Simd< T, N > in)
Compute exp(x) in an optimizer friendly way as follows:
Definition math.h:28
+
mlx::core::simd::log
Simd< float16_t, N > log(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:37
+
mlx::core::simd::floor
Simd< T, N > floor(Simd< T, N > v)
Definition accelerate_simd.h:113
+
mlx::core::simd::expm1
Simd< float16_t, N > expm1(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:36
+
mlx::core::simd::imag
auto imag(Simd< T, 1 > in) -> Simd< decltype(std::imag(in.value)), 1 >
Definition base_simd.h:102
+
mlx::core::simd::asin
Simd< float16_t, N > asin(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:31
+
mlx::core::simd::any
bool any(Simd< T, N > x)
Definition accelerate_simd.h:275
+
mlx::core::simd::fma
Simd< T, N > fma(Simd< T, N > x, Simd< T, N > y, U z)
Definition accelerate_simd.h:264
+
mlx::core::simd::tanh
Simd< float16_t, N > tanh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:43
+
mlx::core::simd::atan
Simd< float16_t, N > atan(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:33
+
mlx::core::simd::asinh
Simd< float16_t, N > asinh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:32
+
mlx::core::simd::remainder
Simd< float16_t, N > remainder(Simd< float16_t, N > x, Simd< float16_t, N > y)
Definition accelerate_fp16_simd.h:53
+
mlx::core::simd::max_size
static constexpr int max_size
Definition base_simd.h:13
+
mlx::core::simd::min
T min(Simd< T, N > x)
Definition accelerate_simd.h:287
+
mlx::core::simd::real
auto real(Simd< T, 1 > in) -> Simd< decltype(std::real(in.value)), 1 >
Definition base_simd.h:98
+
mlx::core::simd::log1p
Simd< float16_t, N > log1p(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:40
+
mlx::core::simd::ceil
Simd< T, N > ceil(Simd< T, N > v)
Definition accelerate_simd.h:120
+
mlx::core::simd::recip
Simd< T, N > recip(Simd< T, N > v)
Definition accelerate_simd.h:131
+
mlx::core::simd::sqrt
Simd< T, N > sqrt(Simd< T, N > v)
Definition accelerate_simd.h:129
+
mlx::core::simd::clamp
Simd< T, N > clamp(Simd< T, N > v, Simd< T, N > min, Simd< T, N > max)
Definition accelerate_simd.h:259
+
mlx::core::simd::acos
Simd< float16_t, N > acos(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:29
+
mlx::core::simd::rsqrt
Simd< T, N > rsqrt(Simd< T, N > v)
Definition accelerate_simd.h:130
+
mlx::core::simd::cosh
Simd< float16_t, N > cosh(Simd< float16_t, N > v)
Definition accelerate_fp16_simd.h:35
+
mlx::core::simd::store
void store(T *dst, Simd< T, N > x)
Definition base_simd.h:32
+
mlx::core::simd::select
Simd< T1, N > select(Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)
Definition accelerate_simd.h:231
+
mlx::core::simd::Simd< T, 1 >::size
static constexpr int size
Definition base_simd.h:17
+
mlx::core::simd::Simd< T, 1 >::Simd
Simd()
Definition base_simd.h:19
+
mlx::core::simd::Simd< T, 1 >::Simd
Simd(Simd< U, 1 > v)
Definition base_simd.h:21
+
mlx::core::simd::Simd< T, 1 >::value
T value
Definition base_simd.h:18
+
mlx::core::simd::Simd< T, 1 >::Simd
Simd(U v)
Definition base_simd.h:23
+
mlx::core::simd::Simd
Definition accelerate_simd.h:55
+
mlx::core::simd::Simd::value
asd::Vector< scalar_t, N >::packed_t value
Definition accelerate_simd.h:80
+
+
+ + + + diff --git a/docs/build/html/bf16__math_8h.html b/docs/build/html/bf16__math_8h.html index d15f9ea21..3fdfd050f 100644 --- a/docs/build/html/bf16__math_8h.html +++ b/docs/build/html/bf16__math_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/bf16_math.h File Reference @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
mlx::core::Fence Member List
+
+
+ +

This is the complete list of members for mlx::core::Fence, including all inherited members.

+ + + + + + +
Fence(const Stream &stream)mlx::core::Fence
update()mlx::core::Fence
update_gpu(const array &x)mlx::core::Fence
wait()mlx::core::Fence
wait_gpu(array &x)mlx::core::Fence
+
+ + + + diff --git a/docs/build/html/classmlx_1_1core_1_1_fence.html b/docs/build/html/classmlx_1_1core_1_1_fence.html new file mode 100644 index 000000000..a1408c747 --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_fence.html @@ -0,0 +1,227 @@ + + + + + + + +MLX: mlx::core::Fence Class Reference + + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
+Public Member Functions | +List of all members
+
mlx::core::Fence Class Reference
+
+
+ +

#include <fence.h>

+ + + + + + + + + + + + +

+Public Member Functions

 Fence (const Stream &stream)
 
void update_gpu (const array &x)
 
void wait_gpu (array &x)
 
void wait ()
 
void update ()
 
+

Constructor & Destructor Documentation

+ +

◆ Fence()

+ +
+
+ + + + + + + +
mlx::core::Fence::Fence (const Stream & stream)
+
+ +
+
+

Member Function Documentation

+ +

◆ update()

+ +
+
+ + + + + + + +
void mlx::core::Fence::update ()
+
+ +
+
+ +

◆ update_gpu()

+ +
+
+ + + + + + + +
void mlx::core::Fence::update_gpu (const array & x)
+
+ +
+
+ +

◆ wait()

+ +
+
+ + + + + + + +
void mlx::core::Fence::wait ()
+
+ +
+
+ +

◆ wait_gpu()

+ +
+
+ + + + + + + +
void mlx::core::Fence::wait_gpu (array & x)
+
+ +
+
+
The documentation for this class was generated from the following file: +
+
+ + + + diff --git a/docs/build/html/classmlx_1_1core_1_1_fence.js b/docs/build/html/classmlx_1_1core_1_1_fence.js new file mode 100644 index 000000000..d4342d49e --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_fence.js @@ -0,0 +1,8 @@ +var classmlx_1_1core_1_1_fence = +[ + [ "Fence", "classmlx_1_1core_1_1_fence.html#a3e3ed08eb6a1025b051b5a435f6f95f7", null ], + [ "update", "classmlx_1_1core_1_1_fence.html#a653279d4023d69751a930a91d3bf010a", null ], + [ "update_gpu", "classmlx_1_1core_1_1_fence.html#a6c5652aad6e93b06c72258bb8d9c19fc", null ], + [ "wait", "classmlx_1_1core_1_1_fence.html#a1ccbe354d043e6c4d76286c6635a38e2", null ], + [ "wait_gpu", "classmlx_1_1core_1_1_fence.html#ab6d783dee02656ebb8ffcbbfa6de5b53", null ] +]; \ No newline at end of file diff --git a/docs/build/html/classmlx_1_1core_1_1_flatten-members.html b/docs/build/html/classmlx_1_1core_1_1_flatten-members.html index 98993e7ec..8f5f8b317 100644 --- a/docs/build/html/classmlx_1_1core_1_1_flatten-members.html +++ b/docs/build/html/classmlx_1_1core_1_1_flatten-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
mlx::core::GatherAxis Member List
+
+
+ +

This is the complete list of members for mlx::core::GatherAxis, including all inherited members.

+ + + + + + + + + + + + + + + + + + + + + + + + + + + +
device()mlx::core::Primitiveinline
eval_cpu(const std::vector< array > &inputs, array &out) overridemlx::core::GatherAxisvirtual
mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual
eval_gpu(const std::vector< array > &inputs, array &out) overridemlx::core::GatherAxisvirtual
mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual
GatherAxis(Stream stream, int axis)mlx::core::GatherAxisinlineexplicit
is_equivalent(const Primitive &other) const overridemlx::core::GatherAxisvirtual
jvp(const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums) overridemlx::core::GatherAxisvirtual
operator=(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive
operator=(UnaryPrimitive &&other)=deletemlx::core::UnaryPrimitive
mlx::core::Primitive::operator=(const Primitive &other)=deletemlx::core::Primitive
mlx::core::Primitive::operator=(Primitive &&other)=deletemlx::core::Primitive
output_shapes(const std::vector< array > &inputs) overridemlx::core::GatherAxisvirtual
Primitive(Stream stream)mlx::core::Primitiveinlineexplicit
Primitive(const Primitive &other)=deletemlx::core::Primitive
Primitive(Primitive &&other)=deletemlx::core::Primitive
print(std::ostream &os) overridemlx::core::GatherAxisinlinevirtual
state() constmlx::core::GatherAxisinline
stream()mlx::core::Primitiveinline
UnaryPrimitive(Stream stream)mlx::core::UnaryPrimitiveinlineexplicit
UnaryPrimitive(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive
UnaryPrimitive(UnaryPrimitive &&other)=deletemlx::core::UnaryPrimitive
vjp(const std::vector< array > &primals, const std::vector< array > &cotangents, const std::vector< int > &argnums, const std::vector< array > &outputs) overridemlx::core::GatherAxisvirtual
vmap(const std::vector< array > &inputs, const std::vector< int > &axes) overridemlx::core::GatherAxisvirtual
~Primitive()=defaultmlx::core::Primitivevirtual
~UnaryPrimitive()=defaultmlx::core::UnaryPrimitivevirtual
+
+ + + + diff --git a/docs/build/html/classmlx_1_1core_1_1_gather_axis.html b/docs/build/html/classmlx_1_1core_1_1_gather_axis.html new file mode 100644 index 000000000..3cc0637ac --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_gather_axis.html @@ -0,0 +1,525 @@ + + + + + + + +MLX: mlx::core::GatherAxis Class Reference + + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
+Public Member Functions | +List of all members
+
mlx::core::GatherAxis Class Reference
+
+
+ +

#include <primitives.h>

+
+Inheritance diagram for mlx::core::GatherAxis:
+
+
+ + +mlx::core::UnaryPrimitive +mlx::core::Primitive + +
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

+Public Member Functions

 GatherAxis (Stream stream, int axis)
 
void eval_cpu (const std::vector< array > &inputs, array &out) override
 
void eval_gpu (const std::vector< array > &inputs, array &out) override
 
virtual std::pair< std::vector< array >, std::vector< int > > vmap (const std::vector< array > &inputs, const std::vector< int > &axes) override
 The primitive must know how to vectorize itself across the given axes.
 
std::vector< array > jvp (const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums) override
 The Jacobian-vector product.
 
std::vector< array > vjp (const std::vector< array > &primals, const std::vector< array > &cotangents, const std::vector< int > &argnums, const std::vector< array > &outputs) override
 The vector-Jacobian product.
 
void print (std::ostream &os) override
 Print the primitive.
 
bool is_equivalent (const Primitive &other) const override
 Equivalence check defaults to false unless overridden by the primitive.
 
std::vector< Shape > output_shapes (const std::vector< array > &inputs) override
 Get the output shapes of the primitive.
 
auto state () const
 
- Public Member Functions inherited from mlx::core::UnaryPrimitive
 UnaryPrimitive (Stream stream)
 An abstract base class for a primitive with a single output.
 
void eval_cpu (const std::vector< array > &inputs, std::vector< array > &outputs) override
 A primitive must know how to evaluate itself on the CPU/GPU for the given inputs and populate the output arrays.
 
void eval_gpu (const std::vector< array > &inputs, std::vector< array > &outputs) override
 
virtual ~UnaryPrimitive ()=default
 
 UnaryPrimitive (const UnaryPrimitive &other)=delete
 
 UnaryPrimitive (UnaryPrimitive &&other)=delete
 
UnaryPrimitive & operator= (const UnaryPrimitive &other)=delete
 
UnaryPrimitive & operator= (UnaryPrimitive &&other)=delete
 
- Public Member Functions inherited from mlx::core::Primitive
 Primitive (Stream stream)
 
const Device & device ()
 The device the primitive will run on.
 
const Stream & stream ()
 The stream the primitive will run on.
 
virtual ~Primitive ()=default
 
 Primitive (const Primitive &other)=delete
 
 Primitive (Primitive &&other)=delete
 
Primitive & operator= (const Primitive &other)=delete
 
Primitive & operator= (Primitive &&other)=delete
 
+

Constructor & Destructor Documentation

+ +

◆ GatherAxis()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
mlx::core::GatherAxis::GatherAxis (Stream stream,
int axis )
+ 		+inlineexplicit
+
+ +
+
+

Member Function Documentation

+ +

◆ eval_cpu()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
void mlx::core::GatherAxis::eval_cpu (const std::vector< array > & inputs,
array & out )
+ 		+overridevirtual
+
+ +

Implements mlx::core::UnaryPrimitive.

+ +
+
+ +

◆ eval_gpu()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
void mlx::core::GatherAxis::eval_gpu (const std::vector< array > & inputs,
array & out )
+ 		+overridevirtual
+
+ +

Implements mlx::core::UnaryPrimitive.

+ +
+
+ +

◆ is_equivalent()

+ +
+
+ + + + + +
+ + + + + + + +
bool mlx::core::GatherAxis::is_equivalent (const Primitive & other) const
+ 		+overridevirtual
+
+ +

Equivalence check defaults to false unless overridden by the primitive.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ jvp()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + +
std::vector< array > mlx::core::GatherAxis::jvp (const std::vector< array > & primals,
const std::vector< array > & tangents,
const std::vector< int > & argnums )
+ 		+overridevirtual
+
+ +

The Jacobian-vector product.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ output_shapes()

+ +
+
+ + + + + +
+ + + + + + + +
std::vector< Shape > mlx::core::GatherAxis::output_shapes (const std::vector< array > & inputs)
+ 		+overridevirtual
+
+ +

Get the output shapes of the primitive.

+

This is not required to be implemented by derived classes, in which case it will throw.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ print()

+ +
+
+ + + + + +
+ + + + + + + +
void mlx::core::GatherAxis::print (std::ostream & os)
+ 		+inlineoverridevirtual
+
+ +

Print the primitive.

+ +

Implements mlx::core::Primitive.

+ +
+
+ +

◆ state()

+ +
+
+ + + + + +
+ + + + + + + +
auto mlx::core::GatherAxis::state () const
+ 		+inline
+
+ +
+
+ +

◆ vjp()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + + + + +
std::vector< array > mlx::core::GatherAxis::vjp (const std::vector< array > & primals,
const std::vector< array > & cotangents,
const std::vector< int > & argnums,
const std::vector< array > & outputs )
+ 		+overridevirtual
+
+ +

The vector-Jacobian product.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ vmap()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
virtual std::pair< std::vector< array >, std::vector< int > > mlx::core::GatherAxis::vmap (const std::vector< array > & inputs,
const std::vector< int > & axes )
+ 		+overridevirtual
+
+ +

The primitive must know how to vectorize itself across the given axes.

+

The output is a pair containing the output arrays representing the vectorized computation and the axes which corresponds to the vectorized dimensions of each output.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+
The documentation for this class was generated from the following file: +
+
+ + + + diff --git a/docs/build/html/classmlx_1_1core_1_1_gather_axis.js b/docs/build/html/classmlx_1_1core_1_1_gather_axis.js new file mode 100644 index 000000000..af440a9e9 --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_gather_axis.js @@ -0,0 +1,13 @@ +var classmlx_1_1core_1_1_gather_axis = +[ + [ "GatherAxis", "classmlx_1_1core_1_1_gather_axis.html#ad8fc6400954c52079f0a2f2b711df060", null ], + [ "eval_cpu", "classmlx_1_1core_1_1_gather_axis.html#a474eae1d024e676e668318bf10928e2a", null ], + [ "eval_gpu", "classmlx_1_1core_1_1_gather_axis.html#a1344749d33e4ea2cb80b69a5a4a21afc", null ], + [ "is_equivalent", "classmlx_1_1core_1_1_gather_axis.html#a8f603c5c46d566654bd8a615d24c1089", null ], + [ "jvp", "classmlx_1_1core_1_1_gather_axis.html#a4f6015bf2c9bb8773118eb51be45b378", null ], + [ "output_shapes", "classmlx_1_1core_1_1_gather_axis.html#abc483c7da7747263b2f1498f98b4d96d", null ], + [ "print", "classmlx_1_1core_1_1_gather_axis.html#a9108bd9dfc153e6260e6340ff923ba38", null ], + [ "state", "classmlx_1_1core_1_1_gather_axis.html#adff37b05799654b1a589e334d1cd6b46", null ], + [ "vjp", "classmlx_1_1core_1_1_gather_axis.html#a9c73b4ebed01bbdbaa316eddb6b5606d", null ], + [ "vmap", "classmlx_1_1core_1_1_gather_axis.html#a48d50bad33b69e29f75bedc794f7b785", null ] +]; \ No newline at end of file diff --git a/docs/build/html/classmlx_1_1core_1_1_gather_axis.png b/docs/build/html/classmlx_1_1core_1_1_gather_axis.png new file mode 100644 index 000000000..afacc2806 Binary files /dev/null and b/docs/build/html/classmlx_1_1core_1_1_gather_axis.png differ diff --git a/docs/build/html/classmlx_1_1core_1_1_gather_m_m-members.html b/docs/build/html/classmlx_1_1core_1_1_gather_m_m-members.html index 8a8bd0b61..8925e0f0b 100644 --- a/docs/build/html/classmlx_1_1core_1_1_gather_m_m-members.html +++ b/docs/build/html/classmlx_1_1core_1_1_gather_m_m-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
mlx::core::ScatterAxis Member List
+
+
+ +

This is the complete list of members for mlx::core::ScatterAxis, including all inherited members.

+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
device()mlx::core::Primitiveinline
eval_cpu(const std::vector< array > &inputs, array &out) overridemlx::core::ScatterAxisvirtual
mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual
eval_gpu(const std::vector< array > &inputs, array &out) overridemlx::core::ScatterAxisvirtual
mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual
is_equivalent(const Primitive &other) const overridemlx::core::ScatterAxisvirtual
jvp(const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums) overridemlx::core::ScatterAxisvirtual
None enum valuemlx::core::ScatterAxis
operator=(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive
operator=(UnaryPrimitive &&other)=deletemlx::core::UnaryPrimitive
mlx::core::Primitive::operator=(const Primitive &other)=deletemlx::core::Primitive
mlx::core::Primitive::operator=(Primitive &&other)=deletemlx::core::Primitive
output_shapes(const std::vector< array > &inputs) overridemlx::core::ScatterAxisvirtual
Primitive(Stream stream)mlx::core::Primitiveinlineexplicit
Primitive(const Primitive &other)=deletemlx::core::Primitive
Primitive(Primitive &&other)=deletemlx::core::Primitive
print(std::ostream &os) overridemlx::core::ScatterAxisinlinevirtual
ReduceType enum namemlx::core::ScatterAxis
ScatterAxis(Stream stream, ReduceType reduce_type, int axis)mlx::core::ScatterAxisinlineexplicit
state() constmlx::core::ScatterAxisinline
stream()mlx::core::Primitiveinline
Sum enum valuemlx::core::ScatterAxis
UnaryPrimitive(Stream stream)mlx::core::UnaryPrimitiveinlineexplicit
UnaryPrimitive(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive
UnaryPrimitive(UnaryPrimitive &&other)=deletemlx::core::UnaryPrimitive
vjp(const std::vector< array > &primals, const std::vector< array > &cotangents, const std::vector< int > &argnums, const std::vector< array > &outputs) overridemlx::core::ScatterAxisvirtual
vmap(const std::vector< array > &inputs, const std::vector< int > &axes) overridemlx::core::ScatterAxisvirtual
~Primitive()=defaultmlx::core::Primitivevirtual
~UnaryPrimitive()=defaultmlx::core::UnaryPrimitivevirtual
+
+ + + + diff --git a/docs/build/html/classmlx_1_1core_1_1_scatter_axis.html b/docs/build/html/classmlx_1_1core_1_1_scatter_axis.html new file mode 100644 index 000000000..de0d7cc45 --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_scatter_axis.html @@ -0,0 +1,557 @@ + + + + + + + +MLX: mlx::core::ScatterAxis Class Reference + + + + + + + + + + + + + + + + +
+
+ + + + + + + +
+
MLX +
+
+ +   + + + + +
+
+
+ + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+
+
+
+
+
Loading...
+
Searching...
+
No Matches
+
+
+
+
+ +
+
+Public Types | +Public Member Functions | +List of all members
+
mlx::core::ScatterAxis Class Reference
+
+
+ +

#include <primitives.h>

+
+Inheritance diagram for mlx::core::ScatterAxis:
+
+
+ + +mlx::core::UnaryPrimitive +mlx::core::Primitive + +
+ + + + +

+Public Types

enum  ReduceType { Sum +, None + }
 
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

+Public Member Functions

 ScatterAxis (Stream stream, ReduceType reduce_type, int axis)
 
void eval_cpu (const std::vector< array > &inputs, array &out) override
 
void eval_gpu (const std::vector< array > &inputs, array &out) override
 
virtual std::pair< std::vector< array >, std::vector< int > > vmap (const std::vector< array > &inputs, const std::vector< int > &axes) override
 The primitive must know how to vectorize itself across the given axes.
 
std::vector< array > jvp (const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums) override
 The Jacobian-vector product.
 
std::vector< array > vjp (const std::vector< array > &primals, const std::vector< array > &cotangents, const std::vector< int > &argnums, const std::vector< array > &outputs) override
 The vector-Jacobian product.
 
void print (std::ostream &os) override
 Print the primitive.
 
bool is_equivalent (const Primitive &other) const override
 Equivalence check defaults to false unless overridden by the primitive.
 
std::vector< Shape > output_shapes (const std::vector< array > &inputs) override
 Get the output shapes of the primitive.
 
std::pair< ReduceType, int > state () const
 
- Public Member Functions inherited from mlx::core::UnaryPrimitive
 UnaryPrimitive (Stream stream)
 An abstract base class for a primitive with a single output.
 
void eval_cpu (const std::vector< array > &inputs, std::vector< array > &outputs) override
 A primitive must know how to evaluate itself on the CPU/GPU for the given inputs and populate the output arrays.
 
void eval_gpu (const std::vector< array > &inputs, std::vector< array > &outputs) override
 
virtual ~UnaryPrimitive ()=default
 
 UnaryPrimitive (const UnaryPrimitive &other)=delete
 
 UnaryPrimitive (UnaryPrimitive &&other)=delete
 
UnaryPrimitive & operator= (const UnaryPrimitive &other)=delete
 
UnaryPrimitive & operator= (UnaryPrimitive &&other)=delete
 
- Public Member Functions inherited from mlx::core::Primitive
 Primitive (Stream stream)
 
const Device & device ()
 The device the primitive will run on.
 
const Stream & stream ()
 The stream the primitive will run on.
 
virtual ~Primitive ()=default
 
 Primitive (const Primitive &other)=delete
 
 Primitive (Primitive &&other)=delete
 
Primitive & operator= (const Primitive &other)=delete
 
Primitive & operator= (Primitive &&other)=delete
 
+

Member Enumeration Documentation

+ +

◆ ReduceType

+ +
+
+ + + + +
enum mlx::core::ScatterAxis::ReduceType
+
+ + + +
Enumerator
Sum 
None 
+ +
+
+

Constructor & Destructor Documentation

+ +

◆ ScatterAxis()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + +
mlx::core::ScatterAxis::ScatterAxis (Stream stream,
ReduceType reduce_type,
int axis )
+ 		+inlineexplicit
+
+ +
+
+

Member Function Documentation

+ +

◆ eval_cpu()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
void mlx::core::ScatterAxis::eval_cpu (const std::vector< array > & inputs,
array & out )
+ 		+overridevirtual
+
+ +

Implements mlx::core::UnaryPrimitive.

+ +
+
+ +

◆ eval_gpu()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
void mlx::core::ScatterAxis::eval_gpu (const std::vector< array > & inputs,
array & out )
+ 		+overridevirtual
+
+ +

Implements mlx::core::UnaryPrimitive.

+ +
+
+ +

◆ is_equivalent()

+ +
+
+ + + + + +
+ + + + + + + +
bool mlx::core::ScatterAxis::is_equivalent (const Primitive & other) const
+ 		+overridevirtual
+
+ +

Equivalence check defaults to false unless overridden by the primitive.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ jvp()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + +
std::vector< array > mlx::core::ScatterAxis::jvp (const std::vector< array > & primals,
const std::vector< array > & tangents,
const std::vector< int > & argnums )
+ 		+overridevirtual
+
+ +

The Jacobian-vector product.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ output_shapes()

+ +
+
+ + + + + +
+ + + + + + + +
std::vector< Shape > mlx::core::ScatterAxis::output_shapes (const std::vector< array > & inputs)
+ 		+overridevirtual
+
+ +

Get the output shapes of the primitive.

+

This is not required to be implemented by derived classes, in which case it will throw.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ print()

+ +
+
+ + + + + +
+ + + + + + + +
void mlx::core::ScatterAxis::print (std::ostream & os)
+ 		+inlineoverridevirtual
+
+ +

Print the primitive.

+ +

Implements mlx::core::Primitive.

+ +
+
+ +

◆ state()

+ +
+
+ + + + + +
+ + + + + + + +
std::pair< ReduceType, int > mlx::core::ScatterAxis::state () const
+ 		+inline
+
+ +
+
+ +

◆ vjp()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + + + + +
std::vector< array > mlx::core::ScatterAxis::vjp (const std::vector< array > & primals,
const std::vector< array > & cotangents,
const std::vector< int > & argnums,
const std::vector< array > & outputs )
+ 		+overridevirtual
+
+ +

The vector-Jacobian product.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+ +

◆ vmap()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
virtual std::pair< std::vector< array >, std::vector< int > > mlx::core::ScatterAxis::vmap (const std::vector< array > & inputs,
const std::vector< int > & axes )
+ 		+overridevirtual
+
+ +

The primitive must know how to vectorize itself across the given axes.

+

The output is a pair containing the output arrays representing the vectorized computation and the axes which corresponds to the vectorized dimensions of each output.

+ +

Reimplemented from mlx::core::Primitive.

+ +
+
+
The documentation for this class was generated from the following file: +
+
+ + + + diff --git a/docs/build/html/classmlx_1_1core_1_1_scatter_axis.js b/docs/build/html/classmlx_1_1core_1_1_scatter_axis.js new file mode 100644 index 000000000..9ea77fa72 --- /dev/null +++ b/docs/build/html/classmlx_1_1core_1_1_scatter_axis.js @@ -0,0 +1,17 @@ +var classmlx_1_1core_1_1_scatter_axis = +[ + [ "ReduceType", "classmlx_1_1core_1_1_scatter_axis.html#aa292e6cb2a4b32c42ad4f7a258b334f2", [ + [ "Sum", "classmlx_1_1core_1_1_scatter_axis.html#aa292e6cb2a4b32c42ad4f7a258b334f2a702b8cfdaf7fe3e063873595ff0508f2", null ], + [ "None", "classmlx_1_1core_1_1_scatter_axis.html#aa292e6cb2a4b32c42ad4f7a258b334f2a93146c4280504f1f67459e6ae0d25c38", null ] + ] ], + [ "ScatterAxis", "classmlx_1_1core_1_1_scatter_axis.html#a7365a2c5fddb1c39509998598de411db", null ], + [ "eval_cpu", "classmlx_1_1core_1_1_scatter_axis.html#abf9d24565abdd7e1034daacac603cc54", null ], + [ "eval_gpu", "classmlx_1_1core_1_1_scatter_axis.html#a715c3b959dc904faefb16edbb11f29d7", null ], + [ "is_equivalent", "classmlx_1_1core_1_1_scatter_axis.html#af511c39926d5b85ca59558d64e0608fb", null ], + [ "jvp", "classmlx_1_1core_1_1_scatter_axis.html#a77129b601e5ca9d97669a8b0fdc69805", null ], + [ "output_shapes", "classmlx_1_1core_1_1_scatter_axis.html#af9688c010e1abee9b7b3788f11d91cc5", null ], + [ "print", "classmlx_1_1core_1_1_scatter_axis.html#a657843d4d9846ecd56e35e066986eb96", null ], + [ "state", "classmlx_1_1core_1_1_scatter_axis.html#a1a0125be908a1d80875236c817f34495", null ], + [ "vjp", "classmlx_1_1core_1_1_scatter_axis.html#a450f97b0be61a2bdfbfef4b2eb7cd198", null ], + [ "vmap", "classmlx_1_1core_1_1_scatter_axis.html#ae78709d1be122618f210ff595d888df8", null ] +]; \ No newline at end of file diff --git a/docs/build/html/classmlx_1_1core_1_1_scatter_axis.png b/docs/build/html/classmlx_1_1core_1_1_scatter_axis.png new file mode 100644 index 000000000..55c630a0f Binary files /dev/null and b/docs/build/html/classmlx_1_1core_1_1_scatter_axis.png differ diff --git a/docs/build/html/classmlx_1_1core_1_1_select-members.html b/docs/build/html/classmlx_1_1core_1_1_select-members.html index d4ebe34a4..c2062a9ce 100644 --- a/docs/build/html/classmlx_1_1core_1_1_select-members.html +++ b/docs/build/html/classmlx_1_1core_1_1_select-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + - - - - - - - - - - - - - - -
-
- - - - - - - -
-
MLX -
-
- -   - - - - -
-
-
- - - - -
-
- -
-
-
- -
- -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- -
-
unary.h
-
-
-Go to the documentation of this file.
1// Copyright © 2023 Apple Inc.
-
2
-
3#pragma once
-
4
-
5#include "mlx/allocator.h"
-
6#include "mlx/array.h"
-
7#include "mlx/backend/common/utils.h"
-
8#include "mlx/utils.h"
-
9
-
10namespace mlx::core {
-
11
-
12namespace {
-
13
-
14void set_unary_output_data(const array& in, array& out) {
-
15 if (is_donatable(in, out)) {
-
16 out.copy_shared_buffer(in);
-
17 } else {
-
18 auto size = in.data_size();
-
19 out.set_data(
-
20 allocator::malloc_or_wait(size * out.itemsize()),
-
21 size,
-
22 in.strides(),
-
23 in.flags());
-
24 }
-
25}
-
26
-
27template <typename T, typename U = T, typename Op>
-
28void unary_op(const T* a, U* out, Op op, size_t shape, size_t stride) {
-
29 for (size_t i = 0; i < shape; i += 1) {
-
30 out[i] = op(*a);
-
31 a += stride;
-
32 }
-
33}
-
34
-
35template <typename T, typename U = T, typename Op>
-
36void unary_op(const array& a, array& out, Op op) {
-
37 const T* a_ptr = a.data<T>();
-
38 if (a.flags().contiguous) {
-
39 set_unary_output_data(a, out);
-
40 U* dst = out.data<U>();
-
41 for (size_t i = 0; i < a.data_size(); ++i) {
-
42 dst[i] = op(a_ptr[i]);
-
43 }
-
44 } else {
-
45 out.set_data(allocator::malloc_or_wait(out.nbytes()));
-
46 U* dst = out.data<U>();
-
47 size_t shape = a.ndim() > 0 ? a.shape(-1) : 1;
-
48 size_t stride = a.ndim() > 0 ? a.strides(-1) : 1;
-
49 if (a.ndim() <= 1) {
-
50 unary_op(a_ptr, dst, op, shape, stride);
-
51 return;
-
52 }
-
53 ContiguousIterator it(a.shape(), a.strides(), a.ndim() - 1);
-
54 for (size_t elem = 0; elem < a.size(); elem += shape) {
-
55 unary_op(a_ptr + it.loc, dst + elem, op, shape, stride);
-
56 it.step();
-
57 }
-
58 }
-
59}
-
60
-
61template <typename Op>
-
62void unary(const array& a, array& out, Op op) {
-
63 switch (out.dtype()) {
-
64 case bool_:
-
65 unary_op<bool>(a, out, op);
-
66 break;
-
67 case uint8:
-
68 unary_op<uint8_t>(a, out, op);
-
69 break;
-
70 case uint16:
-
71 unary_op<uint16_t>(a, out, op);
-
72 break;
-
73 case uint32:
-
74 unary_op<uint32_t>(a, out, op);
-
75 break;
-
76 case uint64:
-
77 unary_op<uint64_t>(a, out, op);
-
78 break;
-
79 case int8:
-
80 unary_op<int8_t>(a, out, op);
-
81 break;
-
82 case int16:
-
83 unary_op<int16_t>(a, out, op);
-
84 break;
-
85 case int32:
-
86 unary_op<int32_t>(a, out, op);
-
87 break;
-
88 case int64:
-
89 unary_op<int64_t>(a, out, op);
-
90 break;
-
91 case float16:
-
92 unary_op<float16_t>(a, out, op);
-
93 break;
-
94 case float32:
-
95 unary_op<float>(a, out, op);
-
96 break;
-
97 case bfloat16:
-
98 unary_op<bfloat16_t>(a, out, op);
-
99 break;
-
100 case complex64:
-
101 unary_op<complex64_t>(a, out, op);
-
102 break;
-
103 }
-
104}
-
105
-
106template <typename Op>
-
107void unary_fp(const array& a, array& out, Op op) {
-
108 switch (out.dtype()) {
-
109 case bfloat16:
-
110 unary_op<bfloat16_t>(a, out, op);
-
111 break;
-
112 case float16:
-
113 unary_op<float16_t>(a, out, op);
-
114 break;
-
115 case float32:
-
116 unary_op<float>(a, out, op);
-
117 break;
-
118 case complex64:
-
119 unary_op<complex64_t>(a, out, op);
-
120 break;
-
121 default:
-
122 std::ostringstream err;
-
123 err << "[unary_fp] Does not support " << out.dtype();
-
124 throw std::runtime_error(err.str());
-
125 }
-
126}
-
127
-
128} // namespace
-
129
-
130} // namespace mlx::core
- - - -
mlx::core::array
Definition array.h:24
-
mlx::core::allocator::malloc_or_wait
Buffer malloc_or_wait(size_t size)
-
mlx::core::metal::unary
const char * unary()
-
mlx::core
Definition allocator.h:7
-
mlx::core::bool_
constexpr Dtype bool_
Definition dtype.h:67
-
mlx::core::uint64
constexpr Dtype uint64
Definition dtype.h:72
-
mlx::core::uint16
constexpr Dtype uint16
Definition dtype.h:70
-
mlx::core::bfloat16
constexpr Dtype bfloat16
Definition dtype.h:81
-
mlx::core::int32
constexpr Dtype int32
Definition dtype.h:76
-
mlx::core::float32
constexpr Dtype float32
Definition dtype.h:80
-
mlx::core::int16
constexpr Dtype int16
Definition dtype.h:75
-
mlx::core::int8
constexpr Dtype int8
Definition dtype.h:74
-
mlx::core::int64
constexpr Dtype int64
Definition dtype.h:77
-
mlx::core::uint8
constexpr Dtype uint8
Definition dtype.h:69
-
mlx::core::float16
constexpr Dtype float16
Definition dtype.h:79
-
mlx::core::uint32
constexpr Dtype uint32
Definition dtype.h:71
-
mlx::core::is_donatable
bool is_donatable(const array &in, const array &out)
Definition utils.h:155
-
mlx::core::complex64
constexpr Dtype complex64
Definition dtype.h:82
-
pocketfft::detail::dst
void dst(const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, int type, const T *data_in, T *data_out, T fct, bool ortho, size_t nthreads=1)
Definition pocketfft.h:3416
-
mlx::core::ContiguousIterator
Definition utils.h:73
- -
-
- - - - diff --git a/docs/build/html/compile_8h.html b/docs/build/html/compile_8h.html index 5f0f9527f..11fcad64b 100644 --- a/docs/build/html/compile_8h.html +++ b/docs/build/html/compile_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/compile.h File Reference @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - + + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -924,7 +924,7 @@
  • split()
  • split()
  • split()
    • -
  • meshgrid()
    • +
  • meshgrid()
  • clip()
  • concatenate()
  • concatenate()
    • @@ -937,10 +937,10 @@
  • transpose()
  • swapaxes()
  • moveaxis()
    • -
  • pad()
    • -
  • pad()
    • -
  • pad()
    • -
  • pad()
    • +
  • pad()
    • +
  • pad()
    • +
  • pad()
    • +
  • pad()
  • transpose()
  • broadcast_to()
  • broadcast_arrays()
    • @@ -1110,6 +1110,7 @@
  • take()
  • take_along_axis()
  • put_along_axis()
    • +
  • scatter_add_axis()
  • scatter()
  • scatter()
  • scatter_add()
    • @@ -1499,8 +1500,8 @@
    -
    -std::vector<array> meshgrid(const std::vector<array> &arrays, bool sparse = false, std::string indexing = "xy", StreamOrDevice s = {})#
    +
    +std::vector<array> meshgrid(const std::vector<array> &arrays, bool sparse = false, const std::string &indexing = "xy", StreamOrDevice s = {})#

    A vector of coordinate arrays from coordinate vectors.

    @@ -1572,25 +1573,25 @@
    -
    -array pad(const array &a, const std::vector<int> &axes, const Shape &low_pad_size, const Shape &high_pad_size, const array &pad_value = array(0), const std::string mode = "constant", StreamOrDevice s = {})#
    +
    +array pad(const array &a, const std::vector<int> &axes, const Shape &low_pad_size, const Shape &high_pad_size, const array &pad_value = array(0), const std::string &mode = "constant", StreamOrDevice s = {})#

    Pad an array with a constant value.

    -
    -array pad(const array &a, const std::vector<std::pair<int, int>> &pad_width, const array &pad_value = array(0), const std::string mode = "constant", StreamOrDevice s = {})#
    +
    +array pad(const array &a, const std::vector<std::pair<int, int>> &pad_width, const array &pad_value = array(0), const std::string &mode = "constant", StreamOrDevice s = {})#

    Pad an array with a constant value along all axes.

    -
    -array pad(const array &a, const std::pair<int, int> &pad_width, const array &pad_value = array(0), const std::string mode = "constant", StreamOrDevice s = {})#
    +
    +array pad(const array &a, const std::pair<int, int> &pad_width, const array &pad_value = array(0), const std::string &mode = "constant", StreamOrDevice s = {})#
    -
    -array pad(const array &a, int pad_width, const array &pad_value = array(0), const std::string mode = "constant", StreamOrDevice s = {})#
    +
    +array pad(const array &a, int pad_width, const array &pad_value = array(0), const std::string &mode = "constant", StreamOrDevice s = {})#
    @@ -2554,6 +2555,12 @@

    Put the values into the array at the given indices along the axis.

    +
    +
    +array scatter_add_axis(const array &a, const array &indices, const array &values, int axis, StreamOrDevice s = {})#
    +

    Add the values into the array at the given indices along the axis.

    +
    +
    array scatter(const array &a, const std::vector<array> &indices, const array &updates, const std::vector<int> &axes, StreamOrDevice s = {})#
    @@ -3129,7 +3136,7 @@
  • split()
  • split()
  • split()
    • -
  • meshgrid()
    • +
  • meshgrid()
  • clip()
  • concatenate()
  • concatenate()
    • @@ -3142,10 +3149,10 @@
  • transpose()
  • swapaxes()
  • moveaxis()
    • -
  • pad()
    • -
  • pad()
    • -
  • pad()
    • -
  • pad()
    • +
  • pad()
    • +
  • pad()
    • +
  • pad()
    • +
  • pad()
  • transpose()
  • broadcast_to()
  • broadcast_arrays()
    • @@ -3315,6 +3322,7 @@
  • take()
  • take_along_axis()
  • put_along_axis()
    • +
  • scatter_add_axis()
  • scatter()
  • scatter()
  • scatter_add()
    • diff --git a/docs/build/html/common_2arange_8h.html b/docs/build/html/cpu_2arange_8h.html similarity index 92% rename from docs/build/html/common_2arange_8h.html rename to docs/build/html/cpu_2arange_8h.html index 7da1bb15c..1da9dd3a2 100644 --- a/docs/build/html/common_2arange_8h.html +++ b/docs/build/html/cpu_2arange_8h.html @@ -3,9 +3,9 @@ - + -MLX: mlx/backend/common/arange.h File Reference +MLX: mlx/backend/cpu/arange.h File Reference @@ -52,7 +52,7 @@ - +
    @@ -111,7 +111,7 @@ $(function(){initNavTree('common_2arange_8h.html',''); initResizable(true); });
    #include "mlx/allocator.h"
    #include "mlx/array.h"
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    @@ -130,8 +130,8 @@ Functions diff --git a/docs/build/html/common_2arange_8h.js b/docs/build/html/cpu_2arange_8h.js similarity index 80% rename from docs/build/html/common_2arange_8h.js rename to docs/build/html/cpu_2arange_8h.js index a0d6612ed..b5d9ef600 100644 --- a/docs/build/html/common_2arange_8h.js +++ b/docs/build/html/cpu_2arange_8h.js @@ -1,4 +1,4 @@ -var common_2arange_8h = +var cpu_2arange_8h = [ [ "mlx::core::arange", "namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b", null ] ]; \ No newline at end of file diff --git a/docs/build/html/common_2arange_8h_source.html b/docs/build/html/cpu_2arange_8h_source.html similarity index 97% rename from docs/build/html/common_2arange_8h_source.html rename to docs/build/html/cpu_2arange_8h_source.html index 586098c23..87f5475fc 100644 --- a/docs/build/html/common_2arange_8h_source.html +++ b/docs/build/html/cpu_2arange_8h_source.html @@ -3,9 +3,9 @@ - + -MLX: mlx/backend/common/arange.h Source File +MLX: mlx/backend/cpu/arange.h Source File @@ -52,7 +52,7 @@

    Namespaces

    - +
    @@ -105,7 +105,7 @@ $(function(){initNavTree('common_2arange_8h_source.html',''); initResizable(true
    arange.h
    -Go to the documentation of this file.
    1// Copyright © 2023 Apple Inc.
    +Go to the documentation of this file.
    1// Copyright © 2023 Apple Inc.
    2
    3#pragma once
    4
    @@ -209,8 +209,8 @@ $(function(){initNavTree('common_2arange_8h_source.html',''); initResizable(true diff --git a/docs/build/html/cpu_2binary_8h.html b/docs/build/html/cpu_2binary_8h.html new file mode 100644 index 000000000..d10115299 --- /dev/null +++ b/docs/build/html/cpu_2binary_8h.html @@ -0,0 +1,165 @@ + + + + + + + +MLX: mlx/backend/cpu/binary.h File Reference + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Classes | +Namespaces | +Functions
    +
    binary.h File Reference
    +
    +
    +
    #include <cassert>
    +#include "mlx/allocator.h"
    +#include "mlx/array.h"
    +#include "mlx/backend/common/binary.h"
    +#include "mlx/backend/common/utils.h"
    +#include "mlx/backend/cpu/simd/simd.h"
    +
    +

    Go to the source code of this file.

    + + + + + + + + +

    +Classes

    struct  mlx::core::VectorScalar< Op >
     
    struct  mlx::core::ScalarVector< Op >
     
    struct  mlx::core::VectorVector< Op >
     
    + + + + + +

    +Namespaces

    namespace  mlx
     
    namespace  mlx::core
     
    + + + + + + + + + + + + + + + + +

    +Functions

    template<typename T, typename U, typename Op, int D, bool Strided>
    void mlx::core::binary_op_dims (const T *a, const T *b, U *out, Op op, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &out_strides, int axis)
     
    template<typename T, typename U, bool Strided, typename Op>
    void mlx::core::binary_op_dispatch_dims (const array &a, const array &b, array &out, Op op, int dim, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &out_strides)
     
    template<typename T, typename U, typename Op>
    void mlx::core::binary_op (const array &a, const array &b, array &out, Op op)
     
    template<typename T, typename Op>
    void mlx::core::binary_op (const array &a, const array &b, array &out, Op op)
     
    template<typename Op>
    void mlx::core::binary (const array &a, const array &b, array &out, Op op)
     
    +
    +
    + + + + diff --git a/docs/build/html/cpu_2binary_8h.js b/docs/build/html/cpu_2binary_8h.js new file mode 100644 index 000000000..3c8896aa8 --- /dev/null +++ b/docs/build/html/cpu_2binary_8h.js @@ -0,0 +1,11 @@ +var cpu_2binary_8h = +[ + [ "mlx::core::VectorScalar< Op >", "structmlx_1_1core_1_1_vector_scalar.html", "structmlx_1_1core_1_1_vector_scalar" ], + [ "mlx::core::ScalarVector< Op >", "structmlx_1_1core_1_1_scalar_vector.html", "structmlx_1_1core_1_1_scalar_vector" ], + [ "mlx::core::VectorVector< Op >", "structmlx_1_1core_1_1_vector_vector.html", "structmlx_1_1core_1_1_vector_vector" ], + [ "mlx::core::binary", "namespacemlx_1_1core.html#ae374861abd45cf019c3e6be2026f3798", null ], + [ "mlx::core::binary_op", "namespacemlx_1_1core.html#a9c1c1fdf9a0840a16a4d10a8f74f761d", null ], + [ "mlx::core::binary_op", "namespacemlx_1_1core.html#a2aca3458c56605a74d07ec39876549bc", null ], + [ "mlx::core::binary_op_dims", "namespacemlx_1_1core.html#a7ca09ebf776fe32db580f9038587ec31", null ], + [ "mlx::core::binary_op_dispatch_dims", "namespacemlx_1_1core.html#a66c9ee5018168b9101de52e0122d9755", null ] +]; \ No newline at end of file diff --git a/docs/build/html/cpu_2binary_8h_source.html b/docs/build/html/cpu_2binary_8h_source.html new file mode 100644 index 000000000..f348da1cd --- /dev/null +++ b/docs/build/html/cpu_2binary_8h_source.html @@ -0,0 +1,568 @@ + + + + + + + +MLX: mlx/backend/cpu/binary.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    binary.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2023 Apple Inc.
    +
    2
    +
    3#pragma once
    +
    4#include <cassert>
    +
    5
    +
    6#include "mlx/allocator.h"
    +
    7#include "mlx/array.h"
    +
    8#include "mlx/backend/common/binary.h"
    +
    9#include "mlx/backend/common/utils.h"
    +
    10
    +
    11#include "mlx/backend/cpu/simd/simd.h"
    +
    12
    +
    13namespace mlx::core {
    +
    14
    +
    15template <typename Op>
    +
    +
    16struct VectorScalar {
    +
    17 Op op;
    +
    18
    +
    19 VectorScalar(Op op_) : op(op_) {}
    +
    20
    +
    21 template <typename T, typename U>
    +
    +
    22 void operator()(const T* a, const T* b, U* dst, int size) {
    +
    23 T scalar = *b;
    +
    24 constexpr int N = simd::max_size<T>;
    +
    25 while (size >= N) {
    +
    26 simd::store(dst, op(simd::load<T, N>(a), simd::Simd<T, N>(scalar)));
    +
    27 dst += N;
    +
    28 a += N;
    +
    29 size -= N;
    +
    30 }
    +
    31 while (size-- > 0) {
    +
    32 *dst = op(*a, scalar);
    +
    33 dst++;
    +
    34 a++;
    +
    35 }
    +
    36 }
    +
    +
    37};
    +
    +
    38
    +
    39template <typename Op>
    +
    +
    40struct ScalarVector {
    +
    41 Op op;
    +
    42
    +
    43 ScalarVector(Op op_) : op(op_) {}
    +
    44
    +
    45 template <typename T, typename U>
    +
    +
    46 void operator()(const T* a, const T* b, U* dst, int size) {
    +
    47 T scalar = *a;
    +
    48 constexpr int N = simd::max_size<T>;
    +
    49 while (size >= N) {
    +
    50 simd::store(dst, op(simd::Simd<T, N>(scalar), simd::load<T, N>(b)));
    +
    51 dst += N;
    +
    52 b += N;
    +
    53 size -= N;
    +
    54 }
    +
    55 while (size-- > 0) {
    +
    56 *dst = op(scalar, *b);
    +
    57 dst++;
    +
    58 b++;
    +
    59 }
    +
    60 }
    +
    +
    61};
    +
    +
    62
    +
    63template <typename Op>
    +
    +
    64struct VectorVector {
    +
    65 Op op;
    +
    66
    +
    67 VectorVector(Op op_) : op(op_) {}
    +
    68
    +
    69 template <typename T, typename U>
    +
    +
    70 void operator()(const T* a, const T* b, U* dst, int size) {
    +
    71 constexpr int N = simd::max_size<T>;
    +
    72 while (size >= N) {
    +
    73 simd::store(dst, op(simd::load<T, N>(a), simd::load<T, N>(b)));
    +
    74 dst += N;
    +
    75 a += N;
    +
    76 b += N;
    +
    77 size -= N;
    +
    78 }
    +
    79 while (size-- > 0) {
    +
    80 *dst = op(*a, *b);
    +
    81 dst++;
    +
    82 a++;
    +
    83 b++;
    +
    84 }
    +
    85 }
    +
    +
    86};
    +
    +
    87
    +
    88template <typename T, typename U, typename Op, int D, bool Strided>
    +
    +
    89void binary_op_dims(
    +
    90 const T* a,
    +
    91 const T* b,
    +
    92 U* out,
    +
    93 Op op,
    +
    94 const Shape& shape,
    +
    95 const Strides& a_strides,
    +
    96 const Strides& b_strides,
    +
    97 const Strides& out_strides,
    +
    98 int axis) {
    +
    99 auto stride_a = a_strides[axis];
    +
    100 auto stride_b = b_strides[axis];
    +
    101 auto stride_out = out_strides[axis];
    +
    102 auto N = shape[axis];
    +
    103
    +
    104 for (int i = 0; i < N; i++) {
    +
    105 if constexpr (D > 1) {
    +
    106 binary_op_dims<T, U, Op, D - 1, Strided>(
    +
    107 a, b, out, op, shape, a_strides, b_strides, out_strides, axis + 1);
    +
    108 } else {
    +
    109 if constexpr (Strided) {
    +
    110 op(a, b, out, stride_out);
    +
    111 } else {
    +
    112 *out = op(*a, *b);
    +
    113 }
    +
    114 }
    +
    115 out += stride_out;
    +
    116 a += stride_a;
    +
    117 b += stride_b;
    +
    118 }
    +
    119}
    +
    +
    120
    +
    121template <typename T, typename U, bool Strided, typename Op>
    +
    +
    122void binary_op_dispatch_dims(
    +
    123 const array& a,
    +
    124 const array& b,
    +
    125 array& out,
    +
    126 Op op,
    +
    127 int dim,
    +
    128 const Shape& shape,
    +
    129 const Strides& a_strides,
    +
    130 const Strides& b_strides,
    +
    131 const Strides& out_strides) {
    +
    132 const T* a_ptr = a.data<T>();
    +
    133 const T* b_ptr = b.data<T>();
    +
    134 U* out_ptr = out.data<U>();
    +
    135 switch (dim) {
    +
    136 case 1:
    +
    137 binary_op_dims<T, U, Op, 1, Strided>(
    +
    138 a_ptr,
    +
    139 b_ptr,
    +
    140 out_ptr,
    +
    141 op,
    +
    142 shape,
    +
    143 a_strides,
    +
    144 b_strides,
    +
    145 out_strides,
    +
    146 0);
    +
    147 return;
    +
    148 case 2:
    +
    149 binary_op_dims<T, U, Op, 2, Strided>(
    +
    150 a_ptr,
    +
    151 b_ptr,
    +
    152 out_ptr,
    +
    153 op,
    +
    154 shape,
    +
    155 a_strides,
    +
    156 b_strides,
    +
    157 out_strides,
    +
    158 0);
    +
    159 return;
    +
    160 case 3:
    +
    161 binary_op_dims<T, U, Op, 3, Strided>(
    +
    162 a_ptr,
    +
    163 b_ptr,
    +
    164 out_ptr,
    +
    165 op,
    +
    166 shape,
    +
    167 a_strides,
    +
    168 b_strides,
    +
    169 out_strides,
    +
    170 0);
    +
    171 return;
    +
    172 }
    +
    173
    +
    174 ContiguousIterator a_it(shape, a_strides, dim - 3);
    +
    175 ContiguousIterator b_it(shape, b_strides, dim - 3);
    +
    176 auto stride = out_strides[dim - 4];
    +
    177 for (int64_t elem = 0; elem < a.size(); elem += stride) {
    +
    178 binary_op_dims<T, U, Op, 3, Strided>(
    +
    179 a_ptr + a_it.loc,
    +
    180 b_ptr + b_it.loc,
    +
    181 out_ptr + elem,
    +
    182 op,
    +
    183 shape,
    +
    184 a_strides,
    +
    185 b_strides,
    +
    186 out_strides,
    +
    187 dim - 3);
    +
    188 a_it.step();
    +
    189 b_it.step();
    +
    190 }
    +
    191}
    +
    +
    192
    +
    193template <typename T, typename U, typename Op>
    +
    +
    194void binary_op(const array& a, const array& b, array& out, Op op) {
    +
    195 auto bopt = get_binary_op_type(a, b);
    +
    196 set_binary_op_output_data(a, b, out, bopt);
    +
    197
    +
    198 // The full computation is scalar scalar so call the base op once
    +
    199 if (bopt == BinaryOpType::ScalarScalar) {
    +
    200 *(out.data<U>()) = op(*a.data<T>(), *b.data<T>());
    +
    201 return;
    +
    202 }
    +
    203
    +
    204 // The full computation is scalar vector so delegate to the op
    +
    205 if (bopt == BinaryOpType::ScalarVector) {
    +
    206 ScalarVector{op}(a.data<T>(), b.data<T>(), out.data<U>(), b.data_size());
    +
    207 return;
    +
    208 }
    +
    209
    +
    210 // The full computation is vector scalar so delegate to the op
    +
    211 if (bopt == BinaryOpType::VectorScalar) {
    +
    212 VectorScalar{op}(a.data<T>(), b.data<T>(), out.data<U>(), a.data_size());
    +
    213 return;
    +
    214 }
    +
    215
    +
    216 // The full computation is vector vector so delegate to the op
    +
    217 if (bopt == BinaryOpType::VectorVector) {
    +
    218 VectorVector{op}(a.data<T>(), b.data<T>(), out.data<U>(), out.size());
    +
    219 return;
    +
    220 }
    +
    221
    +
    222 // General computation so let's try to optimize
    +
    223 auto [new_shape, new_strides] = collapse_contiguous_dims(
    +
    224 a.shape(), {a.strides(), b.strides(), out.strides()});
    +
    225 const auto& a_strides = new_strides[0];
    +
    226 const auto& b_strides = new_strides[1];
    +
    227 const auto& strides = new_strides[2];
    +
    228
    +
    229 // Get the left-most dim such that the array is row contiguous after
    +
    230 auto leftmost_rc_dim = [&strides](const auto& arr_strides) {
    +
    231 int d = arr_strides.size() - 1;
    +
    232 for (; d >= 0 && arr_strides[d] == strides[d]; d--) {
    +
    233 }
    +
    234 return d + 1;
    +
    235 };
    +
    236 auto a_rc_dim = leftmost_rc_dim(a_strides);
    +
    237 auto b_rc_dim = leftmost_rc_dim(b_strides);
    +
    238
    +
    239 // Get the left-most dim such that the array is a broadcasted "scalar" after
    +
    240 auto leftmost_s_dim = [](const auto& arr_strides) {
    +
    241 int d = arr_strides.size() - 1;
    +
    242 for (; d >= 0 && arr_strides[d] == 0; d--) {
    +
    243 }
    +
    244 return d + 1;
    +
    245 };
    +
    246 auto a_s_dim = leftmost_s_dim(a_strides);
    +
    247 auto b_s_dim = leftmost_s_dim(b_strides);
    +
    248
    +
    249 auto ndim = new_shape.size();
    +
    250
    +
    251 // Case 1: LxM and FxM where L and F are broadcastable and M is row contiguous
    +
    252 int dim = ndim;
    +
    253 if (int d = std::max(a_rc_dim, b_rc_dim); d < ndim) {
    +
    254 bopt = BinaryOpType::VectorVector;
    +
    255 dim = d;
    +
    256 // Case 2: LxM and Fx1 where L and F are broadcastable and M is row
    +
    257 // contiguous
    +
    258 } else if (int d = std::max(a_rc_dim, b_s_dim); d < ndim) {
    +
    259 bopt = BinaryOpType::VectorScalar;
    +
    260 dim = d;
    +
    261 // Case 3: Lx1 and FxM where L and F are broadcastable and M is row
    +
    262 // contiguous
    +
    263 } else if (int d = std::max(a_s_dim, b_rc_dim); d < ndim) {
    +
    264 bopt = BinaryOpType::ScalarVector;
    +
    265 dim = d;
    +
    266 }
    +
    267
    +
    268 // Can be sure dim > 0 since otherwise we would have used one of the fully
    +
    269 // contiguous methods above. Except for the case that the flags do not
    +
    270 // correspond to the underlying contiguity.
    +
    271 if (dim == 0 || strides[dim - 1] < 16) {
    +
    272 bopt = BinaryOpType::General;
    +
    273 dim = ndim;
    +
    274 }
    +
    275
    +
    276 switch (bopt) {
    +
    277 case BinaryOpType::VectorVector:
    +
    278 binary_op_dispatch_dims<T, U, true>(
    +
    279 a,
    +
    280 b,
    +
    281 out,
    +
    282 VectorVector{op},
    +
    283 dim,
    +
    284 new_shape,
    +
    285 a_strides,
    +
    286 b_strides,
    +
    287 strides);
    +
    288 break;
    +
    289 case BinaryOpType::VectorScalar:
    +
    290 binary_op_dispatch_dims<T, U, true>(
    +
    291 a,
    +
    292 b,
    +
    293 out,
    +
    294 VectorScalar{op},
    +
    295 dim,
    +
    296 new_shape,
    +
    297 a_strides,
    +
    298 b_strides,
    +
    299 strides);
    +
    300 break;
    +
    301 case BinaryOpType::ScalarVector:
    +
    302 binary_op_dispatch_dims<T, U, true>(
    +
    303 a,
    +
    304 b,
    +
    305 out,
    +
    306 ScalarVector{op},
    +
    307 dim,
    +
    308 new_shape,
    +
    309 a_strides,
    +
    310 b_strides,
    +
    311 strides);
    +
    312 break;
    +
    313 default:
    +
    314 binary_op_dispatch_dims<T, U, false>(
    +
    315 a, b, out, op, dim, new_shape, a_strides, b_strides, strides);
    +
    316 break;
    +
    317 }
    +
    318}
    +
    +
    319
    +
    320template <typename T, typename Op>
    +
    +
    321void binary_op(const array& a, const array& b, array& out, Op op) {
    +
    322 binary_op<T, T>(a, b, out, op);
    +
    323}
    +
    +
    324
    +
    325template <typename Op>
    +
    +
    326void binary(const array& a, const array& b, array& out, Op op) {
    +
    327 switch (out.dtype()) {
    +
    328 case bool_:
    +
    329 binary_op<bool>(a, b, out, op);
    +
    330 break;
    +
    331 case uint8:
    +
    332 binary_op<uint8_t>(a, b, out, op);
    +
    333 break;
    +
    334 case uint16:
    +
    335 binary_op<uint16_t>(a, b, out, op);
    +
    336 break;
    +
    337 case uint32:
    +
    338 binary_op<uint32_t>(a, b, out, op);
    +
    339 break;
    +
    340 case uint64:
    +
    341 binary_op<uint64_t>(a, b, out, op);
    +
    342 break;
    +
    343 case int8:
    +
    344 binary_op<int8_t>(a, b, out, op);
    +
    345 break;
    +
    346 case int16:
    +
    347 binary_op<int16_t>(a, b, out, op);
    +
    348 break;
    +
    349 case int32:
    +
    350 binary_op<int32_t>(a, b, out, op);
    +
    351 break;
    +
    352 case int64:
    +
    353 binary_op<int64_t>(a, b, out, op);
    +
    354 break;
    +
    355 case float16:
    +
    356 binary_op<float16_t>(a, b, out, op);
    +
    357 break;
    +
    358 case float32:
    +
    359 binary_op<float>(a, b, out, op);
    +
    360 break;
    +
    361 case bfloat16:
    +
    362 binary_op<bfloat16_t>(a, b, out, op);
    +
    363 break;
    +
    364 case complex64:
    +
    365 binary_op<complex64_t>(a, b, out, op);
    +
    366 break;
    +
    367 }
    +
    368}
    +
    +
    369
    +
    370} // namespace mlx::core
    + + + +
    mlx::core::array
    Definition array.h:24
    +
    mlx::core::array::shape
    const Shape & shape() const
    The shape of the array as a vector of integers.
    Definition array.h:103
    +
    mlx::core::array::size
    size_t size() const
    The number of elements in the array.
    Definition array.h:88
    +
    mlx::core::array::data
    T * data()
    Definition array.h:342
    +
    mlx::core::array::dtype
    Dtype dtype() const
    Get the arrays data type.
    Definition array.h:131
    +
    mlx::core::array::data_size
    size_t data_size() const
    The size (in elements) of the underlying buffer the array points to.
    Definition array.h:320
    + +
    mlx::core::simd::load
    Simd< T, N > load(const T *x)
    Definition base_simd.h:27
    +
    mlx::core::simd::max_size
    static constexpr int max_size
    Definition base_simd.h:13
    +
    mlx::core::simd::store
    void store(T *dst, Simd< T, N > x)
    Definition base_simd.h:32
    +
    mlx::core
    Definition allocator.h:7
    +
    mlx::core::bool_
    constexpr Dtype bool_
    Definition dtype.h:67
    +
    mlx::core::uint64
    constexpr Dtype uint64
    Definition dtype.h:72
    +
    mlx::core::get_binary_op_type
    BinaryOpType get_binary_op_type(const array &a, const array &b)
    Definition binary.h:19
    +
    mlx::core::uint16
    constexpr Dtype uint16
    Definition dtype.h:70
    +
    mlx::core::collapse_contiguous_dims
    std::tuple< Shape, std::vector< Strides > > collapse_contiguous_dims(const Shape &shape, const std::vector< Strides > &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())
    +
    mlx::core::bfloat16
    constexpr Dtype bfloat16
    Definition dtype.h:81
    +
    mlx::core::BinaryOpType::General
    @ General
    Definition binary.h:16
    +
    mlx::core::BinaryOpType::VectorVector
    @ VectorVector
    Definition binary.h:15
    +
    mlx::core::BinaryOpType::ScalarScalar
    @ ScalarScalar
    Definition binary.h:12
    +
    mlx::core::BinaryOpType::VectorScalar
    @ VectorScalar
    Definition binary.h:14
    +
    mlx::core::BinaryOpType::ScalarVector
    @ ScalarVector
    Definition binary.h:13
    +
    mlx::core::int32
    constexpr Dtype int32
    Definition dtype.h:76
    +
    mlx::core::binary_op_dispatch_dims
    void binary_op_dispatch_dims(const array &a, const array &b, array &out, Op op, int dim, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &out_strides)
    Definition binary.h:122
    +
    mlx::core::float32
    constexpr Dtype float32
    Definition dtype.h:80
    +
    mlx::core::Shape
    std::vector< ShapeElem > Shape
    Definition array.h:21
    +
    mlx::core::set_binary_op_output_data
    void set_binary_op_output_data(const array &a, const array &b, array &out, BinaryOpType bopt, bool donate_with_move=false)
    Definition binary.h:37
    +
    mlx::core::int16
    constexpr Dtype int16
    Definition dtype.h:75
    +
    mlx::core::Strides
    std::vector< int64_t > Strides
    Definition array.h:22
    +
    mlx::core::binary_op_dims
    void binary_op_dims(const T *a, const T *b, U *out, Op op, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &out_strides, int axis)
    Definition binary.h:89
    +
    mlx::core::int8
    constexpr Dtype int8
    Definition dtype.h:74
    +
    mlx::core::int64
    constexpr Dtype int64
    Definition dtype.h:77
    +
    mlx::core::uint8
    constexpr Dtype uint8
    Definition dtype.h:69
    +
    mlx::core::binary_op
    void binary_op(const array &a, const array &b, array &out, Op op)
    Definition binary.h:194
    +
    mlx::core::float16
    constexpr Dtype float16
    Definition dtype.h:79
    +
    mlx::core::uint32
    constexpr Dtype uint32
    Definition dtype.h:71
    +
    mlx::core::binary
    void binary(const array &a, const array &b, array &out, Op op)
    Definition binary.h:326
    +
    mlx::core::complex64
    constexpr Dtype complex64
    Definition dtype.h:82
    + +
    mlx::core::ContiguousIterator
    Definition utils.h:73
    +
    mlx::core::ContiguousIterator::loc
    int64_t loc
    Definition utils.h:126
    +
    mlx::core::ContiguousIterator::step
    void step()
    Definition utils.h:74
    +
    mlx::core::ScalarVector
    Definition binary.h:40
    +
    mlx::core::ScalarVector::ScalarVector
    ScalarVector(Op op_)
    Definition binary.h:43
    +
    mlx::core::ScalarVector::operator()
    void operator()(const T *a, const T *b, U *dst, int size)
    Definition binary.h:46
    +
    mlx::core::ScalarVector::op
    Op op
    Definition binary.h:41
    +
    mlx::core::VectorScalar
    Definition binary.h:16
    +
    mlx::core::VectorScalar::operator()
    void operator()(const T *a, const T *b, U *dst, int size)
    Definition binary.h:22
    +
    mlx::core::VectorScalar::op
    Op op
    Definition binary.h:17
    +
    mlx::core::VectorScalar::VectorScalar
    VectorScalar(Op op_)
    Definition binary.h:19
    +
    mlx::core::VectorVector
    Definition binary.h:64
    +
    mlx::core::VectorVector::VectorVector
    VectorVector(Op op_)
    Definition binary.h:67
    +
    mlx::core::VectorVector::op
    Op op
    Definition binary.h:65
    +
    mlx::core::VectorVector::operator()
    void operator()(const T *a, const T *b, U *dst, int size)
    Definition binary.h:70
    +
    mlx::core::simd::Simd
    Definition accelerate_simd.h:55
    +
    +
    + + + + diff --git a/docs/build/html/cpu_2binary__ops_8h.html b/docs/build/html/cpu_2binary__ops_8h.html new file mode 100644 index 000000000..a484492bc --- /dev/null +++ b/docs/build/html/cpu_2binary__ops_8h.html @@ -0,0 +1,278 @@ + + + + + + + +MLX: mlx/backend/cpu/binary_ops.h File Reference + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Classes | +Namespaces | +Macros
    +
    binary_ops.h File Reference
    +
    +
    +
    #include "mlx/backend/cpu/simd/simd.h"
    +
    +

    Go to the source code of this file.

    + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

    +Classes

    struct  mlx::core::detail::Add
     
    struct  mlx::core::detail::ArcTan2
     
    struct  mlx::core::detail::Divide
     
    struct  mlx::core::detail::Multiply
     
    struct  mlx::core::detail::Subtract
     
    struct  mlx::core::detail::LogicalAnd
     
    struct  mlx::core::detail::LogicalOr
     
    struct  mlx::core::detail::BitwiseAnd
     
    struct  mlx::core::detail::BitwiseOr
     
    struct  mlx::core::detail::BitwiseXor
     
    struct  mlx::core::detail::LeftShift
     
    struct  mlx::core::detail::RightShift
     
    struct  mlx::core::detail::Remainder
     
    struct  mlx::core::detail::Maximum
     
    struct  mlx::core::detail::Minimum
     
    struct  mlx::core::detail::Power
     
    struct  mlx::core::detail::Equal
     
    struct  mlx::core::detail::Greater
     
    struct  mlx::core::detail::GreaterEqual
     
    struct  mlx::core::detail::Less
     
    struct  mlx::core::detail::LessEqual
     
    struct  mlx::core::detail::NotEqual
     
    struct  mlx::core::detail::NaNEqual
     
    struct  mlx::core::detail::LogAddExp
     
    struct  mlx::core::detail::Select
     
    + + + + + + + +

    +Namespaces

    namespace  mlx
     
    namespace  mlx::core
     
    namespace  mlx::core::detail
     
    + + + + + + + +

    +Macros

    #define BINARY_SINGLE()
     
    #define DEFAULT_BINARY_OP(Op, op)
     
    #define DEFAULT_BOOL_OP(Op, op)
     
    +

    Macro Definition Documentation

    + +

    ◆ BINARY_SINGLE

    + +
    +
    + + + + + + + +
    #define BINARY_SINGLE()
    +
    +Value:
    template <typename T> \
    +
    T operator()(T x, T y) { \
    +
    return (*this)(Simd<T, 1>(x), Simd<T, 1>(y)).value; \
    +
    }
    +
    +
    +
    + +

    ◆ DEFAULT_BINARY_OP

    + +
    +
    + + + + + + + + + + + +
    #define DEFAULT_BINARY_OP( Op,
    op )
    +
    +Value:
    struct Op { \
    +
    template <int N, typename T> \
    +
    Simd<T, N> operator()(Simd<T, N> x, Simd<T, N> y) { \
    +
    return op(x, y); \
    +
    } \
    +
    BINARY_SINGLE() \
    +
    };
    +
    +
    +
    + +

    ◆ DEFAULT_BOOL_OP

    + +
    +
    + + + + + + + + + + + +
    #define DEFAULT_BOOL_OP( Op,
    op )
    +
    +Value:
    struct Op { \
    +
    template <int N, typename T> \
    +
    Simd<bool, N> operator()(Simd<T, N> x, Simd<T, N> y) { \
    +
    return op(x, y); \
    +
    } \
    +
    template <typename T> \
    +
    bool operator()(T x, T y) { \
    +
    return (*this)(Simd<T, 1>(x), Simd<T, 1>(y)).value; \
    +
    } \
    +
    };
    +
    +
    +
    +
    +
    + + + + diff --git a/docs/build/html/cpu_2binary__ops_8h.js b/docs/build/html/cpu_2binary__ops_8h.js new file mode 100644 index 000000000..a6c93c52e --- /dev/null +++ b/docs/build/html/cpu_2binary__ops_8h.js @@ -0,0 +1,31 @@ +var cpu_2binary__ops_8h = +[ + [ "mlx::core::detail::Add", "structmlx_1_1core_1_1detail_1_1_add.html", "structmlx_1_1core_1_1detail_1_1_add" ], + [ "mlx::core::detail::ArcTan2", "structmlx_1_1core_1_1detail_1_1_arc_tan2.html", "structmlx_1_1core_1_1detail_1_1_arc_tan2" ], + [ "mlx::core::detail::Divide", "structmlx_1_1core_1_1detail_1_1_divide.html", "structmlx_1_1core_1_1detail_1_1_divide" ], + [ "mlx::core::detail::Multiply", "structmlx_1_1core_1_1detail_1_1_multiply.html", "structmlx_1_1core_1_1detail_1_1_multiply" ], + [ "mlx::core::detail::Subtract", "structmlx_1_1core_1_1detail_1_1_subtract.html", "structmlx_1_1core_1_1detail_1_1_subtract" ], + [ "mlx::core::detail::LogicalAnd", "structmlx_1_1core_1_1detail_1_1_logical_and.html", "structmlx_1_1core_1_1detail_1_1_logical_and" ], + [ "mlx::core::detail::LogicalOr", "structmlx_1_1core_1_1detail_1_1_logical_or.html", "structmlx_1_1core_1_1detail_1_1_logical_or" ], + [ "mlx::core::detail::BitwiseAnd", "structmlx_1_1core_1_1detail_1_1_bitwise_and.html", "structmlx_1_1core_1_1detail_1_1_bitwise_and" ], + [ "mlx::core::detail::BitwiseOr", "structmlx_1_1core_1_1detail_1_1_bitwise_or.html", "structmlx_1_1core_1_1detail_1_1_bitwise_or" ], + [ "mlx::core::detail::BitwiseXor", "structmlx_1_1core_1_1detail_1_1_bitwise_xor.html", "structmlx_1_1core_1_1detail_1_1_bitwise_xor" ], + [ "mlx::core::detail::LeftShift", "structmlx_1_1core_1_1detail_1_1_left_shift.html", "structmlx_1_1core_1_1detail_1_1_left_shift" ], + [ "mlx::core::detail::RightShift", "structmlx_1_1core_1_1detail_1_1_right_shift.html", "structmlx_1_1core_1_1detail_1_1_right_shift" ], + [ "mlx::core::detail::Remainder", "structmlx_1_1core_1_1detail_1_1_remainder.html", "structmlx_1_1core_1_1detail_1_1_remainder" ], + [ "mlx::core::detail::Maximum", "structmlx_1_1core_1_1detail_1_1_maximum.html", "structmlx_1_1core_1_1detail_1_1_maximum" ], + [ "mlx::core::detail::Minimum", "structmlx_1_1core_1_1detail_1_1_minimum.html", "structmlx_1_1core_1_1detail_1_1_minimum" ], + [ "mlx::core::detail::Power", "structmlx_1_1core_1_1detail_1_1_power.html", "structmlx_1_1core_1_1detail_1_1_power" ], + [ "mlx::core::detail::Equal", "structmlx_1_1core_1_1detail_1_1_equal.html", "structmlx_1_1core_1_1detail_1_1_equal" ], + [ "mlx::core::detail::Greater", "structmlx_1_1core_1_1detail_1_1_greater.html", "structmlx_1_1core_1_1detail_1_1_greater" ], + [ "mlx::core::detail::GreaterEqual", "structmlx_1_1core_1_1detail_1_1_greater_equal.html", "structmlx_1_1core_1_1detail_1_1_greater_equal" ], + [ "mlx::core::detail::Less", "structmlx_1_1core_1_1detail_1_1_less.html", "structmlx_1_1core_1_1detail_1_1_less" ], + [ "mlx::core::detail::LessEqual", "structmlx_1_1core_1_1detail_1_1_less_equal.html", "structmlx_1_1core_1_1detail_1_1_less_equal" ], + [ "mlx::core::detail::NotEqual", "structmlx_1_1core_1_1detail_1_1_not_equal.html", "structmlx_1_1core_1_1detail_1_1_not_equal" ], + [ "mlx::core::detail::NaNEqual", "structmlx_1_1core_1_1detail_1_1_na_n_equal.html", "structmlx_1_1core_1_1detail_1_1_na_n_equal" ], + [ "mlx::core::detail::LogAddExp", "structmlx_1_1core_1_1detail_1_1_log_add_exp.html", "structmlx_1_1core_1_1detail_1_1_log_add_exp" ], + [ "mlx::core::detail::Select", "structmlx_1_1core_1_1detail_1_1_select.html", "structmlx_1_1core_1_1detail_1_1_select" ], + [ "BINARY_SINGLE", "cpu_2binary__ops_8h.html#ad606fe7b9586413e760cfe461c9be119", null ], + [ "DEFAULT_BINARY_OP", "cpu_2binary__ops_8h.html#a3f97e0a58306201b54eb73fac0affc8a", null ], + [ "DEFAULT_BOOL_OP", "cpu_2binary__ops_8h.html#a98fb0ee78e9fd26ad6d0a2b365f7caa7", null ] +]; \ No newline at end of file diff --git a/docs/build/html/cpu_2binary__ops_8h_source.html b/docs/build/html/cpu_2binary__ops_8h_source.html new file mode 100644 index 000000000..0273cb1ba --- /dev/null +++ b/docs/build/html/cpu_2binary__ops_8h_source.html @@ -0,0 +1,287 @@ + + + + + + + +MLX: mlx/backend/cpu/binary_ops.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    binary_ops.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2023-2024 Apple Inc.
    +
    2
    +
    3#pragma once
    +
    4
    +
    5#include "mlx/backend/cpu/simd/simd.h"
    +
    6
    +
    +
    7namespace mlx::core::detail {
    +
    8
    +
    9using namespace mlx::core::simd;
    +
    10
    +
    +
    11#define BINARY_SINGLE() \
    +
    12 template <typename T> \
    +
    13 T operator()(T x, T y) { \
    +
    14 return (*this)(Simd<T, 1>(x), Simd<T, 1>(y)).value; \
    +
    15 }
    +
    +
    16
    +
    +
    17#define DEFAULT_BINARY_OP(Op, op) \
    +
    18 struct Op { \
    +
    19 template <int N, typename T> \
    +
    20 Simd<T, N> operator()(Simd<T, N> x, Simd<T, N> y) { \
    +
    21 return op(x, y); \
    +
    22 } \
    +
    23 BINARY_SINGLE() \
    +
    24 };
    +
    +
    25
    +
    26DEFAULT_BINARY_OP(Add, operator+)
    +
    27DEFAULT_BINARY_OP(ArcTan2, atan2)
    +
    28DEFAULT_BINARY_OP(Divide, operator/)
    +
    29DEFAULT_BINARY_OP(Multiply, operator*)
    +
    30DEFAULT_BINARY_OP(Subtract, operator-)
    +
    31DEFAULT_BINARY_OP(LogicalAnd, operator&&)
    +
    32DEFAULT_BINARY_OP(LogicalOr, operator||)
    +
    33DEFAULT_BINARY_OP(BitwiseAnd, operator&)
    +
    34DEFAULT_BINARY_OP(BitwiseOr, operator|)
    +
    35DEFAULT_BINARY_OP(BitwiseXor, operator^)
    +
    36DEFAULT_BINARY_OP(LeftShift, operator<<)
    +
    37DEFAULT_BINARY_OP(RightShift, operator>>)
    +
    38DEFAULT_BINARY_OP(Remainder, remainder)
    +
    39DEFAULT_BINARY_OP(Maximum, maximum)
    +
    40DEFAULT_BINARY_OP(Minimum, minimum)
    +
    41DEFAULT_BINARY_OP(Power, pow)
    +
    42
    +
    +
    43#define DEFAULT_BOOL_OP(Op, op) \
    +
    44 struct Op { \
    +
    45 template <int N, typename T> \
    +
    46 Simd<bool, N> operator()(Simd<T, N> x, Simd<T, N> y) { \
    +
    47 return op(x, y); \
    +
    48 } \
    +
    49 template <typename T> \
    +
    50 bool operator()(T x, T y) { \
    +
    51 return (*this)(Simd<T, 1>(x), Simd<T, 1>(y)).value; \
    +
    52 } \
    +
    53 };
    +
    +
    54
    +
    55DEFAULT_BOOL_OP(Equal, operator==)
    +
    56DEFAULT_BOOL_OP(Greater, operator>)
    +
    57DEFAULT_BOOL_OP(GreaterEqual, operator>=)
    +
    58DEFAULT_BOOL_OP(Less, operator<)
    +
    59DEFAULT_BOOL_OP(LessEqual, operator<=)
    +
    60DEFAULT_BOOL_OP(NotEqual, operator!=)
    +
    61
    +
    +
    62struct NaNEqual {
    +
    63 template <int N, typename T>
    +
    +
    64 Simd<bool, N> operator()(Simd<T, N> x, Simd<T, N> y) {
    +
    65 return x == y || (isnan(x) && isnan(y));
    +
    66 }
    +
    +
    67 template <typename T>
    +
    +
    68 bool operator()(T x, T y) {
    +
    69 return (*this)(Simd<T, 1>(x), Simd<T, 1>(y)).value;
    +
    70 }
    +
    +
    71};
    +
    +
    72
    +
    +
    73struct LogAddExp {
    +
    74 template <int N, typename T>
    +
    +
    75 Simd<T, N> operator()(Simd<T, N> x, Simd<T, N> y) {
    +
    76 auto maxval = maximum(x, y);
    +
    77 auto minval = minimum(x, y);
    +
    78 auto mask = minval == -inf || maxval == inf;
    +
    79 auto out = maxval + log1p(exp(minval - maxval));
    +
    80 return select(mask, Simd<T, N>(maxval), Simd<T, N>(out));
    +
    81 }
    +
    +
    82 BINARY_SINGLE()
    +
    83};
    +
    +
    84
    +
    +
    85struct Select {
    +
    86 template <typename T>
    +
    +
    87 T operator()(bool condition, T x, T y) {
    +
    88 return (*this)(Simd<bool, 1>(condition), Simd<T, 1>(x), Simd<T, 1>(y))
    +
    89 .value;
    +
    90 }
    +
    +
    91
    +
    92 template <int N, typename T>
    +
    +
    93 Simd<T, N> operator()(Simd<bool, N> condition, Simd<T, N> x, Simd<T, N> y) {
    +
    94 return select(condition, x, y);
    +
    95 }
    +
    +
    96};
    +
    +
    97
    +
    98} // namespace mlx::core::detail
    +
    +
    DEFAULT_BINARY_OP
    #define DEFAULT_BINARY_OP(Op, op)
    Definition binary_ops.h:17
    +
    DEFAULT_BOOL_OP
    #define DEFAULT_BOOL_OP(Op, op)
    Definition binary_ops.h:43
    +
    BINARY_SINGLE
    #define BINARY_SINGLE()
    Definition binary_ops.h:11
    +
    mlx::core::detail
    Definition binary_ops.h:7
    +
    mlx::core::simd
    Definition accelerate_fp16_simd.h:9
    +
    mlx::core::simd::isnan
    Simd< bool, N > isnan(Simd< T, N > v)
    Definition accelerate_simd.h:141
    +
    mlx::core::simd::minimum
    Simd< T, N > minimum(Simd< T, N > a, Simd< T, N > b)
    Definition accelerate_simd.h:210
    +
    mlx::core::simd::pow
    Simd< float16_t, N > pow(Simd< float16_t, N > x, Simd< float16_t, N > y)
    Definition accelerate_fp16_simd.h:54
    +
    mlx::core::simd::atan2
    Simd< float16_t, N > atan2(Simd< float16_t, N > x, Simd< float16_t, N > y)
    Definition accelerate_fp16_simd.h:52
    +
    mlx::core::simd::inf
    constexpr float inf
    Definition math.h:9
    +
    mlx::core::simd::maximum
    Simd< T, N > maximum(Simd< T, N > a, Simd< T, N > b)
    Definition accelerate_simd.h:204
    +
    mlx::core::simd::exp
    Simd< T, N > exp(Simd< T, N > in)
    Compute exp(x) in an optimizer friendly way as follows:
    Definition math.h:28
    +
    mlx::core::simd::remainder
    Simd< float16_t, N > remainder(Simd< float16_t, N > x, Simd< float16_t, N > y)
    Definition accelerate_fp16_simd.h:53
    +
    mlx::core::simd::log1p
    Simd< float16_t, N > log1p(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:40
    +
    mlx::core::simd::select
    Simd< T1, N > select(Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)
    Definition accelerate_simd.h:231
    + +
    mlx::core::detail::Add
    Definition binary_ops.h:26
    +
    mlx::core::detail::ArcTan2
    Definition binary_ops.h:27
    +
    mlx::core::detail::BitwiseAnd
    Definition binary_ops.h:33
    +
    mlx::core::detail::BitwiseOr
    Definition binary_ops.h:34
    +
    mlx::core::detail::BitwiseXor
    Definition binary_ops.h:35
    +
    mlx::core::detail::Divide
    Definition binary_ops.h:28
    +
    mlx::core::detail::Equal
    Definition binary_ops.h:55
    +
    mlx::core::detail::GreaterEqual
    Definition binary_ops.h:57
    +
    mlx::core::detail::Greater
    Definition binary_ops.h:56
    +
    mlx::core::detail::LeftShift
    Definition binary_ops.h:36
    +
    mlx::core::detail::LessEqual
    Definition binary_ops.h:59
    +
    mlx::core::detail::Less
    Definition binary_ops.h:58
    +
    mlx::core::detail::LogAddExp
    Definition binary_ops.h:73
    +
    mlx::core::detail::LogAddExp::operator()
    Simd< T, N > operator()(Simd< T, N > x, Simd< T, N > y)
    Definition binary_ops.h:75
    +
    mlx::core::detail::LogicalAnd
    Definition binary_ops.h:31
    +
    mlx::core::detail::LogicalOr
    Definition binary_ops.h:32
    +
    mlx::core::detail::Maximum
    Definition binary_ops.h:39
    +
    mlx::core::detail::Minimum
    Definition binary_ops.h:40
    +
    mlx::core::detail::Multiply
    Definition binary_ops.h:29
    +
    mlx::core::detail::NaNEqual
    Definition binary_ops.h:62
    +
    mlx::core::detail::NaNEqual::operator()
    bool operator()(T x, T y)
    Definition binary_ops.h:68
    +
    mlx::core::detail::NaNEqual::operator()
    Simd< bool, N > operator()(Simd< T, N > x, Simd< T, N > y)
    Definition binary_ops.h:64
    +
    mlx::core::detail::NotEqual
    Definition binary_ops.h:60
    +
    mlx::core::detail::Power
    Definition binary_ops.h:41
    +
    mlx::core::detail::Remainder
    Definition binary_ops.h:38
    +
    mlx::core::detail::RightShift
    Definition binary_ops.h:37
    +
    mlx::core::detail::Select
    Definition binary_ops.h:85
    +
    mlx::core::detail::Select::operator()
    Simd< T, N > operator()(Simd< bool, N > condition, Simd< T, N > x, Simd< T, N > y)
    Definition binary_ops.h:93
    +
    mlx::core::detail::Select::operator()
    T operator()(bool condition, T x, T y)
    Definition binary_ops.h:87
    +
    mlx::core::detail::Subtract
    Definition binary_ops.h:30
    +
    mlx::core::simd::Simd
    Definition accelerate_simd.h:55
    +
    +
    + + + + diff --git a/docs/build/html/common_2binary__two_8h.html b/docs/build/html/cpu_2binary__two_8h.html similarity index 87% rename from docs/build/html/common_2binary__two_8h.html rename to docs/build/html/cpu_2binary__two_8h.html index cf4c65523..a396493f6 100644 --- a/docs/build/html/common_2binary__two_8h.html +++ b/docs/build/html/cpu_2binary__two_8h.html @@ -3,9 +3,9 @@ - + -MLX: mlx/backend/common/binary_two.h File Reference +MLX: mlx/backend/cpu/binary_two.h File Reference @@ -52,7 +52,7 @@
    - +
    @@ -107,10 +107,10 @@ $(function(){initNavTree('common_2binary__two_8h.html',''); initResizable(true);
    binary_two.h File Reference
    -
    #include "mlx/backend/common/binary.h"
    -#include "mlx/backend/common/utils.h"
    +
    #include "mlx/backend/common/utils.h"
    +#include "mlx/backend/cpu/binary.h"
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    @@ -124,8 +124,8 @@ Namespaces diff --git a/docs/build/html/common_2binary__two_8h_source.html b/docs/build/html/cpu_2binary__two_8h_source.html similarity index 78% rename from docs/build/html/common_2binary__two_8h_source.html rename to docs/build/html/cpu_2binary__two_8h_source.html index e8bd059ef..f10f73e7b 100644 --- a/docs/build/html/common_2binary__two_8h_source.html +++ b/docs/build/html/cpu_2binary__two_8h_source.html @@ -3,9 +3,9 @@ - + -MLX: mlx/backend/common/binary_two.h Source File +MLX: mlx/backend/cpu/binary_two.h Source File @@ -52,7 +52,7 @@

    Namespaces

    - +
    @@ -105,19 +105,19 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    binary_two.h
    -Go to the documentation of this file.
    1// Copyright © 2023 Apple Inc.
    +Go to the documentation of this file.
    1// Copyright © 2023 Apple Inc.
    2
    3#pragma once
    4
    -
    5#include "mlx/backend/common/binary.h"
    -
    6#include "mlx/backend/common/utils.h"
    +
    5#include "mlx/backend/common/utils.h"
    +
    6#include "mlx/backend/cpu/binary.h"
    7
    8namespace mlx::core {
    9
    10namespace {
    11
    12template <typename T, typename U, typename Op, int D>
    -
    13void binary_op_dims(
    +
    13void binary_op_dims(
    14 const T* a,
    15 const T* b,
    16 U* out_a,
    @@ -131,11 +131,11 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    24 auto stride_a = a_strides[axis];
    25 auto stride_b = b_strides[axis];
    26 auto stride_out = out_strides[axis];
    -
    27 auto N = shape[axis];
    +
    27 auto N = shape[axis];
    28
    -
    29 for (int i = 0; i < N; i++) {
    +
    29 for (int i = 0; i < N; i++) {
    30 if constexpr (D > 1) {
    -
    31 binary_op_dims<T, U, Op, D - 1>(
    +
    31 binary_op_dims<T, U, Op, D - 1>(
    32 a,
    33 b,
    34 out_a,
    @@ -157,7 +157,7 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    50}
    51
    52template <typename T, typename U, typename Op>
    -
    53void binary_op_dispatch_dims(
    +
    53void binary_op_dispatch_dims(
    54 const array& a,
    55 const array& b,
    56 array& out_a,
    @@ -176,7 +176,7 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    69 int ndim = shape.size();
    70 switch (ndim) {
    71 case 1:
    -
    72 binary_op_dims<T, U, Op, 1>(
    +
    72 binary_op_dims<T, U, Op, 1>(
    73 a_ptr,
    74 b_ptr,
    75 out_a_ptr,
    @@ -189,7 +189,7 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    82 0);
    83 return;
    84 case 2:
    -
    85 binary_op_dims<T, U, Op, 2>(
    +
    85 binary_op_dims<T, U, Op, 2>(
    86 a_ptr,
    87 b_ptr,
    88 out_a_ptr,
    @@ -207,7 +207,7 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    100 ContiguousIterator b_it(shape, b_strides, ndim - 2);
    101 auto stride = out_strides[ndim - 3];
    102 for (size_t elem = 0; elem < a.size(); elem += stride) {
    -
    103 binary_op_dims<T, U, Op, 2>(
    +
    103 binary_op_dims<T, U, Op, 2>(
    104 a_ptr + a_it.loc,
    105 b_ptr + b_it.loc,
    106 out_a_ptr + elem,
    @@ -224,20 +224,20 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    117}
    118
    119template <typename T, typename U = T, typename Op>
    -
    120void binary_op(
    +
    120void binary_op(
    121 const array& a,
    122 const array& b,
    123 std::vector<array>& outputs,
    124 Op op) {
    -
    125 auto bopt = get_binary_op_type(a, b);
    +
    125 auto bopt = get_binary_op_type(a, b);
    126 auto& out_a = outputs[0];
    127 auto& out_b = outputs[1];
    -
    128 set_binary_op_output_data(a, b, out_a, bopt);
    -
    129 set_binary_op_output_data(a, b, out_b, bopt);
    +
    128 set_binary_op_output_data(a, b, out_a, bopt);
    +
    129 set_binary_op_output_data(a, b, out_b, bopt);
    130
    131 // The full computation is scalar scalar so call the base op once
    -
    132 if (bopt == BinaryOpType::General) {
    -
    133 binary_op_dispatch_dims<T, U, Op>(a, b, out_a, out_b, op);
    +
    132 if (bopt == BinaryOpType::General) {
    +
    133 binary_op_dispatch_dims<T, U, Op>(a, b, out_a, out_b, op);
    134 return;
    135 }
    136
    @@ -245,16 +245,16 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    138 auto b_ptr = b.data<T>();
    139 auto out_a_ptr = out_a.data<U>();
    140 auto out_b_ptr = out_b.data<U>();
    -
    141 if (bopt == BinaryOpType::ScalarScalar) {
    +
    141 if (bopt == BinaryOpType::ScalarScalar) {
    142 std::tie(*out_a_ptr, *out_b_ptr) = op(*a_ptr, *b_ptr);
    -
    143 } else if (bopt == BinaryOpType::ScalarVector) {
    +
    143 } else if (bopt == BinaryOpType::ScalarVector) {
    144 for (size_t i = 0; i < b.size(); ++i) {
    145 std::tie(*out_a_ptr, *out_b_ptr) = op(*a_ptr, *b_ptr);
    146 out_a_ptr++;
    147 out_b_ptr++;
    148 b_ptr++;
    149 }
    -
    150 } else if (bopt == BinaryOpType::VectorScalar) {
    +
    150 } else if (bopt == BinaryOpType::VectorScalar) {
    151 for (size_t i = 0; i < a.size(); ++i) {
    152 std::tie(*out_a_ptr, *out_b_ptr) = op(*a_ptr, *b_ptr);
    153 out_a_ptr++;
    @@ -273,50 +273,50 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    166}
    167
    168template <typename Op>
    -
    169void binary(
    +
    169void binary(
    170 const array& a,
    171 const array& b,
    172 std::vector<array>& outputs,
    173 Op op) {
    174 switch (outputs[0].dtype()) {
    175 case bool_:
    -
    176 binary_op<bool>(a, b, outputs, op);
    +
    176 binary_op<bool>(a, b, outputs, op);
    177 break;
    178 case uint8:
    -
    179 binary_op<uint8_t>(a, b, outputs, op);
    +
    179 binary_op<uint8_t>(a, b, outputs, op);
    180 break;
    181 case uint16:
    -
    182 binary_op<uint16_t>(a, b, outputs, op);
    +
    182 binary_op<uint16_t>(a, b, outputs, op);
    183 break;
    184 case uint32:
    -
    185 binary_op<uint32_t>(a, b, outputs, op);
    +
    185 binary_op<uint32_t>(a, b, outputs, op);
    186 break;
    187 case uint64:
    -
    188 binary_op<uint64_t>(a, b, outputs, op);
    +
    188 binary_op<uint64_t>(a, b, outputs, op);
    189 break;
    190 case int8:
    -
    191 binary_op<int8_t>(a, b, outputs, op);
    +
    191 binary_op<int8_t>(a, b, outputs, op);
    192 break;
    193 case int16:
    -
    194 binary_op<int16_t>(a, b, outputs, op);
    +
    194 binary_op<int16_t>(a, b, outputs, op);
    195 break;
    196 case int32:
    -
    197 binary_op<int32_t>(a, b, outputs, op);
    +
    197 binary_op<int32_t>(a, b, outputs, op);
    198 break;
    199 case int64:
    -
    200 binary_op<int64_t>(a, b, outputs, op);
    +
    200 binary_op<int64_t>(a, b, outputs, op);
    201 break;
    202 case float16:
    -
    203 binary_op<float16_t>(a, b, outputs, op);
    +
    203 binary_op<float16_t>(a, b, outputs, op);
    204 break;
    205 case float32:
    -
    206 binary_op<float>(a, b, outputs, op);
    +
    206 binary_op<float>(a, b, outputs, op);
    207 break;
    208 case bfloat16:
    -
    209 binary_op<bfloat16_t>(a, b, outputs, op);
    +
    209 binary_op<bfloat16_t>(a, b, outputs, op);
    210 break;
    211 case complex64:
    -
    212 binary_op<complex64_t>(a, b, outputs, op);
    +
    212 binary_op<complex64_t>(a, b, outputs, op);
    213 break;
    214 }
    215}
    @@ -326,24 +326,34 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable
    219} // namespace mlx::core
    mlx::core::array
    Definition array.h:24
    - -
    mlx::core::metal::binary
    const char * binary()
    + +
    mlx::core::simd::N
    constexpr int N
    Definition neon_fp16_simd.h:9
    mlx::core
    Definition allocator.h:7
    mlx::core::bool_
    constexpr Dtype bool_
    Definition dtype.h:67
    mlx::core::uint64
    constexpr Dtype uint64
    Definition dtype.h:72
    +
    mlx::core::get_binary_op_type
    BinaryOpType get_binary_op_type(const array &a, const array &b)
    Definition binary.h:19
    mlx::core::uint16
    constexpr Dtype uint16
    Definition dtype.h:70
    mlx::core::collapse_contiguous_dims
    std::tuple< Shape, std::vector< Strides > > collapse_contiguous_dims(const Shape &shape, const std::vector< Strides > &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())
    mlx::core::bfloat16
    constexpr Dtype bfloat16
    Definition dtype.h:81
    +
    mlx::core::BinaryOpType::General
    @ General
    Definition binary.h:16
    +
    mlx::core::BinaryOpType::ScalarScalar
    @ ScalarScalar
    Definition binary.h:12
    +
    mlx::core::BinaryOpType::VectorScalar
    @ VectorScalar
    Definition binary.h:14
    +
    mlx::core::BinaryOpType::ScalarVector
    @ ScalarVector
    Definition binary.h:13
    mlx::core::int32
    constexpr Dtype int32
    Definition dtype.h:76
    +
    mlx::core::binary_op_dispatch_dims
    void binary_op_dispatch_dims(const array &a, const array &b, array &out, Op op, int dim, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &out_strides)
    Definition binary.h:122
    mlx::core::float32
    constexpr Dtype float32
    Definition dtype.h:80
    mlx::core::Shape
    std::vector< ShapeElem > Shape
    Definition array.h:21
    +
    mlx::core::set_binary_op_output_data
    void set_binary_op_output_data(const array &a, const array &b, array &out, BinaryOpType bopt, bool donate_with_move=false)
    Definition binary.h:37
    mlx::core::int16
    constexpr Dtype int16
    Definition dtype.h:75
    mlx::core::Strides
    std::vector< int64_t > Strides
    Definition array.h:22
    +
    mlx::core::binary_op_dims
    void binary_op_dims(const T *a, const T *b, U *out, Op op, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &out_strides, int axis)
    Definition binary.h:89
    mlx::core::int8
    constexpr Dtype int8
    Definition dtype.h:74
    mlx::core::int64
    constexpr Dtype int64
    Definition dtype.h:77
    mlx::core::uint8
    constexpr Dtype uint8
    Definition dtype.h:69
    +
    mlx::core::binary_op
    void binary_op(const array &a, const array &b, array &out, Op op)
    Definition binary.h:194
    mlx::core::float16
    constexpr Dtype float16
    Definition dtype.h:79
    mlx::core::uint32
    constexpr Dtype uint32
    Definition dtype.h:71
    +
    mlx::core::binary
    void binary(const array &a, const array &b, array &out, Op op)
    Definition binary.h:326
    mlx::core::complex64
    constexpr Dtype complex64
    Definition dtype.h:82
    mlx::core::ContiguousIterator
    Definition utils.h:73
    @@ -351,8 +361,8 @@ $(function(){initNavTree('common_2binary__two_8h_source.html',''); initResizable diff --git a/docs/build/html/cpu_2copy_8h.html b/docs/build/html/cpu_2copy_8h.html new file mode 100644 index 000000000..de69b295f --- /dev/null +++ b/docs/build/html/cpu_2copy_8h.html @@ -0,0 +1,143 @@ + + + + + + + +MLX: mlx/backend/cpu/copy.h File Reference + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Namespaces | +Functions
    +
    copy.h File Reference
    +
    +
    +
    #include "mlx/array.h"
    +#include "mlx/backend/common/copy.h"
    +#include "mlx/backend/common/utils.h"
    +
    +

    Go to the source code of this file.

    + + + + + + +

    +Namespaces

    namespace  mlx
     
    namespace  mlx::core
     
    + + + + + + + +

    +Functions

    void mlx::core::copy (const array &src, array &dst, CopyType ctype)
     
    void mlx::core::copy_inplace (const array &src, array &dst, CopyType ctype)
     
    void mlx::core::copy_inplace (const array &src, array &dst, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)
     
    +
    +
    + + + + diff --git a/docs/build/html/cpu_2copy_8h.js b/docs/build/html/cpu_2copy_8h.js new file mode 100644 index 000000000..c9361c23b --- /dev/null +++ b/docs/build/html/cpu_2copy_8h.js @@ -0,0 +1,6 @@ +var cpu_2copy_8h = +[ + [ "mlx::core::copy", "namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2", null ], + [ "mlx::core::copy_inplace", "namespacemlx_1_1core.html#ae85bafda5ab0b4b2289591260cf07685", null ], + [ "mlx::core::copy_inplace", "namespacemlx_1_1core.html#a98495894a796b2cc6d022e7a03432c64", null ] +]; \ No newline at end of file diff --git a/docs/build/html/cpu_2copy_8h_source.html b/docs/build/html/cpu_2copy_8h_source.html new file mode 100644 index 000000000..f2839828e --- /dev/null +++ b/docs/build/html/cpu_2copy_8h_source.html @@ -0,0 +1,152 @@ + + + + + + + +MLX: mlx/backend/cpu/copy.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    copy.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2023-2024 Apple Inc.
    +
    2
    +
    3#pragma once
    +
    4
    +
    5#include "mlx/array.h"
    +
    6#include "mlx/backend/common/copy.h"
    +
    7#include "mlx/backend/common/utils.h"
    +
    8
    +
    9namespace mlx::core {
    +
    10
    +
    11void copy(const array& src, array& dst, CopyType ctype);
    +
    12void copy_inplace(const array& src, array& dst, CopyType ctype);
    +
    13
    +
    14void copy_inplace(
    +
    15 const array& src,
    +
    16 array& dst,
    +
    17 const Shape& data_shape,
    +
    18 const Strides& i_strides,
    +
    19 const Strides& o_strides,
    +
    20 int64_t i_offset,
    +
    21 int64_t o_offset,
    +
    22 CopyType ctype);
    +
    23
    +
    24} // namespace mlx::core
    + + +
    mlx::core::array
    Definition array.h:24
    + +
    mlx::core
    Definition allocator.h:7
    +
    mlx::core::copy
    void copy(const array &src, array &dst, CopyType ctype)
    +
    mlx::core::Shape
    std::vector< ShapeElem > Shape
    Definition array.h:21
    +
    mlx::core::Strides
    std::vector< int64_t > Strides
    Definition array.h:22
    +
    mlx::core::copy_inplace
    void copy_inplace(const array &src, array &dst, CopyType ctype)
    +
    mlx::core::CopyType
    CopyType
    Definition copy.h:9
    +
    +
    + + + + diff --git a/docs/build/html/backend_2accelerate_2utils_8h.html b/docs/build/html/cpu_2gemm_8h.html similarity index 78% rename from docs/build/html/backend_2accelerate_2utils_8h.html rename to docs/build/html/cpu_2gemm_8h.html index 16c2725d9..6a41765ea 100644 --- a/docs/build/html/backend_2accelerate_2utils_8h.html +++ b/docs/build/html/cpu_2gemm_8h.html @@ -3,9 +3,9 @@ - + -MLX: mlx/backend/accelerate/utils.h File Reference +MLX: mlx/backend/cpu/gemm.h File Reference @@ -52,7 +52,7 @@
    - +
    @@ -105,13 +105,12 @@ $(function(){initNavTree('backend_2accelerate_2utils_8h.html',''); initResizable
    Namespaces | Functions
    -
    utils.h File Reference
    +
    gemm.h File Reference
    -
    #include <Accelerate/Accelerate.h>
    -#include "mlx/dtype.h"
    +
    #include "mlx/array.h"
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    @@ -122,16 +121,17 @@ Namespaces

    Namespaces

    - - + + +

    Functions

    BNNSDataType mlx::core::to_bnns_dtype (Dtype mlx_dtype)
     
    template<typename T>
    void mlx::core::matmul (const array &a, const array &b, array &out, bool a_transposed, bool b_transposed, size_t lda, size_t ldb, float alpha, float beta)
     
    diff --git a/docs/build/html/cpu_2gemm_8h.js b/docs/build/html/cpu_2gemm_8h.js new file mode 100644 index 000000000..4b39c7431 --- /dev/null +++ b/docs/build/html/cpu_2gemm_8h.js @@ -0,0 +1,4 @@ +var cpu_2gemm_8h = +[ + [ "mlx::core::matmul", "namespacemlx_1_1core.html#aaacf0afe13d77a5c49ce96f1e833eb2d", null ] +]; \ No newline at end of file diff --git a/docs/build/html/jit_2gemv__masked_8h_source.html b/docs/build/html/cpu_2gemm_8h_source.html similarity index 57% rename from docs/build/html/jit_2gemv__masked_8h_source.html rename to docs/build/html/cpu_2gemm_8h_source.html index d1ad2504b..6ccfed1c7 100644 --- a/docs/build/html/jit_2gemv__masked_8h_source.html +++ b/docs/build/html/cpu_2gemm_8h_source.html @@ -3,9 +3,9 @@ - + -MLX: mlx/backend/metal/jit/gemv_masked.h Source File +MLX: mlx/backend/cpu/gemm.h Source File @@ -52,7 +52,7 @@
    - +
    @@ -102,42 +102,40 @@ $(function(){initNavTree('jit_2gemv__masked_8h_source.html',''); initResizable(t
    -
    gemv_masked.h
    +
    gemm.h
    -Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    +Go to the documentation of this file.
    1// Copyright © 2025 Apple Inc.
    2
    -
    3constexpr std::string_view gemv_masked_kernel = R"(
    -
    4template [[host_name("{name}")]] [[kernel]] void
    -
    5gemv_{trans}masked<{itype}, {outm_t}, {opm_t}, {bm}, {bn}, {sm}, {sn}, {tm}, {tn}, {nc}>(
    -
    6 const device {itype}* mat [[buffer(0)]],
    -
    7 const device {itype}* in_vec [[buffer(1)]],
    -
    8 device {itype}* out_vec [[buffer(3)]],
    -
    9 const constant int& in_vec_size [[buffer(4)]],
    -
    10 const constant int& out_vec_size [[buffer(5)]],
    -
    11 const constant int& marix_ld [[buffer(6)]],
    -
    12 const constant int& batch_ndim [[buffer(9)]],
    -
    13 const constant int* batch_shape [[buffer(10)]],
    -
    14 const constant int64_t* vector_batch_stride [[buffer(11)]],
    -
    15 const constant int64_t* matrix_batch_stride [[buffer(12)]],
    -
    16 const device {outm_t}* out_mask [[buffer(20)]],
    -
    17 const device {opm_t}* mat_mask [[buffer(21)]],
    -
    18 const device {opm_t}* vec_mask [[buffer(22)]],
    -
    19 const constant int* mask_strides [[buffer(23)]],
    -
    20 const constant int64_t* mask_batch_strides [[buffer(24)]],
    -
    21 uint3 tid [[threadgroup_position_in_grid]],
    -
    22 uint3 lid [[thread_position_in_threadgroup]],
    -
    23 uint simd_gid [[simdgroup_index_in_threadgroup]],
    -
    24 uint simd_lid [[thread_index_in_simdgroup]]);
    -
    25)";
    -
    gemv_masked_kernel
    constexpr std::string_view gemv_masked_kernel
    Definition gemv_masked.h:3
    +
    3#pragma once
    +
    4#include "mlx/array.h"
    +
    5
    +
    6namespace mlx::core {
    +
    7
    +
    8template <typename T>
    +
    9void matmul(
    +
    10 const array& a,
    +
    11 const array& b,
    +
    12 array& out,
    +
    13 bool a_transposed,
    +
    14 bool b_transposed,
    +
    15 size_t lda,
    +
    16 size_t ldb,
    +
    17 float alpha,
    +
    18 float beta);
    +
    19
    +
    20} // namespace mlx::core
    + +
    mlx::core::array
    Definition array.h:24
    +
    mlx::core
    Definition allocator.h:7
    +
    mlx::core::matmul
    void matmul(const array &a, const array &b, array &out, bool a_transposed, bool b_transposed, size_t lda, size_t ldb, float alpha, float beta)
    diff --git a/docs/build/html/cpu_2slicing_8h.html b/docs/build/html/cpu_2slicing_8h.html new file mode 100644 index 000000000..91507e01e --- /dev/null +++ b/docs/build/html/cpu_2slicing_8h.html @@ -0,0 +1,139 @@ + + + + + + + +MLX: mlx/backend/cpu/slicing.h File Reference + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Namespaces | +Functions
    +
    slicing.h File Reference
    +
    +
    +
    #include "mlx/array.h"
    +
    +

    Go to the source code of this file.

    + + + + + + +

    +Namespaces

    namespace  mlx
     
    namespace  mlx::core
     
    + + + + + +

    +Functions

    std::tuple< int64_t, Strides > mlx::core::prepare_slice (const array &in, const Shape &start_indices, const Shape &strides)
     
    void mlx::core::shared_buffer_slice (const array &in, const Strides &out_strides, size_t data_offset, size_t data_size, array &out)
     
    +
    +
    + + + + diff --git a/docs/build/html/cpu_2slicing_8h.js b/docs/build/html/cpu_2slicing_8h.js new file mode 100644 index 000000000..6ce5e1767 --- /dev/null +++ b/docs/build/html/cpu_2slicing_8h.js @@ -0,0 +1,5 @@ +var cpu_2slicing_8h = +[ + [ "mlx::core::prepare_slice", "namespacemlx_1_1core.html#a9fac4b96a3d783c6392ebc08c81ebdbd", null ], + [ "mlx::core::shared_buffer_slice", "namespacemlx_1_1core.html#a349a9fc2bfd950f679a3fe39b8bdedad", null ] +]; \ No newline at end of file diff --git a/docs/build/html/cpu_2slicing_8h_source.html b/docs/build/html/cpu_2slicing_8h_source.html new file mode 100644 index 000000000..c10da9bf8 --- /dev/null +++ b/docs/build/html/cpu_2slicing_8h_source.html @@ -0,0 +1,146 @@ + + + + + + + +MLX: mlx/backend/cpu/slicing.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    slicing.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    +
    2
    +
    3#pragma once
    +
    4
    +
    5#include "mlx/array.h"
    +
    6
    +
    7namespace mlx::core {
    +
    8
    +
    9std::tuple<int64_t, Strides> prepare_slice(
    +
    10 const array& in,
    +
    11 const Shape& start_indices,
    +
    12 const Shape& strides);
    +
    13
    +
    14void shared_buffer_slice(
    +
    15 const array& in,
    +
    16 const Strides& out_strides,
    +
    17 size_t data_offset,
    +
    18 size_t data_size,
    +
    19 array& out);
    +
    20
    +
    21} // namespace mlx::core
    + +
    mlx::core::array
    Definition array.h:24
    +
    mlx::core
    Definition allocator.h:7
    +
    mlx::core::shared_buffer_slice
    void shared_buffer_slice(const array &in, const Strides &out_strides, size_t data_offset, size_t data_size, array &out)
    +
    mlx::core::Shape
    std::vector< ShapeElem > Shape
    Definition array.h:21
    +
    mlx::core::Strides
    std::vector< int64_t > Strides
    Definition array.h:22
    +
    mlx::core::prepare_slice
    std::tuple< int64_t, Strides > prepare_slice(const array &in, const Shape &start_indices, const Shape &strides)
    +
    +
    + + + + diff --git a/docs/build/html/cpu_2ternary_8h.html b/docs/build/html/cpu_2ternary_8h.html new file mode 100644 index 000000000..4214ae3c0 --- /dev/null +++ b/docs/build/html/cpu_2ternary_8h.html @@ -0,0 +1,147 @@ + + + + + + + +MLX: mlx/backend/cpu/ternary.h File Reference + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Namespaces | +Functions
    +
    ternary.h File Reference
    +
    +
    +
    #include "mlx/allocator.h"
    +#include "mlx/array.h"
    +#include "mlx/backend/common/ternary.h"
    +#include "mlx/backend/common/utils.h"
    +
    +

    Go to the source code of this file.

    + + + + + + +

    +Namespaces

    namespace  mlx
     
    namespace  mlx::core
     
    + + + + + + + + + + +

    +Functions

    template<typename T1, typename T2, typename T3, typename U, typename Op, int D>
    void mlx::core::ternary_op_dims (const T1 *a, const T2 *b, const T3 *c, U *out, Op op, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &c_strides, const Strides &out_strides, int axis)
     
    template<typename T1, typename T2, typename T3, typename U, typename Op>
    void mlx::core::ternary_op_dispatch_dims (const array &a, const array &b, const array &c, array &out, Op op)
     
    template<typename T1, typename T2, typename T3, typename U, typename Op>
    void mlx::core::ternary_op (const array &a, const array &b, const array &c, array &out, Op op)
     
    +
    +
    + + + + diff --git a/docs/build/html/cpu_2ternary_8h.js b/docs/build/html/cpu_2ternary_8h.js new file mode 100644 index 000000000..8a6129550 --- /dev/null +++ b/docs/build/html/cpu_2ternary_8h.js @@ -0,0 +1,6 @@ +var cpu_2ternary_8h = +[ + [ "mlx::core::ternary_op", "namespacemlx_1_1core.html#a9dcc3018702ee31c21c8652bdc2182b1", null ], + [ "mlx::core::ternary_op_dims", "namespacemlx_1_1core.html#a8096c7a688ac3f09cca69a3a85f7f157", null ], + [ "mlx::core::ternary_op_dispatch_dims", "namespacemlx_1_1core.html#ac1c085e305954247d042f5d8803cd85b", null ] +]; \ No newline at end of file diff --git a/docs/build/html/cpu_2ternary_8h_source.html b/docs/build/html/cpu_2ternary_8h_source.html new file mode 100644 index 000000000..4f1832834 --- /dev/null +++ b/docs/build/html/cpu_2ternary_8h_source.html @@ -0,0 +1,304 @@ + + + + + + + +MLX: mlx/backend/cpu/ternary.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    ternary.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2023 Apple Inc.
    +
    2
    +
    3#pragma once
    +
    4#include "mlx/allocator.h"
    +
    5#include "mlx/array.h"
    +
    6#include "mlx/backend/common/ternary.h"
    +
    7#include "mlx/backend/common/utils.h"
    +
    8
    +
    9namespace mlx::core {
    +
    10
    +
    11template <typename T1, typename T2, typename T3, typename U, typename Op, int D>
    +
    +
    12void ternary_op_dims(
    +
    13 const T1* a,
    +
    14 const T2* b,
    +
    15 const T3* c,
    +
    16 U* out,
    +
    17 Op op,
    +
    18 const Shape& shape,
    +
    19 const Strides& a_strides,
    +
    20 const Strides& b_strides,
    +
    21 const Strides& c_strides,
    +
    22 const Strides& out_strides,
    +
    23 int axis) {
    +
    24 auto stride_a = a_strides[axis];
    +
    25 auto stride_b = b_strides[axis];
    +
    26 auto stride_c = c_strides[axis];
    +
    27 auto stride_out = out_strides[axis];
    +
    28 auto N = shape[axis];
    +
    29
    +
    30 for (int i = 0; i < N; i++) {
    +
    31 if constexpr (D > 1) {
    +
    32 ternary_op_dims<T1, T2, T3, U, Op, D - 1>(
    +
    33 a,
    +
    34 b,
    +
    35 c,
    +
    36 out,
    +
    37 op,
    +
    38 shape,
    +
    39 a_strides,
    +
    40 b_strides,
    +
    41 c_strides,
    +
    42 out_strides,
    +
    43 axis + 1);
    +
    44 } else {
    +
    45 *out = op(*a, *b, *c);
    +
    46 }
    +
    47 a += stride_a;
    +
    48 b += stride_b;
    +
    49 c += stride_c;
    +
    50 out += stride_out;
    +
    51 }
    +
    52}
    +
    +
    53
    +
    54template <typename T1, typename T2, typename T3, typename U, typename Op>
    +
    +
    55void ternary_op_dispatch_dims(
    +
    56 const array& a,
    +
    57 const array& b,
    +
    58 const array& c,
    +
    59 array& out,
    +
    60 Op op) {
    +
    61 auto [shape, strides] = collapse_contiguous_dims(
    +
    62 a.shape(), {a.strides(), b.strides(), c.strides(), out.strides()});
    +
    63 const auto& a_strides = strides[0];
    +
    64 const auto& b_strides = strides[1];
    +
    65 const auto& c_strides = strides[2];
    +
    66 const auto& out_strides = strides[3];
    +
    67
    +
    68 const T1* a_ptr = a.data<T1>();
    +
    69 const T2* b_ptr = b.data<T2>();
    +
    70 const T3* c_ptr = c.data<T3>();
    +
    71 U* out_ptr = out.data<T3>();
    +
    72 int ndim = shape.size();
    +
    73 switch (ndim) {
    +
    74 case 1:
    +
    75 ternary_op_dims<T1, T2, T3, U, Op, 1>(
    +
    76 a_ptr,
    +
    77 b_ptr,
    +
    78 c_ptr,
    +
    79 out_ptr,
    +
    80 op,
    +
    81 shape,
    +
    82 a_strides,
    +
    83 b_strides,
    +
    84 c_strides,
    +
    85 out_strides,
    +
    86 0);
    +
    87 return;
    +
    88 case 2:
    +
    89 ternary_op_dims<T1, T2, T3, U, Op, 2>(
    +
    90 a_ptr,
    +
    91 b_ptr,
    +
    92 c_ptr,
    +
    93 out_ptr,
    +
    94 op,
    +
    95 shape,
    +
    96 a_strides,
    +
    97 b_strides,
    +
    98 c_strides,
    +
    99 out_strides,
    +
    100 0);
    +
    101 return;
    +
    102 }
    +
    103
    +
    104 ContiguousIterator a_it(shape, a_strides, ndim - 2);
    +
    105 ContiguousIterator b_it(shape, b_strides, ndim - 2);
    +
    106 ContiguousIterator c_it(shape, c_strides, ndim - 2);
    +
    107 auto stride = out_strides[ndim - 3];
    +
    108 for (size_t elem = 0; elem < a.size(); elem += stride) {
    +
    109 ternary_op_dims<T1, T2, T3, U, Op, 2>(
    +
    110 a_ptr + a_it.loc,
    +
    111 b_ptr + b_it.loc,
    +
    112 c_ptr + c_it.loc,
    +
    113 out_ptr + elem,
    +
    114 op,
    +
    115 shape,
    +
    116 a_strides,
    +
    117 b_strides,
    +
    118 c_strides,
    +
    119 out_strides,
    +
    120 ndim - 2);
    +
    121 a_it.step();
    +
    122 b_it.step();
    +
    123 c_it.step();
    +
    124 }
    +
    125}
    +
    +
    126
    +
    127template <typename T1, typename T2, typename T3, typename U, typename Op>
    +
    +
    128void ternary_op(
    +
    129 const array& a,
    +
    130 const array& b,
    +
    131 const array& c,
    +
    132 array& out,
    +
    133 Op op) {
    +
    134 TernaryOpType topt = get_ternary_op_type(a, b, c);
    +
    135 set_ternary_op_output_data(a, b, c, out, topt);
    +
    136
    +
    137 // The full computation is scalar-scalar-scalar so we call the base op once.
    +
    138 if (topt == TernaryOpType::ScalarScalarScalar) {
    +
    139 *(out.data<U>()) = op(*a.data<T1>(), *b.data<T2>(), *c.data<T3>());
    +
    140 } else if (topt == TernaryOpType::VectorVectorVector) {
    +
    141 const T1* a_ptr = a.data<T1>();
    +
    142 const T2* b_ptr = b.data<T2>();
    +
    143 const T3* c_ptr = c.data<T3>();
    +
    144 U* out_ptr = out.data<U>();
    +
    145 for (size_t i = 0; i < out.size(); ++i) {
    +
    146 *out_ptr = op(*a_ptr, *b_ptr, *c_ptr);
    +
    147 a_ptr++;
    +
    148 b_ptr++;
    +
    149 c_ptr++;
    +
    150 out_ptr++;
    +
    151 }
    +
    152 } else {
    +
    153 ternary_op_dispatch_dims<T1, T2, T3, U>(a, b, c, out, op);
    +
    154 }
    +
    155}
    +
    +
    156
    +
    157} // namespace mlx::core
    + + + +
    mlx::core::array
    Definition array.h:24
    +
    mlx::core::array::shape
    const Shape & shape() const
    The shape of the array as a vector of integers.
    Definition array.h:103
    +
    mlx::core::array::size
    size_t size() const
    The number of elements in the array.
    Definition array.h:88
    +
    mlx::core::array::data
    T * data()
    Definition array.h:342
    + +
    mlx::core
    Definition allocator.h:7
    +
    mlx::core::collapse_contiguous_dims
    std::tuple< Shape, std::vector< Strides > > collapse_contiguous_dims(const Shape &shape, const std::vector< Strides > &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())
    +
    mlx::core::get_ternary_op_type
    TernaryOpType get_ternary_op_type(const array &a, const array &b, const array &c)
    Definition ternary.h:18
    +
    mlx::core::Shape
    std::vector< ShapeElem > Shape
    Definition array.h:21
    +
    mlx::core::set_ternary_op_output_data
    void set_ternary_op_output_data(const array &a, const array &b, const array &c, array &out, TernaryOpType topt, bool donate_with_move=false)
    Definition ternary.h:34
    +
    mlx::core::Strides
    std::vector< int64_t > Strides
    Definition array.h:22
    +
    mlx::core::ternary_op_dims
    void ternary_op_dims(const T1 *a, const T2 *b, const T3 *c, U *out, Op op, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &c_strides, const Strides &out_strides, int axis)
    Definition ternary.h:12
    +
    mlx::core::ternary_op
    void ternary_op(const array &a, const array &b, const array &c, array &out, Op op)
    Definition ternary.h:128
    +
    mlx::core::ternary_op_dispatch_dims
    void ternary_op_dispatch_dims(const array &a, const array &b, const array &c, array &out, Op op)
    Definition ternary.h:55
    +
    mlx::core::TernaryOpType
    TernaryOpType
    Definition ternary.h:11
    +
    mlx::core::TernaryOpType::ScalarScalarScalar
    @ ScalarScalarScalar
    Definition ternary.h:12
    +
    mlx::core::TernaryOpType::VectorVectorVector
    @ VectorVectorVector
    Definition ternary.h:13
    +
    mlx::core::ContiguousIterator
    Definition utils.h:73
    +
    mlx::core::ContiguousIterator::loc
    int64_t loc
    Definition utils.h:126
    +
    mlx::core::ContiguousIterator::step
    void step()
    Definition utils.h:74
    +
    +
    + + + + diff --git a/docs/build/html/cpu_2unary_8h.html b/docs/build/html/cpu_2unary_8h.html new file mode 100644 index 000000000..8d5653fae --- /dev/null +++ b/docs/build/html/cpu_2unary_8h.html @@ -0,0 +1,153 @@ + + + + + + + +MLX: mlx/backend/cpu/unary.h File Reference + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Namespaces | +Functions
    +
    unary.h File Reference
    +
    +
    +
    #include "mlx/allocator.h"
    +#include "mlx/array.h"
    +#include "mlx/backend/common/utils.h"
    +#include "mlx/backend/cpu/simd/simd.h"
    +#include "mlx/utils.h"
    +
    +

    Go to the source code of this file.

    + + + + + + +

    +Namespaces

    namespace  mlx
     
    namespace  mlx::core
     
    + + + + + + + + + + + + + + + +

    +Functions

    void mlx::core::set_unary_output_data (const array &in, array &out)
     
    template<typename T, typename U = T, typename Op>
    void mlx::core::unary_op (const T *a, U *out, Op op, size_t shape, size_t stride)
     
    template<typename T, typename U = T, typename Op>
    void mlx::core::unary_op (const array &a, array &out, Op op)
     
    template<typename Op>
    void mlx::core::unary (const array &a, array &out, Op op)
     
    template<typename Op>
    void mlx::core::unary_fp (const array &a, array &out, Op op)
     
    +
    +
    + + + + diff --git a/docs/build/html/cpu_2unary_8h.js b/docs/build/html/cpu_2unary_8h.js new file mode 100644 index 000000000..a2741a1fd --- /dev/null +++ b/docs/build/html/cpu_2unary_8h.js @@ -0,0 +1,8 @@ +var cpu_2unary_8h = +[ + [ "mlx::core::set_unary_output_data", "namespacemlx_1_1core.html#a4c6a4241bfcdd7bbf30d0e521b79e5a3", null ], + [ "mlx::core::unary", "namespacemlx_1_1core.html#a6c8fdd03ef891d7f47804bf02e9a8507", null ], + [ "mlx::core::unary_fp", "namespacemlx_1_1core.html#a76a2cb4634f5fd6970a8c3b3753d7a4a", null ], + [ "mlx::core::unary_op", "namespacemlx_1_1core.html#ae20f207ad1ed3badc17cecf08f118b5e", null ], + [ "mlx::core::unary_op", "namespacemlx_1_1core.html#a27f00519f9756896734fd4d47fec0625", null ] +]; \ No newline at end of file diff --git a/docs/build/html/cpu_2unary_8h_source.html b/docs/build/html/cpu_2unary_8h_source.html new file mode 100644 index 000000000..3437a7e9b --- /dev/null +++ b/docs/build/html/cpu_2unary_8h_source.html @@ -0,0 +1,311 @@ + + + + + + + +MLX: mlx/backend/cpu/unary.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    unary.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2023 Apple Inc.
    +
    2
    +
    3#pragma once
    +
    4
    +
    5#include "mlx/allocator.h"
    +
    6#include "mlx/array.h"
    +
    7#include "mlx/backend/common/utils.h"
    +
    8#include "mlx/backend/cpu/simd/simd.h"
    +
    9#include "mlx/utils.h"
    +
    10
    +
    11namespace mlx::core {
    +
    12
    +
    +
    13void set_unary_output_data(const array& in, array& out) {
    +
    14 if (is_donatable(in, out)) {
    +
    15 out.copy_shared_buffer(in);
    +
    16 } else {
    +
    17 auto size = in.data_size();
    +
    18 out.set_data(
    +
    19 allocator::malloc_or_wait(size * out.itemsize()),
    +
    20 size,
    +
    21 in.strides(),
    +
    22 in.flags());
    +
    23 }
    +
    24}
    +
    +
    25
    +
    26template <typename T, typename U = T, typename Op>
    +
    +
    27void unary_op(const T* a, U* out, Op op, size_t shape, size_t stride) {
    +
    28 for (size_t i = 0; i < shape; i += 1) {
    +
    29 out[i] = op(*a);
    +
    30 a += stride;
    +
    31 }
    +
    32}
    +
    +
    33
    +
    34template <typename T, typename U = T, typename Op>
    +
    +
    35void unary_op(const array& a, array& out, Op op) {
    +
    36 const T* a_ptr = a.data<T>();
    +
    37 if (a.flags().contiguous) {
    +
    38 set_unary_output_data(a, out);
    +
    39 U* dst = out.data<U>();
    +
    40 constexpr int N = simd::max_size<T>;
    +
    41 size_t size = a.data_size();
    +
    42 while (size >= N) {
    +
    43 simd::store(dst, op(simd::load<T, N>(a_ptr)));
    +
    44 size -= N;
    +
    45 a_ptr += N;
    +
    46 dst += N;
    +
    47 }
    +
    48 while (size > 0) {
    +
    49 *dst = op(*a_ptr);
    +
    50 size--;
    +
    51 dst++;
    +
    52 a_ptr++;
    +
    53 }
    +
    54 } else {
    +
    55 out.set_data(allocator::malloc_or_wait(out.nbytes()));
    +
    56 U* dst = out.data<U>();
    +
    57 size_t shape = a.ndim() > 0 ? a.shape(-1) : 1;
    +
    58 size_t stride = a.ndim() > 0 ? a.strides(-1) : 1;
    +
    59 if (a.ndim() <= 1) {
    +
    60 unary_op(a_ptr, dst, op, shape, stride);
    +
    61 return;
    +
    62 }
    +
    63 ContiguousIterator it(a.shape(), a.strides(), a.ndim() - 1);
    +
    64 for (size_t elem = 0; elem < a.size(); elem += shape) {
    +
    65 unary_op(a_ptr + it.loc, dst + elem, op, shape, stride);
    +
    66 it.step();
    +
    67 }
    +
    68 }
    +
    69}
    +
    +
    70
    +
    71template <typename Op>
    +
    +
    72void unary(const array& a, array& out, Op op) {
    +
    73 switch (out.dtype()) {
    +
    74 case bool_:
    +
    75 unary_op<bool>(a, out, op);
    +
    76 break;
    +
    77 case uint8:
    +
    78 unary_op<uint8_t>(a, out, op);
    +
    79 break;
    +
    80 case uint16:
    +
    81 unary_op<uint16_t>(a, out, op);
    +
    82 break;
    +
    83 case uint32:
    +
    84 unary_op<uint32_t>(a, out, op);
    +
    85 break;
    +
    86 case uint64:
    +
    87 unary_op<uint64_t>(a, out, op);
    +
    88 break;
    +
    89 case int8:
    +
    90 unary_op<int8_t>(a, out, op);
    +
    91 break;
    +
    92 case int16:
    +
    93 unary_op<int16_t>(a, out, op);
    +
    94 break;
    +
    95 case int32:
    +
    96 unary_op<int32_t>(a, out, op);
    +
    97 break;
    +
    98 case int64:
    +
    99 unary_op<int64_t>(a, out, op);
    +
    100 break;
    +
    101 case float16:
    +
    102 unary_op<float16_t>(a, out, op);
    +
    103 break;
    +
    104 case float32:
    +
    105 unary_op<float>(a, out, op);
    +
    106 break;
    +
    107 case bfloat16:
    +
    108 unary_op<bfloat16_t>(a, out, op);
    +
    109 break;
    +
    110 case complex64:
    +
    111 unary_op<complex64_t>(a, out, op);
    +
    112 break;
    +
    113 }
    +
    114}
    +
    +
    115
    +
    116template <typename Op>
    +
    +
    117void unary_fp(const array& a, array& out, Op op) {
    +
    118 switch (out.dtype()) {
    +
    119 case bfloat16:
    +
    120 unary_op<bfloat16_t>(a, out, op);
    +
    121 break;
    +
    122 case float16:
    +
    123 unary_op<float16_t>(a, out, op);
    +
    124 break;
    +
    125 case float32:
    +
    126 unary_op<float>(a, out, op);
    +
    127 break;
    +
    128 case complex64:
    +
    129 unary_op<complex64_t>(a, out, op);
    +
    130 break;
    +
    131 default:
    +
    132 std::ostringstream err;
    +
    133 err << "[unary_fp] Does not support " << out.dtype();
    +
    134 throw std::runtime_error(err.str());
    +
    135 }
    +
    136}
    +
    +
    137
    +
    138} // namespace mlx::core
    + + + +
    mlx::core::array
    Definition array.h:24
    +
    mlx::core::array::flags
    const Flags & flags() const
    Get the Flags bit-field.
    Definition array.h:306
    +
    mlx::core::array::shape
    const Shape & shape() const
    The shape of the array as a vector of integers.
    Definition array.h:103
    +
    mlx::core::array::strides
    const Strides & strides() const
    The strides of the array.
    Definition array.h:117
    +
    mlx::core::array::nbytes
    size_t nbytes() const
    The number of bytes in the array.
    Definition array.h:93
    +
    mlx::core::array::ndim
    size_t ndim() const
    The number of dimensions of the array.
    Definition array.h:98
    +
    mlx::core::array::size
    size_t size() const
    The number of elements in the array.
    Definition array.h:88
    +
    mlx::core::array::data
    T * data()
    Definition array.h:342
    +
    mlx::core::array::copy_shared_buffer
    void copy_shared_buffer(const array &other, const Strides &strides, Flags flags, size_t data_size, size_t offset=0)
    +
    mlx::core::array::dtype
    Dtype dtype() const
    Get the arrays data type.
    Definition array.h:131
    +
    mlx::core::array::itemsize
    size_t itemsize() const
    The size of the array's datatype in bytes.
    Definition array.h:83
    +
    mlx::core::array::set_data
    void set_data(allocator::Buffer buffer, Deleter d=allocator::free)
    +
    mlx::core::array::data_size
    size_t data_size() const
    The size (in elements) of the underlying buffer the array points to.
    Definition array.h:320
    +
    mlx::core::allocator::malloc_or_wait
    Buffer malloc_or_wait(size_t size)
    +
    mlx::core::simd::load
    Simd< T, N > load(const T *x)
    Definition base_simd.h:27
    +
    mlx::core::simd::max_size
    static constexpr int max_size
    Definition base_simd.h:13
    +
    mlx::core::simd::store
    void store(T *dst, Simd< T, N > x)
    Definition base_simd.h:32
    +
    mlx::core
    Definition allocator.h:7
    +
    mlx::core::bool_
    constexpr Dtype bool_
    Definition dtype.h:67
    +
    mlx::core::uint64
    constexpr Dtype uint64
    Definition dtype.h:72
    +
    mlx::core::unary_op
    void unary_op(const T *a, U *out, Op op, size_t shape, size_t stride)
    Definition unary.h:27
    +
    mlx::core::uint16
    constexpr Dtype uint16
    Definition dtype.h:70
    +
    mlx::core::set_unary_output_data
    void set_unary_output_data(const array &in, array &out)
    Definition unary.h:13
    +
    mlx::core::bfloat16
    constexpr Dtype bfloat16
    Definition dtype.h:81
    +
    mlx::core::int32
    constexpr Dtype int32
    Definition dtype.h:76
    +
    mlx::core::float32
    constexpr Dtype float32
    Definition dtype.h:80
    +
    mlx::core::unary
    void unary(const array &a, array &out, Op op)
    Definition unary.h:72
    +
    mlx::core::int16
    constexpr Dtype int16
    Definition dtype.h:75
    +
    mlx::core::unary_fp
    void unary_fp(const array &a, array &out, Op op)
    Definition unary.h:117
    +
    mlx::core::int8
    constexpr Dtype int8
    Definition dtype.h:74
    +
    mlx::core::int64
    constexpr Dtype int64
    Definition dtype.h:77
    +
    mlx::core::uint8
    constexpr Dtype uint8
    Definition dtype.h:69
    +
    mlx::core::float16
    constexpr Dtype float16
    Definition dtype.h:79
    +
    mlx::core::uint32
    constexpr Dtype uint32
    Definition dtype.h:71
    +
    mlx::core::is_donatable
    bool is_donatable(const array &in, const array &out)
    Definition utils.h:155
    +
    mlx::core::complex64
    constexpr Dtype complex64
    Definition dtype.h:82
    + +
    mlx::core::ContiguousIterator
    Definition utils.h:73
    +
    mlx::core::ContiguousIterator::loc
    int64_t loc
    Definition utils.h:126
    +
    mlx::core::ContiguousIterator::step
    void step()
    Definition utils.h:74
    +
    mlx::core::array::Flags::contiguous
    bool contiguous
    Definition array.h:231
    + +
    +
    + + + + diff --git a/docs/build/html/backend_2common_2ops_8h.html b/docs/build/html/cpu_2unary__ops_8h.html similarity index 62% rename from docs/build/html/backend_2common_2ops_8h.html rename to docs/build/html/cpu_2unary__ops_8h.html index fbd4e3758..6abe68814 100644 --- a/docs/build/html/backend_2common_2ops_8h.html +++ b/docs/build/html/cpu_2unary__ops_8h.html @@ -3,9 +3,9 @@ - + -MLX: mlx/backend/common/ops.h File Reference +MLX: mlx/backend/cpu/unary_ops.h File Reference @@ -52,7 +52,7 @@
    - +
    @@ -105,20 +105,19 @@ $(function(){initNavTree('backend_2common_2ops_8h.html',''); initResizable(true)
    Classes | Namespaces | -Functions
    -
    ops.h File Reference
    +Macros
    +
    unary_ops.h File Reference
    #include <stdint.h>
    #include <cmath>
    #include <complex>
    +#include "mlx/backend/cpu/simd/simd.h"
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    - - @@ -131,8 +130,6 @@ Classes - - @@ -153,8 +150,6 @@ Classes - - @@ -167,20 +162,12 @@ Classes - - - - - - - - @@ -189,53 +176,15 @@ Classes - + - + - + - + - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - +

    Classes

    union  mlx::core::detail::IntOrFloat
     
    struct  mlx::core::detail::Abs
     
    struct  mlx::core::detail::ArcCos
     
    struct  mlx::core::detail::ArcTan
     
    struct  mlx::core::detail::ArcTan2
     
    struct  mlx::core::detail::ArcTanh
     
    struct  mlx::core::detail::Ceil
     
    struct  mlx::core::detail::Floor
     
    struct  mlx::core::detail::Imag
     
    struct  mlx::core::detail::Log
     
    struct  mlx::core::detail::Log2
     
    struct  mlx::core::detail::Negative
     
    struct  mlx::core::detail::Real
     
    struct  mlx::core::detail::Round
     
    struct  mlx::core::detail::Sigmoid
     
    struct  mlx::core::detail::Sign
     
    struct  mlx::core::detail::Sin
     
    struct  mlx::core::detail::Sinh
     
    struct  mlx::core::detail::Square
     
    struct  mlx::core::detail::Sqrt
     
    struct  mlx::core::detail::Rsqrt
     
    struct  mlx::core::detail::Tanh
     
    struct  mlx::core::detail::Add
    struct  mlx::core::detail::Imag
     
    struct  mlx::core::detail::Divide
    struct  mlx::core::detail::Real
     
    struct  mlx::core::detail::Remainder
    struct  mlx::core::detail::Sigmoid
     
    struct  mlx::core::detail::Equal
    struct  mlx::core::detail::Sign
     
    struct  mlx::core::detail::NaNEqual
     
    struct  mlx::core::detail::Greater
     
    struct  mlx::core::detail::GreaterEqual
     
    struct  mlx::core::detail::Less
     
    struct  mlx::core::detail::LessEqual
     
    struct  mlx::core::detail::Maximum
     
    struct  mlx::core::detail::Minimum
     
    struct  mlx::core::detail::LogAddExp
     
    struct  mlx::core::detail::Multiply
     
    struct  mlx::core::detail::NotEqual
     
    struct  mlx::core::detail::Power
     
    struct  mlx::core::detail::Subtract
     
    struct  mlx::core::detail::LogicalAnd
     
    struct  mlx::core::detail::LogicalOr
     
    struct  mlx::core::detail::Select
     
    struct  mlx::core::detail::BitwiseAnd
     
    struct  mlx::core::detail::BitwiseOr
     
    struct  mlx::core::detail::BitwiseXor
     
    struct  mlx::core::detail::LeftShift
     
    struct  mlx::core::detail::RightShift
    struct  mlx::core::detail::Square
     

    @@ -247,22 +196,70 @@ Namespaces

    namespace  mlx::core::detail
     
    - - - - - - - + + + + +

    -Functions

    float mlx::core::detail::fast_exp (float x)
     
    float mlx::core::detail::fast_erf (float a)
     
    float mlx::core::detail::fast_erfinv (float a)
     

    +Macros

    #define SINGLE()
     
    #define DEFAULT_OP(Op, op)
     
    +

    Macro Definition Documentation

    + +

    ◆ DEFAULT_OP

    + +
    +
    + + + + + + + + + + + +
    #define DEFAULT_OP( Op,
    op )
    +
    +Value:
    struct Op { \
    +
    template <int N, typename T> \
    +
    Simd<T, N> operator()(Simd<T, N> x) { \
    +
    return simd::op(x); \
    +
    } \
    +
    SINGLE() \
    +
    };
    +
    +
    +
    + +

    ◆ SINGLE

    + +
    +
    + + + + + + + +
    #define SINGLE()
    +
    +Value:
    template <typename T> \
    +
    T operator()(T x) { \
    +
    return (*this)(Simd<T, 1>(x)).value; \
    +
    }
    +
    +
    +
    diff --git a/docs/build/html/backend_2common_2ops_8h.js b/docs/build/html/cpu_2unary__ops_8h.js similarity index 52% rename from docs/build/html/backend_2common_2ops_8h.js rename to docs/build/html/cpu_2unary__ops_8h.js index ce72fbd73..517b07cd4 100644 --- a/docs/build/html/backend_2common_2ops_8h.js +++ b/docs/build/html/cpu_2unary__ops_8h.js @@ -1,13 +1,11 @@ -var backend_2common_2ops_8h = +var cpu_2unary__ops_8h = [ - [ "mlx::core::detail::IntOrFloat", "unionmlx_1_1core_1_1detail_1_1_int_or_float.html", "unionmlx_1_1core_1_1detail_1_1_int_or_float" ], [ "mlx::core::detail::Abs", "structmlx_1_1core_1_1detail_1_1_abs.html", "structmlx_1_1core_1_1detail_1_1_abs" ], [ "mlx::core::detail::ArcCos", "structmlx_1_1core_1_1detail_1_1_arc_cos.html", "structmlx_1_1core_1_1detail_1_1_arc_cos" ], [ "mlx::core::detail::ArcCosh", "structmlx_1_1core_1_1detail_1_1_arc_cosh.html", "structmlx_1_1core_1_1detail_1_1_arc_cosh" ], [ "mlx::core::detail::ArcSin", "structmlx_1_1core_1_1detail_1_1_arc_sin.html", "structmlx_1_1core_1_1detail_1_1_arc_sin" ], [ "mlx::core::detail::ArcSinh", "structmlx_1_1core_1_1detail_1_1_arc_sinh.html", "structmlx_1_1core_1_1detail_1_1_arc_sinh" ], [ "mlx::core::detail::ArcTan", "structmlx_1_1core_1_1detail_1_1_arc_tan.html", "structmlx_1_1core_1_1detail_1_1_arc_tan" ], - [ "mlx::core::detail::ArcTan2", "structmlx_1_1core_1_1detail_1_1_arc_tan2.html", "structmlx_1_1core_1_1detail_1_1_arc_tan2" ], [ "mlx::core::detail::ArcTanh", "structmlx_1_1core_1_1detail_1_1_arc_tanh.html", "structmlx_1_1core_1_1detail_1_1_arc_tanh" ], [ "mlx::core::detail::Ceil", "structmlx_1_1core_1_1detail_1_1_ceil.html", "structmlx_1_1core_1_1detail_1_1_ceil" ], [ "mlx::core::detail::Conjugate", "structmlx_1_1core_1_1detail_1_1_conjugate.html", "structmlx_1_1core_1_1detail_1_1_conjugate" ], @@ -18,49 +16,24 @@ var backend_2common_2ops_8h = [ "mlx::core::detail::Exp", "structmlx_1_1core_1_1detail_1_1_exp.html", "structmlx_1_1core_1_1detail_1_1_exp" ], [ "mlx::core::detail::Expm1", "structmlx_1_1core_1_1detail_1_1_expm1.html", "structmlx_1_1core_1_1detail_1_1_expm1" ], [ "mlx::core::detail::Floor", "structmlx_1_1core_1_1detail_1_1_floor.html", "structmlx_1_1core_1_1detail_1_1_floor" ], - [ "mlx::core::detail::Imag", "structmlx_1_1core_1_1detail_1_1_imag.html", "structmlx_1_1core_1_1detail_1_1_imag" ], [ "mlx::core::detail::Log", "structmlx_1_1core_1_1detail_1_1_log.html", "structmlx_1_1core_1_1detail_1_1_log" ], [ "mlx::core::detail::Log2", "structmlx_1_1core_1_1detail_1_1_log2.html", "structmlx_1_1core_1_1detail_1_1_log2" ], [ "mlx::core::detail::Log10", "structmlx_1_1core_1_1detail_1_1_log10.html", "structmlx_1_1core_1_1detail_1_1_log10" ], [ "mlx::core::detail::Log1p", "structmlx_1_1core_1_1detail_1_1_log1p.html", "structmlx_1_1core_1_1detail_1_1_log1p" ], [ "mlx::core::detail::LogicalNot", "structmlx_1_1core_1_1detail_1_1_logical_not.html", "structmlx_1_1core_1_1detail_1_1_logical_not" ], [ "mlx::core::detail::Negative", "structmlx_1_1core_1_1detail_1_1_negative.html", "structmlx_1_1core_1_1detail_1_1_negative" ], - [ "mlx::core::detail::Real", "structmlx_1_1core_1_1detail_1_1_real.html", "structmlx_1_1core_1_1detail_1_1_real" ], [ "mlx::core::detail::Round", "structmlx_1_1core_1_1detail_1_1_round.html", "structmlx_1_1core_1_1detail_1_1_round" ], - [ "mlx::core::detail::Sigmoid", "structmlx_1_1core_1_1detail_1_1_sigmoid.html", "structmlx_1_1core_1_1detail_1_1_sigmoid" ], - [ "mlx::core::detail::Sign", "structmlx_1_1core_1_1detail_1_1_sign.html", "structmlx_1_1core_1_1detail_1_1_sign" ], [ "mlx::core::detail::Sin", "structmlx_1_1core_1_1detail_1_1_sin.html", "structmlx_1_1core_1_1detail_1_1_sin" ], [ "mlx::core::detail::Sinh", "structmlx_1_1core_1_1detail_1_1_sinh.html", "structmlx_1_1core_1_1detail_1_1_sinh" ], - [ "mlx::core::detail::Square", "structmlx_1_1core_1_1detail_1_1_square.html", "structmlx_1_1core_1_1detail_1_1_square" ], [ "mlx::core::detail::Sqrt", "structmlx_1_1core_1_1detail_1_1_sqrt.html", "structmlx_1_1core_1_1detail_1_1_sqrt" ], [ "mlx::core::detail::Rsqrt", "structmlx_1_1core_1_1detail_1_1_rsqrt.html", "structmlx_1_1core_1_1detail_1_1_rsqrt" ], [ "mlx::core::detail::Tan", "structmlx_1_1core_1_1detail_1_1_tan.html", "structmlx_1_1core_1_1detail_1_1_tan" ], [ "mlx::core::detail::Tanh", "structmlx_1_1core_1_1detail_1_1_tanh.html", "structmlx_1_1core_1_1detail_1_1_tanh" ], - [ "mlx::core::detail::Add", "structmlx_1_1core_1_1detail_1_1_add.html", "structmlx_1_1core_1_1detail_1_1_add" ], - [ "mlx::core::detail::Divide", "structmlx_1_1core_1_1detail_1_1_divide.html", "structmlx_1_1core_1_1detail_1_1_divide" ], - [ "mlx::core::detail::Remainder", "structmlx_1_1core_1_1detail_1_1_remainder.html", "structmlx_1_1core_1_1detail_1_1_remainder" ], - [ "mlx::core::detail::Equal", "structmlx_1_1core_1_1detail_1_1_equal.html", "structmlx_1_1core_1_1detail_1_1_equal" ], - [ "mlx::core::detail::NaNEqual", "structmlx_1_1core_1_1detail_1_1_na_n_equal.html", "structmlx_1_1core_1_1detail_1_1_na_n_equal" ], - [ "mlx::core::detail::Greater", "structmlx_1_1core_1_1detail_1_1_greater.html", "structmlx_1_1core_1_1detail_1_1_greater" ], - [ "mlx::core::detail::GreaterEqual", "structmlx_1_1core_1_1detail_1_1_greater_equal.html", "structmlx_1_1core_1_1detail_1_1_greater_equal" ], - [ "mlx::core::detail::Less", "structmlx_1_1core_1_1detail_1_1_less.html", "structmlx_1_1core_1_1detail_1_1_less" ], - [ "mlx::core::detail::LessEqual", "structmlx_1_1core_1_1detail_1_1_less_equal.html", "structmlx_1_1core_1_1detail_1_1_less_equal" ], - [ "mlx::core::detail::Maximum", "structmlx_1_1core_1_1detail_1_1_maximum.html", "structmlx_1_1core_1_1detail_1_1_maximum" ], - [ "mlx::core::detail::Minimum", "structmlx_1_1core_1_1detail_1_1_minimum.html", "structmlx_1_1core_1_1detail_1_1_minimum" ], - [ "mlx::core::detail::LogAddExp", "structmlx_1_1core_1_1detail_1_1_log_add_exp.html", "structmlx_1_1core_1_1detail_1_1_log_add_exp" ], - [ "mlx::core::detail::Multiply", "structmlx_1_1core_1_1detail_1_1_multiply.html", "structmlx_1_1core_1_1detail_1_1_multiply" ], - [ "mlx::core::detail::NotEqual", "structmlx_1_1core_1_1detail_1_1_not_equal.html", "structmlx_1_1core_1_1detail_1_1_not_equal" ], - [ "mlx::core::detail::Power", "structmlx_1_1core_1_1detail_1_1_power.html", "structmlx_1_1core_1_1detail_1_1_power" ], - [ "mlx::core::detail::Subtract", "structmlx_1_1core_1_1detail_1_1_subtract.html", "structmlx_1_1core_1_1detail_1_1_subtract" ], - [ "mlx::core::detail::LogicalAnd", "structmlx_1_1core_1_1detail_1_1_logical_and.html", "structmlx_1_1core_1_1detail_1_1_logical_and" ], - [ "mlx::core::detail::LogicalOr", "structmlx_1_1core_1_1detail_1_1_logical_or.html", "structmlx_1_1core_1_1detail_1_1_logical_or" ], - [ "mlx::core::detail::Select", "structmlx_1_1core_1_1detail_1_1_select.html", "structmlx_1_1core_1_1detail_1_1_select" ], - [ "mlx::core::detail::BitwiseAnd", "structmlx_1_1core_1_1detail_1_1_bitwise_and.html", "structmlx_1_1core_1_1detail_1_1_bitwise_and" ], - [ "mlx::core::detail::BitwiseOr", "structmlx_1_1core_1_1detail_1_1_bitwise_or.html", "structmlx_1_1core_1_1detail_1_1_bitwise_or" ], - [ "mlx::core::detail::BitwiseXor", "structmlx_1_1core_1_1detail_1_1_bitwise_xor.html", "structmlx_1_1core_1_1detail_1_1_bitwise_xor" ], - [ "mlx::core::detail::LeftShift", "structmlx_1_1core_1_1detail_1_1_left_shift.html", "structmlx_1_1core_1_1detail_1_1_left_shift" ], - [ "mlx::core::detail::RightShift", "structmlx_1_1core_1_1detail_1_1_right_shift.html", "structmlx_1_1core_1_1detail_1_1_right_shift" ], - [ "mlx::core::detail::fast_erf", "namespacemlx_1_1core_1_1detail.html#a90c9f6149af5adf4e2a95608d5f7b790", null ], - [ "mlx::core::detail::fast_erfinv", "namespacemlx_1_1core_1_1detail.html#ad0ff3975e4c96317df1a2de0f7a30c1d", null ], - [ "mlx::core::detail::fast_exp", "namespacemlx_1_1core_1_1detail.html#a2726436fc72d4a3f0030c89579b4d374", null ] + [ "mlx::core::detail::Imag", "structmlx_1_1core_1_1detail_1_1_imag.html", "structmlx_1_1core_1_1detail_1_1_imag" ], + [ "mlx::core::detail::Real", "structmlx_1_1core_1_1detail_1_1_real.html", "structmlx_1_1core_1_1detail_1_1_real" ], + [ "mlx::core::detail::Sigmoid", "structmlx_1_1core_1_1detail_1_1_sigmoid.html", "structmlx_1_1core_1_1detail_1_1_sigmoid" ], + [ "mlx::core::detail::Sign", "structmlx_1_1core_1_1detail_1_1_sign.html", "structmlx_1_1core_1_1detail_1_1_sign" ], + [ "mlx::core::detail::Square", "structmlx_1_1core_1_1detail_1_1_square.html", "structmlx_1_1core_1_1detail_1_1_square" ], + [ "DEFAULT_OP", "cpu_2unary__ops_8h.html#a83702f31e6dbd79c339a6aad67319f64", null ], + [ "SINGLE", "cpu_2unary__ops_8h.html#a602aea95990389a45c255195f849d5de", null ] ]; \ No newline at end of file diff --git a/docs/build/html/cpu_2unary__ops_8h_source.html b/docs/build/html/cpu_2unary__ops_8h_source.html new file mode 100644 index 000000000..0a2168a9a --- /dev/null +++ b/docs/build/html/cpu_2unary__ops_8h_source.html @@ -0,0 +1,324 @@ + + + + + + + +MLX: mlx/backend/cpu/unary_ops.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    unary_ops.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    +
    2
    +
    3#pragma once
    +
    4
    +
    5#include <stdint.h>
    +
    6#include <cmath>
    +
    7#include <complex>
    +
    8
    +
    9#include "mlx/backend/cpu/simd/simd.h"
    +
    10
    +
    11namespace mlx::core::detail {
    +
    12
    +
    13using namespace mlx::core::simd;
    +
    14
    +
    +
    15#define SINGLE() \
    +
    16 template <typename T> \
    +
    17 T operator()(T x) { \
    +
    18 return (*this)(Simd<T, 1>(x)).value; \
    +
    19 }
    +
    +
    20
    +
    +
    21#define DEFAULT_OP(Op, op) \
    +
    22 struct Op { \
    +
    23 template <int N, typename T> \
    +
    24 Simd<T, N> operator()(Simd<T, N> x) { \
    +
    25 return simd::op(x); \
    +
    26 } \
    +
    27 SINGLE() \
    +
    28 };
    +
    +
    29
    +
    30DEFAULT_OP(Abs, abs)
    +
    31DEFAULT_OP(ArcCos, acos)
    +
    32DEFAULT_OP(ArcCosh, acosh)
    +
    33DEFAULT_OP(ArcSin, asin)
    +
    34DEFAULT_OP(ArcSinh, asinh)
    +
    35DEFAULT_OP(ArcTan, atan)
    +
    36DEFAULT_OP(ArcTanh, atanh)
    +
    37DEFAULT_OP(Ceil, ceil)
    +
    38DEFAULT_OP(Conjugate, conj)
    +
    39DEFAULT_OP(Cos, cos)
    +
    40DEFAULT_OP(Cosh, cosh)
    +
    41DEFAULT_OP(Erf, erf)
    +
    42DEFAULT_OP(ErfInv, erfinv)
    +
    43DEFAULT_OP(Exp, exp)
    +
    44DEFAULT_OP(Expm1, expm1)
    +
    45DEFAULT_OP(Floor, floor);
    +
    46DEFAULT_OP(Log, log);
    +
    47DEFAULT_OP(Log2, log2);
    +
    48DEFAULT_OP(Log10, log10);
    +
    49DEFAULT_OP(Log1p, log1p);
    +
    50DEFAULT_OP(LogicalNot, operator!)
    +
    51DEFAULT_OP(Negative, operator-)
    +
    52DEFAULT_OP(Round, rint);
    +
    53DEFAULT_OP(Sin, sin)
    +
    54DEFAULT_OP(Sinh, sinh)
    +
    55DEFAULT_OP(Sqrt, sqrt)
    +
    56DEFAULT_OP(Rsqrt, rsqrt)
    +
    57DEFAULT_OP(Tan, tan)
    +
    58DEFAULT_OP(Tanh, tanh)
    +
    59
    +
    +
    60struct Imag {
    +
    61 template <int N>
    + +
    65 SINGLE()
    +
    66};
    +
    +
    67
    +
    +
    68struct Real {
    +
    69 template <int N>
    + +
    73 SINGLE()
    +
    74};
    +
    +
    75
    +
    +
    76struct Sigmoid {
    +
    77 template <int N, typename T>
    +
    +
    78 Simd<T, N> operator()(Simd<T, N> x) {
    +
    79 return 1.0f / (1.0f + simd::exp(-x));
    +
    80 }
    +
    +
    81 SINGLE()
    +
    82};
    +
    +
    83
    +
    +
    84struct Sign {
    +
    85 template <int N, typename T>
    +
    +
    86 Simd<T, N> operator()(Simd<T, N> x) {
    +
    87 auto z = Simd<T, N>{0};
    +
    88 if constexpr (std::is_unsigned_v<T>) {
    +
    89 return x != z;
    +
    90 } else if constexpr (std::is_same_v<T, complex64_t>) {
    +
    91 return simd::select(x == z, x, Simd<T, N>(x / simd::abs(x)));
    +
    92 } else {
    +
    93 return simd::select(
    +
    94 x < z, Simd<T, N>{-1}, simd::select(x > z, Simd<T, N>{1}, z));
    +
    95 }
    +
    96 }
    +
    +
    97 SINGLE()
    +
    98};
    +
    +
    99
    +
    +
    100struct Square {
    +
    101 template <int N, typename T>
    +
    +
    102 Simd<T, N> operator()(Simd<T, N> x) {
    +
    103 return x * x;
    +
    104 }
    +
    +
    105 SINGLE()
    +
    106};
    +
    +
    107
    +
    108} // namespace mlx::core::detail
    +
    SINGLE
    #define SINGLE()
    Definition unary_ops.h:15
    +
    DEFAULT_OP
    #define DEFAULT_OP(Op, op)
    Definition unary_ops.h:21
    +
    mlx::core::detail
    Definition binary_ops.h:7
    +
    mlx::core::simd::sinh
    Simd< float16_t, N > sinh(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:41
    +
    mlx::core::simd::atanh
    Simd< float16_t, N > atanh(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:34
    +
    mlx::core::simd::log10
    Simd< float16_t, N > log10(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:39
    +
    mlx::core::simd::rint
    Simd< T, N > rint(Simd< T, N > v)
    Definition accelerate_simd.h:127
    +
    mlx::core::simd::tan
    Simd< float16_t, N > tan(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:42
    +
    mlx::core::simd::abs
    Simd< T, N > abs(Simd< T, N > v)
    Definition accelerate_simd.h:112
    +
    mlx::core::simd::acosh
    Simd< float16_t, N > acosh(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:30
    +
    mlx::core::simd::erf
    Simd< T, N > erf(Simd< T, N > x)
    Definition math.h:137
    +
    mlx::core::simd::conj
    Simd< T, 1 > conj(Simd< T, 1 > in)
    Definition base_simd.h:84
    +
    mlx::core::simd::log2
    Simd< float16_t, N > log2(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:38
    +
    mlx::core::simd::erfinv
    Simd< T, N > erfinv(Simd< T, N > a_)
    Definition math.h:151
    +
    mlx::core::simd::exp
    Simd< T, N > exp(Simd< T, N > in)
    Compute exp(x) in an optimizer friendly way as follows:
    Definition math.h:28
    +
    mlx::core::simd::log
    Simd< float16_t, N > log(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:37
    +
    mlx::core::simd::floor
    Simd< T, N > floor(Simd< T, N > v)
    Definition accelerate_simd.h:113
    +
    mlx::core::simd::expm1
    Simd< float16_t, N > expm1(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:36
    +
    mlx::core::simd::imag
    auto imag(Simd< T, 1 > in) -> Simd< decltype(std::imag(in.value)), 1 >
    Definition base_simd.h:102
    +
    mlx::core::simd::asin
    Simd< float16_t, N > asin(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:31
    +
    mlx::core::simd::tanh
    Simd< float16_t, N > tanh(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:43
    +
    mlx::core::simd::atan
    Simd< float16_t, N > atan(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:33
    +
    mlx::core::simd::asinh
    Simd< float16_t, N > asinh(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:32
    +
    mlx::core::simd::cos
    Simd< T, N > cos(Simd< T, N > x)
    Definition math.h:128
    +
    mlx::core::simd::sin
    Simd< T, N > sin(Simd< T, N > x)
    Definition math.h:119
    +
    mlx::core::simd::real
    auto real(Simd< T, 1 > in) -> Simd< decltype(std::real(in.value)), 1 >
    Definition base_simd.h:98
    +
    mlx::core::simd::log1p
    Simd< float16_t, N > log1p(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:40
    +
    mlx::core::simd::ceil
    Simd< T, N > ceil(Simd< T, N > v)
    Definition accelerate_simd.h:120
    +
    mlx::core::simd::sqrt
    Simd< T, N > sqrt(Simd< T, N > v)
    Definition accelerate_simd.h:129
    +
    mlx::core::simd::acos
    Simd< float16_t, N > acos(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:29
    +
    mlx::core::simd::rsqrt
    Simd< T, N > rsqrt(Simd< T, N > v)
    Definition accelerate_simd.h:130
    +
    mlx::core::simd::cosh
    Simd< float16_t, N > cosh(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:35
    +
    mlx::core::simd::select
    Simd< T1, N > select(Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)
    Definition accelerate_simd.h:231
    + +
    mlx::core::detail::Abs
    Definition unary_ops.h:30
    +
    mlx::core::detail::ArcCos
    Definition unary_ops.h:31
    +
    mlx::core::detail::ArcCosh
    Definition unary_ops.h:32
    +
    mlx::core::detail::ArcSin
    Definition unary_ops.h:33
    +
    mlx::core::detail::ArcSinh
    Definition unary_ops.h:34
    +
    mlx::core::detail::ArcTan
    Definition unary_ops.h:35
    +
    mlx::core::detail::ArcTanh
    Definition unary_ops.h:36
    +
    mlx::core::detail::Ceil
    Definition unary_ops.h:37
    +
    mlx::core::detail::Conjugate
    Definition unary_ops.h:38
    +
    mlx::core::detail::Cos
    Definition unary_ops.h:39
    +
    mlx::core::detail::Cosh
    Definition unary_ops.h:40
    +
    mlx::core::detail::Erf
    Definition unary_ops.h:41
    +
    mlx::core::detail::ErfInv
    Definition unary_ops.h:42
    +
    mlx::core::detail::Exp
    Definition unary_ops.h:43
    +
    mlx::core::detail::Expm1
    Definition unary_ops.h:44
    +
    mlx::core::detail::Floor
    Definition unary_ops.h:45
    +
    mlx::core::detail::Imag
    Definition unary_ops.h:60
    +
    mlx::core::detail::Imag::operator()
    Simd< float, N > operator()(Simd< complex64_t, N > x)
    Definition unary_ops.h:62
    +
    mlx::core::detail::Log10
    Definition unary_ops.h:48
    +
    mlx::core::detail::Log1p
    Definition unary_ops.h:49
    +
    mlx::core::detail::Log2
    Definition unary_ops.h:47
    +
    mlx::core::detail::Log
    Definition unary_ops.h:46
    +
    mlx::core::detail::LogicalNot
    Definition unary_ops.h:50
    +
    mlx::core::detail::Negative
    Definition unary_ops.h:51
    +
    mlx::core::detail::Real
    Definition unary_ops.h:68
    +
    mlx::core::detail::Real::operator()
    Simd< float, N > operator()(Simd< complex64_t, N > x)
    Definition unary_ops.h:70
    +
    mlx::core::detail::Round
    Definition unary_ops.h:52
    +
    mlx::core::detail::Rsqrt
    Definition unary_ops.h:56
    +
    mlx::core::detail::Sigmoid
    Definition unary_ops.h:76
    +
    mlx::core::detail::Sigmoid::operator()
    Simd< T, N > operator()(Simd< T, N > x)
    Definition unary_ops.h:78
    +
    mlx::core::detail::Sign
    Definition unary_ops.h:84
    +
    mlx::core::detail::Sign::operator()
    Simd< T, N > operator()(Simd< T, N > x)
    Definition unary_ops.h:86
    +
    mlx::core::detail::Sin
    Definition unary_ops.h:53
    +
    mlx::core::detail::Sinh
    Definition unary_ops.h:54
    +
    mlx::core::detail::Sqrt
    Definition unary_ops.h:55
    +
    mlx::core::detail::Square
    Definition unary_ops.h:100
    +
    mlx::core::detail::Square::operator()
    Simd< T, N > operator()(Simd< T, N > x)
    Definition unary_ops.h:102
    +
    mlx::core::detail::Tan
    Definition unary_ops.h:57
    +
    mlx::core::detail::Tanh
    Definition unary_ops.h:58
    +
    mlx::core::simd::Simd
    Definition accelerate_simd.h:55
    +
    +
    + + + + diff --git a/docs/build/html/defines_8h.html b/docs/build/html/defines_8h.html index 4240aa4d7..3ca57decc 100644 --- a/docs/build/html/defines_8h.html +++ b/docs/build/html/defines_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/defines.h File Reference @@ -52,7 +52,7 @@ - + + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/dev/extensions.html b/docs/build/html/dev/extensions.html index b179c3d05..f9270149d 100644 --- a/docs/build/html/dev/extensions.html +++ b/docs/build/html/dev/extensions.html @@ -8,7 +8,7 @@ - Custom Extensions in MLX — MLX 0.22.0 documentation + Custom Extensions in MLX — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/dev/metal_debugger.html b/docs/build/html/dev/metal_debugger.html index b54921bbe..8eee1ab87 100644 --- a/docs/build/html/dev/metal_debugger.html +++ b/docs/build/html/dev/metal_debugger.html @@ -8,7 +8,7 @@ - Metal Debugger — MLX 0.22.0 documentation + Metal Debugger — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/dev/mlx_in_cpp.html b/docs/build/html/dev/mlx_in_cpp.html index a9c6f9e58..4f56324fe 100644 --- a/docs/build/html/dev/mlx_in_cpp.html +++ b/docs/build/html/dev/mlx_in_cpp.html @@ -8,7 +8,7 @@ - Using MLX in C++ — MLX 0.22.0 documentation + Using MLX in C++ — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -50,7 +50,7 @@ - + @@ -129,8 +129,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/device_8h.html b/docs/build/html/device_8h.html index 3a46626d0..28de1fedd 100644 --- a/docs/build/html/device_8h.html +++ b/docs/build/html/device_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/device.h File Reference @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - +
    @@ -102,13 +102,13 @@ $(function(){initNavTree('dir_86b95e7b1d0d6e25466bb9213752d32f.html',''); initRe
    -
    accelerate Directory Reference
    +
    ring Directory Reference
    - +

    Files

     utils.h
     ring.h
     
    @@ -116,8 +116,8 @@ Files diff --git a/docs/build/html/dir_3afd476a29b32961d09b8b179155b807.js b/docs/build/html/dir_3afd476a29b32961d09b8b179155b807.js new file mode 100644 index 000000000..76d87e8dd --- /dev/null +++ b/docs/build/html/dir_3afd476a29b32961d09b8b179155b807.js @@ -0,0 +1,4 @@ +var dir_3afd476a29b32961d09b8b179155b807 = +[ + [ "ring.h", "ring_8h.html", "ring_8h" ] +]; \ No newline at end of file diff --git a/docs/build/html/dir_4336740ec0075891704443b417fef6cb.html b/docs/build/html/dir_4336740ec0075891704443b417fef6cb.html index 701d9ce0e..de360ad55 100644 --- a/docs/build/html/dir_4336740ec0075891704443b417fef6cb.html +++ b/docs/build/html/dir_4336740ec0075891704443b417fef6cb.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/jit Directory Reference @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    cpu Directory Reference
    +
    +
    + + + + +

    +Directories

     simd
     
    + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

    +Files

     arange.h
     
     binary.h
     
     binary_ops.h
     
     binary_two.h
     
     compiled_preamble.h
     
     copy.h
     
     gemm.h
     
     jit_compiler.h
     
     lapack.h
     
     slicing.h
     
     ternary.h
     
     threefry.h
     
     unary.h
     
     unary_ops.h
     
    +
    +
    + + + + diff --git a/docs/build/html/dir_48c8bf40aae7e42226b4fe31ea48af19.js b/docs/build/html/dir_48c8bf40aae7e42226b4fe31ea48af19.js new file mode 100644 index 000000000..1adc2785b --- /dev/null +++ b/docs/build/html/dir_48c8bf40aae7e42226b4fe31ea48af19.js @@ -0,0 +1,18 @@ +var dir_48c8bf40aae7e42226b4fe31ea48af19 = +[ + [ "simd", "dir_777905fddc177f731a39846ae16b0314.html", "dir_777905fddc177f731a39846ae16b0314" ], + [ "arange.h", "cpu_2arange_8h.html", "cpu_2arange_8h" ], + [ "binary.h", "cpu_2binary_8h.html", "cpu_2binary_8h" ], + [ "binary_ops.h", "cpu_2binary__ops_8h.html", "cpu_2binary__ops_8h" ], + [ "binary_two.h", "cpu_2binary__two_8h.html", null ], + [ "compiled_preamble.h", "compiled__preamble_8h.html", "compiled__preamble_8h" ], + [ "copy.h", "cpu_2copy_8h.html", "cpu_2copy_8h" ], + [ "gemm.h", "cpu_2gemm_8h.html", "cpu_2gemm_8h" ], + [ "jit_compiler.h", "jit__compiler_8h.html", "jit__compiler_8h" ], + [ "lapack.h", "lapack_8h.html", "lapack_8h" ], + [ "slicing.h", "cpu_2slicing_8h.html", "cpu_2slicing_8h" ], + [ "ternary.h", "cpu_2ternary_8h.html", "cpu_2ternary_8h" ], + [ "threefry.h", "threefry_8h.html", "threefry_8h" ], + [ "unary.h", "cpu_2unary_8h.html", "cpu_2unary_8h" ], + [ "unary_ops.h", "cpu_2unary__ops_8h.html", "cpu_2unary__ops_8h" ] +]; \ No newline at end of file diff --git a/docs/build/html/dir_5aea41cce495e77a0857a0aecf063e33.html b/docs/build/html/dir_5aea41cce495e77a0857a0aecf063e33.html index fc2e82639..0cf32c80a 100644 --- a/docs/build/html/dir_5aea41cce495e77a0857a0aecf063e33.html +++ b/docs/build/html/dir_5aea41cce495e77a0857a0aecf063e33.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/steel/attn/kernels Directory Reference @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    simd Directory Reference
    +
    +
    + + + + + + + + + + + + + + + + +

    +Files

     accelerate_fp16_simd.h
     
     accelerate_simd.h
     
     base_simd.h
     
     math.h
     
     neon_fp16_simd.h
     
     simd.h
     
     type.h
     
    +
    +
    + + + + diff --git a/docs/build/html/dir_777905fddc177f731a39846ae16b0314.js b/docs/build/html/dir_777905fddc177f731a39846ae16b0314.js new file mode 100644 index 000000000..7f65f2ad9 --- /dev/null +++ b/docs/build/html/dir_777905fddc177f731a39846ae16b0314.js @@ -0,0 +1,10 @@ +var dir_777905fddc177f731a39846ae16b0314 = +[ + [ "accelerate_fp16_simd.h", "accelerate__fp16__simd_8h.html", "accelerate__fp16__simd_8h" ], + [ "accelerate_simd.h", "accelerate__simd_8h.html", "accelerate__simd_8h" ], + [ "base_simd.h", "base__simd_8h.html", "base__simd_8h" ], + [ "math.h", "math_8h.html", "math_8h" ], + [ "neon_fp16_simd.h", "neon__fp16__simd_8h.html", "neon__fp16__simd_8h" ], + [ "simd.h", "simd_8h.html", null ], + [ "type.h", "type_8h.html", null ] +]; \ No newline at end of file diff --git a/docs/build/html/dir_83367edb60e23ad59b1a493d8c883287.html b/docs/build/html/dir_83367edb60e23ad59b1a493d8c883287.html index 164095ae5..c97c05bfa 100644 --- a/docs/build/html/dir_83367edb60e23ad59b1a493d8c883287.html +++ b/docs/build/html/dir_83367edb60e23ad59b1a493d8c883287.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/metal_3_1 Directory Reference @@ -52,7 +52,7 @@ - + + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/examples/llama-inference.html b/docs/build/html/examples/llama-inference.html index 206a26fed..4facb9856 100644 --- a/docs/build/html/examples/llama-inference.html +++ b/docs/build/html/examples/llama-inference.html @@ -8,7 +8,7 @@ - LLM inference — MLX 0.22.0 documentation + LLM inference — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/examples/mlp.html b/docs/build/html/examples/mlp.html index fc85efa0f..12c573fcd 100644 --- a/docs/build/html/examples/mlp.html +++ b/docs/build/html/examples/mlp.html @@ -8,7 +8,7 @@ - Multi-Layer Perceptron — MLX 0.22.0 documentation + Multi-Layer Perceptron — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/expm1f_8h.html b/docs/build/html/expm1f_8h.html index c51169b18..08170f3ea 100644 --- a/docs/build/html/expm1f_8h.html +++ b/docs/build/html/expm1f_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/expm1f.h File Reference @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - +
    @@ -103,17 +103,20 @@ $(function(){initNavTree('common_2unary_8h.html',''); initResizable(true); });
    +Classes | Namespaces
    -
    unary.h File Reference
    +
    fence.h File Reference
    -
    #include "mlx/allocator.h"
    -#include "mlx/array.h"
    -#include "mlx/backend/common/utils.h"
    -#include "mlx/utils.h"
    +
    #include "mlx/backend/metal/device.h"
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    + + + +

    +Classes

    class  mlx::core::Fence
     
    @@ -126,8 +129,8 @@ Namespaces diff --git a/docs/build/html/fence_8h.js b/docs/build/html/fence_8h.js new file mode 100644 index 000000000..f5283bb58 --- /dev/null +++ b/docs/build/html/fence_8h.js @@ -0,0 +1,4 @@ +var fence_8h = +[ + [ "mlx::core::Fence", "classmlx_1_1core_1_1_fence.html", "classmlx_1_1core_1_1_fence" ] +]; \ No newline at end of file diff --git a/docs/build/html/fence_8h_source.html b/docs/build/html/fence_8h_source.html new file mode 100644 index 000000000..45ecd60c3 --- /dev/null +++ b/docs/build/html/fence_8h_source.html @@ -0,0 +1,169 @@ + + + + + + + +MLX: mlx/backend/metal/fence.h Source File + + + + + + + + + + + + + + + + +
    +
    +

    Namespaces

    namespace  mlx
    + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    fence.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    +
    2
    +
    3#include "mlx/backend/metal/device.h"
    +
    4
    +
    5namespace mlx::core {
    +
    6
    +
    7/* A fence to be used for synchronizing work between the CPU and GPU
    +
    8 *
    +
    9 * Calls to `update_gpu` should be paired with calls to `wait`. This ensures
    +
    10 * that the array passed to `update_gpu` is computed and visible to the CPU
    +
    11 * after the call to `wait` returns.
    +
    12 *
    +
    13 * Calls to `update` should be paired with calls to `wait_gpu`. This ensures
    +
    14 * that the array passed to `wait_gpu` will not be read by the GPU until the CPU
    +
    15 * has called `update`.
    +
    16 *
    +
    17 * The fence supports slow (default) and fast mode. Fast mode requires setting
    +
    18 * the environment variable `MLX_METAL_FAST_SYNCH=1`. Fast mode also requires
    +
    19 * Metal 3.2+ (macOS 15+, iOS 18+).
    +
    20 */
    +
    +
    21class Fence {
    +
    22 public:
    +
    23 Fence(const Stream& stream);
    +
    24
    +
    25 void update_gpu(const array& x);
    +
    26 void wait_gpu(array& x);
    +
    27
    +
    28 void wait();
    +
    29 void update();
    +
    30
    +
    31 private:
    +
    32 Stream stream_;
    +
    33 std::shared_ptr<void> fence_;
    +
    34 uint32_t cpu_count_{0};
    +
    35 uint32_t gpu_count_{0};
    +
    36 bool use_fast_;
    +
    37 std::atomic_uint* cpu_value();
    +
    38};
    +
    +
    39
    +
    40} // namespace mlx::core
    + + +
    mlx::core::Fence::Fence
    Fence(const Stream &stream)
    + +
    mlx::core::Fence::update_gpu
    void update_gpu(const array &x)
    +
    mlx::core::Fence::wait_gpu
    void wait_gpu(array &x)
    +
    mlx::core::array
    Definition array.h:24
    +
    mlx::core
    Definition allocator.h:7
    +
    mlx::core::Stream
    Definition stream.h:9
    +
    +
    + + + + diff --git a/docs/build/html/fft_8h.html b/docs/build/html/fft_8h.html index 74224c6e9..e0802cee7 100644 --- a/docs/build/html/fft_8h.html +++ b/docs/build/html/fft_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/fft.h File Reference @@ -52,7 +52,7 @@
    - + @@ -52,7 +52,7 @@ - +
    @@ -103,62 +103,96 @@ $(function(){initNavTree('jit_2gemv__masked_8h.html',''); initResizable(true); }
    -Variables
    -
    gemv_masked.h File Reference
    +Functions
    +
    gather_axis.h File Reference
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    - - - + + + +

    -Variables

    constexpr std::string_view gemv_masked_kernel
     

    +Functions

    template<typename T, typename IdxT, typename LocT, bool SrcC, bool IdxC>
    void gather_axis (const device T *src, const device IdxT *indices, device T *out, const constant int *shape, const constant int64_t *src_strides, const constant int64_t *idx_strides, const constant size_t &ndim, const constant int &axis, const constant int &axis_size, const constant size_t &src_ax_stride, const constant size_t &idx_ax_stride, uint3 index, uint3 grid_dim)
     
    -

    Variable Documentation

    - -

    ◆ gemv_masked_kernel

    +

    Function Documentation

    + +

    ◆ gather_axis()

    - - - - - -
    +
    +template<typename T, typename IdxT, typename LocT, bool SrcC, bool IdxC>
    - + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
    std::string_view gemv_masked_kernelvoid gather_axis (const device T * src,
    const device IdxT * indices,
    device T * out,
    const constant int * shape,
    const constant int64_t * src_strides,
    const constant int64_t * idx_strides,
    const constant size_t & ndim,
    const constant int & axis,
    const constant int & axis_size,
    const constant size_t & src_ax_stride,
    const constant size_t & idx_ax_stride,
    uint3 index,
    uint3 grid_dim )
    -     		-constexpr
    -Initial value:
    = R"(
    -
    template [[host_name("{name}")]] [[kernel]] void
    -
    gemv_{trans}masked<{itype}, {outm_t}, {opm_t}, {bm}, {bn}, {sm}, {sn}, {tm}, {tn}, {nc}>(
    -
    const device {itype}* mat [[buffer(0)]],
    -
    const device {itype}* in_vec [[buffer(1)]],
    -
    device {itype}* out_vec [[buffer(3)]],
    -
    const constant int& in_vec_size [[buffer(4)]],
    -
    const constant int& out_vec_size [[buffer(5)]],
    -
    const constant int& marix_ld [[buffer(6)]],
    -
    const constant int& batch_ndim [[buffer(9)]],
    -
    const constant int* batch_shape [[buffer(10)]],
    -
    const constant int64_t* vector_batch_stride [[buffer(11)]],
    -
    const constant int64_t* matrix_batch_stride [[buffer(12)]],
    -
    const device {outm_t}* out_mask [[buffer(20)]],
    -
    const device {opm_t}* mat_mask [[buffer(21)]],
    -
    const device {opm_t}* vec_mask [[buffer(22)]],
    -
    const constant int* mask_strides [[buffer(23)]],
    -
    const constant int64_t* mask_batch_strides [[buffer(24)]],
    -
    uint3 tid [[threadgroup_position_in_grid]],
    -
    uint3 lid [[thread_position_in_threadgroup]],
    -
    uint simd_gid [[simdgroup_index_in_threadgroup]],
    -
    uint simd_lid [[thread_index_in_simdgroup]]);
    -
    )"
    -
    +
    @@ -166,8 +200,8 @@ Variables diff --git a/docs/build/html/gather__axis_8h.js b/docs/build/html/gather__axis_8h.js new file mode 100644 index 000000000..641f5e6c7 --- /dev/null +++ b/docs/build/html/gather__axis_8h.js @@ -0,0 +1,4 @@ +var gather__axis_8h = +[ + [ "gather_axis", "gather__axis_8h.html#a0d7d3866afe2007e280781502dc767c6", null ] +]; \ No newline at end of file diff --git a/docs/build/html/gather__axis_8h_source.html b/docs/build/html/gather__axis_8h_source.html new file mode 100644 index 000000000..34ddfe36a --- /dev/null +++ b/docs/build/html/gather__axis_8h_source.html @@ -0,0 +1,166 @@ + + + + + + + +MLX: mlx/backend/metal/kernels/gather_axis.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    gather_axis.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2025 Apple Inc.
    +
    2
    +
    3#pragma once
    +
    4
    +
    5template <typename T, typename IdxT, typename LocT, bool SrcC, bool IdxC>
    +
    +
    6[[kernel]] void gather_axis(
    +
    7 const device T* src [[buffer(0)]],
    +
    8 const device IdxT* indices [[buffer(1)]],
    +
    9 device T* out [[buffer(2)]],
    +
    10 const constant int* shape [[buffer(3)]],
    +
    11 const constant int64_t* src_strides [[buffer(4)]],
    +
    12 const constant int64_t* idx_strides [[buffer(5)]],
    +
    13 const constant size_t& ndim [[buffer(6)]],
    +
    14 const constant int& axis [[buffer(7)]],
    +
    15 const constant int& axis_size [[buffer(8)]],
    +
    16 const constant size_t& src_ax_stride [[buffer(9)]],
    +
    17 const constant size_t& idx_ax_stride [[buffer(10)]],
    +
    18 uint3 index [[thread_position_in_grid]],
    +
    19 uint3 grid_dim [[threads_per_grid]]) {
    +
    20 LocT elem_idx = index.z * static_cast<LocT>(grid_dim.x);
    +
    21 LocT out_idx = elem_idx * grid_dim.y + index.x;
    +
    22
    +
    23 LocT idx_loc = index.y * static_cast<LocT>(idx_ax_stride);
    +
    24 if (IdxC) {
    +
    25 idx_loc += out_idx;
    +
    26 } else {
    +
    27 idx_loc += elem_to_loc<LocT>(elem_idx + index.x, shape, idx_strides, ndim);
    +
    28 }
    +
    29
    +
    30 auto idx_val = indices[idx_loc];
    +
    31 if (is_signed_v<IdxT>) {
    +
    32 idx_val = (idx_val < 0) ? idx_val + axis_size : idx_val;
    +
    33 }
    +
    34
    +
    35 LocT src_idx = idx_val * static_cast<LocT>(src_ax_stride);
    +
    36 if (SrcC) {
    +
    37 src_idx += elem_idx * axis_size + index.x;
    +
    38 } else {
    +
    39 src_idx += elem_to_loc<LocT>(elem_idx + index.x, shape, src_strides, ndim);
    +
    40 }
    +
    41
    +
    42 out_idx += index.y * static_cast<LocT>(grid_dim.x);
    +
    43 out[out_idx] = src[src_idx];
    +
    44}
    +
    +
    elem_to_loc
    METAL_FUNC IdxT elem_to_loc(IdxT elem, constant const int *shape, constant const int64_t *strides, int ndim)
    Definition utils.h:93
    +
    gather_axis
    void gather_axis(const device T *src, const device IdxT *indices, device T *out, const constant int *shape, const constant int64_t *src_strides, const constant int64_t *idx_strides, const constant size_t &ndim, const constant int &axis, const constant int &axis_size, const constant size_t &src_ax_stride, const constant size_t &idx_ax_stride, uint3 index, uint3 grid_dim)
    Definition gather_axis.h:6
    +
    +
    + + + + diff --git a/docs/build/html/gemm_2loader_8h.html b/docs/build/html/gemm_2loader_8h.html index e583c1369..2ec4c237b 100644 --- a/docs/build/html/gemm_2loader_8h.html +++ b/docs/build/html/gemm_2loader_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/steel/gemm/loader.h File Reference @@ -52,7 +52,7 @@ - +
    @@ -112,7 +112,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h.html',''); initResizable(true
    #include "mlx/backend/metal/kernels/steel/utils.h"
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    @@ -419,8 +419,8 @@ template<typename T, typename out_mask_t, typename op_mask_t, const int BM, c diff --git a/docs/build/html/gemv__masked_8h.js b/docs/build/html/gemv__masked_8h.js new file mode 100644 index 000000000..5759be884 --- /dev/null +++ b/docs/build/html/gemv__masked_8h.js @@ -0,0 +1,12 @@ +var gemv__masked_8h = +[ + [ "_NoMask", "struct___no_mask.html", "struct___no_mask" ], + [ "ScaleOp< OutT, InT >", "struct_scale_op.html", "struct_scale_op" ], + [ "GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >", "struct_g_e_m_v_kernel.html", "struct_g_e_m_v_kernel" ], + [ "GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >", "struct_g_e_m_v_t_kernel.html", "struct_g_e_m_v_t_kernel" ], + [ "MLX_MTL_CONST", "gemv__masked_8h.html#a0386011c52d03e60885a31e6fbd903dd", null ], + [ "MLX_MTL_PRAGMA_UNROLL", "gemv__masked_8h.html#a069b682d7d21827461544817d722bfd3", null ], + [ "nomask_t", "gemv__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca", null ], + [ "gemv_masked", "gemv__masked_8h.html#af890b6ac155165f8ee0c600363938341", null ], + [ "gemv_t_masked", "gemv__masked_8h.html#ae5b4a5124ddf92a984258a0be1ff0f4f", null ] +]; \ No newline at end of file diff --git a/docs/build/html/kernels_2gemv__masked_8h_source.html b/docs/build/html/gemv__masked_8h_source.html similarity index 90% rename from docs/build/html/kernels_2gemv__masked_8h_source.html rename to docs/build/html/gemv__masked_8h_source.html index faa6c7506..1f0ff5687 100644 --- a/docs/build/html/kernels_2gemv__masked_8h_source.html +++ b/docs/build/html/gemv__masked_8h_source.html @@ -3,7 +3,7 @@ - +MLX: mlx/backend/metal/kernels/gemv_masked.h Source File @@ -52,7 +52,7 @@

    Classes

    - +
    @@ -105,14 +105,14 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    gemv_masked.h
    -Go to the documentation of this file.
    1// Copyright © 2023-2024 Apple Inc.
    +Go to the documentation of this file.
    1// Copyright © 2023-2024 Apple Inc.
    2
    3#include "mlx/backend/metal/kernels/steel/utils.h"
    4
    5using namespace metal;
    6
    -
    7#define MLX_MTL_CONST static constant constexpr const
    -
    8#define MLX_MTL_PRAGMA_UNROLL _Pragma("clang loop unroll(full)")
    +
    7#define MLX_MTL_CONST static constant constexpr const
    +
    8#define MLX_MTL_PRAGMA_UNROLL _Pragma("clang loop unroll(full)")
    9
    10struct _NoMask {
    @@ -141,7 +141,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    25};
    26
    -
    27typedef struct _NoMask nomask_t;
    +
    27typedef struct _NoMask nomask_t;
    28
    29template <typename OutT, typename InT = OutT>
    @@ -168,11 +168,11 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    47 const int TN> /* Thread cols (in elements) */
    48struct GEMVKernel {
    -
    49 MLX_MTL_CONST int threadsM = BM * SM;
    -
    50 MLX_MTL_CONST int threadsN = BN * SN;
    +
    49 MLX_MTL_CONST int threadsM = BM * SM;
    +
    50 MLX_MTL_CONST int threadsN = BN * SN;
    51
    -
    52 MLX_MTL_CONST int blockM = threadsM * TM;
    -
    53 MLX_MTL_CONST int blockN = threadsN * TN;
    +
    52 MLX_MTL_CONST int blockM = threadsM * TM;
    +
    53 MLX_MTL_CONST int blockN = threadsN * TN;
    54
    55 static_assert(SM * SN == 32, "simdgroup can only have 32 threads");
    56
    @@ -182,12 +182,12 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    60
    61 static_assert(blockN >= blockM, "Masked gemv must have blockN >= blockM");
    62
    -
    63 MLX_MTL_CONST bool has_operand_mask = !metal::is_same_v<op_mask_t, nomask_t>;
    -
    64 MLX_MTL_CONST bool has_output_mask = !metal::is_same_v<out_mask_t, nomask_t>;
    +
    63 MLX_MTL_CONST bool has_operand_mask = !metal::is_same_v<op_mask_t, nomask_t>;
    +
    64 MLX_MTL_CONST bool has_output_mask = !metal::is_same_v<out_mask_t, nomask_t>;
    65
    -
    66 MLX_MTL_CONST bool has_mul_operand_mask =
    +
    66 MLX_MTL_CONST bool has_mul_operand_mask =
    67 has_operand_mask && !metal::is_same_v<op_mask_t, bool>;
    -
    68 MLX_MTL_CONST bool has_mul_output_mask =
    +
    68 MLX_MTL_CONST bool has_mul_output_mask =
    69 has_output_mask && !metal::is_same_v<out_mask_t, bool>;
    70
    71 // - The matrix of size (M = out_vec_size, K = in_vec_size) is divided up
    @@ -210,13 +210,13 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    88 // * The last thread that partially overlaps with the matrix is shifted
    89 // inwards such that the thread block fits exactly in the matrix
    90
    -
    91 MLX_MTL_CONST short tgp_mem_size = BN > 1 ? BN*(blockM + TM) : 0;
    -
    92 MLX_MTL_CONST bool needs_tgp_reduction = BN > 1;
    +
    91 MLX_MTL_CONST short tgp_mem_size = BN > 1 ? BN*(blockM + TM) : 0;
    +
    92 MLX_MTL_CONST bool needs_tgp_reduction = BN > 1;
    93
    94 static METAL_FUNC void
    95 load_unsafe(const device T* src, thread T dst[TN], const int src_offset = 0) {
    -
    96 MLX_MTL_PRAGMA_UNROLL
    +
    96 MLX_MTL_PRAGMA_UNROLL
    97 for (int tn = 0; tn < TN; tn++) {
    98 dst[tn] = src[src_offset + tn];
    99 }
    @@ -230,12 +230,12 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    105 const int src_offset = 0,
    106 const int src_size = TN) {
    107 if (src_offset + TN <= src_size) {
    -
    108 MLX_MTL_PRAGMA_UNROLL
    +
    108 MLX_MTL_PRAGMA_UNROLL
    109 for (int tn = 0; tn < TN; tn++) {
    110 dst[tn] = src[src_offset + tn];
    111 }
    112 } else { // Edgecase
    -
    113 MLX_MTL_PRAGMA_UNROLL
    +
    113 MLX_MTL_PRAGMA_UNROLL
    114 for (int tn = 0; tn < TN; tn++) {
    115 dst[tn] = src_offset + tn < src_size ? src[src_offset + tn] : 0;
    116 }
    @@ -316,7 +316,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    189 // Write zeros and return if mask is 0
    190 if (!mask_out) {
    191 if (simdN == 0 && thrN == 0) {
    -
    192 MLX_MTL_PRAGMA_UNROLL
    +
    192 MLX_MTL_PRAGMA_UNROLL
    193 for (int tm = 0; tm < TM; tm++) {
    194 out_vec[out_row + tm] = T(0.);
    195 }
    @@ -356,7 +356,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    229
    230 // Apply scale
    231 if (has_mul_operand_mask) {
    -
    232 MLX_MTL_PRAGMA_UNROLL
    +
    232 MLX_MTL_PRAGMA_UNROLL
    233 for (int tn = 0; tn < TN; tn++) {
    234 v_coeff[tn] *= block_scale;
    235 }
    @@ -364,13 +364,13 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    237
    238 // Per thread work loop
    239 int mat_offset = 0;
    -
    240 MLX_MTL_PRAGMA_UNROLL
    +
    240 MLX_MTL_PRAGMA_UNROLL
    241 for (int tm = 0; tm < TM; tm++) {
    242 // Load for the row
    243 load_unsafe(mat, inter, mat_offset + bn);
    244
    245 // Accumulate results
    -
    246 MLX_MTL_PRAGMA_UNROLL
    +
    246 MLX_MTL_PRAGMA_UNROLL
    247 for (int tn = 0; tn < TN; tn++) {
    248 result[tm] += inter[tn] * v_coeff[tn];
    249 }
    @@ -398,20 +398,20 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    271
    272 // Apply scale
    273 if (has_mul_operand_mask) {
    -
    274 MLX_MTL_PRAGMA_UNROLL
    +
    274 MLX_MTL_PRAGMA_UNROLL
    275 for (int tn = 0; tn < TN; tn++) {
    276 v_coeff[tn] *= block_scale;
    277 }
    278 }
    279
    280 // Per thread work loop
    -
    281 MLX_MTL_PRAGMA_UNROLL
    +
    281 MLX_MTL_PRAGMA_UNROLL
    282 for (int tm = 0; tm < TM; tm++) {
    283 // Load for the row
    284 load_safe(&mat[tm * matrix_ld], inter, bn, in_size);
    285
    286 // Accumulate results
    -
    287 MLX_MTL_PRAGMA_UNROLL
    +
    287 MLX_MTL_PRAGMA_UNROLL
    288 for (int tn = 0; tn < TN; tn++) {
    289 result[tm] += inter[tn] * v_coeff[tn];
    290 }
    @@ -420,16 +420,16 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    293
    294 // Apply out scale
    295 if (has_mul_output_mask) {
    -
    296 MLX_MTL_PRAGMA_UNROLL
    +
    296 MLX_MTL_PRAGMA_UNROLL
    297 for (int tm = 0; tm < TM; tm++) {
    298 result[tm] *= out_scale;
    299 }
    300 }
    301
    302 // Simdgroup accumulations
    -
    303 MLX_MTL_PRAGMA_UNROLL
    +
    303 MLX_MTL_PRAGMA_UNROLL
    304 for (int tm = 0; tm < TM; tm++) {
    -
    305 MLX_MTL_PRAGMA_UNROLL
    +
    305 MLX_MTL_PRAGMA_UNROLL
    306 for (ushort sn = (SN / 2); sn >= 1; sn >>= 1) {
    307 result[tm] += simd_shuffle_down(result[tm], sn);
    308 }
    @@ -439,7 +439,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    312 if (needs_tgp_reduction) {
    313 threadgroup T* tgp_results = tgp_memory + sgN * (blockM + TM) + bm;
    314 if (thrN == 0) {
    -
    315 MLX_MTL_PRAGMA_UNROLL
    +
    315 MLX_MTL_PRAGMA_UNROLL
    316 for (int tm = 0; tm < TM; tm++) {
    317 tgp_results[tm] = result[tm];
    318 }
    @@ -447,9 +447,9 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    320 threadgroup_barrier(mem_flags::mem_none);
    321
    322 if (sgN == 0) {
    -
    323 MLX_MTL_PRAGMA_UNROLL
    +
    323 MLX_MTL_PRAGMA_UNROLL
    324 for (int sgn = 1; sgn < BN; sgn++) {
    -
    325 MLX_MTL_PRAGMA_UNROLL
    +
    325 MLX_MTL_PRAGMA_UNROLL
    326 for (int tm = 0; tm < TM; tm++) {
    327 result[tm] += tgp_results[sgn * (blockM + TM) + tm];
    328 }
    @@ -460,7 +460,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    333
    334 // Write outputs
    335 if (simdN == 0 && thrN == 0) {
    -
    336 MLX_MTL_PRAGMA_UNROLL
    +
    336 MLX_MTL_PRAGMA_UNROLL
    337 for (int tm = 0; tm < TM; tm++) {
    338 out_vec[out_row + tm] = result[tm];
    339 }
    @@ -483,20 +483,20 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    357 const int TN> /* Thread cols (in elements) */
    358struct GEMVTKernel {
    -
    359 MLX_MTL_CONST int threadsM = BM * SM;
    -
    360 MLX_MTL_CONST int threadsN = BN * SN;
    +
    359 MLX_MTL_CONST int threadsM = BM * SM;
    +
    360 MLX_MTL_CONST int threadsN = BN * SN;
    361
    -
    362 MLX_MTL_CONST int blockM = threadsM * TM;
    -
    363 MLX_MTL_CONST int blockN = threadsN * TN;
    +
    362 MLX_MTL_CONST int blockM = threadsM * TM;
    +
    363 MLX_MTL_CONST int blockN = threadsN * TN;
    364
    365 static_assert(SM * SN == 32, "simdgroup can only have 32 threads");
    366
    -
    367 MLX_MTL_CONST bool has_operand_mask = !metal::is_same_v<op_mask_t, nomask_t>;
    -
    368 MLX_MTL_CONST bool has_output_mask = !metal::is_same_v<out_mask_t, nomask_t>;
    +
    367 MLX_MTL_CONST bool has_operand_mask = !metal::is_same_v<op_mask_t, nomask_t>;
    +
    368 MLX_MTL_CONST bool has_output_mask = !metal::is_same_v<out_mask_t, nomask_t>;
    369
    -
    370 MLX_MTL_CONST bool has_mul_operand_mask =
    +
    370 MLX_MTL_CONST bool has_mul_operand_mask =
    371 has_operand_mask && !metal::is_same_v<op_mask_t, bool>;
    -
    372 MLX_MTL_CONST bool has_mul_output_mask =
    +
    372 MLX_MTL_CONST bool has_mul_output_mask =
    373 has_output_mask && !metal::is_same_v<out_mask_t, bool>;
    374
    375 // - The matrix of size (M = in_vec_size, N = out_vec_size) is divided up
    @@ -518,8 +518,8 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    391 // * The last thread that partially overlaps with the matrix is shifted
    392 // inwards such that the thread block fits exactly in the matrix
    393
    -
    394 MLX_MTL_CONST short tgp_mem_size = BM > 1 ? BM*(blockN + TN) : 0;
    -
    395 MLX_MTL_CONST bool needs_tgp_reduction = BM > 1;
    +
    394 MLX_MTL_CONST short tgp_mem_size = BM > 1 ? BM*(blockN + TN) : 0;
    +
    395 MLX_MTL_CONST bool needs_tgp_reduction = BM > 1;
    396
    397 static METAL_FUNC void run(
    @@ -591,7 +591,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    463 if (!mask_out) {
    464 if (cm == 0 && out_col < out_vec_size) {
    465 if (out_col + TN <= out_vec_size) {
    -
    466 MLX_MTL_PRAGMA_UNROLL
    +
    466 MLX_MTL_PRAGMA_UNROLL
    467 for (int tn = 0; tn < TN; tn++) {
    468 out_vec[out_col + tn] = T(0.);
    469 }
    @@ -637,20 +637,20 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    509 T(mat_mask[mat_mask_offset]) * T(vec_mask[vec_mask_offset]);
    510 }
    511
    -
    512 MLX_MTL_PRAGMA_UNROLL
    +
    512 MLX_MTL_PRAGMA_UNROLL
    513 for (int tm = 0; tm < TM; tm++) {
    514 v_coeff[tm] = in_vec[bm + tm];
    515 }
    516
    517 // Apply scale
    518 if (has_mul_operand_mask) {
    -
    519 MLX_MTL_PRAGMA_UNROLL
    +
    519 MLX_MTL_PRAGMA_UNROLL
    520 for (int tm = 0; tm < TM; tm++) {
    521 v_coeff[tm] *= block_scale;
    522 }
    523 }
    524
    -
    525 MLX_MTL_PRAGMA_UNROLL
    +
    525 MLX_MTL_PRAGMA_UNROLL
    526 for (int tm = 0; tm < TM; tm++) {
    527 for (int tn = 0; tn < TN; tn++) {
    528 inter[tn] = mat[(bm + tm) * marix_ld + out_col + tn];
    @@ -683,12 +683,12 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    555 v_coeff[tm] *= block_scale;
    556 }
    557
    -
    558 MLX_MTL_PRAGMA_UNROLL
    +
    558 MLX_MTL_PRAGMA_UNROLL
    559 for (int tn = 0; tn < TN; tn++) {
    560 inter[tn] = mat[(bm + tm) * marix_ld + out_col + tn];
    561 }
    562
    -
    563 MLX_MTL_PRAGMA_UNROLL
    +
    563 MLX_MTL_PRAGMA_UNROLL
    564 for (int tn = 0; tn < TN; tn++) {
    565 result[tn] += v_coeff[tm] * inter[tn];
    566 }
    @@ -698,16 +698,16 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    570
    571 // Apply out scale
    572 if (has_mul_output_mask) {
    -
    573 MLX_MTL_PRAGMA_UNROLL
    +
    573 MLX_MTL_PRAGMA_UNROLL
    574 for (int tn = 0; tn < TN; tn++) {
    575 result[tn] *= out_scale;
    576 }
    577 }
    578
    579 // Simdgroup accumulations
    -
    580 MLX_MTL_PRAGMA_UNROLL
    +
    580 MLX_MTL_PRAGMA_UNROLL
    581 for (int tn = 0; tn < TN; tn++) {
    -
    582 MLX_MTL_PRAGMA_UNROLL
    +
    582 MLX_MTL_PRAGMA_UNROLL
    583 for (ushort sm = (SM / 2); sm >= 1; sm >>= 1) {
    584 result[tn] += simd_shuffle_down(result[tn], SN * sm);
    585 }
    @@ -717,7 +717,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    589 if (needs_tgp_reduction) {
    590 threadgroup T* tgp_results = tgp_memory + sgM * (blockN + TN) + bn;
    591 if (thrM == 0) {
    -
    592 MLX_MTL_PRAGMA_UNROLL
    +
    592 MLX_MTL_PRAGMA_UNROLL
    593 for (int tn = 0; tn < TN; tn++) {
    594 tgp_results[tn] = result[tn];
    595 }
    @@ -725,9 +725,9 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    597 threadgroup_barrier(mem_flags::mem_none);
    598
    599 if (sgM == 0) {
    -
    600 MLX_MTL_PRAGMA_UNROLL
    +
    600 MLX_MTL_PRAGMA_UNROLL
    601 for (int sgm = 1; sgm < BM; sgm++) {
    -
    602 MLX_MTL_PRAGMA_UNROLL
    +
    602 MLX_MTL_PRAGMA_UNROLL
    603 for (int tn = 0; tn < TN; tn++) {
    604 result[tn] += tgp_results[sgm * (blockN + TN) + tn];
    605 }
    @@ -738,7 +738,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    610
    611 // Threadgroup accumulation and writing out results
    612 if (cm == 0 && out_col < out_vec_size) {
    -
    613 MLX_MTL_PRAGMA_UNROLL
    +
    613 MLX_MTL_PRAGMA_UNROLL
    614 for (int j = 0; j < TN; j++) {
    615 out_vec[out_col + j] = result[j];
    616 }
    @@ -761,7 +761,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    634 const int TN, /* Thread cols (in elements) */
    635 const bool kDoNCBatch> /* Batch ndim > 1 */
    -
    636[[kernel, max_total_threads_per_threadgroup(BM* BN * 32)]] void gemv_masked(
    +
    636[[kernel, max_total_threads_per_threadgroup(BM* BN * 32)]] void gemv_masked(
    637 const device T* mat [[buffer(0)]],
    638 const device T* in_vec [[buffer(1)]],
    639 device T* out_vec [[buffer(3)]],
    @@ -860,7 +860,7 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    734 const int TN, /* Thread cols (in elements) */
    735 const bool kDoNCBatch> /* Batch ndim > 1 */
    -
    736[[kernel, max_total_threads_per_threadgroup(BM* BN * 32)]] void gemv_t_masked(
    +
    736[[kernel, max_total_threads_per_threadgroup(BM* BN * 32)]] void gemv_t_masked(
    737 const device T* mat [[buffer(0)]],
    738 const device T* in_vec [[buffer(1)]],
    739 device T* out_vec [[buffer(3)]],
    @@ -948,11 +948,11 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab
    elem_to_loc_broadcast
    METAL_FUNC ulong2 elem_to_loc_broadcast(uint elem, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, int ndim)
    Definition utils.h:7
    elem_to_loc
    METAL_FUNC IdxT elem_to_loc(IdxT elem, constant const int *shape, constant const int64_t *strides, int ndim)
    Definition utils.h:93
    -
    MLX_MTL_CONST
    #define MLX_MTL_CONST
    Definition gemv_masked.h:7
    -
    MLX_MTL_PRAGMA_UNROLL
    #define MLX_MTL_PRAGMA_UNROLL
    Definition gemv_masked.h:8
    -
    nomask_t
    struct _NoMask nomask_t
    Definition gemv_masked.h:27
    -
    gemv_t_masked
    void gemv_t_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant int64_t *vector_batch_stride, const constant int64_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant int64_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)
    Vector matrix multiplication.
    Definition gemv_masked.h:736
    -
    gemv_masked
    void gemv_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant int64_t *vector_batch_stride, const constant int64_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant int64_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)
    Matrix vector multiplication.
    Definition gemv_masked.h:636
    +
    MLX_MTL_CONST
    #define MLX_MTL_CONST
    Definition gemv_masked.h:7
    +
    MLX_MTL_PRAGMA_UNROLL
    #define MLX_MTL_PRAGMA_UNROLL
    Definition gemv_masked.h:8
    +
    nomask_t
    struct _NoMask nomask_t
    Definition gemv_masked.h:27
    +
    gemv_t_masked
    void gemv_t_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant int64_t *vector_batch_stride, const constant int64_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant int64_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)
    Vector matrix multiplication.
    Definition gemv_masked.h:736
    +
    gemv_masked
    void gemv_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant int64_t *vector_batch_stride, const constant int64_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant int64_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)
    Matrix vector multiplication.
    Definition gemv_masked.h:636
    metal
    Definition bf16_math.h:226
    metal::simd_shuffle_down
    METAL_FUNC bfloat16_t simd_shuffle_down(bfloat16_t data, ushort delta)
    Definition bf16_math.h:377
    _NoMask
    Definition gemv_masked.h:10
    @@ -991,8 +991,8 @@ $(function(){initNavTree('kernels_2gemv__masked_8h_source.html',''); initResizab diff --git a/docs/build/html/genindex.html b/docs/build/html/genindex.html index fb7ba6a04..92c582f30 100644 --- a/docs/build/html/genindex.html +++ b/docs/build/html/genindex.html @@ -7,7 +7,7 @@ - Index — MLX 0.22.0 documentation + Index — MLX 0.22.1 documentation @@ -38,7 +38,7 @@ - + @@ -48,7 +48,7 @@ - + @@ -129,8 +129,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home
    @@ -1750,7 +1750,7 @@
  • (in module mlx.core)
    • -
  • meshgrid (C++ function) +
  • meshgrid (C++ function)
  • meshgrid() (in module mlx.core)
    • @@ -1900,7 +1900,7 @@

    P

    - + + @@ -51,7 +51,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home
    diff --git a/docs/build/html/install.html b/docs/build/html/install.html index 3f31e0526..61da43fac 100644 --- a/docs/build/html/install.html +++ b/docs/build/html/install.html @@ -8,7 +8,7 @@ - Build and Install — MLX 0.22.0 documentation + Build and Install — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home
    diff --git a/docs/build/html/integral__constant_8h.html b/docs/build/html/integral__constant_8h.html index 253c629c4..66e06236f 100644 --- a/docs/build/html/integral__constant_8h.html +++ b/docs/build/html/integral__constant_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/steel/utils/integral_constant.h File Reference @@ -52,7 +52,7 @@
    - + - - - - - - - - - - - - - - -
    -
    - - - - - - - -
    -
    MLX -
    -
    - -   - - - - -
    -
    -
    - - - - -
    -
    - -
    -
    -
    - -
    - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - -
    -
    -Variables
    -
    steel_conv.h File Reference
    -
    -
    - -

    Go to the source code of this file.

    - - - - - - -

    -Variables

    constexpr std::string_view steel_conv_kernels
     
    constexpr std::string_view steel_conv_general_kernels
     
    -

    Variable Documentation

    - -

    ◆ steel_conv_general_kernels

    - -
    -
    - - - - - -
    - - - - -
    std::string_view steel_conv_general_kernels
    -     		-constexpr
    -
    -Initial value:
    = R"(
    -
    template [[host_name("{name}")]] [[kernel]] void
    -
    implicit_gemm_conv_2d_general<{itype}, {bm}, {bn}, {bk}, {wm}, {wn}>(
    -
    const device {itype}* A [[buffer(0)]],
    -
    const device {itype}* B [[buffer(1)]],
    -
    device {itype}* C [[buffer(2)]],
    -
    const constant MLXConvParams<2>* params [[buffer(3)]],
    -
    const constant ImplicitGemmConv2DParams* gemm_params [[buffer(4)]],
    -
    const constant Conv2DGeneralJumpParams* jump_params [[buffer(5)]],
    -
    const constant Conv2DGeneralBaseInfo* base_h [[buffer(6)]],
    -
    const constant Conv2DGeneralBaseInfo* base_w [[buffer(7)]],
    -
    uint3 tid [[threadgroup_position_in_grid]],
    -
    uint3 lid [[thread_position_in_threadgroup]],
    -
    uint simd_gid [[simdgroup_index_in_threadgroup]],
    -
    uint simd_lid [[thread_index_in_simdgroup]]);
    -
    )"
    -
    -
    -
    - -

    ◆ steel_conv_kernels

    - -
    -
    - - - - - -
    - - - - -
    std::string_view steel_conv_kernels
    -     		-constexpr
    -
    -Initial value:
    = R"(
    -
    template [[host_name("{name}")]] [[kernel]] void
    -
    implicit_gemm_conv_2d<{itype}, {bm}, {bn}, {bk}, {wm}, {wn}, {n_channels}, {small_filter}>(
    -
    const device {itype}* A [[buffer(0)]],
    -
    const device {itype}* B [[buffer(1)]],
    -
    device {itype}* C [[buffer(2)]],
    -
    const constant MLXConvParams<2>* params [[buffer(3)]],
    -
    const constant ImplicitGemmConv2DParams* gemm_params [[buffer(4)]],
    -
    uint3 tid [[threadgroup_position_in_grid]],
    -
    uint3 lid [[thread_position_in_threadgroup]],
    -
    uint simd_gid [[simdgroup_index_in_threadgroup]],
    -
    uint simd_lid [[thread_index_in_simdgroup]]);
    -
    )"
    -
    -
    -
    -
    -
    - - - - diff --git a/docs/build/html/jit_2steel__conv_8h.js b/docs/build/html/jit_2steel__conv_8h.js deleted file mode 100644 index 3cdad7c60..000000000 --- a/docs/build/html/jit_2steel__conv_8h.js +++ /dev/null @@ -1,5 +0,0 @@ -var jit_2steel__conv_8h = -[ - [ "steel_conv_general_kernels", "jit_2steel__conv_8h.html#ae4ca1720029316b08ea92b7662347d47", null ], - [ "steel_conv_kernels", "jit_2steel__conv_8h.html#a386d79077465df56659416fd84adb513", null ] -]; \ No newline at end of file diff --git a/docs/build/html/jit_2steel__conv_8h_source.html b/docs/build/html/jit_2steel__conv_8h_source.html deleted file mode 100644 index 16a1fc074..000000000 --- a/docs/build/html/jit_2steel__conv_8h_source.html +++ /dev/null @@ -1,152 +0,0 @@ - - - - - - - -MLX: mlx/backend/metal/jit/steel_conv.h Source File - - - - - - - - - - - - - - - - -
    -
    - - - - - - - -
    -
    MLX -
    -
    - -   - - - - -
    -
    -
    - - - - -
    -
    - -
    -
    -
    - -
    - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - -
    -
    steel_conv.h
    -
    -
    -Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    -
    2
    -
    3constexpr std::string_view steel_conv_kernels = R"(
    -
    4template [[host_name("{name}")]] [[kernel]] void
    -
    5implicit_gemm_conv_2d<{itype}, {bm}, {bn}, {bk}, {wm}, {wn}, {n_channels}, {small_filter}>(
    -
    6 const device {itype}* A [[buffer(0)]],
    -
    7 const device {itype}* B [[buffer(1)]],
    -
    8 device {itype}* C [[buffer(2)]],
    -
    9 const constant MLXConvParams<2>* params [[buffer(3)]],
    -
    10 const constant ImplicitGemmConv2DParams* gemm_params [[buffer(4)]],
    -
    11 uint3 tid [[threadgroup_position_in_grid]],
    -
    12 uint3 lid [[thread_position_in_threadgroup]],
    -
    13 uint simd_gid [[simdgroup_index_in_threadgroup]],
    -
    14 uint simd_lid [[thread_index_in_simdgroup]]);
    -
    15)";
    -
    16
    -
    17constexpr std::string_view steel_conv_general_kernels = R"(
    -
    18template [[host_name("{name}")]] [[kernel]] void
    -
    19 implicit_gemm_conv_2d_general<{itype}, {bm}, {bn}, {bk}, {wm}, {wn}>(
    -
    20 const device {itype}* A [[buffer(0)]],
    -
    21 const device {itype}* B [[buffer(1)]],
    -
    22 device {itype}* C [[buffer(2)]],
    -
    23 const constant MLXConvParams<2>* params [[buffer(3)]],
    -
    24 const constant ImplicitGemmConv2DParams* gemm_params [[buffer(4)]],
    -
    25 const constant Conv2DGeneralJumpParams* jump_params [[buffer(5)]],
    -
    26 const constant Conv2DGeneralBaseInfo* base_h [[buffer(6)]],
    -
    27 const constant Conv2DGeneralBaseInfo* base_w [[buffer(7)]],
    -
    28 uint3 tid [[threadgroup_position_in_grid]],
    -
    29 uint3 lid [[thread_position_in_threadgroup]],
    -
    30 uint simd_gid [[simdgroup_index_in_threadgroup]],
    -
    31 uint simd_lid [[thread_index_in_simdgroup]]);
    -
    32)";
    -
    steel_conv_kernels
    constexpr std::string_view steel_conv_kernels
    Definition steel_conv.h:3
    -
    steel_conv_general_kernels
    constexpr std::string_view steel_conv_general_kernels
    Definition steel_conv.h:17
    -
    -
    - - - - diff --git a/docs/build/html/jit__compiler_8h.html b/docs/build/html/jit__compiler_8h.html index 90a1c2d53..f5d230c98 100644 --- a/docs/build/html/jit__compiler_8h.html +++ b/docs/build/html/jit__compiler_8h.html @@ -3,9 +3,9 @@ - + -MLX: mlx/backend/common/jit_compiler.h File Reference +MLX: mlx/backend/cpu/jit_compiler.h File Reference @@ -52,7 +52,7 @@
    - + @@ -52,7 +52,7 @@
    - + @@ -52,7 +52,7 @@
    - + @@ -52,7 +52,7 @@
    - + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Namespaces | +Functions | +Variables
    +
    math.h File Reference
    +
    +
    +
    #include "mlx/backend/cpu/simd/type.h"
    +
    +

    Go to the source code of this file.

    + + + + + + + + +

    +Namespaces

    namespace  mlx
     
    namespace  mlx::core
     
    namespace  mlx::core::simd
     
    + + + + + + + + + + + + + + + + + + + + +

    +Functions

    template<typename T, int N>
    Simd< T, N > mlx::core::simd::exp (Simd< T, N > in)
     Compute exp(x) in an optimizer friendly way as follows:
     
    template<bool Sine, typename T, int N>
    Simd< T, N > mlx::core::simd::sincos (Simd< T, N > in)
     
    template<typename T, int N>
    Simd< T, N > mlx::core::simd::sin (Simd< T, N > x)
     
    template<typename T, int N>
    Simd< T, N > mlx::core::simd::cos (Simd< T, N > x)
     
    template<typename T, int N>
    Simd< T, N > mlx::core::simd::erf (Simd< T, N > x)
     
    template<typename T, int N>
    Simd< T, N > mlx::core::simd::erfinv (Simd< T, N > a_)
     
    + + + +

    +Variables

    constexpr float mlx::core::simd::inf = std::numeric_limits<float>::infinity()
     
    +
    +
    + + + + diff --git a/docs/build/html/math_8h.js b/docs/build/html/math_8h.js new file mode 100644 index 000000000..edc4855dc --- /dev/null +++ b/docs/build/html/math_8h.js @@ -0,0 +1,10 @@ +var math_8h = +[ + [ "mlx::core::simd::cos", "namespacemlx_1_1core_1_1simd.html#ab179f429e34cd6d5c37050ea7e7c54ad", null ], + [ "mlx::core::simd::erf", "namespacemlx_1_1core_1_1simd.html#a60e33ebb16d9bab375a64aec8015a5c2", null ], + [ "mlx::core::simd::erfinv", "namespacemlx_1_1core_1_1simd.html#a7687f3d14077b51fb421f0efb5b626db", null ], + [ "mlx::core::simd::exp", "namespacemlx_1_1core_1_1simd.html#a835d71dd0bb2f9494a397d9939696ec2", null ], + [ "mlx::core::simd::sin", "namespacemlx_1_1core_1_1simd.html#ab4d582d72c0a7ee313e19c906e43cef1", null ], + [ "mlx::core::simd::sincos", "namespacemlx_1_1core_1_1simd.html#ad78056685c9732c3465c0d8b8ec1bef7", null ], + [ "mlx::core::simd::inf", "namespacemlx_1_1core_1_1simd.html#a771b6597803beb800ff5e7560c41e341", null ] +]; \ No newline at end of file diff --git a/docs/build/html/math_8h_source.html b/docs/build/html/math_8h_source.html new file mode 100644 index 000000000..55957518c --- /dev/null +++ b/docs/build/html/math_8h_source.html @@ -0,0 +1,327 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/math.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    math.h
    +
    +
    +Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    +
    2
    +
    3#pragma once
    +
    4
    +
    5#include "mlx/backend/cpu/simd/type.h"
    +
    6
    +
    7namespace mlx::core::simd {
    +
    8
    +
    9constexpr float inf = std::numeric_limits<float>::infinity();
    +
    10
    +
    27template <typename T, int N>
    +
    +
    28Simd<T, N> exp(Simd<T, N> in) {
    +
    29 if constexpr (is_complex<T>) {
    +
    30 return Simd<T, 1>{std::exp(in.value)};
    +
    31 } else {
    +
    32 Simd<float, N> x_init = in;
    +
    33 auto x = x_init * 1.442695f; // multiply with log_2(e)
    +
    34 Simd<float, N> ipart, fpart;
    +
    35 ipart = floor(x + 0.5);
    +
    36 fpart = x - ipart;
    +
    37
    +
    38 x = 1.535336188319500e-4f;
    +
    39 x = fma(x, fpart, 1.339887440266574e-3f);
    +
    40 x = fma(x, fpart, 9.618437357674640e-3f);
    +
    41 x = fma(x, fpart, 5.550332471162809e-2f);
    +
    42 x = fma(x, fpart, 2.402264791363012e-1f);
    +
    43 x = fma(x, fpart, 6.931472028550421e-1f);
    +
    44 x = fma(x, fpart, 1.000000000000000f);
    +
    45
    +
    46 // generate 2**ipart in the floating point representation using integer
    +
    47 // bitshifting
    +
    48 Simd<int, N> epart = (Simd<int, N>(ipart) + 127) << 23;
    +
    49
    +
    50 // Deal with NaN and Inf
    +
    51 auto result = select(isnan(x_init), x_init, (*(Simd<float, N>*)&epart) * x);
    +
    52 result = select(x_init > 88.0f, Simd<float, N>(inf), result);
    +
    53 result = select(x_init < -88.0f, Simd<float, N>(0), result);
    +
    54 return Simd<T, N>(result);
    +
    55 }
    +
    56}
    +
    +
    57
    +
    58/* Implementation from:
    +
    59 * https://github.com/JishinMaster/simd_utils/blob/3c1433a86fb38edcc9b02039f3c9a65b16640976/neon_mathfun.h#L357
    +
    60 * which originally came from the Cephes math library.
    +
    61 */
    +
    62template <bool Sine, typename T, int N>
    +
    +
    63Simd<T, N> sincos(Simd<T, N> in) {
    +
    64 auto sign_mask_sin = in < 0;
    +
    65 in = abs(in);
    +
    66 Simd<float, N> x = in;
    +
    67
    +
    68 // scale by 4/Pi
    +
    69 auto y = x * 1.27323954473516f;
    +
    70
    +
    71 // store the integer part of y in mm0
    +
    72 Simd<uint32_t, N> emm2 = y;
    +
    73
    +
    74 // j=(j+1) & (~1) (see the cephes sources)
    +
    75 emm2 = emm2 + 1;
    +
    76 emm2 = emm2 & ~1;
    +
    77
    +
    78 y = emm2;
    +
    79
    +
    80 // Get the polynom selection mask. There is one polynom for 0 <= x <= Pi/4
    +
    81 // and another one for Pi/4<x<=Pi/2. Both branches will be computed.
    +
    82 auto poly_mask = (emm2 & 2) != 0;
    +
    83
    +
    84 // The magic pass: "Extended precision modular arithmetic"
    +
    85 // x = ((x - y * DP1) - y * DP2) - y * DP3
    +
    86 x = fma(y, Simd<float, N>(-0.78515625f), x);
    +
    87 x = fma(y, Simd<float, N>(-2.4187564849853515625e-4f), x);
    +
    88 x = fma(y, Simd<float, N>(-3.77489497744594108e-8f), x);
    +
    89
    +
    90 sign_mask_sin = sign_mask_sin ^ ((emm2 & 4) != 0);
    +
    91 auto sign_mask_cos = ((emm2 - 2) & 4) != 0;
    +
    92
    +
    93 // Evaluate the first polynom (0 <= x <= Pi/4) in y1,
    +
    94 // and the second polynom (Pi/4 <= x <= 0) in y2
    +
    95 auto z = x * x;
    +
    96
    +
    97 auto y1 =
    +
    98 fma(z, Simd<float, N>(2.443315711809948e-5f), -1.388731625493765e-3f);
    +
    99 auto y2 = fma(z, Simd<float, N>(-1.9515295891e-4f), 8.3321608736e-3f);
    +
    100 y1 = fma(y1, z, 4.166664568298827e-2f);
    +
    101 y2 = fma(y2, z, -1.6666654611e-1f);
    +
    102 y1 = y1 * z;
    +
    103 y2 = y2 * z;
    +
    104 y1 = y1 * z;
    +
    105 y2 = fma(x, y2, x);
    +
    106 y1 = fma(z, Simd<float, N>(-0.5f), y1);
    +
    107 y1 = y1 + 1.0f;
    +
    108
    +
    109 if constexpr (Sine) {
    +
    110 auto ys = select(poly_mask, y1, y2);
    +
    111 return select(sign_mask_sin, -ys, ys);
    +
    112 } else {
    +
    113 auto yc = select(poly_mask, y2, y1);
    +
    114 return select(sign_mask_cos, yc, -yc);
    +
    115 }
    +
    116}
    +
    +
    117
    +
    118template <typename T, int N>
    +
    +
    119Simd<T, N> sin(Simd<T, N> x) {
    +
    120 if constexpr (is_complex<T>) {
    +
    121 return std::sin(x.value);
    +
    122 } else {
    +
    123 return sincos<true>(x);
    +
    124 }
    +
    125}
    +
    +
    126
    +
    127template <typename T, int N>
    +
    +
    128Simd<T, N> cos(Simd<T, N> x) {
    +
    129 if constexpr (is_complex<T>) {
    +
    130 return std::cos(x.value);
    +
    131 } else {
    +
    132 return sincos<false>(x);
    +
    133 }
    +
    134}
    +
    +
    135
    +
    136template <typename T, int N>
    +
    +
    137Simd<T, N> erf(Simd<T, N> x) {
    +
    138 // https://github.com/pytorch/pytorch/blob/abf28982a8cb43342e7669d859de9543fd804cc9/aten/src/ATen/cpu/vec/vec256/vec256_float.h#L175
    +
    139 Simd<float, N> v = x;
    +
    140 auto t = recip(fma(Simd<float, N>(0.3275911f), abs(v), 1.0f));
    +
    141 auto r = fma(Simd<float, N>(1.061405429f), t, -1.453152027f);
    +
    142 r = fma(r, t, 1.421413741f);
    +
    143 r = fma(r, t, -0.284496736f);
    +
    144 r = fma(r, t, 0.254829592f);
    +
    145 auto e = -exp(-v * v);
    +
    146 auto result = Simd<T, N>(fma(e * t, r, 1.0f));
    +
    147 return select(x > 0, result, -result);
    +
    148}
    +
    +
    149
    +
    150template <typename T, int N>
    +
    +
    151Simd<T, N> erfinv(Simd<T, N> a_) {
    +
    152 Simd<float, N> a = a_;
    +
    153 auto t = fma(a, 0.0f - a, 1.0f);
    +
    154 t = log(t);
    +
    155 auto lhs = [](auto t) {
    +
    156 Simd<float, N> p;
    +
    157 p = 3.03697567e-10f; // 0x1.4deb44p-32
    +
    158 p = fma(p, t, 2.93243101e-8f); // 0x1.f7c9aep-26
    +
    159 p = fma(p, t, 1.22150334e-6f); // 0x1.47e512p-20
    +
    160 p = fma(p, t, 2.84108955e-5f); // 0x1.dca7dep-16
    +
    161 p = fma(p, t, 3.93552968e-4f); // 0x1.9cab92p-12
    +
    162 p = fma(p, t, 3.02698812e-3f); // 0x1.8cc0dep-9
    +
    163 p = fma(p, t, 4.83185798e-3f); // 0x1.3ca920p-8
    +
    164 p = fma(p, t, -2.64646143e-1f); // -0x1.0eff66p-2
    +
    165 return fma(p, t, 8.40016484e-1f); // 0x1.ae16a4p-1
    +
    166 };
    +
    167 auto rhs = [](auto t) {
    +
    168 Simd<float, N> p;
    +
    169 p = 5.43877832e-9f; // 0x1.75c000p-28
    +
    170 p = fma(p, t, 1.43285448e-7f); // 0x1.33b402p-23
    +
    171 p = fma(p, t, 1.22774793e-6f); // 0x1.499232p-20
    +
    172 p = fma(p, t, 1.12963626e-7f); // 0x1.e52cd2p-24
    +
    173 p = fma(p, t, -5.61530760e-5f); // -0x1.d70bd0p-15
    +
    174 p = fma(p, t, -1.47697632e-4f); // -0x1.35be90p-13
    +
    175 p = fma(p, t, 2.31468678e-3f); // 0x1.2f6400p-9
    +
    176 p = fma(p, t, 1.15392581e-2f); // 0x1.7a1e50p-7
    +
    177 p = fma(p, t, -2.32015476e-1f); // -0x1.db2aeep-3
    +
    178 return fma(p, t, 8.86226892e-1f); // 0x1.c5bf88p-1
    +
    179 };
    +
    180 auto thresh = 6.125f;
    +
    181 // Compute both branches and select if N > 1
    +
    182 if constexpr (N == 1) {
    +
    183 if ((abs(t) > thresh).value) { // maximum ulp error = 2.35793
    +
    184 return a * lhs(t);
    +
    185 } else { // maximum ulp error = 2.35002
    +
    186 return a * rhs(t);
    +
    187 }
    +
    188 } else {
    +
    189 return a * select(t > thresh, lhs(t), rhs(t));
    +
    190 }
    +
    191}
    +
    +
    192
    +
    193} // namespace mlx::core::simd
    +
    mlx::core::simd
    Definition accelerate_fp16_simd.h:9
    +
    mlx::core::simd::isnan
    Simd< bool, N > isnan(Simd< T, N > v)
    Definition accelerate_simd.h:141
    +
    mlx::core::simd::N
    constexpr int N
    Definition neon_fp16_simd.h:9
    +
    mlx::core::simd::abs
    Simd< T, N > abs(Simd< T, N > v)
    Definition accelerate_simd.h:112
    +
    mlx::core::simd::erf
    Simd< T, N > erf(Simd< T, N > x)
    Definition math.h:137
    +
    mlx::core::simd::is_complex
    constexpr bool is_complex
    Definition base_simd.h:42
    +
    mlx::core::simd::erfinv
    Simd< T, N > erfinv(Simd< T, N > a_)
    Definition math.h:151
    +
    mlx::core::simd::inf
    constexpr float inf
    Definition math.h:9
    +
    mlx::core::simd::exp
    Simd< T, N > exp(Simd< T, N > in)
    Compute exp(x) in an optimizer friendly way as follows:
    Definition math.h:28
    +
    mlx::core::simd::log
    Simd< float16_t, N > log(Simd< float16_t, N > v)
    Definition accelerate_fp16_simd.h:37
    +
    mlx::core::simd::floor
    Simd< T, N > floor(Simd< T, N > v)
    Definition accelerate_simd.h:113
    +
    mlx::core::simd::fma
    Simd< T, N > fma(Simd< T, N > x, Simd< T, N > y, U z)
    Definition accelerate_simd.h:264
    +
    mlx::core::simd::cos
    Simd< T, N > cos(Simd< T, N > x)
    Definition math.h:128
    +
    mlx::core::simd::sin
    Simd< T, N > sin(Simd< T, N > x)
    Definition math.h:119
    +
    mlx::core::simd::sincos
    Simd< T, N > sincos(Simd< T, N > in)
    Definition math.h:63
    +
    mlx::core::simd::recip
    Simd< T, N > recip(Simd< T, N > v)
    Definition accelerate_simd.h:131
    +
    mlx::core::simd::select
    Simd< T1, N > select(Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)
    Definition accelerate_simd.h:231
    +
    mlx::core::simd::Simd
    Definition accelerate_simd.h:55
    +
    mlx::core::simd::Simd::value
    asd::Vector< scalar_t, N >::packed_t value
    Definition accelerate_simd.h:80
    + +
    +
    + + + + diff --git a/docs/build/html/matmul_8h.html b/docs/build/html/matmul_8h.html index 1c55eb6c1..91e797560 100644 --- a/docs/build/html/matmul_8h.html +++ b/docs/build/html/matmul_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/matmul.h File Reference @@ -52,7 +52,7 @@
    - +
    @@ -110,7 +110,7 @@ $(function(){initNavTree('binary__ops_8h.html',''); initResizable(true); });
    #include <metal_integer>
    #include <metal_math>
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    @@ -172,8 +172,8 @@ Classes diff --git a/docs/build/html/binary__ops_8h.js b/docs/build/html/metal_2kernels_2binary__ops_8h.js similarity index 97% rename from docs/build/html/binary__ops_8h.js rename to docs/build/html/metal_2kernels_2binary__ops_8h.js index fb22d934e..f1406d2c2 100644 --- a/docs/build/html/binary__ops_8h.js +++ b/docs/build/html/metal_2kernels_2binary__ops_8h.js @@ -1,4 +1,4 @@ -var binary__ops_8h = +var metal_2kernels_2binary__ops_8h = [ [ "Add", "struct_add.html", "struct_add" ], [ "FloorDivide", "struct_floor_divide.html", "struct_floor_divide" ], diff --git a/docs/build/html/binary__ops_8h_source.html b/docs/build/html/metal_2kernels_2binary__ops_8h_source.html similarity index 99% rename from docs/build/html/binary__ops_8h_source.html rename to docs/build/html/metal_2kernels_2binary__ops_8h_source.html index 6f818e693..49b1d5bae 100644 --- a/docs/build/html/binary__ops_8h_source.html +++ b/docs/build/html/metal_2kernels_2binary__ops_8h_source.html @@ -3,7 +3,7 @@ - +MLX: mlx/backend/metal/kernels/binary_ops.h Source File @@ -52,7 +52,7 @@

    Classes

    - +
    @@ -105,7 +105,7 @@ $(function(){initNavTree('binary__ops_8h_source.html',''); initResizable(true);
    binary_ops.h
    -Go to the documentation of this file.
    1// Copyright © 2023-2024 Apple Inc.
    +Go to the documentation of this file.
    1// Copyright © 2023-2024 Apple Inc.
    2
    3#pragma once
    4
    @@ -619,8 +619,8 @@ $(function(){initNavTree('binary__ops_8h_source.html',''); initResizable(true); diff --git a/docs/build/html/metal_2kernels_2binary__two_8h.html b/docs/build/html/metal_2kernels_2binary__two_8h.html index 25419281c..13e76e0a2 100644 --- a/docs/build/html/metal_2kernels_2binary__two_8h.html +++ b/docs/build/html/metal_2kernels_2binary__two_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/binary_two.h File Reference @@ -52,7 +52,7 @@
    - +
    @@ -114,7 +114,7 @@ $(function(){initNavTree('gemm_8h.html',''); initResizable(true); }); #include "mlx/backend/metal/kernels/steel/gemm/transforms.h"
    #include "mlx/backend/metal/kernels/steel/utils.h"
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    @@ -135,8 +135,8 @@ Namespaces diff --git a/docs/build/html/gemm_8h.js b/docs/build/html/metal_2kernels_2steel_2gemm_2gemm_8h.js similarity index 88% rename from docs/build/html/gemm_8h.js rename to docs/build/html/metal_2kernels_2steel_2gemm_2gemm_8h.js index 73c57e862..58e80b7bf 100644 --- a/docs/build/html/gemm_8h.js +++ b/docs/build/html/metal_2kernels_2steel_2gemm_2gemm_8h.js @@ -1,4 +1,4 @@ -var gemm_8h = +var metal_2kernels_2steel_2gemm_2gemm_8h = [ [ "mlx::steel::LoopAlignment< M_aligned, N_aligned, K_aligned >", "structmlx_1_1steel_1_1_loop_alignment.html", null ], [ "mlx::steel::GEMMKernel< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, MN_aligned, K_aligned, AccumType, Epilogue >", "structmlx_1_1steel_1_1_g_e_m_m_kernel.html", "structmlx_1_1steel_1_1_g_e_m_m_kernel" ] diff --git a/docs/build/html/gemm_8h_source.html b/docs/build/html/metal_2kernels_2steel_2gemm_2gemm_8h_source.html similarity index 98% rename from docs/build/html/gemm_8h_source.html rename to docs/build/html/metal_2kernels_2steel_2gemm_2gemm_8h_source.html index 7f467a5cb..c9881cd04 100644 --- a/docs/build/html/gemm_8h_source.html +++ b/docs/build/html/metal_2kernels_2steel_2gemm_2gemm_8h_source.html @@ -3,7 +3,7 @@ - +MLX: mlx/backend/metal/kernels/steel/gemm/gemm.h Source File @@ -52,7 +52,7 @@

    Classes

    - +
    @@ -105,7 +105,7 @@ $(function(){initNavTree('gemm_8h_source.html',''); initResizable(true); });
    gemm.h
    -Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    +Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    2
    3#pragma once
    4
    @@ -123,7 +123,7 @@ $(function(){initNavTree('gemm_8h_source.html',''); initResizable(true); });
    18namespace steel {
    19
    20template <bool M_aligned, bool N_aligned, bool K_aligned>
    -
    21struct LoopAlignment {};
    +
    21struct LoopAlignment {};
    22
    23template <
    24 typename T,
    @@ -139,8 +139,7 @@ $(function(){initNavTree('gemm_8h_source.html',''); initResizable(true); });
    34 bool K_aligned,
    35 typename AccumType = typename AccumHelper<T>::accum_type,
    36 typename Epilogue = TransformNone<U, AccumType>>
    -
    -
    37struct GEMMKernel {
    +
    37struct GEMMKernel {
    38 STEEL_CONST short tgp_padding_a = 16 / sizeof(T);
    39 STEEL_CONST short tgp_padding_b = 16 / sizeof(T);
    40 STEEL_CONST short tgp_mem_size_a =
    @@ -398,7 +397,6 @@ $(function(){initNavTree('gemm_8h_source.html',''); initResizable(true); });
    291 }
    292};
    -
    293
    294} // namespace steel
    295} // namespace mlx
    @@ -414,8 +412,8 @@ $(function(){initNavTree('gemm_8h_source.html',''); initResizable(true); });
    STEEL_CONST
    #define STEEL_CONST
    Definition defines.h:3
    mlx::steel::AccumHelper::accum_type
    float accum_type
    Definition transforms.h:57
    mlx::steel::BlockLoader
    Definition loader.h:25
    -
    mlx::steel::BlockMMA
    Definition mma.h:377
    -
    mlx::steel::GEMMKernel
    Definition gemm.h:37
    +
    mlx::steel::BlockMMA
    Definition mma.h:449
    +
    mlx::steel::GEMMKernel
    Definition attn.h:38
    mlx::steel::GEMMKernel::run
    static METAL_FUNC void run(const device T *A, const device T *B, device U *D, const constant GEMMParams *params, threadgroup T *As, threadgroup T *Bs, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)
    Definition gemm.h:140
    mlx::steel::GEMMKernel::tgp_mem_size_b
    STEEL_CONST short tgp_mem_size_b
    Definition attn.h:43
    mlx::steel::GEMMKernel::tgp_mem_size
    STEEL_CONST short tgp_mem_size
    Definition attn.h:45
    @@ -428,15 +426,15 @@ $(function(){initNavTree('gemm_8h_source.html',''); initResizable(true); });
    mlx::steel::GEMMKernel::tgp_padding_a
    STEEL_CONST short tgp_padding_a
    Definition attn.h:39
    mlx::steel::GEMMKernel::mma_t
    BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, AccumType, Epilogue > mma_t
    Definition attn.h:63
    mlx::steel::GEMMParams
    Definition params.h:12
    -
    mlx::steel::LoopAlignment
    Definition gemm.h:21
    +
    mlx::steel::LoopAlignment
    Definition attn.h:22
    mlx::steel::TransformNone
    Definition transforms.h:15
    diff --git a/docs/build/html/metal_2kernels_2ternary_8h.html b/docs/build/html/metal_2kernels_2ternary_8h.html index 5f66ea9a1..ce1ad4d5d 100644 --- a/docs/build/html/metal_2kernels_2ternary_8h.html +++ b/docs/build/html/metal_2kernels_2ternary_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/ternary.h File Reference @@ -52,7 +52,7 @@
    - +
    @@ -112,7 +112,7 @@ $(function(){initNavTree('unary__ops_8h.html',''); initResizable(true); }); #include "mlx/backend/metal/kernels/erf.h"
    #include "mlx/backend/metal/kernels/expm1f.h"
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    @@ -190,8 +190,8 @@ Classes diff --git a/docs/build/html/unary__ops_8h.js b/docs/build/html/metal_2kernels_2unary__ops_8h.js similarity index 98% rename from docs/build/html/unary__ops_8h.js rename to docs/build/html/metal_2kernels_2unary__ops_8h.js index f8523336e..18897c02c 100644 --- a/docs/build/html/unary__ops_8h.js +++ b/docs/build/html/metal_2kernels_2unary__ops_8h.js @@ -1,4 +1,4 @@ -var unary__ops_8h = +var metal_2kernels_2unary__ops_8h = [ [ "Abs", "struct_abs.html", "struct_abs" ], [ "ArcCos", "struct_arc_cos.html", "struct_arc_cos" ], diff --git a/docs/build/html/unary__ops_8h_source.html b/docs/build/html/metal_2kernels_2unary__ops_8h_source.html similarity index 99% rename from docs/build/html/unary__ops_8h_source.html rename to docs/build/html/metal_2kernels_2unary__ops_8h_source.html index 93653e7a2..0f8e37469 100644 --- a/docs/build/html/unary__ops_8h_source.html +++ b/docs/build/html/metal_2kernels_2unary__ops_8h_source.html @@ -3,7 +3,7 @@ - +MLX: mlx/backend/metal/kernels/unary_ops.h Source File @@ -52,7 +52,7 @@

    Classes

    - +
    @@ -105,7 +105,7 @@ $(function(){initNavTree('unary__ops_8h_source.html',''); initResizable(true); }
    unary_ops.h
    -Go to the documentation of this file.
    1// Copyright © 2023-2024 Apple Inc.
    +Go to the documentation of this file.
    1// Copyright © 2023-2024 Apple Inc.
    2
    3#pragma once
    4
    @@ -116,7 +116,7 @@ $(function(){initNavTree('unary__ops_8h_source.html',''); initResizable(true); }
    9#include "mlx/backend/metal/kernels/expm1f.h"
    10
    11namespace {
    -
    12constant float inf = metal::numeric_limits<float>::infinity();
    +
    12constant float inf = metal::numeric_limits<float>::infinity();
    13}
    14
    @@ -752,6 +752,7 @@ $(function(){initNavTree('unary__ops_8h_source.html',''); initResizable(true); }
    metal::abs
    METAL_FUNC bfloat16_t abs(bfloat16_t x)
    Definition bf16_math.h:232
    metal::exp
    METAL_FUNC bfloat16_t exp(bfloat16_t x)
    Definition bf16_math.h:232
    metal::ceil
    METAL_FUNC bfloat16_t ceil(bfloat16_t x)
    Definition bf16_math.h:232
    +
    mlx::core::simd::inf
    constexpr float inf
    Definition math.h:9
    Abs
    Definition unary_ops.h:15
    Abs::operator()
    uint8_t operator()(uint8_t x)
    Definition unary_ops.h:21
    Abs::operator()
    uint32_t operator()(uint32_t x)
    Definition unary_ops.h:29
    @@ -862,8 +863,8 @@ $(function(){initNavTree('unary__ops_8h_source.html',''); initResizable(true); } diff --git a/docs/build/html/metal_2reduce_8h.html b/docs/build/html/metal_2reduce_8h.html index e13aaa485..7ec897781 100644 --- a/docs/build/html/metal_2reduce_8h.html +++ b/docs/build/html/metal_2reduce_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/reduce.h File Reference @@ -52,7 +52,7 @@
    - + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Typedefs | +Functions
    +
    mlx::core::distributed::ring Namespace Reference
    +
    +
    + + + + +

    +Typedefs

    using GroupImpl = mlx::core::distributed::detail::GroupImpl
     
    + + + + + +

    +Functions

    bool is_available ()
     
    std::shared_ptr< GroupImpl > init (bool strict=false)
     
    +

    Typedef Documentation

    + +

    ◆ GroupImpl

    + +
    +
    + + + + +
    using mlx::core::distributed::ring::GroupImpl = mlx::core::distributed::detail::GroupImpl
    +
    + +
    +
    +

    Function Documentation

    + +

    ◆ init()

    + +
    +
    + + + + + + + +
    std::shared_ptr< GroupImpl > mlx::core::distributed::ring::init (bool strict = false)
    +
    + +
    +
    + +

    ◆ is_available()

    + +
    +
    + + + + + + + +
    bool mlx::core::distributed::ring::is_available ()
    +
    + +
    +
    +
    +
    + + + + diff --git a/docs/build/html/namespacemlx_1_1core_1_1env.html b/docs/build/html/namespacemlx_1_1core_1_1env.html index 0dabf3846..2f440451e 100644 --- a/docs/build/html/namespacemlx_1_1core_1_1env.html +++ b/docs/build/html/namespacemlx_1_1core_1_1env.html @@ -3,7 +3,7 @@ - + MLX: mlx::core::env Namespace Reference @@ -52,7 +52,7 @@
    - + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Classes | +Functions | +Variables
    +
    mlx::core::simd Namespace Reference
    +
    +
    + + + + + + + + + + + + + + + + + + +

    +Classes

    struct  ScalarT
     
    struct  ScalarT< bool, N >
     
    struct  ScalarT< int64_t, N >
     
    struct  ScalarT< int8_t, N >
     
    struct  ScalarT< uint64_t, N >
     
    struct  Simd
     
    struct  Simd< float16_t, N >
     
    struct  Simd< T, 1 >
     
    + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

    +Functions

    template<>
    Simd< float16_t, N > acos (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > acosh (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > asin (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > asinh (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > atan (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > atanh (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > cosh (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > expm1 (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > log (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > log2 (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > log10 (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > log1p (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > sinh (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > tan (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > tanh (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > atan2 (Simd< float16_t, N > x, Simd< float16_t, N > y)
     
    template<>
    Simd< float16_t, N > remainder (Simd< float16_t, N > x, Simd< float16_t, N > y)
     
    template<>
    Simd< float16_t, N > pow (Simd< float16_t, N > x, Simd< float16_t, N > y)
     
    template<typename T, int N>
    Simd< T, N > abs (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > floor (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > acos (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > acosh (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > asin (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > asinh (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > atan (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > atanh (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > ceil (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > cosh (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > expm1 (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > log (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > log2 (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > log10 (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > log1p (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > rint (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > sinh (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > sqrt (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > rsqrt (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > recip (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > tan (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > tanh (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< T, N > operator- (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< bool, N > isnan (Simd< T, N > v)
     
    template<typename T, int N>
    Simd< bool, N > operator! (Simd< T, N > v)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator+ (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator+ (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator+ (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator- (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator- (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator- (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator/ (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator/ (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator/ (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator* (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator* (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator* (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator<< (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator<< (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator<< (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator>> (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator>> (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator>> (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator| (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator| (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator| (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator^ (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator^ (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator^ (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator& (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator& (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator& (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator&& (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator&& (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator&& (Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, typename U, int N>
    Simd< T, N > operator|| (Simd< T, N > x, U y)
     
    template<typename T1, typename T2, int N>
    Simd< T2, N > operator|| (T1 x, Simd< T2, N > y)
     
    template<typename T1, typename T2, int N>
    Simd< T1, N > operator|| (Simd< T1, N > x, Simd< T2, N > y)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator> (Simd< T, N > a, U b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator> (T a, Simd< U, N > b)
     
    template<int N, typename T1, typename T2>
    Simd< bool, N > operator> (Simd< T1, N > a, Simd< T2, N > b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator< (Simd< T, N > a, U b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator< (T a, Simd< U, N > b)
     
    template<int N, typename T1, typename T2>
    Simd< bool, N > operator< (Simd< T1, N > a, Simd< T2, N > b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator>= (Simd< T, N > a, U b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator>= (T a, Simd< U, N > b)
     
    template<int N, typename T1, typename T2>
    Simd< bool, N > operator>= (Simd< T1, N > a, Simd< T2, N > b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator<= (Simd< T, N > a, U b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator<= (T a, Simd< U, N > b)
     
    template<int N, typename T1, typename T2>
    Simd< bool, N > operator<= (Simd< T1, N > a, Simd< T2, N > b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator== (Simd< T, N > a, U b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator== (T a, Simd< U, N > b)
     
    template<int N, typename T1, typename T2>
    Simd< bool, N > operator== (Simd< T1, N > a, Simd< T2, N > b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator!= (Simd< T, N > a, U b)
     
    template<int N, typename T, typename U>
    Simd< bool, N > operator!= (T a, Simd< U, N > b)
     
    template<int N, typename T1, typename T2>
    Simd< bool, N > operator!= (Simd< T1, N > a, Simd< T2, N > b)
     
    template<typename T, int N>
    Simd< T, N > atan2 (Simd< T, N > a, Simd< T, N > b)
     
    template<typename T, int N>
    Simd< T, N > maximum (Simd< T, N > a, Simd< T, N > b)
     
    template<typename T, int N>
    Simd< T, N > minimum (Simd< T, N > a, Simd< T, N > b)
     
    template<typename T, int N>
    Simd< T, N > remainder (Simd< T, N > a, Simd< T, N > b)
     
    template<typename MaskT, typename T1, typename T2, int N>
    Simd< T1, N > select (Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)
     
    template<typename T, int N>
    Simd< T, N > pow (Simd< T, N > base, Simd< T, N > exp)
     
    template<typename T, int N>
    Simd< T, N > clamp (Simd< T, N > v, Simd< T, N > min, Simd< T, N > max)
     
    template<typename T, typename U, int N>
    Simd< T, N > fma (Simd< T, N > x, Simd< T, N > y, U z)
     
    template<typename T, int N>
    bool all (Simd< T, N > x)
     
    template<typename T, int N>
    bool any (Simd< T, N > x)
     
    template<typename T, int N>
    T sum (Simd< T, N > x)
     
    template<typename T, int N>
    T max (Simd< T, N > x)
     
    template<typename T, int N>
    T min (Simd< T, N > x)
     
    template<typename T, int N>
    T prod (Simd< T, N > x)
     
    template<typename T, int N>
    Simd< T, N > load (const T *x)
     
    template<typename T, int N>
    void store (T *dst, Simd< T, N > x)
     
    template<typename T>
    Simd< T, 1 > rint (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > rsqrt (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > recip (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > operator- (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > operator! (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > abs (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > acos (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > acosh (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > asin (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > asinh (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > atan (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > atanh (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > ceil (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > conj (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > cosh (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > expm1 (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > floor (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > log (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > log2 (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > log10 (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > log1p (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > sinh (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > sqrt (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > tan (Simd< T, 1 > in)
     
    template<typename T>
    Simd< T, 1 > tanh (Simd< T, 1 > in)
     
    template<typename T>
    auto real (Simd< T, 1 > in) -> Simd< decltype(std::real(in.value)), 1 >
     
    template<typename T>
    auto imag (Simd< T, 1 > in) -> Simd< decltype(std::imag(in.value)), 1 >
     
    template<typename T>
    Simd< bool, 1 > isnan (Simd< T, 1 > in)
     
    template<typename T1, typename T2>
    auto operator+ (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value+b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator+ (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a+b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator+ (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value+b), 1 >
     
    template<typename T1, typename T2>
    auto operator- (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value - b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator- (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a - b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator- (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value - b), 1 >
     
    template<typename T1, typename T2>
    auto operator* (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value *b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator* (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a *b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator* (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value *b), 1 >
     
    template<typename T1, typename T2>
    auto operator/ (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value/b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator/ (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a/b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator/ (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value/b), 1 >
     
    template<typename T1, typename T2>
    auto operator<< (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value<< b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator<< (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a<< b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator<< (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value<< b), 1 >
     
    template<typename T1, typename T2>
    auto operator>> (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value > > b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator>> (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a > > b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator>> (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value > > b), 1 >
     
    template<typename T1, typename T2>
    auto operator| (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value|b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator| (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a|b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator| (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value|b), 1 >
     
    template<typename T1, typename T2>
    auto operator^ (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value ^ b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator^ (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a ^ b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator^ (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value ^ b), 1 >
     
    template<typename T1, typename T2>
    auto operator& (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value &b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator& (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a &b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator& (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value &b), 1 >
     
    template<typename T1, typename T2>
    auto operator&& (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value &&b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator&& (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a &&b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator&& (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value &&b), 1 >
     
    template<typename T1, typename T2>
    auto operator|| (Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value||b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator|| (T1 a, Simd< T2, 1 > b) -> Simd< decltype(a||b.value), 1 >
     
    template<typename T1, typename T2>
    auto operator|| (Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value||b), 1 >
     
    template<typename T>
    Simd< T, 1 > remainder (Simd< T, 1 > a_, Simd< T, 1 > b_)
     
    template<typename T>
    Simd< T, 1 > maximum (Simd< T, 1 > a_, Simd< T, 1 > b_)
     
    template<typename T>
    Simd< T, 1 > minimum (Simd< T, 1 > a_, Simd< T, 1 > b_)
     
    template<typename T>
    Simd< T, 1 > pow (Simd< T, 1 > a, Simd< T, 1 > b)
     
    template<typename T>
    Simd< T, 1 > atan2 (Simd< T, 1 > a, Simd< T, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator> (Simd< T1, 1 > a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator> (T1 a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator> (Simd< T1, 1 > a, T2 b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator< (Simd< T1, 1 > a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator< (T1 a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator< (Simd< T1, 1 > a, T2 b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator>= (Simd< T1, 1 > a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator>= (T1 a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator>= (Simd< T1, 1 > a, T2 b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator<= (Simd< T1, 1 > a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator<= (T1 a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator<= (Simd< T1, 1 > a, T2 b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator== (Simd< T1, 1 > a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator== (T1 a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator== (Simd< T1, 1 > a, T2 b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator!= (Simd< T1, 1 > a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator!= (T1 a, Simd< T2, 1 > b)
     
    template<typename T1, typename T2>
    Simd< bool, 1 > operator!= (Simd< T1, 1 > a, T2 b)
     
    template<typename MaskT, typename T>
    Simd< T, 1 > select (Simd< MaskT, 1 > mask, Simd< T, 1 > x, Simd< T, 1 > y)
     
    template<typename T>
    Simd< T, 1 > clamp (Simd< T, 1 > v, Simd< T, 1 > min, Simd< T, 1 > max)
     
    template<typename T, typename U>
    Simd< T, 1 > fma (Simd< T, 1 > x, Simd< T, 1 > y, U z)
     
    template<typename T>
    T max (Simd< T, 1 > x)
     
    template<typename T>
    T min (Simd< T, 1 > x)
     
    template<typename T>
    T sum (Simd< T, 1 > x)
     
    template<typename T>
    T prod (Simd< T, 1 > x)
     
    template<typename T>
    bool any (Simd< T, 1 > x)
     
    template<typename T>
    bool all (Simd< T, 1 > x)
     
    template<typename T, int N>
    Simd< T, N > exp (Simd< T, N > in)
     Compute exp(x) in an optimizer friendly way as follows:
     
    template<bool Sine, typename T, int N>
    Simd< T, N > sincos (Simd< T, N > in)
     
    template<typename T, int N>
    Simd< T, N > sin (Simd< T, N > x)
     
    template<typename T, int N>
    Simd< T, N > cos (Simd< T, N > x)
     
    template<typename T, int N>
    Simd< T, N > erf (Simd< T, N > x)
     
    template<typename T, int N>
    Simd< T, N > erfinv (Simd< T, N > a_)
     
    Simd< float16_t, N > abs (Simd< float16_t, N > a)
     
    Simd< float16_t, N > ceil (Simd< float16_t, N > a)
     
    Simd< float16_t, N > floor (Simd< float16_t, N > a)
     
    Simd< float16_t, N > sqrt (Simd< float16_t, N > a)
     
    Simd< float16_t, N > rsqrt (Simd< float16_t, N > a)
     
    Simd< float16_t, N > recip (Simd< float16_t, N > a)
     
    Simd< float16_t, N > rint (Simd< float16_t, N > a)
     
    Simd< float16_t, N > operator! (Simd< float16_t, N > v)
     
    Simd< float16_t, N > operator- (Simd< float16_t, N > v)
     
    Simd< float16_t, N > maximum (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > maximum (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > maximum (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > minimum (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > minimum (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > minimum (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > operator+ (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > operator+ (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > operator+ (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > operator- (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > operator- (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > operator- (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > operator* (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > operator* (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > operator* (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > operator/ (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > operator/ (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > operator/ (T a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > operator== (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > operator== (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > operator== (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > operator>= (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > operator>= (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > operator>= (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > operator<= (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > operator<= (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > operator<= (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > operator> (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > operator> (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > operator> (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > operator< (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > operator< (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > operator< (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > operator!= (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > operator!= (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > operator!= (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > operator|| (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > operator|| (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > operator|| (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > operator&& (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > operator&& (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > operator&& (T a, Simd< float16_t, N > b)
     
    template<>
    Simd< bool, N > isnan (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > clamp (Simd< float16_t, N > v, Simd< float16_t, N > min, Simd< float16_t, N > max)
     
    template<typename T>
    Simd< float16_t, N > fma (Simd< float16_t, N > x, Simd< float16_t, N > y, T z)
     
    template<typename MaskT>
    Simd< float16_t, N > select (Simd< MaskT, N > mask, Simd< float16_t, N > x, Simd< float16_t, N > y)
     
    float16_t max (Simd< float16_t, N > x)
     
    float16_t min (Simd< float16_t, N > x)
     
    float16_t sum (Simd< float16_t, N > x)
     
    float16_t prod (Simd< float16_t, N > x)
     
    + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

    +Variables

    template<>
    static constexpr int max_size< float16_t > = N
     
    template<>
    static constexpr int max_size< int8_t > = 16
     
    template<>
    static constexpr int max_size< int16_t > = 16
     
    template<>
    static constexpr int max_size< int > = 8
     
    template<>
    static constexpr int max_size< int64_t > = 4
     
    template<>
    static constexpr int max_size< uint8_t > = 16
     
    template<>
    static constexpr int max_size< uint16_t > = 16
     
    template<>
    static constexpr int max_size< uint32_t > = 8
     
    template<>
    static constexpr int max_size< uint64_t > = 4
     
    template<>
    static constexpr int max_size< float > = 8
     
    template<>
    static constexpr int max_size< double > = 4
     
    template<typename T>
    static constexpr int max_size = 1
     
    template<typename, typename = void>
    constexpr bool is_complex = false
     
    template<typename T>
    constexpr bool is_complex< T, std::void_t< decltype(std::declval< T >().real())> >
     
    constexpr float inf = std::numeric_limits<float>::infinity()
     
    constexpr int N = 8
     
    +

    Function Documentation

    + +

    ◆ abs() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::abs (Simd< float16_t, N > a)
    +     		+inline
    +
    + +
    +
    + +

    ◆ abs() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::abs (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ abs() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::abs (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ acos() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::acos (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ acos() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::acos (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ acos() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::acos (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ acosh() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::acosh (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ acosh() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::acosh (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ acosh() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::acosh (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ all() [1/2]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    bool mlx::core::simd::all (Simd< T, 1 > x)
    +
    + +
    +
    + +

    ◆ all() [2/2]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    bool mlx::core::simd::all (Simd< T, N > x)
    +
    + +
    +
    + +

    ◆ any() [1/2]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    bool mlx::core::simd::any (Simd< T, 1 > x)
    +
    + +
    +
    + +

    ◆ any() [2/2]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    bool mlx::core::simd::any (Simd< T, N > x)
    +
    + +
    +
    + +

    ◆ asin() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::asin (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ asin() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::asin (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ asin() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::asin (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ asinh() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::asinh (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ asinh() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::asinh (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ asinh() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::asinh (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ atan() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::atan (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ atan() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::atan (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ atan() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::atan (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ atan2() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::atan2 (Simd< float16_t, N > x,
    Simd< float16_t, N > y )
    +     		+inline
    +
    + +
    +
    + +

    ◆ atan2() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< T, 1 > mlx::core::simd::atan2 (Simd< T, 1 > a,
    Simd< T, 1 > b )
    +
    + +
    +
    + +

    ◆ atan2() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::atan2 (Simd< T, N > a,
    Simd< T, N > b )
    +
    + +
    +
    + +

    ◆ atanh() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::atanh (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ atanh() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::atanh (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ atanh() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::atanh (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ ceil() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::ceil (Simd< float16_t, N > a)
    +     		+inline
    +
    + +
    +
    + +

    ◆ ceil() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::ceil (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ ceil() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::ceil (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ clamp() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::clamp (Simd< float16_t, N > v,
    Simd< float16_t, N > min,
    Simd< float16_t, N > max )
    +     		+inline
    +
    + +
    +
    + +

    ◆ clamp() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + + + + + + +
    Simd< T, 1 > mlx::core::simd::clamp (Simd< T, 1 > v,
    Simd< T, 1 > min,
    Simd< T, 1 > max )
    +
    + +
    +
    + +

    ◆ clamp() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::clamp (Simd< T, N > v,
    Simd< T, N > min,
    Simd< T, N > max )
    +
    + +
    +
    + +

    ◆ conj()

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::conj (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ cos()

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::cos (Simd< T, N > x)
    +
    + +
    +
    + +

    ◆ cosh() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::cosh (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ cosh() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::cosh (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ cosh() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::cosh (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ erf()

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::erf (Simd< T, N > x)
    +
    + +
    +
    + +

    ◆ erfinv()

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::erfinv (Simd< T, N > a_)
    +
    + +
    +
    + +

    ◆ exp()

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::exp (Simd< T, N > in)
    +
    + +

    Compute exp(x) in an optimizer friendly way as follows:

    +

    First change the problem to computing 2**y where y = x / ln(2).

    +

    Now we will compute 2**y as 2**y1 * 2**y2 where y1 is the integer part ipart and y2 is fractional part. For the integer part we perform bit shifting and for the fractional part we use a polynomial approximation.

    +

    The algorithm and constants of the polynomial taken from https://github.com/akohlmey/fastermath/blob/master/src/exp.c which took them from Cephes math library.

    +

    Note: The implementation below is a general fast exp. There could be faster implementations for numbers strictly < 0.

    + +
    +
    + +

    ◆ expm1() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::expm1 (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ expm1() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::expm1 (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ expm1() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::expm1 (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ floor() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::floor (Simd< float16_t, N > a)
    +     		+inline
    +
    + +
    +
    + +

    ◆ floor() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::floor (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ floor() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::floor (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ fma() [1/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::fma (Simd< float16_t, N > x,
    Simd< float16_t, N > y,
    T z )
    +
    + +
    +
    + +

    ◆ fma() [2/3]

    + +
    +
    +
    +template<typename T, typename U>
    + + + + + + + + + + + + + + + + +
    Simd< T, 1 > mlx::core::simd::fma (Simd< T, 1 > x,
    Simd< T, 1 > y,
    U z )
    +
    + +
    +
    + +

    ◆ fma() [3/3]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::fma (Simd< T, N > x,
    Simd< T, N > y,
    U z )
    +
    + +
    +
    + +

    ◆ imag()

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    auto mlx::core::simd::imag (Simd< T, 1 > in) -> Simd<decltype(std::imag(in.value)), 1>
    +
    + +
    +
    + +

    ◆ isnan() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< bool, N > mlx::core::simd::isnan (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ isnan() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::isnan (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ isnan() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< bool, N > mlx::core::simd::isnan (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ load()

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::load (const T * x)
    +
    + +
    +
    + +

    ◆ log() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::log (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ log() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::log (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ log() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::log (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ log10() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::log10 (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ log10() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::log10 (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ log10() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::log10 (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ log1p() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::log1p (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ log1p() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::log1p (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ log1p() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::log1p (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ log2() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::log2 (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ log2() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::log2 (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ log2() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::log2 (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ max() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    float16_t mlx::core::simd::max (Simd< float16_t, N > x)
    +     		+inline
    +
    + +
    +
    + +

    ◆ max() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    T mlx::core::simd::max (Simd< T, 1 > x)
    +
    + +
    +
    + +

    ◆ max() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    T mlx::core::simd::max (Simd< T, N > x)
    +
    + +
    +
    + +

    ◆ maximum() [1/5]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::maximum (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ maximum() [2/5]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::maximum (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ maximum() [3/5]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< T, 1 > mlx::core::simd::maximum (Simd< T, 1 > a_,
    Simd< T, 1 > b_ )
    +
    + +
    +
    + +

    ◆ maximum() [4/5]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::maximum (Simd< T, N > a,
    Simd< T, N > b )
    +
    + +
    +
    + +

    ◆ maximum() [5/5]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::maximum (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ min() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    float16_t mlx::core::simd::min (Simd< float16_t, N > x)
    +     		+inline
    +
    + +
    +
    + +

    ◆ min() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    T mlx::core::simd::min (Simd< T, 1 > x)
    +
    + +
    +
    + +

    ◆ min() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    T mlx::core::simd::min (Simd< T, N > x)
    +
    + +
    +
    + +

    ◆ minimum() [1/5]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::minimum (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ minimum() [2/5]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::minimum (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ minimum() [3/5]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< T, 1 > mlx::core::simd::minimum (Simd< T, 1 > a_,
    Simd< T, 1 > b_ )
    +
    + +
    +
    + +

    ◆ minimum() [4/5]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::minimum (Simd< T, N > a,
    Simd< T, N > b )
    +
    + +
    +
    + +

    ◆ minimum() [5/5]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::minimum (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator!() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator! (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator!() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::operator! (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ operator!() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator! (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ operator!=() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator!= (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator!=() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator!= (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator!=() [3/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator!= (Simd< T, N > a,
    U b )
    +
    + +
    +
    + +

    ◆ operator!=() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator!= (Simd< T1, 1 > a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator!=() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator!= (Simd< T1, 1 > a,
    T2 b )
    +
    + +
    +
    + +

    ◆ operator!=() [6/9]

    + +
    +
    +
    +template<int N, typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator!= (Simd< T1, N > a,
    Simd< T2, N > b )
    +
    + +
    +
    + +

    ◆ operator!=() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator!= (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator!=() [8/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator!= (T a,
    Simd< U, N > b )
    +
    + +
    +
    + +

    ◆ operator!=() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator!= (T1 a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator&() [1/6]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator& (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator&() [2/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator& (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value & b.value), 1>
    +
    + +
    +
    + +

    ◆ operator&() [3/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator& (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value & b), 1>
    +
    + +
    +
    + +

    ◆ operator&() [4/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator& (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator&() [5/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator& (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a & b.value), 1>
    +
    + +
    +
    + +

    ◆ operator&() [6/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator& (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator&&() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator&& (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator&&() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator&& (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator&&() [3/9]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator&& (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator&&() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator&& (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value && b.value), 1>
    +
    + +
    +
    + +

    ◆ operator&&() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator&& (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value && b), 1>
    +
    + +
    +
    + +

    ◆ operator&&() [6/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator&& (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator&&() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator&& (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator&&() [8/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator&& (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a && b.value), 1>
    +
    + +
    +
    + +

    ◆ operator&&() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator&& (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator*() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator* (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator*() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator* (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator*() [3/9]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator* (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator*() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator* (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value * b.value), 1>
    +
    + +
    +
    + +

    ◆ operator*() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator* (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value * b), 1>
    +
    + +
    +
    + +

    ◆ operator*() [6/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator* (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator*() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator* (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator*() [8/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator* (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a * b.value), 1>
    +
    + +
    +
    + +

    ◆ operator*() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator* (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator+() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator+ (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator+() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator+ (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator+() [3/9]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator+ (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator+() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator+ (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value + b.value), 1>
    +
    + +
    +
    + +

    ◆ operator+() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator+ (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value + b), 1>
    +
    + +
    +
    + +

    ◆ operator+() [6/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator+ (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator+() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator+ (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator+() [8/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator+ (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a + b.value), 1>
    +
    + +
    +
    + +

    ◆ operator+() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator+ (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator-() [1/12]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator- (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator-() [2/12]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator- (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator-() [3/12]

    + +
    +
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator- (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator-() [4/12]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::operator- (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ operator-() [5/12]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::operator- (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ operator-() [6/12]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator- (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator-() [7/12]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator- (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value - b.value), 1>
    +
    + +
    +
    + +

    ◆ operator-() [8/12]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator- (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value - b), 1>
    +
    + +
    +
    + +

    ◆ operator-() [9/12]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator- (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator-() [10/12]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator- (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator-() [11/12]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator- (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a - b.value), 1>
    +
    + +
    +
    + +

    ◆ operator-() [12/12]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator- (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator/() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator/ (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator/() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator/ (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator/() [3/9]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator/ (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator/() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator/ (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value / b.value), 1>
    +
    + +
    +
    + +

    ◆ operator/() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator/ (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value / b), 1>
    +
    + +
    +
    + +

    ◆ operator/() [6/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator/ (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator/() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator/ (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator/() [8/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator/ (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a / b.value), 1>
    +
    + +
    +
    + +

    ◆ operator/() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator/ (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator<() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator< (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator<() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator< (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator<() [3/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator< (Simd< T, N > a,
    U b )
    +
    + +
    +
    + +

    ◆ operator<() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator< (Simd< T1, 1 > a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator<() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator< (Simd< T1, 1 > a,
    T2 b )
    +
    + +
    +
    + +

    ◆ operator<() [6/9]

    + +
    +
    +
    +template<int N, typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator< (Simd< T1, N > a,
    Simd< T2, N > b )
    +
    + +
    +
    + +

    ◆ operator<() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator< (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator<() [8/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator< (T a,
    Simd< U, N > b )
    +
    + +
    +
    + +

    ◆ operator<() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator< (T1 a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator<<() [1/6]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator<< (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator<<() [2/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator<< (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value << b.value), 1>
    +
    + +
    +
    + +

    ◆ operator<<() [3/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator<< (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value << b), 1>
    +
    + +
    +
    + +

    ◆ operator<<() [4/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator<< (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator<<() [5/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator<< (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a << b.value), 1>
    +
    + +
    +
    + +

    ◆ operator<<() [6/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator<< (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator<=() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator<= (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator<=() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator<= (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator<=() [3/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator<= (Simd< T, N > a,
    U b )
    +
    + +
    +
    + +

    ◆ operator<=() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator<= (Simd< T1, 1 > a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator<=() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator<= (Simd< T1, 1 > a,
    T2 b )
    +
    + +
    +
    + +

    ◆ operator<=() [6/9]

    + +
    +
    +
    +template<int N, typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator<= (Simd< T1, N > a,
    Simd< T2, N > b )
    +
    + +
    +
    + +

    ◆ operator<=() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator<= (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator<=() [8/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator<= (T a,
    Simd< U, N > b )
    +
    + +
    +
    + +

    ◆ operator<=() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator<= (T1 a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator==() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator== (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator==() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator== (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator==() [3/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator== (Simd< T, N > a,
    U b )
    +
    + +
    +
    + +

    ◆ operator==() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator== (Simd< T1, 1 > a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator==() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator== (Simd< T1, 1 > a,
    T2 b )
    +
    + +
    +
    + +

    ◆ operator==() [6/9]

    + +
    +
    +
    +template<int N, typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator== (Simd< T1, N > a,
    Simd< T2, N > b )
    +
    + +
    +
    + +

    ◆ operator==() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator== (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator==() [8/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator== (T a,
    Simd< U, N > b )
    +
    + +
    +
    + +

    ◆ operator==() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator== (T1 a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator>() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator> (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator>() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator> (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator>() [3/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator> (Simd< T, N > a,
    U b )
    +
    + +
    +
    + +

    ◆ operator>() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator> (Simd< T1, 1 > a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator>() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator> (Simd< T1, 1 > a,
    T2 b )
    +
    + +
    +
    + +

    ◆ operator>() [6/9]

    + +
    +
    +
    +template<int N, typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator> (Simd< T1, N > a,
    Simd< T2, N > b )
    +
    + +
    +
    + +

    ◆ operator>() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator> (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator>() [8/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator> (T a,
    Simd< U, N > b )
    +
    + +
    +
    + +

    ◆ operator>() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator> (T1 a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator>=() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator>= (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator>=() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator>= (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator>=() [3/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator>= (Simd< T, N > a,
    U b )
    +
    + +
    +
    + +

    ◆ operator>=() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator>= (Simd< T1, 1 > a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator>=() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator>= (Simd< T1, 1 > a,
    T2 b )
    +
    + +
    +
    + +

    ◆ operator>=() [6/9]

    + +
    +
    +
    +template<int N, typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator>= (Simd< T1, N > a,
    Simd< T2, N > b )
    +
    + +
    +
    + +

    ◆ operator>=() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator>= (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator>=() [8/9]

    + +
    +
    +
    +template<int N, typename T, typename U>
    + + + + + + + + + + + +
    Simd< bool, N > mlx::core::simd::operator>= (T a,
    Simd< U, N > b )
    +
    + +
    +
    + +

    ◆ operator>=() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    Simd< bool, 1 > mlx::core::simd::operator>= (T1 a,
    Simd< T2, 1 > b )
    +
    + +
    +
    + +

    ◆ operator>>() [1/6]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator>> (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator>>() [2/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator>> (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value >> b.value), 1>
    +
    + +
    +
    + +

    ◆ operator>>() [3/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator>> (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value >> b), 1>
    +
    + +
    +
    + +

    ◆ operator>>() [4/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator>> (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator>>() [5/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator>> (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a >> b.value), 1>
    +
    + +
    +
    + +

    ◆ operator>>() [6/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator>> (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator^() [1/6]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator^ (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator^() [2/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator^ (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value ^ b.value), 1>
    +
    + +
    +
    + +

    ◆ operator^() [3/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator^ (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value ^ b), 1>
    +
    + +
    +
    + +

    ◆ operator^() [4/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator^ (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator^() [5/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator^ (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a ^ b.value), 1>
    +
    + +
    +
    + +

    ◆ operator^() [6/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator^ (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator|() [1/6]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator| (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator|() [2/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator| (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value | b.value), 1>
    +
    + +
    +
    + +

    ◆ operator|() [3/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator| (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value | b), 1>
    +
    + +
    +
    + +

    ◆ operator|() [4/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator| (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator|() [5/6]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator| (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a | b.value), 1>
    +
    + +
    +
    + +

    ◆ operator|() [6/6]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator| (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator||() [1/9]

    + +
    +
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator|| (Simd< float16_t, N > a,
    Simd< float16_t, N > b )
    +     		+inline
    +
    + +
    +
    + +

    ◆ operator||() [2/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator|| (Simd< float16_t, N > a,
    T b )
    +
    + +
    +
    + +

    ◆ operator||() [3/9]

    + +
    +
    +
    +template<typename T, typename U, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::operator|| (Simd< T, N > x,
    U y )
    +
    + +
    +
    + +

    ◆ operator||() [4/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator|| (Simd< T1, 1 > a,
    Simd< T2, 1 > b ) -> Simd<decltype(a.value || b.value), 1>
    +
    + +
    +
    + +

    ◆ operator||() [5/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator|| (Simd< T1, 1 > a,
    T2 b ) -> Simd<decltype(a.value || b), 1>
    +
    + +
    +
    + +

    ◆ operator||() [6/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::operator|| (Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ operator||() [7/9]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::operator|| (T a,
    Simd< float16_t, N > b )
    +
    + +
    +
    + +

    ◆ operator||() [8/9]

    + +
    +
    +
    +template<typename T1, typename T2>
    + + + + + + + + + + + +
    auto mlx::core::simd::operator|| (T1 a,
    Simd< T2, 1 > b ) -> Simd<decltype(a || b.value), 1>
    +
    + +
    +
    + +

    ◆ operator||() [9/9]

    + +
    +
    +
    +template<typename T1, typename T2, int N>
    + + + + + + + + + + + +
    Simd< T2, N > mlx::core::simd::operator|| (T1 x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ pow() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::pow (Simd< float16_t, N > x,
    Simd< float16_t, N > y )
    +     		+inline
    +
    + +
    +
    + +

    ◆ pow() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< T, 1 > mlx::core::simd::pow (Simd< T, 1 > a,
    Simd< T, 1 > b )
    +
    + +
    +
    + +

    ◆ pow() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::pow (Simd< T, N > base,
    Simd< T, N > exp )
    +
    + +
    +
    + +

    ◆ prod() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    float16_t mlx::core::simd::prod (Simd< float16_t, N > x)
    +     		+inline
    +
    + +
    +
    + +

    ◆ prod() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    T mlx::core::simd::prod (Simd< T, 1 > x)
    +
    + +
    +
    + +

    ◆ prod() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    T mlx::core::simd::prod (Simd< T, N > x)
    +
    + +
    +
    + +

    ◆ real()

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    auto mlx::core::simd::real (Simd< T, 1 > in) -> Simd<decltype(std::real(in.value)), 1>
    +
    + +
    +
    + +

    ◆ recip() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::recip (Simd< float16_t, N > a)
    +     		+inline
    +
    + +
    +
    + +

    ◆ recip() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::recip (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ recip() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::recip (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ remainder() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::remainder (Simd< float16_t, N > x,
    Simd< float16_t, N > y )
    +     		+inline
    +
    + +
    +
    + +

    ◆ remainder() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + + + + + +
    Simd< T, 1 > mlx::core::simd::remainder (Simd< T, 1 > a_,
    Simd< T, 1 > b_ )
    +
    + +
    +
    + +

    ◆ remainder() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + + + + + +
    Simd< T, N > mlx::core::simd::remainder (Simd< T, N > a,
    Simd< T, N > b )
    +
    + +
    +
    + +

    ◆ rint() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::rint (Simd< float16_t, N > a)
    +     		+inline
    +
    + +
    +
    + +

    ◆ rint() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::rint (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ rint() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::rint (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ rsqrt() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::rsqrt (Simd< float16_t, N > a)
    +     		+inline
    +
    + +
    +
    + +

    ◆ rsqrt() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::rsqrt (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ rsqrt() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::rsqrt (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ select() [1/3]

    + +
    +
    +
    +template<typename MaskT, typename T>
    + + + + + + + + + + + + + + + + +
    Simd< T, 1 > mlx::core::simd::select (Simd< MaskT, 1 > mask,
    Simd< T, 1 > x,
    Simd< T, 1 > y )
    +
    + +
    +
    + +

    ◆ select() [2/3]

    + +
    +
    +
    +template<typename MaskT>
    + + + + + + + + + + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::select (Simd< MaskT, N > mask,
    Simd< float16_t, N > x,
    Simd< float16_t, N > y )
    +
    + +
    +
    + +

    ◆ select() [3/3]

    + +
    +
    +
    +template<typename MaskT, typename T1, typename T2, int N>
    + + + + + + + + + + + + + + + + +
    Simd< T1, N > mlx::core::simd::select (Simd< MaskT, N > mask,
    Simd< T1, N > x,
    Simd< T2, N > y )
    +
    + +
    +
    + +

    ◆ sin()

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::sin (Simd< T, N > x)
    +
    + +
    +
    + +

    ◆ sincos()

    + +
    +
    +
    +template<bool Sine, typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::sincos (Simd< T, N > in)
    +
    + +
    +
    + +

    ◆ sinh() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::sinh (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ sinh() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::sinh (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ sinh() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::sinh (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ sqrt() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::sqrt (Simd< float16_t, N > a)
    +     		+inline
    +
    + +
    +
    + +

    ◆ sqrt() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::sqrt (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ sqrt() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::sqrt (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ store()

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + + + + + +
    void mlx::core::simd::store (T * dst,
    Simd< T, N > x )
    +
    + +
    +
    + +

    ◆ sum() [1/3]

    + +
    +
    + + + + + +
    + + + + + + + +
    float16_t mlx::core::simd::sum (Simd< float16_t, N > x)
    +     		+inline
    +
    + +
    +
    + +

    ◆ sum() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    T mlx::core::simd::sum (Simd< T, 1 > x)
    +
    + +
    +
    + +

    ◆ sum() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    T mlx::core::simd::sum (Simd< T, N > x)
    +
    + +
    +
    + +

    ◆ tan() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::tan (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ tan() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::tan (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ tan() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::tan (Simd< T, N > v)
    +
    + +
    +
    + +

    ◆ tanh() [1/3]

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + + + + +
    Simd< float16_t, N > mlx::core::simd::tanh (Simd< float16_t, N > v)
    +     		+inline
    +
    + +
    +
    + +

    ◆ tanh() [2/3]

    + +
    +
    +
    +template<typename T>
    + + + + + + + +
    Simd< T, 1 > mlx::core::simd::tanh (Simd< T, 1 > in)
    +
    + +
    +
    + +

    ◆ tanh() [3/3]

    + +
    +
    +
    +template<typename T, int N>
    + + + + + + + +
    Simd< T, N > mlx::core::simd::tanh (Simd< T, N > v)
    +
    + +
    +
    +

    Variable Documentation

    + +

    ◆ inf

    + +
    +
    + + + + + +
    + + + + +
    float mlx::core::simd::inf = std::numeric_limits<float>::infinity()
    +     		+constexpr
    +
    + +
    +
    + +

    ◆ is_complex

    + +
    +
    +
    +template<typename, typename = void>
    + + + + + +
    + + + + +
    bool mlx::core::simd::is_complex = false
    +     		+constexpr
    +
    + +
    +
    + +

    ◆ is_complex< T, std::void_t< decltype(std::declval< T >().real())> >

    + +
    +
    +
    +template<typename T>
    + + + + + +
    + + + + +
    bool mlx::core::simd::is_complex< T, std::void_t< decltype(std::declval< T >().real())> >
    +     		+constexpr
    +
    +Initial value:
    =
    +
    true
    +
    +
    +
    + +

    ◆ max_size

    + +
    +
    +
    +template<typename T>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size = 1
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< double >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< double > = 4
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< float >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< float > = 8
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< float16_t >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< float16_t > = N
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< int >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< int > = 8
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< int16_t >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< int16_t > = 16
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< int64_t >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< int64_t > = 4
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< int8_t >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< int8_t > = 16
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< uint16_t >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< uint16_t > = 16
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< uint32_t >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< uint32_t > = 8
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< uint64_t >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< uint64_t > = 4
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ max_size< uint8_t >

    + +
    +
    +
    +template<>
    + + + + + +
    + + + + +
    int mlx::core::simd::max_size< uint8_t > = 16
    +     		+staticconstexpr
    +
    + +
    +
    + +

    ◆ N

    + +
    +
    + + + + + +
    + + + + +
    int mlx::core::simd::N = 8
    +     		+constexpr
    +
    + +
    +
    +
    +
    + + + + diff --git a/docs/build/html/namespacemlx_1_1core_1_1simd.js b/docs/build/html/namespacemlx_1_1core_1_1simd.js new file mode 100644 index 000000000..5c108f7c0 --- /dev/null +++ b/docs/build/html/namespacemlx_1_1core_1_1simd.js @@ -0,0 +1,295 @@ +var namespacemlx_1_1core_1_1simd = +[ + [ "ScalarT", "structmlx_1_1core_1_1simd_1_1_scalar_t.html", "structmlx_1_1core_1_1simd_1_1_scalar_t" ], + [ "ScalarT< bool, N >", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4" ], + [ "ScalarT< int64_t, N >", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4" ], + [ "ScalarT< int8_t, N >", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4" ], + [ "ScalarT< uint64_t, N >", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4" ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html", "structmlx_1_1core_1_1simd_1_1_simd" ], + [ "Simd< float16_t, N >", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4" ], + [ "Simd< T, 1 >", "structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html", "structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4" ], + [ "abs", "namespacemlx_1_1core_1_1simd.html#a7696a0628a1c6ccb293ebd6f2328ea48", null ], + [ "abs", "namespacemlx_1_1core_1_1simd.html#a1958f026f26f313d17155ac87ea6eca3", null ], + [ "abs", "namespacemlx_1_1core_1_1simd.html#a4f3cc8b2493586e83fd65640df3b60ad", null ], + [ "acos", "namespacemlx_1_1core_1_1simd.html#ae623449dfa7aab3031aa2f14c1b10a2d", null ], + [ "acos", "namespacemlx_1_1core_1_1simd.html#a3d4f9d08d1902e3d62c6f63d39329dbd", null ], + [ "acos", "namespacemlx_1_1core_1_1simd.html#a85999467c83b07e4fa5f093f7ddf19e1", null ], + [ "acosh", "namespacemlx_1_1core_1_1simd.html#a4f8a64e7742fcd8f759f723a36a7c826", null ], + [ "acosh", "namespacemlx_1_1core_1_1simd.html#a51071c8104494b5bd8097990da3bf943", null ], + [ "acosh", "namespacemlx_1_1core_1_1simd.html#a90092f3826ad3be4b2b1785f7ff4a86b", null ], + [ "all", "namespacemlx_1_1core_1_1simd.html#a4ba3690489c2bf861e22e1175255438c", null ], + [ "all", "namespacemlx_1_1core_1_1simd.html#a5109118acb6766855878b9e8a56b156a", null ], + [ "any", "namespacemlx_1_1core_1_1simd.html#ae3b138b4bbcee0ca70b58a3e2ebd818c", null ], + [ "any", "namespacemlx_1_1core_1_1simd.html#a9d968537ad5ef18630f5afce8453b30e", null ], + [ "asin", "namespacemlx_1_1core_1_1simd.html#a9c7723fc49137394fa817136a7ffb50f", null ], + [ "asin", "namespacemlx_1_1core_1_1simd.html#a75349994f899aecb68553c2247580163", null ], + [ "asin", "namespacemlx_1_1core_1_1simd.html#a069963ffb15f06d1c48258054750dadf", null ], + [ "asinh", "namespacemlx_1_1core_1_1simd.html#aaa76bdf1db09261d84da51d394837f5d", null ], + [ "asinh", "namespacemlx_1_1core_1_1simd.html#a9323e370f6740651ebfd51367985d0e2", null ], + [ "asinh", "namespacemlx_1_1core_1_1simd.html#a7b47a5f370e8e59e1debfa5405e13266", null ], + [ "atan", "namespacemlx_1_1core_1_1simd.html#aa7550a1210e50c996d0db84034b8a22e", null ], + [ "atan", "namespacemlx_1_1core_1_1simd.html#a421845a6f68f88c58f520d2c1fa15914", null ], + [ "atan", "namespacemlx_1_1core_1_1simd.html#a99e84cece5722fb764844a2badc5426b", null ], + [ "atan2", "namespacemlx_1_1core_1_1simd.html#a23dba4ee3f0811b41c381733a6e6ff16", null ], + [ "atan2", "namespacemlx_1_1core_1_1simd.html#af5e8e8230c7d7af8201a3aaa7f491a2d", null ], + [ "atan2", "namespacemlx_1_1core_1_1simd.html#ac6104b5667e0eb379528bf7e2de23bee", null ], + [ "atanh", "namespacemlx_1_1core_1_1simd.html#a18d330fd2c7360b2890a722232ba35b7", null ], + [ "atanh", "namespacemlx_1_1core_1_1simd.html#a5ebae2e6cce1889513f15be3adb265ea", null ], + [ "atanh", "namespacemlx_1_1core_1_1simd.html#ae55fd26c3e18a6a27679d2b47566f8bc", null ], + [ "ceil", "namespacemlx_1_1core_1_1simd.html#ae5714693df24c8e26384fe5b5888376d", null ], + [ "ceil", "namespacemlx_1_1core_1_1simd.html#a4e54bd4ceb51ec41b0f95ebabe558713", null ], + [ "ceil", "namespacemlx_1_1core_1_1simd.html#ae1d5460c58c507a0104d8dfa90343f12", null ], + [ "clamp", "namespacemlx_1_1core_1_1simd.html#a146d2a834c936a381c1f86caffa822d7", null ], + [ "clamp", "namespacemlx_1_1core_1_1simd.html#a567c06bf988af03988478679055a6c45", null ], + [ "clamp", "namespacemlx_1_1core_1_1simd.html#ae4be4d88cd8eba7a8c1784fd53b86edb", null ], + [ "conj", "namespacemlx_1_1core_1_1simd.html#a660b79a51fb439f4aba91e2aea276300", null ], + [ "cos", "namespacemlx_1_1core_1_1simd.html#ab179f429e34cd6d5c37050ea7e7c54ad", null ], + [ "cosh", "namespacemlx_1_1core_1_1simd.html#aedc18b6fdb820cce9125c977c02833aa", null ], + [ "cosh", "namespacemlx_1_1core_1_1simd.html#ae1265896d855818d20f2de2a9ebb684a", null ], + [ "cosh", "namespacemlx_1_1core_1_1simd.html#aa5b4f7d3b776e8d16907e15a11800f01", null ], + [ "erf", "namespacemlx_1_1core_1_1simd.html#a60e33ebb16d9bab375a64aec8015a5c2", null ], + [ "erfinv", "namespacemlx_1_1core_1_1simd.html#a7687f3d14077b51fb421f0efb5b626db", null ], + [ "exp", "namespacemlx_1_1core_1_1simd.html#a835d71dd0bb2f9494a397d9939696ec2", null ], + [ "expm1", "namespacemlx_1_1core_1_1simd.html#a9407980793ecff5d5eb19c9a2cbda1eb", null ], + [ "expm1", "namespacemlx_1_1core_1_1simd.html#a8f73d1dac82177e0aeadaeda349c4f96", null ], + [ "expm1", "namespacemlx_1_1core_1_1simd.html#a464687a8809d0180035acc9af2943a94", null ], + [ "floor", "namespacemlx_1_1core_1_1simd.html#ad6b89aecafefe57b6ce69bec143ccd6e", null ], + [ "floor", "namespacemlx_1_1core_1_1simd.html#aa396efa6e9c94f4ac1f8381d5e07f069", null ], + [ "floor", "namespacemlx_1_1core_1_1simd.html#a8e22c484298d9af10b6604c835e52052", null ], + [ "fma", "namespacemlx_1_1core_1_1simd.html#a99099c338377518773b55d4042f9410d", null ], + [ "fma", "namespacemlx_1_1core_1_1simd.html#a8aa81ebff4c26f21cae2253d885fd87a", null ], + [ "fma", "namespacemlx_1_1core_1_1simd.html#a9ddc7f119cc1dc04372ec1adcaf55f70", null ], + [ "imag", "namespacemlx_1_1core_1_1simd.html#a97c69b04852ccba242f1348fda17ca20", null ], + [ "isnan", "namespacemlx_1_1core_1_1simd.html#a20ffdefe25beda96860a1dc9a6f4aa02", null ], + [ "isnan", "namespacemlx_1_1core_1_1simd.html#ac1c6c9b8bc7f3cd32ae39fa84975194d", null ], + [ "isnan", "namespacemlx_1_1core_1_1simd.html#a05f4422a037c3bef343fb11f71363b65", null ], + [ "load", "namespacemlx_1_1core_1_1simd.html#a4041676517d96870293e5448c7e2b5a4", null ], + [ "log", "namespacemlx_1_1core_1_1simd.html#a8cec82f4fb15bfd31d7554c6c09ceed4", null ], + [ "log", "namespacemlx_1_1core_1_1simd.html#abd09d3f5989558ce5156549a94d0fb04", null ], + [ "log", "namespacemlx_1_1core_1_1simd.html#a97a8ca857fe0edd84c68dc0f3dc2c6c4", null ], + [ "log10", "namespacemlx_1_1core_1_1simd.html#a3cb6ea94836e999c07329b34c501ed85", null ], + [ "log10", "namespacemlx_1_1core_1_1simd.html#adf754ade6cc1dd0e0bae0e31c7b513a2", null ], + [ "log10", "namespacemlx_1_1core_1_1simd.html#aba81b735e8f99cedf8b4846b2ab4e236", null ], + [ "log1p", "namespacemlx_1_1core_1_1simd.html#ad06680bbc041e76efe2dbff4e11b9a13", null ], + [ "log1p", "namespacemlx_1_1core_1_1simd.html#a92fcc8037ddb767bff517814ab55c259", null ], + [ "log1p", "namespacemlx_1_1core_1_1simd.html#a830591eb3007fef5d87dc296f5615108", null ], + [ "log2", "namespacemlx_1_1core_1_1simd.html#a66426c28a4324b9f617b7018d9354ea1", null ], + [ "log2", "namespacemlx_1_1core_1_1simd.html#af9eafa15692dec783860ddae3dd8c072", null ], + [ "log2", "namespacemlx_1_1core_1_1simd.html#acb1c49b90d029bc4a7eed257ec52791d", null ], + [ "max", "namespacemlx_1_1core_1_1simd.html#a995da0f1b4ca8077abbbc6f6a6dfd663", null ], + [ "max", "namespacemlx_1_1core_1_1simd.html#a312ecd0ae1c38d32147cee71fd8539d7", null ], + [ "max", "namespacemlx_1_1core_1_1simd.html#a6fcea259041cecfd042d0c4e6afc4b8f", null ], + [ "maximum", "namespacemlx_1_1core_1_1simd.html#ae1f11d9c2c15ebecf001d11b3fca5da2", null ], + [ "maximum", "namespacemlx_1_1core_1_1simd.html#aa78385c9cf0b87aabc377b1b47b2929d", null ], + [ "maximum", "namespacemlx_1_1core_1_1simd.html#ab54ff0f073be504e8428912f8e21effd", null ], + [ "maximum", "namespacemlx_1_1core_1_1simd.html#a7f7a298284e71ddbd2ba0bb6d98b0d16", null ], + [ "maximum", "namespacemlx_1_1core_1_1simd.html#a0ff63db5f193a57ef3b1fffa374eb15a", null ], + [ "min", "namespacemlx_1_1core_1_1simd.html#a160075943b92d541f2e7f7472eaa5167", null ], + [ "min", "namespacemlx_1_1core_1_1simd.html#a96db878d780a8da6abad19ac772d08ca", null ], + [ "min", "namespacemlx_1_1core_1_1simd.html#acd4196d0c66204cfae70b064c305e146", null ], + [ "minimum", "namespacemlx_1_1core_1_1simd.html#abaa09259e92f0fe758dc979d54c327e8", null ], + [ "minimum", "namespacemlx_1_1core_1_1simd.html#ae9ce2f34c97aba7b99223792a86d5c83", null ], + [ "minimum", "namespacemlx_1_1core_1_1simd.html#ac836568622a3e5957c275e115e2fcaf3", null ], + [ "minimum", "namespacemlx_1_1core_1_1simd.html#a1996e77a8c3c24b1ba706113ed9028c4", null ], + [ "minimum", "namespacemlx_1_1core_1_1simd.html#a17f7baec6300f2ff96ec53fb1943cb49", null ], + [ "operator!", "namespacemlx_1_1core_1_1simd.html#aadb0ed44c238d8d643c056298d5b20ca", null ], + [ "operator!", "namespacemlx_1_1core_1_1simd.html#afaa6ce61de4d80a4b7e9b2ab7454fff4", null ], + [ "operator!", "namespacemlx_1_1core_1_1simd.html#a745e05627c77152ec13d8d90c19cc9bf", null ], + [ "operator!=", "namespacemlx_1_1core_1_1simd.html#abc6a26b6e28d3d532fc356f96c97df1d", null ], + [ "operator!=", "namespacemlx_1_1core_1_1simd.html#a6cce6db46c391a5d06dcb262e21b81fc", null ], + [ "operator!=", "namespacemlx_1_1core_1_1simd.html#a4971bfe7f9f9319f859b3040c18f39ca", null ], + [ "operator!=", "namespacemlx_1_1core_1_1simd.html#a757838b9d56e132e797a381d3bb0dc86", null ], + [ "operator!=", "namespacemlx_1_1core_1_1simd.html#a7f1cebaff9cb88df59b5ec7557b5d167", null ], + [ "operator!=", "namespacemlx_1_1core_1_1simd.html#a125cbaa7c5dd0931b0abd11003ab584a", null ], + [ "operator!=", "namespacemlx_1_1core_1_1simd.html#a3699410174385f5e597cfccad57fc736", null ], + [ "operator!=", "namespacemlx_1_1core_1_1simd.html#a5c49123bf2647a5ca4f0579a54f3e53a", null ], + [ "operator!=", "namespacemlx_1_1core_1_1simd.html#ae8ca6615d51866d876b5efb3425600ed", null ], + [ "operator&", "namespacemlx_1_1core_1_1simd.html#a0727c897502944659b3e32b3cde9ee9b", null ], + [ "operator&", "namespacemlx_1_1core_1_1simd.html#a3c42ac1dc74f6c0bb934dfa45986875b", null ], + [ "operator&", "namespacemlx_1_1core_1_1simd.html#a3a060a225b6ead483ca93247c9ad8e4d", null ], + [ "operator&", "namespacemlx_1_1core_1_1simd.html#ac790406f4cf51cbc40d750d377dd741b", null ], + [ "operator&", "namespacemlx_1_1core_1_1simd.html#a8beb567724ab9735b616afb777b93abd", null ], + [ "operator&", "namespacemlx_1_1core_1_1simd.html#a832bbc02ed5589e70106c831c04500f1", null ], + [ "operator&&", "namespacemlx_1_1core_1_1simd.html#a2a381e5ec89406074b8d1921304238bb", null ], + [ "operator&&", "namespacemlx_1_1core_1_1simd.html#af9d5f107ce0c40c3b6a2f176cbb70cd7", null ], + [ "operator&&", "namespacemlx_1_1core_1_1simd.html#a85c23e7ed6fe0ec6dfe4c61f7412a362", null ], + [ "operator&&", "namespacemlx_1_1core_1_1simd.html#a82676bd32059d1172296f8074a841de6", null ], + [ "operator&&", "namespacemlx_1_1core_1_1simd.html#a1eca7cf07b2a238307459c28204319fb", null ], + [ "operator&&", "namespacemlx_1_1core_1_1simd.html#a070f1fa094cf2da5ab7d6baecbbf4f56", null ], + [ "operator&&", "namespacemlx_1_1core_1_1simd.html#af8f245dfc5154c04c0865a208ab1cfe9", null ], + [ "operator&&", "namespacemlx_1_1core_1_1simd.html#af97917ef704103c6ea1d0e44f22ec0d3", null ], + [ "operator&&", "namespacemlx_1_1core_1_1simd.html#a8a2c8aea209236b06c594c8451017ecb", null ], + [ "operator*", "namespacemlx_1_1core_1_1simd.html#a6f6d26e3fe39ee1ba0a7380d0ecf7b45", null ], + [ "operator*", "namespacemlx_1_1core_1_1simd.html#a5373c1af09825b5f701ebd106508fa6b", null ], + [ "operator*", "namespacemlx_1_1core_1_1simd.html#a08c1e7a00b1b4bc60e30d1554f4f46f2", null ], + [ "operator*", "namespacemlx_1_1core_1_1simd.html#ab6a73491bcb185cd91ae4db6b0f21e49", null ], + [ "operator*", "namespacemlx_1_1core_1_1simd.html#acd5ac48dc7895f06daf55f0a7e0667fb", null ], + [ "operator*", "namespacemlx_1_1core_1_1simd.html#a4555cd6a3b50af00700f97fdf00f63a7", null ], + [ "operator*", "namespacemlx_1_1core_1_1simd.html#ac50da923a4b7ac682554bd1d74c306d9", null ], + [ "operator*", "namespacemlx_1_1core_1_1simd.html#a4030444ea38ce1529a8cbb8c183a28bd", null ], + [ "operator*", "namespacemlx_1_1core_1_1simd.html#ae4ec5f1f081d20b46b13eb83eb1b6431", null ], + [ "operator+", "namespacemlx_1_1core_1_1simd.html#ae690b57b386cbad40565487d6d2393bb", null ], + [ "operator+", "namespacemlx_1_1core_1_1simd.html#a417109cdd61f35954ba2cc37af9b4460", null ], + [ "operator+", "namespacemlx_1_1core_1_1simd.html#aac6acd134f1498b4fb45fdbc882335bf", null ], + [ "operator+", "namespacemlx_1_1core_1_1simd.html#a27dfc3843dbefbbebed5b7137bacbb59", null ], + [ "operator+", "namespacemlx_1_1core_1_1simd.html#a98b77f1ca24bff373f48ef62f0013a02", null ], + [ "operator+", "namespacemlx_1_1core_1_1simd.html#aed655ffa017ade5e0f954f906d9f7ae6", null ], + [ "operator+", "namespacemlx_1_1core_1_1simd.html#ac33643b5f3cdbd3be0fa7d5784e35007", null ], + [ "operator+", "namespacemlx_1_1core_1_1simd.html#aa78806bf6a3be64b44e9a1f04bad3862", null ], + [ "operator+", "namespacemlx_1_1core_1_1simd.html#a8b622c47d07b171b2303ea744bf72284", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#a6e39cc693b30ad8e530392baf4bb5b0e", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#aad9cc064528e4189a5b7dd816a134ae6", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#a727a13b3d26f9e7cae7f091105867904", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#a70563bcd6c28802d11199812ffef38c8", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#af5be79b8dada8f8e91ae7c03c16606ec", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#ad5761065b4a655cd086d88846ae08d97", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#ab1f7f553d3a9176a70404a29cad06619", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#ab35a129d6e31b86c06b61252c7b26d4e", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#a678cddce777549a39474449d56fd1de6", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#a7434ba1ab2ad798fe8557a9b45035e81", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#aa73282cb05b65b931b97ce35c46bae20", null ], + [ "operator-", "namespacemlx_1_1core_1_1simd.html#acc490f7f5195acfa7b7c5df7afb39438", null ], + [ "operator/", "namespacemlx_1_1core_1_1simd.html#aab8837750c84794369e630d8ea0b408c", null ], + [ "operator/", "namespacemlx_1_1core_1_1simd.html#a0585ea196b665710115e48b7ebef0fc1", null ], + [ "operator/", "namespacemlx_1_1core_1_1simd.html#ac86a54a5e2ccc79bc92739f143bc0bef", null ], + [ "operator/", "namespacemlx_1_1core_1_1simd.html#a1d45c3b97cecfff86a2e43ae1f7fa185", null ], + [ "operator/", "namespacemlx_1_1core_1_1simd.html#a1c61bd3ac3ec5d8d2da65b45d59f543e", null ], + [ "operator/", "namespacemlx_1_1core_1_1simd.html#a18a2689f4ae197c5b204fe9b3370da4c", null ], + [ "operator/", "namespacemlx_1_1core_1_1simd.html#a075f637ff3f983ada0fd6288ab8d91d7", null ], + [ "operator/", "namespacemlx_1_1core_1_1simd.html#a89be64949908f19dd42aa7e38b320b0c", null ], + [ "operator/", "namespacemlx_1_1core_1_1simd.html#ac5d10f465c21ab259041042ff0159187", null ], + [ "operator<", "namespacemlx_1_1core_1_1simd.html#a9e0c9b3e986809be5e87aacc4612bb8e", null ], + [ "operator<", "namespacemlx_1_1core_1_1simd.html#a3f63139b42029ba8d7b3b8ef10f5ac96", null ], + [ "operator<", "namespacemlx_1_1core_1_1simd.html#a6cd6e41660608d17ca8d38658d5e385c", null ], + [ "operator<", "namespacemlx_1_1core_1_1simd.html#a23b59272b0760326844fffe20db9b3e2", null ], + [ "operator<", "namespacemlx_1_1core_1_1simd.html#acf35d81032bb9043804fd1de43540f60", null ], + [ "operator<", "namespacemlx_1_1core_1_1simd.html#ac962a14c88c87082fc70a9c0370f35b0", null ], + [ "operator<", "namespacemlx_1_1core_1_1simd.html#aaf29bfdcfdbb9a0acb9f4a6ed622868f", null ], + [ "operator<", "namespacemlx_1_1core_1_1simd.html#ad9bebf95b37fa0c6517be82af5ccd4eb", null ], + [ "operator<", "namespacemlx_1_1core_1_1simd.html#a01259c9188e6ecd48979cdc2fd766372", null ], + [ "operator<<", "namespacemlx_1_1core_1_1simd.html#ae21cbfd232edd7fe0f6f6c9fa11a354e", null ], + [ "operator<<", "namespacemlx_1_1core_1_1simd.html#a33232e2342d5a3e542c9428924a25830", null ], + [ "operator<<", "namespacemlx_1_1core_1_1simd.html#a68e7b952915e629d246d1ffac98b54ce", null ], + [ "operator<<", "namespacemlx_1_1core_1_1simd.html#a4ecd782ffa497ac7dc2482a232b0dd00", null ], + [ "operator<<", "namespacemlx_1_1core_1_1simd.html#a50044315dc365f026830416f6b615c77", null ], + [ "operator<<", "namespacemlx_1_1core_1_1simd.html#a56fccba38270fe3ae9fa7b2ecdeb5e87", null ], + [ "operator<=", "namespacemlx_1_1core_1_1simd.html#accd17f741cab18590fdbe388d4783967", null ], + [ "operator<=", "namespacemlx_1_1core_1_1simd.html#ad1570f6937d194a09e61d0e3a70ef578", null ], + [ "operator<=", "namespacemlx_1_1core_1_1simd.html#a4d5e4c31af23d2871e09b88c1f6e418c", null ], + [ "operator<=", "namespacemlx_1_1core_1_1simd.html#aec6783f79ca181d6782a810ffb267482", null ], + [ "operator<=", "namespacemlx_1_1core_1_1simd.html#a914e821c358e05dfe8d0208888646793", null ], + [ "operator<=", "namespacemlx_1_1core_1_1simd.html#aadd49786edc08f867e592d234327a031", null ], + [ "operator<=", "namespacemlx_1_1core_1_1simd.html#a46c6ea18a9edd2a9cdba2ab62ca4782c", null ], + [ "operator<=", "namespacemlx_1_1core_1_1simd.html#ae0fcb84973e4762a543ad3843db4f153", null ], + [ "operator<=", "namespacemlx_1_1core_1_1simd.html#a05240b8fd6f54632b676d4b66449f799", null ], + [ "operator==", "namespacemlx_1_1core_1_1simd.html#aaacbf6671080409e822fbb218e3fdf00", null ], + [ "operator==", "namespacemlx_1_1core_1_1simd.html#acafae9e62680565cd1f1c50c64d7ce4f", null ], + [ "operator==", "namespacemlx_1_1core_1_1simd.html#a273fcc5387c1c9878e658ba6bc32f00c", null ], + [ "operator==", "namespacemlx_1_1core_1_1simd.html#a7928482ed5d25932be80413c7239125c", null ], + [ "operator==", "namespacemlx_1_1core_1_1simd.html#a4877ae5406d081680b785a86ad656e03", null ], + [ "operator==", "namespacemlx_1_1core_1_1simd.html#a63768090c16e5dcffccadf550d169abc", null ], + [ "operator==", "namespacemlx_1_1core_1_1simd.html#aa837052ddcb02f4d9bc39b07399b4d91", null ], + [ "operator==", "namespacemlx_1_1core_1_1simd.html#a46ede415296683771bb22246a813482a", null ], + [ "operator==", "namespacemlx_1_1core_1_1simd.html#a51de2acf3dcd55c7c52e3ce7ed6ed9d7", null ], + [ "operator>", "namespacemlx_1_1core_1_1simd.html#aa17e031474fa87f6ea7855257dcc9ece", null ], + [ "operator>", "namespacemlx_1_1core_1_1simd.html#a35d875fa7bce02a6171f37240a346e1d", null ], + [ "operator>", "namespacemlx_1_1core_1_1simd.html#abd37e62eff936a64677b5aba787b4d18", null ], + [ "operator>", "namespacemlx_1_1core_1_1simd.html#ad8b67f9ced9c7f3cb472b9c3df817f08", null ], + [ "operator>", "namespacemlx_1_1core_1_1simd.html#ac971bfa5c7ec8abc432eab5f3c5646aa", null ], + [ "operator>", "namespacemlx_1_1core_1_1simd.html#ab7b291b3559792e18208e17432d25342", null ], + [ "operator>", "namespacemlx_1_1core_1_1simd.html#acf2391cc4d945887d7820501ba14ba89", null ], + [ "operator>", "namespacemlx_1_1core_1_1simd.html#a71a6902e729e3facdc609e93cd12d485", null ], + [ "operator>", "namespacemlx_1_1core_1_1simd.html#a4113a94fb8dcd0d88f14ec9d82089508", null ], + [ "operator>=", "namespacemlx_1_1core_1_1simd.html#ab9097573af69cc66d1427d0f52507e7a", null ], + [ "operator>=", "namespacemlx_1_1core_1_1simd.html#a8d7dcf1914ce8fe8518d84b0f2a5fe91", null ], + [ "operator>=", "namespacemlx_1_1core_1_1simd.html#a87e11ab36aae3328fe3d5230bdf31692", null ], + [ "operator>=", "namespacemlx_1_1core_1_1simd.html#a530ac8728e4d7e7be2482d5b2467906c", null ], + [ "operator>=", "namespacemlx_1_1core_1_1simd.html#a034d7b57cb3c6ca711c573515327d1a8", null ], + [ "operator>=", "namespacemlx_1_1core_1_1simd.html#a673b4d8d228f35f06cf5b882335f04d5", null ], + [ "operator>=", "namespacemlx_1_1core_1_1simd.html#aecdc08fcc70b158749a93a7a0f688aa3", null ], + [ "operator>=", "namespacemlx_1_1core_1_1simd.html#a4e65febbfa8b4df2970c1d78801b3c66", null ], + [ "operator>=", "namespacemlx_1_1core_1_1simd.html#ac7f3848b48c8e23c71c85fcc9909b933", null ], + [ "operator>>", "namespacemlx_1_1core_1_1simd.html#a6e45c9c2f0591d9d5dd37a07ebcc3c2a", null ], + [ "operator>>", "namespacemlx_1_1core_1_1simd.html#a1108d186d57c2010c743d3f9297befc7", null ], + [ "operator>>", "namespacemlx_1_1core_1_1simd.html#a4bf8c887eb6943563ceb1e603d1325b1", null ], + [ "operator>>", "namespacemlx_1_1core_1_1simd.html#aebf93b8179621e83bb3f3c4a8816eca8", null ], + [ "operator>>", "namespacemlx_1_1core_1_1simd.html#a9ac36abfb7dffc7ad24b4d0c295452e5", null ], + [ "operator>>", "namespacemlx_1_1core_1_1simd.html#aa35a2aab733e4bfc80a9f4e3f508daee", null ], + [ "operator^", "namespacemlx_1_1core_1_1simd.html#a25b3de1947dbab7c4864b41ec226453b", null ], + [ "operator^", "namespacemlx_1_1core_1_1simd.html#a369178519e0e91fa936c0fd4aa9ee109", null ], + [ "operator^", "namespacemlx_1_1core_1_1simd.html#a16fa3c809e46b5cae3e8abfaf98199a4", null ], + [ "operator^", "namespacemlx_1_1core_1_1simd.html#a16c4a2c8fc59a2e2fcc05db243289706", null ], + [ "operator^", "namespacemlx_1_1core_1_1simd.html#a5b877b5eb7044d9b2a42a9af4af21f01", null ], + [ "operator^", "namespacemlx_1_1core_1_1simd.html#a93e69a8170b8fe14f0a3188b4e8ccd49", null ], + [ "operator|", "namespacemlx_1_1core_1_1simd.html#ab2b540d7329491000e7722f9b3ef797d", null ], + [ "operator|", "namespacemlx_1_1core_1_1simd.html#a60805b5f57ddbbf74f700b54cd3fc4f8", null ], + [ "operator|", "namespacemlx_1_1core_1_1simd.html#a6449faa1666afe1186d55b61bb3e5b5a", null ], + [ "operator|", "namespacemlx_1_1core_1_1simd.html#acd57dc91aa205d9d3f8804df4261a7fb", null ], + [ "operator|", "namespacemlx_1_1core_1_1simd.html#a495d15a18ee4a6dda22e37e8dc02e45b", null ], + [ "operator|", "namespacemlx_1_1core_1_1simd.html#a0cd57bba23daed624df5e2b06b676dca", null ], + [ "operator||", "namespacemlx_1_1core_1_1simd.html#a38e83534a648d0743dc4c7deb9a7fd49", null ], + [ "operator||", "namespacemlx_1_1core_1_1simd.html#acdcdaea84869a0b05c08139c10f13a06", null ], + [ "operator||", "namespacemlx_1_1core_1_1simd.html#ab380b8f73672727a38ea0931e731fe4a", null ], + [ "operator||", "namespacemlx_1_1core_1_1simd.html#a7a1c3be1c37d41e450469f2e98cd9dde", null ], + [ "operator||", "namespacemlx_1_1core_1_1simd.html#aad2d440fbb9e5478b5ed24400a859942", null ], + [ "operator||", "namespacemlx_1_1core_1_1simd.html#ab2bc61c02b9096163e9db91a3f88788f", null ], + [ "operator||", "namespacemlx_1_1core_1_1simd.html#aa9ac1951153211b2ff95dd34a3427797", null ], + [ "operator||", "namespacemlx_1_1core_1_1simd.html#a0c8bd67982681ecd53cd8d739be3a5a9", null ], + [ "operator||", "namespacemlx_1_1core_1_1simd.html#ac34f6b278627949d2ee68cdbf3d2f50f", null ], + [ "pow", "namespacemlx_1_1core_1_1simd.html#a19d535de1fc179cc39ec9643c9863cbc", null ], + [ "pow", "namespacemlx_1_1core_1_1simd.html#afe3d50bc4a11061898aa57377fa9536d", null ], + [ "pow", "namespacemlx_1_1core_1_1simd.html#ab18b3a88a2439fd026b6551b38d1f14a", null ], + [ "prod", "namespacemlx_1_1core_1_1simd.html#a0a26dff48b078fb3e9fef688232183ed", null ], + [ "prod", "namespacemlx_1_1core_1_1simd.html#ac27cdc630e86b25ad607ca409de2b274", null ], + [ "prod", "namespacemlx_1_1core_1_1simd.html#a271cedfc48efc69db43813e8c424bf7c", null ], + [ "real", "namespacemlx_1_1core_1_1simd.html#acdf822b7626bbab6a495552aea3457b5", null ], + [ "recip", "namespacemlx_1_1core_1_1simd.html#a6235990c43aaf0e0c126c82d10f01b45", null ], + [ "recip", "namespacemlx_1_1core_1_1simd.html#afc915aed256295475ac88fde3a736f1f", null ], + [ "recip", "namespacemlx_1_1core_1_1simd.html#ae344abefc91c7d9c0a9506c868a84d61", null ], + [ "remainder", "namespacemlx_1_1core_1_1simd.html#ac66bdf1a8e86a4d350c85037bc764da5", null ], + [ "remainder", "namespacemlx_1_1core_1_1simd.html#a54c7f2f2b995eb767462b1228982967f", null ], + [ "remainder", "namespacemlx_1_1core_1_1simd.html#ab020d2c434fad0cdf79fd37b0f6c1676", null ], + [ "rint", "namespacemlx_1_1core_1_1simd.html#a8c200919c0eeefb2e2e5d9d19741a805", null ], + [ "rint", "namespacemlx_1_1core_1_1simd.html#a797196eccc3690aac5c45e5f9c804ceb", null ], + [ "rint", "namespacemlx_1_1core_1_1simd.html#a400d89d040f43d471b306a8e8bdb3974", null ], + [ "rsqrt", "namespacemlx_1_1core_1_1simd.html#a3345cb53830d1afd625acc7bdc3a0435", null ], + [ "rsqrt", "namespacemlx_1_1core_1_1simd.html#a74ac0fd799967b0f303bfd26fc6a17cf", null ], + [ "rsqrt", "namespacemlx_1_1core_1_1simd.html#aea75ddf8c696efc2e5e924667ed48e70", null ], + [ "select", "namespacemlx_1_1core_1_1simd.html#a9e3e7b35d564c70de8fa0b6150570ed8", null ], + [ "select", "namespacemlx_1_1core_1_1simd.html#a3b5ebb46e7beae839c97b2e7ed9c7426", null ], + [ "select", "namespacemlx_1_1core_1_1simd.html#afb3bcbd8d8b34128cd0c8eb677a170ef", null ], + [ "sin", "namespacemlx_1_1core_1_1simd.html#ab4d582d72c0a7ee313e19c906e43cef1", null ], + [ "sincos", "namespacemlx_1_1core_1_1simd.html#ad78056685c9732c3465c0d8b8ec1bef7", null ], + [ "sinh", "namespacemlx_1_1core_1_1simd.html#a09a2f3f2bc999c16babf3d8d90994d6e", null ], + [ "sinh", "namespacemlx_1_1core_1_1simd.html#a40879bf874309c0a5abef783aea2057d", null ], + [ "sinh", "namespacemlx_1_1core_1_1simd.html#a479ccddac341bd0760857b77e449e5e1", null ], + [ "sqrt", "namespacemlx_1_1core_1_1simd.html#a7e80d3e33f2edd02310641d3e3dd5658", null ], + [ "sqrt", "namespacemlx_1_1core_1_1simd.html#a2ba6c75c0821db3e9ac525a89b3ac859", null ], + [ "sqrt", "namespacemlx_1_1core_1_1simd.html#ae39b8e1d1fff94947406eeb8ec6e0414", null ], + [ "store", "namespacemlx_1_1core_1_1simd.html#afa2236afddfdec312eb7e27b89a5316a", null ], + [ "sum", "namespacemlx_1_1core_1_1simd.html#ab0e7c082fc6bed52d522765ef91d205d", null ], + [ "sum", "namespacemlx_1_1core_1_1simd.html#a96ce7d90b3b8b6dddab36ef5b49fffc2", null ], + [ "sum", "namespacemlx_1_1core_1_1simd.html#a53b547b886918dc13d4da88eeb8811d2", null ], + [ "tan", "namespacemlx_1_1core_1_1simd.html#a445ddc4ed928656df64d889942588cfd", null ], + [ "tan", "namespacemlx_1_1core_1_1simd.html#a5abc381a85fe8b0e9cb472f874704652", null ], + [ "tan", "namespacemlx_1_1core_1_1simd.html#a829842f854aecfae93b7d42f83aec9a7", null ], + [ "tanh", "namespacemlx_1_1core_1_1simd.html#aa244fbe7456b653aa50a473108fd6a2b", null ], + [ "tanh", "namespacemlx_1_1core_1_1simd.html#ab80a7db8d99e3f4032e761c60216027d", null ], + [ "tanh", "namespacemlx_1_1core_1_1simd.html#ad78f543dc5da87a14ca113a1dd9852fd", null ], + [ "inf", "namespacemlx_1_1core_1_1simd.html#a771b6597803beb800ff5e7560c41e341", null ], + [ "is_complex", "namespacemlx_1_1core_1_1simd.html#a64e80f096a8baf99ba8d396414473cc7", null ], + [ "is_complex< T, std::void_t< decltype(std::declval< T >().real())> >", "namespacemlx_1_1core_1_1simd.html#a29fe8445e54a61f6bccc8d50f142ca54", null ], + [ "max_size", "namespacemlx_1_1core_1_1simd.html#ac91bd36c7caafd3c7ff176e7e2f81887", null ], + [ "max_size< double >", "namespacemlx_1_1core_1_1simd.html#a3fa3d1f571027c5cdd1dce5d2cd041e3", null ], + [ "max_size< float >", "namespacemlx_1_1core_1_1simd.html#ae745e117cacfe455df39aa4569c34c11", null ], + [ "max_size< float16_t >", "namespacemlx_1_1core_1_1simd.html#a155df1de3c26e1a3725b63e9e97c0b53", null ], + [ "max_size< int >", "namespacemlx_1_1core_1_1simd.html#ab25fc96fa6f00d0a8c335b8da293fbbb", null ], + [ "max_size< int16_t >", "namespacemlx_1_1core_1_1simd.html#a7e63a5eb08898b84fd4000dadc460fd9", null ], + [ "max_size< int64_t >", "namespacemlx_1_1core_1_1simd.html#a7913cb2854ffc37efcf26635a097f0a9", null ], + [ "max_size< int8_t >", "namespacemlx_1_1core_1_1simd.html#ac368e4701363cfece4935e57f3c709b1", null ], + [ "max_size< uint16_t >", "namespacemlx_1_1core_1_1simd.html#a0cc9ca2925c25d2eb225af9125bd6bc4", null ], + [ "max_size< uint32_t >", "namespacemlx_1_1core_1_1simd.html#a06cb29f91deeaec69471058044abd2aa", null ], + [ "max_size< uint64_t >", "namespacemlx_1_1core_1_1simd.html#ab367b9b65be2fda4830a56fc9cc0cd2f", null ], + [ "max_size< uint8_t >", "namespacemlx_1_1core_1_1simd.html#a8f731e5a287c714dfc92879fe37503d5", null ], + [ "N", "namespacemlx_1_1core_1_1simd.html#a12b1553495a0c99d52472bd2a6626ddb", null ] +]; \ No newline at end of file diff --git a/docs/build/html/namespacemlx_1_1steel.html b/docs/build/html/namespacemlx_1_1steel.html index 880a97f7b..5b1b1c4b5 100644 --- a/docs/build/html/namespacemlx_1_1steel.html +++ b/docs/build/html/namespacemlx_1_1steel.html @@ -3,7 +3,7 @@ - + MLX: mlx::steel Namespace Reference @@ -52,7 +52,7 @@
    - + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    +Classes | +Namespaces | +Macros | +Functions | +Variables
    +
    neon_fp16_simd.h File Reference
    +
    +
    +
    #include <arm_neon.h>
    +#include "mlx/backend/cpu/simd/base_simd.h"
    +
    +

    Go to the source code of this file.

    + + + + +

    +Classes

    struct  mlx::core::simd::Simd< float16_t, N >
     
    + + + + + + + +

    +Namespaces

    namespace  mlx
     
    namespace  mlx::core
     
    namespace  mlx::core::simd
     
    + + + + + + + +

    +Macros

    #define DEFINE_NEON_UNARY_OP(name, op)
     
    #define DEFINE_NEON_BINARY_OP(name, op)
     
    #define DEFINE_NEON_COMPARISON(Op, op)
     
    + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

    +Functions

    Simd< float16_t, N > mlx::core::simd::abs (Simd< float16_t, N > a)
     
    Simd< float16_t, N > mlx::core::simd::ceil (Simd< float16_t, N > a)
     
    Simd< float16_t, N > mlx::core::simd::floor (Simd< float16_t, N > a)
     
    Simd< float16_t, N > mlx::core::simd::sqrt (Simd< float16_t, N > a)
     
    Simd< float16_t, N > mlx::core::simd::rsqrt (Simd< float16_t, N > a)
     
    Simd< float16_t, N > mlx::core::simd::recip (Simd< float16_t, N > a)
     
    Simd< float16_t, N > mlx::core::simd::rint (Simd< float16_t, N > a)
     
    Simd< float16_t, N > mlx::core::simd::operator! (Simd< float16_t, N > v)
     
    Simd< float16_t, N > mlx::core::simd::operator- (Simd< float16_t, N > v)
     
    Simd< float16_t, N > mlx::core::simd::maximum (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::maximum (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::maximum (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > mlx::core::simd::minimum (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::minimum (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::minimum (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > mlx::core::simd::operator+ (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator+ (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator+ (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > mlx::core::simd::operator- (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator- (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator- (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > mlx::core::simd::operator* (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator* (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator* (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > mlx::core::simd::operator/ (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator/ (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator/ (T a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator== (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator== (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > mlx::core::simd::operator== (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator>= (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator>= (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > mlx::core::simd::operator>= (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator<= (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator<= (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > mlx::core::simd::operator<= (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator> (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator> (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > mlx::core::simd::operator> (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator< (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator< (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > mlx::core::simd::operator< (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator!= (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< bool, N > mlx::core::simd::operator!= (T a, Simd< float16_t, N > b)
     
    Simd< bool, N > mlx::core::simd::operator!= (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > mlx::core::simd::operator|| (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator|| (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator|| (T a, Simd< float16_t, N > b)
     
    Simd< float16_t, N > mlx::core::simd::operator&& (Simd< float16_t, N > a, Simd< float16_t, N > b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator&& (Simd< float16_t, N > a, T b)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::operator&& (T a, Simd< float16_t, N > b)
     
    template<>
    Simd< bool, N > mlx::core::simd::isnan (Simd< float16_t, N > v)
     
    template<>
    Simd< float16_t, N > mlx::core::simd::clamp (Simd< float16_t, N > v, Simd< float16_t, N > min, Simd< float16_t, N > max)
     
    template<typename T>
    Simd< float16_t, N > mlx::core::simd::fma (Simd< float16_t, N > x, Simd< float16_t, N > y, T z)
     
    template<typename MaskT>
    Simd< float16_t, N > mlx::core::simd::select (Simd< MaskT, N > mask, Simd< float16_t, N > x, Simd< float16_t, N > y)
     
    float16_t mlx::core::simd::max (Simd< float16_t, N > x)
     
    float16_t mlx::core::simd::min (Simd< float16_t, N > x)
     
    float16_t mlx::core::simd::sum (Simd< float16_t, N > x)
     
    float16_t mlx::core::simd::prod (Simd< float16_t, N > x)
     
    + + + +

    +Variables

    constexpr int mlx::core::simd::N = 8
     
    +

    Macro Definition Documentation

    + +

    ◆ DEFINE_NEON_BINARY_OP

    + +
    +
    + + + + + + + + + + + +
    #define DEFINE_NEON_BINARY_OP( name,
    op )
    +
    +Value:
    inline Simd<float16_t, N> name(Simd<float16_t, N> a, Simd<float16_t, N> b) { \
    +
    return op(a.value, b.value); \
    +
    } \
    +
    template <typename T> \
    +
    Simd<float16_t, N> name(Simd<float16_t, N> a, T b) { \
    +
    return op(a.value, Simd<float16_t, N>(b).value); \
    +
    } \
    +
    template <typename T> \
    +
    Simd<float16_t, N> name(T a, Simd<float16_t, N> b) { \
    +
    return op(Simd<float16_t, N>(a).value, b.value); \
    +
    }
    +
    +
    +
    + +

    ◆ DEFINE_NEON_COMPARISON

    + +
    +
    + + + + + + + + + + + +
    #define DEFINE_NEON_COMPARISON( Op,
    op )
    +
    +Value:
    template <typename T> \
    +
    Simd<bool, N> operator Op(Simd<float16_t, N> a, T b) { \
    +
    auto out = op(a.value, Simd<float16_t, N>(b).value); \
    +
    return Simd<uint16_t, N>(*(uint16_t*)(&out)); \
    +
    } \
    +
    template <typename T> \
    +
    Simd<bool, N> operator Op(T a, Simd<float16_t, N> b) { \
    +
    auto out = op(Simd<float16_t, N>(a).value, b.value); \
    +
    return Simd<uint16_t, N>(*(uint16_t*)(&out)); \
    +
    } \
    +
    inline Simd<bool, N> operator Op( \
    +
    Simd<float16_t, N> a, Simd<float16_t, N> b) { \
    +
    auto out = op(a.value, b.value); \
    +
    return Simd<uint16_t, N>(*(uint16_t*)(&out)); \
    +
    }
    +
    +
    +
    + +

    ◆ DEFINE_NEON_UNARY_OP

    + +
    +
    + + + + + + + + + + + +
    #define DEFINE_NEON_UNARY_OP( name,
    op )
    +
    +Value:
    inline Simd<float16_t, N> name(Simd<float16_t, N> a) { \
    +
    return Simd<float16_t, N>{op(a.value)}; \
    +
    }
    +
    +
    +
    +
    +
    + + + + diff --git a/docs/build/html/neon__fp16__simd_8h.js b/docs/build/html/neon__fp16__simd_8h.js new file mode 100644 index 000000000..a4efbc48d --- /dev/null +++ b/docs/build/html/neon__fp16__simd_8h.js @@ -0,0 +1,67 @@ +var neon__fp16__simd_8h = +[ + [ "mlx::core::simd::Simd< float16_t, N >", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4" ], + [ "DEFINE_NEON_BINARY_OP", "neon__fp16__simd_8h.html#a63ba6e115ef14e134081310f7d4579a5", null ], + [ "DEFINE_NEON_COMPARISON", "neon__fp16__simd_8h.html#a9fcd94c7369a6b4437f9c310a805c79d", null ], + [ "DEFINE_NEON_UNARY_OP", "neon__fp16__simd_8h.html#af8138a463be93b9e0c9b685e94a1fd00", null ], + [ "mlx::core::simd::abs", "namespacemlx_1_1core_1_1simd.html#a7696a0628a1c6ccb293ebd6f2328ea48", null ], + [ "mlx::core::simd::ceil", "namespacemlx_1_1core_1_1simd.html#ae5714693df24c8e26384fe5b5888376d", null ], + [ "mlx::core::simd::clamp", "namespacemlx_1_1core_1_1simd.html#a146d2a834c936a381c1f86caffa822d7", null ], + [ "mlx::core::simd::floor", "namespacemlx_1_1core_1_1simd.html#ad6b89aecafefe57b6ce69bec143ccd6e", null ], + [ "mlx::core::simd::fma", "namespacemlx_1_1core_1_1simd.html#a99099c338377518773b55d4042f9410d", null ], + [ "mlx::core::simd::isnan", "namespacemlx_1_1core_1_1simd.html#a20ffdefe25beda96860a1dc9a6f4aa02", null ], + [ "mlx::core::simd::max", "namespacemlx_1_1core_1_1simd.html#a995da0f1b4ca8077abbbc6f6a6dfd663", null ], + [ "mlx::core::simd::maximum", "namespacemlx_1_1core_1_1simd.html#ae1f11d9c2c15ebecf001d11b3fca5da2", null ], + [ "mlx::core::simd::maximum", "namespacemlx_1_1core_1_1simd.html#aa78385c9cf0b87aabc377b1b47b2929d", null ], + [ "mlx::core::simd::maximum", "namespacemlx_1_1core_1_1simd.html#a0ff63db5f193a57ef3b1fffa374eb15a", null ], + [ "mlx::core::simd::min", "namespacemlx_1_1core_1_1simd.html#a160075943b92d541f2e7f7472eaa5167", null ], + [ "mlx::core::simd::minimum", "namespacemlx_1_1core_1_1simd.html#abaa09259e92f0fe758dc979d54c327e8", null ], + [ "mlx::core::simd::minimum", "namespacemlx_1_1core_1_1simd.html#ae9ce2f34c97aba7b99223792a86d5c83", null ], + [ "mlx::core::simd::minimum", "namespacemlx_1_1core_1_1simd.html#a17f7baec6300f2ff96ec53fb1943cb49", null ], + [ "mlx::core::simd::operator!", "namespacemlx_1_1core_1_1simd.html#aadb0ed44c238d8d643c056298d5b20ca", null ], + [ "mlx::core::simd::operator!=", "namespacemlx_1_1core_1_1simd.html#abc6a26b6e28d3d532fc356f96c97df1d", null ], + [ "mlx::core::simd::operator!=", "namespacemlx_1_1core_1_1simd.html#a6cce6db46c391a5d06dcb262e21b81fc", null ], + [ "mlx::core::simd::operator!=", "namespacemlx_1_1core_1_1simd.html#a3699410174385f5e597cfccad57fc736", null ], + [ "mlx::core::simd::operator&&", "namespacemlx_1_1core_1_1simd.html#a2a381e5ec89406074b8d1921304238bb", null ], + [ "mlx::core::simd::operator&&", "namespacemlx_1_1core_1_1simd.html#af9d5f107ce0c40c3b6a2f176cbb70cd7", null ], + [ "mlx::core::simd::operator&&", "namespacemlx_1_1core_1_1simd.html#af8f245dfc5154c04c0865a208ab1cfe9", null ], + [ "mlx::core::simd::operator*", "namespacemlx_1_1core_1_1simd.html#a6f6d26e3fe39ee1ba0a7380d0ecf7b45", null ], + [ "mlx::core::simd::operator*", "namespacemlx_1_1core_1_1simd.html#a5373c1af09825b5f701ebd106508fa6b", null ], + [ "mlx::core::simd::operator*", "namespacemlx_1_1core_1_1simd.html#ac50da923a4b7ac682554bd1d74c306d9", null ], + [ "mlx::core::simd::operator+", "namespacemlx_1_1core_1_1simd.html#ae690b57b386cbad40565487d6d2393bb", null ], + [ "mlx::core::simd::operator+", "namespacemlx_1_1core_1_1simd.html#a417109cdd61f35954ba2cc37af9b4460", null ], + [ "mlx::core::simd::operator+", "namespacemlx_1_1core_1_1simd.html#ac33643b5f3cdbd3be0fa7d5784e35007", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#a6e39cc693b30ad8e530392baf4bb5b0e", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#aad9cc064528e4189a5b7dd816a134ae6", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#a727a13b3d26f9e7cae7f091105867904", null ], + [ "mlx::core::simd::operator-", "namespacemlx_1_1core_1_1simd.html#a7434ba1ab2ad798fe8557a9b45035e81", null ], + [ "mlx::core::simd::operator/", "namespacemlx_1_1core_1_1simd.html#aab8837750c84794369e630d8ea0b408c", null ], + [ "mlx::core::simd::operator/", "namespacemlx_1_1core_1_1simd.html#a0585ea196b665710115e48b7ebef0fc1", null ], + [ "mlx::core::simd::operator/", "namespacemlx_1_1core_1_1simd.html#a075f637ff3f983ada0fd6288ab8d91d7", null ], + [ "mlx::core::simd::operator<", "namespacemlx_1_1core_1_1simd.html#a9e0c9b3e986809be5e87aacc4612bb8e", null ], + [ "mlx::core::simd::operator<", "namespacemlx_1_1core_1_1simd.html#a3f63139b42029ba8d7b3b8ef10f5ac96", null ], + [ "mlx::core::simd::operator<", "namespacemlx_1_1core_1_1simd.html#aaf29bfdcfdbb9a0acb9f4a6ed622868f", null ], + [ "mlx::core::simd::operator<=", "namespacemlx_1_1core_1_1simd.html#accd17f741cab18590fdbe388d4783967", null ], + [ "mlx::core::simd::operator<=", "namespacemlx_1_1core_1_1simd.html#ad1570f6937d194a09e61d0e3a70ef578", null ], + [ "mlx::core::simd::operator<=", "namespacemlx_1_1core_1_1simd.html#a46c6ea18a9edd2a9cdba2ab62ca4782c", null ], + [ "mlx::core::simd::operator==", "namespacemlx_1_1core_1_1simd.html#aaacbf6671080409e822fbb218e3fdf00", null ], + [ "mlx::core::simd::operator==", "namespacemlx_1_1core_1_1simd.html#acafae9e62680565cd1f1c50c64d7ce4f", null ], + [ "mlx::core::simd::operator==", "namespacemlx_1_1core_1_1simd.html#aa837052ddcb02f4d9bc39b07399b4d91", null ], + [ "mlx::core::simd::operator>", "namespacemlx_1_1core_1_1simd.html#aa17e031474fa87f6ea7855257dcc9ece", null ], + [ "mlx::core::simd::operator>", "namespacemlx_1_1core_1_1simd.html#a35d875fa7bce02a6171f37240a346e1d", null ], + [ "mlx::core::simd::operator>", "namespacemlx_1_1core_1_1simd.html#acf2391cc4d945887d7820501ba14ba89", null ], + [ "mlx::core::simd::operator>=", "namespacemlx_1_1core_1_1simd.html#ab9097573af69cc66d1427d0f52507e7a", null ], + [ "mlx::core::simd::operator>=", "namespacemlx_1_1core_1_1simd.html#a8d7dcf1914ce8fe8518d84b0f2a5fe91", null ], + [ "mlx::core::simd::operator>=", "namespacemlx_1_1core_1_1simd.html#aecdc08fcc70b158749a93a7a0f688aa3", null ], + [ "mlx::core::simd::operator||", "namespacemlx_1_1core_1_1simd.html#a38e83534a648d0743dc4c7deb9a7fd49", null ], + [ "mlx::core::simd::operator||", "namespacemlx_1_1core_1_1simd.html#acdcdaea84869a0b05c08139c10f13a06", null ], + [ "mlx::core::simd::operator||", "namespacemlx_1_1core_1_1simd.html#aa9ac1951153211b2ff95dd34a3427797", null ], + [ "mlx::core::simd::prod", "namespacemlx_1_1core_1_1simd.html#a0a26dff48b078fb3e9fef688232183ed", null ], + [ "mlx::core::simd::recip", "namespacemlx_1_1core_1_1simd.html#a6235990c43aaf0e0c126c82d10f01b45", null ], + [ "mlx::core::simd::rint", "namespacemlx_1_1core_1_1simd.html#a8c200919c0eeefb2e2e5d9d19741a805", null ], + [ "mlx::core::simd::rsqrt", "namespacemlx_1_1core_1_1simd.html#a3345cb53830d1afd625acc7bdc3a0435", null ], + [ "mlx::core::simd::select", "namespacemlx_1_1core_1_1simd.html#a3b5ebb46e7beae839c97b2e7ed9c7426", null ], + [ "mlx::core::simd::sqrt", "namespacemlx_1_1core_1_1simd.html#a7e80d3e33f2edd02310641d3e3dd5658", null ], + [ "mlx::core::simd::sum", "namespacemlx_1_1core_1_1simd.html#ab0e7c082fc6bed52d522765ef91d205d", null ], + [ "mlx::core::simd::N", "namespacemlx_1_1core_1_1simd.html#a12b1553495a0c99d52472bd2a6626ddb", null ] +]; \ No newline at end of file diff --git a/docs/build/html/neon__fp16__simd_8h_source.html b/docs/build/html/neon__fp16__simd_8h_source.html new file mode 100644 index 000000000..01130a218 --- /dev/null +++ b/docs/build/html/neon__fp16__simd_8h_source.html @@ -0,0 +1,429 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/neon_fp16_simd.h Source File + + + + + + + + + + + + + + + + +
    +
    + + + + + + + +
    +
    MLX +
    +
    + +   + + + + +
    +
    +
    + + + + +
    +
    + +
    +
    +
    + +
    + +
    +
    + + +
    +
    +
    +
    +
    +
    Loading...
    +
    Searching...
    +
    No Matches
    +
    +
    +
    +
    + +
    +
    neon_fp16_simd.h
    +
    +
    +Go to the documentation of this file.
    1#pragma once
    +
    2
    +
    3#include <arm_neon.h>
    +
    4
    +
    5#include "mlx/backend/cpu/simd/base_simd.h"
    +
    6
    +
    7namespace mlx::core::simd {
    +
    8
    +
    9constexpr int N = 8;
    +
    10
    +
    11template <>
    +
    +
    12struct Simd<float16_t, N> {
    +
    +
    13 static constexpr int size = N;
    +
    14 using scalar_t = float16_t;
    +
    15
    +
    16 Simd<float16_t, N>() {}
    +
    +
    17
    +
    18 template <typename U>
    +
    19 Simd<float16_t, N>(U v) : value(vdupq_n_f16(v)){};
    +
    20
    +
    21 Simd<float16_t, N>(float16x8_t v) : value(v){};
    +
    22
    +
    23 Simd<float16_t, N>(Simd<float, N> other) {
    +
    24 auto f32x4_a = *(float32x4_t*)(&other);
    +
    25 auto f32x4_b = *((float32x4_t*)(&other) + 1);
    +
    26 value = vcvt_high_f16_f32(vcvt_f16_f32(f32x4_a), f32x4_b);
    +
    27 };
    +
    28
    +
    29 Simd<float16_t, N>(Simd<uint16_t, N> other) {
    +
    30 value = vcvtq_f16_u16(*(uint16x8_t*)(&other.value));
    +
    31 };
    +
    32
    +
    +
    33 operator Simd<int16_t, N>() {
    +
    34 auto v = vcvtq_s16_f16(value);
    +
    35 return load<int16_t, N>((int16_t*)&v);
    +
    36 };
    +
    +
    37
    +
    +
    38 operator Simd<float, N>() {
    +
    39 float32x4x2_t v;
    +
    40 v.val[0] = vcvt_f32_f16(*(float16x4_t*)(&value));
    +
    41 v.val[1] = vcvt_high_f32_f16(value);
    +
    42 return load<float, N>((float*)&v);
    +
    43 }
    +
    +
    +
    44 float16_t operator[](int idx) const {
    +
    45 return reinterpret_cast<const float16_t*>(&value)[idx];
    +
    46 }
    +
    +
    47
    +
    +
    48 float16_t& operator[](int idx) {
    +
    49 return reinterpret_cast<float16_t*>(&value)[idx];
    +
    50 }
    +
    +
    51
    +
    52 float16x8_t value;
    +
    53};
    +
    +
    54
    +
    +
    55#define DEFINE_NEON_UNARY_OP(name, op) \
    +
    56 inline Simd<float16_t, N> name(Simd<float16_t, N> a) { \
    +
    57 return Simd<float16_t, N>{op(a.value)}; \
    +
    58 }
    +
    +
    59
    +
    60DEFINE_NEON_UNARY_OP(abs, vabsq_f16)
    +
    61DEFINE_NEON_UNARY_OP(ceil, vrndpq_f16)
    +
    62DEFINE_NEON_UNARY_OP(floor, vrndmq_f16)
    +
    63DEFINE_NEON_UNARY_OP(sqrt, vsqrtq_f16)
    +
    64DEFINE_NEON_UNARY_OP(rsqrt, vrsqrteq_f16)
    +
    65DEFINE_NEON_UNARY_OP(recip, vrecpeq_f16)
    +
    66DEFINE_NEON_UNARY_OP(rint, vrndnq_f16)
    +
    67
    +
    +
    68#define DEFINE_NEON_BINARY_OP(name, op) \
    +
    69 inline Simd<float16_t, N> name(Simd<float16_t, N> a, Simd<float16_t, N> b) { \
    +
    70 return op(a.value, b.value); \
    +
    71 } \
    +
    72 template <typename T> \
    +
    73 Simd<float16_t, N> name(Simd<float16_t, N> a, T b) { \
    +
    74 return op(a.value, Simd<float16_t, N>(b).value); \
    +
    75 } \
    +
    76 template <typename T> \
    +
    77 Simd<float16_t, N> name(T a, Simd<float16_t, N> b) { \
    +
    78 return op(Simd<float16_t, N>(a).value, b.value); \
    +
    79 }
    +
    +
    80
    +
    +
    81inline Simd<float16_t, N> operator!(Simd<float16_t, N> v) {
    +
    82 auto out = vceqzq_f16(v.value);
    +
    83 return Simd<uint16_t, N>(*(uint16_t*)&out);
    +
    84}
    +
    +
    85
    +
    +
    86inline Simd<float16_t, N> operator-(Simd<float16_t, N> v) {
    +
    87 return vnegq_f16(v.value);
    +
    88}
    +
    +
    89
    +
    90DEFINE_NEON_BINARY_OP(maximum, vmaxq_f16)
    +
    91DEFINE_NEON_BINARY_OP(minimum, vminq_f16)
    +
    92DEFINE_NEON_BINARY_OP(operator+, vaddq_f16)
    +
    93DEFINE_NEON_BINARY_OP(operator-, vsubq_f16)
    +
    94DEFINE_NEON_BINARY_OP(operator*, vmulq_f16)
    +
    95DEFINE_NEON_BINARY_OP(operator/, vdivq_f16)
    +
    96
    +
    +
    97#define DEFINE_NEON_COMPARISON(Op, op) \
    +
    98 template <typename T> \
    +
    99 Simd<bool, N> operator Op(Simd<float16_t, N> a, T b) { \
    +
    100 auto out = op(a.value, Simd<float16_t, N>(b).value); \
    +
    101 return Simd<uint16_t, N>(*(uint16_t*)(&out)); \
    +
    102 } \
    +
    103 template <typename T> \
    +
    104 Simd<bool, N> operator Op(T a, Simd<float16_t, N> b) { \
    +
    105 auto out = op(Simd<float16_t, N>(a).value, b.value); \
    +
    106 return Simd<uint16_t, N>(*(uint16_t*)(&out)); \
    +
    107 } \
    +
    108 inline Simd<bool, N> operator Op( \
    +
    109 Simd<float16_t, N> a, Simd<float16_t, N> b) { \
    +
    110 auto out = op(a.value, b.value); \
    +
    111 return Simd<uint16_t, N>(*(uint16_t*)(&out)); \
    +
    112 }
    +
    +
    113
    +
    114DEFINE_NEON_COMPARISON(==, vceqq_f16)
    +
    +
    115DEFINE_NEON_COMPARISON(>=, vcgeq_f16)
    +
    116DEFINE_NEON_COMPARISON(<=, vcleq_f16)
    +
    +
    +
    117DEFINE_NEON_COMPARISON(>, vcgtq_f16)
    +
    118DEFINE_NEON_COMPARISON(<, vcltq_f16)
    +
    +
    119
    +
    120template <typename T>
    +
    +
    121Simd<bool, N> operator!=(Simd<float16_t, N> a, T b) {
    +
    122 return !(a == b);
    +
    123}
    +
    +
    124template <typename T>
    +
    +
    125Simd<bool, N> operator!=(T a, Simd<float16_t, N> b) {
    +
    126 return !(a == b);
    +
    127}
    +
    +
    +
    128inline Simd<bool, N> operator!=(Simd<float16_t, N> a, Simd<float16_t, N> b) {
    +
    129 return !(a == b);
    +
    130}
    +
    +
    131
    +
    +
    132inline Simd<float16_t, N> operator||(
    +
    133 Simd<float16_t, N> a,
    +
    134 Simd<float16_t, N> b) {
    +
    135 return Simd<uint16_t, N>((a != 0) || (b != 0));
    +
    136}
    +
    +
    137template <typename T>
    +
    +
    138Simd<float16_t, N> operator||(Simd<float16_t, N> a, T b) {
    +
    139 return Simd<uint16_t, N>((a != 0) || (b != 0));
    +
    140}
    +
    +
    141template <typename T>
    +
    +
    142Simd<float16_t, N> operator||(T a, Simd<float16_t, N> b) {
    +
    143 return Simd<uint16_t, N>((a != 0) || (b != 0));
    +
    144}
    +
    +
    +
    145inline Simd<float16_t, N> operator&&(
    +
    146 Simd<float16_t, N> a,
    +
    147 Simd<float16_t, N> b) {
    +
    148 return Simd<uint16_t, N>((a != 0) && (b != 0));
    +
    149}
    +
    +
    150template <typename T>
    +
    +
    151Simd<float16_t, N> operator&&(Simd<float16_t, N> a, T b) {
    +
    152 return Simd<uint16_t, N>((a != 0) && (b != 0));
    +
    153}
    +
    +
    154template <typename T>
    +
    +
    155Simd<float16_t, N> operator&&(T a, Simd<float16_t, N> b) {
    +
    156 return Simd<uint16_t, N>((a != 0) && (b != 0));
    +
    157}
    +
    +
    158
    +
    159template <>
    +
    +
    160inline Simd<bool, N> isnan(Simd<float16_t, N> v) {
    +
    161 return v != v;
    +
    162}
    +
    +
    163
    +
    164template <>
    +
    165inline Simd<float16_t, N>
    + +
    169
    +
    170template <typename T>
    +
    +
    171Simd<float16_t, N> fma(Simd<float16_t, N> x, Simd<float16_t, N> y, T z) {
    +
    172 return vfmaq_f16(x.value, y.value, Simd<float16_t, N>(z).value);
    +
    173}
    +
    +
    174
    +
    175template <typename MaskT>
    +
    176Simd<float16_t, N>
    +
    +
    177select(Simd<MaskT, N> mask, Simd<float16_t, N> x, Simd<float16_t, N> y) {
    +
    178 return vbslq_f16(Simd<uint16_t, N>(mask).value, x.value, y.value);
    +
    179}
    +
    +
    180
    +
    181// Reductions
    +
    +
    182inline float16_t max(Simd<float16_t, N> x) {
    +
    183 float16x4_t y;
    +
    184 y = vpmax_f16(vget_low_f16(x.value), vget_high_f16(x.value));
    +
    185 y = vpmax_f16(y, y);
    +
    186 y = vpmax_f16(y, y);
    +
    187 return vget_lane_f16(y, 0);
    +
    188}
    +
    +
    +
    189inline float16_t min(Simd<float16_t, N> x) {
    +
    190 float16x4_t y;
    +
    191 y = vpmin_f16(vget_low_f16(x.value), vget_high_f16(x.value));
    +
    192 y = vpmin_f16(y, y);
    +
    193 y = vpmin_f16(y, y);
    +
    194 return vget_lane_f16(y, 0);
    +
    195}
    +
    +
    +
    196inline float16_t sum(Simd<float16_t, N> x) {
    +
    197 float16x4_t y;
    +
    198 y = vpadd_f16(vget_low_f16(x.value), vget_high_f16(x.value));
    +
    199 y = vpadd_f16(y, y);
    +
    200 y = vpadd_f16(y, y);
    +
    201 return vget_lane_f16(y, 0);
    +
    202}
    +
    +
    +
    203inline float16_t prod(Simd<float16_t, N> x) {
    +
    204 auto hx = vmul_f16(vget_low_f16(x.value), vget_high_f16(x.value));
    +
    205 auto out = hx[0];
    +
    206 hx[0] *= hx[1];
    +
    207 hx[0] *= hx[2];
    +
    208 hx[0] *= hx[3];
    +
    209 return hx[0];
    +
    210}
    +
    +
    211
    +
    212} // namespace mlx::core::simd
    +
    base_simd.h
    +
    mlx::core::simd
    Definition accelerate_fp16_simd.h:9
    +
    mlx::core::simd::isnan
    Simd< bool, N > isnan(Simd< T, N > v)
    Definition accelerate_simd.h:141
    +
    mlx::core::simd::N
    constexpr int N
    Definition neon_fp16_simd.h:9
    +
    mlx::core::simd::minimum
    Simd< T, N > minimum(Simd< T, N > a, Simd< T, N > b)
    Definition accelerate_simd.h:210
    +
    mlx::core::simd::prod
    T prod(Simd< T, N > x)
    Definition accelerate_simd.h:292
    +
    mlx::core::simd::rint
    Simd< T, N > rint(Simd< T, N > v)
    Definition accelerate_simd.h:127
    +
    mlx::core::simd::load
    Simd< T, N > load(const T *x)
    Definition base_simd.h:27
    +
    mlx::core::simd::operator!=
    Simd< bool, N > operator!=(Simd< T, N > a, U b)
    Definition accelerate_simd.h:196
    +
    mlx::core::simd::abs
    Simd< T, N > abs(Simd< T, N > v)
    Definition accelerate_simd.h:112
    +
    mlx::core::simd::sum
    T sum(Simd< T, N > x)
    Definition accelerate_simd.h:279
    +
    mlx::core::simd::max
    T max(Simd< T, N > x)
    Definition accelerate_simd.h:283
    +
    mlx::core::simd::operator!
    Simd< bool, N > operator!(Simd< T, N > v)
    Definition accelerate_simd.h:147
    +
    mlx::core::simd::maximum
    Simd< T, N > maximum(Simd< T, N > a, Simd< T, N > b)
    Definition accelerate_simd.h:204
    +
    mlx::core::simd::operator&&
    Simd< T, N > operator&&(Simd< T, N > x, U y)
    Definition accelerate_simd.h:174
    +
    mlx::core::simd::floor
    Simd< T, N > floor(Simd< T, N > v)
    Definition accelerate_simd.h:113
    +
    mlx::core::simd::fma
    Simd< T, N > fma(Simd< T, N > x, Simd< T, N > y, U z)
    Definition accelerate_simd.h:264
    +
    mlx::core::simd::operator||
    Simd< T, N > operator||(Simd< T, N > x, U y)
    Definition accelerate_simd.h:175
    +
    mlx::core::simd::min
    T min(Simd< T, N > x)
    Definition accelerate_simd.h:287
    +
    mlx::core::simd::ceil
    Simd< T, N > ceil(Simd< T, N > v)
    Definition accelerate_simd.h:120
    +
    mlx::core::simd::recip
    Simd< T, N > recip(Simd< T, N > v)
    Definition accelerate_simd.h:131
    +
    mlx::core::simd::sqrt
    Simd< T, N > sqrt(Simd< T, N > v)
    Definition accelerate_simd.h:129
    +
    mlx::core::simd::clamp
    Simd< T, N > clamp(Simd< T, N > v, Simd< T, N > min, Simd< T, N > max)
    Definition accelerate_simd.h:259
    +
    mlx::core::simd::rsqrt
    Simd< T, N > rsqrt(Simd< T, N > v)
    Definition accelerate_simd.h:130
    +
    mlx::core::simd::operator-
    Simd< T, N > operator-(Simd< T, N > v)
    Definition accelerate_simd.h:136
    +
    mlx::core::simd::select
    Simd< T1, N > select(Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)
    Definition accelerate_simd.h:231
    +
    mlx::core::float16_t
    struct _MLX_Float16 float16_t
    Definition half_types.h:17
    +
    DEFINE_NEON_BINARY_OP
    #define DEFINE_NEON_BINARY_OP(name, op)
    Definition neon_fp16_simd.h:68
    +
    DEFINE_NEON_COMPARISON
    #define DEFINE_NEON_COMPARISON(Op, op)
    Definition neon_fp16_simd.h:97
    +
    DEFINE_NEON_UNARY_OP
    #define DEFINE_NEON_UNARY_OP(name, op)
    Definition neon_fp16_simd.h:55
    +
    mlx::core::simd::Simd< float16_t, N >::Simd
    Simd()
    Definition neon_fp16_simd.h:16
    +
    mlx::core::simd::Simd< float16_t, N >::size
    static constexpr int size
    Definition neon_fp16_simd.h:13
    +
    mlx::core::simd::Simd< float16_t, N >::scalar_t
    float16_t scalar_t
    Definition neon_fp16_simd.h:14
    +
    mlx::core::simd::Simd< float16_t, N >::operator[]
    float16_t operator[](int idx) const
    Definition neon_fp16_simd.h:44
    +
    mlx::core::simd::Simd< float16_t, N >::operator[]
    float16_t & operator[](int idx)
    Definition neon_fp16_simd.h:48
    +
    mlx::core::simd::Simd< float16_t, N >::value
    float16x8_t value
    Definition neon_fp16_simd.h:52
    +
    mlx::core::simd::Simd
    Definition accelerate_simd.h:55
    +
    mlx::core::simd::Simd::value
    asd::Vector< scalar_t, N >::packed_t value
    Definition accelerate_simd.h:80
    +
    mlx::core::simd::Simd::Simd
    Simd()
    Definition accelerate_simd.h:59
    +
    +
    + + + + diff --git a/docs/build/html/objects.inv b/docs/build/html/objects.inv index c47b24455..798290458 100644 Binary files a/docs/build/html/objects.inv and b/docs/build/html/objects.inv differ diff --git a/docs/build/html/ops_8h.html b/docs/build/html/ops_8h.html index de0cb6ce4..6e46fd7c0 100644 --- a/docs/build/html/ops_8h.html +++ b/docs/build/html/ops_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/ops.h File Reference @@ -52,7 +52,7 @@
    - + + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.Dtype.html b/docs/build/html/python/_autosummary/mlx.core.Dtype.html index b2786f676..a1a56ef2b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.Dtype.html +++ b/docs/build/html/python/_autosummary/mlx.core.Dtype.html @@ -8,7 +8,7 @@ - mlx.core.Dtype — MLX 0.22.0 documentation + mlx.core.Dtype — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html b/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html index 3525b0166..5662b7582 100644 --- a/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html +++ b/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html @@ -8,7 +8,7 @@ - mlx.core.DtypeCategory — MLX 0.22.0 documentation + mlx.core.DtypeCategory — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.abs.html b/docs/build/html/python/_autosummary/mlx.core.abs.html index 99974ef40..e2641c424 100644 --- a/docs/build/html/python/_autosummary/mlx.core.abs.html +++ b/docs/build/html/python/_autosummary/mlx.core.abs.html @@ -8,7 +8,7 @@ - mlx.core.abs — MLX 0.22.0 documentation + mlx.core.abs — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.add.html b/docs/build/html/python/_autosummary/mlx.core.add.html index 9a718a830..fcb997785 100644 --- a/docs/build/html/python/_autosummary/mlx.core.add.html +++ b/docs/build/html/python/_autosummary/mlx.core.add.html @@ -8,7 +8,7 @@ - mlx.core.add — MLX 0.22.0 documentation + mlx.core.add — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.addmm.html b/docs/build/html/python/_autosummary/mlx.core.addmm.html index 0442d2198..0ad5cdeca 100644 --- a/docs/build/html/python/_autosummary/mlx.core.addmm.html +++ b/docs/build/html/python/_autosummary/mlx.core.addmm.html @@ -8,7 +8,7 @@ - mlx.core.addmm — MLX 0.22.0 documentation + mlx.core.addmm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.all.html b/docs/build/html/python/_autosummary/mlx.core.all.html index 2e9e1800d..a70fd9349 100644 --- a/docs/build/html/python/_autosummary/mlx.core.all.html +++ b/docs/build/html/python/_autosummary/mlx.core.all.html @@ -8,7 +8,7 @@ - mlx.core.all — MLX 0.22.0 documentation + mlx.core.all — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.allclose.html b/docs/build/html/python/_autosummary/mlx.core.allclose.html index a1b5a2c8c..133584dc5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.allclose.html +++ b/docs/build/html/python/_autosummary/mlx.core.allclose.html @@ -8,7 +8,7 @@ - mlx.core.allclose — MLX 0.22.0 documentation + mlx.core.allclose — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.any.html b/docs/build/html/python/_autosummary/mlx.core.any.html index 39d84d3e2..4e943dcde 100644 --- a/docs/build/html/python/_autosummary/mlx.core.any.html +++ b/docs/build/html/python/_autosummary/mlx.core.any.html @@ -8,7 +8,7 @@ - mlx.core.any — MLX 0.22.0 documentation + mlx.core.any — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.arange.html b/docs/build/html/python/_autosummary/mlx.core.arange.html index b8d64a754..a094ac61e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arange.html +++ b/docs/build/html/python/_autosummary/mlx.core.arange.html @@ -8,7 +8,7 @@ - mlx.core.arange — MLX 0.22.0 documentation + mlx.core.arange — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.arccos.html b/docs/build/html/python/_autosummary/mlx.core.arccos.html index 50b0929d6..414348d61 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arccos.html +++ b/docs/build/html/python/_autosummary/mlx.core.arccos.html @@ -8,7 +8,7 @@ - mlx.core.arccos — MLX 0.22.0 documentation + mlx.core.arccos — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.arccosh.html b/docs/build/html/python/_autosummary/mlx.core.arccosh.html index 3191fe5b4..24cce2804 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arccosh.html +++ b/docs/build/html/python/_autosummary/mlx.core.arccosh.html @@ -8,7 +8,7 @@ - mlx.core.arccosh — MLX 0.22.0 documentation + mlx.core.arccosh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsin.html b/docs/build/html/python/_autosummary/mlx.core.arcsin.html index b130e8842..8ba841ff2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arcsin.html +++ b/docs/build/html/python/_autosummary/mlx.core.arcsin.html @@ -8,7 +8,7 @@ - mlx.core.arcsin — MLX 0.22.0 documentation + mlx.core.arcsin — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsinh.html b/docs/build/html/python/_autosummary/mlx.core.arcsinh.html index fc4ad34e7..fdaeb243e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arcsinh.html +++ b/docs/build/html/python/_autosummary/mlx.core.arcsinh.html @@ -8,7 +8,7 @@ - mlx.core.arcsinh — MLX 0.22.0 documentation + mlx.core.arcsinh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan.html b/docs/build/html/python/_autosummary/mlx.core.arctan.html index 4a150f287..a2eebd025 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arctan.html +++ b/docs/build/html/python/_autosummary/mlx.core.arctan.html @@ -8,7 +8,7 @@ - mlx.core.arctan — MLX 0.22.0 documentation + mlx.core.arctan — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan2.html b/docs/build/html/python/_autosummary/mlx.core.arctan2.html index d14490358..9e478f74e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arctan2.html +++ b/docs/build/html/python/_autosummary/mlx.core.arctan2.html @@ -8,7 +8,7 @@ - mlx.core.arctan2 — MLX 0.22.0 documentation + mlx.core.arctan2 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.arctanh.html b/docs/build/html/python/_autosummary/mlx.core.arctanh.html index 6573f955f..b10d1d032 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arctanh.html +++ b/docs/build/html/python/_autosummary/mlx.core.arctanh.html @@ -8,7 +8,7 @@ - mlx.core.arctanh — MLX 0.22.0 documentation + mlx.core.arctanh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.argmax.html b/docs/build/html/python/_autosummary/mlx.core.argmax.html index 78836ed85..ecec4742d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.argmax.html +++ b/docs/build/html/python/_autosummary/mlx.core.argmax.html @@ -8,7 +8,7 @@ - mlx.core.argmax — MLX 0.22.0 documentation + mlx.core.argmax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.argmin.html b/docs/build/html/python/_autosummary/mlx.core.argmin.html index dd87a3c27..425f537b5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.argmin.html +++ b/docs/build/html/python/_autosummary/mlx.core.argmin.html @@ -8,7 +8,7 @@ - mlx.core.argmin — MLX 0.22.0 documentation + mlx.core.argmin — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.argpartition.html b/docs/build/html/python/_autosummary/mlx.core.argpartition.html index 727956296..4105bd9b5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.argpartition.html +++ b/docs/build/html/python/_autosummary/mlx.core.argpartition.html @@ -8,7 +8,7 @@ - mlx.core.argpartition — MLX 0.22.0 documentation + mlx.core.argpartition — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.argsort.html b/docs/build/html/python/_autosummary/mlx.core.argsort.html index 212b0b44b..0fb6c744a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.argsort.html +++ b/docs/build/html/python/_autosummary/mlx.core.argsort.html @@ -8,7 +8,7 @@ - mlx.core.argsort — MLX 0.22.0 documentation + mlx.core.argsort — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.T.html b/docs/build/html/python/_autosummary/mlx.core.array.T.html index 2b059b545..ee75eb4ee 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.T.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.T.html @@ -8,7 +8,7 @@ - mlx.core.array.T — MLX 0.22.0 documentation + mlx.core.array.T — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.abs.html b/docs/build/html/python/_autosummary/mlx.core.array.abs.html index 77ea18ed6..e3ebaac14 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.abs.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.abs.html @@ -8,7 +8,7 @@ - mlx.core.array.abs — MLX 0.22.0 documentation + mlx.core.array.abs — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.all.html b/docs/build/html/python/_autosummary/mlx.core.array.all.html index 22253b5b1..9f40384d6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.all.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.all.html @@ -8,7 +8,7 @@ - mlx.core.array.all — MLX 0.22.0 documentation + mlx.core.array.all — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.any.html b/docs/build/html/python/_autosummary/mlx.core.array.any.html index 967f9de2b..a817833c3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.any.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.any.html @@ -8,7 +8,7 @@ - mlx.core.array.any — MLX 0.22.0 documentation + mlx.core.array.any — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.argmax.html b/docs/build/html/python/_autosummary/mlx.core.array.argmax.html index 6f47985c4..3d25fbd07 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.argmax.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.argmax.html @@ -8,7 +8,7 @@ - mlx.core.array.argmax — MLX 0.22.0 documentation + mlx.core.array.argmax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.argmin.html b/docs/build/html/python/_autosummary/mlx.core.array.argmin.html index 097c1a0e8..3e0f52aa4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.argmin.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.argmin.html @@ -8,7 +8,7 @@ - mlx.core.array.argmin — MLX 0.22.0 documentation + mlx.core.array.argmin — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.astype.html b/docs/build/html/python/_autosummary/mlx.core.array.astype.html index 0f2378601..6e34bd4b0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.astype.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.astype.html @@ -8,7 +8,7 @@ - mlx.core.array.astype — MLX 0.22.0 documentation + mlx.core.array.astype — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.at.html b/docs/build/html/python/_autosummary/mlx.core.array.at.html index 8ccfeb777..dd49bd56e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.at.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.at.html @@ -8,7 +8,7 @@ - mlx.core.array.at — MLX 0.22.0 documentation + mlx.core.array.at — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.conj.html b/docs/build/html/python/_autosummary/mlx.core.array.conj.html index 6dcddc97a..2f48f9183 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.conj.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.conj.html @@ -8,7 +8,7 @@ - mlx.core.array.conj — MLX 0.22.0 documentation + mlx.core.array.conj — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cos.html b/docs/build/html/python/_autosummary/mlx.core.array.cos.html index fe7ca076a..73212c58e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cos.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cos.html @@ -8,7 +8,7 @@ - mlx.core.array.cos — MLX 0.22.0 documentation + mlx.core.array.cos — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cummax.html b/docs/build/html/python/_autosummary/mlx.core.array.cummax.html index e41c187b4..96089efc2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cummax.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cummax.html @@ -8,7 +8,7 @@ - mlx.core.array.cummax — MLX 0.22.0 documentation + mlx.core.array.cummax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cummin.html b/docs/build/html/python/_autosummary/mlx.core.array.cummin.html index 4260bcc9e..c5d9e28d6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cummin.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cummin.html @@ -8,7 +8,7 @@ - mlx.core.array.cummin — MLX 0.22.0 documentation + mlx.core.array.cummin — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html b/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html index 3e476dfaf..5ca24905a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html @@ -8,7 +8,7 @@ - mlx.core.array.cumprod — MLX 0.22.0 documentation + mlx.core.array.cumprod — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html b/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html index efe86cf5f..3db5a695b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html @@ -8,7 +8,7 @@ - mlx.core.array.cumsum — MLX 0.22.0 documentation + mlx.core.array.cumsum — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.diag.html b/docs/build/html/python/_autosummary/mlx.core.array.diag.html index 4df4781a5..d5092ad4e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.diag.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.diag.html @@ -8,7 +8,7 @@ - mlx.core.array.diag — MLX 0.22.0 documentation + mlx.core.array.diag — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html b/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html index f2b5849a1..6bc55578f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html @@ -8,7 +8,7 @@ - mlx.core.array.diagonal — MLX 0.22.0 documentation + mlx.core.array.diagonal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.dtype.html b/docs/build/html/python/_autosummary/mlx.core.array.dtype.html index 3d08b2b26..371a9d20f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.dtype.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.dtype.html @@ -8,7 +8,7 @@ - mlx.core.array.dtype — MLX 0.22.0 documentation + mlx.core.array.dtype — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.exp.html b/docs/build/html/python/_autosummary/mlx.core.array.exp.html index 11c2ba175..0d505aa53 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.exp.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.exp.html @@ -8,7 +8,7 @@ - mlx.core.array.exp — MLX 0.22.0 documentation + mlx.core.array.exp — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.flatten.html b/docs/build/html/python/_autosummary/mlx.core.array.flatten.html index cb75fab21..133fa5132 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.flatten.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.flatten.html @@ -8,7 +8,7 @@ - mlx.core.array.flatten — MLX 0.22.0 documentation + mlx.core.array.flatten — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.html b/docs/build/html/python/_autosummary/mlx.core.array.html index 21ef3f40b..65945696c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.html @@ -8,7 +8,7 @@ - mlx.core.array — MLX 0.22.0 documentation + mlx.core.array — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.item.html b/docs/build/html/python/_autosummary/mlx.core.array.item.html index bc884d595..6e0ab14ff 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.item.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.item.html @@ -8,7 +8,7 @@ - mlx.core.array.item — MLX 0.22.0 documentation + mlx.core.array.item — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html b/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html index f859082da..8bf65647f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html @@ -8,7 +8,7 @@ - mlx.core.array.itemsize — MLX 0.22.0 documentation + mlx.core.array.itemsize — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log.html b/docs/build/html/python/_autosummary/mlx.core.array.log.html index 8ce6ec9ca..d6454d4a1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.log.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.log.html @@ -8,7 +8,7 @@ - mlx.core.array.log — MLX 0.22.0 documentation + mlx.core.array.log — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log10.html b/docs/build/html/python/_autosummary/mlx.core.array.log10.html index a26198ed1..ed0805d41 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.log10.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.log10.html @@ -8,7 +8,7 @@ - mlx.core.array.log10 — MLX 0.22.0 documentation + mlx.core.array.log10 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log1p.html b/docs/build/html/python/_autosummary/mlx.core.array.log1p.html index 1f6688432..a15131341 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.log1p.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.log1p.html @@ -8,7 +8,7 @@ - mlx.core.array.log1p — MLX 0.22.0 documentation + mlx.core.array.log1p — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log2.html b/docs/build/html/python/_autosummary/mlx.core.array.log2.html index e1baece77..95df4db6f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.log2.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.log2.html @@ -8,7 +8,7 @@ - mlx.core.array.log2 — MLX 0.22.0 documentation + mlx.core.array.log2 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html b/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html index 9c01c8e5f..0b307a028 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html @@ -8,7 +8,7 @@ - mlx.core.array.logsumexp — MLX 0.22.0 documentation + mlx.core.array.logsumexp — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.max.html b/docs/build/html/python/_autosummary/mlx.core.array.max.html index 729a788ef..f29cd61fc 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.max.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.max.html @@ -8,7 +8,7 @@ - mlx.core.array.max — MLX 0.22.0 documentation + mlx.core.array.max — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.mean.html b/docs/build/html/python/_autosummary/mlx.core.array.mean.html index 3d1bea927..9b3458c01 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.mean.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.mean.html @@ -8,7 +8,7 @@ - mlx.core.array.mean — MLX 0.22.0 documentation + mlx.core.array.mean — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.min.html b/docs/build/html/python/_autosummary/mlx.core.array.min.html index 57df959ab..0139f111e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.min.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.min.html @@ -8,7 +8,7 @@ - mlx.core.array.min — MLX 0.22.0 documentation + mlx.core.array.min — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html b/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html index d4ed852e0..62c40df06 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html @@ -8,7 +8,7 @@ - mlx.core.array.moveaxis — MLX 0.22.0 documentation + mlx.core.array.moveaxis — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html b/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html index 7ad618efb..d8d35b141 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html @@ -8,7 +8,7 @@ - mlx.core.array.nbytes — MLX 0.22.0 documentation + mlx.core.array.nbytes — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.ndim.html b/docs/build/html/python/_autosummary/mlx.core.array.ndim.html index 17909d951..2c408742b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.ndim.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.ndim.html @@ -8,7 +8,7 @@ - mlx.core.array.ndim — MLX 0.22.0 documentation + mlx.core.array.ndim — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.prod.html b/docs/build/html/python/_autosummary/mlx.core.array.prod.html index 94eb62ec5..1cd8a5535 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.prod.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.prod.html @@ -8,7 +8,7 @@ - mlx.core.array.prod — MLX 0.22.0 documentation + mlx.core.array.prod — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html b/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html index a2e9ce24f..b68f5a592 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html @@ -8,7 +8,7 @@ - mlx.core.array.reciprocal — MLX 0.22.0 documentation + mlx.core.array.reciprocal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.reshape.html b/docs/build/html/python/_autosummary/mlx.core.array.reshape.html index d0e127d61..97e428055 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.reshape.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.reshape.html @@ -8,7 +8,7 @@ - mlx.core.array.reshape — MLX 0.22.0 documentation + mlx.core.array.reshape — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.round.html b/docs/build/html/python/_autosummary/mlx.core.array.round.html index a5d00233c..176b5ee1c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.round.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.round.html @@ -8,7 +8,7 @@ - mlx.core.array.round — MLX 0.22.0 documentation + mlx.core.array.round — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html b/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html index 1f7841ba6..6637dcb1c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html @@ -8,7 +8,7 @@ - mlx.core.array.rsqrt — MLX 0.22.0 documentation + mlx.core.array.rsqrt — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.shape.html b/docs/build/html/python/_autosummary/mlx.core.array.shape.html index 7918b7d27..b912f07da 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.shape.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.shape.html @@ -8,7 +8,7 @@ - mlx.core.array.shape — MLX 0.22.0 documentation + mlx.core.array.shape — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.sin.html b/docs/build/html/python/_autosummary/mlx.core.array.sin.html index 836f6988f..4ebaa5878 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.sin.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.sin.html @@ -8,7 +8,7 @@ - mlx.core.array.sin — MLX 0.22.0 documentation + mlx.core.array.sin — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.size.html b/docs/build/html/python/_autosummary/mlx.core.array.size.html index 0cf3d4667..06a230c20 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.size.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.size.html @@ -8,7 +8,7 @@ - mlx.core.array.size — MLX 0.22.0 documentation + mlx.core.array.size — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.split.html b/docs/build/html/python/_autosummary/mlx.core.array.split.html index 341b444ff..d909cbd91 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.split.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.split.html @@ -8,7 +8,7 @@ - mlx.core.array.split — MLX 0.22.0 documentation + mlx.core.array.split — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html b/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html index 64a538f53..e1346b512 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html @@ -8,7 +8,7 @@ - mlx.core.array.sqrt — MLX 0.22.0 documentation + mlx.core.array.sqrt — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.square.html b/docs/build/html/python/_autosummary/mlx.core.array.square.html index df0a9ae03..d8ac62e1f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.square.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.square.html @@ -8,7 +8,7 @@ - mlx.core.array.square — MLX 0.22.0 documentation + mlx.core.array.square — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html b/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html index 18b3b6323..85f6513dc 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html @@ -8,7 +8,7 @@ - mlx.core.array.squeeze — MLX 0.22.0 documentation + mlx.core.array.squeeze — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.std.html b/docs/build/html/python/_autosummary/mlx.core.array.std.html index 339e85bb2..953ba45af 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.std.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.std.html @@ -8,7 +8,7 @@ - mlx.core.array.std — MLX 0.22.0 documentation + mlx.core.array.std — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.sum.html b/docs/build/html/python/_autosummary/mlx.core.array.sum.html index 281d1bbfd..c951c06a2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.sum.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.sum.html @@ -8,7 +8,7 @@ - mlx.core.array.sum — MLX 0.22.0 documentation + mlx.core.array.sum — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html b/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html index cf899c80d..f10963659 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html @@ -8,7 +8,7 @@ - mlx.core.array.swapaxes — MLX 0.22.0 documentation + mlx.core.array.swapaxes — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.tolist.html b/docs/build/html/python/_autosummary/mlx.core.array.tolist.html index 41ac02881..6655e4e9e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.tolist.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.tolist.html @@ -8,7 +8,7 @@ - mlx.core.array.tolist — MLX 0.22.0 documentation + mlx.core.array.tolist — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.transpose.html b/docs/build/html/python/_autosummary/mlx.core.array.transpose.html index 8ed29a11b..03712270b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.transpose.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.transpose.html @@ -8,7 +8,7 @@ - mlx.core.array.transpose — MLX 0.22.0 documentation + mlx.core.array.transpose — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.var.html b/docs/build/html/python/_autosummary/mlx.core.array.var.html index 1ff9c5624..eb87eeea6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.var.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.var.html @@ -8,7 +8,7 @@ - mlx.core.array.var — MLX 0.22.0 documentation + mlx.core.array.var — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array.view.html b/docs/build/html/python/_autosummary/mlx.core.array.view.html index 72a57223e..b05b16746 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.view.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.view.html @@ -8,7 +8,7 @@ - mlx.core.array.view — MLX 0.22.0 documentation + mlx.core.array.view — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.array_equal.html b/docs/build/html/python/_autosummary/mlx.core.array_equal.html index 5e37d5baf..13879f304 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array_equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.array_equal.html @@ -8,7 +8,7 @@ - mlx.core.array_equal — MLX 0.22.0 documentation + mlx.core.array_equal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.as_strided.html b/docs/build/html/python/_autosummary/mlx.core.as_strided.html index 21984298d..8bfe199fb 100644 --- a/docs/build/html/python/_autosummary/mlx.core.as_strided.html +++ b/docs/build/html/python/_autosummary/mlx.core.as_strided.html @@ -8,7 +8,7 @@ - mlx.core.as_strided — MLX 0.22.0 documentation + mlx.core.as_strided — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html index d79784adb..30874acd3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html +++ b/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html @@ -8,7 +8,7 @@ - mlx.core.atleast_1d — MLX 0.22.0 documentation + mlx.core.atleast_1d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html index 4020151ad..c20dca552 100644 --- a/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html +++ b/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html @@ -8,7 +8,7 @@ - mlx.core.atleast_2d — MLX 0.22.0 documentation + mlx.core.atleast_2d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html index 94fc9d6f7..6ec9315ab 100644 --- a/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html +++ b/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html @@ -8,7 +8,7 @@ - mlx.core.atleast_3d — MLX 0.22.0 documentation + mlx.core.atleast_3d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html index 35be4154e..fbbd2e941 100644 --- a/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html +++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html @@ -8,7 +8,7 @@ - mlx.core.bitwise_and — MLX 0.22.0 documentation + mlx.core.bitwise_and — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html index 7d15a5278..8ec32a62b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html +++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html @@ -8,7 +8,7 @@ - mlx.core.bitwise_or — MLX 0.22.0 documentation + mlx.core.bitwise_or — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html index 0406a6d82..ee58adec0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html +++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html @@ -8,7 +8,7 @@ - mlx.core.bitwise_xor — MLX 0.22.0 documentation + mlx.core.bitwise_xor — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html b/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html index 710e07365..1e67353c5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html +++ b/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html @@ -8,7 +8,7 @@ - mlx.core.block_masked_mm — MLX 0.22.0 documentation + mlx.core.block_masked_mm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.broadcast_to.html b/docs/build/html/python/_autosummary/mlx.core.broadcast_to.html index ee19454f1..b3648d352 100644 --- a/docs/build/html/python/_autosummary/mlx.core.broadcast_to.html +++ b/docs/build/html/python/_autosummary/mlx.core.broadcast_to.html @@ -8,7 +8,7 @@ - mlx.core.broadcast_to — MLX 0.22.0 documentation + mlx.core.broadcast_to — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.ceil.html b/docs/build/html/python/_autosummary/mlx.core.ceil.html index 88ffa2239..b3529fb48 100644 --- a/docs/build/html/python/_autosummary/mlx.core.ceil.html +++ b/docs/build/html/python/_autosummary/mlx.core.ceil.html @@ -8,7 +8,7 @@ - mlx.core.ceil — MLX 0.22.0 documentation + mlx.core.ceil — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.clip.html b/docs/build/html/python/_autosummary/mlx.core.clip.html index 54e7d3861..2bc6be17e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.clip.html +++ b/docs/build/html/python/_autosummary/mlx.core.clip.html @@ -8,7 +8,7 @@ - mlx.core.clip — MLX 0.22.0 documentation + mlx.core.clip — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.compile.html b/docs/build/html/python/_autosummary/mlx.core.compile.html index 9c8bfeb31..f52dd46ce 100644 --- a/docs/build/html/python/_autosummary/mlx.core.compile.html +++ b/docs/build/html/python/_autosummary/mlx.core.compile.html @@ -8,7 +8,7 @@ - mlx.core.compile — MLX 0.22.0 documentation + mlx.core.compile — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.concatenate.html b/docs/build/html/python/_autosummary/mlx.core.concatenate.html index 3a5692ca1..d9e74ea9b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.concatenate.html +++ b/docs/build/html/python/_autosummary/mlx.core.concatenate.html @@ -8,7 +8,7 @@ - mlx.core.concatenate — MLX 0.22.0 documentation + mlx.core.concatenate — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.conj.html b/docs/build/html/python/_autosummary/mlx.core.conj.html index 3e8f8215d..6e66269bb 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conj.html +++ b/docs/build/html/python/_autosummary/mlx.core.conj.html @@ -8,7 +8,7 @@ - mlx.core.conj — MLX 0.22.0 documentation + mlx.core.conj — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.conjugate.html b/docs/build/html/python/_autosummary/mlx.core.conjugate.html index e43eedd78..b69a47fae 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conjugate.html +++ b/docs/build/html/python/_autosummary/mlx.core.conjugate.html @@ -8,7 +8,7 @@ - mlx.core.conjugate — MLX 0.22.0 documentation + mlx.core.conjugate — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.conv1d.html b/docs/build/html/python/_autosummary/mlx.core.conv1d.html index c78ffec49..00683ad5a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv1d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv1d.html @@ -8,7 +8,7 @@ - mlx.core.conv1d — MLX 0.22.0 documentation + mlx.core.conv1d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.conv2d.html b/docs/build/html/python/_autosummary/mlx.core.conv2d.html index a1d9496fe..40f183d42 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv2d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv2d.html @@ -8,7 +8,7 @@ - mlx.core.conv2d — MLX 0.22.0 documentation + mlx.core.conv2d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.conv3d.html b/docs/build/html/python/_autosummary/mlx.core.conv3d.html index 6765497b3..514f8f182 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv3d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv3d.html @@ -8,7 +8,7 @@ - mlx.core.conv3d — MLX 0.22.0 documentation + mlx.core.conv3d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.conv_general.html b/docs/build/html/python/_autosummary/mlx.core.conv_general.html index 7ac417f73..fe3686b94 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv_general.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv_general.html @@ -8,7 +8,7 @@ - mlx.core.conv_general — MLX 0.22.0 documentation + mlx.core.conv_general — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.conv_transpose1d.html b/docs/build/html/python/_autosummary/mlx.core.conv_transpose1d.html index 97ba73aa2..0d34306a7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv_transpose1d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv_transpose1d.html @@ -8,7 +8,7 @@ - mlx.core.conv_transpose1d — MLX 0.22.0 documentation + mlx.core.conv_transpose1d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.conv_transpose2d.html b/docs/build/html/python/_autosummary/mlx.core.conv_transpose2d.html index 8aea067de..0d8de88d1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv_transpose2d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv_transpose2d.html @@ -8,7 +8,7 @@ - mlx.core.conv_transpose2d — MLX 0.22.0 documentation + mlx.core.conv_transpose2d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.conv_transpose3d.html b/docs/build/html/python/_autosummary/mlx.core.conv_transpose3d.html index da6747272..6901a3cbe 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv_transpose3d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv_transpose3d.html @@ -8,7 +8,7 @@ - mlx.core.conv_transpose3d — MLX 0.22.0 documentation + mlx.core.conv_transpose3d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.convolve.html b/docs/build/html/python/_autosummary/mlx.core.convolve.html index 06b04a5a5..f6c1d112b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.convolve.html +++ b/docs/build/html/python/_autosummary/mlx.core.convolve.html @@ -8,7 +8,7 @@ - mlx.core.convolve — MLX 0.22.0 documentation + mlx.core.convolve — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.cos.html b/docs/build/html/python/_autosummary/mlx.core.cos.html index 956aa6104..5323e5371 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cos.html +++ b/docs/build/html/python/_autosummary/mlx.core.cos.html @@ -8,7 +8,7 @@ - mlx.core.cos — MLX 0.22.0 documentation + mlx.core.cos — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.cosh.html b/docs/build/html/python/_autosummary/mlx.core.cosh.html index 0b742a4af..1417ff90a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cosh.html +++ b/docs/build/html/python/_autosummary/mlx.core.cosh.html @@ -8,7 +8,7 @@ - mlx.core.cosh — MLX 0.22.0 documentation + mlx.core.cosh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.cummax.html b/docs/build/html/python/_autosummary/mlx.core.cummax.html index d80cecb3c..80aacb288 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cummax.html +++ b/docs/build/html/python/_autosummary/mlx.core.cummax.html @@ -8,7 +8,7 @@ - mlx.core.cummax — MLX 0.22.0 documentation + mlx.core.cummax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.cummin.html b/docs/build/html/python/_autosummary/mlx.core.cummin.html index 2069fb1ce..8e517ce07 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cummin.html +++ b/docs/build/html/python/_autosummary/mlx.core.cummin.html @@ -8,7 +8,7 @@ - mlx.core.cummin — MLX 0.22.0 documentation + mlx.core.cummin — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.cumprod.html b/docs/build/html/python/_autosummary/mlx.core.cumprod.html index 7c8b8fd8b..552dd00b0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cumprod.html +++ b/docs/build/html/python/_autosummary/mlx.core.cumprod.html @@ -8,7 +8,7 @@ - mlx.core.cumprod — MLX 0.22.0 documentation + mlx.core.cumprod — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.cumsum.html b/docs/build/html/python/_autosummary/mlx.core.cumsum.html index f730e1b4b..8ffa25b9a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cumsum.html +++ b/docs/build/html/python/_autosummary/mlx.core.cumsum.html @@ -8,7 +8,7 @@ - mlx.core.cumsum — MLX 0.22.0 documentation + mlx.core.cumsum — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.custom_function.html b/docs/build/html/python/_autosummary/mlx.core.custom_function.html index 490c7d05c..63e4aa38f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.custom_function.html +++ b/docs/build/html/python/_autosummary/mlx.core.custom_function.html @@ -8,7 +8,7 @@ - mlx.core.custom_function — MLX 0.22.0 documentation + mlx.core.custom_function — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.default_device.html b/docs/build/html/python/_autosummary/mlx.core.default_device.html index 5325b2ea8..8539028c8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.default_device.html +++ b/docs/build/html/python/_autosummary/mlx.core.default_device.html @@ -8,7 +8,7 @@ - mlx.core.default_device — MLX 0.22.0 documentation + mlx.core.default_device — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.default_stream.html b/docs/build/html/python/_autosummary/mlx.core.default_stream.html index 7a0337caa..b841779ff 100644 --- a/docs/build/html/python/_autosummary/mlx.core.default_stream.html +++ b/docs/build/html/python/_autosummary/mlx.core.default_stream.html @@ -8,7 +8,7 @@ - mlx.core.default_stream — MLX 0.22.0 documentation + mlx.core.default_stream — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.degrees.html b/docs/build/html/python/_autosummary/mlx.core.degrees.html index ae25fd699..d6defe383 100644 --- a/docs/build/html/python/_autosummary/mlx.core.degrees.html +++ b/docs/build/html/python/_autosummary/mlx.core.degrees.html @@ -8,7 +8,7 @@ - mlx.core.degrees — MLX 0.22.0 documentation + mlx.core.degrees — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.dequantize.html b/docs/build/html/python/_autosummary/mlx.core.dequantize.html index ad21cc664..28704d31b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.dequantize.html +++ b/docs/build/html/python/_autosummary/mlx.core.dequantize.html @@ -8,7 +8,7 @@ - mlx.core.dequantize — MLX 0.22.0 documentation + mlx.core.dequantize — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.diag.html b/docs/build/html/python/_autosummary/mlx.core.diag.html index 459fa2349..079c3a304 100644 --- a/docs/build/html/python/_autosummary/mlx.core.diag.html +++ b/docs/build/html/python/_autosummary/mlx.core.diag.html @@ -8,7 +8,7 @@ - mlx.core.diag — MLX 0.22.0 documentation + mlx.core.diag — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.diagonal.html b/docs/build/html/python/_autosummary/mlx.core.diagonal.html index b822e6763..a28d57529 100644 --- a/docs/build/html/python/_autosummary/mlx.core.diagonal.html +++ b/docs/build/html/python/_autosummary/mlx.core.diagonal.html @@ -8,7 +8,7 @@ - mlx.core.diagonal — MLX 0.22.0 documentation + mlx.core.diagonal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.disable_compile.html b/docs/build/html/python/_autosummary/mlx.core.disable_compile.html index 375b63d14..6d0b43c64 100644 --- a/docs/build/html/python/_autosummary/mlx.core.disable_compile.html +++ b/docs/build/html/python/_autosummary/mlx.core.disable_compile.html @@ -8,7 +8,7 @@ - mlx.core.disable_compile — MLX 0.22.0 documentation + mlx.core.disable_compile — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html b/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html index 2579dc991..993d53e41 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html @@ -8,7 +8,7 @@ - mlx.core.distributed.Group — MLX 0.22.0 documentation + mlx.core.distributed.Group — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html b/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html index cd269fefa..db717762c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html @@ -8,7 +8,7 @@ - mlx.core.distributed.all_gather — MLX 0.22.0 documentation + mlx.core.distributed.all_gather — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html b/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html index e457ea144..f89d9b3d2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html @@ -8,7 +8,7 @@ - mlx.core.distributed.all_sum — MLX 0.22.0 documentation + mlx.core.distributed.all_sum — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.init.html b/docs/build/html/python/_autosummary/mlx.core.distributed.init.html index db1a677ee..d88b7fe24 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.init.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.init.html @@ -8,7 +8,7 @@ - mlx.core.distributed.init — MLX 0.22.0 documentation + mlx.core.distributed.init — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -885,13 +885,22 @@

    mlx.core.distributed.init#

    -init(strict: bool = False) → Group#
    +init(strict: bool = False, backend: str = 'any') → Group#

    Initialize the communication backend and create the global communication group.

    +

    Example

    +

    import mlx.core as mx

    +

    group = mx.distributed.init(backend=”ring”)

    Parameters:
    -

    strict (bool, optional) – If set to False it returns a singleton group +

      +

    • strict (bool, optional) – If set to False it returns a singleton group in case mx.distributed.is_available() returns False otherwise -it throws a runtime error. Default: False

      +it throws a runtime error. Default: False

      • +

    • backend (str, optional) – Select a specific distributed backend to +initialize. If set to any then try all available backends and +return the first one that succeeds. Subsequent calls will return +the first backend that was initialized. Default: any

      • +
    Returns:

    The group representing all the launched processes.

    diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html b/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html index 54cbed33f..ec5da1158 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html @@ -8,7 +8,7 @@ - mlx.core.distributed.is_available — MLX 0.22.0 documentation + mlx.core.distributed.is_available — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.recv.html b/docs/build/html/python/_autosummary/mlx.core.distributed.recv.html index 4549018c7..a0e7f4725 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.recv.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.recv.html @@ -8,7 +8,7 @@ - mlx.core.distributed.recv — MLX 0.22.0 documentation + mlx.core.distributed.recv — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.recv_like.html b/docs/build/html/python/_autosummary/mlx.core.distributed.recv_like.html index 468dc78ef..c35d32ba2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.recv_like.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.recv_like.html @@ -8,7 +8,7 @@ - mlx.core.distributed.recv_like — MLX 0.22.0 documentation + mlx.core.distributed.recv_like — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.send.html b/docs/build/html/python/_autosummary/mlx.core.distributed.send.html index cf4b6c705..83bbd7c2b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.send.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.send.html @@ -8,7 +8,7 @@ - mlx.core.distributed.send — MLX 0.22.0 documentation + mlx.core.distributed.send — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.divide.html b/docs/build/html/python/_autosummary/mlx.core.divide.html index e803a6174..f5d13f056 100644 --- a/docs/build/html/python/_autosummary/mlx.core.divide.html +++ b/docs/build/html/python/_autosummary/mlx.core.divide.html @@ -8,7 +8,7 @@ - mlx.core.divide — MLX 0.22.0 documentation + mlx.core.divide — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.divmod.html b/docs/build/html/python/_autosummary/mlx.core.divmod.html index a68e6f173..af07df1a6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.divmod.html +++ b/docs/build/html/python/_autosummary/mlx.core.divmod.html @@ -8,7 +8,7 @@ - mlx.core.divmod — MLX 0.22.0 documentation + mlx.core.divmod — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.einsum.html b/docs/build/html/python/_autosummary/mlx.core.einsum.html index c7e8f6362..06ec78bde 100644 --- a/docs/build/html/python/_autosummary/mlx.core.einsum.html +++ b/docs/build/html/python/_autosummary/mlx.core.einsum.html @@ -8,7 +8,7 @@ - mlx.core.einsum — MLX 0.22.0 documentation + mlx.core.einsum — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.einsum_path.html b/docs/build/html/python/_autosummary/mlx.core.einsum_path.html index bc80b1917..ced9529f2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.einsum_path.html +++ b/docs/build/html/python/_autosummary/mlx.core.einsum_path.html @@ -8,7 +8,7 @@ - mlx.core.einsum_path — MLX 0.22.0 documentation + mlx.core.einsum_path — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.enable_compile.html b/docs/build/html/python/_autosummary/mlx.core.enable_compile.html index 5cb0c4ab1..0478e72cf 100644 --- a/docs/build/html/python/_autosummary/mlx.core.enable_compile.html +++ b/docs/build/html/python/_autosummary/mlx.core.enable_compile.html @@ -8,7 +8,7 @@ - mlx.core.enable_compile — MLX 0.22.0 documentation + mlx.core.enable_compile — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.equal.html b/docs/build/html/python/_autosummary/mlx.core.equal.html index 7da48e75e..e93573c46 100644 --- a/docs/build/html/python/_autosummary/mlx.core.equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.equal.html @@ -8,7 +8,7 @@ - mlx.core.equal — MLX 0.22.0 documentation + mlx.core.equal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.erf.html b/docs/build/html/python/_autosummary/mlx.core.erf.html index 146897962..58d03df1b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.erf.html +++ b/docs/build/html/python/_autosummary/mlx.core.erf.html @@ -8,7 +8,7 @@ - mlx.core.erf — MLX 0.22.0 documentation + mlx.core.erf — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.erfinv.html b/docs/build/html/python/_autosummary/mlx.core.erfinv.html index 9500a704a..677f091b7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.erfinv.html +++ b/docs/build/html/python/_autosummary/mlx.core.erfinv.html @@ -8,7 +8,7 @@ - mlx.core.erfinv — MLX 0.22.0 documentation + mlx.core.erfinv — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.eval.html b/docs/build/html/python/_autosummary/mlx.core.eval.html index 7f09eb5a3..c13d97337 100644 --- a/docs/build/html/python/_autosummary/mlx.core.eval.html +++ b/docs/build/html/python/_autosummary/mlx.core.eval.html @@ -8,7 +8,7 @@ - mlx.core.eval — MLX 0.22.0 documentation + mlx.core.eval — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.exp.html b/docs/build/html/python/_autosummary/mlx.core.exp.html index 56140f8a3..967c93580 100644 --- a/docs/build/html/python/_autosummary/mlx.core.exp.html +++ b/docs/build/html/python/_autosummary/mlx.core.exp.html @@ -8,7 +8,7 @@ - mlx.core.exp — MLX 0.22.0 documentation + mlx.core.exp — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.expand_dims.html b/docs/build/html/python/_autosummary/mlx.core.expand_dims.html index bb2afed76..bc924acac 100644 --- a/docs/build/html/python/_autosummary/mlx.core.expand_dims.html +++ b/docs/build/html/python/_autosummary/mlx.core.expand_dims.html @@ -8,7 +8,7 @@ - mlx.core.expand_dims — MLX 0.22.0 documentation + mlx.core.expand_dims — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.expm1.html b/docs/build/html/python/_autosummary/mlx.core.expm1.html index 1df02b42b..7e462a405 100644 --- a/docs/build/html/python/_autosummary/mlx.core.expm1.html +++ b/docs/build/html/python/_autosummary/mlx.core.expm1.html @@ -8,7 +8,7 @@ - mlx.core.expm1 — MLX 0.22.0 documentation + mlx.core.expm1 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.export_function.html b/docs/build/html/python/_autosummary/mlx.core.export_function.html index 5fc73cfd4..83e392ef7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.export_function.html +++ b/docs/build/html/python/_autosummary/mlx.core.export_function.html @@ -8,7 +8,7 @@ - mlx.core.export_function — MLX 0.22.0 documentation + mlx.core.export_function — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.export_to_dot.html b/docs/build/html/python/_autosummary/mlx.core.export_to_dot.html index cf7b67d35..b5b7a2a9c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.export_to_dot.html +++ b/docs/build/html/python/_autosummary/mlx.core.export_to_dot.html @@ -8,7 +8,7 @@ - mlx.core.export_to_dot — MLX 0.22.0 documentation + mlx.core.export_to_dot — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.exporter.html b/docs/build/html/python/_autosummary/mlx.core.exporter.html index b79fc55f4..52dc7efec 100644 --- a/docs/build/html/python/_autosummary/mlx.core.exporter.html +++ b/docs/build/html/python/_autosummary/mlx.core.exporter.html @@ -8,7 +8,7 @@ - mlx.core.exporter — MLX 0.22.0 documentation + mlx.core.exporter — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.eye.html b/docs/build/html/python/_autosummary/mlx.core.eye.html index c91c0d3fd..24c124b59 100644 --- a/docs/build/html/python/_autosummary/mlx.core.eye.html +++ b/docs/build/html/python/_autosummary/mlx.core.eye.html @@ -8,7 +8,7 @@ - mlx.core.eye — MLX 0.22.0 documentation + mlx.core.eye — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.layer_norm.html b/docs/build/html/python/_autosummary/mlx.core.fast.layer_norm.html index 4ad7cb5f2..33b393318 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.layer_norm.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.layer_norm.html @@ -8,7 +8,7 @@ - mlx.core.fast.layer_norm — MLX 0.22.0 documentation + mlx.core.fast.layer_norm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.metal_kernel.html b/docs/build/html/python/_autosummary/mlx.core.fast.metal_kernel.html index 4a184ce89..2632979e5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.metal_kernel.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.metal_kernel.html @@ -8,7 +8,7 @@ - mlx.core.fast.metal_kernel — MLX 0.22.0 documentation + mlx.core.fast.metal_kernel — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.rms_norm.html b/docs/build/html/python/_autosummary/mlx.core.fast.rms_norm.html index a4645524a..8506b4a96 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.rms_norm.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.rms_norm.html @@ -8,7 +8,7 @@ - mlx.core.fast.rms_norm — MLX 0.22.0 documentation + mlx.core.fast.rms_norm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.rope.html b/docs/build/html/python/_autosummary/mlx.core.fast.rope.html index a410262ec..2a6436517 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.rope.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.rope.html @@ -8,7 +8,7 @@ - mlx.core.fast.rope — MLX 0.22.0 documentation + mlx.core.fast.rope — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.scaled_dot_product_attention.html b/docs/build/html/python/_autosummary/mlx.core.fast.scaled_dot_product_attention.html index 66f1fa390..25bd353e4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.scaled_dot_product_attention.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.scaled_dot_product_attention.html @@ -8,7 +8,7 @@ - mlx.core.fast.scaled_dot_product_attention — MLX 0.22.0 documentation + mlx.core.fast.scaled_dot_product_attention — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.fft.html b/docs/build/html/python/_autosummary/mlx.core.fft.fft.html index fdd5f56fa..e516ebe71 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.fft.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.fft.html @@ -8,7 +8,7 @@ - mlx.core.fft.fft — MLX 0.22.0 documentation + mlx.core.fft.fft — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.fft2.html b/docs/build/html/python/_autosummary/mlx.core.fft.fft2.html index ebe3a7fdf..59f690ba5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.fft2.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.fft2.html @@ -8,7 +8,7 @@ - mlx.core.fft.fft2 — MLX 0.22.0 documentation + mlx.core.fft.fft2 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.fftn.html b/docs/build/html/python/_autosummary/mlx.core.fft.fftn.html index 647cbe63f..4fc19c31f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.fftn.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.fftn.html @@ -8,7 +8,7 @@ - mlx.core.fft.fftn — MLX 0.22.0 documentation + mlx.core.fft.fftn — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.ifft.html b/docs/build/html/python/_autosummary/mlx.core.fft.ifft.html index 0011a90fd..f850f2703 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.ifft.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.ifft.html @@ -8,7 +8,7 @@ - mlx.core.fft.ifft — MLX 0.22.0 documentation + mlx.core.fft.ifft — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.ifft2.html b/docs/build/html/python/_autosummary/mlx.core.fft.ifft2.html index 8ea100733..6e20f3aa2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.ifft2.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.ifft2.html @@ -8,7 +8,7 @@ - mlx.core.fft.ifft2 — MLX 0.22.0 documentation + mlx.core.fft.ifft2 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.ifftn.html b/docs/build/html/python/_autosummary/mlx.core.fft.ifftn.html index 883490ff7..c2fb5356e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.ifftn.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.ifftn.html @@ -8,7 +8,7 @@ - mlx.core.fft.ifftn — MLX 0.22.0 documentation + mlx.core.fft.ifftn — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.irfft.html b/docs/build/html/python/_autosummary/mlx.core.fft.irfft.html index 5b3896fde..71f0013f7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.irfft.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.irfft.html @@ -8,7 +8,7 @@ - mlx.core.fft.irfft — MLX 0.22.0 documentation + mlx.core.fft.irfft — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.irfft2.html b/docs/build/html/python/_autosummary/mlx.core.fft.irfft2.html index 6dca01a0a..0003c739d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.irfft2.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.irfft2.html @@ -8,7 +8,7 @@ - mlx.core.fft.irfft2 — MLX 0.22.0 documentation + mlx.core.fft.irfft2 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.irfftn.html b/docs/build/html/python/_autosummary/mlx.core.fft.irfftn.html index 6acf9808a..ba7f493de 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.irfftn.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.irfftn.html @@ -8,7 +8,7 @@ - mlx.core.fft.irfftn — MLX 0.22.0 documentation + mlx.core.fft.irfftn — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.rfft.html b/docs/build/html/python/_autosummary/mlx.core.fft.rfft.html index 7d3ee261b..215094132 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.rfft.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.rfft.html @@ -8,7 +8,7 @@ - mlx.core.fft.rfft — MLX 0.22.0 documentation + mlx.core.fft.rfft — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.rfft2.html b/docs/build/html/python/_autosummary/mlx.core.fft.rfft2.html index 4646759e6..e51969a48 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.rfft2.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.rfft2.html @@ -8,7 +8,7 @@ - mlx.core.fft.rfft2 — MLX 0.22.0 documentation + mlx.core.fft.rfft2 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.rfftn.html b/docs/build/html/python/_autosummary/mlx.core.fft.rfftn.html index e24fb9711..706f66c57 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.rfftn.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.rfftn.html @@ -8,7 +8,7 @@ - mlx.core.fft.rfftn — MLX 0.22.0 documentation + mlx.core.fft.rfftn — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.finfo.html b/docs/build/html/python/_autosummary/mlx.core.finfo.html index ed102edff..bec74e8b9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.finfo.html +++ b/docs/build/html/python/_autosummary/mlx.core.finfo.html @@ -8,7 +8,7 @@ - mlx.core.finfo — MLX 0.22.0 documentation + mlx.core.finfo — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.flatten.html b/docs/build/html/python/_autosummary/mlx.core.flatten.html index fedebaab9..e80beb2f4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.flatten.html +++ b/docs/build/html/python/_autosummary/mlx.core.flatten.html @@ -8,7 +8,7 @@ - mlx.core.flatten — MLX 0.22.0 documentation + mlx.core.flatten — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.floor.html b/docs/build/html/python/_autosummary/mlx.core.floor.html index 05303a525..ee11eedf2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.floor.html +++ b/docs/build/html/python/_autosummary/mlx.core.floor.html @@ -8,7 +8,7 @@ - mlx.core.floor — MLX 0.22.0 documentation + mlx.core.floor — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.floor_divide.html b/docs/build/html/python/_autosummary/mlx.core.floor_divide.html index f850bb7b2..8d29d92f5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.floor_divide.html +++ b/docs/build/html/python/_autosummary/mlx.core.floor_divide.html @@ -8,7 +8,7 @@ - mlx.core.floor_divide — MLX 0.22.0 documentation + mlx.core.floor_divide — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.full.html b/docs/build/html/python/_autosummary/mlx.core.full.html index cd839e700..78adce337 100644 --- a/docs/build/html/python/_autosummary/mlx.core.full.html +++ b/docs/build/html/python/_autosummary/mlx.core.full.html @@ -8,7 +8,7 @@ - mlx.core.full — MLX 0.22.0 documentation + mlx.core.full — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.gather_mm.html b/docs/build/html/python/_autosummary/mlx.core.gather_mm.html index 2461a7660..b0a0be664 100644 --- a/docs/build/html/python/_autosummary/mlx.core.gather_mm.html +++ b/docs/build/html/python/_autosummary/mlx.core.gather_mm.html @@ -8,7 +8,7 @@ - mlx.core.gather_mm — MLX 0.22.0 documentation + mlx.core.gather_mm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.gather_qmm.html b/docs/build/html/python/_autosummary/mlx.core.gather_qmm.html index cdad80b92..93833fcad 100644 --- a/docs/build/html/python/_autosummary/mlx.core.gather_qmm.html +++ b/docs/build/html/python/_autosummary/mlx.core.gather_qmm.html @@ -8,7 +8,7 @@ - mlx.core.gather_qmm — MLX 0.22.0 documentation + mlx.core.gather_qmm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.grad.html b/docs/build/html/python/_autosummary/mlx.core.grad.html index 1cb788d34..a277b084c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.grad.html +++ b/docs/build/html/python/_autosummary/mlx.core.grad.html @@ -8,7 +8,7 @@ - mlx.core.grad — MLX 0.22.0 documentation + mlx.core.grad — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -885,7 +885,7 @@

    mlx.core.grad#

    -grad(fun: Callable, argnums: int | list[int] | None = None, argnames: str | list[str] = []) → Callable#
    +grad(fun: Callable, argnums: int | Sequence[int] | None = None, argnames: str | Sequence[str] = []) → Callable#

    Returns a function which computes the gradient of fun.

    Parameters:
    diff --git a/docs/build/html/python/_autosummary/mlx.core.greater.html b/docs/build/html/python/_autosummary/mlx.core.greater.html index ac9f09154..9feb0c357 100644 --- a/docs/build/html/python/_autosummary/mlx.core.greater.html +++ b/docs/build/html/python/_autosummary/mlx.core.greater.html @@ -8,7 +8,7 @@ - mlx.core.greater — MLX 0.22.0 documentation + mlx.core.greater — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.greater_equal.html b/docs/build/html/python/_autosummary/mlx.core.greater_equal.html index ca13543c8..b91d442e2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.greater_equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.greater_equal.html @@ -8,7 +8,7 @@ - mlx.core.greater_equal — MLX 0.22.0 documentation + mlx.core.greater_equal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html b/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html index 8b090946b..8d3f65638 100644 --- a/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html +++ b/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html @@ -8,7 +8,7 @@ - mlx.core.hadamard_transform — MLX 0.22.0 documentation + mlx.core.hadamard_transform — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.identity.html b/docs/build/html/python/_autosummary/mlx.core.identity.html index 29de4623f..203b6b294 100644 --- a/docs/build/html/python/_autosummary/mlx.core.identity.html +++ b/docs/build/html/python/_autosummary/mlx.core.identity.html @@ -8,7 +8,7 @@ - mlx.core.identity — MLX 0.22.0 documentation + mlx.core.identity — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.imag.html b/docs/build/html/python/_autosummary/mlx.core.imag.html index ef721de9c..df94455be 100644 --- a/docs/build/html/python/_autosummary/mlx.core.imag.html +++ b/docs/build/html/python/_autosummary/mlx.core.imag.html @@ -8,7 +8,7 @@ - mlx.core.imag — MLX 0.22.0 documentation + mlx.core.imag — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.import_function.html b/docs/build/html/python/_autosummary/mlx.core.import_function.html index 3e2e957b0..a7ccd14f5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.import_function.html +++ b/docs/build/html/python/_autosummary/mlx.core.import_function.html @@ -8,7 +8,7 @@ - mlx.core.import_function — MLX 0.22.0 documentation + mlx.core.import_function — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.inner.html b/docs/build/html/python/_autosummary/mlx.core.inner.html index 35b12bae9..e8c255836 100644 --- a/docs/build/html/python/_autosummary/mlx.core.inner.html +++ b/docs/build/html/python/_autosummary/mlx.core.inner.html @@ -8,7 +8,7 @@ - mlx.core.inner — MLX 0.22.0 documentation + mlx.core.inner — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.isclose.html b/docs/build/html/python/_autosummary/mlx.core.isclose.html index 2cef3ee22..df3936de9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isclose.html +++ b/docs/build/html/python/_autosummary/mlx.core.isclose.html @@ -8,7 +8,7 @@ - mlx.core.isclose — MLX 0.22.0 documentation + mlx.core.isclose — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.isfinite.html b/docs/build/html/python/_autosummary/mlx.core.isfinite.html index 689f83a24..2fa9e6322 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isfinite.html +++ b/docs/build/html/python/_autosummary/mlx.core.isfinite.html @@ -8,7 +8,7 @@ - mlx.core.isfinite — MLX 0.22.0 documentation + mlx.core.isfinite — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.isinf.html b/docs/build/html/python/_autosummary/mlx.core.isinf.html index 60f70fc82..ff0c22353 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isinf.html +++ b/docs/build/html/python/_autosummary/mlx.core.isinf.html @@ -8,7 +8,7 @@ - mlx.core.isinf — MLX 0.22.0 documentation + mlx.core.isinf — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.isnan.html b/docs/build/html/python/_autosummary/mlx.core.isnan.html index b80bb4bd9..701e54e84 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isnan.html +++ b/docs/build/html/python/_autosummary/mlx.core.isnan.html @@ -8,7 +8,7 @@ - mlx.core.isnan — MLX 0.22.0 documentation + mlx.core.isnan — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.isneginf.html b/docs/build/html/python/_autosummary/mlx.core.isneginf.html index 5ac67f276..e6383f136 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isneginf.html +++ b/docs/build/html/python/_autosummary/mlx.core.isneginf.html @@ -8,7 +8,7 @@ - mlx.core.isneginf — MLX 0.22.0 documentation + mlx.core.isneginf — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.isposinf.html b/docs/build/html/python/_autosummary/mlx.core.isposinf.html index e4e346b7f..f89e1561c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isposinf.html +++ b/docs/build/html/python/_autosummary/mlx.core.isposinf.html @@ -8,7 +8,7 @@ - mlx.core.isposinf — MLX 0.22.0 documentation + mlx.core.isposinf — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.issubdtype.html b/docs/build/html/python/_autosummary/mlx.core.issubdtype.html index db921fd1a..733128401 100644 --- a/docs/build/html/python/_autosummary/mlx.core.issubdtype.html +++ b/docs/build/html/python/_autosummary/mlx.core.issubdtype.html @@ -8,7 +8,7 @@ - mlx.core.issubdtype — MLX 0.22.0 documentation + mlx.core.issubdtype — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.jvp.html b/docs/build/html/python/_autosummary/mlx.core.jvp.html index baec3ecbd..e5b2fe14c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.jvp.html +++ b/docs/build/html/python/_autosummary/mlx.core.jvp.html @@ -8,7 +8,7 @@ - mlx.core.jvp — MLX 0.22.0 documentation + mlx.core.jvp — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.kron.html b/docs/build/html/python/_autosummary/mlx.core.kron.html index e9ed415a4..6f248df76 100644 --- a/docs/build/html/python/_autosummary/mlx.core.kron.html +++ b/docs/build/html/python/_autosummary/mlx.core.kron.html @@ -8,7 +8,7 @@ - mlx.core.kron — MLX 0.22.0 documentation + mlx.core.kron — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.left_shift.html b/docs/build/html/python/_autosummary/mlx.core.left_shift.html index 5b477f816..3d1fb054c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.left_shift.html +++ b/docs/build/html/python/_autosummary/mlx.core.left_shift.html @@ -8,7 +8,7 @@ - mlx.core.left_shift — MLX 0.22.0 documentation + mlx.core.left_shift — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.less.html b/docs/build/html/python/_autosummary/mlx.core.less.html index 06a70517b..466e281ef 100644 --- a/docs/build/html/python/_autosummary/mlx.core.less.html +++ b/docs/build/html/python/_autosummary/mlx.core.less.html @@ -8,7 +8,7 @@ - mlx.core.less — MLX 0.22.0 documentation + mlx.core.less — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.less_equal.html b/docs/build/html/python/_autosummary/mlx.core.less_equal.html index 2916bbb1e..bf5b8bd1f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.less_equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.less_equal.html @@ -8,7 +8,7 @@ - mlx.core.less_equal — MLX 0.22.0 documentation + mlx.core.less_equal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html index 6005e405f..1e789029c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html @@ -8,7 +8,7 @@ - mlx.core.linalg.cholesky — MLX 0.22.0 documentation + mlx.core.linalg.cholesky — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -903,8 +903,8 @@ in which case the default stream of the default device is used.

    Returns:

    If upper = False, it returns a lower triangular L matrix such -that dot(L, L.T) = a. If upper = True, it returns an upper triangular -U matrix such that dot(U.T, U) = a.

    +that L @ L.T = a. If upper = True, it returns an upper triangular +U matrix such that U.T @ U = a.

    Return type:

    array

    diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html index 0aec2eb0b..453a033fe 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html @@ -8,7 +8,7 @@ - mlx.core.linalg.cholesky_inv — MLX 0.22.0 documentation + mlx.core.linalg.cholesky_inv — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.cross.html b/docs/build/html/python/_autosummary/mlx.core.linalg.cross.html index bc7b414d8..869837958 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.cross.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.cross.html @@ -8,7 +8,7 @@ - mlx.core.linalg.cross — MLX 0.22.0 documentation + mlx.core.linalg.cross — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.eigh.html b/docs/build/html/python/_autosummary/mlx.core.linalg.eigh.html index b84d4aab5..d1e8b3542 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.eigh.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.eigh.html @@ -8,7 +8,7 @@ - mlx.core.linalg.eigh — MLX 0.22.0 documentation + mlx.core.linalg.eigh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.eigvalsh.html b/docs/build/html/python/_autosummary/mlx.core.linalg.eigvalsh.html index 9604829bb..9a2a2b670 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.eigvalsh.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.eigvalsh.html @@ -8,7 +8,7 @@ - mlx.core.linalg.eigvalsh — MLX 0.22.0 documentation + mlx.core.linalg.eigvalsh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html b/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html index 4f6ec1c78..ca839e9d5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html @@ -8,7 +8,7 @@ - mlx.core.linalg.inv — MLX 0.22.0 documentation + mlx.core.linalg.inv — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html b/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html index 9903b2861..2d1342b5a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html @@ -8,7 +8,7 @@ - mlx.core.linalg.norm — MLX 0.22.0 documentation + mlx.core.linalg.norm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html b/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html index 7da1f00cc..72b0445a5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html @@ -8,7 +8,7 @@ - mlx.core.linalg.qr — MLX 0.22.0 documentation + mlx.core.linalg.qr — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -899,7 +899,7 @@ in which case the default stream of the default device is used.

    Returns:
    -

    The Q and R matrices.

    +

    Q and R matrices such that Q @ R = a.

    Return type:

    tuple(array, array)

    diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html b/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html index 6fbf794ef..b08a3d610 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html @@ -8,7 +8,7 @@ - mlx.core.linalg.svd — MLX 0.22.0 documentation + mlx.core.linalg.svd — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html b/docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html index a819f96e5..30a449d4f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html @@ -8,7 +8,7 @@ - mlx.core.linalg.tri_inv — MLX 0.22.0 documentation + mlx.core.linalg.tri_inv — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.linspace.html b/docs/build/html/python/_autosummary/mlx.core.linspace.html index bb07ea213..54ceb2249 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linspace.html +++ b/docs/build/html/python/_autosummary/mlx.core.linspace.html @@ -8,7 +8,7 @@ - mlx.core.linspace — MLX 0.22.0 documentation + mlx.core.linspace — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.load.html b/docs/build/html/python/_autosummary/mlx.core.load.html index 657a65ab9..c4b0eab13 100644 --- a/docs/build/html/python/_autosummary/mlx.core.load.html +++ b/docs/build/html/python/_autosummary/mlx.core.load.html @@ -8,7 +8,7 @@ - mlx.core.load — MLX 0.22.0 documentation + mlx.core.load — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.log.html b/docs/build/html/python/_autosummary/mlx.core.log.html index 084bdfe69..ebf47edd1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.log.html +++ b/docs/build/html/python/_autosummary/mlx.core.log.html @@ -8,7 +8,7 @@ - mlx.core.log — MLX 0.22.0 documentation + mlx.core.log — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.log10.html b/docs/build/html/python/_autosummary/mlx.core.log10.html index 5c0372b0a..02f168584 100644 --- a/docs/build/html/python/_autosummary/mlx.core.log10.html +++ b/docs/build/html/python/_autosummary/mlx.core.log10.html @@ -8,7 +8,7 @@ - mlx.core.log10 — MLX 0.22.0 documentation + mlx.core.log10 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.log1p.html b/docs/build/html/python/_autosummary/mlx.core.log1p.html index 8b91616d9..5dc489b15 100644 --- a/docs/build/html/python/_autosummary/mlx.core.log1p.html +++ b/docs/build/html/python/_autosummary/mlx.core.log1p.html @@ -8,7 +8,7 @@ - mlx.core.log1p — MLX 0.22.0 documentation + mlx.core.log1p — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.log2.html b/docs/build/html/python/_autosummary/mlx.core.log2.html index 0c331e5a3..df86ada48 100644 --- a/docs/build/html/python/_autosummary/mlx.core.log2.html +++ b/docs/build/html/python/_autosummary/mlx.core.log2.html @@ -8,7 +8,7 @@ - mlx.core.log2 — MLX 0.22.0 documentation + mlx.core.log2 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.logaddexp.html b/docs/build/html/python/_autosummary/mlx.core.logaddexp.html index 9aa2a1a35..226355f3d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logaddexp.html +++ b/docs/build/html/python/_autosummary/mlx.core.logaddexp.html @@ -8,7 +8,7 @@ - mlx.core.logaddexp — MLX 0.22.0 documentation + mlx.core.logaddexp — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.logical_and.html b/docs/build/html/python/_autosummary/mlx.core.logical_and.html index 35846014d..d0b927229 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logical_and.html +++ b/docs/build/html/python/_autosummary/mlx.core.logical_and.html @@ -8,7 +8,7 @@ - mlx.core.logical_and — MLX 0.22.0 documentation + mlx.core.logical_and — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.logical_not.html b/docs/build/html/python/_autosummary/mlx.core.logical_not.html index d0ad5286d..1aad91770 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logical_not.html +++ b/docs/build/html/python/_autosummary/mlx.core.logical_not.html @@ -8,7 +8,7 @@ - mlx.core.logical_not — MLX 0.22.0 documentation + mlx.core.logical_not — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.logical_or.html b/docs/build/html/python/_autosummary/mlx.core.logical_or.html index 7b2390520..845639780 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logical_or.html +++ b/docs/build/html/python/_autosummary/mlx.core.logical_or.html @@ -8,7 +8,7 @@ - mlx.core.logical_or — MLX 0.22.0 documentation + mlx.core.logical_or — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.logsumexp.html b/docs/build/html/python/_autosummary/mlx.core.logsumexp.html index 02d4032fe..c76617e33 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logsumexp.html +++ b/docs/build/html/python/_autosummary/mlx.core.logsumexp.html @@ -8,7 +8,7 @@ - mlx.core.logsumexp — MLX 0.22.0 documentation + mlx.core.logsumexp — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.matmul.html b/docs/build/html/python/_autosummary/mlx.core.matmul.html index 608ff03d6..d5edfdc53 100644 --- a/docs/build/html/python/_autosummary/mlx.core.matmul.html +++ b/docs/build/html/python/_autosummary/mlx.core.matmul.html @@ -8,7 +8,7 @@ - mlx.core.matmul — MLX 0.22.0 documentation + mlx.core.matmul — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.max.html b/docs/build/html/python/_autosummary/mlx.core.max.html index aba596ace..dc83f025d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.max.html +++ b/docs/build/html/python/_autosummary/mlx.core.max.html @@ -8,7 +8,7 @@ - mlx.core.max — MLX 0.22.0 documentation + mlx.core.max — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.maximum.html b/docs/build/html/python/_autosummary/mlx.core.maximum.html index ada1fdcab..13082a027 100644 --- a/docs/build/html/python/_autosummary/mlx.core.maximum.html +++ b/docs/build/html/python/_autosummary/mlx.core.maximum.html @@ -8,7 +8,7 @@ - mlx.core.maximum — MLX 0.22.0 documentation + mlx.core.maximum — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.mean.html b/docs/build/html/python/_autosummary/mlx.core.mean.html index f1ce601e9..0810d46a2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.mean.html +++ b/docs/build/html/python/_autosummary/mlx.core.mean.html @@ -8,7 +8,7 @@ - mlx.core.mean — MLX 0.22.0 documentation + mlx.core.mean — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.meshgrid.html b/docs/build/html/python/_autosummary/mlx.core.meshgrid.html index 56a1fffcc..1d92d310f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.meshgrid.html +++ b/docs/build/html/python/_autosummary/mlx.core.meshgrid.html @@ -8,7 +8,7 @@ - mlx.core.meshgrid — MLX 0.22.0 documentation + mlx.core.meshgrid — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html b/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html index d43aa5127..acffd5821 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html @@ -8,7 +8,7 @@ - mlx.core.metal.clear_cache — MLX 0.22.0 documentation + mlx.core.metal.clear_cache — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html b/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html index 14059ab49..2e6c2d252 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html @@ -8,7 +8,7 @@ - mlx.core.metal.device_info — MLX 0.22.0 documentation + mlx.core.metal.device_info — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html index 5747565fc..97a026f14 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html @@ -8,7 +8,7 @@ - mlx.core.metal.get_active_memory — MLX 0.22.0 documentation + mlx.core.metal.get_active_memory — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html index dfee9cdd3..1a9e6b1e5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html @@ -8,7 +8,7 @@ - mlx.core.metal.get_cache_memory — MLX 0.22.0 documentation + mlx.core.metal.get_cache_memory — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html index c79674e03..36b787a85 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html @@ -8,7 +8,7 @@ - mlx.core.metal.get_peak_memory — MLX 0.22.0 documentation + mlx.core.metal.get_peak_memory — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html b/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html index b0dbb43b0..7fad8008f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html @@ -8,7 +8,7 @@ - mlx.core.metal.is_available — MLX 0.22.0 documentation + mlx.core.metal.is_available — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html index 5b06b3dbe..77f5a9147 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html @@ -8,7 +8,7 @@ - mlx.core.metal.reset_peak_memory — MLX 0.22.0 documentation + mlx.core.metal.reset_peak_memory — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html b/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html index 814e91ede..37f699a99 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html @@ -8,7 +8,7 @@ - mlx.core.metal.set_cache_limit — MLX 0.22.0 documentation + mlx.core.metal.set_cache_limit — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html b/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html index de4b4dc66..37438cdc6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html @@ -8,7 +8,7 @@ - mlx.core.metal.set_memory_limit — MLX 0.22.0 documentation + mlx.core.metal.set_memory_limit — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.set_wired_limit.html b/docs/build/html/python/_autosummary/mlx.core.metal.set_wired_limit.html index c92feedaf..2d621e386 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.set_wired_limit.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.set_wired_limit.html @@ -8,7 +8,7 @@ - mlx.core.metal.set_wired_limit — MLX 0.22.0 documentation + mlx.core.metal.set_wired_limit — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html b/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html index e4421c671..9ba457641 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html @@ -8,7 +8,7 @@ - mlx.core.metal.start_capture — MLX 0.22.0 documentation + mlx.core.metal.start_capture — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html b/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html index 5623ff8fc..ef3629975 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html @@ -8,7 +8,7 @@ - mlx.core.metal.stop_capture — MLX 0.22.0 documentation + mlx.core.metal.stop_capture — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.min.html b/docs/build/html/python/_autosummary/mlx.core.min.html index 865658678..24e43c671 100644 --- a/docs/build/html/python/_autosummary/mlx.core.min.html +++ b/docs/build/html/python/_autosummary/mlx.core.min.html @@ -8,7 +8,7 @@ - mlx.core.min — MLX 0.22.0 documentation + mlx.core.min — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.minimum.html b/docs/build/html/python/_autosummary/mlx.core.minimum.html index 74ed98398..cb78a8e46 100644 --- a/docs/build/html/python/_autosummary/mlx.core.minimum.html +++ b/docs/build/html/python/_autosummary/mlx.core.minimum.html @@ -8,7 +8,7 @@ - mlx.core.minimum — MLX 0.22.0 documentation + mlx.core.minimum — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.moveaxis.html b/docs/build/html/python/_autosummary/mlx.core.moveaxis.html index c380fb7ea..b11d2e4be 100644 --- a/docs/build/html/python/_autosummary/mlx.core.moveaxis.html +++ b/docs/build/html/python/_autosummary/mlx.core.moveaxis.html @@ -8,7 +8,7 @@ - mlx.core.moveaxis — MLX 0.22.0 documentation + mlx.core.moveaxis — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.multiply.html b/docs/build/html/python/_autosummary/mlx.core.multiply.html index 020830415..6ab4f2ca9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.multiply.html +++ b/docs/build/html/python/_autosummary/mlx.core.multiply.html @@ -8,7 +8,7 @@ - mlx.core.multiply — MLX 0.22.0 documentation + mlx.core.multiply — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html b/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html index a3faa9f50..1adf13d20 100644 --- a/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html +++ b/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html @@ -8,7 +8,7 @@ - mlx.core.nan_to_num — MLX 0.22.0 documentation + mlx.core.nan_to_num — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.negative.html b/docs/build/html/python/_autosummary/mlx.core.negative.html index 5624f3a85..4cf7f2328 100644 --- a/docs/build/html/python/_autosummary/mlx.core.negative.html +++ b/docs/build/html/python/_autosummary/mlx.core.negative.html @@ -8,7 +8,7 @@ - mlx.core.negative — MLX 0.22.0 documentation + mlx.core.negative — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.new_stream.html b/docs/build/html/python/_autosummary/mlx.core.new_stream.html index ffa41d9f7..506113f8e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.new_stream.html +++ b/docs/build/html/python/_autosummary/mlx.core.new_stream.html @@ -8,7 +8,7 @@ - mlx.core.new_stream — MLX 0.22.0 documentation + mlx.core.new_stream — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.not_equal.html b/docs/build/html/python/_autosummary/mlx.core.not_equal.html index ac6ba9051..9cf2ef5c9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.not_equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.not_equal.html @@ -8,7 +8,7 @@ - mlx.core.not_equal — MLX 0.22.0 documentation + mlx.core.not_equal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.ones.html b/docs/build/html/python/_autosummary/mlx.core.ones.html index 0cd6af15c..96427cd92 100644 --- a/docs/build/html/python/_autosummary/mlx.core.ones.html +++ b/docs/build/html/python/_autosummary/mlx.core.ones.html @@ -8,7 +8,7 @@ - mlx.core.ones — MLX 0.22.0 documentation + mlx.core.ones — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.ones_like.html b/docs/build/html/python/_autosummary/mlx.core.ones_like.html index 838bd1401..da6197552 100644 --- a/docs/build/html/python/_autosummary/mlx.core.ones_like.html +++ b/docs/build/html/python/_autosummary/mlx.core.ones_like.html @@ -8,7 +8,7 @@ - mlx.core.ones_like — MLX 0.22.0 documentation + mlx.core.ones_like — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.outer.html b/docs/build/html/python/_autosummary/mlx.core.outer.html index ff1bb0bc4..3c0efd74b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.outer.html +++ b/docs/build/html/python/_autosummary/mlx.core.outer.html @@ -8,7 +8,7 @@ - mlx.core.outer — MLX 0.22.0 documentation + mlx.core.outer — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.pad.html b/docs/build/html/python/_autosummary/mlx.core.pad.html index f90387fdb..b709a8d45 100644 --- a/docs/build/html/python/_autosummary/mlx.core.pad.html +++ b/docs/build/html/python/_autosummary/mlx.core.pad.html @@ -8,7 +8,7 @@ - mlx.core.pad — MLX 0.22.0 documentation + mlx.core.pad — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.partition.html b/docs/build/html/python/_autosummary/mlx.core.partition.html index daa8a5d86..1be48fd23 100644 --- a/docs/build/html/python/_autosummary/mlx.core.partition.html +++ b/docs/build/html/python/_autosummary/mlx.core.partition.html @@ -8,7 +8,7 @@ - mlx.core.partition — MLX 0.22.0 documentation + mlx.core.partition — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.power.html b/docs/build/html/python/_autosummary/mlx.core.power.html index 5f2179bcc..11c4b95f5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.power.html +++ b/docs/build/html/python/_autosummary/mlx.core.power.html @@ -8,7 +8,7 @@ - mlx.core.power — MLX 0.22.0 documentation + mlx.core.power — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.prod.html b/docs/build/html/python/_autosummary/mlx.core.prod.html index 442102951..7071f0309 100644 --- a/docs/build/html/python/_autosummary/mlx.core.prod.html +++ b/docs/build/html/python/_autosummary/mlx.core.prod.html @@ -8,7 +8,7 @@ - mlx.core.prod — MLX 0.22.0 documentation + mlx.core.prod — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.put_along_axis.html b/docs/build/html/python/_autosummary/mlx.core.put_along_axis.html index f5d3dd6a8..5f00a9db9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.put_along_axis.html +++ b/docs/build/html/python/_autosummary/mlx.core.put_along_axis.html @@ -8,7 +8,7 @@ - mlx.core.put_along_axis — MLX 0.22.0 documentation + mlx.core.put_along_axis — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.quantize.html b/docs/build/html/python/_autosummary/mlx.core.quantize.html index c16fa5703..e73fa1c73 100644 --- a/docs/build/html/python/_autosummary/mlx.core.quantize.html +++ b/docs/build/html/python/_autosummary/mlx.core.quantize.html @@ -8,7 +8,7 @@ - mlx.core.quantize — MLX 0.22.0 documentation + mlx.core.quantize — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html b/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html index cb504d498..89841629f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html +++ b/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html @@ -8,7 +8,7 @@ - mlx.core.quantized_matmul — MLX 0.22.0 documentation + mlx.core.quantized_matmul — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.radians.html b/docs/build/html/python/_autosummary/mlx.core.radians.html index 4d22a7b12..4390f6f98 100644 --- a/docs/build/html/python/_autosummary/mlx.core.radians.html +++ b/docs/build/html/python/_autosummary/mlx.core.radians.html @@ -8,7 +8,7 @@ - mlx.core.radians — MLX 0.22.0 documentation + mlx.core.radians — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html b/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html index 49be877ed..6a54fc027 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html @@ -8,7 +8,7 @@ - mlx.core.random.bernoulli — MLX 0.22.0 documentation + mlx.core.random.bernoulli — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.categorical.html b/docs/build/html/python/_autosummary/mlx.core.random.categorical.html index 50c2bb112..0b22b7615 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.categorical.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.categorical.html @@ -8,7 +8,7 @@ - mlx.core.random.categorical — MLX 0.22.0 documentation + mlx.core.random.categorical — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html b/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html index d5a89eff6..572a34c5a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html @@ -8,7 +8,7 @@ - mlx.core.random.gumbel — MLX 0.22.0 documentation + mlx.core.random.gumbel — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.key.html b/docs/build/html/python/_autosummary/mlx.core.random.key.html index 6a3761a6f..c9d1b9784 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.key.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.key.html @@ -8,7 +8,7 @@ - mlx.core.random.key — MLX 0.22.0 documentation + mlx.core.random.key — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.laplace.html b/docs/build/html/python/_autosummary/mlx.core.random.laplace.html index 97731e488..39efe3045 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.laplace.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.laplace.html @@ -8,7 +8,7 @@ - mlx.core.random.laplace — MLX 0.22.0 documentation + mlx.core.random.laplace — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.multivariate_normal.html b/docs/build/html/python/_autosummary/mlx.core.random.multivariate_normal.html index b06d6b69f..3f7611a44 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.multivariate_normal.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.multivariate_normal.html @@ -8,7 +8,7 @@ - mlx.core.random.multivariate_normal — MLX 0.22.0 documentation + mlx.core.random.multivariate_normal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.normal.html b/docs/build/html/python/_autosummary/mlx.core.random.normal.html index 980535594..e86a6587a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.normal.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.normal.html @@ -8,7 +8,7 @@ - mlx.core.random.normal — MLX 0.22.0 documentation + mlx.core.random.normal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.permutation.html b/docs/build/html/python/_autosummary/mlx.core.random.permutation.html index c2facc023..d071515f5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.permutation.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.permutation.html @@ -8,7 +8,7 @@ - mlx.core.random.permutation — MLX 0.22.0 documentation + mlx.core.random.permutation — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.randint.html b/docs/build/html/python/_autosummary/mlx.core.random.randint.html index 82990bffe..d2b5ee9dd 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.randint.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.randint.html @@ -8,7 +8,7 @@ - mlx.core.random.randint — MLX 0.22.0 documentation + mlx.core.random.randint — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.seed.html b/docs/build/html/python/_autosummary/mlx.core.random.seed.html index 4ca472164..bdc96871d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.seed.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.seed.html @@ -8,7 +8,7 @@ - mlx.core.random.seed — MLX 0.22.0 documentation + mlx.core.random.seed — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.split.html b/docs/build/html/python/_autosummary/mlx.core.random.split.html index 125a23a01..d4882197d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.split.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.split.html @@ -8,7 +8,7 @@ - mlx.core.random.split — MLX 0.22.0 documentation + mlx.core.random.split — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html b/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html index 9863a25b8..20c27a630 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html @@ -8,7 +8,7 @@ - mlx.core.random.truncated_normal — MLX 0.22.0 documentation + mlx.core.random.truncated_normal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.random.uniform.html b/docs/build/html/python/_autosummary/mlx.core.random.uniform.html index 2192bd822..6302d9b25 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.uniform.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.uniform.html @@ -8,7 +8,7 @@ - mlx.core.random.uniform — MLX 0.22.0 documentation + mlx.core.random.uniform — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.real.html b/docs/build/html/python/_autosummary/mlx.core.real.html index d3d3005bb..f61891964 100644 --- a/docs/build/html/python/_autosummary/mlx.core.real.html +++ b/docs/build/html/python/_autosummary/mlx.core.real.html @@ -8,7 +8,7 @@ - mlx.core.real — MLX 0.22.0 documentation + mlx.core.real — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.reciprocal.html b/docs/build/html/python/_autosummary/mlx.core.reciprocal.html index 49f11f7ec..7118f8299 100644 --- a/docs/build/html/python/_autosummary/mlx.core.reciprocal.html +++ b/docs/build/html/python/_autosummary/mlx.core.reciprocal.html @@ -8,7 +8,7 @@ - mlx.core.reciprocal — MLX 0.22.0 documentation + mlx.core.reciprocal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.remainder.html b/docs/build/html/python/_autosummary/mlx.core.remainder.html index 7cb46b686..28cbe60ab 100644 --- a/docs/build/html/python/_autosummary/mlx.core.remainder.html +++ b/docs/build/html/python/_autosummary/mlx.core.remainder.html @@ -8,7 +8,7 @@ - mlx.core.remainder — MLX 0.22.0 documentation + mlx.core.remainder — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.repeat.html b/docs/build/html/python/_autosummary/mlx.core.repeat.html index 2decc871f..618507254 100644 --- a/docs/build/html/python/_autosummary/mlx.core.repeat.html +++ b/docs/build/html/python/_autosummary/mlx.core.repeat.html @@ -8,7 +8,7 @@ - mlx.core.repeat — MLX 0.22.0 documentation + mlx.core.repeat — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.reshape.html b/docs/build/html/python/_autosummary/mlx.core.reshape.html index e9cee4f77..4aff497c3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.reshape.html +++ b/docs/build/html/python/_autosummary/mlx.core.reshape.html @@ -8,7 +8,7 @@ - mlx.core.reshape — MLX 0.22.0 documentation + mlx.core.reshape — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.right_shift.html b/docs/build/html/python/_autosummary/mlx.core.right_shift.html index 855af3c9e..fa12e4e96 100644 --- a/docs/build/html/python/_autosummary/mlx.core.right_shift.html +++ b/docs/build/html/python/_autosummary/mlx.core.right_shift.html @@ -8,7 +8,7 @@ - mlx.core.right_shift — MLX 0.22.0 documentation + mlx.core.right_shift — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.roll.html b/docs/build/html/python/_autosummary/mlx.core.roll.html index 953b55fc8..73de9d216 100644 --- a/docs/build/html/python/_autosummary/mlx.core.roll.html +++ b/docs/build/html/python/_autosummary/mlx.core.roll.html @@ -8,7 +8,7 @@ - mlx.core.roll — MLX 0.22.0 documentation + mlx.core.roll — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.round.html b/docs/build/html/python/_autosummary/mlx.core.round.html index ae79cea3e..62622a9b8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.round.html +++ b/docs/build/html/python/_autosummary/mlx.core.round.html @@ -8,7 +8,7 @@ - mlx.core.round — MLX 0.22.0 documentation + mlx.core.round — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.rsqrt.html b/docs/build/html/python/_autosummary/mlx.core.rsqrt.html index 33b28f84a..9fa325655 100644 --- a/docs/build/html/python/_autosummary/mlx.core.rsqrt.html +++ b/docs/build/html/python/_autosummary/mlx.core.rsqrt.html @@ -8,7 +8,7 @@ - mlx.core.rsqrt — MLX 0.22.0 documentation + mlx.core.rsqrt — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.save.html b/docs/build/html/python/_autosummary/mlx.core.save.html index 9c958e2d3..91a9aa80f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.save.html +++ b/docs/build/html/python/_autosummary/mlx.core.save.html @@ -8,7 +8,7 @@ - mlx.core.save — MLX 0.22.0 documentation + mlx.core.save — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.save_gguf.html b/docs/build/html/python/_autosummary/mlx.core.save_gguf.html index e82c68124..371676f4d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.save_gguf.html +++ b/docs/build/html/python/_autosummary/mlx.core.save_gguf.html @@ -8,7 +8,7 @@ - mlx.core.save_gguf — MLX 0.22.0 documentation + mlx.core.save_gguf — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.save_safetensors.html b/docs/build/html/python/_autosummary/mlx.core.save_safetensors.html index c2613607d..6e01f0e22 100644 --- a/docs/build/html/python/_autosummary/mlx.core.save_safetensors.html +++ b/docs/build/html/python/_autosummary/mlx.core.save_safetensors.html @@ -8,7 +8,7 @@ - mlx.core.save_safetensors — MLX 0.22.0 documentation + mlx.core.save_safetensors — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.savez.html b/docs/build/html/python/_autosummary/mlx.core.savez.html index e9797c49e..05a2b69f7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.savez.html +++ b/docs/build/html/python/_autosummary/mlx.core.savez.html @@ -8,7 +8,7 @@ - mlx.core.savez — MLX 0.22.0 documentation + mlx.core.savez — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html b/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html index e28a54af6..614b6491a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html +++ b/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html @@ -8,7 +8,7 @@ - mlx.core.savez_compressed — MLX 0.22.0 documentation + mlx.core.savez_compressed — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.set_default_device.html b/docs/build/html/python/_autosummary/mlx.core.set_default_device.html index bb9efd916..b6dd74f0b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.set_default_device.html +++ b/docs/build/html/python/_autosummary/mlx.core.set_default_device.html @@ -8,7 +8,7 @@ - mlx.core.set_default_device — MLX 0.22.0 documentation + mlx.core.set_default_device — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html b/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html index 34de04fe5..5c2339047 100644 --- a/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html +++ b/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html @@ -8,7 +8,7 @@ - mlx.core.set_default_stream — MLX 0.22.0 documentation + mlx.core.set_default_stream — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.sigmoid.html b/docs/build/html/python/_autosummary/mlx.core.sigmoid.html index 6314d6957..08d4a9aa7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sigmoid.html +++ b/docs/build/html/python/_autosummary/mlx.core.sigmoid.html @@ -8,7 +8,7 @@ - mlx.core.sigmoid — MLX 0.22.0 documentation + mlx.core.sigmoid — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.sign.html b/docs/build/html/python/_autosummary/mlx.core.sign.html index 6611678b5..40d16822b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sign.html +++ b/docs/build/html/python/_autosummary/mlx.core.sign.html @@ -8,7 +8,7 @@ - mlx.core.sign — MLX 0.22.0 documentation + mlx.core.sign — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.sin.html b/docs/build/html/python/_autosummary/mlx.core.sin.html index 7fd772ec0..c93046ae4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sin.html +++ b/docs/build/html/python/_autosummary/mlx.core.sin.html @@ -8,7 +8,7 @@ - mlx.core.sin — MLX 0.22.0 documentation + mlx.core.sin — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.sinh.html b/docs/build/html/python/_autosummary/mlx.core.sinh.html index 7c17479cb..83b1e4239 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sinh.html +++ b/docs/build/html/python/_autosummary/mlx.core.sinh.html @@ -8,7 +8,7 @@ - mlx.core.sinh — MLX 0.22.0 documentation + mlx.core.sinh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.slice.html b/docs/build/html/python/_autosummary/mlx.core.slice.html index 7629ba1e9..f7028a568 100644 --- a/docs/build/html/python/_autosummary/mlx.core.slice.html +++ b/docs/build/html/python/_autosummary/mlx.core.slice.html @@ -8,7 +8,7 @@ - mlx.core.slice — MLX 0.22.0 documentation + mlx.core.slice — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.slice_update.html b/docs/build/html/python/_autosummary/mlx.core.slice_update.html index 0be9e5811..f719716b1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.slice_update.html +++ b/docs/build/html/python/_autosummary/mlx.core.slice_update.html @@ -8,7 +8,7 @@ - mlx.core.slice_update — MLX 0.22.0 documentation + mlx.core.slice_update — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.softmax.html b/docs/build/html/python/_autosummary/mlx.core.softmax.html index 73ff6b53c..197812e2a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.softmax.html +++ b/docs/build/html/python/_autosummary/mlx.core.softmax.html @@ -8,7 +8,7 @@ - mlx.core.softmax — MLX 0.22.0 documentation + mlx.core.softmax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.sort.html b/docs/build/html/python/_autosummary/mlx.core.sort.html index 47cd043cf..5c3a2e950 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sort.html +++ b/docs/build/html/python/_autosummary/mlx.core.sort.html @@ -8,7 +8,7 @@ - mlx.core.sort — MLX 0.22.0 documentation + mlx.core.sort — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.split.html b/docs/build/html/python/_autosummary/mlx.core.split.html index c1f2afc0a..a837ce009 100644 --- a/docs/build/html/python/_autosummary/mlx.core.split.html +++ b/docs/build/html/python/_autosummary/mlx.core.split.html @@ -8,7 +8,7 @@ - mlx.core.split — MLX 0.22.0 documentation + mlx.core.split — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.sqrt.html b/docs/build/html/python/_autosummary/mlx.core.sqrt.html index d71e93f2d..aac034427 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sqrt.html +++ b/docs/build/html/python/_autosummary/mlx.core.sqrt.html @@ -8,7 +8,7 @@ - mlx.core.sqrt — MLX 0.22.0 documentation + mlx.core.sqrt — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.square.html b/docs/build/html/python/_autosummary/mlx.core.square.html index ba448bbd6..42956ce46 100644 --- a/docs/build/html/python/_autosummary/mlx.core.square.html +++ b/docs/build/html/python/_autosummary/mlx.core.square.html @@ -8,7 +8,7 @@ - mlx.core.square — MLX 0.22.0 documentation + mlx.core.square — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.squeeze.html b/docs/build/html/python/_autosummary/mlx.core.squeeze.html index 92e9e4bcf..eb6219fc4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.squeeze.html +++ b/docs/build/html/python/_autosummary/mlx.core.squeeze.html @@ -8,7 +8,7 @@ - mlx.core.squeeze — MLX 0.22.0 documentation + mlx.core.squeeze — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.stack.html b/docs/build/html/python/_autosummary/mlx.core.stack.html index 26012db7a..d0f858a9c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.stack.html +++ b/docs/build/html/python/_autosummary/mlx.core.stack.html @@ -8,7 +8,7 @@ - mlx.core.stack — MLX 0.22.0 documentation + mlx.core.stack — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.std.html b/docs/build/html/python/_autosummary/mlx.core.std.html index cba483d28..13279fd01 100644 --- a/docs/build/html/python/_autosummary/mlx.core.std.html +++ b/docs/build/html/python/_autosummary/mlx.core.std.html @@ -8,7 +8,7 @@ - mlx.core.std — MLX 0.22.0 documentation + mlx.core.std — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html b/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html index 22227747d..cd0f97419 100644 --- a/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html +++ b/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html @@ -8,7 +8,7 @@ - mlx.core.stop_gradient — MLX 0.22.0 documentation + mlx.core.stop_gradient — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.stream.html b/docs/build/html/python/_autosummary/mlx.core.stream.html index 356351702..9deaa4c57 100644 --- a/docs/build/html/python/_autosummary/mlx.core.stream.html +++ b/docs/build/html/python/_autosummary/mlx.core.stream.html @@ -8,7 +8,7 @@ - mlx.core.stream — MLX 0.22.0 documentation + mlx.core.stream — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.subtract.html b/docs/build/html/python/_autosummary/mlx.core.subtract.html index 5254db88f..f17493d77 100644 --- a/docs/build/html/python/_autosummary/mlx.core.subtract.html +++ b/docs/build/html/python/_autosummary/mlx.core.subtract.html @@ -8,7 +8,7 @@ - mlx.core.subtract — MLX 0.22.0 documentation + mlx.core.subtract — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.sum.html b/docs/build/html/python/_autosummary/mlx.core.sum.html index a8654d218..0330ca53c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sum.html +++ b/docs/build/html/python/_autosummary/mlx.core.sum.html @@ -8,7 +8,7 @@ - mlx.core.sum — MLX 0.22.0 documentation + mlx.core.sum — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.swapaxes.html b/docs/build/html/python/_autosummary/mlx.core.swapaxes.html index 39e719a21..6ad952035 100644 --- a/docs/build/html/python/_autosummary/mlx.core.swapaxes.html +++ b/docs/build/html/python/_autosummary/mlx.core.swapaxes.html @@ -8,7 +8,7 @@ - mlx.core.swapaxes — MLX 0.22.0 documentation + mlx.core.swapaxes — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.synchronize.html b/docs/build/html/python/_autosummary/mlx.core.synchronize.html index 6dd880726..f410e8c2d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.synchronize.html +++ b/docs/build/html/python/_autosummary/mlx.core.synchronize.html @@ -8,7 +8,7 @@ - mlx.core.synchronize — MLX 0.22.0 documentation + mlx.core.synchronize — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.take.html b/docs/build/html/python/_autosummary/mlx.core.take.html index 8a30c3c60..63e4f54c9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.take.html +++ b/docs/build/html/python/_autosummary/mlx.core.take.html @@ -8,7 +8,7 @@ - mlx.core.take — MLX 0.22.0 documentation + mlx.core.take — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html b/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html index dfd245ae4..9be3806f5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html +++ b/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html @@ -8,7 +8,7 @@ - mlx.core.take_along_axis — MLX 0.22.0 documentation + mlx.core.take_along_axis — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.tan.html b/docs/build/html/python/_autosummary/mlx.core.tan.html index 41d43f214..920097e9b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tan.html +++ b/docs/build/html/python/_autosummary/mlx.core.tan.html @@ -8,7 +8,7 @@ - mlx.core.tan — MLX 0.22.0 documentation + mlx.core.tan — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.tanh.html b/docs/build/html/python/_autosummary/mlx.core.tanh.html index 4bfdf6517..89155a507 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tanh.html +++ b/docs/build/html/python/_autosummary/mlx.core.tanh.html @@ -8,7 +8,7 @@ - mlx.core.tanh — MLX 0.22.0 documentation + mlx.core.tanh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.tensordot.html b/docs/build/html/python/_autosummary/mlx.core.tensordot.html index d5967986c..d9c3cdd27 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tensordot.html +++ b/docs/build/html/python/_autosummary/mlx.core.tensordot.html @@ -8,7 +8,7 @@ - mlx.core.tensordot — MLX 0.22.0 documentation + mlx.core.tensordot — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.tile.html b/docs/build/html/python/_autosummary/mlx.core.tile.html index 7600db888..97f410867 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tile.html +++ b/docs/build/html/python/_autosummary/mlx.core.tile.html @@ -8,7 +8,7 @@ - mlx.core.tile — MLX 0.22.0 documentation + mlx.core.tile — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.topk.html b/docs/build/html/python/_autosummary/mlx.core.topk.html index 568330bfc..ec0502891 100644 --- a/docs/build/html/python/_autosummary/mlx.core.topk.html +++ b/docs/build/html/python/_autosummary/mlx.core.topk.html @@ -8,7 +8,7 @@ - mlx.core.topk — MLX 0.22.0 documentation + mlx.core.topk — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.trace.html b/docs/build/html/python/_autosummary/mlx.core.trace.html index a960bcaf0..641bd494c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.trace.html +++ b/docs/build/html/python/_autosummary/mlx.core.trace.html @@ -8,7 +8,7 @@ - mlx.core.trace — MLX 0.22.0 documentation + mlx.core.trace — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.transpose.html b/docs/build/html/python/_autosummary/mlx.core.transpose.html index 391dd2fd3..c594ae4c7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.transpose.html +++ b/docs/build/html/python/_autosummary/mlx.core.transpose.html @@ -8,7 +8,7 @@ - mlx.core.transpose — MLX 0.22.0 documentation + mlx.core.transpose — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.tri.html b/docs/build/html/python/_autosummary/mlx.core.tri.html index db2f18aae..83b3b5aeb 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tri.html +++ b/docs/build/html/python/_autosummary/mlx.core.tri.html @@ -8,7 +8,7 @@ - mlx.core.tri — MLX 0.22.0 documentation + mlx.core.tri — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.tril.html b/docs/build/html/python/_autosummary/mlx.core.tril.html index bb261989f..d64ea2651 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tril.html +++ b/docs/build/html/python/_autosummary/mlx.core.tril.html @@ -8,7 +8,7 @@ - mlx.core.tril — MLX 0.22.0 documentation + mlx.core.tril — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.triu.html b/docs/build/html/python/_autosummary/mlx.core.triu.html index b703141cb..9da7cf57a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.triu.html +++ b/docs/build/html/python/_autosummary/mlx.core.triu.html @@ -8,7 +8,7 @@ - mlx.core.triu — MLX 0.22.0 documentation + mlx.core.triu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.unflatten.html b/docs/build/html/python/_autosummary/mlx.core.unflatten.html index 32de69c70..0835bd425 100644 --- a/docs/build/html/python/_autosummary/mlx.core.unflatten.html +++ b/docs/build/html/python/_autosummary/mlx.core.unflatten.html @@ -8,7 +8,7 @@ - mlx.core.unflatten — MLX 0.22.0 documentation + mlx.core.unflatten — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html b/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html index 4095dc20d..1b23e5f80 100644 --- a/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html +++ b/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html @@ -8,7 +8,7 @@ - mlx.core.value_and_grad — MLX 0.22.0 documentation + mlx.core.value_and_grad — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -885,7 +885,7 @@

    mlx.core.value_and_grad#

    -value_and_grad(fun: Callable, argnums: int | list[int] | None = None, argnames: str | list[str] = []) → Callable#
    +value_and_grad(fun: Callable, argnums: int | Sequence[int] | None = None, argnames: str | Sequence[str] = []) → Callable#

    Returns a function which computes the value and gradient of fun.

    The function passed to value_and_grad() should return either a scalar loss or a tuple in which the first element is a scalar diff --git a/docs/build/html/python/_autosummary/mlx.core.var.html b/docs/build/html/python/_autosummary/mlx.core.var.html index 8282c3d5a..6e2a8c830 100644 --- a/docs/build/html/python/_autosummary/mlx.core.var.html +++ b/docs/build/html/python/_autosummary/mlx.core.var.html @@ -8,7 +8,7 @@ - mlx.core.var — MLX 0.22.0 documentation + mlx.core.var — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.view.html b/docs/build/html/python/_autosummary/mlx.core.view.html index 4bb2386a9..ea6a741ae 100644 --- a/docs/build/html/python/_autosummary/mlx.core.view.html +++ b/docs/build/html/python/_autosummary/mlx.core.view.html @@ -8,7 +8,7 @@ - mlx.core.view — MLX 0.22.0 documentation + mlx.core.view — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.vjp.html b/docs/build/html/python/_autosummary/mlx.core.vjp.html index 87e6c0965..ed69a591c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.vjp.html +++ b/docs/build/html/python/_autosummary/mlx.core.vjp.html @@ -8,7 +8,7 @@ - mlx.core.vjp — MLX 0.22.0 documentation + mlx.core.vjp — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.vmap.html b/docs/build/html/python/_autosummary/mlx.core.vmap.html index 539d40aca..7f1f7d570 100644 --- a/docs/build/html/python/_autosummary/mlx.core.vmap.html +++ b/docs/build/html/python/_autosummary/mlx.core.vmap.html @@ -8,7 +8,7 @@ - mlx.core.vmap — MLX 0.22.0 documentation + mlx.core.vmap — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.where.html b/docs/build/html/python/_autosummary/mlx.core.where.html index 9bf16accf..fb1348304 100644 --- a/docs/build/html/python/_autosummary/mlx.core.where.html +++ b/docs/build/html/python/_autosummary/mlx.core.where.html @@ -8,7 +8,7 @@ - mlx.core.where — MLX 0.22.0 documentation + mlx.core.where — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.zeros.html b/docs/build/html/python/_autosummary/mlx.core.zeros.html index 5f0b65422..87a74a1ff 100644 --- a/docs/build/html/python/_autosummary/mlx.core.zeros.html +++ b/docs/build/html/python/_autosummary/mlx.core.zeros.html @@ -8,7 +8,7 @@ - mlx.core.zeros — MLX 0.22.0 documentation + mlx.core.zeros — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.core.zeros_like.html b/docs/build/html/python/_autosummary/mlx.core.zeros_like.html index b1e039a9f..b5db20c70 100644 --- a/docs/build/html/python/_autosummary/mlx.core.zeros_like.html +++ b/docs/build/html/python/_autosummary/mlx.core.zeros_like.html @@ -8,7 +8,7 @@ - mlx.core.zeros_like — MLX 0.22.0 documentation + mlx.core.zeros_like — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.nn.quantize.html b/docs/build/html/python/_autosummary/mlx.nn.quantize.html index 000cd69a4..512c769f8 100644 --- a/docs/build/html/python/_autosummary/mlx.nn.quantize.html +++ b/docs/build/html/python/_autosummary/mlx.nn.quantize.html @@ -8,7 +8,7 @@ - mlx.nn.quantize — MLX 0.22.0 documentation + mlx.nn.quantize — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html b/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html index ca4f91797..cdddd859c 100644 --- a/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html +++ b/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html @@ -8,7 +8,7 @@ - mlx.nn.value_and_grad — MLX 0.22.0 documentation + mlx.nn.value_and_grad — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html b/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html index f272db0bd..9b28496a6 100644 --- a/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html +++ b/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html @@ -8,7 +8,7 @@ - mlx.optimizers.clip_grad_norm — MLX 0.22.0 documentation + mlx.optimizers.clip_grad_norm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html b/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html index ee0597210..ae42bf782 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html @@ -8,7 +8,7 @@ - mlx.utils.tree_flatten — MLX 0.22.0 documentation + mlx.utils.tree_flatten — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_map.html b/docs/build/html/python/_autosummary/mlx.utils.tree_map.html index 80bf524d7..4f30dac3b 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_map.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_map.html @@ -8,7 +8,7 @@ - mlx.utils.tree_map — MLX 0.22.0 documentation + mlx.utils.tree_map — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_map_with_path.html b/docs/build/html/python/_autosummary/mlx.utils.tree_map_with_path.html index c34094837..50fe295c4 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_map_with_path.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_map_with_path.html @@ -8,7 +8,7 @@ - mlx.utils.tree_map_with_path — MLX 0.22.0 documentation + mlx.utils.tree_map_with_path — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_reduce.html b/docs/build/html/python/_autosummary/mlx.utils.tree_reduce.html index 036a7b9db..f0a29d556 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_reduce.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_reduce.html @@ -8,7 +8,7 @@ - mlx.utils.tree_reduce — MLX 0.22.0 documentation + mlx.utils.tree_reduce — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html b/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html index 2d713eb6f..639fbe337 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html @@ -8,7 +8,7 @@ - mlx.utils.tree_unflatten — MLX 0.22.0 documentation + mlx.utils.tree_unflatten — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/_autosummary/stream_class.html b/docs/build/html/python/_autosummary/stream_class.html index ffc8d7265..bcd493223 100644 --- a/docs/build/html/python/_autosummary/stream_class.html +++ b/docs/build/html/python/_autosummary/stream_class.html @@ -8,7 +8,7 @@ - mlx.core.Stream — MLX 0.22.0 documentation + mlx.core.Stream — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/array.html b/docs/build/html/python/array.html index c44521a53..c8cb0c27e 100644 --- a/docs/build/html/python/array.html +++ b/docs/build/html/python/array.html @@ -8,7 +8,7 @@ - Array — MLX 0.22.0 documentation + Array — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/data_types.html b/docs/build/html/python/data_types.html index 5f409035c..a7e83d6ad 100644 --- a/docs/build/html/python/data_types.html +++ b/docs/build/html/python/data_types.html @@ -8,7 +8,7 @@ - Data Types — MLX 0.22.0 documentation + Data Types — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/devices_and_streams.html b/docs/build/html/python/devices_and_streams.html index 679b48e87..3e56cd0d0 100644 --- a/docs/build/html/python/devices_and_streams.html +++ b/docs/build/html/python/devices_and_streams.html @@ -8,7 +8,7 @@ - Devices and Streams — MLX 0.22.0 documentation + Devices and Streams — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/distributed.html b/docs/build/html/python/distributed.html index 4c0403b62..64b28fafe 100644 --- a/docs/build/html/python/distributed.html +++ b/docs/build/html/python/distributed.html @@ -8,7 +8,7 @@ - Distributed Communication — MLX 0.22.0 documentation + Distributed Communication — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -884,7 +884,7 @@ made available.

    is_available()

    Check if a communication backend is available.

    -

    init([strict])

    +

    init([strict, backend])

    Initialize the communication backend and create the global communication group.

    all_sum(x, *[, group, stream])

    diff --git a/docs/build/html/python/export.html b/docs/build/html/python/export.html index b0ad7633d..8e460f759 100644 --- a/docs/build/html/python/export.html +++ b/docs/build/html/python/export.html @@ -8,7 +8,7 @@ - Export Functions — MLX 0.22.0 documentation + Export Functions — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/fast.html b/docs/build/html/python/fast.html index e1d85b2fb..32f6a1202 100644 --- a/docs/build/html/python/fast.html +++ b/docs/build/html/python/fast.html @@ -8,7 +8,7 @@ - Fast — MLX 0.22.0 documentation + Fast — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/fft.html b/docs/build/html/python/fft.html index 018ddfcf1..edbe1898f 100644 --- a/docs/build/html/python/fft.html +++ b/docs/build/html/python/fft.html @@ -8,7 +8,7 @@ - FFT — MLX 0.22.0 documentation + FFT — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/linalg.html b/docs/build/html/python/linalg.html index 5abfdbcfb..f46786048 100644 --- a/docs/build/html/python/linalg.html +++ b/docs/build/html/python/linalg.html @@ -8,7 +8,7 @@ - Linear Algebra — MLX 0.22.0 documentation + Linear Algebra — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/metal.html b/docs/build/html/python/metal.html index 383bfa0f9..0560e0114 100644 --- a/docs/build/html/python/metal.html +++ b/docs/build/html/python/metal.html @@ -8,7 +8,7 @@ - Metal — MLX 0.22.0 documentation + Metal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn.html b/docs/build/html/python/nn.html index 96edeaec6..ccbfa1738 100644 --- a/docs/build/html/python/nn.html +++ b/docs/build/html/python/nn.html @@ -8,7 +8,7 @@ - Neural Networks — MLX 0.22.0 documentation + Neural Networks — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html index 6fea5eda8..8c605fa0c 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html @@ -8,7 +8,7 @@ - mlx.nn.ALiBi — MLX 0.22.0 documentation + mlx.nn.ALiBi — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html index bd7d5b811..c714d0190 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html @@ -8,7 +8,7 @@ - mlx.nn.AvgPool1d — MLX 0.22.0 documentation + mlx.nn.AvgPool1d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html index 4a90ab931..4a15c2c15 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html @@ -8,7 +8,7 @@ - mlx.nn.AvgPool2d — MLX 0.22.0 documentation + mlx.nn.AvgPool2d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool3d.html index e7f503d05..32ea46e9d 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool3d.html @@ -8,7 +8,7 @@ - mlx.nn.AvgPool3d — MLX 0.22.0 documentation + mlx.nn.AvgPool3d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html index 3e8cbd772..87f974241 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html @@ -8,7 +8,7 @@ - mlx.nn.BatchNorm — MLX 0.22.0 documentation + mlx.nn.BatchNorm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.CELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.CELU.html index b20d4f4f8..83f32d7ae 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.CELU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.CELU.html @@ -8,7 +8,7 @@ - mlx.nn.CELU — MLX 0.22.0 documentation + mlx.nn.CELU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html index 341d86328..e19f7c3d4 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html @@ -8,7 +8,7 @@ - mlx.nn.Conv1d — MLX 0.22.0 documentation + mlx.nn.Conv1d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html index c8f99e2f8..7219cc9b5 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html @@ -8,7 +8,7 @@ - mlx.nn.Conv2d — MLX 0.22.0 documentation + mlx.nn.Conv2d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html index 2f6156256..4692a6959 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html @@ -8,7 +8,7 @@ - mlx.nn.Conv3d — MLX 0.22.0 documentation + mlx.nn.Conv3d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -885,7 +885,7 @@

    mlx.nn.Conv3d#

    -class Conv3d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, bias: bool = True)#
    +class Conv3d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, bias: bool = True)#

    Applies a 3-dimensional convolution over the multi-channel input image.

    The channels are expected to be last i.e. the input shape should be NDHWC where:

      @@ -903,6 +903,7 @@

    • kernel_size (int or tuple) – The size of the convolution filters.

    • stride (int or tuple, optional) – The size of the stride when applying the filter. Default: 1.

      • +

    • dilation (int or tuple, optional) – The dilation of the convolution.

    • padding (int or tuple, optional) – How many positions to 0-pad the input with. Default: 0.

    • bias (bool, optional) – If True add a learnable bias to the diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose1d.html index 95cf45033..e5c3bdce0 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose1d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose1d.html @@ -8,7 +8,7 @@ - mlx.nn.ConvTranspose1d — MLX 0.22.0 documentation + mlx.nn.ConvTranspose1d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose2d.html index a21098beb..6746c0008 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose2d.html @@ -8,7 +8,7 @@ - mlx.nn.ConvTranspose2d — MLX 0.22.0 documentation + mlx.nn.ConvTranspose2d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose3d.html index 142a62373..89cbe44c2 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose3d.html @@ -8,7 +8,7 @@ - mlx.nn.ConvTranspose3d — MLX 0.22.0 documentation + mlx.nn.ConvTranspose3d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -885,7 +885,7 @@

      mlx.nn.ConvTranspose3d#

      -class ConvTranspose3d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, bias: bool = True)#
      +class ConvTranspose3d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, bias: bool = True)#

      Applies a 3-dimensional transposed convolution over the multi-channel input image.

      The channels are expected to be last i.e. the input shape should be NDHWC where:

        @@ -905,6 +905,7 @@ applying the filter. Default: 1.

      • padding (int or tuple, optional) – How many positions to 0-pad the input with. Default: 0.

        • +

      • dilation (int or tuple, optional) – The dilation of the convolution.

      • bias (bool, optional) – If True add a learnable bias to the output. Default: True

      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html index bb829affe..71432a6a1 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html @@ -8,7 +8,7 @@ - mlx.nn.Dropout — MLX 0.22.0 documentation + mlx.nn.Dropout — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html index 305eae9e3..0b7ef6bb4 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html @@ -8,7 +8,7 @@ - mlx.nn.Dropout2d — MLX 0.22.0 documentation + mlx.nn.Dropout2d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html index 4b4cca120..4eff7128c 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html @@ -8,7 +8,7 @@ - mlx.nn.Dropout3d — MLX 0.22.0 documentation + mlx.nn.Dropout3d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ELU.html index ff55986d8..e039d9d4d 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ELU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ELU.html @@ -8,7 +8,7 @@ - mlx.nn.ELU — MLX 0.22.0 documentation + mlx.nn.ELU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html index 93dfe5c36..722bc2165 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html @@ -8,7 +8,7 @@ - mlx.nn.Embedding — MLX 0.22.0 documentation + mlx.nn.Embedding — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html index 0b63e07db..aafa08b48 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html @@ -8,7 +8,7 @@ - mlx.nn.GELU — MLX 0.22.0 documentation + mlx.nn.GELU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html index 31c3dea9c..dec677d68 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html @@ -8,7 +8,7 @@ - mlx.nn.GLU — MLX 0.22.0 documentation + mlx.nn.GLU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html index bce4311d5..e71121907 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html @@ -8,7 +8,7 @@ - mlx.nn.GRU — MLX 0.22.0 documentation + mlx.nn.GRU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html index 1186010a3..b00e90182 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html @@ -8,7 +8,7 @@ - mlx.nn.GroupNorm — MLX 0.22.0 documentation + mlx.nn.GroupNorm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html b/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html index 54483a99e..f5f29fdd2 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html @@ -8,7 +8,7 @@ - mlx.nn.HardShrink — MLX 0.22.0 documentation + mlx.nn.HardShrink — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html b/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html index 221642fc5..a83936ac0 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html @@ -8,7 +8,7 @@ - mlx.nn.HardTanh — MLX 0.22.0 documentation + mlx.nn.HardTanh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html index c1d5d9181..eef054e42 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html @@ -8,7 +8,7 @@ - mlx.nn.Hardswish — MLX 0.22.0 documentation + mlx.nn.Hardswish — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html index 983cbfb54..6ba89df54 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html @@ -8,7 +8,7 @@ - mlx.nn.InstanceNorm — MLX 0.22.0 documentation + mlx.nn.InstanceNorm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html index 5fa62026d..4314a7bd5 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html @@ -8,7 +8,7 @@ - mlx.nn.LSTM — MLX 0.22.0 documentation + mlx.nn.LSTM — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html index 5b45e3635..c86b4c5bf 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html @@ -8,7 +8,7 @@ - mlx.nn.LayerNorm — MLX 0.22.0 documentation + mlx.nn.LayerNorm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html index 752730cd9..b4d95aa4a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html @@ -8,7 +8,7 @@ - mlx.nn.LeakyReLU — MLX 0.22.0 documentation + mlx.nn.LeakyReLU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html index 0dbcdc28a..836a4054f 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html @@ -8,7 +8,7 @@ - mlx.nn.Linear — MLX 0.22.0 documentation + mlx.nn.Linear — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LogSigmoid.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LogSigmoid.html index 4217cc398..dcaed6f3a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LogSigmoid.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LogSigmoid.html @@ -8,7 +8,7 @@ - mlx.nn.LogSigmoid — MLX 0.22.0 documentation + mlx.nn.LogSigmoid — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LogSoftmax.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LogSoftmax.html index 28c775354..068e2e064 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LogSoftmax.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LogSoftmax.html @@ -8,7 +8,7 @@ - mlx.nn.LogSoftmax — MLX 0.22.0 documentation + mlx.nn.LogSoftmax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html index 693535d15..a9fe91bf5 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html @@ -8,7 +8,7 @@ - mlx.nn.MaxPool1d — MLX 0.22.0 documentation + mlx.nn.MaxPool1d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html index ea53142b0..28b492466 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html @@ -8,7 +8,7 @@ - mlx.nn.MaxPool2d — MLX 0.22.0 documentation + mlx.nn.MaxPool2d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool3d.html index a293949ff..cb83ed111 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool3d.html @@ -8,7 +8,7 @@ - mlx.nn.MaxPool3d — MLX 0.22.0 documentation + mlx.nn.MaxPool3d — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html index a56deaab5..730c600a7 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html @@ -8,7 +8,7 @@ - mlx.nn.Mish — MLX 0.22.0 documentation + mlx.nn.Mish — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html index 15d9b4bd1..afaa86011 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html @@ -8,7 +8,7 @@ - mlx.nn.Module.apply — MLX 0.22.0 documentation + mlx.nn.Module.apply — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html index 9e98d135f..2844f9550 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.apply_to_modules — MLX 0.22.0 documentation + mlx.nn.Module.apply_to_modules — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html index 8742c5cb4..8c9d4598c 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html @@ -8,7 +8,7 @@ - mlx.nn.Module.children — MLX 0.22.0 documentation + mlx.nn.Module.children — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html index 68304e65b..ef9912fee 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html @@ -8,7 +8,7 @@ - mlx.nn.Module.eval — MLX 0.22.0 documentation + mlx.nn.Module.eval — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html index f6cf0d398..0cdb5548f 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html @@ -8,7 +8,7 @@ - mlx.nn.Module.filter_and_map — MLX 0.22.0 documentation + mlx.nn.Module.filter_and_map — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html index 8ab528181..1fd2a4662 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html @@ -8,7 +8,7 @@ - mlx.nn.Module.freeze — MLX 0.22.0 documentation + mlx.nn.Module.freeze — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html index b7543b5a3..5c3207335 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.leaf_modules — MLX 0.22.0 documentation + mlx.nn.Module.leaf_modules — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html index c9f18856c..3fe1d95ca 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html @@ -8,7 +8,7 @@ - mlx.nn.Module.load_weights — MLX 0.22.0 documentation + mlx.nn.Module.load_weights — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html index a6cc77943..a27e6f61e 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.modules — MLX 0.22.0 documentation + mlx.nn.Module.modules — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html index ae7ffb806..0097c37d1 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.named_modules — MLX 0.22.0 documentation + mlx.nn.Module.named_modules — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html index 8d791b68a..00c1c766c 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html @@ -8,7 +8,7 @@ - mlx.nn.Module.parameters — MLX 0.22.0 documentation + mlx.nn.Module.parameters — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html index 65b720091..b9140b17d 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html @@ -8,7 +8,7 @@ - mlx.nn.Module.save_weights — MLX 0.22.0 documentation + mlx.nn.Module.save_weights — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html index a829ac333..874b117ca 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html @@ -8,7 +8,7 @@ - mlx.nn.Module.set_dtype — MLX 0.22.0 documentation + mlx.nn.Module.set_dtype — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html index d5df25b40..c00cdb541 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html @@ -8,7 +8,7 @@ - mlx.nn.Module.state — MLX 0.22.0 documentation + mlx.nn.Module.state — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html index 70fa76e63..bf7a84a76 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html @@ -8,7 +8,7 @@ - mlx.nn.Module.train — MLX 0.22.0 documentation + mlx.nn.Module.train — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html index 981f847ca..fc24c0914 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html @@ -8,7 +8,7 @@ - mlx.nn.Module.trainable_parameters — MLX 0.22.0 documentation + mlx.nn.Module.trainable_parameters — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html index 9fa0e3547..1de7ac7a7 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html @@ -8,7 +8,7 @@ - mlx.nn.Module.training — MLX 0.22.0 documentation + mlx.nn.Module.training — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html index 7fb6b19bf..f02173dd3 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html @@ -8,7 +8,7 @@ - mlx.nn.Module.unfreeze — MLX 0.22.0 documentation + mlx.nn.Module.unfreeze — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html index c199ddb19..9b21c31b7 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html @@ -8,7 +8,7 @@ - mlx.nn.Module.update — MLX 0.22.0 documentation + mlx.nn.Module.update — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html index a3e84f33e..232ea25d1 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.update_modules — MLX 0.22.0 documentation + mlx.nn.Module.update_modules — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html index bfd341493..73192ff3c 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html @@ -8,7 +8,7 @@ - mlx.nn.MultiHeadAttention — MLX 0.22.0 documentation + mlx.nn.MultiHeadAttention — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html index 7d2ef4cd2..c2d279374 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html @@ -8,7 +8,7 @@ - mlx.nn.PReLU — MLX 0.22.0 documentation + mlx.nn.PReLU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html index 45e42c541..c9def9a46 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html @@ -8,7 +8,7 @@ - mlx.nn.QuantizedEmbedding — MLX 0.22.0 documentation + mlx.nn.QuantizedEmbedding — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html index 13869e017..c81fd5f0b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html @@ -8,7 +8,7 @@ - mlx.nn.QuantizedLinear — MLX 0.22.0 documentation + mlx.nn.QuantizedLinear — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html index 43ea44dc3..61934519f 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html @@ -8,7 +8,7 @@ - mlx.nn.RMSNorm — MLX 0.22.0 documentation + mlx.nn.RMSNorm — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html b/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html index 1489ebfac..df415c1df 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html @@ -8,7 +8,7 @@ - mlx.nn.RNN — MLX 0.22.0 documentation + mlx.nn.RNN — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html index 4522563b1..8cd998c17 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html @@ -8,7 +8,7 @@ - mlx.nn.ReLU — MLX 0.22.0 documentation + mlx.nn.ReLU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html index 145e8ceaa..f7b94d2c4 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html @@ -8,7 +8,7 @@ - mlx.nn.ReLU6 — MLX 0.22.0 documentation + mlx.nn.ReLU6 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html b/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html index 60723ab58..88934af85 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html @@ -8,7 +8,7 @@ - mlx.nn.RoPE — MLX 0.22.0 documentation + mlx.nn.RoPE — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html index 6d3067959..7763c5ab7 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html @@ -8,7 +8,7 @@ - mlx.nn.SELU — MLX 0.22.0 documentation + mlx.nn.SELU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html index d2d95da6b..c927ec9f8 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html @@ -8,7 +8,7 @@ - mlx.nn.Sequential — MLX 0.22.0 documentation + mlx.nn.Sequential — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html index eeb56cb5d..792ef3545 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html @@ -8,7 +8,7 @@ - mlx.nn.SiLU — MLX 0.22.0 documentation + mlx.nn.SiLU — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Sigmoid.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Sigmoid.html index 31cbe65d4..0e54c2353 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Sigmoid.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Sigmoid.html @@ -8,7 +8,7 @@ - mlx.nn.Sigmoid — MLX 0.22.0 documentation + mlx.nn.Sigmoid — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html b/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html index c7af4e064..47580da27 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html @@ -8,7 +8,7 @@ - mlx.nn.SinusoidalPositionalEncoding — MLX 0.22.0 documentation + mlx.nn.SinusoidalPositionalEncoding — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html index 9b84b2277..5fabe41da 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html @@ -8,7 +8,7 @@ - mlx.nn.Softmax — MLX 0.22.0 documentation + mlx.nn.Softmax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html index 31ff93015..f81c5e1b6 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html @@ -8,7 +8,7 @@ - mlx.nn.Softmin — MLX 0.22.0 documentation + mlx.nn.Softmin — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html index 3f97e4e65..102129f48 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html @@ -8,7 +8,7 @@ - mlx.nn.Softplus — MLX 0.22.0 documentation + mlx.nn.Softplus — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html index 6bf26a9fd..15b750af6 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html @@ -8,7 +8,7 @@ - mlx.nn.Softshrink — MLX 0.22.0 documentation + mlx.nn.Softshrink — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html index 1d5cd2f64..9634662b0 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html @@ -8,7 +8,7 @@ - mlx.nn.Softsign — MLX 0.22.0 documentation + mlx.nn.Softsign — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html index c4c68f7eb..182a122b4 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html @@ -8,7 +8,7 @@ - mlx.nn.Step — MLX 0.22.0 documentation + mlx.nn.Step — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html index 7e45c9fbd..ccddf8fa3 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html @@ -8,7 +8,7 @@ - mlx.nn.Tanh — MLX 0.22.0 documentation + mlx.nn.Tanh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html index 0a00735fe..b74b4d6ee 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html @@ -8,7 +8,7 @@ - mlx.nn.Transformer — MLX 0.22.0 documentation + mlx.nn.Transformer — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html index abb7b6795..14d432a88 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html @@ -8,7 +8,7 @@ - mlx.nn.Upsample — MLX 0.22.0 documentation + mlx.nn.Upsample — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html index 118196163..c333507cf 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html @@ -8,7 +8,7 @@ - mlx.nn.init.constant — MLX 0.22.0 documentation + mlx.nn.init.constant — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html index 34e27639d..64f53b0c3 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html @@ -8,7 +8,7 @@ - mlx.nn.init.glorot_normal — MLX 0.22.0 documentation + mlx.nn.init.glorot_normal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html index 6d83bec31..667a94a0b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html @@ -8,7 +8,7 @@ - mlx.nn.init.glorot_uniform — MLX 0.22.0 documentation + mlx.nn.init.glorot_uniform — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html index 585ca60b6..8f3e36a6b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html @@ -8,7 +8,7 @@ - mlx.nn.init.he_normal — MLX 0.22.0 documentation + mlx.nn.init.he_normal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html index 6e4e77933..6cb2ad8c1 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html @@ -8,7 +8,7 @@ - mlx.nn.init.he_uniform — MLX 0.22.0 documentation + mlx.nn.init.he_uniform — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html index 56e8ca8fa..5b56cc389 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html @@ -8,7 +8,7 @@ - mlx.nn.init.identity — MLX 0.22.0 documentation + mlx.nn.init.identity — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html index 78edf7b5a..b1fc38dbe 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html @@ -8,7 +8,7 @@ - mlx.nn.init.normal — MLX 0.22.0 documentation + mlx.nn.init.normal — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html index 4d446f036..af09c532b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html @@ -8,7 +8,7 @@ - mlx.nn.init.uniform — MLX 0.22.0 documentation + mlx.nn.init.uniform — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.celu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.celu.html index b2a1c5f2b..09c3ff31d 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.celu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.celu.html @@ -8,7 +8,7 @@ - mlx.nn.celu — MLX 0.22.0 documentation + mlx.nn.celu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html index 896c57eb1..a4fdb8039 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html @@ -8,7 +8,7 @@ - mlx.nn.elu — MLX 0.22.0 documentation + mlx.nn.elu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html index ab0e89da4..00ef1a375 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html @@ -8,7 +8,7 @@ - mlx.nn.gelu — MLX 0.22.0 documentation + mlx.nn.gelu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html index ffdf1f9f4..e58765daa 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html @@ -8,7 +8,7 @@ - mlx.nn.gelu_approx — MLX 0.22.0 documentation + mlx.nn.gelu_approx — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html index 8ccbea44b..4d5f7f9bf 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html @@ -8,7 +8,7 @@ - mlx.nn.gelu_fast_approx — MLX 0.22.0 documentation + mlx.nn.gelu_fast_approx — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html index 00928d0ce..3e723158a 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html @@ -8,7 +8,7 @@ - mlx.nn.glu — MLX 0.22.0 documentation + mlx.nn.glu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html index 2f6a67af9..f4908dc33 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html @@ -8,7 +8,7 @@ - mlx.nn.hard_shrink — MLX 0.22.0 documentation + mlx.nn.hard_shrink — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html index e03f11001..940644bce 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html @@ -8,7 +8,7 @@ - mlx.nn.hard_tanh — MLX 0.22.0 documentation + mlx.nn.hard_tanh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html index 4cc9294e4..7f0421dca 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html @@ -8,7 +8,7 @@ - mlx.nn.hardswish — MLX 0.22.0 documentation + mlx.nn.hardswish — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html index ff92371a7..f063e612e 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html @@ -8,7 +8,7 @@ - mlx.nn.leaky_relu — MLX 0.22.0 documentation + mlx.nn.leaky_relu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html index 61e581642..5a8aa03b1 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html @@ -8,7 +8,7 @@ - mlx.nn.log_sigmoid — MLX 0.22.0 documentation + mlx.nn.log_sigmoid — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html index 0f39933c1..7372e5510 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html @@ -8,7 +8,7 @@ - mlx.nn.log_softmax — MLX 0.22.0 documentation + mlx.nn.log_softmax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html index 4b15b4220..07d90ea7a 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html @@ -8,7 +8,7 @@ - mlx.nn.losses.binary_cross_entropy — MLX 0.22.0 documentation + mlx.nn.losses.binary_cross_entropy — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html index e60b43488..544ace61d 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.cosine_similarity_loss — MLX 0.22.0 documentation + mlx.nn.losses.cosine_similarity_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html index 09caa5645..0e912855f 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html @@ -8,7 +8,7 @@ - mlx.nn.losses.cross_entropy — MLX 0.22.0 documentation + mlx.nn.losses.cross_entropy — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html index 626c59b81..7843c7797 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.gaussian_nll_loss — MLX 0.22.0 documentation + mlx.nn.losses.gaussian_nll_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html index bb6f23458..1fbf38807 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.hinge_loss — MLX 0.22.0 documentation + mlx.nn.losses.hinge_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html index 5f1188222..e67a47c00 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.huber_loss — MLX 0.22.0 documentation + mlx.nn.losses.huber_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html index 18478a804..2550c596b 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.kl_div_loss — MLX 0.22.0 documentation + mlx.nn.losses.kl_div_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html index 91c198af3..aa9a5bc40 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.l1_loss — MLX 0.22.0 documentation + mlx.nn.losses.l1_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html index 027261b89..1587a36a3 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.log_cosh_loss — MLX 0.22.0 documentation + mlx.nn.losses.log_cosh_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html index cb8669974..0fb1ddbb9 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.margin_ranking_loss — MLX 0.22.0 documentation + mlx.nn.losses.margin_ranking_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html index dae7a8e53..87203292a 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.mse_loss — MLX 0.22.0 documentation + mlx.nn.losses.mse_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html index 5d583e114..e49e68ff4 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.nll_loss — MLX 0.22.0 documentation + mlx.nn.losses.nll_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html index a420a0af3..95176e58b 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.smooth_l1_loss — MLX 0.22.0 documentation + mlx.nn.losses.smooth_l1_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html index fc83f79dd..0faae7d7e 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.triplet_loss — MLX 0.22.0 documentation + mlx.nn.losses.triplet_loss — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html index 5cf19708f..3eb454946 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html @@ -8,7 +8,7 @@ - mlx.nn.mish — MLX 0.22.0 documentation + mlx.nn.mish — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html index 34e88d60a..b4d38b35f 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html @@ -8,7 +8,7 @@ - mlx.nn.prelu — MLX 0.22.0 documentation + mlx.nn.prelu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html index 90c89fc36..8f3321cc7 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html @@ -8,7 +8,7 @@ - mlx.nn.relu — MLX 0.22.0 documentation + mlx.nn.relu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html index c2a7aee46..843f7ae0f 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html @@ -8,7 +8,7 @@ - mlx.nn.relu6 — MLX 0.22.0 documentation + mlx.nn.relu6 — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html index dc4f7d35f..236cdd47c 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html @@ -8,7 +8,7 @@ - mlx.nn.selu — MLX 0.22.0 documentation + mlx.nn.selu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html index 0f587d3c4..7e45b513f 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html @@ -8,7 +8,7 @@ - mlx.nn.sigmoid — MLX 0.22.0 documentation + mlx.nn.sigmoid — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html index 977c3c3d9..eebf50c82 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html @@ -8,7 +8,7 @@ - mlx.nn.silu — MLX 0.22.0 documentation + mlx.nn.silu — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html index 3688df40e..4e7e9ac42 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html @@ -8,7 +8,7 @@ - mlx.nn.softmax — MLX 0.22.0 documentation + mlx.nn.softmax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html index 46cc1ca43..967761dee 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html @@ -8,7 +8,7 @@ - mlx.nn.softmin — MLX 0.22.0 documentation + mlx.nn.softmin — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html index e1a932e8d..ae7125402 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html @@ -8,7 +8,7 @@ - mlx.nn.softplus — MLX 0.22.0 documentation + mlx.nn.softplus — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html index bd4f243ca..e6c8246c8 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html @@ -8,7 +8,7 @@ - mlx.nn.softshrink — MLX 0.22.0 documentation + mlx.nn.softshrink — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html index 56ba93b90..d759e0447 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html @@ -8,7 +8,7 @@ - mlx.nn.step — MLX 0.22.0 documentation + mlx.nn.step — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html index e1b2e8999..7eb073c3e 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html @@ -8,7 +8,7 @@ - mlx.nn.tanh — MLX 0.22.0 documentation + mlx.nn.tanh — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/functions.html b/docs/build/html/python/nn/functions.html index 69ec57c29..a07661f31 100644 --- a/docs/build/html/python/nn/functions.html +++ b/docs/build/html/python/nn/functions.html @@ -8,7 +8,7 @@ - Functions — MLX 0.22.0 documentation + Functions — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/init.html b/docs/build/html/python/nn/init.html index 4a87e7686..6e2bc3571 100644 --- a/docs/build/html/python/nn/init.html +++ b/docs/build/html/python/nn/init.html @@ -8,7 +8,7 @@ - Initializers — MLX 0.22.0 documentation + Initializers — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/layers.html b/docs/build/html/python/nn/layers.html index 8337273df..d136fcf90 100644 --- a/docs/build/html/python/nn/layers.html +++ b/docs/build/html/python/nn/layers.html @@ -8,7 +8,7 @@ - Layers — MLX 0.22.0 documentation + Layers — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/losses.html b/docs/build/html/python/nn/losses.html index 78f35f512..e3d40216d 100644 --- a/docs/build/html/python/nn/losses.html +++ b/docs/build/html/python/nn/losses.html @@ -8,7 +8,7 @@ - Loss Functions — MLX 0.22.0 documentation + Loss Functions — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/nn/module.html b/docs/build/html/python/nn/module.html index 1b91450f4..3bf146097 100644 --- a/docs/build/html/python/nn/module.html +++ b/docs/build/html/python/nn/module.html @@ -8,7 +8,7 @@ - Module — MLX 0.22.0 documentation + Module — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/ops.html b/docs/build/html/python/ops.html index a6b58fd80..d68756832 100644 --- a/docs/build/html/python/ops.html +++ b/docs/build/html/python/ops.html @@ -8,7 +8,7 @@ - Operations — MLX 0.22.0 documentation + Operations — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers.html b/docs/build/html/python/optimizers.html index 861662cb2..db48d236f 100644 --- a/docs/build/html/python/optimizers.html +++ b/docs/build/html/python/optimizers.html @@ -8,7 +8,7 @@ - Optimizers — MLX 0.22.0 documentation + Optimizers — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdaDelta.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdaDelta.html index 81aba7dab..697f6fc2b 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdaDelta.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdaDelta.html @@ -8,7 +8,7 @@ - mlx.optimizers.AdaDelta — MLX 0.22.0 documentation + mlx.optimizers.AdaDelta — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html index 2758037b8..bc42cce9e 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html @@ -8,7 +8,7 @@ - mlx.optimizers.Adafactor — MLX 0.22.0 documentation + mlx.optimizers.Adafactor — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html index f4a477065..1a42ba67b 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html @@ -8,7 +8,7 @@ - mlx.optimizers.Adagrad — MLX 0.22.0 documentation + mlx.optimizers.Adagrad — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html index 62a7625e1..da1d51e68 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html @@ -8,7 +8,7 @@ - mlx.optimizers.Adam — MLX 0.22.0 documentation + mlx.optimizers.Adam — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html index 76baf081d..5171766f8 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html @@ -8,7 +8,7 @@ - mlx.optimizers.AdamW — MLX 0.22.0 documentation + mlx.optimizers.AdamW — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html index 3b22ba0a4..b21653c4c 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html @@ -8,7 +8,7 @@ - mlx.optimizers.Adamax — MLX 0.22.0 documentation + mlx.optimizers.Adamax — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html index 6ea482ed2..f9b55f1a2 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html @@ -8,7 +8,7 @@ - mlx.optimizers.Lion — MLX 0.22.0 documentation + mlx.optimizers.Lion — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html index 60c81f910..0265f3443 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html @@ -8,7 +8,7 @@ - mlx.optimizers.Optimizer.apply_gradients — MLX 0.22.0 documentation + mlx.optimizers.Optimizer.apply_gradients — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html index a1acfafab..42e137dd4 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html @@ -8,7 +8,7 @@ - mlx.optimizers.Optimizer.init — MLX 0.22.0 documentation + mlx.optimizers.Optimizer.init — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html index 09621a12f..970849a8e 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html @@ -8,7 +8,7 @@ - mlx.optimizers.Optimizer.state — MLX 0.22.0 documentation + mlx.optimizers.Optimizer.state — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html index 058c1de57..4043a4bd6 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html @@ -8,7 +8,7 @@ - mlx.optimizers.Optimizer.update — MLX 0.22.0 documentation + mlx.optimizers.Optimizer.update — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html index 58179041a..54e5b00aa 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html @@ -8,7 +8,7 @@ - mlx.optimizers.RMSprop — MLX 0.22.0 documentation + mlx.optimizers.RMSprop — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html index 21f4f2ba1..0927a2833 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html @@ -8,7 +8,7 @@ - mlx.optimizers.SGD — MLX 0.22.0 documentation + mlx.optimizers.SGD — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html index 91ca53dd7..19a7678b9 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html @@ -8,7 +8,7 @@ - mlx.optimizers.cosine_decay — MLX 0.22.0 documentation + mlx.optimizers.cosine_decay — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html index 18576db01..d32d66ddd 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html @@ -8,7 +8,7 @@ - mlx.optimizers.exponential_decay — MLX 0.22.0 documentation + mlx.optimizers.exponential_decay — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html index bab3e61f3..adab773b3 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html @@ -8,7 +8,7 @@ - mlx.optimizers.join_schedules — MLX 0.22.0 documentation + mlx.optimizers.join_schedules — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html index eb051bd4c..a687d490e 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html @@ -8,7 +8,7 @@ - mlx.optimizers.linear_schedule — MLX 0.22.0 documentation + mlx.optimizers.linear_schedule — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html index 3ab03b892..48df818f6 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html @@ -8,7 +8,7 @@ - mlx.optimizers.step_decay — MLX 0.22.0 documentation + mlx.optimizers.step_decay — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/common_optimizers.html b/docs/build/html/python/optimizers/common_optimizers.html index 2dbb508d7..89bcbd34e 100644 --- a/docs/build/html/python/optimizers/common_optimizers.html +++ b/docs/build/html/python/optimizers/common_optimizers.html @@ -8,7 +8,7 @@ - Common Optimizers — MLX 0.22.0 documentation + Common Optimizers — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/optimizer.html b/docs/build/html/python/optimizers/optimizer.html index e5cb9500a..17dd43dd2 100644 --- a/docs/build/html/python/optimizers/optimizer.html +++ b/docs/build/html/python/optimizers/optimizer.html @@ -8,7 +8,7 @@ - Optimizer — MLX 0.22.0 documentation + Optimizer — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/optimizers/schedulers.html b/docs/build/html/python/optimizers/schedulers.html index 24a57768d..70023b155 100644 --- a/docs/build/html/python/optimizers/schedulers.html +++ b/docs/build/html/python/optimizers/schedulers.html @@ -8,7 +8,7 @@ - Schedulers — MLX 0.22.0 documentation + Schedulers — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/random.html b/docs/build/html/python/random.html index dc957d242..786166fc1 100644 --- a/docs/build/html/python/random.html +++ b/docs/build/html/python/random.html @@ -8,7 +8,7 @@ - Random — MLX 0.22.0 documentation + Random — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/transforms.html b/docs/build/html/python/transforms.html index 0d582a971..4638ca5aa 100644 --- a/docs/build/html/python/transforms.html +++ b/docs/build/html/python/transforms.html @@ -8,7 +8,7 @@ - Transforms — MLX 0.22.0 documentation + Transforms — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/python/tree_utils.html b/docs/build/html/python/tree_utils.html index 529b322b0..750b58867 100644 --- a/docs/build/html/python/tree_utils.html +++ b/docs/build/html/python/tree_utils.html @@ -8,7 +8,7 @@ - Tree Utils — MLX 0.22.0 documentation + Tree Utils — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/quantized_8h.html b/docs/build/html/quantized_8h.html index df5e1d508..7fb48e3f7 100644 --- a/docs/build/html/quantized_8h.html +++ b/docs/build/html/quantized_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/quantized.h File Reference @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Namespaces | +Typedefs | +Functions
      +
      ring.h File Reference
      +
      +
      +
      #include "mlx/distributed/distributed.h"
      +
      +

      Go to the source code of this file.

      + + + + + + + + + + +

      +Namespaces

      namespace  mlx
       
      namespace  mlx::core
       
      namespace  mlx::core::distributed
       
      namespace  mlx::core::distributed::ring
       
      + + + +

      +Typedefs

      using mlx::core::distributed::ring::GroupImpl = mlx::core::distributed::detail::GroupImpl
       
      + + + + + +

      +Functions

      bool mlx::core::distributed::ring::is_available ()
       
      std::shared_ptr< GroupImpl > mlx::core::distributed::ring::init (bool strict=false)
       
      +
      +
      + + + + diff --git a/docs/build/html/ring_8h.js b/docs/build/html/ring_8h.js new file mode 100644 index 000000000..659c76a34 --- /dev/null +++ b/docs/build/html/ring_8h.js @@ -0,0 +1,6 @@ +var ring_8h = +[ + [ "mlx::core::distributed::ring::GroupImpl", "namespacemlx_1_1core_1_1distributed_1_1ring.html#a6e4d590e07f0cf3cc2d15f258f9438ed", null ], + [ "mlx::core::distributed::ring::init", "namespacemlx_1_1core_1_1distributed_1_1ring.html#a81a13abe6f334d2f6b058b39a2221e67", null ], + [ "mlx::core::distributed::ring::is_available", "namespacemlx_1_1core_1_1distributed_1_1ring.html#a1238e89ee95ba016741f0abe91b540ac", null ] +]; \ No newline at end of file diff --git a/docs/build/html/ring_8h_source.html b/docs/build/html/ring_8h_source.html new file mode 100644 index 000000000..6e7411cb4 --- /dev/null +++ b/docs/build/html/ring_8h_source.html @@ -0,0 +1,138 @@ + + + + + + + +MLX: mlx/distributed/ring/ring.h Source File + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      ring.h
      +
      +
      +Go to the documentation of this file.
      1// Copyright © 2024 Apple Inc.
      +
      2
      +
      3#include "mlx/distributed/distributed.h"
      +
      4
      +
      +
      5namespace mlx::core::distributed::ring {
      +
      6
      +
      7using GroupImpl = mlx::core::distributed::detail::GroupImpl;
      +
      8
      +
      9bool is_available();
      +
      10std::shared_ptr<GroupImpl> init(bool strict = false);
      +
      11
      +
      12} // namespace mlx::core::distributed::ring
      +
      +
      mlx::core::distributed::detail::GroupImpl
      Abstract base class of a distributed group implementation.
      Definition distributed_impl.h:12
      + +
      mlx::core::distributed::ring
      Definition ring.h:5
      + +
      mlx::core::distributed::ring::GroupImpl
      mlx::core::distributed::detail::GroupImpl GroupImpl
      Definition ring.h:7
      +
      mlx::core::distributed::ring::init
      std::shared_ptr< GroupImpl > init(bool strict=false)
      +
      +
      + + + + diff --git a/docs/build/html/scan_8h.html b/docs/build/html/scan_8h.html index 076eeff10..708ab50cb 100644 --- a/docs/build/html/scan_8h.html +++ b/docs/build/html/scan_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/scan.h File Reference @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Functions
      +
      scatter_axis.h File Reference
      +
      +
      + +

      Go to the source code of this file.

      + + + + + +

      +Functions

      template<typename T, typename IdxT, typename LocT, typename Op, bool UpdC, bool IdxC>
      void scatter_axis (const device T *upd, const device IdxT *indices, device mlx_atomic< T > *out, const constant int *shape, const constant int64_t *upd_strides, const constant int64_t *idx_strides, const constant size_t &ndim, const constant int &axis, const constant int &out_axis_size, const constant size_t &upd_ax_stride, const constant size_t &idx_ax_stride, uint3 index, uint3 grid_dim)
       
      +

      Function Documentation

      + +

      ◆ scatter_axis()

      + +
      +
      +
      +template<typename T, typename IdxT, typename LocT, typename Op, bool UpdC, bool IdxC>
      + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      void scatter_axis (const device T * upd,
      const device IdxT * indices,
      device mlx_atomic< T > * out,
      const constant int * shape,
      const constant int64_t * upd_strides,
      const constant int64_t * idx_strides,
      const constant size_t & ndim,
      const constant int & axis,
      const constant int & out_axis_size,
      const constant size_t & upd_ax_stride,
      const constant size_t & idx_ax_stride,
      uint3 index,
      uint3 grid_dim )
      +
      + +
      +
      +
      +
      + + + + diff --git a/docs/build/html/scatter__axis_8h.js b/docs/build/html/scatter__axis_8h.js new file mode 100644 index 000000000..61d2e1d8a --- /dev/null +++ b/docs/build/html/scatter__axis_8h.js @@ -0,0 +1,4 @@ +var scatter__axis_8h = +[ + [ "scatter_axis", "scatter__axis_8h.html#af78a7935b05dabd42c2cdff4cf375130", null ] +]; \ No newline at end of file diff --git a/docs/build/html/scatter__axis_8h_source.html b/docs/build/html/scatter__axis_8h_source.html new file mode 100644 index 000000000..dd3e5b68e --- /dev/null +++ b/docs/build/html/scatter__axis_8h_source.html @@ -0,0 +1,175 @@ + + + + + + + +MLX: mlx/backend/metal/kernels/scatter_axis.h Source File + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      scatter_axis.h
      +
      +
      +Go to the documentation of this file.
      1// Copyright © 2025 Apple Inc.
      +
      2
      +
      3#pragma once
      +
      4
      +
      5template <
      +
      6 typename T,
      +
      7 typename IdxT,
      +
      8 typename LocT,
      +
      9 typename Op,
      +
      10 bool UpdC,
      +
      11 bool IdxC>
      +
      +
      12[[kernel]] void scatter_axis(
      +
      13 const device T* upd [[buffer(0)]],
      +
      14 const device IdxT* indices [[buffer(1)]],
      +
      15 device mlx_atomic<T>* out [[buffer(2)]],
      +
      16 const constant int* shape [[buffer(3)]],
      +
      17 const constant int64_t* upd_strides [[buffer(4)]],
      +
      18 const constant int64_t* idx_strides [[buffer(5)]],
      +
      19 const constant size_t& ndim [[buffer(6)]],
      +
      20 const constant int& axis [[buffer(7)]],
      +
      21 const constant int& out_axis_size [[buffer(8)]],
      +
      22 const constant size_t& upd_ax_stride [[buffer(9)]],
      +
      23 const constant size_t& idx_ax_stride [[buffer(10)]],
      +
      24 uint3 index [[thread_position_in_grid]],
      +
      25 uint3 grid_dim [[threads_per_grid]]) {
      +
      26 Op op;
      +
      27
      +
      28 LocT elem_idx = index.z * static_cast<LocT>(grid_dim.x);
      +
      29
      +
      30 LocT idx_loc = index.y * static_cast<LocT>(idx_ax_stride);
      +
      31 if (IdxC) {
      +
      32 idx_loc += elem_idx * grid_dim.y + index.x;
      +
      33 } else {
      +
      34 idx_loc += elem_to_loc<LocT>(elem_idx + index.x, shape, idx_strides, ndim);
      +
      35 }
      +
      36
      +
      37 auto idx_val = indices[idx_loc];
      +
      38 if (is_signed_v<IdxT>) {
      +
      39 idx_val = (idx_val < 0) ? idx_val + out_axis_size : idx_val;
      +
      40 }
      +
      41
      +
      42 LocT upd_idx = index.y * static_cast<LocT>(upd_ax_stride);
      +
      43 if (UpdC) {
      +
      44 upd_idx += elem_idx * grid_dim.y + index.x;
      +
      45 } else {
      +
      46 upd_idx += elem_to_loc<LocT>(elem_idx + index.x, shape, upd_strides, ndim);
      +
      47 }
      +
      48
      +
      49 LocT out_idx = elem_idx * static_cast<LocT>(out_axis_size) +
      +
      50 idx_val * grid_dim.x + index.x;
      +
      51 op.atomic_update(out, upd[upd_idx], out_idx);
      +
      52}
      +
      +
      elem_to_loc
      METAL_FUNC IdxT elem_to_loc(IdxT elem, constant const int *shape, constant const int64_t *strides, int ndim)
      Definition utils.h:93
      +
      scatter_axis
      void scatter_axis(const device T *upd, const device IdxT *indices, device mlx_atomic< T > *out, const constant int *shape, const constant int64_t *upd_strides, const constant int64_t *idx_strides, const constant size_t &ndim, const constant int &axis, const constant int &out_axis_size, const constant size_t &upd_ax_stride, const constant size_t &idx_ax_stride, uint3 index, uint3 grid_dim)
      Definition scatter_axis.h:12
      +
      mlx_atomic
      Definition atomic.h:25
      +
      +
      + + + + diff --git a/docs/build/html/scheduler_8h.html b/docs/build/html/scheduler_8h.html index 678418c79..c9fc33caa 100644 --- a/docs/build/html/scheduler_8h.html +++ b/docs/build/html/scheduler_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/scheduler.h File Reference @@ -52,7 +52,7 @@ - + + @@ -50,7 +50,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/search/all_1.js b/docs/build/html/search/all_1.js index 7da1edcd9..5a53b034a 100644 --- a/docs/build/html/search/all_1.js +++ b/docs/build/html/search/all_1.js @@ -5,116 +5,118 @@ var searchData= ['a_5fstr_5fm_2',['A_str_m',['../structmlx_1_1steel_1_1_block_m_m_a.html#ab9c7f5386594497f5f4df7e59670b877',1,'mlx::steel::BlockMMA']]], ['abort_5fwith_5fexception_3',['abort_with_exception',['../namespacemlx_1_1core.html#ae1b6e5cfd27b1526285648686898e011',1,'mlx::core']]], ['abs_4',['Abs',['../struct_abs.html',1,'Abs'],['../classmlx_1_1core_1_1_abs.html',1,'mlx::core::Abs'],['../structmlx_1_1core_1_1detail_1_1_abs.html',1,'mlx::core::detail::Abs'],['../classmlx_1_1core_1_1_abs.html#a1247e72feb640fb562d036b2dd1ae4ad',1,'mlx::core::Abs::Abs()']]], - ['abs_5',['abs',['../namespacemetal.html#a87c5122c60f9a12afceb9925a5b78ffb',1,'metal::abs()'],['../namespacemetal_1_1fast.html#a90d2973f71f83180e7f02e38d11c7a8f',1,'metal::fast::abs()'],['../namespacemetal_1_1precise.html#a99f2b2746e813b9ca7b4249afbaf2a14',1,'metal::precise::abs()'],['../group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920',1,'mlx::core::abs()']]], - ['accum_5ftype_6',['accum_type',['../structmlx_1_1steel_1_1_accum_helper.html#ae52abf69e7ba6af1a73d65d57182ed26',1,'mlx::steel::AccumHelper::accum_type'],['../structmlx_1_1steel_1_1_accum_helper.html#ae52abf69e7ba6af1a73d65d57182ed26',1,'mlx::steel::AccumHelper::accum_type']]], - ['accumhelper_7',['AccumHelper',['../structmlx_1_1steel_1_1_accum_helper.html',1,'mlx::steel']]], - ['acos_8',['acos',['../namespacemetal.html#ad4537748b3c832b6569ff7ccb209fcb2',1,'metal::acos()'],['../namespacemetal_1_1fast.html#a805ce5c3a94b618b7349d70bbb82f0b2',1,'metal::fast::acos()'],['../namespacemetal_1_1precise.html#a8a2bcc89fc0b7e74f0453f82f89a8604',1,'metal::precise::acos()']]], - ['acosh_9',['acosh',['../namespacemetal.html#a2d0efb92b7f61eff342d776bd6c5f3a0',1,'metal::acosh()'],['../namespacemetal_1_1fast.html#afb656fc3406649a238b6f1e0509de751',1,'metal::fast::acosh()'],['../namespacemetal_1_1precise.html#a1f489fabffab969b8677b56bb1136067',1,'metal::precise::acosh()']]], - ['add_10',['Add',['../struct_add.html',1,'Add'],['../classmlx_1_1core_1_1_add.html',1,'mlx::core::Add'],['../structmlx_1_1core_1_1detail_1_1_add.html',1,'mlx::core::detail::Add'],['../classmlx_1_1core_1_1_add.html#ae3fd5483f3454eac3df256e3f5f3cdae',1,'mlx::core::Add::Add()']]], - ['add_11',['add',['../group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7',1,'mlx::core']]], - ['add_5fhalf_5fbinops_12',['ADD_HALF_BINOPS',['../half__types_8h.html#a6bc906918877a7084068a9f0ed571dca',1,'ADD_HALF_BINOPS: half_types.h'],['../half__types_8h.html#a6bc906918877a7084068a9f0ed571dca',1,'ADD_HALF_BINOPS: half_types.h']]], - ['add_5ftemporaries_13',['add_temporaries',['../classmlx_1_1core_1_1metal_1_1_device.html#a72ad17c96fc6ce825bc77f0bed657901',1,'mlx::core::metal::Device']]], - ['add_5ftemporary_14',['add_temporary',['../classmlx_1_1core_1_1metal_1_1_device.html#acb90010af0cffe27fd8cc6c253d3a576',1,'mlx::core::metal::Device']]], - ['add_5fvec_15',['add_vec',['../structpocketfft_1_1detail_1_1add__vec.html',1,'pocketfft::detail']]], - ['add_5fvec_3c_20cmplx_3c_20t_20_3e_20_3e_16',['add_vec< cmplx< T > >',['../structpocketfft_1_1detail_1_1add__vec.html',1,'pocketfft::detail']]], - ['add_5fvec_5ft_17',['add_vec_t',['../namespacepocketfft_1_1detail.html#a421aa74fbee775a96463246f72b144d6',1,'pocketfft::detail']]], - ['addmm_18',['AddMM',['../classmlx_1_1core_1_1_add_m_m.html',1,'mlx::core::AddMM'],['../classmlx_1_1core_1_1_add_m_m.html#a8ae4372b3f96e72e8a5a06d59de8a550',1,'mlx::core::AddMM::AddMM()']]], - ['addmm_19',['addmm',['../group__ops.html#ga82a53e083205a965387b3c3e2463244a',1,'mlx::core']]], - ['adj_5fimplicit_5fm_20',['adj_implicit_m',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a5bfca3bc43055013d28430cb1f023756',1,'mlx::steel::Conv2DGeneralJumpParams']]], - ['adj_5fout_5fh_21',['adj_out_h',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a879cc9757f59605a87d936ec4156040d',1,'mlx::steel::Conv2DGeneralJumpParams']]], - ['adj_5fout_5fhw_22',['adj_out_hw',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#aed0ffd63fbc85fd5d5c4cc7b43f68363',1,'mlx::steel::Conv2DGeneralJumpParams']]], - ['adj_5fout_5fw_23',['adj_out_w',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#ab971bf879079895189331fbeaf33c211',1,'mlx::steel::Conv2DGeneralJumpParams']]], - ['adjust_5fmatrix_5foffsets_24',['adjust_matrix_offsets',['../quantized_8h.html#a3e448f8f23c12ffc83bff64ae66bbc66',1,'adjust_matrix_offsets(const device T *&x, const device uint32_t *&w, const device T *&scales, const device T *&biases, device T *&y, int output_stride, const constant int &x_batch_ndims, const constant int *x_shape, const constant int64_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant int64_t *w_strides, const constant int64_t *s_strides, const constant int64_t *b_strides, uint3 tid): quantized.h'],['../quantized_8h.html#acb9a7a0653084295657b630d49c71707',1,'adjust_matrix_offsets(const device T *&x, const device uint32_t *&w, const device T *&scales, const device T *&biases, const device uint32_t *lhs_indices, const device uint32_t *rhs_indices, device T *&y, int output_stride, const constant int &batch_ndims, const constant int *batch_shape, const constant int64_t *lhs_strides, const constant int64_t *rhs_strides, const constant int &x_batch_ndims, const constant int *x_shape, const constant int64_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant int64_t *w_strides, const constant int64_t *s_strides, const constant int64_t *b_strides, uint3 tid): quantized.h']]], - ['advance_25',['advance',['../classpocketfft_1_1detail_1_1multi__iter.html#a5ddcc0666125b3cb6c0d62b30befdd2c',1,'pocketfft::detail::multi_iter::advance()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a73a9ecd3008d2bd35aaa00bf9fac074f',1,'pocketfft::detail::simple_iter::advance()'],['../classpocketfft_1_1detail_1_1rev__iter.html#ad1918c84ae963188afc7599629b29686',1,'pocketfft::detail::rev_iter::advance()']]], - ['affine_5fdequantize_26',['affine_dequantize',['../quantized_8h.html#a6076203615038eb06816158f7b3869c6',1,'affine_dequantize(): quantized.h'],['../namespacemlx_1_1core_1_1fast.html#a12c7ef41409d6fb378008e67b6fab328',1,'mlx::core::fast::affine_dequantize()']]], - ['affine_5fquantize_27',['affine_quantize',['../quantized_8h.html#a47bcf4a14566e01e14bd3c155811db59',1,'affine_quantize(): quantized.h'],['../namespacemlx_1_1core_1_1fast.html#aa4b5f6886b2288cb6dfdd8598579f080',1,'mlx::core::fast::affine_quantize(const array &w, int group_size=64, int bits=4, StreamOrDevice s={})']]], - ['affinequantize_28',['AffineQuantize',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html',1,'mlx::core::fast::AffineQuantize'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a84d5fa9e8c3de407fbcc5f38d2ed1473',1,'mlx::core::fast::AffineQuantize::AffineQuantize()']]], - ['align_5fk_29',['align_K',['../steel__attention_8h.html#a8bdd2cecf97aa5b033152b1d0f0d2416',1,'align_K: steel_attention.h'],['../steel__gemm__fused_8h.html#a8bdd2cecf97aa5b033152b1d0f0d2416',1,'align_K: steel_gemm_fused.h']]], - ['align_5fm_30',['align_M',['../steel__gemm__fused_8h.html#a55af226dc74b0026b7d4b865142a6d21',1,'steel_gemm_fused.h']]], - ['align_5fn_31',['align_N',['../steel__gemm__fused_8h.html#aa3b267252df2dcbfdde8c5f174d27036',1,'steel_gemm_fused.h']]], - ['align_5fq_32',['align_Q',['../steel__attention_8h.html#a171fdea1b23976453f5dc5e6b3161982',1,'steel_attention.h']]], - ['aligned_5falloc_33',['aligned_alloc',['../namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76',1,'pocketfft::detail']]], - ['aligned_5fallocator_34',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html',1,'pocketfft::detail::threading::aligned_allocator< T >'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator(const aligned_allocator< U > &)'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator()=default']]], - ['aligned_5fdealloc_35',['aligned_dealloc',['../namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8',1,'pocketfft::detail']]], - ['all_36',['all',['../group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68',1,'mlx::core::all(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4',1,'mlx::core::all(const array &a, StreamOrDevice s={})'],['../group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59',1,'mlx::core::all(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc',1,'mlx::core::all(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['all_5fgather_37',['all_gather',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a04bb1df23abe5b1f3fa0126375c6cea4',1,'mlx::core::distributed::detail::GroupImpl::all_gather()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04',1,'mlx::core::distributed::detail::all_gather()'],['../namespacemlx_1_1core_1_1distributed.html#a82ef5e8cc7ac62cd228e51b1c1b77cb7',1,'mlx::core::distributed::all_gather()']]], - ['all_5freduce_38',['all_reduce',['../reduce__all_8h.html#a9086a585eda5a887160ee24baae0a7b8',1,'reduce_all.h']]], - ['all_5freduce_5fdispatch_39',['all_reduce_dispatch',['../namespacemlx_1_1core.html#a3ab0fd997d9a35782106ff083a72e098',1,'mlx::core']]], - ['all_5fsum_40',['all_sum',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ae163a6f444c6cc8820288b20f294e483',1,'mlx::core::distributed::detail::GroupImpl::all_sum()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aa1d225b25f7b6426c48c5e35860ee960',1,'mlx::core::distributed::detail::all_sum()'],['../namespacemlx_1_1core_1_1distributed.html#a67ccb1a5445fc6f5db49dd36a15e5980',1,'mlx::core::distributed::all_sum()']]], - ['allclose_41',['allclose',['../group__ops.html#gaf0cd4257de7542daf9faf5e605e31020',1,'mlx::core']]], - ['allgather_42',['AllGather',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html',1,'mlx::core::distributed::AllGather'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#af4b10a5b61f160fb64353057c185b661',1,'mlx::core::distributed::AllGather::AllGather()']]], - ['alloc_5ftmp_43',['alloc_tmp',['../namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, size_t axsize, size_t elemsize)'],['../namespacepocketfft_1_1detail.html#a13832735696303b9559c4663631d5475',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, const shape_t &axes, size_t elemsize)']]], - ['allocate_44',['allocate',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a4f785747b898980756af9e5515363826',1,'pocketfft::detail::threading::aligned_allocator']]], - ['allocator_45',['Allocator',['../classmlx_1_1core_1_1allocator_1_1_allocator.html',1,'mlx::core::allocator::Allocator'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a5803678a418fef687fc65fa9d5c37b65',1,'mlx::core::allocator::Allocator::Allocator()=default'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#aa05c081ce80dc036f9d3dd8c195259d2',1,'mlx::core::allocator::Allocator::Allocator(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a8e8ce346a16cf0c62847bed9289f9959',1,'mlx::core::allocator::Allocator::Allocator(Allocator &&other)=delete']]], - ['allocator_46',['allocator',['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#abf84c726a37df68345589b897b2e35f0',1,'mlx::core::allocator::CommonAllocator::allocator()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#afa1c5a725309caff163c492b5b84491e',1,'mlx::core::metal::MetalAllocator::allocator()'],['../namespacemlx_1_1core_1_1allocator.html#aa23e2f20a336d0b159c097087194634e',1,'mlx::core::allocator::allocator()'],['../namespacemlx_1_1core_1_1metal.html#a74b3558bd518aecde6b14b0ba5e1a0d5',1,'mlx::core::metal::allocator()']]], - ['allocator_2eh_47',['allocator.h',['../allocator_8h.html',1,'(Global Namespace)'],['../backend_2metal_2allocator_8h.html',1,'(Global Namespace)']]], - ['allreduce_48',['AllReduce',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html',1,'mlx::core::distributed::AllReduce'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a2d1ea56cbf72a316680ea90aa6da1c2d',1,'mlx::core::distributed::AllReduce::AllReduce()']]], - ['alpha_49',['alpha',['../structmlx_1_1steel_1_1_transform_axpby.html#ab3223b49c6b3b7f89eba91aeaff9dcff',1,'mlx::steel::TransformAxpby::alpha'],['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#af8693d96512eff3e125d33d203920710',1,'mlx::steel::GEMMAddMMParams::alpha']]], - ['and_50',['And',['../struct_and.html',1,'And< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924aba3b7fb927f6b6c8b198a9cdc3dd9e02',1,'mlx::core::distributed::AllReduce::And'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23dab14e7d426f45ae7f029f4e00210fbae4',1,'mlx::core::BitwiseBinary::And'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a5cc3412a1f243dcb11661bca42daea93',1,'mlx::core::Reduce::And']]], - ['any_51',['any',['../group__ops.html#ga8598dd718fb05cb28535e250372d4e6f',1,'mlx::core::any(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gad37df97f253a963bece124198dbaf9ba',1,'mlx::core::any(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef',1,'mlx::core::any(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce',1,'mlx::core::any(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['apply_52',['apply',['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply()'],['../struct_transform_scale.html#a9dd329422e5b8da43486cdce17132e16',1,'TransformScale::apply()'],['../struct_max_op.html#ab3d3c3040017a13c170e7bdd1ffac46e',1,'MaxOp::apply()'],['../struct_sum_op.html#aa9563a98cbbe1b1921ade0c63ab38b4d',1,'SumOp::apply()'],['../struct_mul_op.html#a1b93d804653d92fc7e46747de9e9c756',1,'MulOp::apply()'],['../struct_sub_op.html#ad211f879a212ed0e98136217ca8e4143',1,'SubOp::apply()'],['../struct_exp_sub_op.html#a00e457a01cb38f959dfd789455e7f334',1,'ExpSubOp::apply()'],['../struct_div_op.html#a1b8df47142dc6ea15315ce3a310f9221',1,'DivOp::apply()'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const'],['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply()'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const']]], - ['apply_5fepilogue_53',['apply_epilogue',['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)']]], - ['apply_5fepilogue_5fsafe_54',['apply_epilogue_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA::apply_epilogue_safe(const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const BinaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA::apply_epilogue_safe(const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const BinaryEpilogue &epilogue_op)']]], - ['apply_5finplace_5fop_55',['apply_inplace_op',['../structmlx_1_1steel_1_1_block_loader.html#adb4ca2cc193630a779de552fa8847ddf',1,'mlx::steel::BlockLoader::apply_inplace_op()'],['../structmlx_1_1steel_1_1_block_loader_t.html#a2b136fad00dc54300e68aa6b905eff97',1,'mlx::steel::BlockLoaderT::apply_inplace_op()'],['../structmlx_1_1steel_1_1_block_loader.html#adb4ca2cc193630a779de552fa8847ddf',1,'mlx::steel::BlockLoader::apply_inplace_op()']]], - ['arange_56',['Arange',['../classmlx_1_1core_1_1_arange.html',1,'mlx::core::Arange'],['../classmlx_1_1core_1_1_arange.html#a1a70c3b0b9c67d5a9446c141c5b7c574',1,'mlx::core::Arange::Arange()']]], - ['arange_57',['arange',['../namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b',1,'mlx::core::arange()'],['../namespacemlx_1_1core_1_1metal.html#a272c36f0faf2570cbb2f36030e9a3f26',1,'mlx::core::metal::arange()'],['../metal_2kernels_2arange_8h.html#a1e5126ee6ae0164c2343230c4d87c03e',1,'arange(): arange.h'],['../group__ops.html#ga7ca088b8090b9f84f2e08345cf3f835a',1,'mlx::core::arange(double start, double stop, double step, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga4c36b841dc5cba391dad029be5a0ad98',1,'mlx::core::arange(double start, double stop, double step, StreamOrDevice s={})'],['../group__ops.html#ga8d7cf9eb15e2daf1469058907e8abc85',1,'mlx::core::arange(double start, double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga74566a14e69ba6a25f5a35e7ade5c282',1,'mlx::core::arange(double start, double stop, StreamOrDevice s={})'],['../group__ops.html#ga345aa27af3dae3646b8b4b1068e89a3e',1,'mlx::core::arange(double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaae179075d0fe23f4bd53fdf8c41f4c70',1,'mlx::core::arange(double stop, StreamOrDevice s={})'],['../group__ops.html#ga6b945f513077c2978afc1a952c884860',1,'mlx::core::arange(int start, int stop, int step, StreamOrDevice s={})'],['../group__ops.html#ga1c39fcc6eaa1c1867735c7f849d708d6',1,'mlx::core::arange(int start, int stop, StreamOrDevice s={})'],['../group__ops.html#gafe6e4580452c873cac294f16129e633f',1,'mlx::core::arange(int stop, StreamOrDevice s={})']]], - ['arange_2eh_58',['arange.h',['../common_2arange_8h.html',1,'(Global Namespace)'],['../metal_2jit_2arange_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2arange_8h.html',1,'(Global Namespace)']]], - ['arange_5fkernels_59',['arange_kernels',['../metal_2jit_2arange_8h.html#a2f49fb7bdc0a90230077fe2023e6e5c0',1,'arange.h']]], - ['arccos_60',['ArcCos',['../struct_arc_cos.html',1,'ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html',1,'mlx::core::ArcCos'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html',1,'mlx::core::detail::ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html#a66f4ee841d17923d93241b71ea5103e9',1,'mlx::core::ArcCos::ArcCos()']]], - ['arccos_61',['arccos',['../group__ops.html#ga08bec7cb10c84466487b507fc5bf9776',1,'mlx::core']]], - ['arccosh_62',['ArcCosh',['../struct_arc_cosh.html',1,'ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html',1,'mlx::core::ArcCosh'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html',1,'mlx::core::detail::ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html#a34597054db467941a2a883c653ba4d71',1,'mlx::core::ArcCosh::ArcCosh()']]], - ['arccosh_63',['arccosh',['../group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5',1,'mlx::core']]], - ['arcsin_64',['ArcSin',['../struct_arc_sin.html',1,'ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html',1,'mlx::core::ArcSin'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html',1,'mlx::core::detail::ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html#a97cb8c3d4d9d6abc627dec49a404f013',1,'mlx::core::ArcSin::ArcSin()']]], - ['arcsin_65',['arcsin',['../group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619',1,'mlx::core']]], - ['arcsinh_66',['ArcSinh',['../struct_arc_sinh.html',1,'ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html',1,'mlx::core::ArcSinh'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html',1,'mlx::core::detail::ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html#a30076b222788deeaaf9ad92d3c535f20',1,'mlx::core::ArcSinh::ArcSinh()']]], - ['arcsinh_67',['arcsinh',['../group__ops.html#gac62e2cedc49ef2c90dd8584000317450',1,'mlx::core']]], - ['arctan_68',['ArcTan',['../struct_arc_tan.html',1,'ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html',1,'mlx::core::ArcTan'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html',1,'mlx::core::detail::ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html#a3511153bbd421e89fd9294cdb3f79b44',1,'mlx::core::ArcTan::ArcTan()']]], - ['arctan_69',['arctan',['../group__ops.html#gaa041f3f070e68f4946db07516b7d092e',1,'mlx::core']]], - ['arctan2_70',['ArcTan2',['../struct_arc_tan2.html',1,'ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html',1,'mlx::core::ArcTan2'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html',1,'mlx::core::detail::ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html#aa1a4ebab9924b6bcc80df5b52ed0121a',1,'mlx::core::ArcTan2::ArcTan2()']]], - ['arctan2_71',['arctan2',['../group__ops.html#ga6caba9c92b5989123501f909cc7da354',1,'mlx::core']]], - ['arctanh_72',['ArcTanh',['../struct_arc_tanh.html',1,'ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html',1,'mlx::core::ArcTanh'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html',1,'mlx::core::detail::ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html#a17857bd0e2a3ecf1f7bf8e1a3d354358',1,'mlx::core::ArcTanh::ArcTanh()']]], - ['arctanh_73',['arctanh',['../group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e',1,'mlx::core']]], - ['argmax_74',['ArgMax',['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3acc6659315ab0001abd37cbfcbe837e7e',1,'mlx::core::ArgReduce']]], - ['argmax_75',['argmax',['../group__ops.html#gae60b0b5339b9c50b9970260faf613e83',1,'mlx::core::argmax(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gae6f6c5a840320b336fdc9687e0ed56c8',1,'mlx::core::argmax(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2efa67466510fc26ab9ea8dff30f2ba5',1,'mlx::core::argmax(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['argmin_76',['ArgMin',['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3a93a8a9221545ae9518d289d9ac4d09e9',1,'mlx::core::ArgReduce']]], - ['argmin_77',['argmin',['../group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71',1,'mlx::core::argmin(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga6bc577c5ab10cd9c848ba81321595070',1,'mlx::core::argmin(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf66dc3c77b88e4009e0678eda41eca81',1,'mlx::core::argmin(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['argpartition_78',['ArgPartition',['../classmlx_1_1core_1_1_arg_partition.html',1,'mlx::core::ArgPartition'],['../classmlx_1_1core_1_1_arg_partition.html#ab54b13dbf92351ba1ac06fd3e5a802df',1,'mlx::core::ArgPartition::ArgPartition()']]], - ['argpartition_79',['argpartition',['../group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd',1,'mlx::core::argpartition(const array &a, int kth, StreamOrDevice s={})'],['../group__ops.html#ga7b15c654c7463def57857a0e239989a3',1,'mlx::core::argpartition(const array &a, int kth, int axis, StreamOrDevice s={})']]], - ['argreduce_80',['ArgReduce',['../classmlx_1_1core_1_1_arg_reduce.html',1,'mlx::core::ArgReduce'],['../classmlx_1_1core_1_1_arg_reduce.html#aaccf8021dc24895656e25142eb65aa03',1,'mlx::core::ArgReduce::ArgReduce()']]], - ['args_81',['Args',['../namespacemlx_1_1core.html#acace1870dbbc6a0af0c054e8e71adc1f',1,'mlx::core']]], - ['argsort_82',['ArgSort',['../classmlx_1_1core_1_1_arg_sort.html',1,'mlx::core::ArgSort'],['../classmlx_1_1core_1_1_arg_sort.html#a38507a8445302a81cb44674c4a5fc0b0',1,'mlx::core::ArgSort::ArgSort()']]], - ['argsort_83',['argsort',['../group__ops.html#ga8df3b2703bf671457422894dd870cdc5',1,'mlx::core::argsort(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7878e0daa5a75f44e57b5fe948fa3ef6',1,'mlx::core::argsort(const array &a, int axis, StreamOrDevice s={})']]], - ['argument_5fencoder_84',['argument_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#a6e33e2b1287324fb4a6575e0da5e5881',1,'mlx::core::metal::Device']]], - ['arr_85',['arr',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail::arr< T >'],['../classpocketfft_1_1detail_1_1arr.html#a961a24410638b35129cd6b81850d2a42',1,'pocketfft::detail::arr::arr()'],['../classpocketfft_1_1detail_1_1arr.html#a04f832b780a4453fdf3b69bf75b182bd',1,'pocketfft::detail::arr::arr(size_t n)'],['../classpocketfft_1_1detail_1_1arr.html#a0cd8fb4a588a74d428a7349d38b477d0',1,'pocketfft::detail::arr::arr(arr &&other)']]], - ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20t0_20_3e_20_3e_86',['arr< pocketfft::detail::cmplx< T0 > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20thigh_20_3e_20_3e_87',['arr< pocketfft::detail::cmplx< Thigh > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_3c_20t0_20_3e_88',['arr< T0 >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_5finfo_89',['arr_info',['../classpocketfft_1_1detail_1_1arr__info.html',1,'pocketfft::detail::arr_info'],['../classpocketfft_1_1detail_1_1arr__info.html#a0dbddb7d86ca306159fc9ef9a453b21e',1,'pocketfft::detail::arr_info::arr_info()']]], - ['array_90',['array',['../classmlx_1_1core_1_1array.html',1,'mlx::core::array'],['../classmlx_1_1core_1_1array.html#a75fac72da3ce214fa3737df92a64b232',1,'mlx::core::array::array(T val, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a6db4b8c28c767cc16ad2785ece496dca',1,'mlx::core::array::array(const std::complex< float > &val, Dtype dtype=complex64)'],['../classmlx_1_1core_1_1array.html#abcc030a1c2434ec75ad9425751bffdc7',1,'mlx::core::array::array(It data, Shape shape, Dtype dtype=TypeToDtype< typename std::iterator_traits< It >::value_type >())'],['../classmlx_1_1core_1_1array.html#a87f170384f4fb93decf2b80ae7280f00',1,'mlx::core::array::array(std::initializer_list< T > data, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a46642301da11e3eb4312c37349fbc9d7',1,'mlx::core::array::array(std::initializer_list< float > data)'],['../classmlx_1_1core_1_1array.html#a5e1812029394bfb1a706c83611286f49',1,'mlx::core::array::array(std::initializer_list< int > data, Dtype dtype)'],['../classmlx_1_1core_1_1array.html#a89a7b0c02366ca456232d347ebb11507',1,'mlx::core::array::array(std::initializer_list< T > data, Shape shape, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a485399a6680a370cabb08470306b63d4',1,'mlx::core::array::array(allocator::Buffer data, Shape shape, Dtype dtype, Deleter deleter=allocator::free)'],['../classmlx_1_1core_1_1array.html#a297df274e2da5cb884257bbeffd6b187',1,'mlx::core::array::array(const array &other)=default'],['../classmlx_1_1core_1_1array.html#ab6cbccbba66cc54acda4390b19f0397c',1,'mlx::core::array::array(array &&other)=default'],['../classmlx_1_1core_1_1array.html#abc26528271076510822e374d1668a94b',1,'mlx::core::array::array(Shape shape, Dtype dtype, std::shared_ptr< Primitive > primitive, std::vector< array > inputs)']]], - ['array_20operations_91',['Core array operations',['../group__ops.html',1,'']]], - ['array_2eh_92',['array.h',['../array_8h.html',1,'']]], - ['array_5fequal_93',['array_equal',['../group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645',1,'mlx::core::array_equal(const array &a, const array &b, bool equal_nan, StreamOrDevice s={})'],['../group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14',1,'mlx::core::array_equal(const array &a, const array &b, StreamOrDevice s={})']]], - ['arrayiterator_94',['ArrayIterator',['../structmlx_1_1core_1_1array_1_1_array_iterator.html',1,'mlx::core::array::ArrayIterator'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ad3afcb24c6db7642bbc06835f7f3e27a',1,'mlx::core::array::ArrayIterator::ArrayIterator()']]], - ['as_5foffset_95',['As_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a138ed1bbad2ca88d3a3c7d162cd36562',1,'mlx::steel::BlockMMA']]], - ['as_5fstrided_96',['as_strided',['../group__ops.html#ga6085b03f2662ef2a61de523fd609f3bf',1,'mlx::core']]], - ['asin_97',['asin',['../namespacemetal.html#a16e843194df3fd136404bf80ba5ac95c',1,'metal::asin()'],['../namespacemetal_1_1fast.html#a769455a283da99654b6e42c3acf13eb1',1,'metal::fast::asin()'],['../namespacemetal_1_1precise.html#adc7b8b6e12e320cb32030f728dcbf438',1,'metal::precise::asin()']]], - ['asinh_98',['asinh',['../namespacemetal.html#abcc3251866930cfe880f89e7473d0e63',1,'metal::asinh()'],['../namespacemetal_1_1fast.html#a4367034b7b3e14310803bb2be975a556',1,'metal::fast::asinh()'],['../namespacemetal_1_1precise.html#aaad1cdde6687c8011fbc5fda1bb13424',1,'metal::precise::asinh()']]], - ['asstrided_99',['AsStrided',['../classmlx_1_1core_1_1_as_strided.html',1,'mlx::core::AsStrided'],['../classmlx_1_1core_1_1_as_strided.html#aee21aadc21343fd15aacb8f2f8ac3761',1,'mlx::core::AsStrided::AsStrided()']]], - ['astype_100',['AsType',['../classmlx_1_1core_1_1_as_type.html',1,'mlx::core::AsType'],['../classmlx_1_1core_1_1_as_type.html#a8c3241d402a8977bb4db037e225f5b47',1,'mlx::core::AsType::AsType()']]], - ['astype_101',['astype',['../group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62',1,'mlx::core']]], - ['async_5feval_102',['async_eval',['../namespacemlx_1_1core.html#a15dda19aa7fa1fc5fca35df5cf963297',1,'mlx::core::async_eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#abf8dae3b87a03a409711691cd9c097b6',1,'mlx::core::async_eval(Arrays &&... outputs)']]], - ['atan_103',['atan',['../namespacemetal.html#a80a771553d9a0012b93620d19c48b00f',1,'metal::atan()'],['../namespacemetal_1_1fast.html#a769503b4b7f89071d0983258c5a3ac5a',1,'metal::fast::atan()'],['../namespacemetal_1_1precise.html#aaaf4b5f4786a912089bbf0ae7619a6be',1,'metal::precise::atan()']]], - ['atan2_104',['atan2',['../namespacemetal.html#a1d430793eaa38ccf0d07145e3fcd1e61',1,'metal::atan2()'],['../namespacemetal_1_1fast.html#a00e687ea46f5affe26e6aef8fd62b89a',1,'metal::fast::atan2()'],['../namespacemetal_1_1precise.html#a6f161b049cc6884f87b09b33c2d1cd7f',1,'metal::precise::atan2()']]], - ['atanh_105',['atanh',['../namespacemetal.html#a57116427997ba71dd3863bfb15de33bf',1,'metal::atanh()'],['../namespacemetal_1_1fast.html#af24608fc605db9a14427d37c36dc1c53',1,'metal::fast::atanh()'],['../namespacemetal_1_1precise.html#a902994837653b90c47f4285673e712c4',1,'metal::precise::atanh()']]], - ['atile_106',['Atile',['../structmlx_1_1steel_1_1_block_m_m_a.html#a47e614120c650f7479db79f23a0df586',1,'mlx::steel::BlockMMA']]], - ['atleast_5f1d_107',['atleast_1d',['../group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396',1,'mlx::core::atleast_1d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5',1,'mlx::core::atleast_1d(const std::vector< array > &a, StreamOrDevice s={})']]], - ['atleast_5f2d_108',['atleast_2d',['../group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087',1,'mlx::core::atleast_2d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga9950299a80c2562f13448758f856d1f5',1,'mlx::core::atleast_2d(const std::vector< array > &a, StreamOrDevice s={})']]], - ['atleast_5f3d_109',['atleast_3d',['../group__ops.html#ga4afd919601e67782ff964465919956a0',1,'mlx::core::atleast_3d(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaffdf742ad79440a60dda40062a8074fe',1,'mlx::core::atleast_3d(const std::vector< array > &a, StreamOrDevice s={})']]], - ['atomic_2eh_110',['atomic.h',['../atomic_8h.html',1,'']]], - ['atomic_5fupdate_111',['atomic_update',['../struct_none.html#aecbce7c97e8b1d5dc4afd2e788c24e06',1,'None']]], - ['attach_5fevent_112',['attach_event',['../classmlx_1_1core_1_1array.html#a000c3cfe13cb378bf0523b62816190da',1,'mlx::core::array']]], - ['attention_113',['attention',['../steel__attention_8h.html#a5423b2a414f5e3c14166d568dedfbd33',1,'steel_attention.h']]], - ['attn_2eh_114',['attn.h',['../attn_8h.html',1,'']]], - ['attnparams_115',['AttnParams',['../structmlx_1_1steel_1_1_attn_params.html',1,'mlx::steel']]], - ['available_116',['available',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078a308bd3e5bf976888b120dd36d0c2d2ae',1,'mlx::core::array']]] + ['abs_5',['abs',['../namespacemlx_1_1core_1_1simd.html#a4f3cc8b2493586e83fd65640df3b60ad',1,'mlx::core::simd::abs(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a1958f026f26f313d17155ac87ea6eca3',1,'mlx::core::simd::abs(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a7696a0628a1c6ccb293ebd6f2328ea48',1,'mlx::core::simd::abs(Simd< float16_t, N > a)'],['../namespacemetal.html#a87c5122c60f9a12afceb9925a5b78ffb',1,'metal::abs()'],['../namespacemetal_1_1fast.html#a90d2973f71f83180e7f02e38d11c7a8f',1,'metal::fast::abs()'],['../namespacemetal_1_1precise.html#a99f2b2746e813b9ca7b4249afbaf2a14',1,'metal::precise::abs()'],['../group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920',1,'mlx::core::abs()']]], + ['accelerate_5ffp16_5fsimd_2eh_6',['accelerate_fp16_simd.h',['../accelerate__fp16__simd_8h.html',1,'']]], + ['accelerate_5fsimd_2eh_7',['accelerate_simd.h',['../accelerate__simd_8h.html',1,'']]], + ['accum_5ftype_8',['accum_type',['../structmlx_1_1steel_1_1_accum_helper.html#ae52abf69e7ba6af1a73d65d57182ed26',1,'mlx::steel::AccumHelper::accum_type'],['../structmlx_1_1steel_1_1_accum_helper.html#ae52abf69e7ba6af1a73d65d57182ed26',1,'mlx::steel::AccumHelper::accum_type']]], + ['accumhelper_9',['AccumHelper',['../structmlx_1_1steel_1_1_accum_helper.html',1,'mlx::steel']]], + ['acos_10',['acos',['../namespacemlx_1_1core_1_1simd.html#ae623449dfa7aab3031aa2f14c1b10a2d',1,'mlx::core::simd::acos(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a85999467c83b07e4fa5f093f7ddf19e1',1,'mlx::core::simd::acos(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a3d4f9d08d1902e3d62c6f63d39329dbd',1,'mlx::core::simd::acos(Simd< T, 1 > in)'],['../namespacemetal.html#ad4537748b3c832b6569ff7ccb209fcb2',1,'metal::acos()'],['../namespacemetal_1_1fast.html#a805ce5c3a94b618b7349d70bbb82f0b2',1,'metal::fast::acos()'],['../namespacemetal_1_1precise.html#a8a2bcc89fc0b7e74f0453f82f89a8604',1,'metal::precise::acos()']]], + ['acosh_11',['acosh',['../namespacemlx_1_1core_1_1simd.html#a4f8a64e7742fcd8f759f723a36a7c826',1,'mlx::core::simd::acosh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a90092f3826ad3be4b2b1785f7ff4a86b',1,'mlx::core::simd::acosh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a51071c8104494b5bd8097990da3bf943',1,'mlx::core::simd::acosh(Simd< T, 1 > in)'],['../namespacemetal.html#a2d0efb92b7f61eff342d776bd6c5f3a0',1,'metal::acosh()'],['../namespacemetal_1_1fast.html#afb656fc3406649a238b6f1e0509de751',1,'metal::fast::acosh()'],['../namespacemetal_1_1precise.html#a1f489fabffab969b8677b56bb1136067',1,'metal::precise::acosh()']]], + ['add_12',['Add',['../struct_add.html',1,'Add'],['../classmlx_1_1core_1_1_add.html',1,'mlx::core::Add'],['../structmlx_1_1core_1_1detail_1_1_add.html',1,'mlx::core::detail::Add'],['../classmlx_1_1core_1_1_add.html#ae3fd5483f3454eac3df256e3f5f3cdae',1,'mlx::core::Add::Add()']]], + ['add_13',['add',['../group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7',1,'mlx::core']]], + ['add_5fhalf_5fbinops_14',['ADD_HALF_BINOPS',['../half__types_8h.html#a6bc906918877a7084068a9f0ed571dca',1,'ADD_HALF_BINOPS: half_types.h'],['../half__types_8h.html#a6bc906918877a7084068a9f0ed571dca',1,'ADD_HALF_BINOPS: half_types.h']]], + ['add_5ftemporaries_15',['add_temporaries',['../classmlx_1_1core_1_1metal_1_1_device.html#a72ad17c96fc6ce825bc77f0bed657901',1,'mlx::core::metal::Device']]], + ['add_5ftemporary_16',['add_temporary',['../classmlx_1_1core_1_1metal_1_1_device.html#acb90010af0cffe27fd8cc6c253d3a576',1,'mlx::core::metal::Device']]], + ['add_5fvec_17',['add_vec',['../structpocketfft_1_1detail_1_1add__vec.html',1,'pocketfft::detail']]], + ['add_5fvec_3c_20cmplx_3c_20t_20_3e_20_3e_18',['add_vec< cmplx< T > >',['../structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html',1,'pocketfft::detail']]], + ['add_5fvec_5ft_19',['add_vec_t',['../namespacepocketfft_1_1detail.html#a421aa74fbee775a96463246f72b144d6',1,'pocketfft::detail']]], + ['addmm_20',['AddMM',['../classmlx_1_1core_1_1_add_m_m.html',1,'mlx::core::AddMM'],['../classmlx_1_1core_1_1_add_m_m.html#a8ae4372b3f96e72e8a5a06d59de8a550',1,'mlx::core::AddMM::AddMM()']]], + ['addmm_21',['addmm',['../group__ops.html#ga82a53e083205a965387b3c3e2463244a',1,'mlx::core']]], + ['adj_5fimplicit_5fm_22',['adj_implicit_m',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a5bfca3bc43055013d28430cb1f023756',1,'mlx::steel::Conv2DGeneralJumpParams']]], + ['adj_5fout_5fh_23',['adj_out_h',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a879cc9757f59605a87d936ec4156040d',1,'mlx::steel::Conv2DGeneralJumpParams']]], + ['adj_5fout_5fhw_24',['adj_out_hw',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#aed0ffd63fbc85fd5d5c4cc7b43f68363',1,'mlx::steel::Conv2DGeneralJumpParams']]], + ['adj_5fout_5fw_25',['adj_out_w',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#ab971bf879079895189331fbeaf33c211',1,'mlx::steel::Conv2DGeneralJumpParams']]], + ['adjust_5fmatrix_5foffsets_26',['adjust_matrix_offsets',['../quantized_8h.html#a3e448f8f23c12ffc83bff64ae66bbc66',1,'adjust_matrix_offsets(const device T *&x, const device uint32_t *&w, const device T *&scales, const device T *&biases, device T *&y, int output_stride, const constant int &x_batch_ndims, const constant int *x_shape, const constant int64_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant int64_t *w_strides, const constant int64_t *s_strides, const constant int64_t *b_strides, uint3 tid): quantized.h'],['../quantized_8h.html#acb9a7a0653084295657b630d49c71707',1,'adjust_matrix_offsets(const device T *&x, const device uint32_t *&w, const device T *&scales, const device T *&biases, const device uint32_t *lhs_indices, const device uint32_t *rhs_indices, device T *&y, int output_stride, const constant int &batch_ndims, const constant int *batch_shape, const constant int64_t *lhs_strides, const constant int64_t *rhs_strides, const constant int &x_batch_ndims, const constant int *x_shape, const constant int64_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant int64_t *w_strides, const constant int64_t *s_strides, const constant int64_t *b_strides, uint3 tid): quantized.h']]], + ['advance_27',['advance',['../classpocketfft_1_1detail_1_1multi__iter.html#a5ddcc0666125b3cb6c0d62b30befdd2c',1,'pocketfft::detail::multi_iter::advance()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a73a9ecd3008d2bd35aaa00bf9fac074f',1,'pocketfft::detail::simple_iter::advance()'],['../classpocketfft_1_1detail_1_1rev__iter.html#ad1918c84ae963188afc7599629b29686',1,'pocketfft::detail::rev_iter::advance()']]], + ['affine_5fdequantize_28',['affine_dequantize',['../quantized_8h.html#a6076203615038eb06816158f7b3869c6',1,'affine_dequantize(): quantized.h'],['../namespacemlx_1_1core_1_1fast.html#a12c7ef41409d6fb378008e67b6fab328',1,'mlx::core::fast::affine_dequantize()']]], + ['affine_5fquantize_29',['affine_quantize',['../quantized_8h.html#a47bcf4a14566e01e14bd3c155811db59',1,'affine_quantize(): quantized.h'],['../namespacemlx_1_1core_1_1fast.html#aa4b5f6886b2288cb6dfdd8598579f080',1,'mlx::core::fast::affine_quantize(const array &w, int group_size=64, int bits=4, StreamOrDevice s={})']]], + ['affinequantize_30',['AffineQuantize',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html',1,'mlx::core::fast::AffineQuantize'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a84d5fa9e8c3de407fbcc5f38d2ed1473',1,'mlx::core::fast::AffineQuantize::AffineQuantize()']]], + ['align_5fk_31',['align_K',['../steel__attention_8h.html#a8bdd2cecf97aa5b033152b1d0f0d2416',1,'align_K: steel_attention.h'],['../steel__gemm__fused_8h.html#a8bdd2cecf97aa5b033152b1d0f0d2416',1,'align_K: steel_gemm_fused.h']]], + ['align_5fm_32',['align_M',['../steel__gemm__fused_8h.html#a55af226dc74b0026b7d4b865142a6d21',1,'steel_gemm_fused.h']]], + ['align_5fn_33',['align_N',['../steel__gemm__fused_8h.html#aa3b267252df2dcbfdde8c5f174d27036',1,'steel_gemm_fused.h']]], + ['align_5fq_34',['align_Q',['../steel__attention_8h.html#a171fdea1b23976453f5dc5e6b3161982',1,'steel_attention.h']]], + ['aligned_5falloc_35',['aligned_alloc',['../namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76',1,'pocketfft::detail']]], + ['aligned_5fallocator_36',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html',1,'pocketfft::detail::threading::aligned_allocator< T >'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator(const aligned_allocator< U > &)'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator()=default']]], + ['aligned_5fdealloc_37',['aligned_dealloc',['../namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8',1,'pocketfft::detail']]], + ['all_38',['all',['../namespacemlx_1_1core_1_1simd.html#a5109118acb6766855878b9e8a56b156a',1,'mlx::core::simd::all(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#a4ba3690489c2bf861e22e1175255438c',1,'mlx::core::simd::all(Simd< T, 1 > x)'],['../group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68',1,'mlx::core::all(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4',1,'mlx::core::all(const array &a, StreamOrDevice s={})'],['../group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59',1,'mlx::core::all(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc',1,'mlx::core::all(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['all_5fgather_39',['all_gather',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a04bb1df23abe5b1f3fa0126375c6cea4',1,'mlx::core::distributed::detail::GroupImpl::all_gather()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04',1,'mlx::core::distributed::detail::all_gather()'],['../namespacemlx_1_1core_1_1distributed.html#a82ef5e8cc7ac62cd228e51b1c1b77cb7',1,'mlx::core::distributed::all_gather()']]], + ['all_5freduce_40',['all_reduce',['../reduce__all_8h.html#a9086a585eda5a887160ee24baae0a7b8',1,'reduce_all.h']]], + ['all_5freduce_5fdispatch_41',['all_reduce_dispatch',['../namespacemlx_1_1core.html#a3ab0fd997d9a35782106ff083a72e098',1,'mlx::core']]], + ['all_5fsum_42',['all_sum',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ae163a6f444c6cc8820288b20f294e483',1,'mlx::core::distributed::detail::GroupImpl::all_sum()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aa1d225b25f7b6426c48c5e35860ee960',1,'mlx::core::distributed::detail::all_sum()'],['../namespacemlx_1_1core_1_1distributed.html#a67ccb1a5445fc6f5db49dd36a15e5980',1,'mlx::core::distributed::all_sum()']]], + ['allclose_43',['allclose',['../group__ops.html#gaf0cd4257de7542daf9faf5e605e31020',1,'mlx::core']]], + ['allgather_44',['AllGather',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html',1,'mlx::core::distributed::AllGather'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#af4b10a5b61f160fb64353057c185b661',1,'mlx::core::distributed::AllGather::AllGather()']]], + ['alloc_5ftmp_45',['alloc_tmp',['../namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, size_t axsize, size_t elemsize)'],['../namespacepocketfft_1_1detail.html#a13832735696303b9559c4663631d5475',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, const shape_t &axes, size_t elemsize)']]], + ['allocate_46',['allocate',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a4f785747b898980756af9e5515363826',1,'pocketfft::detail::threading::aligned_allocator']]], + ['allocator_47',['Allocator',['../classmlx_1_1core_1_1allocator_1_1_allocator.html',1,'mlx::core::allocator::Allocator'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a5803678a418fef687fc65fa9d5c37b65',1,'mlx::core::allocator::Allocator::Allocator()=default'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#aa05c081ce80dc036f9d3dd8c195259d2',1,'mlx::core::allocator::Allocator::Allocator(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a8e8ce346a16cf0c62847bed9289f9959',1,'mlx::core::allocator::Allocator::Allocator(Allocator &&other)=delete']]], + ['allocator_48',['allocator',['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#abf84c726a37df68345589b897b2e35f0',1,'mlx::core::allocator::CommonAllocator::allocator()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#afa1c5a725309caff163c492b5b84491e',1,'mlx::core::metal::MetalAllocator::allocator()'],['../namespacemlx_1_1core_1_1allocator.html#aa23e2f20a336d0b159c097087194634e',1,'mlx::core::allocator::allocator()'],['../namespacemlx_1_1core_1_1metal.html#a74b3558bd518aecde6b14b0ba5e1a0d5',1,'mlx::core::metal::allocator()']]], + ['allocator_2eh_49',['allocator.h',['../allocator_8h.html',1,'(Global Namespace)'],['../backend_2metal_2allocator_8h.html',1,'(Global Namespace)']]], + ['allreduce_50',['AllReduce',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html',1,'mlx::core::distributed::AllReduce'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a2d1ea56cbf72a316680ea90aa6da1c2d',1,'mlx::core::distributed::AllReduce::AllReduce()']]], + ['alpha_51',['alpha',['../structmlx_1_1steel_1_1_transform_axpby.html#ab3223b49c6b3b7f89eba91aeaff9dcff',1,'mlx::steel::TransformAxpby::alpha'],['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#af8693d96512eff3e125d33d203920710',1,'mlx::steel::GEMMAddMMParams::alpha']]], + ['and_52',['And',['../struct_and.html',1,'And< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924aba3b7fb927f6b6c8b198a9cdc3dd9e02',1,'mlx::core::distributed::AllReduce::And'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23dab14e7d426f45ae7f029f4e00210fbae4',1,'mlx::core::BitwiseBinary::And'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a5cc3412a1f243dcb11661bca42daea93',1,'mlx::core::Reduce::And']]], + ['any_53',['any',['../namespacemlx_1_1core_1_1simd.html#a9d968537ad5ef18630f5afce8453b30e',1,'mlx::core::simd::any(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#ae3b138b4bbcee0ca70b58a3e2ebd818c',1,'mlx::core::simd::any(Simd< T, 1 > x)'],['../group__ops.html#ga8598dd718fb05cb28535e250372d4e6f',1,'mlx::core::any(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gad37df97f253a963bece124198dbaf9ba',1,'mlx::core::any(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef',1,'mlx::core::any(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce',1,'mlx::core::any(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['apply_54',['apply',['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply()'],['../struct_transform_scale.html#a9dd329422e5b8da43486cdce17132e16',1,'TransformScale::apply()'],['../struct_max_op.html#ab3d3c3040017a13c170e7bdd1ffac46e',1,'MaxOp::apply()'],['../struct_sum_op.html#aa9563a98cbbe1b1921ade0c63ab38b4d',1,'SumOp::apply()'],['../struct_mul_op.html#a1b93d804653d92fc7e46747de9e9c756',1,'MulOp::apply()'],['../struct_sub_op.html#ad211f879a212ed0e98136217ca8e4143',1,'SubOp::apply()'],['../struct_exp_sub_op.html#a00e457a01cb38f959dfd789455e7f334',1,'ExpSubOp::apply()'],['../struct_div_op.html#a1b8df47142dc6ea15315ce3a310f9221',1,'DivOp::apply()'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const'],['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply()'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const']]], + ['apply_5fepilogue_55',['apply_epilogue',['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)']]], + ['apply_5fepilogue_5fsafe_56',['apply_epilogue_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA::apply_epilogue_safe(const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const BinaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA::apply_epilogue_safe(const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const BinaryEpilogue &epilogue_op)']]], + ['apply_5finplace_5fop_57',['apply_inplace_op',['../structmlx_1_1steel_1_1_block_loader.html#adb4ca2cc193630a779de552fa8847ddf',1,'mlx::steel::BlockLoader::apply_inplace_op()'],['../structmlx_1_1steel_1_1_block_loader_t.html#a2b136fad00dc54300e68aa6b905eff97',1,'mlx::steel::BlockLoaderT::apply_inplace_op()'],['../structmlx_1_1steel_1_1_block_loader.html#adb4ca2cc193630a779de552fa8847ddf',1,'mlx::steel::BlockLoader::apply_inplace_op()']]], + ['arange_58',['Arange',['../classmlx_1_1core_1_1_arange.html',1,'mlx::core::Arange'],['../classmlx_1_1core_1_1_arange.html#a1a70c3b0b9c67d5a9446c141c5b7c574',1,'mlx::core::Arange::Arange()']]], + ['arange_59',['arange',['../namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b',1,'mlx::core::arange()'],['../namespacemlx_1_1core_1_1metal.html#a272c36f0faf2570cbb2f36030e9a3f26',1,'mlx::core::metal::arange()'],['../metal_2kernels_2arange_8h.html#a1e5126ee6ae0164c2343230c4d87c03e',1,'arange(): arange.h'],['../group__ops.html#ga7ca088b8090b9f84f2e08345cf3f835a',1,'mlx::core::arange(double start, double stop, double step, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga4c36b841dc5cba391dad029be5a0ad98',1,'mlx::core::arange(double start, double stop, double step, StreamOrDevice s={})'],['../group__ops.html#ga8d7cf9eb15e2daf1469058907e8abc85',1,'mlx::core::arange(double start, double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga74566a14e69ba6a25f5a35e7ade5c282',1,'mlx::core::arange(double start, double stop, StreamOrDevice s={})'],['../group__ops.html#ga345aa27af3dae3646b8b4b1068e89a3e',1,'mlx::core::arange(double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaae179075d0fe23f4bd53fdf8c41f4c70',1,'mlx::core::arange(double stop, StreamOrDevice s={})'],['../group__ops.html#ga6b945f513077c2978afc1a952c884860',1,'mlx::core::arange(int start, int stop, int step, StreamOrDevice s={})'],['../group__ops.html#ga1c39fcc6eaa1c1867735c7f849d708d6',1,'mlx::core::arange(int start, int stop, StreamOrDevice s={})'],['../group__ops.html#gafe6e4580452c873cac294f16129e633f',1,'mlx::core::arange(int stop, StreamOrDevice s={})']]], + ['arange_2eh_60',['arange.h',['../cpu_2arange_8h.html',1,'(Global Namespace)'],['../metal_2jit_2arange_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2arange_8h.html',1,'(Global Namespace)']]], + ['arange_5fkernels_61',['arange_kernels',['../metal_2jit_2arange_8h.html#a2f49fb7bdc0a90230077fe2023e6e5c0',1,'arange.h']]], + ['arccos_62',['ArcCos',['../struct_arc_cos.html',1,'ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html',1,'mlx::core::ArcCos'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html',1,'mlx::core::detail::ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html#a66f4ee841d17923d93241b71ea5103e9',1,'mlx::core::ArcCos::ArcCos()']]], + ['arccos_63',['arccos',['../group__ops.html#ga08bec7cb10c84466487b507fc5bf9776',1,'mlx::core']]], + ['arccosh_64',['ArcCosh',['../struct_arc_cosh.html',1,'ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html',1,'mlx::core::ArcCosh'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html',1,'mlx::core::detail::ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html#a34597054db467941a2a883c653ba4d71',1,'mlx::core::ArcCosh::ArcCosh()']]], + ['arccosh_65',['arccosh',['../group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5',1,'mlx::core']]], + ['arcsin_66',['ArcSin',['../struct_arc_sin.html',1,'ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html',1,'mlx::core::ArcSin'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html',1,'mlx::core::detail::ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html#a97cb8c3d4d9d6abc627dec49a404f013',1,'mlx::core::ArcSin::ArcSin()']]], + ['arcsin_67',['arcsin',['../group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619',1,'mlx::core']]], + ['arcsinh_68',['ArcSinh',['../struct_arc_sinh.html',1,'ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html',1,'mlx::core::ArcSinh'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html',1,'mlx::core::detail::ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html#a30076b222788deeaaf9ad92d3c535f20',1,'mlx::core::ArcSinh::ArcSinh()']]], + ['arcsinh_69',['arcsinh',['../group__ops.html#gac62e2cedc49ef2c90dd8584000317450',1,'mlx::core']]], + ['arctan_70',['ArcTan',['../struct_arc_tan.html',1,'ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html',1,'mlx::core::ArcTan'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html',1,'mlx::core::detail::ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html#a3511153bbd421e89fd9294cdb3f79b44',1,'mlx::core::ArcTan::ArcTan()']]], + ['arctan_71',['arctan',['../group__ops.html#gaa041f3f070e68f4946db07516b7d092e',1,'mlx::core']]], + ['arctan2_72',['ArcTan2',['../struct_arc_tan2.html',1,'ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html',1,'mlx::core::ArcTan2'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html',1,'mlx::core::detail::ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html#aa1a4ebab9924b6bcc80df5b52ed0121a',1,'mlx::core::ArcTan2::ArcTan2()']]], + ['arctan2_73',['arctan2',['../group__ops.html#ga6caba9c92b5989123501f909cc7da354',1,'mlx::core']]], + ['arctanh_74',['ArcTanh',['../struct_arc_tanh.html',1,'ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html',1,'mlx::core::ArcTanh'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html',1,'mlx::core::detail::ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html#a17857bd0e2a3ecf1f7bf8e1a3d354358',1,'mlx::core::ArcTanh::ArcTanh()']]], + ['arctanh_75',['arctanh',['../group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e',1,'mlx::core']]], + ['argmax_76',['ArgMax',['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3acc6659315ab0001abd37cbfcbe837e7e',1,'mlx::core::ArgReduce']]], + ['argmax_77',['argmax',['../group__ops.html#gae60b0b5339b9c50b9970260faf613e83',1,'mlx::core::argmax(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gae6f6c5a840320b336fdc9687e0ed56c8',1,'mlx::core::argmax(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2efa67466510fc26ab9ea8dff30f2ba5',1,'mlx::core::argmax(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['argmin_78',['ArgMin',['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3a93a8a9221545ae9518d289d9ac4d09e9',1,'mlx::core::ArgReduce']]], + ['argmin_79',['argmin',['../group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71',1,'mlx::core::argmin(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga6bc577c5ab10cd9c848ba81321595070',1,'mlx::core::argmin(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf66dc3c77b88e4009e0678eda41eca81',1,'mlx::core::argmin(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['argpartition_80',['ArgPartition',['../classmlx_1_1core_1_1_arg_partition.html',1,'mlx::core::ArgPartition'],['../classmlx_1_1core_1_1_arg_partition.html#ab54b13dbf92351ba1ac06fd3e5a802df',1,'mlx::core::ArgPartition::ArgPartition()']]], + ['argpartition_81',['argpartition',['../group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd',1,'mlx::core::argpartition(const array &a, int kth, StreamOrDevice s={})'],['../group__ops.html#ga7b15c654c7463def57857a0e239989a3',1,'mlx::core::argpartition(const array &a, int kth, int axis, StreamOrDevice s={})']]], + ['argreduce_82',['ArgReduce',['../classmlx_1_1core_1_1_arg_reduce.html',1,'mlx::core::ArgReduce'],['../classmlx_1_1core_1_1_arg_reduce.html#aaccf8021dc24895656e25142eb65aa03',1,'mlx::core::ArgReduce::ArgReduce()']]], + ['args_83',['Args',['../namespacemlx_1_1core.html#acace1870dbbc6a0af0c054e8e71adc1f',1,'mlx::core']]], + ['argsort_84',['ArgSort',['../classmlx_1_1core_1_1_arg_sort.html',1,'mlx::core::ArgSort'],['../classmlx_1_1core_1_1_arg_sort.html#a38507a8445302a81cb44674c4a5fc0b0',1,'mlx::core::ArgSort::ArgSort()']]], + ['argsort_85',['argsort',['../group__ops.html#ga8df3b2703bf671457422894dd870cdc5',1,'mlx::core::argsort(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7878e0daa5a75f44e57b5fe948fa3ef6',1,'mlx::core::argsort(const array &a, int axis, StreamOrDevice s={})']]], + ['argument_5fencoder_86',['argument_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#a6e33e2b1287324fb4a6575e0da5e5881',1,'mlx::core::metal::Device']]], + ['arr_87',['arr',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail::arr< T >'],['../classpocketfft_1_1detail_1_1arr.html#a961a24410638b35129cd6b81850d2a42',1,'pocketfft::detail::arr::arr()'],['../classpocketfft_1_1detail_1_1arr.html#a04f832b780a4453fdf3b69bf75b182bd',1,'pocketfft::detail::arr::arr(size_t n)'],['../classpocketfft_1_1detail_1_1arr.html#a0cd8fb4a588a74d428a7349d38b477d0',1,'pocketfft::detail::arr::arr(arr &&other)']]], + ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20t0_20_3e_20_3e_88',['arr< pocketfft::detail::cmplx< T0 > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20thigh_20_3e_20_3e_89',['arr< pocketfft::detail::cmplx< Thigh > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_3c_20t0_20_3e_90',['arr< T0 >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_5finfo_91',['arr_info',['../classpocketfft_1_1detail_1_1arr__info.html',1,'pocketfft::detail::arr_info'],['../classpocketfft_1_1detail_1_1arr__info.html#a0dbddb7d86ca306159fc9ef9a453b21e',1,'pocketfft::detail::arr_info::arr_info()']]], + ['array_92',['array',['../classmlx_1_1core_1_1array.html',1,'mlx::core::array'],['../classmlx_1_1core_1_1array.html#a75fac72da3ce214fa3737df92a64b232',1,'mlx::core::array::array(T val, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a6db4b8c28c767cc16ad2785ece496dca',1,'mlx::core::array::array(const std::complex< float > &val, Dtype dtype=complex64)'],['../classmlx_1_1core_1_1array.html#abcc030a1c2434ec75ad9425751bffdc7',1,'mlx::core::array::array(It data, Shape shape, Dtype dtype=TypeToDtype< typename std::iterator_traits< It >::value_type >())'],['../classmlx_1_1core_1_1array.html#a87f170384f4fb93decf2b80ae7280f00',1,'mlx::core::array::array(std::initializer_list< T > data, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a46642301da11e3eb4312c37349fbc9d7',1,'mlx::core::array::array(std::initializer_list< float > data)'],['../classmlx_1_1core_1_1array.html#a5e1812029394bfb1a706c83611286f49',1,'mlx::core::array::array(std::initializer_list< int > data, Dtype dtype)'],['../classmlx_1_1core_1_1array.html#a89a7b0c02366ca456232d347ebb11507',1,'mlx::core::array::array(std::initializer_list< T > data, Shape shape, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a485399a6680a370cabb08470306b63d4',1,'mlx::core::array::array(allocator::Buffer data, Shape shape, Dtype dtype, Deleter deleter=allocator::free)'],['../classmlx_1_1core_1_1array.html#a297df274e2da5cb884257bbeffd6b187',1,'mlx::core::array::array(const array &other)=default'],['../classmlx_1_1core_1_1array.html#ab6cbccbba66cc54acda4390b19f0397c',1,'mlx::core::array::array(array &&other)=default'],['../classmlx_1_1core_1_1array.html#abc26528271076510822e374d1668a94b',1,'mlx::core::array::array(Shape shape, Dtype dtype, std::shared_ptr< Primitive > primitive, std::vector< array > inputs)']]], + ['array_20operations_93',['Core array operations',['../group__ops.html',1,'']]], + ['array_2eh_94',['array.h',['../array_8h.html',1,'']]], + ['array_5fequal_95',['array_equal',['../group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645',1,'mlx::core::array_equal(const array &a, const array &b, bool equal_nan, StreamOrDevice s={})'],['../group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14',1,'mlx::core::array_equal(const array &a, const array &b, StreamOrDevice s={})']]], + ['arrayiterator_96',['ArrayIterator',['../structmlx_1_1core_1_1array_1_1_array_iterator.html',1,'mlx::core::array::ArrayIterator'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ad3afcb24c6db7642bbc06835f7f3e27a',1,'mlx::core::array::ArrayIterator::ArrayIterator()']]], + ['as_5foffset_97',['As_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a138ed1bbad2ca88d3a3c7d162cd36562',1,'mlx::steel::BlockMMA']]], + ['as_5fstrided_98',['as_strided',['../group__ops.html#ga6085b03f2662ef2a61de523fd609f3bf',1,'mlx::core']]], + ['asin_99',['asin',['../namespacemlx_1_1core_1_1simd.html#a9c7723fc49137394fa817136a7ffb50f',1,'mlx::core::simd::asin(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a069963ffb15f06d1c48258054750dadf',1,'mlx::core::simd::asin(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a75349994f899aecb68553c2247580163',1,'mlx::core::simd::asin(Simd< T, 1 > in)'],['../namespacemetal.html#a16e843194df3fd136404bf80ba5ac95c',1,'metal::asin()'],['../namespacemetal_1_1fast.html#a769455a283da99654b6e42c3acf13eb1',1,'metal::fast::asin()'],['../namespacemetal_1_1precise.html#adc7b8b6e12e320cb32030f728dcbf438',1,'metal::precise::asin()']]], + ['asinh_100',['asinh',['../namespacemlx_1_1core_1_1simd.html#aaa76bdf1db09261d84da51d394837f5d',1,'mlx::core::simd::asinh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a7b47a5f370e8e59e1debfa5405e13266',1,'mlx::core::simd::asinh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a9323e370f6740651ebfd51367985d0e2',1,'mlx::core::simd::asinh(Simd< T, 1 > in)'],['../namespacemetal.html#abcc3251866930cfe880f89e7473d0e63',1,'metal::asinh()'],['../namespacemetal_1_1fast.html#a4367034b7b3e14310803bb2be975a556',1,'metal::fast::asinh()'],['../namespacemetal_1_1precise.html#aaad1cdde6687c8011fbc5fda1bb13424',1,'metal::precise::asinh()']]], + ['asstrided_101',['AsStrided',['../classmlx_1_1core_1_1_as_strided.html',1,'mlx::core::AsStrided'],['../classmlx_1_1core_1_1_as_strided.html#aee21aadc21343fd15aacb8f2f8ac3761',1,'mlx::core::AsStrided::AsStrided()']]], + ['astype_102',['AsType',['../classmlx_1_1core_1_1_as_type.html',1,'mlx::core::AsType'],['../classmlx_1_1core_1_1_as_type.html#a8c3241d402a8977bb4db037e225f5b47',1,'mlx::core::AsType::AsType()']]], + ['astype_103',['astype',['../group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62',1,'mlx::core']]], + ['async_5feval_104',['async_eval',['../namespacemlx_1_1core.html#a15dda19aa7fa1fc5fca35df5cf963297',1,'mlx::core::async_eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#abf8dae3b87a03a409711691cd9c097b6',1,'mlx::core::async_eval(Arrays &&... outputs)']]], + ['atan_105',['atan',['../namespacemlx_1_1core_1_1simd.html#aa7550a1210e50c996d0db84034b8a22e',1,'mlx::core::simd::atan(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a99e84cece5722fb764844a2badc5426b',1,'mlx::core::simd::atan(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a421845a6f68f88c58f520d2c1fa15914',1,'mlx::core::simd::atan(Simd< T, 1 > in)'],['../namespacemetal.html#a80a771553d9a0012b93620d19c48b00f',1,'metal::atan()'],['../namespacemetal_1_1fast.html#a769503b4b7f89071d0983258c5a3ac5a',1,'metal::fast::atan()'],['../namespacemetal_1_1precise.html#aaaf4b5f4786a912089bbf0ae7619a6be',1,'metal::precise::atan()']]], + ['atan2_106',['atan2',['../namespacemlx_1_1core_1_1simd.html#a23dba4ee3f0811b41c381733a6e6ff16',1,'mlx::core::simd::atan2(Simd< float16_t, N > x, Simd< float16_t, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ac6104b5667e0eb379528bf7e2de23bee',1,'mlx::core::simd::atan2(Simd< T, N > a, Simd< T, N > b)'],['../namespacemlx_1_1core_1_1simd.html#af5e8e8230c7d7af8201a3aaa7f491a2d',1,'mlx::core::simd::atan2(Simd< T, 1 > a, Simd< T, 1 > b)'],['../namespacemetal.html#a1d430793eaa38ccf0d07145e3fcd1e61',1,'metal::atan2()'],['../namespacemetal_1_1fast.html#a00e687ea46f5affe26e6aef8fd62b89a',1,'metal::fast::atan2()'],['../namespacemetal_1_1precise.html#a6f161b049cc6884f87b09b33c2d1cd7f',1,'metal::precise::atan2()']]], + ['atanh_107',['atanh',['../namespacemlx_1_1core_1_1simd.html#a18d330fd2c7360b2890a722232ba35b7',1,'mlx::core::simd::atanh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ae55fd26c3e18a6a27679d2b47566f8bc',1,'mlx::core::simd::atanh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a5ebae2e6cce1889513f15be3adb265ea',1,'mlx::core::simd::atanh(Simd< T, 1 > in)'],['../namespacemetal.html#a57116427997ba71dd3863bfb15de33bf',1,'metal::atanh()'],['../namespacemetal_1_1fast.html#af24608fc605db9a14427d37c36dc1c53',1,'metal::fast::atanh()'],['../namespacemetal_1_1precise.html#a902994837653b90c47f4285673e712c4',1,'metal::precise::atanh()']]], + ['atile_108',['Atile',['../structmlx_1_1steel_1_1_block_m_m_a.html#a47e614120c650f7479db79f23a0df586',1,'mlx::steel::BlockMMA']]], + ['atleast_5f1d_109',['atleast_1d',['../group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396',1,'mlx::core::atleast_1d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5',1,'mlx::core::atleast_1d(const std::vector< array > &a, StreamOrDevice s={})']]], + ['atleast_5f2d_110',['atleast_2d',['../group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087',1,'mlx::core::atleast_2d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga9950299a80c2562f13448758f856d1f5',1,'mlx::core::atleast_2d(const std::vector< array > &a, StreamOrDevice s={})']]], + ['atleast_5f3d_111',['atleast_3d',['../group__ops.html#ga4afd919601e67782ff964465919956a0',1,'mlx::core::atleast_3d(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaffdf742ad79440a60dda40062a8074fe',1,'mlx::core::atleast_3d(const std::vector< array > &a, StreamOrDevice s={})']]], + ['atomic_2eh_112',['atomic.h',['../atomic_8h.html',1,'']]], + ['atomic_5fupdate_113',['atomic_update',['../struct_none.html#aecbce7c97e8b1d5dc4afd2e788c24e06',1,'None']]], + ['attach_5fevent_114',['attach_event',['../classmlx_1_1core_1_1array.html#a000c3cfe13cb378bf0523b62816190da',1,'mlx::core::array']]], + ['attention_115',['attention',['../steel__attention_8h.html#a5423b2a414f5e3c14166d568dedfbd33',1,'steel_attention.h']]], + ['attn_2eh_116',['attn.h',['../attn_8h.html',1,'']]], + ['attnparams_117',['AttnParams',['../structmlx_1_1steel_1_1_attn_params.html',1,'mlx::steel']]], + ['available_118',['available',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078a308bd3e5bf976888b120dd36d0c2d2ae',1,'mlx::core::array']]] ]; diff --git a/docs/build/html/search/all_10.js b/docs/build/html/search/all_10.js index ba1c6b4b6..31bdbbda5 100644 --- a/docs/build/html/search/all_10.js +++ b/docs/build/html/search/all_10.js @@ -2,7 +2,7 @@ var searchData= [ ['pack_5ffactor_0',['pack_factor',['../struct_quantized_block_loader.html#a8eae73a0c04bf1e41fb96131f6aa500d',1,'QuantizedBlockLoader']]], ['pad_1',['Pad',['../classmlx_1_1core_1_1_pad.html',1,'mlx::core::Pad'],['../classmlx_1_1core_1_1_pad.html#aa55090a94f574c29678d841d091cdf44',1,'mlx::core::Pad::Pad()']]], - ['pad_2',['pad',['../struct_m_l_x_conv_params.html#ae84a9afb3a95b57e0b763bb0ebda0753',1,'MLXConvParams::pad'],['../group__ops.html#ga58e03f39aa5d63aba1646f530706a920',1,'mlx::core::pad(const array &a, const std::vector< int > &axes, const Shape &low_pad_size, const Shape &high_pad_size, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#ga8495c9902b0a67f288918cb528d5abeb',1,'mlx::core::pad(const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#gaeaae2f9637b0b3e2b4c6c0517e285935',1,'mlx::core::pad(const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#ga5b8684361f57cbfd5e6eedf82a62ded0',1,'mlx::core::pad(const array &a, int pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})']]], + ['pad_2',['pad',['../struct_m_l_x_conv_params.html#ae84a9afb3a95b57e0b763bb0ebda0753',1,'MLXConvParams::pad'],['../group__ops.html#gab95ebd20bd7c6d1c840007cc020cbc0c',1,'mlx::core::pad(const array &a, const std::vector< int > &axes, const Shape &low_pad_size, const Shape &high_pad_size, const array &pad_value=array(0), const std::string &mode="constant", StreamOrDevice s={})'],['../group__ops.html#gad89f464e92c356faab1f1c2e763b1fb7',1,'mlx::core::pad(const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), const std::string &mode="constant", StreamOrDevice s={})'],['../group__ops.html#gab2d5a17d3d5225fed34905e786c31c5f',1,'mlx::core::pad(const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), const std::string &mode="constant", StreamOrDevice s={})'],['../group__ops.html#gaa73ac5674467b1d5c74de6fef7204c44',1,'mlx::core::pad(const array &a, int pad_width, const array &pad_value=array(0), const std::string &mode="constant", StreamOrDevice s={})']]], ['pad_5fgpu_3',['pad_gpu',['../namespacemlx_1_1core.html#a830324cd1b6231218b3e561e247e69b9',1,'mlx::core']]], ['parallelfilereader_4',['ParallelFileReader',['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html',1,'mlx::core::io::ParallelFileReader'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a6cdb4547408f8cbca9e2ddd82514e697',1,'mlx::core::io::ParallelFileReader::ParallelFileReader()']]], ['params_5',['params',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a09b4719415c5bddb0bb70c704b1d8d02',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a4f8c792ede675d14b70dd19fcf3c5aee',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::params'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a1fed11be2e8d9d594dcdf60e32b936b1',1,'mlx::steel::Conv2DWeightBlockLoader::params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a335c573456ede3dd34bda1eec9842fe2',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::params'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#af67adf4550d69231a259e79c1aae9acc',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a09fd92c74ef57c20b48bc780153365ba',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::params'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ae4759d18c0e5cc3530b3da8493008419',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::params']]], @@ -42,13 +42,13 @@ var searchData= ['pocketfft_5frearrange_39',['POCKETFFT_REARRANGE',['../pocketfft_8h.html#acffdf2e1ab84f36a7a097e1b8b87a9f9',1,'pocketfft.h']]], ['pocketfft_5frestrict_40',['POCKETFFT_RESTRICT',['../pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82',1,'pocketfft.h']]], ['pointer_5felement_41',['pointer_element',['../structmetal_1_1pointer__element.html',1,'metal']]], - ['pointer_5felement_3c_20constant_20t_20_2a_20_3e_42',['pointer_element< constant T * >',['../structmetal_1_1pointer__element.html',1,'metal']]], - ['pointer_5felement_3c_20device_20t_20_2a_20_3e_43',['pointer_element< device T * >',['../structmetal_1_1pointer__element.html',1,'metal']]], - ['pointer_5felement_3c_20thread_20t_20_2a_20_3e_44',['pointer_element< thread T * >',['../structmetal_1_1pointer__element.html',1,'metal']]], - ['pointer_5felement_3c_20threadgroup_20t_20_2a_20_3e_45',['pointer_element< threadgroup T * >',['../structmetal_1_1pointer__element.html',1,'metal']]], + ['pointer_5felement_3c_20constant_20t_20_2a_20_3e_42',['pointer_element< constant T * >',['../structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.html',1,'metal']]], + ['pointer_5felement_3c_20device_20t_20_2a_20_3e_43',['pointer_element< device T * >',['../structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.html',1,'metal']]], + ['pointer_5felement_3c_20thread_20t_20_2a_20_3e_44',['pointer_element< thread T * >',['../structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.html',1,'metal']]], + ['pointer_5felement_3c_20threadgroup_20t_20_2a_20_3e_45',['pointer_element< threadgroup T * >',['../structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.html',1,'metal']]], ['pointer_5felement_5ft_46',['pointer_element_t',['../namespacemetal.html#ac82ee6c3fbe9ec5c78c07329424aaec9',1,'metal']]], ['post_5fin_47',['post_in',['../struct_read_writer.html#ab555cec93b66eead607e6a03d9324e1c',1,'ReadWriter::post_in(float2 elem) const'],['../struct_read_writer.html#a6c47a25b2135393045fa5f95ada59d9d',1,'ReadWriter::post_in(float elem) const']]], - ['pow_48',['pow',['../namespacemetal.html#acd288d4552215bd10455584a214c57b8',1,'metal::pow()'],['../namespacemetal_1_1fast.html#ade2367eaec894bd2e14a1351c363e003',1,'metal::fast::pow()'],['../namespacemetal_1_1precise.html#a4cce64f1f20c1c6dfd29115bdb7c8d42',1,'metal::precise::pow()']]], + ['pow_48',['pow',['../namespacemlx_1_1core_1_1simd.html#a19d535de1fc179cc39ec9643c9863cbc',1,'mlx::core::simd::pow(Simd< float16_t, N > x, Simd< float16_t, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ab18b3a88a2439fd026b6551b38d1f14a',1,'mlx::core::simd::pow(Simd< T, N > base, Simd< T, N > exp)'],['../namespacemlx_1_1core_1_1simd.html#afe3d50bc4a11061898aa57377fa9536d',1,'mlx::core::simd::pow(Simd< T, 1 > a, Simd< T, 1 > b)'],['../namespacemetal.html#acd288d4552215bd10455584a214c57b8',1,'metal::pow()'],['../namespacemetal_1_1fast.html#ade2367eaec894bd2e14a1351c363e003',1,'metal::fast::pow()'],['../namespacemetal_1_1precise.html#a4cce64f1f20c1c6dfd29115bdb7c8d42',1,'metal::precise::pow()']]], ['power_49',['Power',['../structmlx_1_1core_1_1detail_1_1_power.html',1,'mlx::core::detail::Power'],['../classmlx_1_1core_1_1_power.html',1,'mlx::core::Power'],['../struct_power.html',1,'Power'],['../classmlx_1_1core_1_1_power.html#a7bc6c64179b7a2aef56fe1dafb6459b2',1,'mlx::core::Power::Power()']]], ['power_50',['power',['../group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d',1,'mlx::core']]], ['powr_51',['powr',['../namespacemetal.html#ae529e431f178bafedc18a889323c0bc2',1,'metal::powr()'],['../namespacemetal_1_1fast.html#a4293cbc94175b4dcc724fe4747eb5d5a',1,'metal::fast::powr()'],['../namespacemetal_1_1precise.html#ac9dbab0bd99b2b94e364aba5353bdcd7',1,'metal::precise::powr()']]], @@ -60,7 +60,7 @@ var searchData= ['primitive_5fid_57',['primitive_id',['../classmlx_1_1core_1_1array.html#af5ad83605d4eea81561246873bee1d7c',1,'mlx::core::array']]], ['primitive_5fptr_58',['primitive_ptr',['../classmlx_1_1core_1_1array.html#a5119cd616ec3c05d65878944b8889469',1,'mlx::core::array']]], ['primitives_2eh_59',['primitives.h',['../distributed_2primitives_8h.html',1,'(Global Namespace)'],['../primitives_8h.html',1,'(Global Namespace)']]], - ['print_60',['print',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a6814f9008a683c6911d5b8991ef770ab',1,'mlx::core::distributed::AllReduce::print()'],['../classmlx_1_1core_1_1_primitive.html#ae1aff91354ce036596088a3e19474ecb',1,'mlx::core::Primitive::print()'],['../classmlx_1_1core_1_1_abs.html#a643d6db5116eed978e3208804a992107',1,'mlx::core::Abs::print()'],['../classmlx_1_1core_1_1_add.html#a8a96345aa63724f22b68bca7b861211d',1,'mlx::core::Add::print()'],['../classmlx_1_1core_1_1_add_m_m.html#a1262ac2c4c6e9ff6b6047bf7605e5cc9',1,'mlx::core::AddMM::print()'],['../classmlx_1_1core_1_1_arange.html#abd73d2b793da796dc7cf04c9f7d5c19e',1,'mlx::core::Arange::print()'],['../classmlx_1_1core_1_1_arc_cos.html#aa48d8bec4efbac569d809cf11648b739',1,'mlx::core::ArcCos::print()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6a9a2ab0cc360d7e2f9676db17f8e630',1,'mlx::core::ArcCosh::print()'],['../classmlx_1_1core_1_1_arc_sin.html#a895a35c9dd22fdb06e7b971bfd6fde87',1,'mlx::core::ArcSin::print()'],['../classmlx_1_1core_1_1_arc_sinh.html#aa8b2934a8a0b2eedec8257bbb5726430',1,'mlx::core::ArcSinh::print()'],['../classmlx_1_1core_1_1_arc_tan.html#ab0309e4feca36f221b3d672dc92cac05',1,'mlx::core::ArcTan::print()'],['../classmlx_1_1core_1_1_arc_tan2.html#abdfef9f572d06df1251c28222756a361',1,'mlx::core::ArcTan2::print()'],['../classmlx_1_1core_1_1_arc_tanh.html#aa9549311240d7ba225b84e1df9ad8523',1,'mlx::core::ArcTanh::print()'],['../classmlx_1_1core_1_1_arg_partition.html#aa8678d94fa1571ea71a7bf790cdb8d63',1,'mlx::core::ArgPartition::print()'],['../classmlx_1_1core_1_1_arg_reduce.html#a153a6d8dba7301c4fcd0e429154ead8f',1,'mlx::core::ArgReduce::print()'],['../classmlx_1_1core_1_1_arg_sort.html#a0b59ce43e0982d634a01631728b419bd',1,'mlx::core::ArgSort::print()'],['../classmlx_1_1core_1_1_as_type.html#aa617e29147c14bd5d1fa8ad0bf65af0c',1,'mlx::core::AsType::print()'],['../classmlx_1_1core_1_1_as_strided.html#af2e21b77ea9e6c70bca45224967745bf',1,'mlx::core::AsStrided::print()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a69b28e239da7fdb89f0a9f9467dd797d',1,'mlx::core::BitwiseBinary::print()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a37ecf6fa296d28efb7651a3c510fe159',1,'mlx::core::BlockMaskedMM::print()'],['../classmlx_1_1core_1_1_gather_m_m.html#ae7a6f4eecb15e95b21e6c87068ebd758',1,'mlx::core::GatherMM::print()'],['../classmlx_1_1core_1_1_broadcast_axes.html#aa15f81d08cabe43ac92de7534fb940df',1,'mlx::core::BroadcastAxes::print()'],['../classmlx_1_1core_1_1_broadcast.html#a6a610412861c6e472f930b6721b99a11',1,'mlx::core::Broadcast::print()'],['../classmlx_1_1core_1_1_ceil.html#a14a0048dd6496341cacaddada68276ee',1,'mlx::core::Ceil::print()'],['../classmlx_1_1core_1_1_compiled.html#a271521f92eef49c39799f38e26b64a9b',1,'mlx::core::Compiled::print()'],['../classmlx_1_1core_1_1_concatenate.html#a56f29b585a6d1d958954a68dcc893f33',1,'mlx::core::Concatenate::print()'],['../classmlx_1_1core_1_1_conjugate.html#a40281539bbd543ac8fd8e28650de17e4',1,'mlx::core::Conjugate::print()'],['../classmlx_1_1core_1_1_contiguous.html#aca8a4ba9a58cc10f063e6b082fa2fc23',1,'mlx::core::Contiguous::print()'],['../classmlx_1_1core_1_1_convolution.html#a844eab7c4cc99e775cfb561265ed14fd',1,'mlx::core::Convolution::print()'],['../classmlx_1_1core_1_1_copy.html#acfa1a02ab9cdab593e928faa515a8008',1,'mlx::core::Copy::print()'],['../classmlx_1_1core_1_1_cos.html#a81858457e4bea931a4bc6f6e38b0f696',1,'mlx::core::Cos::print()'],['../classmlx_1_1core_1_1_cosh.html#ac247faad68c1050cda9f72d7d6d040e2',1,'mlx::core::Cosh::print()'],['../classmlx_1_1core_1_1_custom_transforms.html#a2ddbacbc468271b11caee0ad97005298',1,'mlx::core::CustomTransforms::print()'],['../classmlx_1_1core_1_1_depends.html#aed575b0d927f4341f60442c70adeeb82',1,'mlx::core::Depends::print()'],['../classmlx_1_1core_1_1_divide.html#af3c15337ac15522cc34ed98b97895bb6',1,'mlx::core::Divide::print()'],['../classmlx_1_1core_1_1_div_mod.html#a7edbed50d07869d921e529157931b7a1',1,'mlx::core::DivMod::print()'],['../classmlx_1_1core_1_1_select.html#a678285f2c0b9dae85692399c3aa692a7',1,'mlx::core::Select::print()'],['../classmlx_1_1core_1_1_remainder.html#aeaecac5ea8e606d7ecd393d8019029e4',1,'mlx::core::Remainder::print()'],['../classmlx_1_1core_1_1_equal.html#a0787bf32f0b405a8b2ac809d2d990774',1,'mlx::core::Equal::print()'],['../classmlx_1_1core_1_1_erf.html#a186af7b783cf832c3b25eec3a09f5a0c',1,'mlx::core::Erf::print()'],['../classmlx_1_1core_1_1_erf_inv.html#a0acb31bd5780abf61877bd1a3e0fd4f9',1,'mlx::core::ErfInv::print()'],['../classmlx_1_1core_1_1_exp.html#ad87cc1b2ae595a613b03b0fdca63ae6a',1,'mlx::core::Exp::print()'],['../classmlx_1_1core_1_1_expm1.html#af1a99266fc50aa5948cdd298e2916ef1',1,'mlx::core::Expm1::print()'],['../classmlx_1_1core_1_1_expand_dims.html#ac8f1d849562b2222158fbe476fc2dc2e',1,'mlx::core::ExpandDims::print()'],['../classmlx_1_1core_1_1_f_f_t.html#a15a2a5f7647f5fb78611a251d3270edf',1,'mlx::core::FFT::print()'],['../classmlx_1_1core_1_1_flatten.html#ad0495ee66601c7527d836d2db77a6aec',1,'mlx::core::Flatten::print()'],['../classmlx_1_1core_1_1_floor.html#ac289e87c5fac15e2f491e2513be610f6',1,'mlx::core::Floor::print()'],['../classmlx_1_1core_1_1_full.html#a68e08303f4960ab373b84a3312edc013',1,'mlx::core::Full::print()'],['../classmlx_1_1core_1_1_gather.html#a9d57637a8a65008683c3847251bdcf91',1,'mlx::core::Gather::print()'],['../classmlx_1_1core_1_1_greater.html#aa2980e45cd2c79ebfb394012d3108a04',1,'mlx::core::Greater::print()'],['../classmlx_1_1core_1_1_greater_equal.html#ab98045c861d2d2ffb0398c2c1d671cef',1,'mlx::core::GreaterEqual::print()'],['../classmlx_1_1core_1_1_hadamard.html#a3df6e7e3b3b71bf50be5f1a05d0870b6',1,'mlx::core::Hadamard::print()'],['../classmlx_1_1core_1_1_imag.html#a0c8d48e2a1474d80a314ea9b96dbaa8d',1,'mlx::core::Imag::print()'],['../classmlx_1_1core_1_1_less.html#ad67e6f66d7b75546fd98dbee6b631d78',1,'mlx::core::Less::print()'],['../classmlx_1_1core_1_1_less_equal.html#a409842d3862113c53cbbdf7467a06950',1,'mlx::core::LessEqual::print()'],['../classmlx_1_1core_1_1_load.html#a54e08a0ca41b7c9f1a76b00c889f0bfa',1,'mlx::core::Load::print()'],['../classmlx_1_1core_1_1_log.html#a7b946d98d4a228c6be9f606a3bd8a30d',1,'mlx::core::Log::print()'],['../classmlx_1_1core_1_1_log1p.html#a8a1569dde30440ce11ea466ccc69d2d4',1,'mlx::core::Log1p::print()'],['../classmlx_1_1core_1_1_logical_not.html#a001ff3eca46440f0d8a287e0b98a8a2c',1,'mlx::core::LogicalNot::print()'],['../classmlx_1_1core_1_1_logical_and.html#a9a5220eb56e1fd94fd879394ee5ad397',1,'mlx::core::LogicalAnd::print()'],['../classmlx_1_1core_1_1_logical_or.html#a6becc5fbfadde850de9857099dcd5003',1,'mlx::core::LogicalOr::print()'],['../classmlx_1_1core_1_1_log_add_exp.html#a702a2eff0bd1ae7b6fb829dd0b0b11b9',1,'mlx::core::LogAddExp::print()'],['../classmlx_1_1core_1_1_matmul.html#abb4a16a265a05d56a2f5d2e89d6f9dfd',1,'mlx::core::Matmul::print()'],['../classmlx_1_1core_1_1_maximum.html#a3b708a1d6b526719c62850294776f8ca',1,'mlx::core::Maximum::print()'],['../classmlx_1_1core_1_1_minimum.html#a137677bf32c626a768b732a7b8575512',1,'mlx::core::Minimum::print()'],['../classmlx_1_1core_1_1_multiply.html#aa4f1f7af68346ce80c2636df415c9909',1,'mlx::core::Multiply::print()'],['../classmlx_1_1core_1_1_negative.html#a0d5c30e267ff6468d64f1987f9f83f91',1,'mlx::core::Negative::print()'],['../classmlx_1_1core_1_1_not_equal.html#a12aa2f764880d29e627540610b63af09',1,'mlx::core::NotEqual::print()'],['../classmlx_1_1core_1_1_number_of_elements.html#aecde30826970938f3aa688979a668f52',1,'mlx::core::NumberOfElements::print()'],['../classmlx_1_1core_1_1_pad.html#af87754daaf51f6a6cf8bd4949ca1e70a',1,'mlx::core::Pad::print()'],['../classmlx_1_1core_1_1_partition.html#ab5c7aa4fed325475b33d4004649f0dc0',1,'mlx::core::Partition::print()'],['../classmlx_1_1core_1_1_power.html#a33e2d7ff078426fe66ea2370ceb5af60',1,'mlx::core::Power::print()'],['../classmlx_1_1core_1_1_quantized_matmul.html#aaef8c96d4d40b4fa08ced540d341a4db',1,'mlx::core::QuantizedMatmul::print()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a53c3fa7beb51ce2e1c2da28633406fe0',1,'mlx::core::GatherQMM::print()'],['../classmlx_1_1core_1_1_random_bits.html#a8a5593c34fd868d94b36a8ced1390271',1,'mlx::core::RandomBits::print()'],['../classmlx_1_1core_1_1_real.html#a740a0dfb54c2a4467a0a59f11fe69e1b',1,'mlx::core::Real::print()'],['../classmlx_1_1core_1_1_reshape.html#a0f2323d5d67ece0eb25ecff565b21862',1,'mlx::core::Reshape::print()'],['../classmlx_1_1core_1_1_reduce.html#a399be3a89553787a0a687706881f03cd',1,'mlx::core::Reduce::print()'],['../classmlx_1_1core_1_1_round.html#af0dfe8943109c936b35ab0082f566f72',1,'mlx::core::Round::print()'],['../classmlx_1_1core_1_1_scan.html#ad5b6308c79e9b985a49df35eadd15b22',1,'mlx::core::Scan::print()'],['../classmlx_1_1core_1_1_scatter.html#aa9d45cbfb27b814517f6016092b30efa',1,'mlx::core::Scatter::print()'],['../classmlx_1_1core_1_1_sigmoid.html#ad4cd19938e5159754aa7516f405580c2',1,'mlx::core::Sigmoid::print()'],['../classmlx_1_1core_1_1_sign.html#a2aa0720fe0a6d2408eb43c25d3d45b0a',1,'mlx::core::Sign::print()'],['../classmlx_1_1core_1_1_sin.html#a73b31005551015897f15c00e8b0222e4',1,'mlx::core::Sin::print()'],['../classmlx_1_1core_1_1_sinh.html#a5b4753d52e80799d4fea0b9172d25a77',1,'mlx::core::Sinh::print()'],['../classmlx_1_1core_1_1_slice.html#a50851148948d924b71817cfbd4401504',1,'mlx::core::Slice::print()'],['../classmlx_1_1core_1_1_slice_update.html#a751eefb9922c56479b4b0de2ad45439b',1,'mlx::core::SliceUpdate::print()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a421283744fe5554ac9a8288cf47edeab',1,'mlx::core::DynamicSlice::print()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a3669f4d939ba36256c43143b603eb12b',1,'mlx::core::DynamicSliceUpdate::print()'],['../classmlx_1_1core_1_1_softmax.html#aa783610ef6b82b92681e78fc99412d83',1,'mlx::core::Softmax::print()'],['../classmlx_1_1core_1_1_sort.html#ada81b9343f80958174eba708452927a2',1,'mlx::core::Sort::print()'],['../classmlx_1_1core_1_1_split.html#ad0c31fe5972643cc75fde10445fc47f2',1,'mlx::core::Split::print()'],['../classmlx_1_1core_1_1_square.html#a75feb558cd1d615e96309dd7d1e81384',1,'mlx::core::Square::print()'],['../classmlx_1_1core_1_1_sqrt.html#a8681c8de2f50049848d320c47f713c0f',1,'mlx::core::Sqrt::print()'],['../classmlx_1_1core_1_1_stop_gradient.html#acc7a7d51cbf014dae8ba3d20bedcad50',1,'mlx::core::StopGradient::print()'],['../classmlx_1_1core_1_1_subtract.html#a3834fd305435fb5a8e512566832e372b',1,'mlx::core::Subtract::print()'],['../classmlx_1_1core_1_1_squeeze.html#a74c9c825b5b968badb9bca8159eabcdf',1,'mlx::core::Squeeze::print()'],['../classmlx_1_1core_1_1_tan.html#aeea7c284d595a2a928d5f28a55e9be7f',1,'mlx::core::Tan::print()'],['../classmlx_1_1core_1_1_tanh.html#a73f4976d641daf697cc1a231d773d78e',1,'mlx::core::Tanh::print()'],['../classmlx_1_1core_1_1_unflatten.html#a77820cf21bd1277c173305b72599bdef',1,'mlx::core::Unflatten::print()'],['../classmlx_1_1core_1_1_view.html#a513b034919a8a494add3155f910a360c',1,'mlx::core::View::print()'],['../classmlx_1_1core_1_1_transpose.html#ac6c87b850f4e5560aa13a5e1e9f9fe04',1,'mlx::core::Transpose::print()'],['../classmlx_1_1core_1_1_q_r_f.html#aba3526722b3a52b41fa8103b909f7f3b',1,'mlx::core::QRF::print()'],['../classmlx_1_1core_1_1_s_v_d.html#ab87a4e7ef857936bea66ba9e24662f53',1,'mlx::core::SVD::print()'],['../classmlx_1_1core_1_1_inverse.html#a543f18f1ce5c06c897141091e95a66e9',1,'mlx::core::Inverse::print()'],['../classmlx_1_1core_1_1_cholesky.html#a0a8b51ff7f5369d22bdc58910d4aaf84',1,'mlx::core::Cholesky::print()'],['../classmlx_1_1core_1_1_eigh.html#a2b8e47ecd60cd7330716761c5fb1fe84',1,'mlx::core::Eigh::print()'],['../structmlx_1_1core_1_1_print_formatter.html#a79fad4cf5844db8c92b066539146281b',1,'mlx::core::PrintFormatter::print(std::ostream &os, bool val)'],['../structmlx_1_1core_1_1_print_formatter.html#a8da448a8adae671b26359341ea514316',1,'mlx::core::PrintFormatter::print(std::ostream &os, int16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9d750c134a6fbfa8251c5b1d01d73287',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#adbbb9cbff767f9db73c659a0c07ba633',1,'mlx::core::PrintFormatter::print(std::ostream &os, int32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a520adb07fafd911b22bc24b295e4f6cf',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ab0c702f1ae201e17cd328c9855cf522e',1,'mlx::core::PrintFormatter::print(std::ostream &os, int64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac59a5137ddd8b32aae057bb9826ee80d',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac4b7895d1168cfc1a3d1186d8a414d2f',1,'mlx::core::PrintFormatter::print(std::ostream &os, float16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ae21005f92bc641f2d657096f5d176a6d',1,'mlx::core::PrintFormatter::print(std::ostream &os, bfloat16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a57af5c32561b95d6ac2a3a1dc4f5d43e',1,'mlx::core::PrintFormatter::print(std::ostream &os, float val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9e1dc67c9afb0a09966336504790823d',1,'mlx::core::PrintFormatter::print(std::ostream &os, complex64_t val)']]], + ['print_60',['print',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a6814f9008a683c6911d5b8991ef770ab',1,'mlx::core::distributed::AllReduce::print()'],['../classmlx_1_1core_1_1_primitive.html#ae1aff91354ce036596088a3e19474ecb',1,'mlx::core::Primitive::print()'],['../classmlx_1_1core_1_1_abs.html#a643d6db5116eed978e3208804a992107',1,'mlx::core::Abs::print()'],['../classmlx_1_1core_1_1_add.html#a8a96345aa63724f22b68bca7b861211d',1,'mlx::core::Add::print()'],['../classmlx_1_1core_1_1_add_m_m.html#a1262ac2c4c6e9ff6b6047bf7605e5cc9',1,'mlx::core::AddMM::print()'],['../classmlx_1_1core_1_1_arange.html#abd73d2b793da796dc7cf04c9f7d5c19e',1,'mlx::core::Arange::print()'],['../classmlx_1_1core_1_1_arc_cos.html#aa48d8bec4efbac569d809cf11648b739',1,'mlx::core::ArcCos::print()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6a9a2ab0cc360d7e2f9676db17f8e630',1,'mlx::core::ArcCosh::print()'],['../classmlx_1_1core_1_1_arc_sin.html#a895a35c9dd22fdb06e7b971bfd6fde87',1,'mlx::core::ArcSin::print()'],['../classmlx_1_1core_1_1_arc_sinh.html#aa8b2934a8a0b2eedec8257bbb5726430',1,'mlx::core::ArcSinh::print()'],['../classmlx_1_1core_1_1_arc_tan.html#ab0309e4feca36f221b3d672dc92cac05',1,'mlx::core::ArcTan::print()'],['../classmlx_1_1core_1_1_arc_tan2.html#abdfef9f572d06df1251c28222756a361',1,'mlx::core::ArcTan2::print()'],['../classmlx_1_1core_1_1_arc_tanh.html#aa9549311240d7ba225b84e1df9ad8523',1,'mlx::core::ArcTanh::print()'],['../classmlx_1_1core_1_1_arg_partition.html#aa8678d94fa1571ea71a7bf790cdb8d63',1,'mlx::core::ArgPartition::print()'],['../classmlx_1_1core_1_1_arg_reduce.html#a153a6d8dba7301c4fcd0e429154ead8f',1,'mlx::core::ArgReduce::print()'],['../classmlx_1_1core_1_1_arg_sort.html#a0b59ce43e0982d634a01631728b419bd',1,'mlx::core::ArgSort::print()'],['../classmlx_1_1core_1_1_as_type.html#aa617e29147c14bd5d1fa8ad0bf65af0c',1,'mlx::core::AsType::print()'],['../classmlx_1_1core_1_1_as_strided.html#af2e21b77ea9e6c70bca45224967745bf',1,'mlx::core::AsStrided::print()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a69b28e239da7fdb89f0a9f9467dd797d',1,'mlx::core::BitwiseBinary::print()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a37ecf6fa296d28efb7651a3c510fe159',1,'mlx::core::BlockMaskedMM::print()'],['../classmlx_1_1core_1_1_gather_m_m.html#ae7a6f4eecb15e95b21e6c87068ebd758',1,'mlx::core::GatherMM::print()'],['../classmlx_1_1core_1_1_broadcast_axes.html#aa15f81d08cabe43ac92de7534fb940df',1,'mlx::core::BroadcastAxes::print()'],['../classmlx_1_1core_1_1_broadcast.html#a6a610412861c6e472f930b6721b99a11',1,'mlx::core::Broadcast::print()'],['../classmlx_1_1core_1_1_ceil.html#a14a0048dd6496341cacaddada68276ee',1,'mlx::core::Ceil::print()'],['../classmlx_1_1core_1_1_compiled.html#a271521f92eef49c39799f38e26b64a9b',1,'mlx::core::Compiled::print()'],['../classmlx_1_1core_1_1_concatenate.html#a56f29b585a6d1d958954a68dcc893f33',1,'mlx::core::Concatenate::print()'],['../classmlx_1_1core_1_1_conjugate.html#a40281539bbd543ac8fd8e28650de17e4',1,'mlx::core::Conjugate::print()'],['../classmlx_1_1core_1_1_contiguous.html#aca8a4ba9a58cc10f063e6b082fa2fc23',1,'mlx::core::Contiguous::print()'],['../classmlx_1_1core_1_1_convolution.html#a844eab7c4cc99e775cfb561265ed14fd',1,'mlx::core::Convolution::print()'],['../classmlx_1_1core_1_1_copy.html#acfa1a02ab9cdab593e928faa515a8008',1,'mlx::core::Copy::print()'],['../classmlx_1_1core_1_1_cos.html#a81858457e4bea931a4bc6f6e38b0f696',1,'mlx::core::Cos::print()'],['../classmlx_1_1core_1_1_cosh.html#ac247faad68c1050cda9f72d7d6d040e2',1,'mlx::core::Cosh::print()'],['../classmlx_1_1core_1_1_custom_transforms.html#a2ddbacbc468271b11caee0ad97005298',1,'mlx::core::CustomTransforms::print()'],['../classmlx_1_1core_1_1_depends.html#aed575b0d927f4341f60442c70adeeb82',1,'mlx::core::Depends::print()'],['../classmlx_1_1core_1_1_divide.html#af3c15337ac15522cc34ed98b97895bb6',1,'mlx::core::Divide::print()'],['../classmlx_1_1core_1_1_div_mod.html#a7edbed50d07869d921e529157931b7a1',1,'mlx::core::DivMod::print()'],['../classmlx_1_1core_1_1_select.html#a678285f2c0b9dae85692399c3aa692a7',1,'mlx::core::Select::print()'],['../classmlx_1_1core_1_1_remainder.html#aeaecac5ea8e606d7ecd393d8019029e4',1,'mlx::core::Remainder::print()'],['../classmlx_1_1core_1_1_equal.html#a0787bf32f0b405a8b2ac809d2d990774',1,'mlx::core::Equal::print()'],['../classmlx_1_1core_1_1_erf.html#a186af7b783cf832c3b25eec3a09f5a0c',1,'mlx::core::Erf::print()'],['../classmlx_1_1core_1_1_erf_inv.html#a0acb31bd5780abf61877bd1a3e0fd4f9',1,'mlx::core::ErfInv::print()'],['../classmlx_1_1core_1_1_exp.html#ad87cc1b2ae595a613b03b0fdca63ae6a',1,'mlx::core::Exp::print()'],['../classmlx_1_1core_1_1_expm1.html#af1a99266fc50aa5948cdd298e2916ef1',1,'mlx::core::Expm1::print()'],['../classmlx_1_1core_1_1_expand_dims.html#ac8f1d849562b2222158fbe476fc2dc2e',1,'mlx::core::ExpandDims::print()'],['../classmlx_1_1core_1_1_f_f_t.html#a15a2a5f7647f5fb78611a251d3270edf',1,'mlx::core::FFT::print()'],['../classmlx_1_1core_1_1_flatten.html#ad0495ee66601c7527d836d2db77a6aec',1,'mlx::core::Flatten::print()'],['../classmlx_1_1core_1_1_floor.html#ac289e87c5fac15e2f491e2513be610f6',1,'mlx::core::Floor::print()'],['../classmlx_1_1core_1_1_full.html#a68e08303f4960ab373b84a3312edc013',1,'mlx::core::Full::print()'],['../classmlx_1_1core_1_1_gather.html#a9d57637a8a65008683c3847251bdcf91',1,'mlx::core::Gather::print()'],['../classmlx_1_1core_1_1_gather_axis.html#a9108bd9dfc153e6260e6340ff923ba38',1,'mlx::core::GatherAxis::print()'],['../classmlx_1_1core_1_1_greater.html#aa2980e45cd2c79ebfb394012d3108a04',1,'mlx::core::Greater::print()'],['../classmlx_1_1core_1_1_greater_equal.html#ab98045c861d2d2ffb0398c2c1d671cef',1,'mlx::core::GreaterEqual::print()'],['../classmlx_1_1core_1_1_hadamard.html#a3df6e7e3b3b71bf50be5f1a05d0870b6',1,'mlx::core::Hadamard::print()'],['../classmlx_1_1core_1_1_imag.html#a0c8d48e2a1474d80a314ea9b96dbaa8d',1,'mlx::core::Imag::print()'],['../classmlx_1_1core_1_1_less.html#ad67e6f66d7b75546fd98dbee6b631d78',1,'mlx::core::Less::print()'],['../classmlx_1_1core_1_1_less_equal.html#a409842d3862113c53cbbdf7467a06950',1,'mlx::core::LessEqual::print()'],['../classmlx_1_1core_1_1_load.html#a54e08a0ca41b7c9f1a76b00c889f0bfa',1,'mlx::core::Load::print()'],['../classmlx_1_1core_1_1_log.html#a7b946d98d4a228c6be9f606a3bd8a30d',1,'mlx::core::Log::print()'],['../classmlx_1_1core_1_1_log1p.html#a8a1569dde30440ce11ea466ccc69d2d4',1,'mlx::core::Log1p::print()'],['../classmlx_1_1core_1_1_logical_not.html#a001ff3eca46440f0d8a287e0b98a8a2c',1,'mlx::core::LogicalNot::print()'],['../classmlx_1_1core_1_1_logical_and.html#a9a5220eb56e1fd94fd879394ee5ad397',1,'mlx::core::LogicalAnd::print()'],['../classmlx_1_1core_1_1_logical_or.html#a6becc5fbfadde850de9857099dcd5003',1,'mlx::core::LogicalOr::print()'],['../classmlx_1_1core_1_1_log_add_exp.html#a702a2eff0bd1ae7b6fb829dd0b0b11b9',1,'mlx::core::LogAddExp::print()'],['../classmlx_1_1core_1_1_matmul.html#abb4a16a265a05d56a2f5d2e89d6f9dfd',1,'mlx::core::Matmul::print()'],['../classmlx_1_1core_1_1_maximum.html#a3b708a1d6b526719c62850294776f8ca',1,'mlx::core::Maximum::print()'],['../classmlx_1_1core_1_1_minimum.html#a137677bf32c626a768b732a7b8575512',1,'mlx::core::Minimum::print()'],['../classmlx_1_1core_1_1_multiply.html#aa4f1f7af68346ce80c2636df415c9909',1,'mlx::core::Multiply::print()'],['../classmlx_1_1core_1_1_negative.html#a0d5c30e267ff6468d64f1987f9f83f91',1,'mlx::core::Negative::print()'],['../classmlx_1_1core_1_1_not_equal.html#a12aa2f764880d29e627540610b63af09',1,'mlx::core::NotEqual::print()'],['../classmlx_1_1core_1_1_number_of_elements.html#aecde30826970938f3aa688979a668f52',1,'mlx::core::NumberOfElements::print()'],['../classmlx_1_1core_1_1_pad.html#af87754daaf51f6a6cf8bd4949ca1e70a',1,'mlx::core::Pad::print()'],['../classmlx_1_1core_1_1_partition.html#ab5c7aa4fed325475b33d4004649f0dc0',1,'mlx::core::Partition::print()'],['../classmlx_1_1core_1_1_power.html#a33e2d7ff078426fe66ea2370ceb5af60',1,'mlx::core::Power::print()'],['../classmlx_1_1core_1_1_quantized_matmul.html#aaef8c96d4d40b4fa08ced540d341a4db',1,'mlx::core::QuantizedMatmul::print()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a53c3fa7beb51ce2e1c2da28633406fe0',1,'mlx::core::GatherQMM::print()'],['../classmlx_1_1core_1_1_random_bits.html#a8a5593c34fd868d94b36a8ced1390271',1,'mlx::core::RandomBits::print()'],['../classmlx_1_1core_1_1_real.html#a740a0dfb54c2a4467a0a59f11fe69e1b',1,'mlx::core::Real::print()'],['../classmlx_1_1core_1_1_reshape.html#a0f2323d5d67ece0eb25ecff565b21862',1,'mlx::core::Reshape::print()'],['../classmlx_1_1core_1_1_reduce.html#a399be3a89553787a0a687706881f03cd',1,'mlx::core::Reduce::print()'],['../classmlx_1_1core_1_1_round.html#af0dfe8943109c936b35ab0082f566f72',1,'mlx::core::Round::print()'],['../classmlx_1_1core_1_1_scan.html#ad5b6308c79e9b985a49df35eadd15b22',1,'mlx::core::Scan::print()'],['../classmlx_1_1core_1_1_scatter.html#aa9d45cbfb27b814517f6016092b30efa',1,'mlx::core::Scatter::print()'],['../classmlx_1_1core_1_1_scatter_axis.html#a657843d4d9846ecd56e35e066986eb96',1,'mlx::core::ScatterAxis::print()'],['../classmlx_1_1core_1_1_sigmoid.html#ad4cd19938e5159754aa7516f405580c2',1,'mlx::core::Sigmoid::print()'],['../classmlx_1_1core_1_1_sign.html#a2aa0720fe0a6d2408eb43c25d3d45b0a',1,'mlx::core::Sign::print()'],['../classmlx_1_1core_1_1_sin.html#a73b31005551015897f15c00e8b0222e4',1,'mlx::core::Sin::print()'],['../classmlx_1_1core_1_1_sinh.html#a5b4753d52e80799d4fea0b9172d25a77',1,'mlx::core::Sinh::print()'],['../classmlx_1_1core_1_1_slice.html#a50851148948d924b71817cfbd4401504',1,'mlx::core::Slice::print()'],['../classmlx_1_1core_1_1_slice_update.html#a751eefb9922c56479b4b0de2ad45439b',1,'mlx::core::SliceUpdate::print()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a421283744fe5554ac9a8288cf47edeab',1,'mlx::core::DynamicSlice::print()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a3669f4d939ba36256c43143b603eb12b',1,'mlx::core::DynamicSliceUpdate::print()'],['../classmlx_1_1core_1_1_softmax.html#aa783610ef6b82b92681e78fc99412d83',1,'mlx::core::Softmax::print()'],['../classmlx_1_1core_1_1_sort.html#ada81b9343f80958174eba708452927a2',1,'mlx::core::Sort::print()'],['../classmlx_1_1core_1_1_split.html#ad0c31fe5972643cc75fde10445fc47f2',1,'mlx::core::Split::print()'],['../classmlx_1_1core_1_1_square.html#a75feb558cd1d615e96309dd7d1e81384',1,'mlx::core::Square::print()'],['../classmlx_1_1core_1_1_sqrt.html#a8681c8de2f50049848d320c47f713c0f',1,'mlx::core::Sqrt::print()'],['../classmlx_1_1core_1_1_stop_gradient.html#acc7a7d51cbf014dae8ba3d20bedcad50',1,'mlx::core::StopGradient::print()'],['../classmlx_1_1core_1_1_subtract.html#a3834fd305435fb5a8e512566832e372b',1,'mlx::core::Subtract::print()'],['../classmlx_1_1core_1_1_squeeze.html#a74c9c825b5b968badb9bca8159eabcdf',1,'mlx::core::Squeeze::print()'],['../classmlx_1_1core_1_1_tan.html#aeea7c284d595a2a928d5f28a55e9be7f',1,'mlx::core::Tan::print()'],['../classmlx_1_1core_1_1_tanh.html#a73f4976d641daf697cc1a231d773d78e',1,'mlx::core::Tanh::print()'],['../classmlx_1_1core_1_1_unflatten.html#a77820cf21bd1277c173305b72599bdef',1,'mlx::core::Unflatten::print()'],['../classmlx_1_1core_1_1_view.html#a513b034919a8a494add3155f910a360c',1,'mlx::core::View::print()'],['../classmlx_1_1core_1_1_transpose.html#ac6c87b850f4e5560aa13a5e1e9f9fe04',1,'mlx::core::Transpose::print()'],['../classmlx_1_1core_1_1_q_r_f.html#aba3526722b3a52b41fa8103b909f7f3b',1,'mlx::core::QRF::print()'],['../classmlx_1_1core_1_1_s_v_d.html#ab87a4e7ef857936bea66ba9e24662f53',1,'mlx::core::SVD::print()'],['../classmlx_1_1core_1_1_inverse.html#a543f18f1ce5c06c897141091e95a66e9',1,'mlx::core::Inverse::print()'],['../classmlx_1_1core_1_1_cholesky.html#a0a8b51ff7f5369d22bdc58910d4aaf84',1,'mlx::core::Cholesky::print()'],['../classmlx_1_1core_1_1_eigh.html#a2b8e47ecd60cd7330716761c5fb1fe84',1,'mlx::core::Eigh::print()'],['../structmlx_1_1core_1_1_print_formatter.html#a79fad4cf5844db8c92b066539146281b',1,'mlx::core::PrintFormatter::print(std::ostream &os, bool val)'],['../structmlx_1_1core_1_1_print_formatter.html#a8da448a8adae671b26359341ea514316',1,'mlx::core::PrintFormatter::print(std::ostream &os, int16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9d750c134a6fbfa8251c5b1d01d73287',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#adbbb9cbff767f9db73c659a0c07ba633',1,'mlx::core::PrintFormatter::print(std::ostream &os, int32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a520adb07fafd911b22bc24b295e4f6cf',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ab0c702f1ae201e17cd328c9855cf522e',1,'mlx::core::PrintFormatter::print(std::ostream &os, int64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac59a5137ddd8b32aae057bb9826ee80d',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac4b7895d1168cfc1a3d1186d8a414d2f',1,'mlx::core::PrintFormatter::print(std::ostream &os, float16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ae21005f92bc641f2d657096f5d176a6d',1,'mlx::core::PrintFormatter::print(std::ostream &os, bfloat16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a57af5c32561b95d6ac2a3a1dc4f5d43e',1,'mlx::core::PrintFormatter::print(std::ostream &os, float val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9e1dc67c9afb0a09966336504790823d',1,'mlx::core::PrintFormatter::print(std::ostream &os, complex64_t val)']]], ['print_5fcomplex_5fconstant_61',['print_complex_constant',['../namespacemlx_1_1core.html#a2b78f270942c6eb185e8045f1c5b4286',1,'mlx::core']]], ['print_5fconstant_62',['print_constant',['../namespacemlx_1_1core.html#a7d11b000895d44d183260634f4192d92',1,'mlx::core']]], ['print_5ffloat_5fconstant_63',['print_float_constant',['../namespacemlx_1_1core.html#a93a8ac59c644b801ec8881a58368caf2',1,'mlx::core']]], @@ -68,7 +68,7 @@ var searchData= ['print_5fint_5fconstant_65',['print_int_constant',['../namespacemlx_1_1core.html#a948ce3dfc4520d3aa98b33e42f617c64',1,'mlx::core']]], ['printformatter_66',['PrintFormatter',['../structmlx_1_1core_1_1_print_formatter.html',1,'mlx::core']]], ['prod_67',['Prod',['../struct_prod.html',1,'Prod< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924acdd1ec09a2fd99c81c561b5c63a4b482',1,'mlx::core::distributed::AllReduce::Prod'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9ac5b077bfec55fe2b141b197dfa00ecf7',1,'mlx::core::Reduce::Prod'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1a33edce755ed1a74632c302ad93a14789',1,'mlx::core::Scan::Prod'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca5e43e8ffd1f5ba49826e2e7ac3450466',1,'mlx::core::Scatter::Prod']]], - ['prod_68',['prod',['../structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89',1,'pocketfft::detail::util::prod()'],['../group__ops.html#ga4a09b7241d564d92548bc2773eb1d544',1,'mlx::core::prod(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga61832191f3c42ea549cf04953edc3602',1,'mlx::core::prod(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2b3935108f641e20a70dbf63f540d970',1,'mlx::core::prod(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga8a10a10b81c69996d0aca8ba401f8ff0',1,'mlx::core::prod(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['prod_68',['prod',['../structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89',1,'pocketfft::detail::util::prod()'],['../namespacemlx_1_1core_1_1simd.html#a271cedfc48efc69db43813e8c424bf7c',1,'mlx::core::simd::prod(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#ac27cdc630e86b25ad607ca409de2b274',1,'mlx::core::simd::prod(Simd< T, 1 > x)'],['../namespacemlx_1_1core_1_1simd.html#a0a26dff48b078fb3e9fef688232183ed',1,'mlx::core::simd::prod(Simd< float16_t, N > x)'],['../group__ops.html#ga4a09b7241d564d92548bc2773eb1d544',1,'mlx::core::prod(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga61832191f3c42ea549cf04953edc3602',1,'mlx::core::prod(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2b3935108f641e20a70dbf63f540d970',1,'mlx::core::prod(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga8a10a10b81c69996d0aca8ba401f8ff0',1,'mlx::core::prod(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], ['promote_5ftypes_69',['promote_types',['../namespacemlx_1_1core.html#a505922e54acd43114308e3bdbda0e497',1,'mlx::core']]], ['ptr_70',['ptr',['../classmlx_1_1core_1_1allocator_1_1_buffer.html#a990643feac06961c5599aac098c17b94',1,'mlx::core::allocator::Buffer::ptr() const'],['../classmlx_1_1core_1_1allocator_1_1_buffer.html#acb15b2f057568828ea09635ed968b62a',1,'mlx::core::allocator::Buffer::ptr()'],['../classmlx_1_1core_1_1metal_1_1_buffer.html#a990643feac06961c5599aac098c17b94',1,'mlx::core::metal::Buffer::ptr() const'],['../classmlx_1_1core_1_1metal_1_1_buffer.html#acb15b2f057568828ea09635ed968b62a',1,'mlx::core::metal::Buffer::ptr()']]], ['push_71',['push',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#ae8ebd6bc1b4d18e75bd046005e7dde7b',1,'pocketfft::detail::threading::concurrent_queue']]], diff --git a/docs/build/html/search/all_12.js b/docs/build/html/search/all_12.js index fef63e126..7cbf5dda2 100644 --- a/docs/build/html/search/all_12.js +++ b/docs/build/html/search/all_12.js @@ -61,73 +61,76 @@ var searchData= ['readwrite_2eh_58',['readwrite.h',['../readwrite_8h.html',1,'']]], ['readwriter_59',['ReadWriter',['../struct_read_writer.html',1,'ReadWriter< in_T, out_T, step, four_step_real >'],['../struct_read_writer.html#a1aa07e41d7ac286ad79bd26a072dfa0c',1,'ReadWriter::ReadWriter()']]], ['real_60',['Real',['../structmlx_1_1core_1_1detail_1_1_real.html',1,'mlx::core::detail::Real'],['../classmlx_1_1core_1_1_real.html',1,'mlx::core::Real'],['../struct_real.html',1,'Real'],['../classmlx_1_1core_1_1_real.html#acd4480e3f0834d70ff6b5f1ecef17892',1,'mlx::core::Real::Real()']]], - ['real_61',['real',['../structcomplex64__t.html#abbd4a0092eca9f112c1c5ae1a133a27e',1,'complex64_t::real'],['../group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764',1,'mlx::core::real(const array &a, StreamOrDevice s={})']]], - ['reciprocal_62',['reciprocal',['../group__ops.html#ga4d29556bb93e2f66916116cf1f062b36',1,'mlx::core']]], - ['recv_63',['Recv',['../classmlx_1_1core_1_1distributed_1_1_recv.html',1,'mlx::core::distributed::Recv'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a511dd4e0259da18a181a25579d9b55db',1,'mlx::core::distributed::Recv::Recv()']]], - ['recv_64',['recv',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac4af5fc16a82ba8c72df04d7694f8352',1,'mlx::core::distributed::detail::GroupImpl::recv()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#a003de04deb00ecbb19179b3f557df548',1,'mlx::core::distributed::detail::recv()'],['../namespacemlx_1_1core_1_1distributed.html#af93c1680b656e98158d5f6eed8e092e8',1,'mlx::core::distributed::recv(Shape shape, Dtype dtype, int src, std::optional< Group > group=std::nullopt, StreamOrDevice s={})']]], - ['recv_5flike_65',['recv_like',['../namespacemlx_1_1core_1_1distributed.html#a2822b78bce2c679e6ff940b2fca944f0',1,'mlx::core::distributed']]], - ['reduce_66',['Reduce',['../classmlx_1_1core_1_1_reduce.html',1,'mlx::core::Reduce'],['../classmlx_1_1core_1_1_reduce.html#a055368c1d036fb953a23ef230e33dcbf',1,'mlx::core::Reduce::Reduce()']]], - ['reduce_67',['reduce',['../namespacemlx_1_1core_1_1metal.html#abb997ccbed4c9a9ccd975b1574755fca',1,'mlx::core::metal']]], - ['reduce_2eh_68',['reduce.h',['../common_2reduce_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2reduce_8h.html',1,'(Global Namespace)'],['../metal_2reduce_8h.html',1,'(Global Namespace)']]], - ['reduce_5fall_2eh_69',['reduce_all.h',['../reduce__all_8h.html',1,'']]], - ['reduce_5fcol_2eh_70',['reduce_col.h',['../reduce__col_8h.html',1,'']]], - ['reduce_5finit_2eh_71',['reduce_init.h',['../reduce__init_8h.html',1,'']]], - ['reduce_5fn_5freads_72',['REDUCE_N_READS',['../defines_8h.html#a2ad505864a2ab786147766900bc18c21',1,'defines.h']]], - ['reduce_5fn_5fwrites_73',['REDUCE_N_WRITES',['../defines_8h.html#a68c33274e15a2f163f7631a36280d82f',1,'defines.h']]], - ['reduce_5frow_2eh_74',['reduce_row.h',['../reduce__row_8h.html',1,'']]], - ['reduce_5futils_75',['reduce_utils',['../namespacemlx_1_1core_1_1metal.html#a2ec39572806310cf528aea06530e8af8',1,'mlx::core::metal']]], - ['reduce_5futils_2eh_76',['reduce_utils.h',['../reduce__utils_8h.html',1,'']]], - ['reducetype_77',['ReduceType',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924',1,'mlx::core::distributed::AllReduce::ReduceType'],['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3',1,'mlx::core::ArgReduce::ReduceType'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9',1,'mlx::core::Reduce::ReduceType'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1',1,'mlx::core::Scan::ReduceType'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613c',1,'mlx::core::Scatter::ReduceType']]], - ['reduction_5fop_78',['reduction_op',['../namespacemlx_1_1core.html#aa08ffc1e8f2c58afb2d463496f827ef0',1,'mlx::core::reduction_op(const array &x, array &out, const std::vector< int > &axes, U init, OpS ops, OpC opc, Op op)'],['../namespacemlx_1_1core.html#a064d61b6ddc9e5d1e261a7e33de71083',1,'mlx::core::reduction_op(const array &x, array &out, const std::vector< int > &axes, U init, Op op)']]], + ['real_61',['real',['../structcomplex64__t.html#abbd4a0092eca9f112c1c5ae1a133a27e',1,'complex64_t::real'],['../namespacemlx_1_1core_1_1simd.html#acdf822b7626bbab6a495552aea3457b5',1,'mlx::core::simd::real()'],['../group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764',1,'mlx::core::real()']]], + ['recip_62',['recip',['../namespacemlx_1_1core_1_1simd.html#ae344abefc91c7d9c0a9506c868a84d61',1,'mlx::core::simd::recip(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#afc915aed256295475ac88fde3a736f1f',1,'mlx::core::simd::recip(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a6235990c43aaf0e0c126c82d10f01b45',1,'mlx::core::simd::recip(Simd< float16_t, N > a)']]], + ['reciprocal_63',['reciprocal',['../group__ops.html#ga4d29556bb93e2f66916116cf1f062b36',1,'mlx::core']]], + ['recv_64',['Recv',['../classmlx_1_1core_1_1distributed_1_1_recv.html',1,'mlx::core::distributed::Recv'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a511dd4e0259da18a181a25579d9b55db',1,'mlx::core::distributed::Recv::Recv()']]], + ['recv_65',['recv',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac4af5fc16a82ba8c72df04d7694f8352',1,'mlx::core::distributed::detail::GroupImpl::recv()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#a003de04deb00ecbb19179b3f557df548',1,'mlx::core::distributed::detail::recv()'],['../namespacemlx_1_1core_1_1distributed.html#af93c1680b656e98158d5f6eed8e092e8',1,'mlx::core::distributed::recv(Shape shape, Dtype dtype, int src, std::optional< Group > group=std::nullopt, StreamOrDevice s={})']]], + ['recv_5flike_66',['recv_like',['../namespacemlx_1_1core_1_1distributed.html#a2822b78bce2c679e6ff940b2fca944f0',1,'mlx::core::distributed']]], + ['reduce_67',['Reduce',['../classmlx_1_1core_1_1_reduce.html',1,'mlx::core::Reduce'],['../classmlx_1_1core_1_1_reduce.html#a055368c1d036fb953a23ef230e33dcbf',1,'mlx::core::Reduce::Reduce()']]], + ['reduce_68',['reduce',['../namespacemlx_1_1core_1_1metal.html#abb997ccbed4c9a9ccd975b1574755fca',1,'mlx::core::metal']]], + ['reduce_2eh_69',['reduce.h',['../common_2reduce_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2reduce_8h.html',1,'(Global Namespace)'],['../metal_2reduce_8h.html',1,'(Global Namespace)']]], + ['reduce_5fall_2eh_70',['reduce_all.h',['../reduce__all_8h.html',1,'']]], + ['reduce_5fcol_2eh_71',['reduce_col.h',['../reduce__col_8h.html',1,'']]], + ['reduce_5finit_2eh_72',['reduce_init.h',['../reduce__init_8h.html',1,'']]], + ['reduce_5fn_5freads_73',['REDUCE_N_READS',['../defines_8h.html#a2ad505864a2ab786147766900bc18c21',1,'defines.h']]], + ['reduce_5fn_5fwrites_74',['REDUCE_N_WRITES',['../defines_8h.html#a68c33274e15a2f163f7631a36280d82f',1,'defines.h']]], + ['reduce_5frow_2eh_75',['reduce_row.h',['../reduce__row_8h.html',1,'']]], + ['reduce_5futils_76',['reduce_utils',['../namespacemlx_1_1core_1_1metal.html#a2ec39572806310cf528aea06530e8af8',1,'mlx::core::metal']]], + ['reduce_5futils_2eh_77',['reduce_utils.h',['../reduce__utils_8h.html',1,'']]], + ['reducetype_78',['ReduceType',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924',1,'mlx::core::distributed::AllReduce::ReduceType'],['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3',1,'mlx::core::ArgReduce::ReduceType'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9',1,'mlx::core::Reduce::ReduceType'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1',1,'mlx::core::Scan::ReduceType'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613c',1,'mlx::core::Scatter::ReduceType'],['../classmlx_1_1core_1_1_scatter_axis.html#aa292e6cb2a4b32c42ad4f7a258b334f2',1,'mlx::core::ScatterAxis::ReduceType']]], ['reductionoptype_79',['ReductionOpType',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65',1,'mlx::core']]], ['reductionplan_80',['ReductionPlan',['../structmlx_1_1core_1_1_reduction_plan.html',1,'mlx::core::ReductionPlan'],['../structmlx_1_1core_1_1_reduction_plan.html#a07d9eb40a259918ce23360416b3e9db8',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_, Shape shape_, Strides strides_)'],['../structmlx_1_1core_1_1_reduction_plan.html#aec7496f3740a0b0d51aaa606f6fd68f4',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_)']]], ['reference_81',['reference',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a44e2e1f29191c20ec4390de4fa0bd59f',1,'mlx::core::array::ArrayIterator']]], ['register_5flibrary_82',['register_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a45945f2efcd242d915ffa2171e92bf9d',1,'mlx::core::metal::Device::register_library(const std::string &lib_name, const std::string &lib_path)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a99ff72689b7beb65ad4541391b0eeabf',1,'mlx::core::metal::Device::register_library(const std::string &lib_name)']]], - ['remainder_83',['Remainder',['../structmlx_1_1core_1_1detail_1_1_remainder.html',1,'mlx::core::detail::Remainder'],['../classmlx_1_1core_1_1_remainder.html',1,'mlx::core::Remainder'],['../struct_remainder.html',1,'Remainder'],['../classmlx_1_1core_1_1_remainder.html#a4f3eada4a21898af4a77d1d27ce14641',1,'mlx::core::Remainder::Remainder()']]], - ['remainder_84',['remainder',['../group__ops.html#ga99f5c904f724156a814d7817188351d2',1,'mlx::core']]], - ['remaining_85',['remaining',['../classpocketfft_1_1detail_1_1multi__iter.html#a034d12f842df90e6471dffd3fa6ba4bd',1,'pocketfft::detail::multi_iter::remaining()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a9267d37f51a9a5aecc69293c7ed1b1f6',1,'pocketfft::detail::simple_iter::remaining()'],['../classpocketfft_1_1detail_1_1rev__iter.html#a143637135c441a4b9a2959c2370d8c63',1,'pocketfft::detail::rev_iter::remaining()']]], - ['repeat_86',['repeat',['../group__ops.html#gab49e3a687e826554ed1574186e8ae974',1,'mlx::core::repeat(const array &arr, int repeats, int axis, StreamOrDevice s={})'],['../group__ops.html#ga4f75f5d5db999f02f43ecbc6dccf3ba6',1,'mlx::core::repeat(const array &arr, int repeats, StreamOrDevice s={})']]], - ['reset_87',['reset',['../structmlx_1_1core_1_1_contiguous_iterator.html#afa2e2bde9bfa57ac759bc7f5b881262a',1,'mlx::core::ContiguousIterator']]], - ['reset_5fpeak_5fmemory_88',['reset_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a26b9c8ac7ed56c3bb7ddc194009ec5a6',1,'mlx::core::metal::MetalAllocator::reset_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#adec8bb375da6c9dd5ff625a3a8434122',1,'mlx::core::metal::reset_peak_memory()']]], - ['reshape_89',['Reshape',['../classmlx_1_1core_1_1_reshape.html',1,'mlx::core::Reshape'],['../classmlx_1_1core_1_1_reshape.html#aa5a5d520b6ec6c8d9ba9d79808e36312',1,'mlx::core::Reshape::Reshape()']]], - ['reshape_90',['reshape',['../group__ops.html#ga084f03ce2b22258afb7c8b45e17af828',1,'mlx::core']]], - ['residencyset_91',['ResidencySet',['../classmlx_1_1core_1_1metal_1_1_residency_set.html',1,'mlx::core::metal::ResidencySet'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#abb69d020da017a7e52e9e3903b877eec',1,'mlx::core::metal::ResidencySet::ResidencySet(MTL::Device *d)'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aabbf8c16f269f38e4c38097b947d18b7',1,'mlx::core::metal::ResidencySet::ResidencySet(const ResidencySet &)=delete']]], - ['resident_2eh_92',['resident.h',['../resident_8h.html',1,'']]], - ['resize_93',['resize',['../classpocketfft_1_1detail_1_1arr.html#a8d73baaefa02dff8714e4398c83917e0',1,'pocketfft::detail::arr::resize()'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#a0364647bca4324ac41ea3900925a69b5',1,'mlx::core::metal::ResidencySet::resize()']]], - ['restart_94',['restart',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288',1,'pocketfft::detail::threading::thread_pool']]], - ['result_5ftype_95',['result_type',['../namespacemlx_1_1core.html#a8b984eef832f757e28cd262d64a49ae7',1,'mlx::core::result_type(const array &a, const array &b)'],['../namespacemlx_1_1core.html#ac457c232f956ba802acb69c5a621633d',1,'mlx::core::result_type(const array &a, const array &b, const array &c)'],['../namespacemlx_1_1core.html#aafaf24a28297428caf6d0c36c623489e',1,'mlx::core::result_type(const std::vector< array > &arrays)']]], - ['retain_5fgraph_96',['retain_graph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a12ead93cb70ebab865c5e9ce7718f814',1,'mlx::core::detail::RetainGraph::retain_graph()'],['../namespacemlx_1_1core_1_1detail.html#a38af45eb92e437207c722a088f381cd3',1,'mlx::core::detail::retain_graph()']]], - ['retaingraph_97',['RetainGraph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html',1,'mlx::core::detail::RetainGraph'],['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a7fac0244c14cc9e8f580bc1298ff68da',1,'mlx::core::detail::RetainGraph::RetainGraph()']]], - ['rev_5fiter_98',['rev_iter',['../classpocketfft_1_1detail_1_1rev__iter.html',1,'pocketfft::detail::rev_iter'],['../classpocketfft_1_1detail_1_1rev__iter.html#af7b8c2f1534d3038ba2a3c6b9919e134',1,'pocketfft::detail::rev_iter::rev_iter(const arr_info &arr_, const shape_t &axes)']]], - ['rev_5fofs_99',['rev_ofs',['../classpocketfft_1_1detail_1_1rev__iter.html#a7f112afa76cb7a4c29cff217a6f5f5a9',1,'pocketfft::detail::rev_iter']]], - ['rfft_100',['rfft',['../namespacemlx_1_1core_1_1fft.html#a9cb0edfb831b1ed607a8124d38540c13',1,'mlx::core::fft::rfft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a464016cbc948bb3af17d43ce39cf54bd',1,'mlx::core::fft::rfft(const array &a, int axis=-1, StreamOrDevice s={})']]], - ['rfft2_101',['rfft2',['../namespacemlx_1_1core_1_1fft.html#a99397f5d9de6551f967120546ec96728',1,'mlx::core::fft::rfft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a59ca0c3c455e4ff1fed3dbd2327c55f0',1,'mlx::core::fft::rfft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], - ['rfftn_102',['rfftn',['../namespacemlx_1_1core_1_1fft.html#ab60d121ff5509c5a144b2fab7ae0f93b',1,'mlx::core::fft::rfftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#ab502e092ba4bb571ecc421a25e4cb968',1,'mlx::core::fft::rfftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a53d44fd9b6c7645f9303c24099755bf2',1,'mlx::core::fft::rfftn(const array &a, StreamOrDevice s={})']]], - ['rfftp_103',['rfftp',['../classpocketfft_1_1detail_1_1rfftp.html',1,'pocketfft::detail::rfftp< T0 >'],['../classpocketfft_1_1detail_1_1rfftp.html#a0c590f917b8e8afa3ff53ccff52e68c5',1,'pocketfft::detail::rfftp::rfftp()']]], - ['right_5fshift_104',['right_shift',['../group__ops.html#gafa376ad57d38ba87378f0272dc379b23',1,'mlx::core']]], - ['rightshift_105',['RightShift',['../structmlx_1_1core_1_1detail_1_1_right_shift.html',1,'mlx::core::detail::RightShift'],['../struct_right_shift.html',1,'RightShift'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23da011e7b275a1f0edbd9345cfcf6501503',1,'mlx::core::BitwiseBinary::RightShift']]], - ['rint_106',['rint',['../namespacemetal.html#a29ab6060527120eee745aec0daa06e01',1,'metal::rint()'],['../namespacemetal_1_1fast.html#aa613bc252f8d8069e175ec9e9d05a7ec',1,'metal::fast::rint()'],['../namespacemetal_1_1precise.html#ab17bd408098270ad92f37bcd1039c254',1,'metal::precise::rint()']]], - ['rms_5flooped_5flimit_107',['RMS_LOOPED_LIMIT',['../defines_8h.html#a717a175676c3f96d74adfde7e751a541',1,'defines.h']]], - ['rms_5fn_5freads_108',['RMS_N_READS',['../defines_8h.html#a89c0a33ba39a881ad3458ffdde62a24f',1,'defines.h']]], - ['rms_5fnorm_109',['rms_norm',['../namespacemlx_1_1core_1_1fast.html#ac7b620275c6386f822b7aacc6b312e62',1,'mlx::core::fast']]], - ['rmsnorm_110',['RMSNorm',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html',1,'mlx::core::fast::RMSNorm'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a22adaff0749711263388ec151fcfebe2',1,'mlx::core::fast::RMSNorm::RMSNorm()']]], - ['rmsnormvjp_111',['RMSNormVJP',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html',1,'mlx::core::fast::RMSNormVJP'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#aac060129b2e1af79bf388bfe705381ca',1,'mlx::core::fast::RMSNormVJP::RMSNormVJP()']]], - ['roll_112',['roll',['../group__ops.html#gac40e48c69f9c715a767912c30836e75c',1,'mlx::core::roll(const array &a, int shift, StreamOrDevice s={})'],['../group__ops.html#ga5011d1a5735c64e5b91afa56c7e2cc02',1,'mlx::core::roll(const array &a, const Shape &shift, StreamOrDevice s={})'],['../group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436',1,'mlx::core::roll(const array &a, int shift, int axis, StreamOrDevice s={})'],['../group__ops.html#ga665f502ecc96f1f4467556b784abf9ae',1,'mlx::core::roll(const array &a, int shift, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga79137f90bc44ac9e35f408c012701df9',1,'mlx::core::roll(const array &a, const Shape &shift, int axis, StreamOrDevice s={})'],['../group__ops.html#ga9d76930fb567a7d459ff96fb851abe36',1,'mlx::core::roll(const array &a, const Shape &shift, const std::vector< int > &axes, StreamOrDevice s={})']]], - ['rope_113',['RoPE',['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html',1,'mlx::core::fast::RoPE'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a60b399d7f38c0f5f50342a6b97f0eb1a',1,'mlx::core::fast::RoPE::RoPE()']]], - ['rope_114',['rope',['../namespacemlx_1_1core_1_1fast.html#a534ef357eae24892684a6ecd866d3fab',1,'mlx::core::fast::rope(const array &x, int dims, bool traditional, std::optional< float > base, float scale, int offset, const std::optional< array > &freqs=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fast.html#a1632b78950f0c8c31b24be7d80faeb39',1,'mlx::core::fast::rope(const array &x, int dims, bool traditional, std::optional< float > base, float scale, const array &offset, const std::optional< array > &freqs=std::nullopt, StreamOrDevice s={})']]], - ['rot90_115',['ROT90',['../namespacepocketfft_1_1detail.html#a928bad5278df636ee47402c0a75f64ef',1,'pocketfft::detail']]], - ['rotx90_116',['ROTX90',['../namespacepocketfft_1_1detail.html#ab6a43dc0cec4291e163e68a0875ac501',1,'pocketfft::detail']]], - ['round_117',['Round',['../structmlx_1_1core_1_1detail_1_1_round.html',1,'mlx::core::detail::Round'],['../classmlx_1_1core_1_1_round.html',1,'mlx::core::Round'],['../struct_round.html',1,'Round'],['../classmlx_1_1core_1_1_round.html#a1327a359b2aed91f576145a0e70d1dde',1,'mlx::core::Round::Round()']]], - ['round_118',['round',['../namespacemetal.html#a46c667e169ff9d51a9204a045305442f',1,'metal::round()'],['../namespacemetal_1_1fast.html#a4cb687257a004726d49e496417eaa40f',1,'metal::fast::round()'],['../namespacemetal_1_1precise.html#a5295ab08055d12534cc3775da855ac12',1,'metal::precise::round()'],['../group__ops.html#ga2d74d43f007a069384e89d8416525331',1,'mlx::core::round(const array &a, int decimals, StreamOrDevice s={})'],['../group__ops.html#gaf18fb7e98bf8cf3b7fbc5e64c988a95b',1,'mlx::core::round(const array &a, StreamOrDevice s={})']]], - ['round_5ferror_119',['round_error',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#afa223448fa4f04c1113a85345dd720c3',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['row_5fbin_5fop_120',['row_bin_op',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a318c4279bdc7b39b7919f108b1cd8010',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::row_bin_op()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a3d0d5b9c7962658cc6d5afbbbb2f19e2',1,'mlx::steel::MMATile::row_bin_op()']]], - ['row_5fcontiguous_121',['row_contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#a3170fa381dc7a90f6eabcc029bdf9bfd',1,'mlx::core::array::Flags::row_contiguous'],['../struct_indices.html#a255e340a39c6ac28ef2c232b106f85d1',1,'Indices::row_contiguous']]], - ['row_5freduce_122',['row_reduce',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a51d662e4cff88b5ad17d7c44bb6b6970',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::row_reduce()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa0ad5cb750ace934bf230385d8bd9f88',1,'mlx::steel::MMATile::row_reduce()']]], - ['row_5freduce_5fgeneral_5fdispatch_123',['row_reduce_general_dispatch',['../namespacemlx_1_1core.html#ab1eeca8ec6fa31819ee108fa6ed2c41b',1,'mlx::core']]], - ['row_5freduce_5flooped_124',['row_reduce_looped',['../reduce__row_8h.html#a72611b8006ae5642b69f4d250d69865b',1,'reduce_row.h']]], - ['row_5freduce_5fsimple_125',['row_reduce_simple',['../reduce__row_8h.html#a7dbd7cc81b1ce5a4271d56b99ce595d4',1,'reduce_row.h']]], - ['row_5freduce_5fsmall_126',['row_reduce_small',['../reduce__row_8h.html#a1e1b59fa73d2b0be978494a759f2c6ee',1,'reduce_row.h']]], - ['rsqrt_127',['Rsqrt',['../structmlx_1_1core_1_1detail_1_1_rsqrt.html',1,'mlx::core::detail::Rsqrt'],['../struct_rsqrt.html',1,'Rsqrt']]], - ['rsqrt_128',['rsqrt',['../namespacemetal.html#a1cf4b605c0aa7ff5bfe5e979a16f5157',1,'metal::rsqrt()'],['../namespacemetal_1_1fast.html#aa62097c750f1e4b69d09277f19976ab1',1,'metal::fast::rsqrt()'],['../namespacemetal_1_1precise.html#afb397b477745f12a44423934fa2b05ac',1,'metal::precise::rsqrt()'],['../group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1',1,'mlx::core::rsqrt()']]], - ['run_129',['run',['../struct_g_e_m_v_kernel.html#ac4a7b5011a0ea938ab1949bb1767fc1a',1,'GEMVKernel::run()'],['../struct_g_e_m_v_t_kernel.html#a5d68656832de892f33db939005713927',1,'GEMVTKernel::run()'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run(const device T *A, const device T *B, device U *D, const constant GEMMParams *params, threadgroup T *As, threadgroup T *Bs, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run(const device T *A, const device T *B, device U *D, const constant GEMMParams *params, threadgroup T *As, threadgroup T *Bs, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)']]] + ['register_5foutput_5farray_83',['register_output_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ada20558738968ca2ecdcd95f228e028a',1,'mlx::core::metal::CommandEncoder::register_output_array()'],['../structmlx_1_1core_1_1_command_encoder.html#ada20558738968ca2ecdcd95f228e028a',1,'mlx::core::CommandEncoder::register_output_array()']]], + ['remainder_84',['Remainder',['../structmlx_1_1core_1_1detail_1_1_remainder.html',1,'mlx::core::detail::Remainder'],['../classmlx_1_1core_1_1_remainder.html',1,'mlx::core::Remainder'],['../struct_remainder.html',1,'Remainder'],['../classmlx_1_1core_1_1_remainder.html#a4f3eada4a21898af4a77d1d27ce14641',1,'mlx::core::Remainder::Remainder()']]], + ['remainder_85',['remainder',['../namespacemlx_1_1core_1_1simd.html#ac66bdf1a8e86a4d350c85037bc764da5',1,'mlx::core::simd::remainder(Simd< float16_t, N > x, Simd< float16_t, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ab020d2c434fad0cdf79fd37b0f6c1676',1,'mlx::core::simd::remainder(Simd< T, N > a, Simd< T, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a54c7f2f2b995eb767462b1228982967f',1,'mlx::core::simd::remainder(Simd< T, 1 > a_, Simd< T, 1 > b_)'],['../group__ops.html#ga99f5c904f724156a814d7817188351d2',1,'mlx::core::remainder()']]], + ['remaining_86',['remaining',['../classpocketfft_1_1detail_1_1multi__iter.html#a034d12f842df90e6471dffd3fa6ba4bd',1,'pocketfft::detail::multi_iter::remaining()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a9267d37f51a9a5aecc69293c7ed1b1f6',1,'pocketfft::detail::simple_iter::remaining()'],['../classpocketfft_1_1detail_1_1rev__iter.html#a143637135c441a4b9a2959c2370d8c63',1,'pocketfft::detail::rev_iter::remaining()']]], + ['repeat_87',['repeat',['../group__ops.html#gab49e3a687e826554ed1574186e8ae974',1,'mlx::core::repeat(const array &arr, int repeats, int axis, StreamOrDevice s={})'],['../group__ops.html#ga4f75f5d5db999f02f43ecbc6dccf3ba6',1,'mlx::core::repeat(const array &arr, int repeats, StreamOrDevice s={})']]], + ['reset_88',['reset',['../structmlx_1_1core_1_1_contiguous_iterator.html#afa2e2bde9bfa57ac759bc7f5b881262a',1,'mlx::core::ContiguousIterator']]], + ['reset_5fpeak_5fmemory_89',['reset_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a26b9c8ac7ed56c3bb7ddc194009ec5a6',1,'mlx::core::metal::MetalAllocator::reset_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#adec8bb375da6c9dd5ff625a3a8434122',1,'mlx::core::metal::reset_peak_memory()']]], + ['reshape_90',['Reshape',['../classmlx_1_1core_1_1_reshape.html',1,'mlx::core::Reshape'],['../classmlx_1_1core_1_1_reshape.html#aa5a5d520b6ec6c8d9ba9d79808e36312',1,'mlx::core::Reshape::Reshape()']]], + ['reshape_91',['reshape',['../group__ops.html#ga084f03ce2b22258afb7c8b45e17af828',1,'mlx::core']]], + ['residencyset_92',['ResidencySet',['../classmlx_1_1core_1_1metal_1_1_residency_set.html',1,'mlx::core::metal::ResidencySet'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#abb69d020da017a7e52e9e3903b877eec',1,'mlx::core::metal::ResidencySet::ResidencySet(MTL::Device *d)'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aabbf8c16f269f38e4c38097b947d18b7',1,'mlx::core::metal::ResidencySet::ResidencySet(const ResidencySet &)=delete']]], + ['resident_2eh_93',['resident.h',['../resident_8h.html',1,'']]], + ['resize_94',['resize',['../classpocketfft_1_1detail_1_1arr.html#a8d73baaefa02dff8714e4398c83917e0',1,'pocketfft::detail::arr::resize()'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#a0364647bca4324ac41ea3900925a69b5',1,'mlx::core::metal::ResidencySet::resize()'],['../class_thread_pool.html#a33d9a848213206e95997eb050702ecbf',1,'ThreadPool::resize()']]], + ['restart_95',['restart',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288',1,'pocketfft::detail::threading::thread_pool']]], + ['result_5ftype_96',['result_type',['../namespacemlx_1_1core.html#a8b984eef832f757e28cd262d64a49ae7',1,'mlx::core::result_type(const array &a, const array &b)'],['../namespacemlx_1_1core.html#ac457c232f956ba802acb69c5a621633d',1,'mlx::core::result_type(const array &a, const array &b, const array &c)'],['../namespacemlx_1_1core.html#aafaf24a28297428caf6d0c36c623489e',1,'mlx::core::result_type(const std::vector< array > &arrays)']]], + ['retain_5fgraph_97',['retain_graph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a12ead93cb70ebab865c5e9ce7718f814',1,'mlx::core::detail::RetainGraph::retain_graph()'],['../namespacemlx_1_1core_1_1detail.html#a38af45eb92e437207c722a088f381cd3',1,'mlx::core::detail::retain_graph()']]], + ['retaingraph_98',['RetainGraph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html',1,'mlx::core::detail::RetainGraph'],['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a7fac0244c14cc9e8f580bc1298ff68da',1,'mlx::core::detail::RetainGraph::RetainGraph()']]], + ['rev_5fiter_99',['rev_iter',['../classpocketfft_1_1detail_1_1rev__iter.html',1,'pocketfft::detail::rev_iter'],['../classpocketfft_1_1detail_1_1rev__iter.html#af7b8c2f1534d3038ba2a3c6b9919e134',1,'pocketfft::detail::rev_iter::rev_iter(const arr_info &arr_, const shape_t &axes)']]], + ['rev_5fofs_100',['rev_ofs',['../classpocketfft_1_1detail_1_1rev__iter.html#a7f112afa76cb7a4c29cff217a6f5f5a9',1,'pocketfft::detail::rev_iter']]], + ['rfft_101',['rfft',['../namespacemlx_1_1core_1_1fft.html#a9cb0edfb831b1ed607a8124d38540c13',1,'mlx::core::fft::rfft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a464016cbc948bb3af17d43ce39cf54bd',1,'mlx::core::fft::rfft(const array &a, int axis=-1, StreamOrDevice s={})']]], + ['rfft2_102',['rfft2',['../namespacemlx_1_1core_1_1fft.html#a99397f5d9de6551f967120546ec96728',1,'mlx::core::fft::rfft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a59ca0c3c455e4ff1fed3dbd2327c55f0',1,'mlx::core::fft::rfft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], + ['rfftn_103',['rfftn',['../namespacemlx_1_1core_1_1fft.html#ab60d121ff5509c5a144b2fab7ae0f93b',1,'mlx::core::fft::rfftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#ab502e092ba4bb571ecc421a25e4cb968',1,'mlx::core::fft::rfftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a53d44fd9b6c7645f9303c24099755bf2',1,'mlx::core::fft::rfftn(const array &a, StreamOrDevice s={})']]], + ['rfftp_104',['rfftp',['../classpocketfft_1_1detail_1_1rfftp.html',1,'pocketfft::detail::rfftp< T0 >'],['../classpocketfft_1_1detail_1_1rfftp.html#a0c590f917b8e8afa3ff53ccff52e68c5',1,'pocketfft::detail::rfftp::rfftp()']]], + ['right_5fshift_105',['right_shift',['../group__ops.html#gafa376ad57d38ba87378f0272dc379b23',1,'mlx::core']]], + ['rightshift_106',['RightShift',['../structmlx_1_1core_1_1detail_1_1_right_shift.html',1,'mlx::core::detail::RightShift'],['../struct_right_shift.html',1,'RightShift'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23da011e7b275a1f0edbd9345cfcf6501503',1,'mlx::core::BitwiseBinary::RightShift']]], + ['ring_2eh_107',['ring.h',['../ring_8h.html',1,'']]], + ['rint_108',['rint',['../namespacemlx_1_1core_1_1simd.html#a400d89d040f43d471b306a8e8bdb3974',1,'mlx::core::simd::rint(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a797196eccc3690aac5c45e5f9c804ceb',1,'mlx::core::simd::rint(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a8c200919c0eeefb2e2e5d9d19741a805',1,'mlx::core::simd::rint(Simd< float16_t, N > a)'],['../namespacemetal.html#a29ab6060527120eee745aec0daa06e01',1,'metal::rint()'],['../namespacemetal_1_1fast.html#aa613bc252f8d8069e175ec9e9d05a7ec',1,'metal::fast::rint()'],['../namespacemetal_1_1precise.html#ab17bd408098270ad92f37bcd1039c254',1,'metal::precise::rint()']]], + ['rms_5flooped_5flimit_109',['RMS_LOOPED_LIMIT',['../defines_8h.html#a717a175676c3f96d74adfde7e751a541',1,'defines.h']]], + ['rms_5fn_5freads_110',['RMS_N_READS',['../defines_8h.html#a89c0a33ba39a881ad3458ffdde62a24f',1,'defines.h']]], + ['rms_5fnorm_111',['rms_norm',['../namespacemlx_1_1core_1_1fast.html#ac7b620275c6386f822b7aacc6b312e62',1,'mlx::core::fast']]], + ['rmsnorm_112',['RMSNorm',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html',1,'mlx::core::fast::RMSNorm'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a22adaff0749711263388ec151fcfebe2',1,'mlx::core::fast::RMSNorm::RMSNorm()']]], + ['rmsnormvjp_113',['RMSNormVJP',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html',1,'mlx::core::fast::RMSNormVJP'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#aac060129b2e1af79bf388bfe705381ca',1,'mlx::core::fast::RMSNormVJP::RMSNormVJP()']]], + ['roll_114',['roll',['../group__ops.html#gac40e48c69f9c715a767912c30836e75c',1,'mlx::core::roll(const array &a, int shift, StreamOrDevice s={})'],['../group__ops.html#ga5011d1a5735c64e5b91afa56c7e2cc02',1,'mlx::core::roll(const array &a, const Shape &shift, StreamOrDevice s={})'],['../group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436',1,'mlx::core::roll(const array &a, int shift, int axis, StreamOrDevice s={})'],['../group__ops.html#ga665f502ecc96f1f4467556b784abf9ae',1,'mlx::core::roll(const array &a, int shift, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga79137f90bc44ac9e35f408c012701df9',1,'mlx::core::roll(const array &a, const Shape &shift, int axis, StreamOrDevice s={})'],['../group__ops.html#ga9d76930fb567a7d459ff96fb851abe36',1,'mlx::core::roll(const array &a, const Shape &shift, const std::vector< int > &axes, StreamOrDevice s={})']]], + ['rope_115',['RoPE',['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html',1,'mlx::core::fast::RoPE'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a60b399d7f38c0f5f50342a6b97f0eb1a',1,'mlx::core::fast::RoPE::RoPE()']]], + ['rope_116',['rope',['../namespacemlx_1_1core_1_1fast.html#a534ef357eae24892684a6ecd866d3fab',1,'mlx::core::fast::rope(const array &x, int dims, bool traditional, std::optional< float > base, float scale, int offset, const std::optional< array > &freqs=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fast.html#a1632b78950f0c8c31b24be7d80faeb39',1,'mlx::core::fast::rope(const array &x, int dims, bool traditional, std::optional< float > base, float scale, const array &offset, const std::optional< array > &freqs=std::nullopt, StreamOrDevice s={})']]], + ['rot90_117',['ROT90',['../namespacepocketfft_1_1detail.html#a928bad5278df636ee47402c0a75f64ef',1,'pocketfft::detail']]], + ['rotx90_118',['ROTX90',['../namespacepocketfft_1_1detail.html#ab6a43dc0cec4291e163e68a0875ac501',1,'pocketfft::detail']]], + ['round_119',['Round',['../structmlx_1_1core_1_1detail_1_1_round.html',1,'mlx::core::detail::Round'],['../classmlx_1_1core_1_1_round.html',1,'mlx::core::Round'],['../struct_round.html',1,'Round'],['../classmlx_1_1core_1_1_round.html#a1327a359b2aed91f576145a0e70d1dde',1,'mlx::core::Round::Round()']]], + ['round_120',['round',['../namespacemetal.html#a46c667e169ff9d51a9204a045305442f',1,'metal::round()'],['../namespacemetal_1_1fast.html#a4cb687257a004726d49e496417eaa40f',1,'metal::fast::round()'],['../namespacemetal_1_1precise.html#a5295ab08055d12534cc3775da855ac12',1,'metal::precise::round()'],['../group__ops.html#ga2d74d43f007a069384e89d8416525331',1,'mlx::core::round(const array &a, int decimals, StreamOrDevice s={})'],['../group__ops.html#gaf18fb7e98bf8cf3b7fbc5e64c988a95b',1,'mlx::core::round(const array &a, StreamOrDevice s={})']]], + ['round_5ferror_121',['round_error',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#afa223448fa4f04c1113a85345dd720c3',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['row_5fbin_5fop_122',['row_bin_op',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a318c4279bdc7b39b7919f108b1cd8010',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::row_bin_op()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a3d0d5b9c7962658cc6d5afbbbb2f19e2',1,'mlx::steel::MMATile::row_bin_op()']]], + ['row_5fcontiguous_123',['row_contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#a3170fa381dc7a90f6eabcc029bdf9bfd',1,'mlx::core::array::Flags::row_contiguous'],['../struct_indices.html#a255e340a39c6ac28ef2c232b106f85d1',1,'Indices::row_contiguous']]], + ['row_5ffrag_5ftype_124',['row_frag_type',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a3dcd4301390937f89ed1dde6d28e341f',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >']]], + ['row_5freduce_125',['row_reduce',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a51d662e4cff88b5ad17d7c44bb6b6970',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::row_reduce()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa0ad5cb750ace934bf230385d8bd9f88',1,'mlx::steel::MMATile::row_reduce()']]], + ['row_5freduce_5fgeneral_5fdispatch_126',['row_reduce_general_dispatch',['../namespacemlx_1_1core.html#ab1eeca8ec6fa31819ee108fa6ed2c41b',1,'mlx::core']]], + ['row_5freduce_5flooped_127',['row_reduce_looped',['../reduce__row_8h.html#a72611b8006ae5642b69f4d250d69865b',1,'reduce_row.h']]], + ['row_5freduce_5fsimple_128',['row_reduce_simple',['../reduce__row_8h.html#a7dbd7cc81b1ce5a4271d56b99ce595d4',1,'reduce_row.h']]], + ['row_5freduce_5fsmall_129',['row_reduce_small',['../reduce__row_8h.html#a1e1b59fa73d2b0be978494a759f2c6ee',1,'reduce_row.h']]], + ['rsqrt_130',['Rsqrt',['../structmlx_1_1core_1_1detail_1_1_rsqrt.html',1,'mlx::core::detail::Rsqrt'],['../struct_rsqrt.html',1,'Rsqrt']]], + ['rsqrt_131',['rsqrt',['../namespacemlx_1_1core_1_1simd.html#aea75ddf8c696efc2e5e924667ed48e70',1,'mlx::core::simd::rsqrt(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a74ac0fd799967b0f303bfd26fc6a17cf',1,'mlx::core::simd::rsqrt(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a3345cb53830d1afd625acc7bdc3a0435',1,'mlx::core::simd::rsqrt(Simd< float16_t, N > a)'],['../namespacemetal.html#a1cf4b605c0aa7ff5bfe5e979a16f5157',1,'metal::rsqrt()'],['../namespacemetal_1_1fast.html#aa62097c750f1e4b69d09277f19976ab1',1,'metal::fast::rsqrt()'],['../namespacemetal_1_1precise.html#afb397b477745f12a44423934fa2b05ac',1,'metal::precise::rsqrt()'],['../group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1',1,'mlx::core::rsqrt()']]], + ['run_132',['run',['../struct_g_e_m_v_kernel.html#ac4a7b5011a0ea938ab1949bb1767fc1a',1,'GEMVKernel::run()'],['../struct_g_e_m_v_t_kernel.html#a5d68656832de892f33db939005713927',1,'GEMVTKernel::run()'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run(const device T *A, const device T *B, device U *D, const constant GEMMParams *params, threadgroup T *As, threadgroup T *Bs, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run(const device T *A, const device T *B, device U *D, const constant GEMMParams *params, threadgroup T *As, threadgroup T *Bs, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)']]] ]; diff --git a/docs/build/html/search/all_13.js b/docs/build/html/search/all_13.js index e151b5277..782a2aa8f 100644 --- a/docs/build/html/search/all_13.js +++ b/docs/build/html/search/all_13.js @@ -6,207 +6,228 @@ var searchData= ['save_5fgguf_3',['save_gguf',['../namespacemlx_1_1core.html#a8bcc29ca8846ec99dce333df4a34dc5f',1,'mlx::core']]], ['save_5fsafetensors_4',['save_safetensors',['../namespacemlx_1_1core.html#a9f158db20c2405557f3ebc397e876de8',1,'mlx::core::save_safetensors(std::shared_ptr< io::Writer > in_stream, std::unordered_map< std::string, array >, std::unordered_map< std::string, std::string > metadata={})'],['../namespacemlx_1_1core.html#a21e256d852d587bcdc0827831b2c5c16',1,'mlx::core::save_safetensors(std::string file, std::unordered_map< std::string, array >, std::unordered_map< std::string, std::string > metadata={})']]], ['scalar_5',['Scalar',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337af60357a8d17e45793298323f1b372a74',1,'mlx::core']]], - ['scale_6',['scale',['../struct_scale_op.html#a02043fac21c68fb8d6863a01f45ede4b',1,'ScaleOp::scale'],['../struct_transform_scale.html#aa56b8e107acf16fdf77006625c2b8bc6',1,'TransformScale::scale'],['../structmlx_1_1steel_1_1_attn_params.html#ad81bcd32e6ff8fec0000eca505fb6826',1,'mlx::steel::AttnParams::scale']]], - ['scaled_5fdot_5fproduct_5fattention_7',['scaled_dot_product_attention',['../namespacemlx_1_1core_1_1fast.html#a3663b50265b0a9c0cca2b5376852e059',1,'mlx::core::fast']]], - ['scaleddotproductattention_8',['ScaledDotProductAttention',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html',1,'mlx::core::fast::ScaledDotProductAttention'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ab3f78d30e5bb3e76cfe701f2358e4748',1,'mlx::core::fast::ScaledDotProductAttention::ScaledDotProductAttention()']]], - ['scaleop_9',['ScaleOp',['../struct_scale_op.html',1,'']]], - ['scales_10',['scales',['../struct_quantized_block_loader.html#a6123e4a9209d6eacb58b2c2344ed1ecf',1,'QuantizedBlockLoader']]], - ['scan_11',['Scan',['../classmlx_1_1core_1_1_scan.html',1,'mlx::core::Scan'],['../classmlx_1_1core_1_1_scan.html#ac93e8f9c6771de825d2186ef34fa7087',1,'mlx::core::Scan::Scan()']]], - ['scan_12',['scan',['../namespacemlx_1_1core_1_1metal.html#a81c2cf124b0803098a54a78f8f6873a6',1,'mlx::core::metal']]], - ['scan_2eh_13',['scan.h',['../scan_8h.html',1,'']]], - ['scatter_14',['Scatter',['../classmlx_1_1core_1_1_scatter.html',1,'mlx::core::Scatter'],['../classmlx_1_1core_1_1_scatter.html#ac9b3eff67389ef9aa820753379ffeaa3',1,'mlx::core::Scatter::Scatter()']]], - ['scatter_15',['scatter',['../namespacemlx_1_1core_1_1metal.html#a32e902c6cd6d35fcc3119ed6685a170f',1,'mlx::core::metal::scatter()'],['../group__ops.html#gad438be8f90bae9d37c6853b8f4225d61',1,'mlx::core::scatter(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gac2c2b379a3ce959dbe1c4a68f112edfe',1,'mlx::core::scatter(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scatter_2eh_16',['scatter.h',['../scatter_8h.html',1,'']]], - ['scatter_5fadd_17',['scatter_add',['../group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174',1,'mlx::core::scatter_add(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gac13318518e5703f1273c5366eb523a5a',1,'mlx::core::scatter_add(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scatter_5fimpl_18',['scatter_impl',['../scatter_8h.html#ab72d4fe1dbd4ae4dc529ee2ec8164fa4',1,'scatter.h']]], - ['scatter_5fkernels_19',['scatter_kernels',['../jit_2indexing_8h.html#a768c949cd650a44c6b402fc1440c1a56',1,'indexing.h']]], - ['scatter_5fmax_20',['scatter_max',['../group__ops.html#ga05881a4157cd113c9392d168a79e6673',1,'mlx::core::scatter_max(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga9adda5f9202bb3486e4d9e1114e3a56f',1,'mlx::core::scatter_max(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scatter_5fmin_21',['scatter_min',['../group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5',1,'mlx::core::scatter_min(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga51fa762a997c243ca7a19e1ed3e83199',1,'mlx::core::scatter_min(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scatter_5fprod_22',['scatter_prod',['../group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0',1,'mlx::core::scatter_prod(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gaf83c53c453faa9083ba27e4b97539339',1,'mlx::core::scatter_prod(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scheduled_23',['scheduled',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078af8a6f8eed2395ab89a758dec434393ae',1,'mlx::core::array']]], - ['scheduler_24',['Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html',1,'mlx::core::scheduler::Scheduler'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a3ae42aed78a2200e9d02776fcd2316ba',1,'mlx::core::scheduler::Scheduler::Scheduler()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a61a74e3628899e66dde600e24a750648',1,'mlx::core::scheduler::Scheduler::Scheduler(const Scheduler &)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ac3f77b7c93220dadd0b3bb2e903b7059',1,'mlx::core::scheduler::Scheduler::Scheduler(Scheduler &&)=delete']]], - ['scheduler_25',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html#ae856e468c2f7c8f8ec672522cc13730b',1,'mlx::core::scheduler']]], - ['scheduler_2eh_26',['scheduler.h',['../scheduler_8h.html',1,'']]], - ['sdpa_5fvector_27',['sdpa_vector',['../sdpa__vector_8h.html#aee8f6015c5b4fc81cde3c5edf30af8af',1,'sdpa_vector.h']]], - ['sdpa_5fvector_2eh_28',['sdpa_vector.h',['../sdpa__vector_8h.html',1,'']]], - ['sdpa_5fvector_5f2pass_5f1_29',['sdpa_vector_2pass_1',['../sdpa__vector_8h.html#adef3c1f2f7ec098b0ef43f5f5bfeab0e',1,'sdpa_vector.h']]], - ['sdpa_5fvector_5f2pass_5f2_30',['sdpa_vector_2pass_2',['../sdpa__vector_8h.html#a1368cf3618a4e03dbf743b3463205efe',1,'sdpa_vector.h']]], - ['seed_31',['seed',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a9f19c5da2031cba50d0ff996924347d8',1,'mlx::core::random::KeySequence::seed()'],['../namespacemlx_1_1core_1_1random.html#ac4ad325b613257306df74595d3d0e23b',1,'mlx::core::random::seed()']]], - ['seek_32',['seek',['../structmlx_1_1core_1_1_contiguous_iterator.html#af08f009e0a72414d274db2ff1b2c7dd5',1,'mlx::core::ContiguousIterator::seek()'],['../classmlx_1_1core_1_1io_1_1_reader.html#acea55078bd39ccaa27a9a36f17a39cd1',1,'mlx::core::io::Reader::seek()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a9c1716dda53aa36faea9c8fb1a3e34d4',1,'mlx::core::io::Writer::seek()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a4434ee18ff8bbf1b4fce670a337b535f',1,'mlx::core::io::ParallelFileReader::seek()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9646f4ea048ae58719daeb588e2de433',1,'mlx::core::io::FileWriter::seek()']]], - ['select_33',['Select',['../structmlx_1_1core_1_1detail_1_1_select.html',1,'mlx::core::detail::Select'],['../classmlx_1_1core_1_1_select.html',1,'mlx::core::Select'],['../struct_select.html',1,'Select'],['../classmlx_1_1core_1_1_select.html#a6f833fe55dd68ad3726bbf9a8f75eec9',1,'mlx::core::Select::Select()']]], - ['send_34',['Send',['../classmlx_1_1core_1_1distributed_1_1_send.html',1,'mlx::core::distributed::Send'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a2481dd876b14d4a13ac466cbca9c4eac',1,'mlx::core::distributed::Send::Send()']]], - ['send_35',['send',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac8472eb2f96d1b14c7e4ccef56268ba0',1,'mlx::core::distributed::detail::GroupImpl::send()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#abf33511660ac71df5fc92f2aad6c6e08',1,'mlx::core::distributed::detail::send()'],['../namespacemlx_1_1core_1_1distributed.html#a5a8360edaa3a528a3927fce4d2cf1777',1,'mlx::core::distributed::send()']]], - ['set_36',['Set',['../structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807',1,'pocketfft::detail::cmplx::Set(T r_, T i_)'],['../structpocketfft_1_1detail_1_1cmplx.html#a447d26b2e07f6e45f29d865e906c0a98',1,'pocketfft::detail::cmplx::Set(T r_)']]], - ['set_5fbytes_37',['set_bytes',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a9c343f791812a45c6c03a5c9f27f74d5',1,'mlx::core::metal::CommandEncoder::set_bytes(const T *v, int n, int idx)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#abc52d18ea87d213c47fd26062c829849',1,'mlx::core::metal::CommandEncoder::set_bytes(const T &v, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a9c343f791812a45c6c03a5c9f27f74d5',1,'mlx::core::CommandEncoder::set_bytes(const T *v, int n, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#abc52d18ea87d213c47fd26062c829849',1,'mlx::core::CommandEncoder::set_bytes(const T &v, int idx)']]], - ['set_5fcache_5flimit_38',['set_cache_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#af392bced29d9e4e3f1a7cc4725d83764',1,'mlx::core::metal::MetalAllocator::set_cache_limit()'],['../namespacemlx_1_1core_1_1metal.html#ab09c9b60f1e886ab859e6a066c9a5b9d',1,'mlx::core::metal::set_cache_limit()']]], - ['set_5fcompile_5fmode_39',['set_compile_mode',['../namespacemlx_1_1core.html#a49445a55f976c4397f25ea18e1e92bef',1,'mlx::core']]], - ['set_5fcompute_5fpipeline_5fstate_40',['set_compute_pipeline_state',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a6d4c03a6585deedb5ccd1a1057d0c6ef',1,'mlx::core::metal::CommandEncoder::set_compute_pipeline_state()'],['../structmlx_1_1core_1_1_command_encoder.html#a6d4c03a6585deedb5ccd1a1057d0c6ef',1,'mlx::core::CommandEncoder::set_compute_pipeline_state()']]], - ['set_5fdata_41',['set_data',['../classmlx_1_1core_1_1array.html#af9e3a02b4c0023c36248dc75c887214f',1,'mlx::core::array::set_data(allocator::Buffer buffer, Deleter d=allocator::free)'],['../classmlx_1_1core_1_1array.html#a5f338202a39d37fa3f4241e851a15838',1,'mlx::core::array::set_data(allocator::Buffer buffer, size_t data_size, Strides strides, Flags flags, Deleter d=allocator::free)']]], - ['set_5fdefault_5fdevice_42',['set_default_device',['../namespacemlx_1_1core.html#a312a2de41367fe52caeaf8c0f596a120',1,'mlx::core']]], - ['set_5fdefault_5fstream_43',['set_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a6d15314ac9cf25efc9bd1278de9a66bb',1,'mlx::core::scheduler::Scheduler::set_default_stream()'],['../namespacemlx_1_1core.html#af35a2b06517d8bb7dbb469692b4f841c',1,'mlx::core::set_default_stream()']]], - ['set_5finput_5farray_44',['set_input_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ab69ff0d7f14b9b59db4df0608193dce4',1,'mlx::core::metal::CommandEncoder::set_input_array()'],['../structmlx_1_1core_1_1_command_encoder.html#ab69ff0d7f14b9b59db4df0608193dce4',1,'mlx::core::CommandEncoder::set_input_array()']]], - ['set_5fmemory_5flimit_45',['set_memory_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a179e3127ef9377ce54295f771c34ba1b',1,'mlx::core::metal::MetalAllocator::set_memory_limit()'],['../namespacemlx_1_1core_1_1metal.html#a3fb2c4a237fa4bfdff798156146c4937',1,'mlx::core::metal::set_memory_limit()']]], - ['set_5fname_46',['set_name',['../structmlx_1_1core_1_1_node_namer.html#a57a574e48f8a9cd122616d80b138c768',1,'mlx::core::NodeNamer']]], - ['set_5foutput_5farray_47',['set_output_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a6a2e28e542eaa2886041bddd51ff6522',1,'mlx::core::metal::CommandEncoder::set_output_array()'],['../structmlx_1_1core_1_1_command_encoder.html#a6a2e28e542eaa2886041bddd51ff6522',1,'mlx::core::CommandEncoder::set_output_array()']]], - ['set_5fresidency_5fset_48',['set_residency_set',['../classmlx_1_1core_1_1metal_1_1_device.html#a03a2f0c712660a1bd437cb16e4aba79f',1,'mlx::core::metal::Device']]], - ['set_5fsiblings_49',['set_siblings',['../classmlx_1_1core_1_1array.html#a8fccbe7a4edfd8cca168161124e263b1',1,'mlx::core::array']]], - ['set_5fstatus_50',['set_status',['../classmlx_1_1core_1_1array.html#a63598018999b49f1340b183cb303f05c',1,'mlx::core::array']]], - ['set_5ftracer_51',['set_tracer',['../classmlx_1_1core_1_1array.html#af26e6be1a9e6239471a4c24310c0c7c8',1,'mlx::core::array']]], - ['set_5fvalue_52',['set_value',['../classmlx_1_1core_1_1_event.html#a0d077b11f4b28f882b42440b7ac6d40d',1,'mlx::core::Event']]], - ['set_5fvector_5fbytes_53',['set_vector_bytes',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a68c3c6a036e11ec40211c09811bbed1b',1,'mlx::core::metal::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, size_t nelems, int idx)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a7375adf9ee5355bcf4b7f5f210efd115',1,'mlx::core::metal::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a68c3c6a036e11ec40211c09811bbed1b',1,'mlx::core::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, size_t nelems, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a7375adf9ee5355bcf4b7f5f210efd115',1,'mlx::core::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, int idx)']]], - ['set_5fwired_5flimit_54',['set_wired_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a84fa0347da18055bc13ba0a5c4b57253',1,'mlx::core::metal::MetalAllocator::set_wired_limit()'],['../namespacemlx_1_1core_1_1metal.html#a31eab4828d31d292bc84e07b0d961e1e',1,'mlx::core::metal::set_wired_limit()']]], - ['shape_55',['Shape',['../namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416',1,'mlx::core']]], - ['shape_56',['shape',['../structmlx_1_1core_1_1_reduction_plan.html#a1576dc3d2e01b3f1e11816151070dd1a',1,'mlx::core::ReductionPlan::shape'],['../structmlx_1_1steel_1_1_layout2_d.html#a23183747ab1ddbdd3f1fcac6d0faa2cd',1,'mlx::steel::Layout2D::shape'],['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63db720fe0c2abc4b71e22a58a015f8a',1,'mlx::core::fast::CustomKernelShapeInfo::shape'],['../classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0',1,'pocketfft::detail::arr_info::shape() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8',1,'pocketfft::detail::arr_info::shape(size_t i) const'],['../classmlx_1_1core_1_1array.html#a1d06c76b0f3010a5c329d0e9e29e0597',1,'mlx::core::array::shape() const'],['../classmlx_1_1core_1_1array.html#ac50382b652f6e8fbd50d42b7ff595810',1,'mlx::core::array::shape(int dim) const']]], - ['shape2d_57',['Shape2D',['../structmlx_1_1steel_1_1_shape2_d.html',1,'mlx::steel::Shape2D< RInt, CInt >'],['../structmlx_1_1steel_1_1_shape2_d.html#a070ce70eb6d84361c7f313159c438a5c',1,'mlx::steel::Shape2D::Shape2D()']]], - ['shape_5ft_58',['shape_t',['../namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::detail::shape_t'],['../namespacepocketfft.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::shape_t']]], - ['shapeelem_59',['ShapeElem',['../namespacemlx_1_1core.html#a167cdec84c0ae62b5b299c617384346e',1,'mlx::core']]], - ['shapes_60',['shapes',['../struct_indices.html#a5ab170f1a77636180889ddfffd4f7d2f',1,'Indices']]], - ['shapes_5fwithout_5freduction_5faxes_61',['shapes_without_reduction_axes',['../namespacemlx_1_1core.html#a0bea91a360a984e72d2815353f97ee25',1,'mlx::core']]], - ['shared_5fbuffer_5freshape_62',['shared_buffer_reshape',['../namespacemlx_1_1core.html#a88d88987bd8bf3ca46bf3b5e8aacce9d',1,'mlx::core']]], - ['shared_5fbuffer_5fslice_63',['shared_buffer_slice',['../namespacemlx_1_1core.html#a349a9fc2bfd950f679a3fe39b8bdedad',1,'mlx::core']]], - ['shp_64',['shp',['../classpocketfft_1_1detail_1_1arr__info.html#a2467e9e01de1ba4d7cd28c1af783da8d',1,'pocketfft::detail::arr_info']]], - ['shutdown_65',['shutdown',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411',1,'pocketfft::detail::threading::thread_pool']]], - ['siblings_66',['siblings',['../classmlx_1_1core_1_1array.html#acf80fde8f743f65ad5b4be69fcb7a74d',1,'mlx::core::array::siblings() const'],['../classmlx_1_1core_1_1array.html#a7263f23e70a580a9bc2129fbcde36e6c',1,'mlx::core::array::siblings()']]], - ['sigmoid_67',['Sigmoid',['../structmlx_1_1core_1_1detail_1_1_sigmoid.html',1,'mlx::core::detail::Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html',1,'mlx::core::Sigmoid'],['../struct_sigmoid.html',1,'Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html#a47eca99113ec19f0eb60b6a0472c592b',1,'mlx::core::Sigmoid::Sigmoid()']]], - ['sigmoid_68',['sigmoid',['../group__ops.html#ga708abf8f79609cd6831db7c38cafac0e',1,'mlx::core']]], - ['sign_69',['Sign',['../structmlx_1_1core_1_1detail_1_1_sign.html',1,'mlx::core::detail::Sign'],['../classmlx_1_1core_1_1_sign.html',1,'mlx::core::Sign'],['../struct_sign.html',1,'Sign'],['../classmlx_1_1core_1_1_sign.html#afe951e50907bc23a601ec5fa9eae5763',1,'mlx::core::Sign::Sign()']]], - ['sign_70',['sign',['../group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa',1,'mlx::core']]], - ['signal_71',['signal',['../classmlx_1_1core_1_1_event.html#a65a858445506a61be5889ae0e3651b89',1,'mlx::core::Event']]], - ['signaling_5fnan_72',['signaling_NaN',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ad1f76a43c7d51a3765174aa6e0dd9f80',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['signedinteger_73',['signedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2daed58b4631ff157bec9e35ed1182d2c10',1,'mlx::core::Dtype::signedinteger'],['../namespacemlx_1_1core.html#a24e1618af591d737d73729665e868001',1,'mlx::core::signedinteger']]], - ['simd_5fbroadcast_74',['simd_broadcast',['../namespacemetal.html#a498f1e85107eb5f01ba4435977f8efe0',1,'metal']]], - ['simd_5fexclusive_5fscan_75',['simd_exclusive_scan',['../struct_cum_prod.html#a1a86e9398bae24182b7be0a6577bf223',1,'CumProd< bool >::simd_exclusive_scan()'],['../struct_cum_max.html#ae11b67aa6c998e9a01615b2a79af4403',1,'CumMax::simd_exclusive_scan()'],['../struct_cum_min.html#a83e65017ff33018b585c043fb803773b',1,'CumMin::simd_exclusive_scan()']]], - ['simd_5fmax_76',['simd_max',['../namespacemetal.html#a048cad0aca52cb737ebf103e76bd1c49',1,'metal']]], - ['simd_5fmin_77',['simd_min',['../namespacemetal.html#ae9e2a23e00724ba2d7868bc4112b386b',1,'metal']]], - ['simd_5fprefix_5fexclusive_5fproduct_78',['simd_prefix_exclusive_product',['../namespacemetal.html#a5ca40242390b632f737e29636829b2e4',1,'metal']]], - ['simd_5fprefix_5fexclusive_5fsum_79',['simd_prefix_exclusive_sum',['../namespacemetal.html#abfbb70c7471f28bf7ff36a612ad014b2',1,'metal']]], - ['simd_5fprefix_5finclusive_5fproduct_80',['simd_prefix_inclusive_product',['../namespacemetal.html#a6ca6a7e1996228fa536e969e9e45c446',1,'metal']]], - ['simd_5fprefix_5finclusive_5fsum_81',['simd_prefix_inclusive_sum',['../namespacemetal.html#a567acb18199ac0107712eb8cb8aeb8e9',1,'metal']]], - ['simd_5fproduct_82',['simd_product',['../namespacemetal.html#ac6e883a04e2265a9790d7db76059e1b4',1,'metal']]], - ['simd_5fscan_83',['simd_scan',['../struct_cum_prod.html#abeb5ec4237b330e7219f4e881cf10d7a',1,'CumProd< bool >::simd_scan()'],['../struct_cum_max.html#adc9ec8bb09b4433d4c2f03022c43d781',1,'CumMax::simd_scan()'],['../struct_cum_min.html#a0a1005d91b1c90e90e2c6dbd6c296649',1,'CumMin::simd_scan()']]], - ['simd_5fshuffle_84',['simd_shuffle',['../namespacemetal.html#a259ed115bc3c58f88eb35830916b26d4',1,'metal::simd_shuffle()'],['../backend_2metal_2kernels_2utils_8h.html#a71986ecdd7d18f975dd22c3df7421ce2',1,'simd_shuffle(uint64_t data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a3bdbdfeb7a1dde40cd3ce1df8d9213b5',1,'simd_shuffle(int64_t data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ab4cbcdb054f9165130da91a3334da0cf',1,'simd_shuffle(bool data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ab8175b66bcc080fb89f738143568c30b',1,'simd_shuffle(complex64_t data, uint16_t lane): utils.h']]], - ['simd_5fshuffle_5fand_5ffill_5fdown_85',['simd_shuffle_and_fill_down',['../namespacemetal.html#ae29a06f0eac636ad7af21dea5b04938b',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a0ee6239fa29a5f9ee0201e0dc5ddc8e0',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], - ['simd_5fshuffle_5fand_5ffill_5fup_86',['simd_shuffle_and_fill_up',['../namespacemetal.html#a1ca14116bf50639b214d8414b5bbaaa6',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a5138d5cdc18139e135707916a243cd8e',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta)'],['../backend_2metal_2kernels_2utils_8h.html#a5862d5ea154c9b76cf56a630cf6385b4',1,'simd_shuffle_and_fill_up(uint64_t data, uint64_t filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a7bb56415c5412a6a26f70a990915f064',1,'simd_shuffle_and_fill_up(int64_t data, int64_t filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ad55bd473647f2c6c68e65e5312c132d1',1,'simd_shuffle_and_fill_up(bool data, bool filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a94e02a6ae8c39cbf4cb23aa44df9dbd5',1,'simd_shuffle_and_fill_up(complex64_t data, complex64_t filling, uint16_t delta): utils.h']]], - ['simd_5fshuffle_5fdown_87',['simd_shuffle_down',['../namespacemetal.html#af6e2dd7ae087aba6abac4f0350b7611c',1,'metal::simd_shuffle_down()'],['../backend_2metal_2kernels_2utils_8h.html#aba6279624b1d30c525efee856a222b5c',1,'simd_shuffle_down(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a0c1e4d782fcc56e1ab5565cef12430dd',1,'simd_shuffle_down(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a48ae83a8caf5c74810df60b6c6cdb062',1,'simd_shuffle_down(bool data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ad9a671a5f9aaa729ae7a77026f16bcb0',1,'simd_shuffle_down(complex64_t data, uint16_t delta): utils.h']]], - ['simd_5fshuffle_5frotate_5fdown_88',['simd_shuffle_rotate_down',['../namespacemetal.html#a4bb203647a421032db47e73cd649841b',1,'metal']]], - ['simd_5fshuffle_5frotate_5fup_89',['simd_shuffle_rotate_up',['../namespacemetal.html#a729b22077d6c944491a6027c18ea80c9',1,'metal']]], - ['simd_5fshuffle_5fup_90',['simd_shuffle_up',['../namespacemetal.html#afe81c5fbde3f4890458b081909242c55',1,'metal::simd_shuffle_up()'],['../backend_2metal_2kernels_2utils_8h.html#a39e436e0a942912266aae7e0bd82d7c0',1,'simd_shuffle_up(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a617f3857caf33c569afa6148135f8b7a',1,'simd_shuffle_up(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ae0f5c42020275a588234e69f1eb7a485',1,'simd_shuffle_up(bool data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a92b455bac6a23af51c35ea83de2383eb',1,'simd_shuffle_up(complex64_t data, uint16_t delta): utils.h']]], - ['simd_5fshuffle_5fxor_91',['simd_shuffle_xor',['../namespacemetal.html#a5017efc9605e069cfb507137cd1a1852',1,'metal']]], - ['simd_5fsize_92',['SIMD_SIZE',['../quantized_8h.html#a62969a218d93680f5e35d0c61b160b99',1,'quantized.h']]], - ['simd_5fsize_93',['simd_size',['../backend_2metal_2kernels_2reduction_2ops_8h.html#a515b75d563a93d3c09ee677948dc83e3',1,'ops.h']]], - ['simd_5fsum_94',['simd_sum',['../namespacemetal.html#a85181e37a00cb4a4217f1bb25389bce5',1,'metal']]], - ['simd_5fxor_95',['simd_xor',['../namespacemetal.html#a1308decbf2d5c33d34d6be523ea1c30f',1,'metal']]], - ['simple_5fiter_96',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html',1,'pocketfft::detail::simple_iter'],['../classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398',1,'pocketfft::detail::simple_iter::simple_iter()']]], - ['simplevalueandgradfn_97',['SimpleValueAndGradFn',['../namespacemlx_1_1core.html#a2689b8f1181648cb1685204fea9f3066',1,'mlx::core']]], - ['sin_98',['Sin',['../structmlx_1_1core_1_1detail_1_1_sin.html',1,'mlx::core::detail::Sin'],['../classmlx_1_1core_1_1_sin.html',1,'mlx::core::Sin'],['../struct_sin.html',1,'Sin'],['../classmlx_1_1core_1_1_sin.html#a10d1ecc0ca96e79cdf55b57073d126ea',1,'mlx::core::Sin::Sin()']]], - ['sin_99',['sin',['../namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258',1,'pocketfft::detail::sin()'],['../namespacemetal.html#a619a159ca5f2ddfe3647d3a6bb6e804c',1,'metal::sin()'],['../namespacemetal_1_1fast.html#a3af771cfe7a135104f9d063147dba270',1,'metal::fast::sin()'],['../namespacemetal_1_1precise.html#a71acf77ffd29c56f56afae0195c98a1c',1,'metal::precise::sin()'],['../group__ops.html#gaebf0a73ad3732fba39df37826c235692',1,'mlx::core::sin()']]], - ['sincos_5f2pibyn_100',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html',1,'pocketfft::detail::sincos_2pibyn< T >'],['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40',1,'pocketfft::detail::sincos_2pibyn::sincos_2pibyn()']]], - ['sinh_101',['Sinh',['../structmlx_1_1core_1_1detail_1_1_sinh.html',1,'mlx::core::detail::Sinh'],['../classmlx_1_1core_1_1_sinh.html',1,'mlx::core::Sinh'],['../struct_sinh.html',1,'Sinh'],['../classmlx_1_1core_1_1_sinh.html#a4a4f6814d403c2ce5d6c574b0dca3c96',1,'mlx::core::Sinh::Sinh()']]], - ['sinh_102',['sinh',['../namespacemetal.html#a83ba4235ae350ab8880a9df09158620b',1,'metal::sinh()'],['../namespacemetal_1_1fast.html#a990d90b3440e38d1fb4ff5065c6c189b',1,'metal::fast::sinh()'],['../namespacemetal_1_1precise.html#abc8f4f59dd6e7204ab5d84f0af96331c',1,'metal::precise::sinh()'],['../group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7',1,'mlx::core::sinh()']]], - ['sinpi_103',['sinpi',['../namespacemetal.html#ae9655f7fa2ba6c0625ca25fbb278e269',1,'metal::sinpi()'],['../namespacemetal_1_1fast.html#ab07a32fe544aa304577d29e0251e87b2',1,'metal::fast::sinpi()'],['../namespacemetal_1_1precise.html#a78b17dab93519d9c82c2575dafec49c9',1,'metal::precise::sinpi()']]], - ['size_104',['size',['../classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952',1,'pocketfft::detail::arr::size()'],['../classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5',1,'pocketfft::detail::arr_info::size()'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a2adf9a9c968f113dde830cc0dc27dcc6',1,'mlx::core::allocator::Allocator::size()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#aafa92e8310db089b1ac72b840777e26b',1,'mlx::core::allocator::CommonAllocator::size()'],['../classmlx_1_1core_1_1array.html#a598f87161926d9e0b516860f0ea2c8f6',1,'mlx::core::array::size()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a51f6587e8065be16f0418ca42a796e05',1,'mlx::core::metal::MetalAllocator::size()'],['../structmlx_1_1core_1_1distributed_1_1_group.html#ac0a5a1e463a9330355e8bfe09c0feaf2',1,'mlx::core::distributed::Group::size()'],['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ab1c8044b05f185c4bcc53002d4587599',1,'mlx::core::distributed::detail::GroupImpl::size()'],['../structmlx_1_1core_1_1_dtype.html#ab54051563d85212c7f0f049166bc9971',1,'mlx::core::Dtype::size()']]], - ['size_5fof_105',['size_of',['../namespacemlx_1_1core.html#add4794cc0ffe5d717fc146084a235d95',1,'mlx::core']]], - ['slice_106',['Slice',['../classmlx_1_1core_1_1_slice.html',1,'mlx::core::Slice'],['../classmlx_1_1core_1_1_slice.html#a3aa025acbf4a9ca9e030a1e6bda102f7',1,'mlx::core::Slice::Slice()']]], - ['slice_107',['slice',['../group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d',1,'mlx::core::slice(const array &a, Shape start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#gaedcbdf4040f5e6a02a74b1ed1c6c2ebc',1,'mlx::core::slice(const array &a, std::initializer_list< int > start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#gaec56dcb94d5e7f7b885fb60b4bf4aa9d',1,'mlx::core::slice(const array &a, Shape start, Shape stop, StreamOrDevice s={})'],['../group__ops.html#ga797996e53ea34317a55dc2f314736b89',1,'mlx::core::slice(const array &a, const array &start, std::vector< int > axes, Shape slice_size, StreamOrDevice s={})']]], - ['slice_5fgpu_108',['slice_gpu',['../namespacemlx_1_1core.html#a327578951a44116e5da2db651661265f',1,'mlx::core']]], - ['slice_5fupdate_109',['slice_update',['../group__ops.html#ga7b7e786985d27789aaed20bb2f9509be',1,'mlx::core::slice_update(const array &src, const array &update, Shape start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#ga3b7c6136ae5a38dd6457b65c1833aa67',1,'mlx::core::slice_update(const array &src, const array &update, Shape start, Shape stop, StreamOrDevice s={})'],['../group__ops.html#gacd906ffb96149a4998c321cbf2231d7e',1,'mlx::core::slice_update(const array &src, const array &update, const array &start, std::vector< int > axes, StreamOrDevice s={})']]], - ['sliceupdate_110',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html',1,'mlx::core::SliceUpdate'],['../classmlx_1_1core_1_1_slice_update.html#a63a49264b18340f2bc442c081a7b4c7a',1,'mlx::core::SliceUpdate::SliceUpdate()']]], - ['slicing_2eh_111',['slicing.h',['../common_2slicing_8h.html',1,'(Global Namespace)'],['../metal_2slicing_8h.html',1,'(Global Namespace)']]], - ['sm_112',['sm',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa85451edf6900fd6af164d4d50889ae3',1,'mlx::steel::BlockMMA']]], - ['sn_113',['sn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ade420e8b811d597345783c324c23a34a',1,'mlx::steel::BlockMMA']]], - ['softmax_114',['Softmax',['../classmlx_1_1core_1_1_softmax.html',1,'mlx::core::Softmax'],['../classmlx_1_1core_1_1_softmax.html#a4ec686aac4e06f0dfe2cbd6801af40eb',1,'mlx::core::Softmax::Softmax()']]], - ['softmax_115',['softmax',['../namespacemlx_1_1core_1_1metal.html#a4fe937c2c584fd646926057f31d54ca6',1,'mlx::core::metal::softmax()'],['../group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6',1,'mlx::core::softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga1ae3614d07d873892a530d14c3857d0b',1,'mlx::core::softmax(const array &a, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga06f570d73716a24303e6de3aaba4457b',1,'mlx::core::softmax(const array &a, int axis, bool precise=false, StreamOrDevice s={})']]], - ['softmax_2eh_116',['softmax.h',['../jit_2softmax_8h.html',1,'(Global Namespace)'],['../kernels_2softmax_8h.html',1,'(Global Namespace)']]], - ['softmax_5fexp_117',['softmax_exp',['../kernels_2softmax_8h.html#a440d4031ee5e86159a4dd715e44a438b',1,'softmax.h']]], - ['softmax_5fkernels_118',['softmax_kernels',['../jit_2softmax_8h.html#a1cbfb210a9a765c6620e9f1247ccef12',1,'softmax.h']]], - ['softmax_5flooped_119',['softmax_looped',['../kernels_2softmax_8h.html#a8c47b0924ebfeebcca25f3dd17373276',1,'softmax.h']]], - ['softmax_5fn_5freads_120',['SOFTMAX_N_READS',['../defines_8h.html#a722995df24286b27b7da3d74b73f768d',1,'defines.h']]], - ['softmax_5fsingle_5frow_121',['softmax_single_row',['../kernels_2softmax_8h.html#a815fe70f879f318e5d6e99acf043f52b',1,'softmax.h']]], - ['sort_122',['Sort',['../classmlx_1_1core_1_1_sort.html',1,'mlx::core::Sort'],['../classmlx_1_1core_1_1_sort.html#a62943032dbd72e85ceb9b4b7211f4a44',1,'mlx::core::Sort::Sort()']]], - ['sort_123',['sort',['../struct_thread_sort.html#ad9ab3e6b47f7e9b91c0f3b773596986d',1,'ThreadSort::sort()'],['../struct_block_merge_sort.html#acc970f5eb963f7f2010f5ae5ea8b8bc0',1,'BlockMergeSort::sort()'],['../namespacemlx_1_1core_1_1metal.html#ab77c9a9ecaeeab8c66b712862777c24b',1,'mlx::core::metal::sort()'],['../group__ops.html#ga7fb616054665b3c2d61fa234f501f079',1,'mlx::core::sort(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaae1bc47aa737f705d0e5884270063fea',1,'mlx::core::sort(const array &a, int axis, StreamOrDevice s={})']]], - ['sort_2eh_124',['sort.h',['../sort_8h.html',1,'']]], - ['special_5fmul_125',['special_mul',['../structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874',1,'pocketfft::detail::cmplx::special_mul()'],['../namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76',1,'pocketfft::detail::special_mul()']]], - ['split_126',['Split',['../classmlx_1_1core_1_1_split.html',1,'mlx::core::Split'],['../classmlx_1_1core_1_1_split.html#ad3f4ed34b85c73683bad5d530309342f',1,'mlx::core::Split::Split()']]], - ['split_127',['split',['../structmlx_1_1core_1_1distributed_1_1_group.html#ad3682f4dc85bfe7e5464b87f6f0fd931',1,'mlx::core::distributed::Group::split()'],['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a87800a23c8160933a2d77a55a959194d',1,'mlx::core::distributed::detail::GroupImpl::split()'],['../group__ops.html#ga7534290bceab5fb3831a05d67bebce7d',1,'mlx::core::split(const array &a, int num_splits, int axis, StreamOrDevice s={})'],['../group__ops.html#ga56882d24e5fde59c266774624c892d41',1,'mlx::core::split(const array &a, int num_splits, StreamOrDevice s={})'],['../group__ops.html#ga19005414e7d8020cd6e94e06bf399b09',1,'mlx::core::split(const array &a, const Shape &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#ga9ea089f42b9940510619052b7166d9ac',1,'mlx::core::split(const array &a, const Shape &indices, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a42847b435d037a977592e355eed072af',1,'mlx::core::random::split(const array &key, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7ec057064c7326c41b536f08178861e5',1,'mlx::core::random::split(const array &key, int num, StreamOrDevice s={})']]], - ['split_5fk_5fpartition_5fsize_128',['split_k_partition_size',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a9f5a67b2343645b570e109c3837d4042',1,'mlx::steel::GEMMSpiltKParams']]], - ['split_5fk_5fpartition_5fstride_129',['split_k_partition_stride',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a12144ce89d404812cd862611d770b9fb',1,'mlx::steel::GEMMSpiltKParams']]], - ['split_5fk_5fpartitions_130',['split_k_partitions',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#ae06c27116905d4ff3b9b436e588a93fd',1,'mlx::steel::GEMMSpiltKParams']]], - ['sqrt_131',['Sqrt',['../structmlx_1_1core_1_1detail_1_1_sqrt.html',1,'mlx::core::detail::Sqrt'],['../classmlx_1_1core_1_1_sqrt.html',1,'mlx::core::Sqrt'],['../struct_sqrt.html',1,'Sqrt'],['../classmlx_1_1core_1_1_sqrt.html#a6682a7c31ca427c9d2c5ddb6a479bf29',1,'mlx::core::Sqrt::Sqrt()']]], - ['sqrt_132',['sqrt',['../namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3',1,'pocketfft::detail::sqrt()'],['../namespacemetal.html#ab3f4d4852ca0e591104fbd8e5b50d31b',1,'metal::sqrt()'],['../namespacemetal_1_1fast.html#a4218a85c7d8a74cb8055b4755205627e',1,'metal::fast::sqrt()'],['../namespacemetal_1_1precise.html#acb213467361cd2cab93a8d5ea1aa5bfd',1,'metal::precise::sqrt()'],['../group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1',1,'mlx::core::sqrt()']]], - ['square_133',['Square',['../structmlx_1_1core_1_1detail_1_1_square.html',1,'mlx::core::detail::Square'],['../classmlx_1_1core_1_1_square.html',1,'mlx::core::Square'],['../struct_square.html',1,'Square'],['../classmlx_1_1core_1_1_square.html#ab94e28d5c92e6febc1c74e525f730dc4',1,'mlx::core::Square::Square()']]], - ['square_134',['square',['../group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e',1,'mlx::core']]], - ['squeeze_135',['Squeeze',['../classmlx_1_1core_1_1_squeeze.html',1,'mlx::core::Squeeze'],['../classmlx_1_1core_1_1_squeeze.html#a032bd53dcc3d71a11d810bc3ca3ef4b0',1,'mlx::core::Squeeze::Squeeze()']]], - ['squeeze_136',['squeeze',['../group__ops.html#ga710daa7ec721bd4d3f326082cb195576',1,'mlx::core::squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3',1,'mlx::core::squeeze(const array &a, int axis, StreamOrDevice s={})'],['../group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad',1,'mlx::core::squeeze(const array &a, StreamOrDevice s={})']]], - ['src_137',['src',['../struct_quantized_block_loader.html#abbf8249ca99e3e87b296ddd60a984b76',1,'QuantizedBlockLoader::src'],['../structmlx_1_1steel_1_1_block_loader.html#ad1db14517568ae9eddfb6986ef31c7aa',1,'mlx::steel::BlockLoader::src'],['../structmlx_1_1steel_1_1_block_loader_t.html#a7004a4efaa483cc79b8b79810a17c777',1,'mlx::steel::BlockLoaderT::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a656a46ee27486482b45ff90b3d626255',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a57552e9cfbafad71d47b2f3a8e027bdf',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7bfbcc4a1e3eef7aef5dd8e8c374a95f',1,'mlx::steel::Conv2DWeightBlockLoader::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#afe21e46e08523232830c25eb1b4ade16',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a8b04a69952404a04029dacc424df6e8f',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1ee2922961b5fcb1db577928c4d9d731',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a50f458dbb74d61be2ed24727d8d43614',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src']]], - ['src_5fld_138',['src_ld',['../struct_quantized_block_loader.html#a8050977d473d1a24fae5c833e609839e',1,'QuantizedBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_block_loader.html#aadafc50f7f06af434149d7469df4714d',1,'mlx::steel::BlockLoader::src_ld'],['../structmlx_1_1steel_1_1_block_loader_t.html#aeba87e81185da6b20a092c5d240d3321',1,'mlx::steel::BlockLoaderT::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7464ec687323fa79050702952ed9084f',1,'mlx::steel::Conv2DWeightBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#aa6bedc0cbb447eaf70c03f2e26df2cb2',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6918c1df7712c4e408e2871467ea7987',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src_ld']]], - ['stack_139',['stack',['../group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032',1,'mlx::core::stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga82216209dce901296fc737fe8efa5c94',1,'mlx::core::stack(const std::vector< array > &arrays, StreamOrDevice s={})']]], - ['start_5fcapture_140',['start_capture',['../namespacemlx_1_1core_1_1metal.html#aa47cb5651bf3b65c46ab216b7e504d77',1,'mlx::core::metal']]], - ['start_5fconcurrent_141',['start_concurrent',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::metal::CommandEncoder::start_concurrent()'],['../structmlx_1_1core_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::CommandEncoder::start_concurrent()']]], - ['start_5frow_142',['start_row',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a220e033b689c8d6a6f319dae02b38334',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], - ['state_143',['state',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a379b27ac336ef351aa81142c5626ad76',1,'mlx::core::fast::RMSNormVJP::state()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a963e672c607b5f86080e6cc32a3cd6e5',1,'mlx::core::fast::LayerNormVJP::state()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#aa5a03284c6f5639d684dd34d86050cf9',1,'mlx::core::fast::AffineQuantize::state()'],['../classmlx_1_1core_1_1_add_m_m.html#a24ab73de46d0589780dac5ade43f93b8',1,'mlx::core::AddMM::state()'],['../classmlx_1_1core_1_1_arange.html#ac4a9f48a11c2af03ed57fdf2422cbfad',1,'mlx::core::Arange::state()'],['../classmlx_1_1core_1_1_arg_partition.html#a5033c46f5aae9b14859cc8b0ca4c8e19',1,'mlx::core::ArgPartition::state()'],['../classmlx_1_1core_1_1_arg_reduce.html#acac3b26364260aac7511b4cb7add3604',1,'mlx::core::ArgReduce::state()'],['../classmlx_1_1core_1_1_arg_sort.html#a90548429765f9e7e2332f01b72692fa2',1,'mlx::core::ArgSort::state()'],['../classmlx_1_1core_1_1_as_type.html#a98ea769fc9cd6d76b07817444e7a78ab',1,'mlx::core::AsType::state()'],['../classmlx_1_1core_1_1_as_strided.html#ae730aeff375498ba774d4207c7af8c36',1,'mlx::core::AsStrided::state()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8a67d6f431b4055ab66656201622af4d',1,'mlx::core::BitwiseBinary::state()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a6bbcc34b256840e4df2953563f2b4a07',1,'mlx::core::BlockMaskedMM::state()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a5b2594b7a70dd4873a07e742140a245f',1,'mlx::core::BroadcastAxes::state()'],['../classmlx_1_1core_1_1_broadcast.html#a49fdb421047860733af7dfbbb478da8d',1,'mlx::core::Broadcast::state()'],['../classmlx_1_1core_1_1_concatenate.html#a60cd572a42b346399ee539af2dfbf29e',1,'mlx::core::Concatenate::state()'],['../classmlx_1_1core_1_1_convolution.html#a7f44f0caea20cc2858717afba1e915d8',1,'mlx::core::Convolution::state()'],['../classmlx_1_1core_1_1_equal.html#aa27ff7525f109edc56b731a6df78f6bc',1,'mlx::core::Equal::state()'],['../classmlx_1_1core_1_1_expand_dims.html#a7cacc704c533c00ba072f0a7872631cf',1,'mlx::core::ExpandDims::state()'],['../classmlx_1_1core_1_1_f_f_t.html#a710c6f6e8412da0af0fdbe58fbae320e',1,'mlx::core::FFT::state()'],['../classmlx_1_1core_1_1_flatten.html#af95dd89c47cd2342233dc0b6d36822a3',1,'mlx::core::Flatten::state()'],['../classmlx_1_1core_1_1_gather.html#aee59ff90127ef4c2d7fcbe2955b95b27',1,'mlx::core::Gather::state()'],['../classmlx_1_1core_1_1_hadamard.html#afd67d09fde38ab3b6ba873b797f03dae',1,'mlx::core::Hadamard::state()'],['../classmlx_1_1core_1_1_log.html#a86fca2ec3766f5d4a2e6d8ba2983c3aa',1,'mlx::core::Log::state()'],['../classmlx_1_1core_1_1_number_of_elements.html#afbfee716b4896e98bdf502ceab87ac09',1,'mlx::core::NumberOfElements::state()'],['../classmlx_1_1core_1_1_pad.html#a00a7cff2ae640f45b43f62cc25d6346c',1,'mlx::core::Pad::state()'],['../classmlx_1_1core_1_1_partition.html#adde13e40924c016473864119465cad4b',1,'mlx::core::Partition::state()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ad83bfd32fda988c29e5ca277a84c0655',1,'mlx::core::QuantizedMatmul::state()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a60c908bc836f930bb33f60b3e9db43af',1,'mlx::core::GatherQMM::state()'],['../classmlx_1_1core_1_1_random_bits.html#a75a34d7541a1c124710dc4d0ec2dfa60',1,'mlx::core::RandomBits::state()'],['../classmlx_1_1core_1_1_reshape.html#aa8ad5958aac8723dd6ce49820eaba029',1,'mlx::core::Reshape::state()'],['../classmlx_1_1core_1_1_reduce.html#afca1398c042a3b1ca9a9a2e40fe62367',1,'mlx::core::Reduce::state()'],['../classmlx_1_1core_1_1_scan.html#a7249ca4c3316b1b1248df32c71fee0ea',1,'mlx::core::Scan::state()'],['../classmlx_1_1core_1_1_scatter.html#a50a65033dc2a1cc84bf529ba718c9c60',1,'mlx::core::Scatter::state()'],['../classmlx_1_1core_1_1_slice.html#a069dafc62bf71e3ebc0bd99d96ec23be',1,'mlx::core::Slice::state()'],['../classmlx_1_1core_1_1_slice_update.html#aac5156a19209274b7de1dff231ef25fd',1,'mlx::core::SliceUpdate::state()'],['../classmlx_1_1core_1_1_dynamic_slice.html#aec9084e603d7562f3a75c5fc32918581',1,'mlx::core::DynamicSlice::state()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a0b0b2a0e4d97305fd6f3c635fcdccd76',1,'mlx::core::DynamicSliceUpdate::state()'],['../classmlx_1_1core_1_1_softmax.html#adf665f7c292e84f56c630016b75427f3',1,'mlx::core::Softmax::state()'],['../classmlx_1_1core_1_1_sort.html#a5ab15d1a89efd8661887c805c35fc617',1,'mlx::core::Sort::state()'],['../classmlx_1_1core_1_1_split.html#a915156cde0448ec26776e329004b1a92',1,'mlx::core::Split::state()'],['../classmlx_1_1core_1_1_sqrt.html#a053853757ad99195e3f2b1cca571e31b',1,'mlx::core::Sqrt::state()'],['../classmlx_1_1core_1_1_squeeze.html#a04f9d2595cb7d4ec988479cd33fe9362',1,'mlx::core::Squeeze::state()'],['../classmlx_1_1core_1_1_unflatten.html#aeba13680064238191811230171365598',1,'mlx::core::Unflatten::state()'],['../classmlx_1_1core_1_1_view.html#a37620f6548630bd2d0dd44e9ab084b93',1,'mlx::core::View::state()'],['../classmlx_1_1core_1_1_transpose.html#a23167291e2bf12e2bb2e51d1db340909',1,'mlx::core::Transpose::state()'],['../classmlx_1_1core_1_1_inverse.html#aa1fce744f4a2d660c65901a7542056f2',1,'mlx::core::Inverse::state()'],['../classmlx_1_1core_1_1_cholesky.html#a64f03d32ed249a3b2a59b6af66d23727',1,'mlx::core::Cholesky::state()'],['../classmlx_1_1core_1_1_eigh.html#aa3b6c33b5679c5528863f3de2ab2f914',1,'mlx::core::Eigh::state()']]], - ['status_144',['Status',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078',1,'mlx::core::array']]], - ['status_145',['status',['../classmlx_1_1core_1_1array.html#a7102659be87e9ef62966696ab9b07dad',1,'mlx::core::array']]], - ['std_146',['std',['../group__ops.html#ga2a466024f8061febc0a64be557644cb0',1,'mlx::core::std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#gafdcb04d77c64405a3990078a77dd984c',1,'mlx::core::std(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8',1,'mlx::core::std(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga62721a206df671ef5797449eea97af9f',1,'mlx::core::std(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], - ['steel_5fattention_2eh_147',['steel_attention.h',['../steel__attention_8h.html',1,'']]], - ['steel_5fconst_148',['STEEL_CONST',['../steel_2defines_8h.html#a90b91c866313ffa46eff6d9cc944ad2b',1,'defines.h']]], - ['steel_5fconv_149',['steel_conv',['../namespacemlx_1_1core_1_1metal.html#a92f1e559b1121d545746f81ff86eaca1',1,'mlx::core::metal']]], - ['steel_5fconv_2eh_150',['steel_conv.h',['../jit_2steel__conv_8h.html',1,'(Global Namespace)'],['../kernels_2steel_2conv_2kernels_2steel__conv_8h.html',1,'(Global Namespace)']]], - ['steel_5fconv_5fgeneral_151',['steel_conv_general',['../namespacemlx_1_1core_1_1metal.html#a02edb6a90bdf30f4c9f0d6c25b0267b5',1,'mlx::core::metal']]], - ['steel_5fconv_5fgeneral_2eh_152',['steel_conv_general.h',['../steel__conv__general_8h.html',1,'']]], - ['steel_5fconv_5fgeneral_5fkernels_153',['steel_conv_general_kernels',['../jit_2steel__conv_8h.html#ae4ca1720029316b08ea92b7662347d47',1,'steel_conv.h']]], - ['steel_5fconv_5fkernels_154',['steel_conv_kernels',['../jit_2steel__conv_8h.html#a386d79077465df56659416fd84adb513',1,'steel_conv.h']]], - ['steel_5fgemm_2eh_155',['steel_gemm.h',['../steel__gemm_8h.html',1,'']]], - ['steel_5fgemm_5ffused_156',['steel_gemm_fused',['../namespacemlx_1_1core_1_1metal.html#a17764366deed71c160fb26091400a803',1,'mlx::core::metal']]], - ['steel_5fgemm_5ffused_2eh_157',['steel_gemm_fused.h',['../steel__gemm__fused_8h.html',1,'']]], - ['steel_5fgemm_5ffused_5fkernels_158',['steel_gemm_fused_kernels',['../steel__gemm_8h.html#a4c6009fd5357b730805f2fd4ba6e093e',1,'steel_gemm.h']]], - ['steel_5fgemm_5fmasked_159',['steel_gemm_masked',['../namespacemlx_1_1core_1_1metal.html#a962272ca73d26c08f76f706a128fd71f',1,'mlx::core::metal']]], - ['steel_5fgemm_5fmasked_2eh_160',['steel_gemm_masked.h',['../steel__gemm__masked_8h.html',1,'']]], - ['steel_5fgemm_5fmasked_5fkernels_161',['steel_gemm_masked_kernels',['../steel__gemm_8h.html#a62a358fd3ec5365081920d07aceb581c',1,'steel_gemm.h']]], - ['steel_5fgemm_5fsplitk_162',['steel_gemm_splitk',['../namespacemlx_1_1core_1_1metal.html#ad0dfd40ba7c09755711ceb731e57a5ac',1,'mlx::core::metal']]], - ['steel_5fgemm_5fsplitk_2eh_163',['steel_gemm_splitk.h',['../steel__gemm__splitk_8h.html',1,'']]], - ['steel_5fgemm_5fsplitk_5faccum_5faxbpy_5fkernels_164',['steel_gemm_splitk_accum_axbpy_kernels',['../steel__gemm_8h.html#a40a86e1381c241aba8511e51a981a4bf',1,'steel_gemm.h']]], - ['steel_5fgemm_5fsplitk_5faccum_5fkernels_165',['steel_gemm_splitk_accum_kernels',['../steel__gemm_8h.html#a144a64b8d94f0371fb144e2cc308fcf9',1,'steel_gemm.h']]], - ['steel_5fgemm_5fsplitk_5fkernels_166',['steel_gemm_splitk_kernels',['../steel__gemm_8h.html#a92108ab01d826e38bca83d8569b947d9',1,'steel_gemm.h']]], - ['steel_5fmatmul_167',['steel_matmul',['../namespacemlx_1_1core.html#aaff208bbac7021c4265580885874499a',1,'mlx::core']]], - ['steel_5fmatmul_5fregular_168',['steel_matmul_regular',['../namespacemlx_1_1core.html#a2a8a09851097571fb51ac5b608550e44',1,'mlx::core']]], - ['steel_5fpragma_5funroll_169',['STEEL_PRAGMA_UNROLL',['../steel_2defines_8h.html#a5a5c3095b132a7589bc19cd5cb80e2c6',1,'defines.h']]], - ['step_170',['step',['../structmlx_1_1core_1_1_contiguous_iterator.html#aad921dd422adb0a0f555e19a2f42239c',1,'mlx::core::ContiguousIterator']]], - ['stop_171',['stop',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a456ad1c0c9e731833a2f8411c4ed51aa',1,'mlx::core::scheduler::StreamThread']]], - ['stop_5fcapture_172',['stop_capture',['../namespacemlx_1_1core_1_1metal.html#ac90714424e36fb01e04550de69b8314f',1,'mlx::core::metal']]], - ['stop_5fgradient_173',['stop_gradient',['../group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f',1,'mlx::core']]], - ['stopgradient_174',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html',1,'mlx::core::StopGradient'],['../classmlx_1_1core_1_1_stop_gradient.html#ac70d1ab819d04e00f76bc25aeebaf84f',1,'mlx::core::StopGradient::StopGradient()']]], - ['store_175',['store',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#aa8f50ea8961ec5b35c1b81366d64f2cb',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a2aadaa3239cb3af0c2ee8af9b88c8a98',1,'mlx::steel::MMATile::store(threadgroup U *dst) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a752f708e4fe5ef37fdd902dae153179f',1,'mlx::steel::MMATile::store(device U *dst, const int ld) const'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#aa8f50ea8961ec5b35c1b81366d64f2cb',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a2aadaa3239cb3af0c2ee8af9b88c8a98',1,'mlx::steel::MMATile::store(threadgroup U *dst) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a752f708e4fe5ef37fdd902dae153179f',1,'mlx::steel::MMATile::store(device U *dst, const int ld) const']]], - ['store_5fresult_176',['store_result',['../structmlx_1_1steel_1_1_block_m_m_a.html#a0461451ffb5041b6a916ea17ed34288b',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a0461451ffb5041b6a916ea17ed34288b',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const']]], - ['store_5fresult_5fsafe_177',['store_result_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a081ba538d30d1d02498a7f341e6bd611',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a081ba538d30d1d02498a7f341e6bd611',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const']]], - ['store_5fsafe_178',['store_safe',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a1f0b00daad8eba2f855bb306e70d2328',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a57703f522c7409dbe2c0a68bb7acc2ba',1,'mlx::steel::MMATile::store_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a1f0b00daad8eba2f855bb306e70d2328',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a57703f522c7409dbe2c0a68bb7acc2ba',1,'mlx::steel::MMATile::store_safe()']]], - ['str_179',['str',['../classpocketfft_1_1detail_1_1arr__info.html#abe1f7b92501b4e0e5a38fd26294ac5a4',1,'pocketfft::detail::arr_info::str'],['../struct_m_l_x_conv_params.html#a862191e8ab1bc8a47aa1396b36d46058',1,'MLXConvParams::str']]], - ['stream_180',['Stream',['../structmlx_1_1core_1_1_stream.html',1,'mlx::core::Stream'],['../structmlx_1_1core_1_1_stream.html#a7f0815ff4886da74cbbff5f93d82dd3e',1,'mlx::core::Stream::Stream()']]], - ['stream_181',['stream',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a8462e4acffcd385c6248bd7102e6bcb1',1,'mlx::core::scheduler::StreamThread::stream'],['../classmlx_1_1core_1_1_event.html#a193143bad31b68c699fa27f135b45614',1,'mlx::core::Event::stream()'],['../classmlx_1_1core_1_1_primitive.html#a46e6257397a662528f9f831842ac456a',1,'mlx::core::Primitive::stream()']]], - ['stream_2eh_182',['stream.h',['../stream_8h.html',1,'']]], - ['streamcontext_183',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html',1,'mlx::core::StreamContext'],['../structmlx_1_1core_1_1_stream_context.html#a89d803151e9d7dce29382aa83d5c6ef1',1,'mlx::core::StreamContext::StreamContext()']]], - ['streamordevice_184',['StreamOrDevice',['../namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58',1,'mlx::core']]], - ['streamthread_185',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html',1,'mlx::core::scheduler::StreamThread'],['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#ac528109a11abcb82e6e221c5efa4493c',1,'mlx::core::scheduler::StreamThread::StreamThread()']]], - ['stride_186',['stride',['../classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093',1,'pocketfft::detail::arr_info::stride() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b',1,'pocketfft::detail::arr_info::stride(size_t i) const']]], - ['stride_5fin_187',['stride_in',['../classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c',1,'pocketfft::detail::multi_iter']]], - ['stride_5fout_188',['stride_out',['../classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7',1,'pocketfft::detail::multi_iter']]], - ['stride_5ft_189',['stride_t',['../namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::detail::stride_t'],['../namespacepocketfft.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::stride_t']]], - ['strided_5fdevice_5fidx_190',['strided_device_idx',['../struct_read_writer.html#a4c0b12484aac4fd6759d67c190391989',1,'ReadWriter']]], - ['strided_5freduce_5fgeneral_5fdispatch_191',['strided_reduce_general_dispatch',['../namespacemlx_1_1core.html#aa0332c64ee9965f05026c30a0b778000',1,'mlx::core']]], - ['strided_5fscan_192',['strided_scan',['../scan_8h.html#a7abb6ffb6c3b96b88c2a63cd4cc2f7ae',1,'scan.h']]], - ['strided_5fshared_5fidx_193',['strided_shared_idx',['../struct_read_writer.html#ace40adb02cfb33d89c98353327c251fc',1,'ReadWriter']]], - ['strides_194',['Strides',['../namespacemlx_1_1core.html#a79939016d0972ded7db37130da2a8b5c',1,'mlx::core']]], - ['strides_195',['strides',['../structmlx_1_1core_1_1_reduction_plan.html#a58bc6189e5e7175dae92632a7bcfd53e',1,'mlx::core::ReductionPlan::strides'],['../struct_indices.html#a510b7fe052c5826911dd17d7ccb9e73f',1,'Indices::strides'],['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63954de7da62942ec69afcaaa19d46f2',1,'mlx::core::fast::CustomKernelShapeInfo::strides'],['../classmlx_1_1core_1_1array.html#a28cf1928f5ec2f972a94ff1c0e71187d',1,'mlx::core::array::strides() const'],['../classmlx_1_1core_1_1array.html#ac9bfc251a9937eaefbe7f8c5ffd304d1',1,'mlx::core::array::strides(int dim) const']]], - ['submit_196',['submit',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4',1,'pocketfft::detail::threading::thread_pool']]], - ['subop_197',['SubOp',['../struct_sub_op.html',1,'']]], - ['subtract_198',['Subtract',['../structmlx_1_1core_1_1detail_1_1_subtract.html',1,'mlx::core::detail::Subtract'],['../classmlx_1_1core_1_1_subtract.html',1,'mlx::core::Subtract'],['../struct_subtract.html',1,'Subtract'],['../classmlx_1_1core_1_1_subtract.html#a834854757394f8de7082af65bf86ed9c',1,'mlx::core::Subtract::Subtract()']]], - ['subtract_199',['subtract',['../group__ops.html#ga196c240d3d0fcbb4713802c485e15133',1,'mlx::core']]], - ['sum_200',['Sum',['../struct_sum.html',1,'Sum< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a1fc7c1f09c80650ab0497e2d6781d65f',1,'mlx::core::distributed::AllReduce::Sum'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a8582875544f1d3d396a1a376473ef1dd',1,'mlx::core::Reduce::Sum'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ade23893033e4849f5596e7ce76a5fc36',1,'mlx::core::Scan::Sum'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca14abe2d8818efa71726be4e156813d6f',1,'mlx::core::Scatter::Sum']]], - ['sum_201',['sum',['../namespacemlx_1_1steel.html#ab4a6ddea4beb7c447cf5b69b9d46cc3b',1,'mlx::steel::sum(T x)'],['../namespacemlx_1_1steel.html#acd6e194d37b617d7a5818bc384a97fe4',1,'mlx::steel::sum(T x, Us... us)'],['../group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6',1,'mlx::core::sum(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3627754d7868487bdab1bd83f05d9c81',1,'mlx::core::sum(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4',1,'mlx::core::sum(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d',1,'mlx::core::sum(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['sumop_202',['SumOp',['../struct_sum_op.html',1,'']]], - ['svd_203',['SVD',['../classmlx_1_1core_1_1_s_v_d.html',1,'mlx::core::SVD'],['../classmlx_1_1core_1_1_s_v_d.html#ae89ff583e34fa894cccb8e7a475ee6d1',1,'mlx::core::SVD::SVD()']]], - ['svd_204',['svd',['../namespacemlx_1_1core_1_1linalg.html#a64364b880e99914cf47bf756fa8dbaf0',1,'mlx::core::linalg']]], - ['swapaxes_205',['swapaxes',['../group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb',1,'mlx::core']]], - ['swizzle_206',['swizzle',['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle::swizzle(uint3 tid, const int swizzle_log)'],['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle::swizzle(uint3 tid, const int swizzle_log)']]], - ['swizzle_5flog_207',['swizzle_log',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#ad0713159d4f710cd9a066596593d8840',1,'mlx::steel::ImplicitGemmConv2DParams::swizzle_log'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#af9ff2c06dd8994126634531440325be7',1,'mlx::steel::GEMMParams::swizzle_log']]], - ['synchronize_208',['synchronize',['../namespacemlx_1_1core.html#a14287949d82ffefad0306cef5eb5f9e4',1,'mlx::core::synchronize()'],['../namespacemlx_1_1core.html#a6648a71937b055e5ff513d98056c2fb5',1,'mlx::core::synchronize(Stream)']]] + ['scalar_5ft_6',['scalar_t',['../structmlx_1_1core_1_1simd_1_1_simd.html#a26040194a37172b6aed7c5d1685362fb',1,'mlx::core::simd::Simd::scalar_t'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a422e15f018cd242dd62617f4213dace0',1,'mlx::core::simd::Simd< float16_t, N >::scalar_t'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a26040194a37172b6aed7c5d1685362fb',1,'mlx::core::simd::Simd< T, 1 >::scalar_t'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a26040194a37172b6aed7c5d1685362fb',1,'mlx::core::simd::Simd< float16_t, N >::scalar_t']]], + ['scalarscalar_7',['ScalarScalar',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6a7b15cb76e0535ea81a5b6af9c96dcde4',1,'mlx::core']]], + ['scalarscalarscalar_8',['ScalarScalarScalar',['../namespacemlx_1_1core.html#ac2b8997537c7f25dd2b244d4c0a865a1a09c2e68746fa22c9903625cea17464db',1,'mlx::core']]], + ['scalart_9',['ScalarT',['../structmlx_1_1core_1_1simd_1_1_scalar_t.html',1,'mlx::core::simd']]], + ['scalart_3c_20bool_2c_20n_20_3e_10',['ScalarT< bool, N >',['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['scalart_3c_20int64_5ft_2c_20n_20_3e_11',['ScalarT< int64_t, N >',['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['scalart_3c_20int8_5ft_2c_20n_20_3e_12',['ScalarT< int8_t, N >',['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['scalart_3c_20uint64_5ft_2c_20n_20_3e_13',['ScalarT< uint64_t, N >',['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['scalarvector_14',['ScalarVector',['../structmlx_1_1core_1_1_scalar_vector.html',1,'mlx::core::ScalarVector< Op >'],['../structmlx_1_1core_1_1_scalar_vector.html#a69d6a3ddd7586e8e19a42c5e6f5a287b',1,'mlx::core::ScalarVector::ScalarVector()'],['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6aabac63719294588466e3c2f00cccb0a6',1,'mlx::core::ScalarVector']]], + ['scale_15',['scale',['../struct_scale_op.html#a02043fac21c68fb8d6863a01f45ede4b',1,'ScaleOp::scale'],['../struct_transform_scale.html#aa56b8e107acf16fdf77006625c2b8bc6',1,'TransformScale::scale'],['../structmlx_1_1steel_1_1_attn_params.html#ad81bcd32e6ff8fec0000eca505fb6826',1,'mlx::steel::AttnParams::scale']]], + ['scaled_5fdot_5fproduct_5fattention_16',['scaled_dot_product_attention',['../namespacemlx_1_1core_1_1fast.html#a3663b50265b0a9c0cca2b5376852e059',1,'mlx::core::fast']]], + ['scaleddotproductattention_17',['ScaledDotProductAttention',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html',1,'mlx::core::fast::ScaledDotProductAttention'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ab3f78d30e5bb3e76cfe701f2358e4748',1,'mlx::core::fast::ScaledDotProductAttention::ScaledDotProductAttention()']]], + ['scaleop_18',['ScaleOp',['../struct_scale_op.html',1,'']]], + ['scales_19',['scales',['../struct_quantized_block_loader.html#a6123e4a9209d6eacb58b2c2344ed1ecf',1,'QuantizedBlockLoader']]], + ['scan_20',['Scan',['../classmlx_1_1core_1_1_scan.html',1,'mlx::core::Scan'],['../classmlx_1_1core_1_1_scan.html#ac93e8f9c6771de825d2186ef34fa7087',1,'mlx::core::Scan::Scan()']]], + ['scan_21',['scan',['../namespacemlx_1_1core_1_1metal.html#a81c2cf124b0803098a54a78f8f6873a6',1,'mlx::core::metal']]], + ['scan_2eh_22',['scan.h',['../scan_8h.html',1,'']]], + ['scatter_23',['Scatter',['../classmlx_1_1core_1_1_scatter.html',1,'mlx::core::Scatter'],['../classmlx_1_1core_1_1_scatter.html#ac9b3eff67389ef9aa820753379ffeaa3',1,'mlx::core::Scatter::Scatter()']]], + ['scatter_24',['scatter',['../namespacemlx_1_1core_1_1metal.html#a32e902c6cd6d35fcc3119ed6685a170f',1,'mlx::core::metal::scatter()'],['../group__ops.html#gad438be8f90bae9d37c6853b8f4225d61',1,'mlx::core::scatter(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gac2c2b379a3ce959dbe1c4a68f112edfe',1,'mlx::core::scatter(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatter_2eh_25',['scatter.h',['../scatter_8h.html',1,'']]], + ['scatter_5fadd_26',['scatter_add',['../group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174',1,'mlx::core::scatter_add(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gac13318518e5703f1273c5366eb523a5a',1,'mlx::core::scatter_add(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatter_5fadd_5faxis_27',['scatter_add_axis',['../group__ops.html#gab3fd98c0d06b84b836f93bddbd7a2a0d',1,'mlx::core']]], + ['scatter_5faxis_28',['scatter_axis',['../namespacemlx_1_1core_1_1metal.html#a88c1d42d525fcdfb2f9e8aa2c3f82ea6',1,'mlx::core::metal::scatter_axis()'],['../scatter__axis_8h.html#af78a7935b05dabd42c2cdff4cf375130',1,'scatter_axis(const device T *upd, const device IdxT *indices, device mlx_atomic< T > *out, const constant int *shape, const constant int64_t *upd_strides, const constant int64_t *idx_strides, const constant size_t &ndim, const constant int &axis, const constant int &out_axis_size, const constant size_t &upd_ax_stride, const constant size_t &idx_ax_stride, uint3 index, uint3 grid_dim): scatter_axis.h']]], + ['scatter_5faxis_2eh_29',['scatter_axis.h',['../scatter__axis_8h.html',1,'']]], + ['scatter_5fimpl_30',['scatter_impl',['../scatter_8h.html#ab72d4fe1dbd4ae4dc529ee2ec8164fa4',1,'scatter.h']]], + ['scatter_5fkernels_31',['scatter_kernels',['../jit_2indexing_8h.html#a768c949cd650a44c6b402fc1440c1a56',1,'indexing.h']]], + ['scatter_5fmax_32',['scatter_max',['../group__ops.html#ga05881a4157cd113c9392d168a79e6673',1,'mlx::core::scatter_max(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga9adda5f9202bb3486e4d9e1114e3a56f',1,'mlx::core::scatter_max(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatter_5fmin_33',['scatter_min',['../group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5',1,'mlx::core::scatter_min(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga51fa762a997c243ca7a19e1ed3e83199',1,'mlx::core::scatter_min(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatter_5fprod_34',['scatter_prod',['../group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0',1,'mlx::core::scatter_prod(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gaf83c53c453faa9083ba27e4b97539339',1,'mlx::core::scatter_prod(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatteraxis_35',['ScatterAxis',['../classmlx_1_1core_1_1_scatter_axis.html',1,'mlx::core::ScatterAxis'],['../classmlx_1_1core_1_1_scatter_axis.html#a7365a2c5fddb1c39509998598de411db',1,'mlx::core::ScatterAxis::ScatterAxis()']]], + ['scheduled_36',['scheduled',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078af8a6f8eed2395ab89a758dec434393ae',1,'mlx::core::array']]], + ['scheduler_37',['Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html',1,'mlx::core::scheduler::Scheduler'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a3ae42aed78a2200e9d02776fcd2316ba',1,'mlx::core::scheduler::Scheduler::Scheduler()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a61a74e3628899e66dde600e24a750648',1,'mlx::core::scheduler::Scheduler::Scheduler(const Scheduler &)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ac3f77b7c93220dadd0b3bb2e903b7059',1,'mlx::core::scheduler::Scheduler::Scheduler(Scheduler &&)=delete']]], + ['scheduler_38',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html#ae856e468c2f7c8f8ec672522cc13730b',1,'mlx::core::scheduler']]], + ['scheduler_2eh_39',['scheduler.h',['../scheduler_8h.html',1,'']]], + ['sdpa_5fvector_40',['sdpa_vector',['../sdpa__vector_8h.html#a826f7a3c7ab843abc0842241db3e57b3',1,'sdpa_vector.h']]], + ['sdpa_5fvector_2eh_41',['sdpa_vector.h',['../sdpa__vector_8h.html',1,'']]], + ['sdpa_5fvector_5f2pass_5f1_42',['sdpa_vector_2pass_1',['../sdpa__vector_8h.html#aae1a2f23b03e24734805b08ebc5c1a59',1,'sdpa_vector.h']]], + ['sdpa_5fvector_5f2pass_5f2_43',['sdpa_vector_2pass_2',['../sdpa__vector_8h.html#a1368cf3618a4e03dbf743b3463205efe',1,'sdpa_vector.h']]], + ['seed_44',['seed',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a9f19c5da2031cba50d0ff996924347d8',1,'mlx::core::random::KeySequence::seed()'],['../namespacemlx_1_1core_1_1random.html#ac4ad325b613257306df74595d3d0e23b',1,'mlx::core::random::seed()']]], + ['seek_45',['seek',['../structmlx_1_1core_1_1_contiguous_iterator.html#af08f009e0a72414d274db2ff1b2c7dd5',1,'mlx::core::ContiguousIterator::seek()'],['../classmlx_1_1core_1_1io_1_1_reader.html#acea55078bd39ccaa27a9a36f17a39cd1',1,'mlx::core::io::Reader::seek()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a9c1716dda53aa36faea9c8fb1a3e34d4',1,'mlx::core::io::Writer::seek()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a4434ee18ff8bbf1b4fce670a337b535f',1,'mlx::core::io::ParallelFileReader::seek()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9646f4ea048ae58719daeb588e2de433',1,'mlx::core::io::FileWriter::seek()']]], + ['select_46',['Select',['../structmlx_1_1core_1_1detail_1_1_select.html',1,'mlx::core::detail::Select'],['../classmlx_1_1core_1_1_select.html',1,'mlx::core::Select'],['../struct_select.html',1,'Select'],['../classmlx_1_1core_1_1_select.html#a6f833fe55dd68ad3726bbf9a8f75eec9',1,'mlx::core::Select::Select()']]], + ['select_47',['select',['../namespacemlx_1_1core_1_1simd.html#afb3bcbd8d8b34128cd0c8eb677a170ef',1,'mlx::core::simd::select(Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a9e3e7b35d564c70de8fa0b6150570ed8',1,'mlx::core::simd::select(Simd< MaskT, 1 > mask, Simd< T, 1 > x, Simd< T, 1 > y)'],['../namespacemlx_1_1core_1_1simd.html#a3b5ebb46e7beae839c97b2e7ed9c7426',1,'mlx::core::simd::select(Simd< MaskT, N > mask, Simd< float16_t, N > x, Simd< float16_t, N > y)']]], + ['send_48',['Send',['../classmlx_1_1core_1_1distributed_1_1_send.html',1,'mlx::core::distributed::Send'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a2481dd876b14d4a13ac466cbca9c4eac',1,'mlx::core::distributed::Send::Send()']]], + ['send_49',['send',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac8472eb2f96d1b14c7e4ccef56268ba0',1,'mlx::core::distributed::detail::GroupImpl::send()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#abf33511660ac71df5fc92f2aad6c6e08',1,'mlx::core::distributed::detail::send()'],['../namespacemlx_1_1core_1_1distributed.html#a5a8360edaa3a528a3927fce4d2cf1777',1,'mlx::core::distributed::send()']]], + ['set_50',['Set',['../structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807',1,'pocketfft::detail::cmplx::Set(T r_, T i_)'],['../structpocketfft_1_1detail_1_1cmplx.html#a447d26b2e07f6e45f29d865e906c0a98',1,'pocketfft::detail::cmplx::Set(T r_)']]], + ['set_5fbinary_5fop_5foutput_5fdata_51',['set_binary_op_output_data',['../namespacemlx_1_1core.html#a6a52856325c2eb031d3983eba2108d59',1,'mlx::core']]], + ['set_5fbuffer_52',['set_buffer',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ae890f5cefa4ae24ae0f5d8e46a313a92',1,'mlx::core::metal::CommandEncoder::set_buffer()'],['../structmlx_1_1core_1_1_command_encoder.html#ae890f5cefa4ae24ae0f5d8e46a313a92',1,'mlx::core::CommandEncoder::set_buffer()']]], + ['set_5fbytes_53',['set_bytes',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a9c343f791812a45c6c03a5c9f27f74d5',1,'mlx::core::metal::CommandEncoder::set_bytes(const T *v, int n, int idx)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#abc52d18ea87d213c47fd26062c829849',1,'mlx::core::metal::CommandEncoder::set_bytes(const T &v, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a9c343f791812a45c6c03a5c9f27f74d5',1,'mlx::core::CommandEncoder::set_bytes(const T *v, int n, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#abc52d18ea87d213c47fd26062c829849',1,'mlx::core::CommandEncoder::set_bytes(const T &v, int idx)']]], + ['set_5fcache_5flimit_54',['set_cache_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#af392bced29d9e4e3f1a7cc4725d83764',1,'mlx::core::metal::MetalAllocator::set_cache_limit()'],['../namespacemlx_1_1core_1_1metal.html#ab09c9b60f1e886ab859e6a066c9a5b9d',1,'mlx::core::metal::set_cache_limit()']]], + ['set_5fcompile_5fmode_55',['set_compile_mode',['../namespacemlx_1_1core.html#a49445a55f976c4397f25ea18e1e92bef',1,'mlx::core']]], + ['set_5fcompute_5fpipeline_5fstate_56',['set_compute_pipeline_state',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a6d4c03a6585deedb5ccd1a1057d0c6ef',1,'mlx::core::metal::CommandEncoder::set_compute_pipeline_state()'],['../structmlx_1_1core_1_1_command_encoder.html#a6d4c03a6585deedb5ccd1a1057d0c6ef',1,'mlx::core::CommandEncoder::set_compute_pipeline_state()']]], + ['set_5fdata_57',['set_data',['../classmlx_1_1core_1_1array.html#af9e3a02b4c0023c36248dc75c887214f',1,'mlx::core::array::set_data(allocator::Buffer buffer, Deleter d=allocator::free)'],['../classmlx_1_1core_1_1array.html#a5f338202a39d37fa3f4241e851a15838',1,'mlx::core::array::set_data(allocator::Buffer buffer, size_t data_size, Strides strides, Flags flags, Deleter d=allocator::free)']]], + ['set_5fdefault_5fdevice_58',['set_default_device',['../namespacemlx_1_1core.html#a312a2de41367fe52caeaf8c0f596a120',1,'mlx::core']]], + ['set_5fdefault_5fstream_59',['set_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a6d15314ac9cf25efc9bd1278de9a66bb',1,'mlx::core::scheduler::Scheduler::set_default_stream()'],['../namespacemlx_1_1core.html#af35a2b06517d8bb7dbb469692b4f841c',1,'mlx::core::set_default_stream()']]], + ['set_5finput_5farray_60',['set_input_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ab69ff0d7f14b9b59db4df0608193dce4',1,'mlx::core::metal::CommandEncoder::set_input_array()'],['../structmlx_1_1core_1_1_command_encoder.html#ab69ff0d7f14b9b59db4df0608193dce4',1,'mlx::core::CommandEncoder::set_input_array()']]], + ['set_5fmemory_5flimit_61',['set_memory_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a179e3127ef9377ce54295f771c34ba1b',1,'mlx::core::metal::MetalAllocator::set_memory_limit()'],['../namespacemlx_1_1core_1_1metal.html#a3fb2c4a237fa4bfdff798156146c4937',1,'mlx::core::metal::set_memory_limit()']]], + ['set_5fname_62',['set_name',['../structmlx_1_1core_1_1_node_namer.html#a57a574e48f8a9cd122616d80b138c768',1,'mlx::core::NodeNamer']]], + ['set_5foutput_5farray_63',['set_output_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a6a2e28e542eaa2886041bddd51ff6522',1,'mlx::core::metal::CommandEncoder::set_output_array()'],['../structmlx_1_1core_1_1_command_encoder.html#a6a2e28e542eaa2886041bddd51ff6522',1,'mlx::core::CommandEncoder::set_output_array()']]], + ['set_5fresidency_5fset_64',['set_residency_set',['../classmlx_1_1core_1_1metal_1_1_device.html#a03a2f0c712660a1bd437cb16e4aba79f',1,'mlx::core::metal::Device']]], + ['set_5fsiblings_65',['set_siblings',['../classmlx_1_1core_1_1array.html#a8fccbe7a4edfd8cca168161124e263b1',1,'mlx::core::array']]], + ['set_5fstatus_66',['set_status',['../classmlx_1_1core_1_1array.html#a63598018999b49f1340b183cb303f05c',1,'mlx::core::array']]], + ['set_5fternary_5fop_5foutput_5fdata_67',['set_ternary_op_output_data',['../namespacemlx_1_1core.html#a6f4528d0d338ea5e1f19d345875c26a2',1,'mlx::core']]], + ['set_5ftracer_68',['set_tracer',['../classmlx_1_1core_1_1array.html#af26e6be1a9e6239471a4c24310c0c7c8',1,'mlx::core::array']]], + ['set_5funary_5foutput_5fdata_69',['set_unary_output_data',['../namespacemlx_1_1core.html#a4c6a4241bfcdd7bbf30d0e521b79e5a3',1,'mlx::core']]], + ['set_5fvalue_70',['set_value',['../classmlx_1_1core_1_1_event.html#a0d077b11f4b28f882b42440b7ac6d40d',1,'mlx::core::Event']]], + ['set_5fvector_5fbytes_71',['set_vector_bytes',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a68c3c6a036e11ec40211c09811bbed1b',1,'mlx::core::metal::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, size_t nelems, int idx)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a7375adf9ee5355bcf4b7f5f210efd115',1,'mlx::core::metal::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a68c3c6a036e11ec40211c09811bbed1b',1,'mlx::core::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, size_t nelems, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a7375adf9ee5355bcf4b7f5f210efd115',1,'mlx::core::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, int idx)']]], + ['set_5fwired_5flimit_72',['set_wired_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a84fa0347da18055bc13ba0a5c4b57253',1,'mlx::core::metal::MetalAllocator::set_wired_limit()'],['../namespacemlx_1_1core_1_1metal.html#a31eab4828d31d292bc84e07b0d961e1e',1,'mlx::core::metal::set_wired_limit()']]], + ['shape_73',['Shape',['../namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416',1,'mlx::core']]], + ['shape_74',['shape',['../structmlx_1_1core_1_1_reduction_plan.html#a1576dc3d2e01b3f1e11816151070dd1a',1,'mlx::core::ReductionPlan::shape'],['../structmlx_1_1steel_1_1_layout2_d.html#a23183747ab1ddbdd3f1fcac6d0faa2cd',1,'mlx::steel::Layout2D::shape'],['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63db720fe0c2abc4b71e22a58a015f8a',1,'mlx::core::fast::CustomKernelShapeInfo::shape'],['../classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0',1,'pocketfft::detail::arr_info::shape() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8',1,'pocketfft::detail::arr_info::shape(size_t i) const'],['../classmlx_1_1core_1_1array.html#a1d06c76b0f3010a5c329d0e9e29e0597',1,'mlx::core::array::shape() const'],['../classmlx_1_1core_1_1array.html#ac50382b652f6e8fbd50d42b7ff595810',1,'mlx::core::array::shape(int dim) const']]], + ['shape2d_75',['Shape2D',['../structmlx_1_1steel_1_1_shape2_d.html',1,'mlx::steel::Shape2D< RInt, CInt >'],['../structmlx_1_1steel_1_1_shape2_d.html#a070ce70eb6d84361c7f313159c438a5c',1,'mlx::steel::Shape2D::Shape2D()']]], + ['shape_5ft_76',['shape_t',['../namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::detail::shape_t'],['../namespacepocketfft.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::shape_t']]], + ['shapeelem_77',['ShapeElem',['../namespacemlx_1_1core.html#a167cdec84c0ae62b5b299c617384346e',1,'mlx::core']]], + ['shapes_78',['shapes',['../struct_indices.html#a5ab170f1a77636180889ddfffd4f7d2f',1,'Indices']]], + ['shapes_5fwithout_5freduction_5faxes_79',['shapes_without_reduction_axes',['../namespacemlx_1_1core.html#a0bea91a360a984e72d2815353f97ee25',1,'mlx::core']]], + ['shared_5fbuffer_5freshape_80',['shared_buffer_reshape',['../namespacemlx_1_1core.html#a88d88987bd8bf3ca46bf3b5e8aacce9d',1,'mlx::core']]], + ['shared_5fbuffer_5fslice_81',['shared_buffer_slice',['../namespacemlx_1_1core.html#a349a9fc2bfd950f679a3fe39b8bdedad',1,'mlx::core']]], + ['shp_82',['shp',['../classpocketfft_1_1detail_1_1arr__info.html#a2467e9e01de1ba4d7cd28c1af783da8d',1,'pocketfft::detail::arr_info']]], + ['shutdown_83',['shutdown',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411',1,'pocketfft::detail::threading::thread_pool']]], + ['siblings_84',['siblings',['../classmlx_1_1core_1_1array.html#acf80fde8f743f65ad5b4be69fcb7a74d',1,'mlx::core::array::siblings() const'],['../classmlx_1_1core_1_1array.html#a7263f23e70a580a9bc2129fbcde36e6c',1,'mlx::core::array::siblings()']]], + ['sigmoid_85',['Sigmoid',['../structmlx_1_1core_1_1detail_1_1_sigmoid.html',1,'mlx::core::detail::Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html',1,'mlx::core::Sigmoid'],['../struct_sigmoid.html',1,'Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html#a47eca99113ec19f0eb60b6a0472c592b',1,'mlx::core::Sigmoid::Sigmoid()']]], + ['sigmoid_86',['sigmoid',['../group__ops.html#ga708abf8f79609cd6831db7c38cafac0e',1,'mlx::core']]], + ['sign_87',['Sign',['../structmlx_1_1core_1_1detail_1_1_sign.html',1,'mlx::core::detail::Sign'],['../classmlx_1_1core_1_1_sign.html',1,'mlx::core::Sign'],['../struct_sign.html',1,'Sign'],['../classmlx_1_1core_1_1_sign.html#afe951e50907bc23a601ec5fa9eae5763',1,'mlx::core::Sign::Sign()']]], + ['sign_88',['sign',['../group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa',1,'mlx::core']]], + ['signal_89',['signal',['../classmlx_1_1core_1_1_event.html#a65a858445506a61be5889ae0e3651b89',1,'mlx::core::Event']]], + ['signaling_5fnan_90',['signaling_NaN',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ad1f76a43c7d51a3765174aa6e0dd9f80',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['signedinteger_91',['signedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2daed58b4631ff157bec9e35ed1182d2c10',1,'mlx::core::Dtype::signedinteger'],['../namespacemlx_1_1core.html#a24e1618af591d737d73729665e868001',1,'mlx::core::signedinteger']]], + ['simd_92',['Simd',['../structmlx_1_1core_1_1simd_1_1_simd.html',1,'mlx::core::simd::Simd< T, N >'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a5c24246e05e833fd81d900226a29e6ab',1,'mlx::core::simd::Simd::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a1693c8e542dddf2ab60d309d64de71b6',1,'mlx::core::simd::Simd::Simd(Simd< U, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a74091cd8b7c72014f73ec215b52ea2ce',1,'mlx::core::simd::Simd::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aabe757c8fd93dadd6f8859cda99f4927',1,'mlx::core::simd::Simd::Simd(Simd< T, N/2 > x, Simd< T, N/2 > y)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#a3f6e4a83ecf897465f44160b6fad5a7a',1,'mlx::core::simd::Simd< T, 1 >::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#a585bc4768c4f7e1313d7e8756fbb00cc',1,'mlx::core::simd::Simd< T, 1 >::Simd(Simd< U, 1 > v)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#acf948f7c5e8829432c0ac17fc9f911e2',1,'mlx::core::simd::Simd< T, 1 >::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a04a3a73f98fa5c9090b6cf6154e99e8d',1,'mlx::core::simd::Simd< float16_t, N >::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#ad8b628f8834e983853d557cc1e4124bb',1,'mlx::core::simd::Simd< float16_t, N >::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a5e76655d70c0e9ae49eea536c0e3b8cf',1,'mlx::core::simd::Simd< float16_t, N >::Simd(float16x8_t v)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#ae1dfcaca51f9a6fcdb757cb8413ac223',1,'mlx::core::simd::Simd< float16_t, N >::Simd(Simd< float, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a1f30c088a6828d0673e927ed6c0a4b2b',1,'mlx::core::simd::Simd< float16_t, N >::Simd(Simd< uint16_t, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a5c24246e05e833fd81d900226a29e6ab',1,'mlx::core::simd::Simd< T, 1 >::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a1693c8e542dddf2ab60d309d64de71b6',1,'mlx::core::simd::Simd< T, 1 >::Simd(Simd< U, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a74091cd8b7c72014f73ec215b52ea2ce',1,'mlx::core::simd::Simd< T, 1 >::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aabe757c8fd93dadd6f8859cda99f4927',1,'mlx::core::simd::Simd< T, 1 >::Simd(Simd< T, N/2 > x, Simd< T, N/2 > y)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a5c24246e05e833fd81d900226a29e6ab',1,'mlx::core::simd::Simd< float16_t, N >::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a1693c8e542dddf2ab60d309d64de71b6',1,'mlx::core::simd::Simd< float16_t, N >::Simd(Simd< U, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a74091cd8b7c72014f73ec215b52ea2ce',1,'mlx::core::simd::Simd< float16_t, N >::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aabe757c8fd93dadd6f8859cda99f4927',1,'mlx::core::simd::Simd< float16_t, N >::Simd(Simd< float16_t, N/2 > x, Simd< float16_t, N/2 > y)']]], + ['simd_2eh_93',['simd.h',['../simd_8h.html',1,'']]], + ['simd_3c_20float16_5ft_2c_20n_20_3e_94',['Simd< float16_t, N >',['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['simd_3c_20t_2c_201_20_3e_95',['Simd< T, 1 >',['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html',1,'mlx::core::simd']]], + ['simd_5fbroadcast_96',['simd_broadcast',['../namespacemetal.html#a498f1e85107eb5f01ba4435977f8efe0',1,'metal']]], + ['simd_5fdefault_5fbinary_97',['SIMD_DEFAULT_BINARY',['../accelerate__simd_8h.html#a9aa795d90ddc485f24cd4a5268fe0e82',1,'accelerate_simd.h']]], + ['simd_5fdefault_5fcomparisons_98',['SIMD_DEFAULT_COMPARISONS',['../accelerate__simd_8h.html#aca22aa431d399cea13c969926689b459',1,'accelerate_simd.h']]], + ['simd_5fdefault_5funary_99',['SIMD_DEFAULT_UNARY',['../accelerate__simd_8h.html#a3b449b9ce2c623ab4dce3f6fe349bb6d',1,'accelerate_simd.h']]], + ['simd_5fexclusive_5fscan_100',['simd_exclusive_scan',['../struct_cum_prod_3_01bool_01_4.html#a1a86e9398bae24182b7be0a6577bf223',1,'CumProd< bool >::simd_exclusive_scan()'],['../struct_cum_max.html#ae11b67aa6c998e9a01615b2a79af4403',1,'CumMax::simd_exclusive_scan()'],['../struct_cum_min.html#a83e65017ff33018b585c043fb803773b',1,'CumMin::simd_exclusive_scan()']]], + ['simd_5ffp16_5fdefault_5fbinary_101',['SIMD_FP16_DEFAULT_BINARY',['../accelerate__fp16__simd_8h.html#a6437645d67e6a8b7c1349b0040aaa6bb',1,'accelerate_fp16_simd.h']]], + ['simd_5ffp16_5fdefault_5funary_102',['SIMD_FP16_DEFAULT_UNARY',['../accelerate__fp16__simd_8h.html#a60af883c0392ecab270ecc1146f5e8a2',1,'accelerate_fp16_simd.h']]], + ['simd_5fmax_103',['simd_max',['../namespacemetal.html#a048cad0aca52cb737ebf103e76bd1c49',1,'metal']]], + ['simd_5fmin_104',['simd_min',['../namespacemetal.html#ae9e2a23e00724ba2d7868bc4112b386b',1,'metal']]], + ['simd_5fprefix_5fexclusive_5fproduct_105',['simd_prefix_exclusive_product',['../namespacemetal.html#a5ca40242390b632f737e29636829b2e4',1,'metal']]], + ['simd_5fprefix_5fexclusive_5fsum_106',['simd_prefix_exclusive_sum',['../namespacemetal.html#abfbb70c7471f28bf7ff36a612ad014b2',1,'metal']]], + ['simd_5fprefix_5finclusive_5fproduct_107',['simd_prefix_inclusive_product',['../namespacemetal.html#a6ca6a7e1996228fa536e969e9e45c446',1,'metal']]], + ['simd_5fprefix_5finclusive_5fsum_108',['simd_prefix_inclusive_sum',['../namespacemetal.html#a567acb18199ac0107712eb8cb8aeb8e9',1,'metal']]], + ['simd_5fproduct_109',['simd_product',['../namespacemetal.html#ac6e883a04e2265a9790d7db76059e1b4',1,'metal']]], + ['simd_5fscan_110',['simd_scan',['../struct_cum_prod_3_01bool_01_4.html#abeb5ec4237b330e7219f4e881cf10d7a',1,'CumProd< bool >::simd_scan()'],['../struct_cum_max.html#adc9ec8bb09b4433d4c2f03022c43d781',1,'CumMax::simd_scan()'],['../struct_cum_min.html#a0a1005d91b1c90e90e2c6dbd6c296649',1,'CumMin::simd_scan()']]], + ['simd_5fshuffle_111',['simd_shuffle',['../namespacemetal.html#a259ed115bc3c58f88eb35830916b26d4',1,'metal::simd_shuffle()'],['../backend_2metal_2kernels_2utils_8h.html#a71986ecdd7d18f975dd22c3df7421ce2',1,'simd_shuffle(uint64_t data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a3bdbdfeb7a1dde40cd3ce1df8d9213b5',1,'simd_shuffle(int64_t data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ab4cbcdb054f9165130da91a3334da0cf',1,'simd_shuffle(bool data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ab8175b66bcc080fb89f738143568c30b',1,'simd_shuffle(complex64_t data, uint16_t lane): utils.h']]], + ['simd_5fshuffle_5fand_5ffill_5fdown_112',['simd_shuffle_and_fill_down',['../namespacemetal.html#ae29a06f0eac636ad7af21dea5b04938b',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a0ee6239fa29a5f9ee0201e0dc5ddc8e0',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], + ['simd_5fshuffle_5fand_5ffill_5fup_113',['simd_shuffle_and_fill_up',['../namespacemetal.html#a1ca14116bf50639b214d8414b5bbaaa6',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a5138d5cdc18139e135707916a243cd8e',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta)'],['../backend_2metal_2kernels_2utils_8h.html#a5862d5ea154c9b76cf56a630cf6385b4',1,'simd_shuffle_and_fill_up(uint64_t data, uint64_t filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a7bb56415c5412a6a26f70a990915f064',1,'simd_shuffle_and_fill_up(int64_t data, int64_t filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ad55bd473647f2c6c68e65e5312c132d1',1,'simd_shuffle_and_fill_up(bool data, bool filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a94e02a6ae8c39cbf4cb23aa44df9dbd5',1,'simd_shuffle_and_fill_up(complex64_t data, complex64_t filling, uint16_t delta): utils.h']]], + ['simd_5fshuffle_5fdown_114',['simd_shuffle_down',['../namespacemetal.html#af6e2dd7ae087aba6abac4f0350b7611c',1,'metal::simd_shuffle_down()'],['../backend_2metal_2kernels_2utils_8h.html#aba6279624b1d30c525efee856a222b5c',1,'simd_shuffle_down(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a0c1e4d782fcc56e1ab5565cef12430dd',1,'simd_shuffle_down(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a48ae83a8caf5c74810df60b6c6cdb062',1,'simd_shuffle_down(bool data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ad9a671a5f9aaa729ae7a77026f16bcb0',1,'simd_shuffle_down(complex64_t data, uint16_t delta): utils.h']]], + ['simd_5fshuffle_5frotate_5fdown_115',['simd_shuffle_rotate_down',['../namespacemetal.html#a4bb203647a421032db47e73cd649841b',1,'metal']]], + ['simd_5fshuffle_5frotate_5fup_116',['simd_shuffle_rotate_up',['../namespacemetal.html#a729b22077d6c944491a6027c18ea80c9',1,'metal']]], + ['simd_5fshuffle_5fup_117',['simd_shuffle_up',['../namespacemetal.html#afe81c5fbde3f4890458b081909242c55',1,'metal::simd_shuffle_up()'],['../backend_2metal_2kernels_2utils_8h.html#a39e436e0a942912266aae7e0bd82d7c0',1,'simd_shuffle_up(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a617f3857caf33c569afa6148135f8b7a',1,'simd_shuffle_up(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ae0f5c42020275a588234e69f1eb7a485',1,'simd_shuffle_up(bool data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a92b455bac6a23af51c35ea83de2383eb',1,'simd_shuffle_up(complex64_t data, uint16_t delta): utils.h']]], + ['simd_5fshuffle_5fxor_118',['simd_shuffle_xor',['../namespacemetal.html#a5017efc9605e069cfb507137cd1a1852',1,'metal']]], + ['simd_5fsize_119',['SIMD_SIZE',['../quantized_8h.html#a62969a218d93680f5e35d0c61b160b99',1,'quantized.h']]], + ['simd_5fsize_120',['simd_size',['../backend_2metal_2kernels_2reduction_2ops_8h.html#a515b75d563a93d3c09ee677948dc83e3',1,'ops.h']]], + ['simd_5fsum_121',['simd_sum',['../namespacemetal.html#a85181e37a00cb4a4217f1bb25389bce5',1,'metal']]], + ['simd_5fxor_122',['simd_xor',['../namespacemetal.html#a1308decbf2d5c33d34d6be523ea1c30f',1,'metal']]], + ['simple_5fiter_123',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html',1,'pocketfft::detail::simple_iter'],['../classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398',1,'pocketfft::detail::simple_iter::simple_iter()']]], + ['simplevalueandgradfn_124',['SimpleValueAndGradFn',['../namespacemlx_1_1core.html#a2689b8f1181648cb1685204fea9f3066',1,'mlx::core']]], + ['sin_125',['Sin',['../structmlx_1_1core_1_1detail_1_1_sin.html',1,'mlx::core::detail::Sin'],['../classmlx_1_1core_1_1_sin.html',1,'mlx::core::Sin'],['../struct_sin.html',1,'Sin'],['../classmlx_1_1core_1_1_sin.html#a10d1ecc0ca96e79cdf55b57073d126ea',1,'mlx::core::Sin::Sin()']]], + ['sin_126',['sin',['../namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258',1,'pocketfft::detail::sin()'],['../namespacemlx_1_1core_1_1simd.html#ab4d582d72c0a7ee313e19c906e43cef1',1,'mlx::core::simd::sin()'],['../namespacemetal.html#a619a159ca5f2ddfe3647d3a6bb6e804c',1,'metal::sin()'],['../namespacemetal_1_1fast.html#a3af771cfe7a135104f9d063147dba270',1,'metal::fast::sin()'],['../namespacemetal_1_1precise.html#a71acf77ffd29c56f56afae0195c98a1c',1,'metal::precise::sin()'],['../group__ops.html#gaebf0a73ad3732fba39df37826c235692',1,'mlx::core::sin()']]], + ['sincos_127',['sincos',['../namespacemlx_1_1core_1_1simd.html#ad78056685c9732c3465c0d8b8ec1bef7',1,'mlx::core::simd']]], + ['sincos_5f2pibyn_128',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html',1,'pocketfft::detail::sincos_2pibyn< T >'],['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40',1,'pocketfft::detail::sincos_2pibyn::sincos_2pibyn()']]], + ['single_129',['SINGLE',['../cpu_2unary__ops_8h.html#a602aea95990389a45c255195f849d5de',1,'unary_ops.h']]], + ['sinh_130',['Sinh',['../structmlx_1_1core_1_1detail_1_1_sinh.html',1,'mlx::core::detail::Sinh'],['../classmlx_1_1core_1_1_sinh.html',1,'mlx::core::Sinh'],['../struct_sinh.html',1,'Sinh'],['../classmlx_1_1core_1_1_sinh.html#a4a4f6814d403c2ce5d6c574b0dca3c96',1,'mlx::core::Sinh::Sinh()']]], + ['sinh_131',['sinh',['../namespacemlx_1_1core_1_1simd.html#a09a2f3f2bc999c16babf3d8d90994d6e',1,'mlx::core::simd::sinh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a479ccddac341bd0760857b77e449e5e1',1,'mlx::core::simd::sinh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a40879bf874309c0a5abef783aea2057d',1,'mlx::core::simd::sinh(Simd< T, 1 > in)'],['../namespacemetal.html#a83ba4235ae350ab8880a9df09158620b',1,'metal::sinh()'],['../namespacemetal_1_1fast.html#a990d90b3440e38d1fb4ff5065c6c189b',1,'metal::fast::sinh()'],['../namespacemetal_1_1precise.html#abc8f4f59dd6e7204ab5d84f0af96331c',1,'metal::precise::sinh()'],['../group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7',1,'mlx::core::sinh()']]], + ['sinpi_132',['sinpi',['../namespacemetal.html#ae9655f7fa2ba6c0625ca25fbb278e269',1,'metal::sinpi()'],['../namespacemetal_1_1fast.html#ab07a32fe544aa304577d29e0251e87b2',1,'metal::fast::sinpi()'],['../namespacemetal_1_1precise.html#a78b17dab93519d9c82c2575dafec49c9',1,'metal::precise::sinpi()']]], + ['size_133',['size',['../structmlx_1_1core_1_1simd_1_1_simd.html#aa2b56facc70ba4e8d33a74def204a1fd',1,'mlx::core::simd::Simd::size'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#a14e16c6e2ef5e89135cf8e85dc9f1f1f',1,'mlx::core::simd::Simd< T, 1 >::size'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a2629cb8da72b6f922ed14cc7b6c43ce7',1,'mlx::core::simd::Simd< float16_t, N >::size'],['../classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952',1,'pocketfft::detail::arr::size()'],['../classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5',1,'pocketfft::detail::arr_info::size()'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a2adf9a9c968f113dde830cc0dc27dcc6',1,'mlx::core::allocator::Allocator::size()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#aafa92e8310db089b1ac72b840777e26b',1,'mlx::core::allocator::CommonAllocator::size()'],['../classmlx_1_1core_1_1array.html#a598f87161926d9e0b516860f0ea2c8f6',1,'mlx::core::array::size()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a51f6587e8065be16f0418ca42a796e05',1,'mlx::core::metal::MetalAllocator::size()'],['../structmlx_1_1core_1_1distributed_1_1_group.html#ac0a5a1e463a9330355e8bfe09c0feaf2',1,'mlx::core::distributed::Group::size()'],['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ab1c8044b05f185c4bcc53002d4587599',1,'mlx::core::distributed::detail::GroupImpl::size()'],['../structmlx_1_1core_1_1_dtype.html#ab54051563d85212c7f0f049166bc9971',1,'mlx::core::Dtype::size()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aa2b56facc70ba4e8d33a74def204a1fd',1,'mlx::core::simd::Simd< T, 1 >::size'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aa2b56facc70ba4e8d33a74def204a1fd',1,'mlx::core::simd::Simd< float16_t, N >::size']]], + ['size_5fof_134',['size_of',['../namespacemlx_1_1core.html#add4794cc0ffe5d717fc146084a235d95',1,'mlx::core']]], + ['slice_135',['Slice',['../classmlx_1_1core_1_1_slice.html',1,'mlx::core::Slice'],['../classmlx_1_1core_1_1_slice.html#a3aa025acbf4a9ca9e030a1e6bda102f7',1,'mlx::core::Slice::Slice()']]], + ['slice_136',['slice',['../namespacemlx_1_1core.html#a5adff87687b78bfc18dafbc654925cdb',1,'mlx::core::slice(const array &in, array &out, const Shape &start_indices, const Shape &strides)'],['../group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d',1,'mlx::core::slice(const array &a, Shape start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#gaedcbdf4040f5e6a02a74b1ed1c6c2ebc',1,'mlx::core::slice(const array &a, std::initializer_list< int > start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#gaec56dcb94d5e7f7b885fb60b4bf4aa9d',1,'mlx::core::slice(const array &a, Shape start, Shape stop, StreamOrDevice s={})'],['../group__ops.html#ga797996e53ea34317a55dc2f314736b89',1,'mlx::core::slice(const array &a, const array &start, std::vector< int > axes, Shape slice_size, StreamOrDevice s={})']]], + ['slice_5fgpu_137',['slice_gpu',['../namespacemlx_1_1core.html#a327578951a44116e5da2db651661265f',1,'mlx::core']]], + ['slice_5fupdate_138',['slice_update',['../group__ops.html#ga7b7e786985d27789aaed20bb2f9509be',1,'mlx::core::slice_update(const array &src, const array &update, Shape start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#ga3b7c6136ae5a38dd6457b65c1833aa67',1,'mlx::core::slice_update(const array &src, const array &update, Shape start, Shape stop, StreamOrDevice s={})'],['../group__ops.html#gacd906ffb96149a4998c321cbf2231d7e',1,'mlx::core::slice_update(const array &src, const array &update, const array &start, std::vector< int > axes, StreamOrDevice s={})']]], + ['sliceupdate_139',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html',1,'mlx::core::SliceUpdate'],['../classmlx_1_1core_1_1_slice_update.html#a63a49264b18340f2bc442c081a7b4c7a',1,'mlx::core::SliceUpdate::SliceUpdate()']]], + ['slicing_2eh_140',['slicing.h',['../common_2slicing_8h.html',1,'(Global Namespace)'],['../cpu_2slicing_8h.html',1,'(Global Namespace)'],['../metal_2slicing_8h.html',1,'(Global Namespace)']]], + ['sm_141',['sm',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa85451edf6900fd6af164d4d50889ae3',1,'mlx::steel::BlockMMA']]], + ['sn_142',['sn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ade420e8b811d597345783c324c23a34a',1,'mlx::steel::BlockMMA']]], + ['softmax_143',['Softmax',['../classmlx_1_1core_1_1_softmax.html',1,'mlx::core::Softmax'],['../classmlx_1_1core_1_1_softmax.html#a4ec686aac4e06f0dfe2cbd6801af40eb',1,'mlx::core::Softmax::Softmax()']]], + ['softmax_144',['softmax',['../namespacemlx_1_1core_1_1metal.html#a4fe937c2c584fd646926057f31d54ca6',1,'mlx::core::metal::softmax()'],['../group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6',1,'mlx::core::softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga1ae3614d07d873892a530d14c3857d0b',1,'mlx::core::softmax(const array &a, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga06f570d73716a24303e6de3aaba4457b',1,'mlx::core::softmax(const array &a, int axis, bool precise=false, StreamOrDevice s={})']]], + ['softmax_2eh_145',['softmax.h',['../jit_2softmax_8h.html',1,'(Global Namespace)'],['../kernels_2softmax_8h.html',1,'(Global Namespace)']]], + ['softmax_5fexp_146',['softmax_exp',['../kernels_2softmax_8h.html#a440d4031ee5e86159a4dd715e44a438b',1,'softmax.h']]], + ['softmax_5fkernels_147',['softmax_kernels',['../jit_2softmax_8h.html#a1cbfb210a9a765c6620e9f1247ccef12',1,'softmax.h']]], + ['softmax_5flooped_148',['softmax_looped',['../kernels_2softmax_8h.html#a8c47b0924ebfeebcca25f3dd17373276',1,'softmax.h']]], + ['softmax_5fn_5freads_149',['SOFTMAX_N_READS',['../defines_8h.html#a722995df24286b27b7da3d74b73f768d',1,'defines.h']]], + ['softmax_5fsingle_5frow_150',['softmax_single_row',['../kernels_2softmax_8h.html#a815fe70f879f318e5d6e99acf043f52b',1,'softmax.h']]], + ['sort_151',['Sort',['../classmlx_1_1core_1_1_sort.html',1,'mlx::core::Sort'],['../classmlx_1_1core_1_1_sort.html#a62943032dbd72e85ceb9b4b7211f4a44',1,'mlx::core::Sort::Sort()']]], + ['sort_152',['sort',['../struct_thread_sort.html#a7c7cf65ffe4362596009fc1c05633e7a',1,'ThreadSort::sort()'],['../struct_block_merge_sort.html#ad25513c2d02802ac35edb086aceb4ce8',1,'BlockMergeSort::sort()'],['../namespacemlx_1_1core_1_1metal.html#ab77c9a9ecaeeab8c66b712862777c24b',1,'mlx::core::metal::sort()'],['../group__ops.html#ga7fb616054665b3c2d61fa234f501f079',1,'mlx::core::sort(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaae1bc47aa737f705d0e5884270063fea',1,'mlx::core::sort(const array &a, int axis, StreamOrDevice s={})']]], + ['sort_2eh_153',['sort.h',['../sort_8h.html',1,'']]], + ['special_5fmul_154',['special_mul',['../structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874',1,'pocketfft::detail::cmplx::special_mul()'],['../namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76',1,'pocketfft::detail::special_mul()']]], + ['split_155',['Split',['../classmlx_1_1core_1_1_split.html',1,'mlx::core::Split'],['../classmlx_1_1core_1_1_split.html#ad3f4ed34b85c73683bad5d530309342f',1,'mlx::core::Split::Split()']]], + ['split_156',['split',['../structmlx_1_1core_1_1distributed_1_1_group.html#ad3682f4dc85bfe7e5464b87f6f0fd931',1,'mlx::core::distributed::Group::split()'],['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a87800a23c8160933a2d77a55a959194d',1,'mlx::core::distributed::detail::GroupImpl::split()'],['../group__ops.html#ga7534290bceab5fb3831a05d67bebce7d',1,'mlx::core::split(const array &a, int num_splits, int axis, StreamOrDevice s={})'],['../group__ops.html#ga56882d24e5fde59c266774624c892d41',1,'mlx::core::split(const array &a, int num_splits, StreamOrDevice s={})'],['../group__ops.html#ga19005414e7d8020cd6e94e06bf399b09',1,'mlx::core::split(const array &a, const Shape &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#ga9ea089f42b9940510619052b7166d9ac',1,'mlx::core::split(const array &a, const Shape &indices, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a42847b435d037a977592e355eed072af',1,'mlx::core::random::split(const array &key, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7ec057064c7326c41b536f08178861e5',1,'mlx::core::random::split(const array &key, int num, StreamOrDevice s={})']]], + ['split_5fk_5fpartition_5fsize_157',['split_k_partition_size',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a9f5a67b2343645b570e109c3837d4042',1,'mlx::steel::GEMMSpiltKParams']]], + ['split_5fk_5fpartition_5fstride_158',['split_k_partition_stride',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a12144ce89d404812cd862611d770b9fb',1,'mlx::steel::GEMMSpiltKParams']]], + ['split_5fk_5fpartitions_159',['split_k_partitions',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#ae06c27116905d4ff3b9b436e588a93fd',1,'mlx::steel::GEMMSpiltKParams']]], + ['sqrt_160',['Sqrt',['../structmlx_1_1core_1_1detail_1_1_sqrt.html',1,'mlx::core::detail::Sqrt'],['../classmlx_1_1core_1_1_sqrt.html',1,'mlx::core::Sqrt'],['../struct_sqrt.html',1,'Sqrt'],['../classmlx_1_1core_1_1_sqrt.html#a6682a7c31ca427c9d2c5ddb6a479bf29',1,'mlx::core::Sqrt::Sqrt()']]], + ['sqrt_161',['sqrt',['../namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3',1,'pocketfft::detail::sqrt()'],['../namespacemlx_1_1core_1_1simd.html#ae39b8e1d1fff94947406eeb8ec6e0414',1,'mlx::core::simd::sqrt(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a2ba6c75c0821db3e9ac525a89b3ac859',1,'mlx::core::simd::sqrt(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a7e80d3e33f2edd02310641d3e3dd5658',1,'mlx::core::simd::sqrt(Simd< float16_t, N > a)'],['../namespacemetal.html#ab3f4d4852ca0e591104fbd8e5b50d31b',1,'metal::sqrt()'],['../namespacemetal_1_1fast.html#a4218a85c7d8a74cb8055b4755205627e',1,'metal::fast::sqrt()'],['../namespacemetal_1_1precise.html#acb213467361cd2cab93a8d5ea1aa5bfd',1,'metal::precise::sqrt()'],['../group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1',1,'mlx::core::sqrt()']]], + ['square_162',['Square',['../structmlx_1_1core_1_1detail_1_1_square.html',1,'mlx::core::detail::Square'],['../classmlx_1_1core_1_1_square.html',1,'mlx::core::Square'],['../struct_square.html',1,'Square'],['../classmlx_1_1core_1_1_square.html#ab94e28d5c92e6febc1c74e525f730dc4',1,'mlx::core::Square::Square()']]], + ['square_163',['square',['../group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e',1,'mlx::core']]], + ['squeeze_164',['Squeeze',['../classmlx_1_1core_1_1_squeeze.html',1,'mlx::core::Squeeze'],['../classmlx_1_1core_1_1_squeeze.html#a032bd53dcc3d71a11d810bc3ca3ef4b0',1,'mlx::core::Squeeze::Squeeze()']]], + ['squeeze_165',['squeeze',['../group__ops.html#ga710daa7ec721bd4d3f326082cb195576',1,'mlx::core::squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3',1,'mlx::core::squeeze(const array &a, int axis, StreamOrDevice s={})'],['../group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad',1,'mlx::core::squeeze(const array &a, StreamOrDevice s={})']]], + ['src_166',['src',['../struct_quantized_block_loader.html#abbf8249ca99e3e87b296ddd60a984b76',1,'QuantizedBlockLoader::src'],['../structmlx_1_1steel_1_1_block_loader.html#ad1db14517568ae9eddfb6986ef31c7aa',1,'mlx::steel::BlockLoader::src'],['../structmlx_1_1steel_1_1_block_loader_t.html#a7004a4efaa483cc79b8b79810a17c777',1,'mlx::steel::BlockLoaderT::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a656a46ee27486482b45ff90b3d626255',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a57552e9cfbafad71d47b2f3a8e027bdf',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7bfbcc4a1e3eef7aef5dd8e8c374a95f',1,'mlx::steel::Conv2DWeightBlockLoader::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#afe21e46e08523232830c25eb1b4ade16',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a8b04a69952404a04029dacc424df6e8f',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1ee2922961b5fcb1db577928c4d9d731',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a50f458dbb74d61be2ed24727d8d43614',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src']]], + ['src_5fld_167',['src_ld',['../struct_quantized_block_loader.html#a8050977d473d1a24fae5c833e609839e',1,'QuantizedBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_block_loader.html#aadafc50f7f06af434149d7469df4714d',1,'mlx::steel::BlockLoader::src_ld'],['../structmlx_1_1steel_1_1_block_loader_t.html#aeba87e81185da6b20a092c5d240d3321',1,'mlx::steel::BlockLoaderT::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7464ec687323fa79050702952ed9084f',1,'mlx::steel::Conv2DWeightBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#aa6bedc0cbb447eaf70c03f2e26df2cb2',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6918c1df7712c4e408e2871467ea7987',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src_ld']]], + ['stack_168',['stack',['../group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032',1,'mlx::core::stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga82216209dce901296fc737fe8efa5c94',1,'mlx::core::stack(const std::vector< array > &arrays, StreamOrDevice s={})']]], + ['start_5fcapture_169',['start_capture',['../namespacemlx_1_1core_1_1metal.html#aa47cb5651bf3b65c46ab216b7e504d77',1,'mlx::core::metal']]], + ['start_5fconcurrent_170',['start_concurrent',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::metal::CommandEncoder::start_concurrent()'],['../structmlx_1_1core_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::CommandEncoder::start_concurrent()']]], + ['start_5frow_171',['start_row',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a220e033b689c8d6a6f319dae02b38334',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], + ['state_172',['state',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a379b27ac336ef351aa81142c5626ad76',1,'mlx::core::fast::RMSNormVJP::state()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a963e672c607b5f86080e6cc32a3cd6e5',1,'mlx::core::fast::LayerNormVJP::state()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#aa5a03284c6f5639d684dd34d86050cf9',1,'mlx::core::fast::AffineQuantize::state()'],['../classmlx_1_1core_1_1_add_m_m.html#a24ab73de46d0589780dac5ade43f93b8',1,'mlx::core::AddMM::state()'],['../classmlx_1_1core_1_1_arange.html#ac4a9f48a11c2af03ed57fdf2422cbfad',1,'mlx::core::Arange::state()'],['../classmlx_1_1core_1_1_arg_partition.html#a5033c46f5aae9b14859cc8b0ca4c8e19',1,'mlx::core::ArgPartition::state()'],['../classmlx_1_1core_1_1_arg_reduce.html#acac3b26364260aac7511b4cb7add3604',1,'mlx::core::ArgReduce::state()'],['../classmlx_1_1core_1_1_arg_sort.html#a90548429765f9e7e2332f01b72692fa2',1,'mlx::core::ArgSort::state()'],['../classmlx_1_1core_1_1_as_type.html#a98ea769fc9cd6d76b07817444e7a78ab',1,'mlx::core::AsType::state()'],['../classmlx_1_1core_1_1_as_strided.html#ae730aeff375498ba774d4207c7af8c36',1,'mlx::core::AsStrided::state()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8a67d6f431b4055ab66656201622af4d',1,'mlx::core::BitwiseBinary::state()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a6bbcc34b256840e4df2953563f2b4a07',1,'mlx::core::BlockMaskedMM::state()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a5b2594b7a70dd4873a07e742140a245f',1,'mlx::core::BroadcastAxes::state()'],['../classmlx_1_1core_1_1_broadcast.html#a49fdb421047860733af7dfbbb478da8d',1,'mlx::core::Broadcast::state()'],['../classmlx_1_1core_1_1_concatenate.html#a60cd572a42b346399ee539af2dfbf29e',1,'mlx::core::Concatenate::state()'],['../classmlx_1_1core_1_1_convolution.html#a7f44f0caea20cc2858717afba1e915d8',1,'mlx::core::Convolution::state()'],['../classmlx_1_1core_1_1_equal.html#aa27ff7525f109edc56b731a6df78f6bc',1,'mlx::core::Equal::state()'],['../classmlx_1_1core_1_1_expand_dims.html#a7cacc704c533c00ba072f0a7872631cf',1,'mlx::core::ExpandDims::state()'],['../classmlx_1_1core_1_1_f_f_t.html#a710c6f6e8412da0af0fdbe58fbae320e',1,'mlx::core::FFT::state()'],['../classmlx_1_1core_1_1_flatten.html#af95dd89c47cd2342233dc0b6d36822a3',1,'mlx::core::Flatten::state()'],['../classmlx_1_1core_1_1_gather.html#aee59ff90127ef4c2d7fcbe2955b95b27',1,'mlx::core::Gather::state()'],['../classmlx_1_1core_1_1_gather_axis.html#adff37b05799654b1a589e334d1cd6b46',1,'mlx::core::GatherAxis::state()'],['../classmlx_1_1core_1_1_hadamard.html#afd67d09fde38ab3b6ba873b797f03dae',1,'mlx::core::Hadamard::state()'],['../classmlx_1_1core_1_1_log.html#a86fca2ec3766f5d4a2e6d8ba2983c3aa',1,'mlx::core::Log::state()'],['../classmlx_1_1core_1_1_number_of_elements.html#afbfee716b4896e98bdf502ceab87ac09',1,'mlx::core::NumberOfElements::state()'],['../classmlx_1_1core_1_1_pad.html#a00a7cff2ae640f45b43f62cc25d6346c',1,'mlx::core::Pad::state()'],['../classmlx_1_1core_1_1_partition.html#adde13e40924c016473864119465cad4b',1,'mlx::core::Partition::state()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ad83bfd32fda988c29e5ca277a84c0655',1,'mlx::core::QuantizedMatmul::state()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a60c908bc836f930bb33f60b3e9db43af',1,'mlx::core::GatherQMM::state()'],['../classmlx_1_1core_1_1_random_bits.html#a75a34d7541a1c124710dc4d0ec2dfa60',1,'mlx::core::RandomBits::state()'],['../classmlx_1_1core_1_1_reshape.html#aa8ad5958aac8723dd6ce49820eaba029',1,'mlx::core::Reshape::state()'],['../classmlx_1_1core_1_1_reduce.html#afca1398c042a3b1ca9a9a2e40fe62367',1,'mlx::core::Reduce::state()'],['../classmlx_1_1core_1_1_scan.html#a7249ca4c3316b1b1248df32c71fee0ea',1,'mlx::core::Scan::state()'],['../classmlx_1_1core_1_1_scatter.html#a50a65033dc2a1cc84bf529ba718c9c60',1,'mlx::core::Scatter::state()'],['../classmlx_1_1core_1_1_scatter_axis.html#a1a0125be908a1d80875236c817f34495',1,'mlx::core::ScatterAxis::state()'],['../classmlx_1_1core_1_1_slice.html#a069dafc62bf71e3ebc0bd99d96ec23be',1,'mlx::core::Slice::state()'],['../classmlx_1_1core_1_1_slice_update.html#aac5156a19209274b7de1dff231ef25fd',1,'mlx::core::SliceUpdate::state()'],['../classmlx_1_1core_1_1_dynamic_slice.html#aec9084e603d7562f3a75c5fc32918581',1,'mlx::core::DynamicSlice::state()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a0b0b2a0e4d97305fd6f3c635fcdccd76',1,'mlx::core::DynamicSliceUpdate::state()'],['../classmlx_1_1core_1_1_softmax.html#adf665f7c292e84f56c630016b75427f3',1,'mlx::core::Softmax::state()'],['../classmlx_1_1core_1_1_sort.html#a5ab15d1a89efd8661887c805c35fc617',1,'mlx::core::Sort::state()'],['../classmlx_1_1core_1_1_split.html#a915156cde0448ec26776e329004b1a92',1,'mlx::core::Split::state()'],['../classmlx_1_1core_1_1_sqrt.html#a053853757ad99195e3f2b1cca571e31b',1,'mlx::core::Sqrt::state()'],['../classmlx_1_1core_1_1_squeeze.html#a04f9d2595cb7d4ec988479cd33fe9362',1,'mlx::core::Squeeze::state()'],['../classmlx_1_1core_1_1_unflatten.html#aeba13680064238191811230171365598',1,'mlx::core::Unflatten::state()'],['../classmlx_1_1core_1_1_view.html#a37620f6548630bd2d0dd44e9ab084b93',1,'mlx::core::View::state()'],['../classmlx_1_1core_1_1_transpose.html#a23167291e2bf12e2bb2e51d1db340909',1,'mlx::core::Transpose::state()'],['../classmlx_1_1core_1_1_inverse.html#aa1fce744f4a2d660c65901a7542056f2',1,'mlx::core::Inverse::state()'],['../classmlx_1_1core_1_1_cholesky.html#a64f03d32ed249a3b2a59b6af66d23727',1,'mlx::core::Cholesky::state()'],['../classmlx_1_1core_1_1_eigh.html#aa3b6c33b5679c5528863f3de2ab2f914',1,'mlx::core::Eigh::state()']]], + ['status_173',['Status',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078',1,'mlx::core::array']]], + ['status_174',['status',['../classmlx_1_1core_1_1array.html#a7102659be87e9ef62966696ab9b07dad',1,'mlx::core::array']]], + ['std_175',['std',['../group__ops.html#ga2a466024f8061febc0a64be557644cb0',1,'mlx::core::std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#gafdcb04d77c64405a3990078a77dd984c',1,'mlx::core::std(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8',1,'mlx::core::std(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga62721a206df671ef5797449eea97af9f',1,'mlx::core::std(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], + ['steel_5fattention_2eh_176',['steel_attention.h',['../steel__attention_8h.html',1,'']]], + ['steel_5fconst_177',['STEEL_CONST',['../steel_2defines_8h.html#a90b91c866313ffa46eff6d9cc944ad2b',1,'defines.h']]], + ['steel_5fconv_178',['steel_conv',['../namespacemlx_1_1core_1_1metal.html#a92f1e559b1121d545746f81ff86eaca1',1,'mlx::core::metal']]], + ['steel_5fconv_2eh_179',['steel_conv.h',['../steel__conv_8h.html',1,'']]], + ['steel_5fconv_5fgeneral_180',['steel_conv_general',['../namespacemlx_1_1core_1_1metal.html#a02edb6a90bdf30f4c9f0d6c25b0267b5',1,'mlx::core::metal']]], + ['steel_5fconv_5fgeneral_2eh_181',['steel_conv_general.h',['../steel__conv__general_8h.html',1,'']]], + ['steel_5fgemm_5ffused_182',['steel_gemm_fused',['../namespacemlx_1_1core_1_1metal.html#a17764366deed71c160fb26091400a803',1,'mlx::core::metal']]], + ['steel_5fgemm_5ffused_2eh_183',['steel_gemm_fused.h',['../steel__gemm__fused_8h.html',1,'']]], + ['steel_5fgemm_5fmasked_184',['steel_gemm_masked',['../namespacemlx_1_1core_1_1metal.html#a962272ca73d26c08f76f706a128fd71f',1,'mlx::core::metal']]], + ['steel_5fgemm_5fmasked_2eh_185',['steel_gemm_masked.h',['../steel__gemm__masked_8h.html',1,'']]], + ['steel_5fgemm_5fsplitk_186',['steel_gemm_splitk',['../namespacemlx_1_1core_1_1metal.html#ad0dfd40ba7c09755711ceb731e57a5ac',1,'mlx::core::metal']]], + ['steel_5fgemm_5fsplitk_2eh_187',['steel_gemm_splitk.h',['../steel__gemm__splitk_8h.html',1,'']]], + ['steel_5fmatmul_188',['steel_matmul',['../namespacemlx_1_1core.html#aaff208bbac7021c4265580885874499a',1,'mlx::core']]], + ['steel_5fmatmul_5fregular_189',['steel_matmul_regular',['../namespacemlx_1_1core.html#a2a8a09851097571fb51ac5b608550e44',1,'mlx::core']]], + ['steel_5fpragma_5funroll_190',['STEEL_PRAGMA_UNROLL',['../steel_2defines_8h.html#a5a5c3095b132a7589bc19cd5cb80e2c6',1,'defines.h']]], + ['step_191',['step',['../structmlx_1_1core_1_1_contiguous_iterator.html#aad921dd422adb0a0f555e19a2f42239c',1,'mlx::core::ContiguousIterator']]], + ['stop_192',['stop',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a456ad1c0c9e731833a2f8411c4ed51aa',1,'mlx::core::scheduler::StreamThread']]], + ['stop_5fcapture_193',['stop_capture',['../namespacemlx_1_1core_1_1metal.html#ac90714424e36fb01e04550de69b8314f',1,'mlx::core::metal']]], + ['stop_5fgradient_194',['stop_gradient',['../group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f',1,'mlx::core']]], + ['stopgradient_195',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html',1,'mlx::core::StopGradient'],['../classmlx_1_1core_1_1_stop_gradient.html#ac70d1ab819d04e00f76bc25aeebaf84f',1,'mlx::core::StopGradient::StopGradient()']]], + ['store_196',['store',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#aa8f50ea8961ec5b35c1b81366d64f2cb',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a2aadaa3239cb3af0c2ee8af9b88c8a98',1,'mlx::steel::MMATile::store(threadgroup U *dst) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a752f708e4fe5ef37fdd902dae153179f',1,'mlx::steel::MMATile::store(device U *dst, const int ld) const'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#aa8f50ea8961ec5b35c1b81366d64f2cb',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a2aadaa3239cb3af0c2ee8af9b88c8a98',1,'mlx::steel::MMATile::store(threadgroup U *dst) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a752f708e4fe5ef37fdd902dae153179f',1,'mlx::steel::MMATile::store(device U *dst, const int ld) const'],['../namespacemlx_1_1core_1_1simd.html#afa2236afddfdec312eb7e27b89a5316a',1,'mlx::core::simd::store()']]], + ['store_5fresult_197',['store_result',['../structmlx_1_1steel_1_1_block_m_m_a.html#a0461451ffb5041b6a916ea17ed34288b',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a0461451ffb5041b6a916ea17ed34288b',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const']]], + ['store_5fresult_5fsafe_198',['store_result_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a081ba538d30d1d02498a7f341e6bd611',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a081ba538d30d1d02498a7f341e6bd611',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const']]], + ['store_5fsafe_199',['store_safe',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1f0b00daad8eba2f855bb306e70d2328',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a57703f522c7409dbe2c0a68bb7acc2ba',1,'mlx::steel::MMATile::store_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1f0b00daad8eba2f855bb306e70d2328',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a57703f522c7409dbe2c0a68bb7acc2ba',1,'mlx::steel::MMATile::store_safe()']]], + ['str_200',['str',['../classpocketfft_1_1detail_1_1arr__info.html#abe1f7b92501b4e0e5a38fd26294ac5a4',1,'pocketfft::detail::arr_info::str'],['../struct_m_l_x_conv_params.html#a862191e8ab1bc8a47aa1396b36d46058',1,'MLXConvParams::str']]], + ['stream_201',['Stream',['../structmlx_1_1core_1_1_stream.html',1,'mlx::core::Stream'],['../structmlx_1_1core_1_1_stream.html#a7f0815ff4886da74cbbff5f93d82dd3e',1,'mlx::core::Stream::Stream()']]], + ['stream_202',['stream',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a8462e4acffcd385c6248bd7102e6bcb1',1,'mlx::core::scheduler::StreamThread::stream'],['../classmlx_1_1core_1_1_event.html#a193143bad31b68c699fa27f135b45614',1,'mlx::core::Event::stream()'],['../classmlx_1_1core_1_1_primitive.html#a46e6257397a662528f9f831842ac456a',1,'mlx::core::Primitive::stream()']]], + ['stream_2eh_203',['stream.h',['../stream_8h.html',1,'']]], + ['streamcontext_204',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html',1,'mlx::core::StreamContext'],['../structmlx_1_1core_1_1_stream_context.html#a89d803151e9d7dce29382aa83d5c6ef1',1,'mlx::core::StreamContext::StreamContext()']]], + ['streamordevice_205',['StreamOrDevice',['../namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58',1,'mlx::core']]], + ['streamthread_206',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html',1,'mlx::core::scheduler::StreamThread'],['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#ac528109a11abcb82e6e221c5efa4493c',1,'mlx::core::scheduler::StreamThread::StreamThread()']]], + ['stride_207',['stride',['../classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093',1,'pocketfft::detail::arr_info::stride() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b',1,'pocketfft::detail::arr_info::stride(size_t i) const']]], + ['stride_5fin_208',['stride_in',['../classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c',1,'pocketfft::detail::multi_iter']]], + ['stride_5fout_209',['stride_out',['../classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7',1,'pocketfft::detail::multi_iter']]], + ['stride_5ft_210',['stride_t',['../namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::detail::stride_t'],['../namespacepocketfft.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::stride_t']]], + ['strided_5fdevice_5fidx_211',['strided_device_idx',['../struct_read_writer.html#a4c0b12484aac4fd6759d67c190391989',1,'ReadWriter']]], + ['strided_5freduce_5fgeneral_5fdispatch_212',['strided_reduce_general_dispatch',['../namespacemlx_1_1core.html#aa0332c64ee9965f05026c30a0b778000',1,'mlx::core']]], + ['strided_5fscan_213',['strided_scan',['../scan_8h.html#a7abb6ffb6c3b96b88c2a63cd4cc2f7ae',1,'scan.h']]], + ['strided_5fshared_5fidx_214',['strided_shared_idx',['../struct_read_writer.html#ace40adb02cfb33d89c98353327c251fc',1,'ReadWriter']]], + ['strides_215',['Strides',['../namespacemlx_1_1core.html#a79939016d0972ded7db37130da2a8b5c',1,'mlx::core']]], + ['strides_216',['strides',['../structmlx_1_1core_1_1_reduction_plan.html#a58bc6189e5e7175dae92632a7bcfd53e',1,'mlx::core::ReductionPlan::strides'],['../struct_indices.html#a510b7fe052c5826911dd17d7ccb9e73f',1,'Indices::strides'],['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63954de7da62942ec69afcaaa19d46f2',1,'mlx::core::fast::CustomKernelShapeInfo::strides'],['../classmlx_1_1core_1_1array.html#a28cf1928f5ec2f972a94ff1c0e71187d',1,'mlx::core::array::strides() const'],['../classmlx_1_1core_1_1array.html#ac9bfc251a9937eaefbe7f8c5ffd304d1',1,'mlx::core::array::strides(int dim) const']]], + ['submit_217',['submit',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4',1,'pocketfft::detail::threading::thread_pool']]], + ['subop_218',['SubOp',['../struct_sub_op.html',1,'']]], + ['subtract_219',['Subtract',['../structmlx_1_1core_1_1detail_1_1_subtract.html',1,'mlx::core::detail::Subtract'],['../classmlx_1_1core_1_1_subtract.html',1,'mlx::core::Subtract'],['../struct_subtract.html',1,'Subtract'],['../classmlx_1_1core_1_1_subtract.html#a834854757394f8de7082af65bf86ed9c',1,'mlx::core::Subtract::Subtract()']]], + ['subtract_220',['subtract',['../group__ops.html#ga196c240d3d0fcbb4713802c485e15133',1,'mlx::core']]], + ['sum_221',['Sum',['../struct_sum.html',1,'Sum< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a1fc7c1f09c80650ab0497e2d6781d65f',1,'mlx::core::distributed::AllReduce::Sum'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a8582875544f1d3d396a1a376473ef1dd',1,'mlx::core::Reduce::Sum'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ade23893033e4849f5596e7ce76a5fc36',1,'mlx::core::Scan::Sum'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca14abe2d8818efa71726be4e156813d6f',1,'mlx::core::Scatter::Sum'],['../classmlx_1_1core_1_1_scatter_axis.html#aa292e6cb2a4b32c42ad4f7a258b334f2a702b8cfdaf7fe3e063873595ff0508f2',1,'mlx::core::ScatterAxis::Sum']]], + ['sum_222',['sum',['../namespacemlx_1_1core_1_1simd.html#a53b547b886918dc13d4da88eeb8811d2',1,'mlx::core::simd::sum(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#a96ce7d90b3b8b6dddab36ef5b49fffc2',1,'mlx::core::simd::sum(Simd< T, 1 > x)'],['../namespacemlx_1_1core_1_1simd.html#ab0e7c082fc6bed52d522765ef91d205d',1,'mlx::core::simd::sum(Simd< float16_t, N > x)'],['../namespacemlx_1_1steel.html#ab4a6ddea4beb7c447cf5b69b9d46cc3b',1,'mlx::steel::sum(T x)'],['../namespacemlx_1_1steel.html#acd6e194d37b617d7a5818bc384a97fe4',1,'mlx::steel::sum(T x, Us... us)'],['../group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6',1,'mlx::core::sum(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3627754d7868487bdab1bd83f05d9c81',1,'mlx::core::sum(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4',1,'mlx::core::sum(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d',1,'mlx::core::sum(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['sumop_223',['SumOp',['../struct_sum_op.html',1,'']]], + ['svd_224',['SVD',['../classmlx_1_1core_1_1_s_v_d.html',1,'mlx::core::SVD'],['../classmlx_1_1core_1_1_s_v_d.html#ae89ff583e34fa894cccb8e7a475ee6d1',1,'mlx::core::SVD::SVD()']]], + ['svd_225',['svd',['../namespacemlx_1_1core_1_1linalg.html#a64364b880e99914cf47bf756fa8dbaf0',1,'mlx::core::linalg']]], + ['swapaxes_226',['swapaxes',['../group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb',1,'mlx::core']]], + ['swizzle_227',['swizzle',['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle::swizzle(uint3 tid, const int swizzle_log)'],['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle::swizzle(uint3 tid, const int swizzle_log)']]], + ['swizzle_5flog_228',['swizzle_log',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#ad0713159d4f710cd9a066596593d8840',1,'mlx::steel::ImplicitGemmConv2DParams::swizzle_log'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#af9ff2c06dd8994126634531440325be7',1,'mlx::steel::GEMMParams::swizzle_log']]], + ['synchronize_229',['synchronize',['../namespacemlx_1_1core.html#a14287949d82ffefad0306cef5eb5f9e4',1,'mlx::core::synchronize()'],['../namespacemlx_1_1core.html#a6648a71937b055e5ff513d98056c2fb5',1,'mlx::core::synchronize(Stream)']]] ]; diff --git a/docs/build/html/search/all_14.js b/docs/build/html/search/all_14.js index 909767f2f..77cb2423b 100644 --- a/docs/build/html/search/all_14.js +++ b/docs/build/html/search/all_14.js @@ -7,9 +7,9 @@ var searchData= ['take_4',['take',['../group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7',1,'mlx::core::take(const array &a, const array &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#ga731af77b7de547a73336f91c829c40b4',1,'mlx::core::take(const array &a, int index, int axis, StreamOrDevice s={})'],['../group__ops.html#ga45d0f423a5e030440ef753f36c5aabf1',1,'mlx::core::take(const array &a, const array &indices, StreamOrDevice s={})'],['../group__ops.html#ga7b3c9b4c8ee02dc23cfd7dcd855dbf20',1,'mlx::core::take(const array &a, int index, StreamOrDevice s={})']]], ['take_5falong_5faxis_5',['take_along_axis',['../group__ops.html#gae0a81d4f983e296a87302e36d65bfc76',1,'mlx::core']]], ['tan_6',['Tan',['../structmlx_1_1core_1_1detail_1_1_tan.html',1,'mlx::core::detail::Tan'],['../classmlx_1_1core_1_1_tan.html',1,'mlx::core::Tan'],['../struct_tan.html',1,'Tan'],['../classmlx_1_1core_1_1_tan.html#a8dcc9ff660210ccf05134dd95f47de08',1,'mlx::core::Tan::Tan()']]], - ['tan_7',['tan',['../namespacemetal.html#a862215a8ddacb086296ba02567c9b158',1,'metal::tan()'],['../namespacemetal_1_1fast.html#ae34754afa152a6170ac2ae3294174506',1,'metal::fast::tan()'],['../namespacemetal_1_1precise.html#a8261ed22f03122ef15b89512358acb1f',1,'metal::precise::tan()'],['../group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5',1,'mlx::core::tan()']]], + ['tan_7',['tan',['../namespacemlx_1_1core_1_1simd.html#a445ddc4ed928656df64d889942588cfd',1,'mlx::core::simd::tan(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a829842f854aecfae93b7d42f83aec9a7',1,'mlx::core::simd::tan(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a5abc381a85fe8b0e9cb472f874704652',1,'mlx::core::simd::tan(Simd< T, 1 > in)'],['../namespacemetal.html#a862215a8ddacb086296ba02567c9b158',1,'metal::tan()'],['../namespacemetal_1_1fast.html#ae34754afa152a6170ac2ae3294174506',1,'metal::fast::tan()'],['../namespacemetal_1_1precise.html#a8261ed22f03122ef15b89512358acb1f',1,'metal::precise::tan()'],['../group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5',1,'mlx::core::tan()']]], ['tanh_8',['Tanh',['../structmlx_1_1core_1_1detail_1_1_tanh.html',1,'mlx::core::detail::Tanh'],['../classmlx_1_1core_1_1_tanh.html',1,'mlx::core::Tanh'],['../struct_tanh.html',1,'Tanh'],['../classmlx_1_1core_1_1_tanh.html#ae551297bf573e1802fb831440276dee4',1,'mlx::core::Tanh::Tanh()']]], - ['tanh_9',['tanh',['../namespacemetal.html#aa97fc50bd6addfc6de0aae8570fe963d',1,'metal::tanh()'],['../namespacemetal_1_1fast.html#a13e6e6ae087b7c558e9a94ddbc864d43',1,'metal::fast::tanh()'],['../namespacemetal_1_1precise.html#a741c27a10cc968dd1e63473d9fcd8f99',1,'metal::precise::tanh()'],['../group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d',1,'mlx::core::tanh()']]], + ['tanh_9',['tanh',['../namespacemlx_1_1core_1_1simd.html#aa244fbe7456b653aa50a473108fd6a2b',1,'mlx::core::simd::tanh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ad78f543dc5da87a14ca113a1dd9852fd',1,'mlx::core::simd::tanh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ab80a7db8d99e3f4032e761c60216027d',1,'mlx::core::simd::tanh(Simd< T, 1 > in)'],['../namespacemetal.html#aa97fc50bd6addfc6de0aae8570fe963d',1,'metal::tanh()'],['../namespacemetal_1_1fast.html#a13e6e6ae087b7c558e9a94ddbc864d43',1,'metal::fast::tanh()'],['../namespacemetal_1_1precise.html#a741c27a10cc968dd1e63473d9fcd8f99',1,'metal::precise::tanh()'],['../group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d',1,'mlx::core::tanh()']]], ['tanpi_10',['tanpi',['../namespacemetal.html#ae2046d163a525fc1822a9ec8a0aeaeb3',1,'metal::tanpi()'],['../namespacemetal_1_1fast.html#a39b2952d4adf1400016c63243798aaf8',1,'metal::fast::tanpi()'],['../namespacemetal_1_1precise.html#a8fae8c20deff43a8e855bba6f3ba20a5',1,'metal::precise::tanpi()']]], ['tcols_11',['TCOLS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a0b5303f3258e0a21862dead8e3f5401e',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a5adbd51e9adb6f7853724d83de4ff755',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a80cb90674f839d5d4ecfde384fa0a7a2',1,'mlx::steel::Conv2DWeightBlockLoader::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ad2508cd5cdb51b2f611057e743b8fc6f',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acd54132d0928d0f6fb15b2f367e5d5e8',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#ae25c676b7318d78462ee89bcd80dc805',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aff021a6fae860b4ac01fb593b2720457',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::TCOLS']]], ['tell_12',['tell',['../classmlx_1_1core_1_1io_1_1_reader.html#a27697ccc1ce45da0233db3bd4f298aed',1,'mlx::core::io::Reader::tell()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a11ad80749894993232fbb5c70fd7b282',1,'mlx::core::io::Writer::tell()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a2e92131428f0ffa98fff781b8c35d9e5',1,'mlx::core::io::ParallelFileReader::tell()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#aa883a722789c962164fd0ddcc5f6ffc5',1,'mlx::core::io::FileWriter::tell()']]], @@ -18,77 +18,81 @@ var searchData= ['ten_15',['ten',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421a394d85b39676763bdf35b8d54b9e43a1',1,'mlx::core::Log']]], ['tensordot_16',['tensordot',['../group__ops.html#gaf5c9735f4690327e1500e04e728fae70',1,'mlx::core::tensordot(const array &a, const array &b, const int axis=2, StreamOrDevice s={})'],['../group__ops.html#gad7fe00b566f89d607639c1a497cabbc6',1,'mlx::core::tensordot(const array &a, const array &b, const std::vector< int > &axes_a, const std::vector< int > &axes_b, StreamOrDevice s={})']]], ['ternary_17',['ternary',['../namespacemlx_1_1core_1_1metal.html#a2d1c92ba6897c0a7a428fed63279b61f',1,'mlx::core::metal']]], - ['ternary_2eh_18',['ternary.h',['../common_2ternary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2ternary_8h.html',1,'(Global Namespace)'],['../metal_2ternary_8h.html',1,'(Global Namespace)']]], + ['ternary_2eh_18',['ternary.h',['../common_2ternary_8h.html',1,'(Global Namespace)'],['../cpu_2ternary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2ternary_8h.html',1,'(Global Namespace)'],['../metal_2ternary_8h.html',1,'(Global Namespace)']]], ['ternary_5fg_19',['ternary_g',['../metal_2kernels_2ternary_8h.html#a34189dad8224d54d74e070ab05b97e9f',1,'ternary.h']]], ['ternary_5fg_5fnd1_20',['ternary_g_nd1',['../metal_2kernels_2ternary_8h.html#af8400389f3bb11498fb1c4057e638a27',1,'ternary.h']]], ['ternary_5fg_5fnd2_21',['ternary_g_nd2',['../metal_2kernels_2ternary_8h.html#a0971bb39ec881e97af7ab9584e1ce22a',1,'ternary.h']]], ['ternary_5fg_5fnd3_22',['ternary_g_nd3',['../metal_2kernels_2ternary_8h.html#ad7968ba7b638b85ff67a65eef768f59c',1,'ternary.h']]], - ['ternary_5fop_5fgpu_23',['ternary_op_gpu',['../namespacemlx_1_1core.html#aa63e62b6d3906e4cac871d498515a1cd',1,'mlx::core']]], - ['ternary_5fop_5fgpu_5finplace_24',['ternary_op_gpu_inplace',['../namespacemlx_1_1core.html#a37645c0adccb3eb46844115def1a68d7',1,'mlx::core']]], - ['ternary_5fops_25',['ternary_ops',['../namespacemlx_1_1core_1_1metal.html#a11b593b07e9a33e5f78fe4695fb99ec9',1,'mlx::core::metal']]], - ['ternary_5fops_2eh_26',['ternary_ops.h',['../ternary__ops_8h.html',1,'']]], - ['ternary_5fv_27',['ternary_v',['../metal_2kernels_2ternary_8h.html#a83f93644d21ee774e06e8190d0725ccb',1,'ternary.h']]], - ['ternary_5fv2_28',['ternary_v2',['../metal_2kernels_2ternary_8h.html#a3e610f3b01966bdbf23fdfebe5d2c508',1,'ternary.h']]], - ['tgp_5fmem_5fsize_29',['tgp_mem_size',['../struct_g_e_m_v_kernel.html#a9ef4d0e62094d7033069f5dda5efb236',1,'GEMVKernel::tgp_mem_size'],['../struct_g_e_m_v_t_kernel.html#a48a09a21d7b822f380d040c752b785d7',1,'GEMVTKernel::tgp_mem_size'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a1ec583584e69dcbbb72106390a4fc5da',1,'mlx::steel::GEMMKernel::tgp_mem_size']]], - ['tgp_5fmem_5fsize_5fa_30',['tgp_mem_size_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ac00b149d76a903c2f91b0f477dc5037f',1,'mlx::steel::GEMMKernel']]], - ['tgp_5fmem_5fsize_5fb_31',['tgp_mem_size_b',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a105af1069668028c6f1bc6d6dd162298',1,'mlx::steel::GEMMKernel']]], - ['tgp_5fpadding_5fa_32',['tgp_padding_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ad547704ccbff6c2076abeffa6628c5a0',1,'mlx::steel::GEMMKernel']]], - ['tgp_5fpadding_5fb_33',['tgp_padding_b',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ad1b03941e869017558423c08b08bc094',1,'mlx::steel::GEMMKernel']]], - ['tgp_5fsize_34',['tgp_size',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a9058ddb73e30e83fb9c548ba22817d64',1,'mlx::steel::GEMMKernel']]], - ['thread_35',['thread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a449de02bf2ac80d8fe2f208fa7eac359',1,'mlx::core::scheduler::StreamThread']]], - ['thread_5fcount_36',['thread_count',['../structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87',1,'pocketfft::detail::util']]], - ['thread_5ffn_37',['thread_fn',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a06a62c21c1174e4eb4d242e50aad7adf',1,'mlx::core::scheduler::StreamThread']]], - ['thread_5fid_38',['thread_id',['../namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b',1,'pocketfft::detail::threading']]], - ['thread_5fidx_39',['thread_idx',['../struct_quantized_block_loader.html#a50821537ea747bc03295a09bb0eef475',1,'QuantizedBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_block_loader.html#a064e2cc77e0b1cf0f8027929e031775b',1,'mlx::steel::BlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_block_loader_t.html#af2838998a02866f22b525f9b6ae004da',1,'mlx::steel::BlockLoaderT::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a70da26a715135d973f88371a70255be9',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ac18de37cde1459595bfe18b0d5ef146d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ab1cb2ade639787243e0325dcd3dc0a11',1,'mlx::steel::Conv2DWeightBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9642399b8066e29123524f36ebc7b482',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acacdac168004c87fee27c8554ac905a7',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a401f0c7cf1588552556603c7ffba2316',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08a517bc50caf41155b98be0690bfe44',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::thread_idx']]], - ['thread_5fmap_40',['thread_map',['../namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e',1,'pocketfft::detail::threading']]], - ['thread_5fpool_41',['thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html',1,'pocketfft::detail::threading::thread_pool'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560',1,'pocketfft::detail::threading::thread_pool::thread_pool(size_t nthreads)'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0',1,'pocketfft::detail::threading::thread_pool::thread_pool()'],['../namespacemlx_1_1core_1_1io.html#a05f27b765443a178a972abae772e863d',1,'mlx::core::io::thread_pool()']]], - ['thread_5freduce_42',['thread_reduce',['../reduce__row_8h.html#afd80a25fa84e6cc884dcc8698859ade1',1,'reduce_row.h']]], - ['thread_5fsort_5ft_43',['thread_sort_t',['../struct_block_merge_sort.html#ad2474d16721f4ceb954125728a0e2ea2',1,'BlockMergeSort']]], - ['thread_5fswap_44',['thread_swap',['../sort_8h.html#a6e8c2da4975a8001fd5ddf211a3058b7',1,'sort.h']]], - ['threadgroup_5freduce_45',['threadgroup_reduce',['../reduce__row_8h.html#aa146bb611069fd2892f03714fd1cc3cf',1,'reduce_row.h']]], - ['threadpool_46',['ThreadPool',['../class_thread_pool.html',1,'ThreadPool'],['../class_thread_pool.html#ac291710e33dbbed96ee20711080d506d',1,'ThreadPool::ThreadPool()']]], - ['threadpool_2eh_47',['threadpool.h',['../threadpool_8h.html',1,'']]], - ['threads_5fper_5ftg_48',['threads_per_tg',['../struct_read_writer.html#a64c58e358da22358df3075448ea23893',1,'ReadWriter']]], - ['threadsm_49',['threadsM',['../struct_g_e_m_v_kernel.html#a1dd943fcbf5e7be435fc36bed589a641',1,'GEMVKernel::threadsM'],['../struct_g_e_m_v_t_kernel.html#a4a53e73a581aa8881b1f86ce653519e6',1,'GEMVTKernel::threadsM']]], - ['threadsn_50',['threadsN',['../struct_g_e_m_v_kernel.html#a47bfab7d21dd18760d3e0937ad36b19d',1,'GEMVKernel::threadsN'],['../struct_g_e_m_v_t_kernel.html#ade6f15a9744616de9dd71498ad7e758d',1,'GEMVTKernel::threadsN']]], - ['threadsort_51',['ThreadSort',['../struct_thread_sort.html',1,'']]], - ['threefry_2eh_52',['threefry.h',['../threefry_8h.html',1,'']]], - ['threefry2x32_5fhash_53',['threefry2x32_hash',['../namespacemlx_1_1core_1_1random.html#ac7e92c89a2bac1b0bed922a3d4c3c66b',1,'mlx::core::random']]], - ['tile_54',['tile',['../group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361',1,'mlx::core']]], - ['tile_5fmatmad_55',['tile_matmad',['../namespacemlx_1_1steel.html#ad583e6038efc119542410f43b603d4ad',1,'mlx::steel']]], - ['tile_5fstride_56',['tile_stride',['../struct_quantized_block_loader.html#ac3f651c1a645291d1037a2cc8ded2320',1,'QuantizedBlockLoader::tile_stride'],['../structmlx_1_1steel_1_1_block_loader.html#ab87876699d55473620c7ea99f9da911d',1,'mlx::steel::BlockLoader::tile_stride'],['../structmlx_1_1steel_1_1_block_loader_t.html#a3abb86e68adb7e4d87cb808d6c25e35f',1,'mlx::steel::BlockLoaderT::tile_stride']]], - ['tile_5fstride_5fa_57',['tile_stride_a',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8fddaa78913cdc8eea5e1cf7d2776330',1,'mlx::steel::BlockMMA']]], - ['tile_5fstride_5fb_58',['tile_stride_b',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae3f35453b3afbaac9df64ad5966b34a4',1,'mlx::steel::BlockMMA']]], - ['tiles_5fm_59',['tiles_m',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a4c5e33edf70be99cf93ac5723c12eb24',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#ad23a5a7f74cd5859741a36e4bc7823ca',1,'mlx::steel::GEMMParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a0970989624e17088d5326c2e198cb95b',1,'mlx::steel::GEMMSpiltKParams::tiles_m']]], - ['tiles_5fn_60',['tiles_n',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a76f9f381e7187a993d65128b9b681b2d',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0e6b8b629232f1b43fbce9a395174bed',1,'mlx::steel::GEMMParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a5b46dfb9cee3606efa05d217349a20a6',1,'mlx::steel::GEMMSpiltKParams::tiles_n']]], - ['tm_61',['TM',['../structmlx_1_1steel_1_1_block_m_m_a.html#aba5f749fdf32d8bd9d9e29f2a9ae4591',1,'mlx::steel::BlockMMA']]], - ['tm_5fstride_62',['TM_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a5b0029866f493363942133b55bff7307',1,'mlx::steel::BlockMMA']]], - ['tn_63',['TN',['../structmlx_1_1steel_1_1_block_m_m_a.html#a706ae779c1f8d2eb18f19c248567d424',1,'mlx::steel::BlockMMA']]], - ['tn_5fstride_64',['TN_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8b3690b383afd26563efb38f9c375e50',1,'mlx::steel::BlockMMA']]], - ['to_5fbnns_5fdtype_65',['to_bnns_dtype',['../namespacemlx_1_1core.html#a7a4193f37b1de9c33c31d1da09c77edb',1,'mlx::core']]], - ['to_5fstream_66',['to_stream',['../namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9',1,'mlx::core']]], - ['topk_67',['topk',['../group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a',1,'mlx::core::topk(const array &a, int k, StreamOrDevice s={})'],['../group__ops.html#ga35b8436c79ff953f6c809598b646f498',1,'mlx::core::topk(const array &a, int k, int axis, StreamOrDevice s={})']]], - ['trace_68',['trace',['../group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, StreamOrDevice s={})'],['../group__ops.html#gaf25c00108feaafaa6350a4434cb0062e',1,'mlx::core::trace(const array &a, StreamOrDevice s={})']]], - ['transformadd_69',['TransformAdd',['../structmlx_1_1steel_1_1_transform_add.html',1,'mlx::steel::TransformAdd< OutT, InT >'],['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd(const float, const float)'],['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd(const float, const float)']]], - ['transformaxpby_70',['TransformAxpby',['../structmlx_1_1steel_1_1_transform_axpby.html',1,'mlx::steel::TransformAxpby< OutT, InT >'],['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby(const float alpha_, const float beta_)'],['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby(const float alpha_, const float beta_)']]], - ['transformnone_71',['TransformNone',['../structmlx_1_1steel_1_1_transform_none.html',1,'mlx::steel']]], - ['transforms_2eh_72',['transforms.h',['../backend_2metal_2kernels_2steel_2attn_2transforms_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2gemm_2transforms_8h.html',1,'(Global Namespace)'],['../transforms_8h.html',1,'(Global Namespace)']]], - ['transforms_5fimpl_2eh_73',['transforms_impl.h',['../transforms__impl_8h.html',1,'']]], - ['transformscale_74',['TransformScale',['../struct_transform_scale.html',1,'TransformScale< T >'],['../struct_transform_scale.html#ae109cf7c963ba13df96977e7563f7b70',1,'TransformScale::TransformScale()']]], - ['transpose_75',['Transpose',['../classmlx_1_1core_1_1_transpose.html',1,'mlx::core::Transpose'],['../classmlx_1_1core_1_1_transpose.html#a1a9ba023584c61c7ac93d6dce536760a',1,'mlx::core::Transpose::Transpose()']]], - ['transpose_76',['transpose',['../group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b',1,'mlx::core::transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5',1,'mlx::core::transpose(const array &a, std::initializer_list< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a',1,'mlx::core::transpose(const array &a, StreamOrDevice s={})']]], - ['tri_77',['tri',['../group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78',1,'mlx::core::tri(int n, int m, int k, Dtype type, StreamOrDevice s={})'],['../group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5',1,'mlx::core::tri(int n, Dtype type, StreamOrDevice s={})']]], - ['tri_5finv_78',['tri_inv',['../namespacemlx_1_1core_1_1linalg.html#aba1994571326326717b5b5e38c2e0661',1,'mlx::core::linalg']]], - ['tril_79',['tril',['../group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2',1,'mlx::core']]], - ['triu_80',['triu',['../group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412',1,'mlx::core']]], - ['trows_81',['TROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a2528ff5ed472e4ed35415ada42276b07',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a3957fb263fe040fe70683fd1d7b06487',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a94f0ce5bb7d87bc1fb6a7c2ba2b892d4',1,'mlx::steel::Conv2DWeightBlockLoader::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a789683f9ac9d9309d07c05f3bdedd2fd',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a5cefb1285ed13ad3490198e9303453de',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a2aff22af70f685f858adea73f5575cf7',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acbc28f364381166faaeec2783dc88e10',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::TROWS']]], - ['true_5ftype_82',['true_type',['../namespacemlx_1_1steel.html#a594a6ccb75b38b5ae4ddd0d9ad047b3a',1,'mlx::steel']]], - ['trunc_83',['trunc',['../namespacemetal.html#a93cb75a11a362bfc8310ea19c554c887',1,'metal::trunc()'],['../namespacemetal_1_1fast.html#aa62e1075e86c626d97038f16e9433415',1,'metal::fast::trunc()'],['../namespacemetal_1_1precise.html#a334183e7a2dd49b983d072d1e8ee2b27',1,'metal::precise::trunc()']]], - ['truncated_5fnormal_84',['truncated_normal',['../namespacemlx_1_1core_1_1random.html#aece7dc5a29e0488d8b9648f340dbff72',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a39663eda0fd7b274d01499a7b1c9035f',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['try_5fpop_85',['try_pop',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43',1,'pocketfft::detail::threading::concurrent_queue']]], - ['two_86',['two',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421a41877eab6fa3db7d7ed2cda9eba14251',1,'mlx::core::Log']]], - ['type_87',['type',['../structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e',1,'pocketfft::detail::add_vec::type'],['../structmlx_1_1steel_1_1integral__constant.html#a6492c15b37d160d3a33e1cbe770aa3f1',1,'mlx::steel::integral_constant::type'],['../structmetal_1_1make__void.html#a53b29d640a06df62047fffdec0571392',1,'metal::make_void::type'],['../struct_conditional_type.html#a00bac71c43763817c4422bf0363dc92b',1,'ConditionalType::type'],['../structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea',1,'pocketfft::detail::ExecDcst::type'],['../structmlx_1_1core_1_1_reduction_plan.html#a24e407f13d4d02156380ecc1a6748a76',1,'mlx::core::ReductionPlan::type'],['../structmlx_1_1core_1_1_device.html#a763264ec90f7f23c5dced36c3f0db2e5',1,'mlx::core::Device::type']]], - ['type_5fto_5fname_88',['type_to_name',['../namespacemlx_1_1core.html#aef60e3a8d9c987c9c338b193673d2164',1,'mlx::core::type_to_name(const Dtype &t)'],['../namespacemlx_1_1core.html#af1fdfdaa5644394362e6baba30701bae',1,'mlx::core::type_to_name(const array &a)']]], - ['type_5ftraits_2eh_89',['type_traits.h',['../type__traits_8h.html',1,'']]], - ['typetodtype_90',['TypeToDtype',['../structmlx_1_1core_1_1_type_to_dtype.html',1,'mlx::core']]] + ['ternary_5fop_23',['ternary_op',['../namespacemlx_1_1core.html#a9dcc3018702ee31c21c8652bdc2182b1',1,'mlx::core']]], + ['ternary_5fop_5fdims_24',['ternary_op_dims',['../namespacemlx_1_1core.html#a8096c7a688ac3f09cca69a3a85f7f157',1,'mlx::core']]], + ['ternary_5fop_5fdispatch_5fdims_25',['ternary_op_dispatch_dims',['../namespacemlx_1_1core.html#ac1c085e305954247d042f5d8803cd85b',1,'mlx::core']]], + ['ternary_5fop_5fgpu_26',['ternary_op_gpu',['../namespacemlx_1_1core.html#aa63e62b6d3906e4cac871d498515a1cd',1,'mlx::core']]], + ['ternary_5fop_5fgpu_5finplace_27',['ternary_op_gpu_inplace',['../namespacemlx_1_1core.html#a37645c0adccb3eb46844115def1a68d7',1,'mlx::core']]], + ['ternary_5fops_28',['ternary_ops',['../namespacemlx_1_1core_1_1metal.html#a11b593b07e9a33e5f78fe4695fb99ec9',1,'mlx::core::metal']]], + ['ternary_5fops_2eh_29',['ternary_ops.h',['../ternary__ops_8h.html',1,'']]], + ['ternary_5fv_30',['ternary_v',['../metal_2kernels_2ternary_8h.html#a83f93644d21ee774e06e8190d0725ccb',1,'ternary.h']]], + ['ternary_5fv2_31',['ternary_v2',['../metal_2kernels_2ternary_8h.html#a3e610f3b01966bdbf23fdfebe5d2c508',1,'ternary.h']]], + ['ternaryoptype_32',['TernaryOpType',['../namespacemlx_1_1core.html#ac2b8997537c7f25dd2b244d4c0a865a1',1,'mlx::core']]], + ['tgp_5fmem_5fsize_33',['tgp_mem_size',['../struct_g_e_m_v_kernel.html#a9ef4d0e62094d7033069f5dda5efb236',1,'GEMVKernel::tgp_mem_size'],['../struct_g_e_m_v_t_kernel.html#a48a09a21d7b822f380d040c752b785d7',1,'GEMVTKernel::tgp_mem_size'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a1ec583584e69dcbbb72106390a4fc5da',1,'mlx::steel::GEMMKernel::tgp_mem_size']]], + ['tgp_5fmem_5fsize_5fa_34',['tgp_mem_size_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ac00b149d76a903c2f91b0f477dc5037f',1,'mlx::steel::GEMMKernel']]], + ['tgp_5fmem_5fsize_5fb_35',['tgp_mem_size_b',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a105af1069668028c6f1bc6d6dd162298',1,'mlx::steel::GEMMKernel']]], + ['tgp_5fpadding_5fa_36',['tgp_padding_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ad547704ccbff6c2076abeffa6628c5a0',1,'mlx::steel::GEMMKernel']]], + ['tgp_5fpadding_5fb_37',['tgp_padding_b',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ad1b03941e869017558423c08b08bc094',1,'mlx::steel::GEMMKernel']]], + ['tgp_5fsize_38',['tgp_size',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a9058ddb73e30e83fb9c548ba22817d64',1,'mlx::steel::GEMMKernel']]], + ['thread_39',['thread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a449de02bf2ac80d8fe2f208fa7eac359',1,'mlx::core::scheduler::StreamThread']]], + ['thread_5fcount_40',['thread_count',['../structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87',1,'pocketfft::detail::util']]], + ['thread_5ffn_41',['thread_fn',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a06a62c21c1174e4eb4d242e50aad7adf',1,'mlx::core::scheduler::StreamThread']]], + ['thread_5fid_42',['thread_id',['../namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b',1,'pocketfft::detail::threading']]], + ['thread_5fidx_43',['thread_idx',['../struct_quantized_block_loader.html#a50821537ea747bc03295a09bb0eef475',1,'QuantizedBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_block_loader.html#a064e2cc77e0b1cf0f8027929e031775b',1,'mlx::steel::BlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_block_loader_t.html#af2838998a02866f22b525f9b6ae004da',1,'mlx::steel::BlockLoaderT::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a70da26a715135d973f88371a70255be9',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ac18de37cde1459595bfe18b0d5ef146d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ab1cb2ade639787243e0325dcd3dc0a11',1,'mlx::steel::Conv2DWeightBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9642399b8066e29123524f36ebc7b482',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acacdac168004c87fee27c8554ac905a7',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a401f0c7cf1588552556603c7ffba2316',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08a517bc50caf41155b98be0690bfe44',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::thread_idx']]], + ['thread_5fmap_44',['thread_map',['../namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e',1,'pocketfft::detail::threading']]], + ['thread_5fpool_45',['thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html',1,'pocketfft::detail::threading::thread_pool'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560',1,'pocketfft::detail::threading::thread_pool::thread_pool(size_t nthreads)'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0',1,'pocketfft::detail::threading::thread_pool::thread_pool()'],['../namespacemlx_1_1core_1_1io.html#a05f27b765443a178a972abae772e863d',1,'mlx::core::io::thread_pool()']]], + ['thread_5freduce_46',['thread_reduce',['../reduce__row_8h.html#afd80a25fa84e6cc884dcc8698859ade1',1,'reduce_row.h']]], + ['thread_5fsort_5ft_47',['thread_sort_t',['../struct_block_merge_sort.html#afdec1074b7de04cbe083e3270b4ece88',1,'BlockMergeSort']]], + ['thread_5fswap_48',['thread_swap',['../sort_8h.html#a6e8c2da4975a8001fd5ddf211a3058b7',1,'sort.h']]], + ['threadgroup_5freduce_49',['threadgroup_reduce',['../reduce__row_8h.html#aa146bb611069fd2892f03714fd1cc3cf',1,'reduce_row.h']]], + ['threadpool_50',['ThreadPool',['../class_thread_pool.html',1,'ThreadPool'],['../class_thread_pool.html#ac291710e33dbbed96ee20711080d506d',1,'ThreadPool::ThreadPool()']]], + ['threadpool_2eh_51',['threadpool.h',['../threadpool_8h.html',1,'']]], + ['threads_5fper_5ftg_52',['threads_per_tg',['../struct_read_writer.html#a64c58e358da22358df3075448ea23893',1,'ReadWriter']]], + ['threadsm_53',['threadsM',['../struct_g_e_m_v_kernel.html#a1dd943fcbf5e7be435fc36bed589a641',1,'GEMVKernel::threadsM'],['../struct_g_e_m_v_t_kernel.html#a4a53e73a581aa8881b1f86ce653519e6',1,'GEMVTKernel::threadsM']]], + ['threadsn_54',['threadsN',['../struct_g_e_m_v_kernel.html#a47bfab7d21dd18760d3e0937ad36b19d',1,'GEMVKernel::threadsN'],['../struct_g_e_m_v_t_kernel.html#ade6f15a9744616de9dd71498ad7e758d',1,'GEMVTKernel::threadsN']]], + ['threadsort_55',['ThreadSort',['../struct_thread_sort.html',1,'']]], + ['threefry_2eh_56',['threefry.h',['../threefry_8h.html',1,'']]], + ['threefry2x32_5fhash_57',['threefry2x32_hash',['../namespacemlx_1_1core_1_1random.html#ac7e92c89a2bac1b0bed922a3d4c3c66b',1,'mlx::core::random']]], + ['tile_58',['tile',['../group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361',1,'mlx::core']]], + ['tile_5fmatmad_59',['tile_matmad',['../namespacemlx_1_1steel.html#ad583e6038efc119542410f43b603d4ad',1,'mlx::steel']]], + ['tile_5fstride_60',['tile_stride',['../struct_quantized_block_loader.html#ac3f651c1a645291d1037a2cc8ded2320',1,'QuantizedBlockLoader::tile_stride'],['../structmlx_1_1steel_1_1_block_loader.html#ab87876699d55473620c7ea99f9da911d',1,'mlx::steel::BlockLoader::tile_stride'],['../structmlx_1_1steel_1_1_block_loader_t.html#a3abb86e68adb7e4d87cb808d6c25e35f',1,'mlx::steel::BlockLoaderT::tile_stride']]], + ['tile_5fstride_5fa_61',['tile_stride_a',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8fddaa78913cdc8eea5e1cf7d2776330',1,'mlx::steel::BlockMMA']]], + ['tile_5fstride_5fb_62',['tile_stride_b',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae3f35453b3afbaac9df64ad5966b34a4',1,'mlx::steel::BlockMMA']]], + ['tiles_5fm_63',['tiles_m',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a4c5e33edf70be99cf93ac5723c12eb24',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#ad23a5a7f74cd5859741a36e4bc7823ca',1,'mlx::steel::GEMMParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a0970989624e17088d5326c2e198cb95b',1,'mlx::steel::GEMMSpiltKParams::tiles_m']]], + ['tiles_5fn_64',['tiles_n',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a76f9f381e7187a993d65128b9b681b2d',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0e6b8b629232f1b43fbce9a395174bed',1,'mlx::steel::GEMMParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a5b46dfb9cee3606efa05d217349a20a6',1,'mlx::steel::GEMMSpiltKParams::tiles_n']]], + ['tm_65',['TM',['../structmlx_1_1steel_1_1_block_m_m_a.html#aba5f749fdf32d8bd9d9e29f2a9ae4591',1,'mlx::steel::BlockMMA']]], + ['tm_5fstride_66',['TM_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a5b0029866f493363942133b55bff7307',1,'mlx::steel::BlockMMA']]], + ['tn_67',['TN',['../structmlx_1_1steel_1_1_block_m_m_a.html#a706ae779c1f8d2eb18f19c248567d424',1,'mlx::steel::BlockMMA']]], + ['tn_5fstride_68',['TN_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8b3690b383afd26563efb38f9c375e50',1,'mlx::steel::BlockMMA']]], + ['to_5fstream_69',['to_stream',['../namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9',1,'mlx::core']]], + ['topk_70',['topk',['../group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a',1,'mlx::core::topk(const array &a, int k, StreamOrDevice s={})'],['../group__ops.html#ga35b8436c79ff953f6c809598b646f498',1,'mlx::core::topk(const array &a, int k, int axis, StreamOrDevice s={})']]], + ['trace_71',['trace',['../group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, StreamOrDevice s={})'],['../group__ops.html#gaf25c00108feaafaa6350a4434cb0062e',1,'mlx::core::trace(const array &a, StreamOrDevice s={})']]], + ['transformadd_72',['TransformAdd',['../structmlx_1_1steel_1_1_transform_add.html',1,'mlx::steel::TransformAdd< OutT, InT >'],['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd(const float, const float)'],['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd(const float, const float)']]], + ['transformaxpby_73',['TransformAxpby',['../structmlx_1_1steel_1_1_transform_axpby.html',1,'mlx::steel::TransformAxpby< OutT, InT >'],['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby(const float alpha_, const float beta_)'],['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby(const float alpha_, const float beta_)']]], + ['transformnone_74',['TransformNone',['../structmlx_1_1steel_1_1_transform_none.html',1,'mlx::steel']]], + ['transforms_2eh_75',['transforms.h',['../backend_2metal_2kernels_2steel_2attn_2transforms_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2gemm_2transforms_8h.html',1,'(Global Namespace)'],['../transforms_8h.html',1,'(Global Namespace)']]], + ['transforms_5fimpl_2eh_76',['transforms_impl.h',['../transforms__impl_8h.html',1,'']]], + ['transformscale_77',['TransformScale',['../struct_transform_scale.html',1,'TransformScale< T >'],['../struct_transform_scale.html#ae109cf7c963ba13df96977e7563f7b70',1,'TransformScale::TransformScale()']]], + ['transpose_78',['Transpose',['../classmlx_1_1core_1_1_transpose.html',1,'mlx::core::Transpose'],['../classmlx_1_1core_1_1_transpose.html#a1a9ba023584c61c7ac93d6dce536760a',1,'mlx::core::Transpose::Transpose()']]], + ['transpose_79',['transpose',['../group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b',1,'mlx::core::transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5',1,'mlx::core::transpose(const array &a, std::initializer_list< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a',1,'mlx::core::transpose(const array &a, StreamOrDevice s={})']]], + ['tri_80',['tri',['../group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78',1,'mlx::core::tri(int n, int m, int k, Dtype type, StreamOrDevice s={})'],['../group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5',1,'mlx::core::tri(int n, Dtype type, StreamOrDevice s={})']]], + ['tri_5finv_81',['tri_inv',['../namespacemlx_1_1core_1_1linalg.html#aba1994571326326717b5b5e38c2e0661',1,'mlx::core::linalg']]], + ['tril_82',['tril',['../group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2',1,'mlx::core']]], + ['triu_83',['triu',['../group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412',1,'mlx::core']]], + ['trows_84',['TROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a2528ff5ed472e4ed35415ada42276b07',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a3957fb263fe040fe70683fd1d7b06487',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a94f0ce5bb7d87bc1fb6a7c2ba2b892d4',1,'mlx::steel::Conv2DWeightBlockLoader::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a789683f9ac9d9309d07c05f3bdedd2fd',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a5cefb1285ed13ad3490198e9303453de',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a2aff22af70f685f858adea73f5575cf7',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acbc28f364381166faaeec2783dc88e10',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::TROWS']]], + ['true_5ftype_85',['true_type',['../namespacemlx_1_1steel.html#a594a6ccb75b38b5ae4ddd0d9ad047b3a',1,'mlx::steel']]], + ['trunc_86',['trunc',['../namespacemetal.html#a93cb75a11a362bfc8310ea19c554c887',1,'metal::trunc()'],['../namespacemetal_1_1fast.html#aa62e1075e86c626d97038f16e9433415',1,'metal::fast::trunc()'],['../namespacemetal_1_1precise.html#a334183e7a2dd49b983d072d1e8ee2b27',1,'metal::precise::trunc()']]], + ['truncated_5fnormal_87',['truncated_normal',['../namespacemlx_1_1core_1_1random.html#aece7dc5a29e0488d8b9648f340dbff72',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a39663eda0fd7b274d01499a7b1c9035f',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['try_5fpop_88',['try_pop',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43',1,'pocketfft::detail::threading::concurrent_queue']]], + ['two_89',['two',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421a41877eab6fa3db7d7ed2cda9eba14251',1,'mlx::core::Log']]], + ['type_90',['type',['../structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e',1,'pocketfft::detail::add_vec::type'],['../structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html#a257b1c81fb9f559c48ee90497013494e',1,'pocketfft::detail::add_vec< cmplx< T > >::type'],['../structmlx_1_1steel_1_1integral__constant.html#a6492c15b37d160d3a33e1cbe770aa3f1',1,'mlx::steel::integral_constant::type'],['../structmetal_1_1make__void.html#a53b29d640a06df62047fffdec0571392',1,'metal::make_void::type'],['../structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.html#a98fbc2aa99dd26bb35aa9cd1826318d8',1,'metal::pointer_element< thread T * >::type'],['../structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.html#ab36a7c5a64c0693dd3d8ccb322c163d4',1,'metal::pointer_element< device T * >::type'],['../structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.html#ad154b55b9e450a6376016488c8e68c53',1,'metal::pointer_element< constant T * >::type'],['../structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.html#a78c718d6da9d393c139a385f42472362',1,'metal::pointer_element< threadgroup T * >::type'],['../struct_conditional_type.html#a00bac71c43763817c4422bf0363dc92b',1,'ConditionalType::type'],['../struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.html#abf6b75b3bfc5e3baf8c37afdb8a7e4fc',1,'ConditionalType< true, T, U >::type'],['../structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea',1,'pocketfft::detail::ExecDcst::type'],['../structmlx_1_1core_1_1_reduction_plan.html#a24e407f13d4d02156380ecc1a6748a76',1,'mlx::core::ReductionPlan::type'],['../structmlx_1_1core_1_1_device.html#a763264ec90f7f23c5dced36c3f0db2e5',1,'mlx::core::Device::type'],['../structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e',1,'pocketfft::detail::add_vec< cmplx< T > >::type'],['../struct_conditional_type.html#a00bac71c43763817c4422bf0363dc92b',1,'ConditionalType< true, T, U >::type']]], + ['type_2eh_91',['type.h',['../type_8h.html',1,'']]], + ['type_5fto_5fname_92',['type_to_name',['../namespacemlx_1_1core.html#aef60e3a8d9c987c9c338b193673d2164',1,'mlx::core::type_to_name(const Dtype &t)'],['../namespacemlx_1_1core.html#af1fdfdaa5644394362e6baba30701bae',1,'mlx::core::type_to_name(const array &a)']]], + ['type_5ftraits_2eh_93',['type_traits.h',['../type__traits_8h.html',1,'']]], + ['typetodtype_94',['TypeToDtype',['../structmlx_1_1core_1_1_type_to_dtype.html',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/all_15.js b/docs/build/html/search/all_15.js index 18b37dd1e..486fbf16d 100644 --- a/docs/build/html/search/all_15.js +++ b/docs/build/html/search/all_15.js @@ -6,24 +6,28 @@ var searchData= ['uint32_3',['uint32',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa3de84ad0700f2a1571f633d399e1900e',1,'mlx::core::Dtype::uint32'],['../namespacemlx_1_1core.html#ac63820d6fe10545907c33faf466a929e',1,'mlx::core::uint32']]], ['uint64_4',['uint64',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa2e8d31865e5d4b9d8611e1b991baed07',1,'mlx::core::Dtype::uint64'],['../namespacemlx_1_1core.html#a1f42e3dd4787d2ecec7114a12daefec8',1,'mlx::core::uint64']]], ['uint8_5',['uint8',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa5f423e669d0a8f4ab7c4c3e6da27161a',1,'mlx::core::Dtype::uint8'],['../namespacemlx_1_1core.html#a9778d50afbf456b0bd738751243b3b68',1,'mlx::core::uint8']]], - ['unary_6',['unary',['../namespacemlx_1_1core_1_1metal.html#afac64fd56ac492d6baf6de7e8a00b039',1,'mlx::core::metal']]], - ['unary_2eh_7',['unary.h',['../common_2unary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2unary_8h.html',1,'(Global Namespace)'],['../metal_2unary_8h.html',1,'(Global Namespace)']]], - ['unary_5fg_8',['unary_g',['../metal_2kernels_2unary_8h.html#af13d20efb568db3ab7cd7ec0311c87be',1,'unary.h']]], - ['unary_5fop_5fgpu_9',['unary_op_gpu',['../namespacemlx_1_1core.html#aba2b4accc059f30d4dca88db9f7a6e13',1,'mlx::core']]], - ['unary_5fop_5fgpu_5finplace_10',['unary_op_gpu_inplace',['../namespacemlx_1_1core.html#a668fde2bd280a88f63a68b68a343d375',1,'mlx::core']]], - ['unary_5fops_11',['unary_ops',['../namespacemlx_1_1core_1_1metal.html#a17b471fa52ea5f24ee63e081f46528f5',1,'mlx::core::metal']]], - ['unary_5fops_2eh_12',['unary_ops.h',['../unary__ops_8h.html',1,'']]], - ['unary_5fv_13',['unary_v',['../metal_2kernels_2unary_8h.html#a64e4f6737edddb72122e262977ee3014',1,'unary.h']]], - ['unary_5fv2_14',['unary_v2',['../metal_2kernels_2unary_8h.html#a7c7690f0df9d2acc60b63be58d9c7777',1,'unary.h']]], - ['unaryprimitive_15',['UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html',1,'mlx::core::UnaryPrimitive'],['../classmlx_1_1core_1_1_unary_primitive.html#a189f6d4ed369f82a4b724a29eb056d4e',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(Stream stream)'],['../classmlx_1_1core_1_1_unary_primitive.html#a9935cffc4f246d3d883bc3d26c5163f2',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a780281fb04e2daf1be630c124bd605e3',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(UnaryPrimitive &&other)=delete']]], - ['unflatten_16',['Unflatten',['../classmlx_1_1core_1_1_unflatten.html',1,'mlx::core::Unflatten'],['../classmlx_1_1core_1_1_unflatten.html#a2d1c32eb1fe2bc7641ade600453c7966',1,'mlx::core::Unflatten::Unflatten()']]], - ['unflatten_17',['unflatten',['../group__ops.html#ga666bcc2187a144247e8c0c224b016625',1,'mlx::core']]], - ['uniform_18',['uniform',['../namespacemlx_1_1core_1_1random.html#ac461a0be91e448c9887b38b832c61cc2',1,'mlx::core::random::uniform(const array &low, const array &high, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#abe65438fbb52624386f50f77863a2c5e',1,'mlx::core::random::uniform(T low, U high, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a52913f952387ee3943b3c1f572583ac0',1,'mlx::core::random::uniform(const Shape &shape, Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a0ffb2f91da490f372f898ca2f82104a8',1,'mlx::core::random::uniform(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['unscheduled_19',['unscheduled',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078ae8a9988458b0355001674020a45656fb',1,'mlx::core::array']]], - ['unsignedinteger_20',['unsignedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da9c035d4e66b2c72f583cde964cf3a0d3',1,'mlx::core::Dtype::unsignedinteger'],['../namespacemlx_1_1core.html#a42e9706a5521bb25eaf12ccad94bfc81',1,'mlx::core::unsignedinteger']]], - ['update_5ffence_21',['update_fence',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aeef08f5f3c015578d40de756a6465aa2',1,'mlx::core::metal::CommandEncoder::update_fence()'],['../structmlx_1_1core_1_1_command_encoder.html#aeef08f5f3c015578d40de756a6465aa2',1,'mlx::core::CommandEncoder::update_fence()']]], - ['use_5fout_5fsource_22',['use_out_source',['../steel__gemm__fused_8h.html#a3fe4e4382bda8a419557a5e6f77bc084',1,'steel_gemm_fused.h']]], - ['util_23',['util',['../structpocketfft_1_1detail_1_1util.html',1,'pocketfft::detail']]], - ['utils_24',['utils',['../namespacemlx_1_1core_1_1metal.html#a529dc6c2d4a37ba544b66b2c3cd792cc',1,'mlx::core::metal']]], - ['utils_2eh_25',['utils.h',['../backend_2accelerate_2utils_8h.html',1,'(Global Namespace)'],['../backend_2common_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2utils_8h.html',1,'(Global Namespace)'],['../utils_8h.html',1,'(Global Namespace)']]] + ['unary_6',['unary',['../namespacemlx_1_1core.html#a6c8fdd03ef891d7f47804bf02e9a8507',1,'mlx::core::unary()'],['../namespacemlx_1_1core_1_1metal.html#afac64fd56ac492d6baf6de7e8a00b039',1,'mlx::core::metal::unary()']]], + ['unary_2eh_7',['unary.h',['../cpu_2unary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2unary_8h.html',1,'(Global Namespace)'],['../metal_2unary_8h.html',1,'(Global Namespace)']]], + ['unary_5ffp_8',['unary_fp',['../namespacemlx_1_1core.html#a76a2cb4634f5fd6970a8c3b3753d7a4a',1,'mlx::core']]], + ['unary_5fg_9',['unary_g',['../metal_2kernels_2unary_8h.html#af13d20efb568db3ab7cd7ec0311c87be',1,'unary.h']]], + ['unary_5fop_10',['unary_op',['../namespacemlx_1_1core.html#a27f00519f9756896734fd4d47fec0625',1,'mlx::core::unary_op(const T *a, U *out, Op op, size_t shape, size_t stride)'],['../namespacemlx_1_1core.html#ae20f207ad1ed3badc17cecf08f118b5e',1,'mlx::core::unary_op(const array &a, array &out, Op op)']]], + ['unary_5fop_5fgpu_11',['unary_op_gpu',['../namespacemlx_1_1core.html#aba2b4accc059f30d4dca88db9f7a6e13',1,'mlx::core']]], + ['unary_5fop_5fgpu_5finplace_12',['unary_op_gpu_inplace',['../namespacemlx_1_1core.html#a668fde2bd280a88f63a68b68a343d375',1,'mlx::core']]], + ['unary_5fops_13',['unary_ops',['../namespacemlx_1_1core_1_1metal.html#a17b471fa52ea5f24ee63e081f46528f5',1,'mlx::core::metal']]], + ['unary_5fops_2eh_14',['unary_ops.h',['../cpu_2unary__ops_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2unary__ops_8h.html',1,'(Global Namespace)']]], + ['unary_5fv_15',['unary_v',['../metal_2kernels_2unary_8h.html#a64e4f6737edddb72122e262977ee3014',1,'unary.h']]], + ['unary_5fv2_16',['unary_v2',['../metal_2kernels_2unary_8h.html#a7c7690f0df9d2acc60b63be58d9c7777',1,'unary.h']]], + ['unaryprimitive_17',['UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html',1,'mlx::core::UnaryPrimitive'],['../classmlx_1_1core_1_1_unary_primitive.html#a189f6d4ed369f82a4b724a29eb056d4e',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(Stream stream)'],['../classmlx_1_1core_1_1_unary_primitive.html#a9935cffc4f246d3d883bc3d26c5163f2',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a780281fb04e2daf1be630c124bd605e3',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(UnaryPrimitive &&other)=delete']]], + ['unflatten_18',['Unflatten',['../classmlx_1_1core_1_1_unflatten.html',1,'mlx::core::Unflatten'],['../classmlx_1_1core_1_1_unflatten.html#a2d1c32eb1fe2bc7641ade600453c7966',1,'mlx::core::Unflatten::Unflatten()']]], + ['unflatten_19',['unflatten',['../group__ops.html#ga666bcc2187a144247e8c0c224b016625',1,'mlx::core']]], + ['uniform_20',['uniform',['../namespacemlx_1_1core_1_1random.html#ac461a0be91e448c9887b38b832c61cc2',1,'mlx::core::random::uniform(const array &low, const array &high, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#abe65438fbb52624386f50f77863a2c5e',1,'mlx::core::random::uniform(T low, U high, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a52913f952387ee3943b3c1f572583ac0',1,'mlx::core::random::uniform(const Shape &shape, Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a0ffb2f91da490f372f898ca2f82104a8',1,'mlx::core::random::uniform(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['unscheduled_21',['unscheduled',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078ae8a9988458b0355001674020a45656fb',1,'mlx::core::array']]], + ['unsignedinteger_22',['unsignedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da9c035d4e66b2c72f583cde964cf3a0d3',1,'mlx::core::Dtype::unsignedinteger'],['../namespacemlx_1_1core.html#a42e9706a5521bb25eaf12ccad94bfc81',1,'mlx::core::unsignedinteger']]], + ['update_23',['update',['../classmlx_1_1core_1_1_fence.html#a653279d4023d69751a930a91d3bf010a',1,'mlx::core::Fence']]], + ['update_5ffence_24',['update_fence',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aeef08f5f3c015578d40de756a6465aa2',1,'mlx::core::metal::CommandEncoder::update_fence()'],['../structmlx_1_1core_1_1_command_encoder.html#aeef08f5f3c015578d40de756a6465aa2',1,'mlx::core::CommandEncoder::update_fence()']]], + ['update_5fgpu_25',['update_gpu',['../classmlx_1_1core_1_1_fence.html#a6c5652aad6e93b06c72258bb8d9c19fc',1,'mlx::core::Fence']]], + ['use_5fout_5fsource_26',['use_out_source',['../steel__gemm__fused_8h.html#a3fe4e4382bda8a419557a5e6f77bc084',1,'steel_gemm_fused.h']]], + ['util_27',['util',['../structpocketfft_1_1detail_1_1util.html',1,'pocketfft::detail']]], + ['utils_28',['utils',['../namespacemlx_1_1core_1_1metal.html#a529dc6c2d4a37ba544b66b2c3cd792cc',1,'mlx::core::metal']]], + ['utils_2eh_29',['utils.h',['../backend_2common_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2utils_8h.html',1,'(Global Namespace)'],['../utils_8h.html',1,'(Global Namespace)']]] ]; diff --git a/docs/build/html/search/all_16.js b/docs/build/html/search/all_16.js index 284e68509..63233ff63 100644 --- a/docs/build/html/search/all_16.js +++ b/docs/build/html/search/all_16.js @@ -1,28 +1,31 @@ var searchData= [ ['v_0',['V',['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a5206560a306a2e085a437fd258eb57ce',1,'mlx::core::Dtype']]], - ['v_1',['v',['../structmlx_1_1steel_1_1_block_loader_1_1_read_vector.html#a20963f7191251defca48bf8a843d019d',1,'mlx::steel::BlockLoader::ReadVector']]], + ['v_1',['v',['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html#a3d47d5ad1ff8981bd9876a5fc1870174',1,'mlx::core::simd::ScalarT< bool, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html#af2775b07509324182bd715aac65b7eb0',1,'mlx::core::simd::ScalarT< int8_t, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html#aaf352e77f7ab310c40a31d3dd2bde0eb',1,'mlx::core::simd::ScalarT< uint64_t, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html#aa36db163e4909aea98b7129764184801',1,'mlx::core::simd::ScalarT< int64_t, N >::v'],['../structmlx_1_1steel_1_1_block_loader_1_1_read_vector.html#a20963f7191251defca48bf8a843d019d',1,'mlx::steel::BlockLoader::ReadVector::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT< bool, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT< int64_t, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT< int8_t, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT< uint64_t, N >::v']]], ['v_5fstrides_2',['V_strides',['../structmlx_1_1steel_1_1_attn_params.html#ad1495980297901b8ded1fb6dd73979b1',1,'mlx::steel::AttnParams']]], ['val_3',['Val',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1da',1,'mlx::core::Dtype']]], - ['val_4',['val',['../structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534',1,'pocketfft::detail::VLEN::val'],['../structmlx__atomic.html#a6f6651b8dd8149917c50cd99b13c6747',1,'mlx_atomic::val'],['../structmlx_1_1core_1_1_dtype.html#a7a99656f121c8922ab82e72c8e9bd7f1',1,'mlx::core::Dtype::val()']]], + ['val_4',['val',['../structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534',1,'pocketfft::detail::VLEN::val'],['../structmlx__atomic.html#a6f6651b8dd8149917c50cd99b13c6747',1,'mlx_atomic::val'],['../structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html#a8dbf729fcd8c4a16e41b546c7405543d',1,'mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >::val'],['../structmlx_1_1core_1_1_dtype.html#a7a99656f121c8922ab82e72c8e9bd7f1',1,'mlx::core::Dtype::val()'],['../structmlx__atomic.html#a6f6651b8dd8149917c50cd99b13c6747',1,'mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >::val']]], ['val_5ffrags_5',['val_frags',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a684e6c6d9f00f583994285b60aaa3b62',1,'mlx::steel::MMATile']]], - ['val_5ft_6',['val_t',['../struct_kernel_merge_sort.html#a4e3f09896275956fc4c23e1f157dca3b',1,'KernelMergeSort']]], - ['valid_7',['valid',['../classmlx_1_1core_1_1_event.html#aa77afd9669e2ef9d5e9ae1c2c6fd24fa',1,'mlx::core::Event']]], - ['value_8',['value',['../structmlx_1_1steel_1_1integral__constant.html#a4efa69cb3fd42ac0dcad46578600d637',1,'mlx::steel::integral_constant::value'],['../classmlx_1_1core_1_1_event.html#ab71c7baee3d1d02ad6a2001bbf90b970',1,'mlx::core::Event::value()']]], + ['valid_6',['valid',['../classmlx_1_1core_1_1_event.html#aa77afd9669e2ef9d5e9ae1c2c6fd24fa',1,'mlx::core::Event']]], + ['valt_7',['ValT',['../struct_kernel_merge_sort.html#a92910d137ea47703820cde5666d80574',1,'KernelMergeSort']]], + ['value_8',['value',['../structmlx_1_1core_1_1simd_1_1_simd.html#a36e2b7db5ce6eb4dd456e99a4cd2c2cf',1,'mlx::core::simd::Simd::value'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#ac348445fd44bce2b6ee77adeac7d82df',1,'mlx::core::simd::Simd< T, 1 >::value'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#aa6042509bb67de25bedbee1ee1d66094',1,'mlx::core::simd::Simd< float16_t, N >::value'],['../structmlx_1_1steel_1_1integral__constant.html#a4efa69cb3fd42ac0dcad46578600d637',1,'mlx::steel::integral_constant::value'],['../classmlx_1_1core_1_1_event.html#ab71c7baee3d1d02ad6a2001bbf90b970',1,'mlx::core::Event::value()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a36e2b7db5ce6eb4dd456e99a4cd2c2cf',1,'mlx::core::simd::Simd< T, 1 >::value'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a36e2b7db5ce6eb4dd456e99a4cd2c2cf',1,'mlx::core::simd::Simd< float16_t, N >::value']]], ['value_5fand_5fgrad_9',['value_and_grad',['../namespacemlx_1_1core.html#abf49b337a00997231c0f7fd389efa8f3',1,'mlx::core::value_and_grad(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#a7b987f404b8699de00f9e0099ab6b1b0',1,'mlx::core::value_and_grad(const std::function< std::vector< array >(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a5a64dc878b29403d27e50bd7a288cc04',1,'mlx::core::value_and_grad(const std::function< array(const array &)> &fun)'],['../namespacemlx_1_1core.html#a7620f1ae298127cb6181db9162f012a7',1,'mlx::core::value_and_grad(const std::function< array(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#a2f69ffc30d66b1fca8f24b65be161a51',1,'mlx::core::value_and_grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)']]], ['value_5ftype_10',['value_type',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#ab86a2740ed9ce3199135372ff1d88c76',1,'pocketfft::detail::threading::aligned_allocator::value_type'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ae24fe304397e961687d0d4c7012b8ae4',1,'mlx::core::array::ArrayIterator::value_type'],['../structmlx_1_1steel_1_1integral__constant.html#a0569cc1334e0bc4f474304b33d365759',1,'mlx::steel::integral_constant::value_type']]], ['valueandgradfn_11',['ValueAndGradFn',['../namespacemlx_1_1core.html#ab79d66ddf1ec38b2f2c01234892a2230',1,'mlx::core']]], ['var_12',['var',['../group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9',1,'mlx::core::var(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga7d7b38d118fa2613214078ef0f7d5a42',1,'mlx::core::var(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga78ddeb966cbe7a5b0aa17e1de43025f2',1,'mlx::core::var(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga4fbf3e3f98f2e4956faf87af320aa9d0',1,'mlx::core::var(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], - ['vec_5fsize_13',['vec_size',['../structmlx_1_1steel_1_1_block_loader.html#a58bdf9b9c81962733e22ecdeae28c092',1,'mlx::steel::BlockLoader::vec_size'],['../structmlx_1_1steel_1_1_block_loader_t.html#a9ac651d9e5097507c57b10dfeb40bfe5',1,'mlx::steel::BlockLoaderT::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#adcc83bf6c02391cc2375e55c06a1c9a4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a71c313e1597a2bb99f7b07d434e119d2',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a10109dc9553207f5a365799e4969c6d2',1,'mlx::steel::Conv2DWeightBlockLoader::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a6b0b18428516d1d6dcae3beb3faee81c',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a006153d274aa13d5fd4448b4607fff3a',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1587047caa339cf5b2c06adc4b332ab8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08dba753ec7c8ea2892775746933b3e7',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::vec_size']]], + ['vec_5fsize_13',['vec_size',['../structmlx_1_1steel_1_1_block_loader.html#a58bdf9b9c81962733e22ecdeae28c092',1,'mlx::steel::BlockLoader::vec_size'],['../structmlx_1_1steel_1_1_block_loader_t.html#a9ac651d9e5097507c57b10dfeb40bfe5',1,'mlx::steel::BlockLoaderT::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#adcc83bf6c02391cc2375e55c06a1c9a4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a71c313e1597a2bb99f7b07d434e119d2',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a10109dc9553207f5a365799e4969c6d2',1,'mlx::steel::Conv2DWeightBlockLoader::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#a71449551bbfe56058440755dfd50fc75',1,'mlx::steel::ChannelHelper< 1 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#acfb18991a77a9d1d4a79918ac5f387af',1,'mlx::steel::ChannelHelper< 2 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#a5cb83774601c29564a6bbc010fc0bf7f',1,'mlx::steel::ChannelHelper< 3 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#af28cdbe2a3c027d95832de07f60448ca',1,'mlx::steel::ChannelHelper< 4 >::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a6b0b18428516d1d6dcae3beb3faee81c',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a006153d274aa13d5fd4448b4607fff3a',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1587047caa339cf5b2c06adc4b332ab8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08dba753ec7c8ea2892775746933b3e7',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper< 1 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper< 2 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper< 3 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper< 4 >::vec_size']]], ['vector_14',['Vector',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a57dea6f5039281b7fee517fc43bf3110',1,'mlx::core']]], - ['view_15',['View',['../classmlx_1_1core_1_1_view.html',1,'mlx::core::View'],['../classmlx_1_1core_1_1_view.html#ad7eed156c308e9a29a8b41f965ec941e',1,'mlx::core::View::View()']]], - ['view_16',['view',['../group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02',1,'mlx::core']]], - ['vjp_17',['vjp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abbf6d1d63dcda207ad7d9eeb4fc36225',1,'mlx::core::distributed::AllReduce::vjp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#aa5eff6fc128b71220899aab8ab9116fb',1,'mlx::core::distributed::AllGather::vjp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a74be4bcd0382f7f6400bf73fd5569c91',1,'mlx::core::fast::Custom::vjp()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#aacfbbbc15fcee0a5ce4f519ca3cca5eb',1,'mlx::core::fast::RMSNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#ae5e1b5df0705a6b1d141691a4396b0b6',1,'mlx::core::fast::LayerNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#ad999105414badd66c8fd9e069454a533',1,'mlx::core::fast::RoPE::vjp()'],['../classmlx_1_1core_1_1_primitive.html#a1dcb6807326eeab62474c6a0e3836d42',1,'mlx::core::Primitive::vjp()'],['../classmlx_1_1core_1_1_abs.html#aa2dd8ec0989e716b77394ac349b34592',1,'mlx::core::Abs::vjp()'],['../classmlx_1_1core_1_1_add.html#ac28e581862880e24ed2b99bb6a916607',1,'mlx::core::Add::vjp()'],['../classmlx_1_1core_1_1_add_m_m.html#ac1562a37cec6928e01281926ebeb47c6',1,'mlx::core::AddMM::vjp()'],['../classmlx_1_1core_1_1_arc_cos.html#a78e73e5e639d1249c7fe9614bf157c92',1,'mlx::core::ArcCos::vjp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a856c677f16e2b3f2edd2491e35db2d26',1,'mlx::core::ArcCosh::vjp()'],['../classmlx_1_1core_1_1_arc_sin.html#ab4057cd5ef1a8359f97493018e10d3a1',1,'mlx::core::ArcSin::vjp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a7988ee5b9e1e7e498dcab73d61ba147e',1,'mlx::core::ArcSinh::vjp()'],['../classmlx_1_1core_1_1_arc_tan.html#a5fefc3634b96a67ff8ae011a8ee180c2',1,'mlx::core::ArcTan::vjp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a99840c282e37b2b2a9c312e6e8ade1d2',1,'mlx::core::ArcTan2::vjp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a07da5797f7aaf3dfe43bf24e8562ac72',1,'mlx::core::ArcTanh::vjp()'],['../classmlx_1_1core_1_1_arg_partition.html#ade23d014717a0b0235d00073503aeac0',1,'mlx::core::ArgPartition::vjp()'],['../classmlx_1_1core_1_1_arg_reduce.html#a60d272685a373e6fe879416481a1ce1a',1,'mlx::core::ArgReduce::vjp()'],['../classmlx_1_1core_1_1_as_type.html#ac38a4f889311a3b5e5be9a67dcb93e18',1,'mlx::core::AsType::vjp()'],['../classmlx_1_1core_1_1_as_strided.html#a34783284c9b2f5b4a62c3c3ee5dd4062',1,'mlx::core::AsStrided::vjp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6131ed1c317ff8700a3e9b13fdaa9d61',1,'mlx::core::BitwiseBinary::vjp()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a1adf20087ee2f685bf39c2724b8e7120',1,'mlx::core::BlockMaskedMM::vjp()'],['../classmlx_1_1core_1_1_gather_m_m.html#a76c9f27c57354f6230b43944882e1bda',1,'mlx::core::GatherMM::vjp()'],['../classmlx_1_1core_1_1_broadcast_axes.html#aea8ef2b2616568a2bb56695381a035be',1,'mlx::core::BroadcastAxes::vjp()'],['../classmlx_1_1core_1_1_broadcast.html#a0318847c9be40f00b23907ad56037d18',1,'mlx::core::Broadcast::vjp()'],['../classmlx_1_1core_1_1_ceil.html#ac2f5a2bd84b8f013e5ce688419a88acb',1,'mlx::core::Ceil::vjp()'],['../classmlx_1_1core_1_1_compiled.html#a32462e65c52f84b708188130cc508133',1,'mlx::core::Compiled::vjp()'],['../classmlx_1_1core_1_1_concatenate.html#a8155db9100ec3b8bd0bc94baeaeee3b0',1,'mlx::core::Concatenate::vjp()'],['../classmlx_1_1core_1_1_contiguous.html#abf488f02057fd5852f38b2e8a600ad2a',1,'mlx::core::Contiguous::vjp()'],['../classmlx_1_1core_1_1_convolution.html#af8eb9c0c055ad20aa74b547016917690',1,'mlx::core::Convolution::vjp()'],['../classmlx_1_1core_1_1_copy.html#a6c4dee582001e9983e9517485ee37efd',1,'mlx::core::Copy::vjp()'],['../classmlx_1_1core_1_1_cos.html#a51d84113728e651ef9d4a1fe671c4d00',1,'mlx::core::Cos::vjp()'],['../classmlx_1_1core_1_1_cosh.html#a0791abd4305a333fb3b181a5357ce0f4',1,'mlx::core::Cosh::vjp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa1da36cef632df767cd9809d6cf06209',1,'mlx::core::CustomTransforms::vjp()'],['../classmlx_1_1core_1_1_depends.html#a02996fa45f01f7cb9f37074d5f8ccab0',1,'mlx::core::Depends::vjp()'],['../classmlx_1_1core_1_1_divide.html#ad3af7c70cad22c1a1a75b4a78ef793b6',1,'mlx::core::Divide::vjp()'],['../classmlx_1_1core_1_1_div_mod.html#a8c914a07f666a1d9377a27ed5d55e7c1',1,'mlx::core::DivMod::vjp()'],['../classmlx_1_1core_1_1_select.html#a9b522487b78fceeca7f827cd1c29a9a3',1,'mlx::core::Select::vjp()'],['../classmlx_1_1core_1_1_remainder.html#ab18f7bca1027ae71847a50da0933cec6',1,'mlx::core::Remainder::vjp()'],['../classmlx_1_1core_1_1_equal.html#af3c1bfcd1bf50922fc00e302bb193736',1,'mlx::core::Equal::vjp()'],['../classmlx_1_1core_1_1_erf.html#a1f529e95a42a2d69a8b18979d3ee2909',1,'mlx::core::Erf::vjp()'],['../classmlx_1_1core_1_1_erf_inv.html#a48afff12a58ddefae7ae0245c3580189',1,'mlx::core::ErfInv::vjp()'],['../classmlx_1_1core_1_1_exp.html#a94b9b7d137c3640d290b96c5e8b7e1a8',1,'mlx::core::Exp::vjp()'],['../classmlx_1_1core_1_1_expm1.html#af6ce416169190479c9792bb9cdbe2f43',1,'mlx::core::Expm1::vjp()'],['../classmlx_1_1core_1_1_expand_dims.html#a2fb3c65ba7a3b2d1f33a3c681fda8896',1,'mlx::core::ExpandDims::vjp()'],['../classmlx_1_1core_1_1_f_f_t.html#aafc895614a6e368c0e6d64af20d01090',1,'mlx::core::FFT::vjp()'],['../classmlx_1_1core_1_1_flatten.html#ab549a8c38b63055e2d5cd672f7676aab',1,'mlx::core::Flatten::vjp()'],['../classmlx_1_1core_1_1_floor.html#a589e2cf99b6fd1a5ba85534a2a31338e',1,'mlx::core::Floor::vjp()'],['../classmlx_1_1core_1_1_full.html#a49e76e7a8641f990701abc1b3bd49969',1,'mlx::core::Full::vjp()'],['../classmlx_1_1core_1_1_gather.html#aacf612a8f5f1cdbbfd19707d8d33c426',1,'mlx::core::Gather::vjp()'],['../classmlx_1_1core_1_1_greater.html#a341766a8a7e41d2a1160d35d4e781679',1,'mlx::core::Greater::vjp()'],['../classmlx_1_1core_1_1_greater_equal.html#a62f07a4ac54c708307c82aac0e5693ee',1,'mlx::core::GreaterEqual::vjp()'],['../classmlx_1_1core_1_1_hadamard.html#af4134775427b8998d66f489468b98656',1,'mlx::core::Hadamard::vjp()'],['../classmlx_1_1core_1_1_imag.html#a80da5fdd0fa549eebd7804c0e261848b',1,'mlx::core::Imag::vjp()'],['../classmlx_1_1core_1_1_less.html#aaf205d389b5e602e0814b68f66de8f50',1,'mlx::core::Less::vjp()'],['../classmlx_1_1core_1_1_less_equal.html#aab2aab7590c299885e815c18eedd1028',1,'mlx::core::LessEqual::vjp()'],['../classmlx_1_1core_1_1_log.html#a40885dccfbf928c4d035881be1d49280',1,'mlx::core::Log::vjp()'],['../classmlx_1_1core_1_1_log1p.html#a3113c1d2b4c5e73d0b470f42dc48a880',1,'mlx::core::Log1p::vjp()'],['../classmlx_1_1core_1_1_logical_not.html#af2c3c241cf3910fbaba013c69d052a50',1,'mlx::core::LogicalNot::vjp()'],['../classmlx_1_1core_1_1_logical_and.html#ae42f8fc454577b0fd6410cae9d5f3b54',1,'mlx::core::LogicalAnd::vjp()'],['../classmlx_1_1core_1_1_logical_or.html#a51aed488f52d5031998689af9cb17847',1,'mlx::core::LogicalOr::vjp()'],['../classmlx_1_1core_1_1_log_add_exp.html#ae231af0ed24a93eb647ee58c2d2b20b4',1,'mlx::core::LogAddExp::vjp()'],['../classmlx_1_1core_1_1_matmul.html#a524136cca481598ea20894d85ca66bb0',1,'mlx::core::Matmul::vjp()'],['../classmlx_1_1core_1_1_maximum.html#a7de15d7b28784e24bbfc7e85ddcbcff3',1,'mlx::core::Maximum::vjp()'],['../classmlx_1_1core_1_1_minimum.html#a48a0cbe3a6c4f7473c00e343f63b5204',1,'mlx::core::Minimum::vjp()'],['../classmlx_1_1core_1_1_multiply.html#a74b7556ec03e2c3d3f971666d06f5db1',1,'mlx::core::Multiply::vjp()'],['../classmlx_1_1core_1_1_negative.html#a889585f056d33bda30c30311257af52a',1,'mlx::core::Negative::vjp()'],['../classmlx_1_1core_1_1_not_equal.html#a0361f29f4ae1235bdf3f3304527e2d4b',1,'mlx::core::NotEqual::vjp()'],['../classmlx_1_1core_1_1_pad.html#ad8a7e547644f2717a24322968e971038',1,'mlx::core::Pad::vjp()'],['../classmlx_1_1core_1_1_partition.html#a7110772b6cd2d430a2b825cf5c952ca9',1,'mlx::core::Partition::vjp()'],['../classmlx_1_1core_1_1_power.html#a1453bb8307d6ff33134f1e00263bf082',1,'mlx::core::Power::vjp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#acb975e272b4a88ab232ef7f7c3a2bf26',1,'mlx::core::QuantizedMatmul::vjp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#ae08a4b7d28902d46f39e66beeb0e23ab',1,'mlx::core::GatherQMM::vjp()'],['../classmlx_1_1core_1_1_real.html#a29f6109339c5141a862ceae72c8b80fe',1,'mlx::core::Real::vjp()'],['../classmlx_1_1core_1_1_reshape.html#ab17294ecc6b5d4e89626fb48c7516365',1,'mlx::core::Reshape::vjp()'],['../classmlx_1_1core_1_1_reduce.html#a684883d2a96315f548ca769510e28e4e',1,'mlx::core::Reduce::vjp()'],['../classmlx_1_1core_1_1_round.html#af8f085e08b7fa8840c52a20b12ca35ce',1,'mlx::core::Round::vjp()'],['../classmlx_1_1core_1_1_scan.html#aaf13f72620b4b5d6a20e1228930e848e',1,'mlx::core::Scan::vjp()'],['../classmlx_1_1core_1_1_scatter.html#a0b51287fba789bb139ed61d40a0c636a',1,'mlx::core::Scatter::vjp()'],['../classmlx_1_1core_1_1_sigmoid.html#aac2f56a4c8362e36a28e232758ca52cf',1,'mlx::core::Sigmoid::vjp()'],['../classmlx_1_1core_1_1_sign.html#aa60ac52edd739fbdf388a997acd01bce',1,'mlx::core::Sign::vjp()'],['../classmlx_1_1core_1_1_sin.html#aedefe550ab4b0687858981bc0bcfbfa0',1,'mlx::core::Sin::vjp()'],['../classmlx_1_1core_1_1_sinh.html#a6b39fdd429bbb4de389e7c904fd561f0',1,'mlx::core::Sinh::vjp()'],['../classmlx_1_1core_1_1_slice.html#a291746a527ff991b66249fb2b54b685f',1,'mlx::core::Slice::vjp()'],['../classmlx_1_1core_1_1_slice_update.html#aedcdc60a0477997a96306c02b66d3f77',1,'mlx::core::SliceUpdate::vjp()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a29caf03256945f7732a52d551191f8fa',1,'mlx::core::DynamicSlice::vjp()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ab2817cb9d1bfcd3de6454d841909da1f',1,'mlx::core::DynamicSliceUpdate::vjp()'],['../classmlx_1_1core_1_1_softmax.html#abb68c311c45ee422a7c966accde9041b',1,'mlx::core::Softmax::vjp()'],['../classmlx_1_1core_1_1_sort.html#a3a8900dce53ee4eb7a1b83806e629358',1,'mlx::core::Sort::vjp()'],['../classmlx_1_1core_1_1_split.html#a7e8730f9cffa9872fff6f8d577031674',1,'mlx::core::Split::vjp()'],['../classmlx_1_1core_1_1_square.html#abcd9516da7f02dc906368c23b0bca263',1,'mlx::core::Square::vjp()'],['../classmlx_1_1core_1_1_sqrt.html#a08a21bd2c3a016f042d95aca294e68f3',1,'mlx::core::Sqrt::vjp()'],['../classmlx_1_1core_1_1_subtract.html#a3a3322be7c3bcaa0397cf099091df16b',1,'mlx::core::Subtract::vjp()'],['../classmlx_1_1core_1_1_squeeze.html#a8d95a13d7cc5586d48a38e9199180d06',1,'mlx::core::Squeeze::vjp()'],['../classmlx_1_1core_1_1_tan.html#a4639836cff03d73c769387d6943e92d7',1,'mlx::core::Tan::vjp()'],['../classmlx_1_1core_1_1_tanh.html#afe7b05e2b36b99c3a1b66f0cd3544e95',1,'mlx::core::Tanh::vjp()'],['../classmlx_1_1core_1_1_unflatten.html#a34f1218fa1d0e28f3ee10b65e6b0e319',1,'mlx::core::Unflatten::vjp()'],['../classmlx_1_1core_1_1_transpose.html#ac7805aa29b34afdf8852554f1e759f80',1,'mlx::core::Transpose::vjp()'],['../namespacemlx_1_1core.html#a1b33e2c2e3471420490cf0be2de6de18',1,'mlx::core::vjp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &cotangents)'],['../namespacemlx_1_1core.html#a2065a11249c3f4356ffd69b7a8c487ff',1,'mlx::core::vjp(const std::function< array(const array &)> &fun, const array &primal, const array &cotangent)']]], - ['vlen_18',['VLEN',['../structpocketfft_1_1detail_1_1_v_l_e_n.html',1,'pocketfft::detail']]], - ['vmap_19',['vmap',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a3f2dc71859847ca675ec4bfbe125035a',1,'mlx::core::distributed::AllReduce::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ad532d1d51f089dec3c84799b724ea031',1,'mlx::core::distributed::AllGather::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a5cfb66191b9e8b86649da77af55b0f93',1,'mlx::core::distributed::Send::vmap()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a7f4c3a4c48c6807faa36fb31e39dad8d',1,'mlx::core::fast::Custom::vmap()'],['../classmlx_1_1core_1_1_primitive.html#ac632b9619dd7a6a0f177bd36202e8103',1,'mlx::core::Primitive::vmap()'],['../classmlx_1_1core_1_1_abs.html#a4c9c98f1d71432fd3752ad9a6a8e7f2f',1,'mlx::core::Abs::vmap()'],['../classmlx_1_1core_1_1_add.html#a0e557d4d896153f84a25532562e4c646',1,'mlx::core::Add::vmap()'],['../classmlx_1_1core_1_1_add_m_m.html#a73ce80b3a37ec2523943028d50ebce81',1,'mlx::core::AddMM::vmap()'],['../classmlx_1_1core_1_1_arc_cos.html#a7548e23ace6827674aa6d284d44ccf83',1,'mlx::core::ArcCos::vmap()'],['../classmlx_1_1core_1_1_arc_cosh.html#af8ff78e910a9e485a203e1d3347bd461',1,'mlx::core::ArcCosh::vmap()'],['../classmlx_1_1core_1_1_arc_sin.html#a7cabb1e5a2bda44944378822c671ec82',1,'mlx::core::ArcSin::vmap()'],['../classmlx_1_1core_1_1_arc_sinh.html#a9e72b9751939387c333b5d4e19a37f6d',1,'mlx::core::ArcSinh::vmap()'],['../classmlx_1_1core_1_1_arc_tan.html#a1fb921554544a56498bc54f82e4a0556',1,'mlx::core::ArcTan::vmap()'],['../classmlx_1_1core_1_1_arc_tan2.html#ae02cb9fbf25e93dc1d7fbc9e3fb28634',1,'mlx::core::ArcTan2::vmap()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6ddcae68873559211cb91e7740dfc040',1,'mlx::core::ArcTanh::vmap()'],['../classmlx_1_1core_1_1_arg_partition.html#a441093795bcc31495ab5fbc9957b740a',1,'mlx::core::ArgPartition::vmap()'],['../classmlx_1_1core_1_1_arg_reduce.html#abfec42fa06ea15edaf393593751fb1ba',1,'mlx::core::ArgReduce::vmap()'],['../classmlx_1_1core_1_1_arg_sort.html#a3522bbbe4626a467394c1a8a9d7ac34e',1,'mlx::core::ArgSort::vmap()'],['../classmlx_1_1core_1_1_as_type.html#a7ebaf86fd6cad4a1ecfd7cde1ee0b0cc',1,'mlx::core::AsType::vmap()'],['../classmlx_1_1core_1_1_bitwise_binary.html#aa10be55f05bc1868bf4b375dc475f965',1,'mlx::core::BitwiseBinary::vmap()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a4e04f564d440e2d312c335db52c308e1',1,'mlx::core::BroadcastAxes::vmap()'],['../classmlx_1_1core_1_1_broadcast.html#aee4c71c2588ad01eb57e10f346cd666f',1,'mlx::core::Broadcast::vmap()'],['../classmlx_1_1core_1_1_ceil.html#ae86819990b43bdb0c2b3a25719b3a7a4',1,'mlx::core::Ceil::vmap()'],['../classmlx_1_1core_1_1_compiled.html#a732e7548f53977b4513bb7f30a04c30d',1,'mlx::core::Compiled::vmap()'],['../classmlx_1_1core_1_1_concatenate.html#a58c54dcf8e4b045d25edd3afc2caffc1',1,'mlx::core::Concatenate::vmap()'],['../classmlx_1_1core_1_1_conjugate.html#a2c7632c8ae0ca07777e23a0a79344e60',1,'mlx::core::Conjugate::vmap()'],['../classmlx_1_1core_1_1_contiguous.html#a563221e90b15aa90bfae23d29c10e4ec',1,'mlx::core::Contiguous::vmap()'],['../classmlx_1_1core_1_1_copy.html#a669b10253c15b769d90058d1ad7d0e61',1,'mlx::core::Copy::vmap()'],['../classmlx_1_1core_1_1_cos.html#aec9460daf0131156734013d03b230cd6',1,'mlx::core::Cos::vmap()'],['../classmlx_1_1core_1_1_cosh.html#a1ab2386e7d96219b6e4a525f7dac0406',1,'mlx::core::Cosh::vmap()'],['../classmlx_1_1core_1_1_custom_transforms.html#a906a2ff30d9c5281fbf1fa927e4c021b',1,'mlx::core::CustomTransforms::vmap()'],['../classmlx_1_1core_1_1_divide.html#a83e7da52831165b3a026e97b63770242',1,'mlx::core::Divide::vmap()'],['../classmlx_1_1core_1_1_div_mod.html#ae709e0fdd83994bd1d156e0d0e6a7942',1,'mlx::core::DivMod::vmap()'],['../classmlx_1_1core_1_1_select.html#a84e80361c8cf02536b4b98098793550f',1,'mlx::core::Select::vmap()'],['../classmlx_1_1core_1_1_remainder.html#a79867e1099a2e3c2d3e87407b2ab6e3d',1,'mlx::core::Remainder::vmap()'],['../classmlx_1_1core_1_1_equal.html#aea9cc3c88924ac824d72c39c2e83b0ca',1,'mlx::core::Equal::vmap()'],['../classmlx_1_1core_1_1_erf.html#abe554f553356654a3e800ba368108aaa',1,'mlx::core::Erf::vmap()'],['../classmlx_1_1core_1_1_erf_inv.html#ad5d7634e8568af8cc4a54a558a48d0e9',1,'mlx::core::ErfInv::vmap()'],['../classmlx_1_1core_1_1_exp.html#a0fcd579fe148b4c3dbc72e514b81bb37',1,'mlx::core::Exp::vmap()'],['../classmlx_1_1core_1_1_expm1.html#aa4caa848b2ea97e71ee3dd33de039296',1,'mlx::core::Expm1::vmap()'],['../classmlx_1_1core_1_1_expand_dims.html#a380c9ddc25a1f973c3d71b42f8a19486',1,'mlx::core::ExpandDims::vmap()'],['../classmlx_1_1core_1_1_f_f_t.html#ac32d6cc9b67289124f855ea68a61ede1',1,'mlx::core::FFT::vmap()'],['../classmlx_1_1core_1_1_flatten.html#a244a03915313286d36ed4d36b01a99f2',1,'mlx::core::Flatten::vmap()'],['../classmlx_1_1core_1_1_floor.html#aea4dc79a65774990e775ad49519a5d10',1,'mlx::core::Floor::vmap()'],['../classmlx_1_1core_1_1_full.html#afc57ab6bd9ebdbbf042af54a59785d95',1,'mlx::core::Full::vmap()'],['../classmlx_1_1core_1_1_gather.html#abab0c4c204e66489825ce80d2194a275',1,'mlx::core::Gather::vmap()'],['../classmlx_1_1core_1_1_greater.html#a6d8267411fc4951de781f9e8e6c53aa0',1,'mlx::core::Greater::vmap()'],['../classmlx_1_1core_1_1_greater_equal.html#ab0e1be93eb01b0ce7fa83e953f5e3e1d',1,'mlx::core::GreaterEqual::vmap()'],['../classmlx_1_1core_1_1_hadamard.html#a9f1a172e6246859e813002abe9b8f99c',1,'mlx::core::Hadamard::vmap()'],['../classmlx_1_1core_1_1_imag.html#ace9906672bd88df0573653883d58ecb3',1,'mlx::core::Imag::vmap()'],['../classmlx_1_1core_1_1_less.html#a5fee5956cf087d8405359121aa62ba7e',1,'mlx::core::Less::vmap()'],['../classmlx_1_1core_1_1_less_equal.html#a3d5df21db184f2b7620cda9da1684480',1,'mlx::core::LessEqual::vmap()'],['../classmlx_1_1core_1_1_log.html#a007ddbcf911093231f607a8b9ed5cd49',1,'mlx::core::Log::vmap()'],['../classmlx_1_1core_1_1_log1p.html#a7122576f95ce479926bbbbc690891f71',1,'mlx::core::Log1p::vmap()'],['../classmlx_1_1core_1_1_logical_not.html#a5308a271619ee74df561b0aaf525915d',1,'mlx::core::LogicalNot::vmap()'],['../classmlx_1_1core_1_1_logical_and.html#aacc5f6f53ffc327b7771485e3da2a4e5',1,'mlx::core::LogicalAnd::vmap()'],['../classmlx_1_1core_1_1_logical_or.html#a6e2e77e6aaf47872b2e96b151c32daf3',1,'mlx::core::LogicalOr::vmap()'],['../classmlx_1_1core_1_1_log_add_exp.html#a82190aa1421a9734b6e9480debffac78',1,'mlx::core::LogAddExp::vmap()'],['../classmlx_1_1core_1_1_matmul.html#a3a1c6e70bac300240760fe41a58340c2',1,'mlx::core::Matmul::vmap()'],['../classmlx_1_1core_1_1_maximum.html#ab664918e0d71cfec1318a9879e78c5d3',1,'mlx::core::Maximum::vmap()'],['../classmlx_1_1core_1_1_minimum.html#adab0f31acf68075a0be908d8eb882980',1,'mlx::core::Minimum::vmap()'],['../classmlx_1_1core_1_1_multiply.html#ae7e82c8fc8cbaf4e00c27eb54fac7dbf',1,'mlx::core::Multiply::vmap()'],['../classmlx_1_1core_1_1_negative.html#a1f8a6079e272f1a0599f88a1a8419cf0',1,'mlx::core::Negative::vmap()'],['../classmlx_1_1core_1_1_not_equal.html#ab8b57932f03c8eee664bf89adeaa43b5',1,'mlx::core::NotEqual::vmap()'],['../classmlx_1_1core_1_1_number_of_elements.html#a977d83eae845b8bd8c0b98b48cb1c6c2',1,'mlx::core::NumberOfElements::vmap()'],['../classmlx_1_1core_1_1_pad.html#a85658812a0f3275ba3eb74b7c75686cf',1,'mlx::core::Pad::vmap()'],['../classmlx_1_1core_1_1_partition.html#aa0cc55e4d4d2cb5d129d32832321df2c',1,'mlx::core::Partition::vmap()'],['../classmlx_1_1core_1_1_power.html#a5e22749592413a9adbdc877b03b87c8f',1,'mlx::core::Power::vmap()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a3434394140177b285f971c9ffe7e8763',1,'mlx::core::QuantizedMatmul::vmap()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a13ce5e138ebddb8780a034452f68892f',1,'mlx::core::GatherQMM::vmap()'],['../classmlx_1_1core_1_1_random_bits.html#a0dc12f053c6492f934bc18031412c415',1,'mlx::core::RandomBits::vmap()'],['../classmlx_1_1core_1_1_real.html#a07fbbefb6a1bc1ebd3985b24c36693b6',1,'mlx::core::Real::vmap()'],['../classmlx_1_1core_1_1_reshape.html#ae239dd3c6cab147e4af572dc58204f9d',1,'mlx::core::Reshape::vmap()'],['../classmlx_1_1core_1_1_reduce.html#abab1b5aa01ccad44f213f510c3596b38',1,'mlx::core::Reduce::vmap()'],['../classmlx_1_1core_1_1_round.html#a6fad8799a7982e1ccbe05be7cc38a7fd',1,'mlx::core::Round::vmap()'],['../classmlx_1_1core_1_1_scan.html#a297c7cc89c9bf9d186ebdebb634c7804',1,'mlx::core::Scan::vmap()'],['../classmlx_1_1core_1_1_scatter.html#a696c38b373a7a7c71bc112bd1117e322',1,'mlx::core::Scatter::vmap()'],['../classmlx_1_1core_1_1_sigmoid.html#a12712c23037e38192cbccd2d4b14cc85',1,'mlx::core::Sigmoid::vmap()'],['../classmlx_1_1core_1_1_sign.html#aa7296045907015b4e0ae8a93e5e6e295',1,'mlx::core::Sign::vmap()'],['../classmlx_1_1core_1_1_sin.html#a45533996f3d72d9dd97d4c61cd684fba',1,'mlx::core::Sin::vmap()'],['../classmlx_1_1core_1_1_sinh.html#ae171df22bc34c32e31b8135dc4caa788',1,'mlx::core::Sinh::vmap()'],['../classmlx_1_1core_1_1_slice.html#ae33583b0db22fcfeae34dfe1c0e3eaa2',1,'mlx::core::Slice::vmap()'],['../classmlx_1_1core_1_1_slice_update.html#adbf1c76de6ab2f986758530d351d6fa3',1,'mlx::core::SliceUpdate::vmap()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a825a6d4d1499b287525462854b841ef2',1,'mlx::core::DynamicSlice::vmap()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a750fb3548d8f3a5c6f4e54958649936f',1,'mlx::core::DynamicSliceUpdate::vmap()'],['../classmlx_1_1core_1_1_softmax.html#ad22d3dcc71054d3dba529cf2dc981e19',1,'mlx::core::Softmax::vmap()'],['../classmlx_1_1core_1_1_sort.html#abfabb9e625cc0cb9335c7454ed27505c',1,'mlx::core::Sort::vmap()'],['../classmlx_1_1core_1_1_split.html#ab7c40e02a842e83bdb4698608472c7a6',1,'mlx::core::Split::vmap()'],['../classmlx_1_1core_1_1_square.html#a55bf43f878d4741c57a08d5fef472ea5',1,'mlx::core::Square::vmap()'],['../classmlx_1_1core_1_1_sqrt.html#a9d30e306ce08980c27d98c898577017e',1,'mlx::core::Sqrt::vmap()'],['../classmlx_1_1core_1_1_stop_gradient.html#aca680c8befef81da414c4375b11b16b0',1,'mlx::core::StopGradient::vmap()'],['../classmlx_1_1core_1_1_subtract.html#aa98f960e621a767c8a03624fd292f098',1,'mlx::core::Subtract::vmap()'],['../classmlx_1_1core_1_1_squeeze.html#aa098a5850741bfb621800c7badce3532',1,'mlx::core::Squeeze::vmap()'],['../classmlx_1_1core_1_1_tan.html#ae2f67ca2adc83b10009cf28498bf58b7',1,'mlx::core::Tan::vmap()'],['../classmlx_1_1core_1_1_tanh.html#a32df3564c1ecb858c1ba9f855376762f',1,'mlx::core::Tanh::vmap()'],['../classmlx_1_1core_1_1_unflatten.html#a0f6ee31b99aca962d887c856414813fe',1,'mlx::core::Unflatten::vmap()'],['../classmlx_1_1core_1_1_view.html#a2230d3e5f434fb2b888de50b529ac121',1,'mlx::core::View::vmap()'],['../classmlx_1_1core_1_1_transpose.html#a5ef848b69def9a246665b67e6e3ffdfe',1,'mlx::core::Transpose::vmap()'],['../classmlx_1_1core_1_1_s_v_d.html#a0366c958f6cdac8d1d9e1a4eda53fae8',1,'mlx::core::SVD::vmap()'],['../classmlx_1_1core_1_1_inverse.html#a98419b9f0b8a6c9185fe012d523552c2',1,'mlx::core::Inverse::vmap()'],['../classmlx_1_1core_1_1_cholesky.html#ab5c3f6199ec3b399c91243a05d116aa5',1,'mlx::core::Cholesky::vmap()'],['../classmlx_1_1core_1_1_eigh.html#ab2f2ea5326e2f6045f9b7250692c240f',1,'mlx::core::Eigh::vmap()'],['../namespacemlx_1_1core.html#ac3caec2fa65375ed4c3bf1206177b84c',1,'mlx::core::vmap(const std::function< array(const array &)> &fun, int in_axis=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a8481a3bb4c12c2b7dc6ba576c2be3d0d',1,'mlx::core::vmap(const std::function< array(const array &, const array &)> &fun, int in_axis_a=0, int in_axis_b=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a95a7757e8d18fced38acfc6a3e8d686a',1,'mlx::core::vmap(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &in_axes={}, const std::vector< int > &out_axes={})']]], - ['vmap_5freplace_20',['vmap_replace',['../namespacemlx_1_1core_1_1detail.html#a31a5582530faea230eb8acafc0f7e154',1,'mlx::core::detail']]], - ['vmap_5ftrace_21',['vmap_trace',['../namespacemlx_1_1core_1_1detail.html#a5ba794afe1a557e0505887cfb481c515',1,'mlx::core::detail']]], - ['void_5ft_22',['void_t',['../namespacemetal.html#a192322c772aa8b168d59edc55fb806f1',1,'metal']]], - ['vtype_23',['VTYPE',['../structpocketfft_1_1detail_1_1_v_t_y_p_e.html',1,'pocketfft::detail']]], - ['vtype_5ft_24',['vtype_t',['../namespacepocketfft_1_1detail.html#a3edfb93aeed2f8258183d463ea291d62',1,'pocketfft::detail']]] + ['vectorscalar_15',['VectorScalar',['../structmlx_1_1core_1_1_vector_scalar.html',1,'mlx::core::VectorScalar< Op >'],['../structmlx_1_1core_1_1_vector_scalar.html#a97088143e6d301d753dcdd1ccdd82287',1,'mlx::core::VectorScalar::VectorScalar()'],['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6a8a94416459b638cebf3bfbce26a6ce78',1,'mlx::core::VectorScalar']]], + ['vectorvector_16',['VectorVector',['../structmlx_1_1core_1_1_vector_vector.html',1,'mlx::core::VectorVector< Op >'],['../structmlx_1_1core_1_1_vector_vector.html#a4867666c95c597a113afb64f173cc022',1,'mlx::core::VectorVector::VectorVector()'],['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6a4d8269410dcd9cadc9722e9a118bddfb',1,'mlx::core::VectorVector']]], + ['vectorvectorvector_17',['VectorVectorVector',['../namespacemlx_1_1core.html#ac2b8997537c7f25dd2b244d4c0a865a1acbcaeeb0e232871afe48bcf063a14b42',1,'mlx::core']]], + ['view_18',['View',['../classmlx_1_1core_1_1_view.html',1,'mlx::core::View'],['../classmlx_1_1core_1_1_view.html#ad7eed156c308e9a29a8b41f965ec941e',1,'mlx::core::View::View()']]], + ['view_19',['view',['../group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02',1,'mlx::core']]], + ['vjp_20',['vjp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abbf6d1d63dcda207ad7d9eeb4fc36225',1,'mlx::core::distributed::AllReduce::vjp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#aa5eff6fc128b71220899aab8ab9116fb',1,'mlx::core::distributed::AllGather::vjp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a74be4bcd0382f7f6400bf73fd5569c91',1,'mlx::core::fast::Custom::vjp()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#aacfbbbc15fcee0a5ce4f519ca3cca5eb',1,'mlx::core::fast::RMSNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#ae5e1b5df0705a6b1d141691a4396b0b6',1,'mlx::core::fast::LayerNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#ad999105414badd66c8fd9e069454a533',1,'mlx::core::fast::RoPE::vjp()'],['../classmlx_1_1core_1_1_primitive.html#a1dcb6807326eeab62474c6a0e3836d42',1,'mlx::core::Primitive::vjp()'],['../classmlx_1_1core_1_1_abs.html#aa2dd8ec0989e716b77394ac349b34592',1,'mlx::core::Abs::vjp()'],['../classmlx_1_1core_1_1_add.html#ac28e581862880e24ed2b99bb6a916607',1,'mlx::core::Add::vjp()'],['../classmlx_1_1core_1_1_add_m_m.html#ac1562a37cec6928e01281926ebeb47c6',1,'mlx::core::AddMM::vjp()'],['../classmlx_1_1core_1_1_arc_cos.html#a78e73e5e639d1249c7fe9614bf157c92',1,'mlx::core::ArcCos::vjp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a856c677f16e2b3f2edd2491e35db2d26',1,'mlx::core::ArcCosh::vjp()'],['../classmlx_1_1core_1_1_arc_sin.html#ab4057cd5ef1a8359f97493018e10d3a1',1,'mlx::core::ArcSin::vjp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a7988ee5b9e1e7e498dcab73d61ba147e',1,'mlx::core::ArcSinh::vjp()'],['../classmlx_1_1core_1_1_arc_tan.html#a5fefc3634b96a67ff8ae011a8ee180c2',1,'mlx::core::ArcTan::vjp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a99840c282e37b2b2a9c312e6e8ade1d2',1,'mlx::core::ArcTan2::vjp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a07da5797f7aaf3dfe43bf24e8562ac72',1,'mlx::core::ArcTanh::vjp()'],['../classmlx_1_1core_1_1_arg_partition.html#ade23d014717a0b0235d00073503aeac0',1,'mlx::core::ArgPartition::vjp()'],['../classmlx_1_1core_1_1_arg_reduce.html#a60d272685a373e6fe879416481a1ce1a',1,'mlx::core::ArgReduce::vjp()'],['../classmlx_1_1core_1_1_as_type.html#ac38a4f889311a3b5e5be9a67dcb93e18',1,'mlx::core::AsType::vjp()'],['../classmlx_1_1core_1_1_as_strided.html#a34783284c9b2f5b4a62c3c3ee5dd4062',1,'mlx::core::AsStrided::vjp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6131ed1c317ff8700a3e9b13fdaa9d61',1,'mlx::core::BitwiseBinary::vjp()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a1adf20087ee2f685bf39c2724b8e7120',1,'mlx::core::BlockMaskedMM::vjp()'],['../classmlx_1_1core_1_1_gather_m_m.html#a76c9f27c57354f6230b43944882e1bda',1,'mlx::core::GatherMM::vjp()'],['../classmlx_1_1core_1_1_broadcast_axes.html#aea8ef2b2616568a2bb56695381a035be',1,'mlx::core::BroadcastAxes::vjp()'],['../classmlx_1_1core_1_1_broadcast.html#a0318847c9be40f00b23907ad56037d18',1,'mlx::core::Broadcast::vjp()'],['../classmlx_1_1core_1_1_ceil.html#ac2f5a2bd84b8f013e5ce688419a88acb',1,'mlx::core::Ceil::vjp()'],['../classmlx_1_1core_1_1_compiled.html#a32462e65c52f84b708188130cc508133',1,'mlx::core::Compiled::vjp()'],['../classmlx_1_1core_1_1_concatenate.html#a8155db9100ec3b8bd0bc94baeaeee3b0',1,'mlx::core::Concatenate::vjp()'],['../classmlx_1_1core_1_1_contiguous.html#abf488f02057fd5852f38b2e8a600ad2a',1,'mlx::core::Contiguous::vjp()'],['../classmlx_1_1core_1_1_convolution.html#af8eb9c0c055ad20aa74b547016917690',1,'mlx::core::Convolution::vjp()'],['../classmlx_1_1core_1_1_copy.html#a6c4dee582001e9983e9517485ee37efd',1,'mlx::core::Copy::vjp()'],['../classmlx_1_1core_1_1_cos.html#a51d84113728e651ef9d4a1fe671c4d00',1,'mlx::core::Cos::vjp()'],['../classmlx_1_1core_1_1_cosh.html#a0791abd4305a333fb3b181a5357ce0f4',1,'mlx::core::Cosh::vjp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa1da36cef632df767cd9809d6cf06209',1,'mlx::core::CustomTransforms::vjp()'],['../classmlx_1_1core_1_1_depends.html#a02996fa45f01f7cb9f37074d5f8ccab0',1,'mlx::core::Depends::vjp()'],['../classmlx_1_1core_1_1_divide.html#ad3af7c70cad22c1a1a75b4a78ef793b6',1,'mlx::core::Divide::vjp()'],['../classmlx_1_1core_1_1_div_mod.html#a8c914a07f666a1d9377a27ed5d55e7c1',1,'mlx::core::DivMod::vjp()'],['../classmlx_1_1core_1_1_select.html#a9b522487b78fceeca7f827cd1c29a9a3',1,'mlx::core::Select::vjp()'],['../classmlx_1_1core_1_1_remainder.html#ab18f7bca1027ae71847a50da0933cec6',1,'mlx::core::Remainder::vjp()'],['../classmlx_1_1core_1_1_equal.html#af3c1bfcd1bf50922fc00e302bb193736',1,'mlx::core::Equal::vjp()'],['../classmlx_1_1core_1_1_erf.html#a1f529e95a42a2d69a8b18979d3ee2909',1,'mlx::core::Erf::vjp()'],['../classmlx_1_1core_1_1_erf_inv.html#a48afff12a58ddefae7ae0245c3580189',1,'mlx::core::ErfInv::vjp()'],['../classmlx_1_1core_1_1_exp.html#a94b9b7d137c3640d290b96c5e8b7e1a8',1,'mlx::core::Exp::vjp()'],['../classmlx_1_1core_1_1_expm1.html#af6ce416169190479c9792bb9cdbe2f43',1,'mlx::core::Expm1::vjp()'],['../classmlx_1_1core_1_1_expand_dims.html#a2fb3c65ba7a3b2d1f33a3c681fda8896',1,'mlx::core::ExpandDims::vjp()'],['../classmlx_1_1core_1_1_f_f_t.html#aafc895614a6e368c0e6d64af20d01090',1,'mlx::core::FFT::vjp()'],['../classmlx_1_1core_1_1_flatten.html#ab549a8c38b63055e2d5cd672f7676aab',1,'mlx::core::Flatten::vjp()'],['../classmlx_1_1core_1_1_floor.html#a589e2cf99b6fd1a5ba85534a2a31338e',1,'mlx::core::Floor::vjp()'],['../classmlx_1_1core_1_1_full.html#a49e76e7a8641f990701abc1b3bd49969',1,'mlx::core::Full::vjp()'],['../classmlx_1_1core_1_1_gather.html#aacf612a8f5f1cdbbfd19707d8d33c426',1,'mlx::core::Gather::vjp()'],['../classmlx_1_1core_1_1_gather_axis.html#a9c73b4ebed01bbdbaa316eddb6b5606d',1,'mlx::core::GatherAxis::vjp()'],['../classmlx_1_1core_1_1_greater.html#a341766a8a7e41d2a1160d35d4e781679',1,'mlx::core::Greater::vjp()'],['../classmlx_1_1core_1_1_greater_equal.html#a62f07a4ac54c708307c82aac0e5693ee',1,'mlx::core::GreaterEqual::vjp()'],['../classmlx_1_1core_1_1_hadamard.html#af4134775427b8998d66f489468b98656',1,'mlx::core::Hadamard::vjp()'],['../classmlx_1_1core_1_1_imag.html#a80da5fdd0fa549eebd7804c0e261848b',1,'mlx::core::Imag::vjp()'],['../classmlx_1_1core_1_1_less.html#aaf205d389b5e602e0814b68f66de8f50',1,'mlx::core::Less::vjp()'],['../classmlx_1_1core_1_1_less_equal.html#aab2aab7590c299885e815c18eedd1028',1,'mlx::core::LessEqual::vjp()'],['../classmlx_1_1core_1_1_log.html#a40885dccfbf928c4d035881be1d49280',1,'mlx::core::Log::vjp()'],['../classmlx_1_1core_1_1_log1p.html#a3113c1d2b4c5e73d0b470f42dc48a880',1,'mlx::core::Log1p::vjp()'],['../classmlx_1_1core_1_1_logical_not.html#af2c3c241cf3910fbaba013c69d052a50',1,'mlx::core::LogicalNot::vjp()'],['../classmlx_1_1core_1_1_logical_and.html#ae42f8fc454577b0fd6410cae9d5f3b54',1,'mlx::core::LogicalAnd::vjp()'],['../classmlx_1_1core_1_1_logical_or.html#a51aed488f52d5031998689af9cb17847',1,'mlx::core::LogicalOr::vjp()'],['../classmlx_1_1core_1_1_log_add_exp.html#ae231af0ed24a93eb647ee58c2d2b20b4',1,'mlx::core::LogAddExp::vjp()'],['../classmlx_1_1core_1_1_matmul.html#a524136cca481598ea20894d85ca66bb0',1,'mlx::core::Matmul::vjp()'],['../classmlx_1_1core_1_1_maximum.html#a7de15d7b28784e24bbfc7e85ddcbcff3',1,'mlx::core::Maximum::vjp()'],['../classmlx_1_1core_1_1_minimum.html#a48a0cbe3a6c4f7473c00e343f63b5204',1,'mlx::core::Minimum::vjp()'],['../classmlx_1_1core_1_1_multiply.html#a74b7556ec03e2c3d3f971666d06f5db1',1,'mlx::core::Multiply::vjp()'],['../classmlx_1_1core_1_1_negative.html#a889585f056d33bda30c30311257af52a',1,'mlx::core::Negative::vjp()'],['../classmlx_1_1core_1_1_not_equal.html#a0361f29f4ae1235bdf3f3304527e2d4b',1,'mlx::core::NotEqual::vjp()'],['../classmlx_1_1core_1_1_pad.html#ad8a7e547644f2717a24322968e971038',1,'mlx::core::Pad::vjp()'],['../classmlx_1_1core_1_1_partition.html#a7110772b6cd2d430a2b825cf5c952ca9',1,'mlx::core::Partition::vjp()'],['../classmlx_1_1core_1_1_power.html#a1453bb8307d6ff33134f1e00263bf082',1,'mlx::core::Power::vjp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#acb975e272b4a88ab232ef7f7c3a2bf26',1,'mlx::core::QuantizedMatmul::vjp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#ae08a4b7d28902d46f39e66beeb0e23ab',1,'mlx::core::GatherQMM::vjp()'],['../classmlx_1_1core_1_1_real.html#a29f6109339c5141a862ceae72c8b80fe',1,'mlx::core::Real::vjp()'],['../classmlx_1_1core_1_1_reshape.html#ab17294ecc6b5d4e89626fb48c7516365',1,'mlx::core::Reshape::vjp()'],['../classmlx_1_1core_1_1_reduce.html#a684883d2a96315f548ca769510e28e4e',1,'mlx::core::Reduce::vjp()'],['../classmlx_1_1core_1_1_round.html#af8f085e08b7fa8840c52a20b12ca35ce',1,'mlx::core::Round::vjp()'],['../classmlx_1_1core_1_1_scan.html#aaf13f72620b4b5d6a20e1228930e848e',1,'mlx::core::Scan::vjp()'],['../classmlx_1_1core_1_1_scatter.html#a0b51287fba789bb139ed61d40a0c636a',1,'mlx::core::Scatter::vjp()'],['../classmlx_1_1core_1_1_scatter_axis.html#a450f97b0be61a2bdfbfef4b2eb7cd198',1,'mlx::core::ScatterAxis::vjp()'],['../classmlx_1_1core_1_1_sigmoid.html#aac2f56a4c8362e36a28e232758ca52cf',1,'mlx::core::Sigmoid::vjp()'],['../classmlx_1_1core_1_1_sign.html#aa60ac52edd739fbdf388a997acd01bce',1,'mlx::core::Sign::vjp()'],['../classmlx_1_1core_1_1_sin.html#aedefe550ab4b0687858981bc0bcfbfa0',1,'mlx::core::Sin::vjp()'],['../classmlx_1_1core_1_1_sinh.html#a6b39fdd429bbb4de389e7c904fd561f0',1,'mlx::core::Sinh::vjp()'],['../classmlx_1_1core_1_1_slice.html#a291746a527ff991b66249fb2b54b685f',1,'mlx::core::Slice::vjp()'],['../classmlx_1_1core_1_1_slice_update.html#aedcdc60a0477997a96306c02b66d3f77',1,'mlx::core::SliceUpdate::vjp()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a29caf03256945f7732a52d551191f8fa',1,'mlx::core::DynamicSlice::vjp()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ab2817cb9d1bfcd3de6454d841909da1f',1,'mlx::core::DynamicSliceUpdate::vjp()'],['../classmlx_1_1core_1_1_softmax.html#abb68c311c45ee422a7c966accde9041b',1,'mlx::core::Softmax::vjp()'],['../classmlx_1_1core_1_1_sort.html#a3a8900dce53ee4eb7a1b83806e629358',1,'mlx::core::Sort::vjp()'],['../classmlx_1_1core_1_1_split.html#a7e8730f9cffa9872fff6f8d577031674',1,'mlx::core::Split::vjp()'],['../classmlx_1_1core_1_1_square.html#abcd9516da7f02dc906368c23b0bca263',1,'mlx::core::Square::vjp()'],['../classmlx_1_1core_1_1_sqrt.html#a08a21bd2c3a016f042d95aca294e68f3',1,'mlx::core::Sqrt::vjp()'],['../classmlx_1_1core_1_1_subtract.html#a3a3322be7c3bcaa0397cf099091df16b',1,'mlx::core::Subtract::vjp()'],['../classmlx_1_1core_1_1_squeeze.html#a8d95a13d7cc5586d48a38e9199180d06',1,'mlx::core::Squeeze::vjp()'],['../classmlx_1_1core_1_1_tan.html#a4639836cff03d73c769387d6943e92d7',1,'mlx::core::Tan::vjp()'],['../classmlx_1_1core_1_1_tanh.html#afe7b05e2b36b99c3a1b66f0cd3544e95',1,'mlx::core::Tanh::vjp()'],['../classmlx_1_1core_1_1_unflatten.html#a34f1218fa1d0e28f3ee10b65e6b0e319',1,'mlx::core::Unflatten::vjp()'],['../classmlx_1_1core_1_1_transpose.html#ac7805aa29b34afdf8852554f1e759f80',1,'mlx::core::Transpose::vjp()'],['../namespacemlx_1_1core.html#a1b33e2c2e3471420490cf0be2de6de18',1,'mlx::core::vjp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &cotangents)'],['../namespacemlx_1_1core.html#a2065a11249c3f4356ffd69b7a8c487ff',1,'mlx::core::vjp(const std::function< array(const array &)> &fun, const array &primal, const array &cotangent)']]], + ['vlen_21',['VLEN',['../structpocketfft_1_1detail_1_1_v_l_e_n.html',1,'pocketfft::detail']]], + ['vmap_22',['vmap',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a3f2dc71859847ca675ec4bfbe125035a',1,'mlx::core::distributed::AllReduce::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ad532d1d51f089dec3c84799b724ea031',1,'mlx::core::distributed::AllGather::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a5cfb66191b9e8b86649da77af55b0f93',1,'mlx::core::distributed::Send::vmap()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a7f4c3a4c48c6807faa36fb31e39dad8d',1,'mlx::core::fast::Custom::vmap()'],['../classmlx_1_1core_1_1_primitive.html#ac632b9619dd7a6a0f177bd36202e8103',1,'mlx::core::Primitive::vmap()'],['../classmlx_1_1core_1_1_abs.html#a4c9c98f1d71432fd3752ad9a6a8e7f2f',1,'mlx::core::Abs::vmap()'],['../classmlx_1_1core_1_1_add.html#a0e557d4d896153f84a25532562e4c646',1,'mlx::core::Add::vmap()'],['../classmlx_1_1core_1_1_add_m_m.html#a73ce80b3a37ec2523943028d50ebce81',1,'mlx::core::AddMM::vmap()'],['../classmlx_1_1core_1_1_arc_cos.html#a7548e23ace6827674aa6d284d44ccf83',1,'mlx::core::ArcCos::vmap()'],['../classmlx_1_1core_1_1_arc_cosh.html#af8ff78e910a9e485a203e1d3347bd461',1,'mlx::core::ArcCosh::vmap()'],['../classmlx_1_1core_1_1_arc_sin.html#a7cabb1e5a2bda44944378822c671ec82',1,'mlx::core::ArcSin::vmap()'],['../classmlx_1_1core_1_1_arc_sinh.html#a9e72b9751939387c333b5d4e19a37f6d',1,'mlx::core::ArcSinh::vmap()'],['../classmlx_1_1core_1_1_arc_tan.html#a1fb921554544a56498bc54f82e4a0556',1,'mlx::core::ArcTan::vmap()'],['../classmlx_1_1core_1_1_arc_tan2.html#ae02cb9fbf25e93dc1d7fbc9e3fb28634',1,'mlx::core::ArcTan2::vmap()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6ddcae68873559211cb91e7740dfc040',1,'mlx::core::ArcTanh::vmap()'],['../classmlx_1_1core_1_1_arg_partition.html#a441093795bcc31495ab5fbc9957b740a',1,'mlx::core::ArgPartition::vmap()'],['../classmlx_1_1core_1_1_arg_reduce.html#abfec42fa06ea15edaf393593751fb1ba',1,'mlx::core::ArgReduce::vmap()'],['../classmlx_1_1core_1_1_arg_sort.html#a3522bbbe4626a467394c1a8a9d7ac34e',1,'mlx::core::ArgSort::vmap()'],['../classmlx_1_1core_1_1_as_type.html#a7ebaf86fd6cad4a1ecfd7cde1ee0b0cc',1,'mlx::core::AsType::vmap()'],['../classmlx_1_1core_1_1_bitwise_binary.html#aa10be55f05bc1868bf4b375dc475f965',1,'mlx::core::BitwiseBinary::vmap()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a4e04f564d440e2d312c335db52c308e1',1,'mlx::core::BroadcastAxes::vmap()'],['../classmlx_1_1core_1_1_broadcast.html#aee4c71c2588ad01eb57e10f346cd666f',1,'mlx::core::Broadcast::vmap()'],['../classmlx_1_1core_1_1_ceil.html#ae86819990b43bdb0c2b3a25719b3a7a4',1,'mlx::core::Ceil::vmap()'],['../classmlx_1_1core_1_1_compiled.html#a732e7548f53977b4513bb7f30a04c30d',1,'mlx::core::Compiled::vmap()'],['../classmlx_1_1core_1_1_concatenate.html#a58c54dcf8e4b045d25edd3afc2caffc1',1,'mlx::core::Concatenate::vmap()'],['../classmlx_1_1core_1_1_conjugate.html#a2c7632c8ae0ca07777e23a0a79344e60',1,'mlx::core::Conjugate::vmap()'],['../classmlx_1_1core_1_1_contiguous.html#a563221e90b15aa90bfae23d29c10e4ec',1,'mlx::core::Contiguous::vmap()'],['../classmlx_1_1core_1_1_copy.html#a669b10253c15b769d90058d1ad7d0e61',1,'mlx::core::Copy::vmap()'],['../classmlx_1_1core_1_1_cos.html#aec9460daf0131156734013d03b230cd6',1,'mlx::core::Cos::vmap()'],['../classmlx_1_1core_1_1_cosh.html#a1ab2386e7d96219b6e4a525f7dac0406',1,'mlx::core::Cosh::vmap()'],['../classmlx_1_1core_1_1_custom_transforms.html#a906a2ff30d9c5281fbf1fa927e4c021b',1,'mlx::core::CustomTransforms::vmap()'],['../classmlx_1_1core_1_1_divide.html#a83e7da52831165b3a026e97b63770242',1,'mlx::core::Divide::vmap()'],['../classmlx_1_1core_1_1_div_mod.html#ae709e0fdd83994bd1d156e0d0e6a7942',1,'mlx::core::DivMod::vmap()'],['../classmlx_1_1core_1_1_select.html#a84e80361c8cf02536b4b98098793550f',1,'mlx::core::Select::vmap()'],['../classmlx_1_1core_1_1_remainder.html#a79867e1099a2e3c2d3e87407b2ab6e3d',1,'mlx::core::Remainder::vmap()'],['../classmlx_1_1core_1_1_equal.html#aea9cc3c88924ac824d72c39c2e83b0ca',1,'mlx::core::Equal::vmap()'],['../classmlx_1_1core_1_1_erf.html#abe554f553356654a3e800ba368108aaa',1,'mlx::core::Erf::vmap()'],['../classmlx_1_1core_1_1_erf_inv.html#ad5d7634e8568af8cc4a54a558a48d0e9',1,'mlx::core::ErfInv::vmap()'],['../classmlx_1_1core_1_1_exp.html#a0fcd579fe148b4c3dbc72e514b81bb37',1,'mlx::core::Exp::vmap()'],['../classmlx_1_1core_1_1_expm1.html#aa4caa848b2ea97e71ee3dd33de039296',1,'mlx::core::Expm1::vmap()'],['../classmlx_1_1core_1_1_expand_dims.html#a380c9ddc25a1f973c3d71b42f8a19486',1,'mlx::core::ExpandDims::vmap()'],['../classmlx_1_1core_1_1_f_f_t.html#ac32d6cc9b67289124f855ea68a61ede1',1,'mlx::core::FFT::vmap()'],['../classmlx_1_1core_1_1_flatten.html#a244a03915313286d36ed4d36b01a99f2',1,'mlx::core::Flatten::vmap()'],['../classmlx_1_1core_1_1_floor.html#aea4dc79a65774990e775ad49519a5d10',1,'mlx::core::Floor::vmap()'],['../classmlx_1_1core_1_1_full.html#afc57ab6bd9ebdbbf042af54a59785d95',1,'mlx::core::Full::vmap()'],['../classmlx_1_1core_1_1_gather.html#abab0c4c204e66489825ce80d2194a275',1,'mlx::core::Gather::vmap()'],['../classmlx_1_1core_1_1_gather_axis.html#a48d50bad33b69e29f75bedc794f7b785',1,'mlx::core::GatherAxis::vmap()'],['../classmlx_1_1core_1_1_greater.html#a6d8267411fc4951de781f9e8e6c53aa0',1,'mlx::core::Greater::vmap()'],['../classmlx_1_1core_1_1_greater_equal.html#ab0e1be93eb01b0ce7fa83e953f5e3e1d',1,'mlx::core::GreaterEqual::vmap()'],['../classmlx_1_1core_1_1_hadamard.html#a9f1a172e6246859e813002abe9b8f99c',1,'mlx::core::Hadamard::vmap()'],['../classmlx_1_1core_1_1_imag.html#ace9906672bd88df0573653883d58ecb3',1,'mlx::core::Imag::vmap()'],['../classmlx_1_1core_1_1_less.html#a5fee5956cf087d8405359121aa62ba7e',1,'mlx::core::Less::vmap()'],['../classmlx_1_1core_1_1_less_equal.html#a3d5df21db184f2b7620cda9da1684480',1,'mlx::core::LessEqual::vmap()'],['../classmlx_1_1core_1_1_log.html#a007ddbcf911093231f607a8b9ed5cd49',1,'mlx::core::Log::vmap()'],['../classmlx_1_1core_1_1_log1p.html#a7122576f95ce479926bbbbc690891f71',1,'mlx::core::Log1p::vmap()'],['../classmlx_1_1core_1_1_logical_not.html#a5308a271619ee74df561b0aaf525915d',1,'mlx::core::LogicalNot::vmap()'],['../classmlx_1_1core_1_1_logical_and.html#aacc5f6f53ffc327b7771485e3da2a4e5',1,'mlx::core::LogicalAnd::vmap()'],['../classmlx_1_1core_1_1_logical_or.html#a6e2e77e6aaf47872b2e96b151c32daf3',1,'mlx::core::LogicalOr::vmap()'],['../classmlx_1_1core_1_1_log_add_exp.html#a82190aa1421a9734b6e9480debffac78',1,'mlx::core::LogAddExp::vmap()'],['../classmlx_1_1core_1_1_matmul.html#a3a1c6e70bac300240760fe41a58340c2',1,'mlx::core::Matmul::vmap()'],['../classmlx_1_1core_1_1_maximum.html#ab664918e0d71cfec1318a9879e78c5d3',1,'mlx::core::Maximum::vmap()'],['../classmlx_1_1core_1_1_minimum.html#adab0f31acf68075a0be908d8eb882980',1,'mlx::core::Minimum::vmap()'],['../classmlx_1_1core_1_1_multiply.html#ae7e82c8fc8cbaf4e00c27eb54fac7dbf',1,'mlx::core::Multiply::vmap()'],['../classmlx_1_1core_1_1_negative.html#a1f8a6079e272f1a0599f88a1a8419cf0',1,'mlx::core::Negative::vmap()'],['../classmlx_1_1core_1_1_not_equal.html#ab8b57932f03c8eee664bf89adeaa43b5',1,'mlx::core::NotEqual::vmap()'],['../classmlx_1_1core_1_1_number_of_elements.html#a977d83eae845b8bd8c0b98b48cb1c6c2',1,'mlx::core::NumberOfElements::vmap()'],['../classmlx_1_1core_1_1_pad.html#a85658812a0f3275ba3eb74b7c75686cf',1,'mlx::core::Pad::vmap()'],['../classmlx_1_1core_1_1_partition.html#aa0cc55e4d4d2cb5d129d32832321df2c',1,'mlx::core::Partition::vmap()'],['../classmlx_1_1core_1_1_power.html#a5e22749592413a9adbdc877b03b87c8f',1,'mlx::core::Power::vmap()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a3434394140177b285f971c9ffe7e8763',1,'mlx::core::QuantizedMatmul::vmap()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a13ce5e138ebddb8780a034452f68892f',1,'mlx::core::GatherQMM::vmap()'],['../classmlx_1_1core_1_1_random_bits.html#a0dc12f053c6492f934bc18031412c415',1,'mlx::core::RandomBits::vmap()'],['../classmlx_1_1core_1_1_real.html#a07fbbefb6a1bc1ebd3985b24c36693b6',1,'mlx::core::Real::vmap()'],['../classmlx_1_1core_1_1_reshape.html#ae239dd3c6cab147e4af572dc58204f9d',1,'mlx::core::Reshape::vmap()'],['../classmlx_1_1core_1_1_reduce.html#abab1b5aa01ccad44f213f510c3596b38',1,'mlx::core::Reduce::vmap()'],['../classmlx_1_1core_1_1_round.html#a6fad8799a7982e1ccbe05be7cc38a7fd',1,'mlx::core::Round::vmap()'],['../classmlx_1_1core_1_1_scan.html#a297c7cc89c9bf9d186ebdebb634c7804',1,'mlx::core::Scan::vmap()'],['../classmlx_1_1core_1_1_scatter.html#a696c38b373a7a7c71bc112bd1117e322',1,'mlx::core::Scatter::vmap()'],['../classmlx_1_1core_1_1_scatter_axis.html#ae78709d1be122618f210ff595d888df8',1,'mlx::core::ScatterAxis::vmap()'],['../classmlx_1_1core_1_1_sigmoid.html#a12712c23037e38192cbccd2d4b14cc85',1,'mlx::core::Sigmoid::vmap()'],['../classmlx_1_1core_1_1_sign.html#aa7296045907015b4e0ae8a93e5e6e295',1,'mlx::core::Sign::vmap()'],['../classmlx_1_1core_1_1_sin.html#a45533996f3d72d9dd97d4c61cd684fba',1,'mlx::core::Sin::vmap()'],['../classmlx_1_1core_1_1_sinh.html#ae171df22bc34c32e31b8135dc4caa788',1,'mlx::core::Sinh::vmap()'],['../classmlx_1_1core_1_1_slice.html#ae33583b0db22fcfeae34dfe1c0e3eaa2',1,'mlx::core::Slice::vmap()'],['../classmlx_1_1core_1_1_slice_update.html#adbf1c76de6ab2f986758530d351d6fa3',1,'mlx::core::SliceUpdate::vmap()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a825a6d4d1499b287525462854b841ef2',1,'mlx::core::DynamicSlice::vmap()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a750fb3548d8f3a5c6f4e54958649936f',1,'mlx::core::DynamicSliceUpdate::vmap()'],['../classmlx_1_1core_1_1_softmax.html#ad22d3dcc71054d3dba529cf2dc981e19',1,'mlx::core::Softmax::vmap()'],['../classmlx_1_1core_1_1_sort.html#abfabb9e625cc0cb9335c7454ed27505c',1,'mlx::core::Sort::vmap()'],['../classmlx_1_1core_1_1_split.html#ab7c40e02a842e83bdb4698608472c7a6',1,'mlx::core::Split::vmap()'],['../classmlx_1_1core_1_1_square.html#a55bf43f878d4741c57a08d5fef472ea5',1,'mlx::core::Square::vmap()'],['../classmlx_1_1core_1_1_sqrt.html#a9d30e306ce08980c27d98c898577017e',1,'mlx::core::Sqrt::vmap()'],['../classmlx_1_1core_1_1_stop_gradient.html#aca680c8befef81da414c4375b11b16b0',1,'mlx::core::StopGradient::vmap()'],['../classmlx_1_1core_1_1_subtract.html#aa98f960e621a767c8a03624fd292f098',1,'mlx::core::Subtract::vmap()'],['../classmlx_1_1core_1_1_squeeze.html#aa098a5850741bfb621800c7badce3532',1,'mlx::core::Squeeze::vmap()'],['../classmlx_1_1core_1_1_tan.html#ae2f67ca2adc83b10009cf28498bf58b7',1,'mlx::core::Tan::vmap()'],['../classmlx_1_1core_1_1_tanh.html#a32df3564c1ecb858c1ba9f855376762f',1,'mlx::core::Tanh::vmap()'],['../classmlx_1_1core_1_1_unflatten.html#a0f6ee31b99aca962d887c856414813fe',1,'mlx::core::Unflatten::vmap()'],['../classmlx_1_1core_1_1_view.html#a2230d3e5f434fb2b888de50b529ac121',1,'mlx::core::View::vmap()'],['../classmlx_1_1core_1_1_transpose.html#a5ef848b69def9a246665b67e6e3ffdfe',1,'mlx::core::Transpose::vmap()'],['../classmlx_1_1core_1_1_s_v_d.html#a0366c958f6cdac8d1d9e1a4eda53fae8',1,'mlx::core::SVD::vmap()'],['../classmlx_1_1core_1_1_inverse.html#a98419b9f0b8a6c9185fe012d523552c2',1,'mlx::core::Inverse::vmap()'],['../classmlx_1_1core_1_1_cholesky.html#ab5c3f6199ec3b399c91243a05d116aa5',1,'mlx::core::Cholesky::vmap()'],['../classmlx_1_1core_1_1_eigh.html#ab2f2ea5326e2f6045f9b7250692c240f',1,'mlx::core::Eigh::vmap()'],['../namespacemlx_1_1core.html#ac3caec2fa65375ed4c3bf1206177b84c',1,'mlx::core::vmap(const std::function< array(const array &)> &fun, int in_axis=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a8481a3bb4c12c2b7dc6ba576c2be3d0d',1,'mlx::core::vmap(const std::function< array(const array &, const array &)> &fun, int in_axis_a=0, int in_axis_b=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a95a7757e8d18fced38acfc6a3e8d686a',1,'mlx::core::vmap(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &in_axes={}, const std::vector< int > &out_axes={})']]], + ['vmap_5freplace_23',['vmap_replace',['../namespacemlx_1_1core_1_1detail.html#a31a5582530faea230eb8acafc0f7e154',1,'mlx::core::detail']]], + ['vmap_5ftrace_24',['vmap_trace',['../namespacemlx_1_1core_1_1detail.html#a5ba794afe1a557e0505887cfb481c515',1,'mlx::core::detail']]], + ['void_5ft_25',['void_t',['../namespacemetal.html#a192322c772aa8b168d59edc55fb806f1',1,'metal']]], + ['vtype_26',['VTYPE',['../structpocketfft_1_1detail_1_1_v_t_y_p_e.html',1,'pocketfft::detail']]], + ['vtype_5ft_27',['vtype_t',['../namespacepocketfft_1_1detail.html#a3edfb93aeed2f8258183d463ea291d62',1,'pocketfft::detail']]] ]; diff --git a/docs/build/html/search/all_17.js b/docs/build/html/search/all_17.js index 21f0622e5..dc9a5fe0a 100644 --- a/docs/build/html/search/all_17.js +++ b/docs/build/html/search/all_17.js @@ -1,20 +1,21 @@ var searchData= [ - ['wait_0',['wait',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#af503189cc9247047fbdfc3ebf1daacc1',1,'pocketfft::detail::threading::latch::wait()'],['../classmlx_1_1core_1_1array.html#a648592006f1c92287734ba2428eaa45e',1,'mlx::core::array::wait()'],['../classmlx_1_1core_1_1_event.html#a634afd918e6ed847f354531ba9f48252',1,'mlx::core::Event::wait()']]], + ['wait_0',['wait',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#af503189cc9247047fbdfc3ebf1daacc1',1,'pocketfft::detail::threading::latch::wait()'],['../classmlx_1_1core_1_1array.html#a648592006f1c92287734ba2428eaa45e',1,'mlx::core::array::wait()'],['../classmlx_1_1core_1_1_fence.html#a1ccbe354d043e6c4d76286c6635a38e2',1,'mlx::core::Fence::wait()'],['../classmlx_1_1core_1_1_event.html#a634afd918e6ed847f354531ba9f48252',1,'mlx::core::Event::wait()']]], ['wait_5ffor_5ffence_1',['wait_for_fence',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aefdadbff4e003dc6f77506840babc088',1,'mlx::core::metal::CommandEncoder::wait_for_fence()'],['../structmlx_1_1core_1_1_command_encoder.html#aefdadbff4e003dc6f77506840babc088',1,'mlx::core::CommandEncoder::wait_for_fence()']]], ['wait_5ffor_5fone_2',['wait_for_one',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a01c574bb388f10d67aaaaa541894d807',1,'mlx::core::scheduler::Scheduler::wait_for_one()'],['../namespacemlx_1_1core_1_1scheduler.html#a8cc4d5fd1f5ce722b377ead1863a2291',1,'mlx::core::scheduler::wait_for_one()']]], - ['weight_5fbase_3',['weight_base',['../structmlx_1_1steel_1_1_conv2_d_general_base_info.html#a1d88677c4617f4bdae157e40a64a407b',1,'mlx::steel::Conv2DGeneralBaseInfo']]], - ['weight_5fh_4',['weight_h',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a3be4815d4090cb27ebe2f9bad1a39e95',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::weight_h'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a366c3cee4ed1165545287c8d5ce49445',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::weight_h'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a397412909eb955babc935a35d97c3fd4',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::weight_h'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a5997fd8ef249e4cd3df7dad7b251d8d5',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::weight_h']]], - ['weight_5fhw_5',['weight_hw',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ae39d43f741c9c87cce9c6d3144dc8b94',1,'mlx::steel::Conv2DWeightBlockLoader::weight_hw'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a7dd320bc5b0a9a2e425d6b292ddac037',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::weight_hw'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a5752e0309a4dc873cb31ce724c11ada6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::weight_hw']]], - ['weight_5fsize_6',['weight_size',['../structmlx_1_1steel_1_1_conv2_d_general_base_info.html#aff119a4325b97fdbd745d8fcaed9f041',1,'mlx::steel::Conv2DGeneralBaseInfo']]], - ['weight_5fw_7',['weight_w',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#add1186c7accb62bfa8a4a7e87fc4cc84',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::weight_w'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a4744bd79fb05e81eaa53d2eabe017446',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::weight_w'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a0261d0349a0a95ca1a02a959b73e9352',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::weight_w'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6efa6268a37f18f4d225674bf1780cf6',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::weight_w']]], - ['where_8',['where',['../group__ops.html#ga8a2056f8c9bb30914c40bcf509386491',1,'mlx::core']]], - ['write_9',['write',['../struct_read_writer.html#ac2ea71e41740ddc863890e3e8e6f09d0',1,'ReadWriter::write()'],['../classmlx_1_1core_1_1io_1_1_writer.html#ad9515b7f007338674de1e124cf77e125',1,'mlx::core::io::Writer::write()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#abca32838c9886f734d93430c34c07d7f',1,'mlx::core::io::FileWriter::write()'],['../struct_read_writer.html#a7a3d1396b0f83aa7506207bd6e7336bf',1,'ReadWriter::write() const'],['../struct_read_writer.html#ae1f0d3555b74998cc2d2288bce72a1f4',1,'ReadWriter::write() const']]], - ['write_5fpadded_10',['write_padded',['../struct_read_writer.html#a95367307acace2aa88226cf8956d2d88',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#abaf2a6ad4c88bd9f65fe1db1f73a8d87',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#a420453a56e77d6b3891ed4b5f178af9c',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const']]], - ['write_5fsafe_11',['write_safe',['../scan_8h.html#ae86aef08e5ebc8790031eb51eefa754c',1,'scan.h']]], - ['write_5fstrided_12',['write_strided',['../struct_read_writer.html#a77a4d7eac217305e22a3c25b3756ef67',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a12e7f43cd9de2d9990054184c0a32839',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a959ccaa08f2999c50cea063b01e492e4',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a5592b24dad5ad030a1e4769b0a278f35',1,'ReadWriter::write_strided(int stride, int overall_n)']]], - ['write_5funsafe_13',['write_unsafe',['../scan_8h.html#a8010e7bdf7a72cbd35ce7cd7ecb08e32',1,'scan.h']]], - ['writer_14',['Writer',['../classmlx_1_1core_1_1io_1_1_writer.html',1,'mlx::core::io']]], - ['ws_15',['wS',['../struct_m_l_x_conv_params.html#aba2074189644b1b59567d018409277a9',1,'MLXConvParams']]], - ['wt_5fstrides_16',['wt_strides',['../struct_m_l_x_conv_params.html#aa5ec3cb7bccbb04d561be16498cf06c3',1,'MLXConvParams']]] + ['wait_5fgpu_3',['wait_gpu',['../classmlx_1_1core_1_1_fence.html#ab6d783dee02656ebb8ffcbbfa6de5b53',1,'mlx::core::Fence']]], + ['weight_5fbase_4',['weight_base',['../structmlx_1_1steel_1_1_conv2_d_general_base_info.html#a1d88677c4617f4bdae157e40a64a407b',1,'mlx::steel::Conv2DGeneralBaseInfo']]], + ['weight_5fh_5',['weight_h',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a3be4815d4090cb27ebe2f9bad1a39e95',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::weight_h'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a366c3cee4ed1165545287c8d5ce49445',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::weight_h'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a397412909eb955babc935a35d97c3fd4',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::weight_h'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a5997fd8ef249e4cd3df7dad7b251d8d5',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::weight_h']]], + ['weight_5fhw_6',['weight_hw',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ae39d43f741c9c87cce9c6d3144dc8b94',1,'mlx::steel::Conv2DWeightBlockLoader::weight_hw'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a7dd320bc5b0a9a2e425d6b292ddac037',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::weight_hw'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a5752e0309a4dc873cb31ce724c11ada6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::weight_hw']]], + ['weight_5fsize_7',['weight_size',['../structmlx_1_1steel_1_1_conv2_d_general_base_info.html#aff119a4325b97fdbd745d8fcaed9f041',1,'mlx::steel::Conv2DGeneralBaseInfo']]], + ['weight_5fw_8',['weight_w',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#add1186c7accb62bfa8a4a7e87fc4cc84',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::weight_w'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a4744bd79fb05e81eaa53d2eabe017446',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::weight_w'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a0261d0349a0a95ca1a02a959b73e9352',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::weight_w'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6efa6268a37f18f4d225674bf1780cf6',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::weight_w']]], + ['where_9',['where',['../group__ops.html#ga8a2056f8c9bb30914c40bcf509386491',1,'mlx::core']]], + ['write_10',['write',['../struct_read_writer.html#ac2ea71e41740ddc863890e3e8e6f09d0',1,'ReadWriter::write()'],['../classmlx_1_1core_1_1io_1_1_writer.html#ad9515b7f007338674de1e124cf77e125',1,'mlx::core::io::Writer::write()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#abca32838c9886f734d93430c34c07d7f',1,'mlx::core::io::FileWriter::write()'],['../struct_read_writer.html#a7a3d1396b0f83aa7506207bd6e7336bf',1,'ReadWriter::write() const'],['../struct_read_writer.html#ae1f0d3555b74998cc2d2288bce72a1f4',1,'ReadWriter::write() const']]], + ['write_5fpadded_11',['write_padded',['../struct_read_writer.html#a95367307acace2aa88226cf8956d2d88',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#abaf2a6ad4c88bd9f65fe1db1f73a8d87',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#a420453a56e77d6b3891ed4b5f178af9c',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const']]], + ['write_5fsafe_12',['write_safe',['../scan_8h.html#ae86aef08e5ebc8790031eb51eefa754c',1,'scan.h']]], + ['write_5fstrided_13',['write_strided',['../struct_read_writer.html#a77a4d7eac217305e22a3c25b3756ef67',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a12e7f43cd9de2d9990054184c0a32839',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a959ccaa08f2999c50cea063b01e492e4',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a5592b24dad5ad030a1e4769b0a278f35',1,'ReadWriter::write_strided(int stride, int overall_n)']]], + ['write_5funsafe_14',['write_unsafe',['../scan_8h.html#a8010e7bdf7a72cbd35ce7cd7ecb08e32',1,'scan.h']]], + ['writer_15',['Writer',['../classmlx_1_1core_1_1io_1_1_writer.html',1,'mlx::core::io']]], + ['ws_16',['wS',['../struct_m_l_x_conv_params.html#aba2074189644b1b59567d018409277a9',1,'MLXConvParams']]], + ['wt_5fstrides_17',['wt_strides',['../struct_m_l_x_conv_params.html#aa5ec3cb7bccbb04d561be16498cf06c3',1,'MLXConvParams']]] ]; diff --git a/docs/build/html/search/all_1a.js b/docs/build/html/search/all_1a.js index fc00a1651..bab955a35 100644 --- a/docs/build/html/search/all_1a.js +++ b/docs/build/html/search/all_1a.js @@ -10,17 +10,18 @@ var searchData= ['_7edevicestream_7',['~DeviceStream',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a1c4397732f64f5811381dd01e30e020e',1,'mlx::core::metal::DeviceStream']]], ['_7efence_8',['~Fence',['../structmlx_1_1core_1_1metal_1_1_fence.html#a4940c1aece13814af7727de9abb511f2',1,'mlx::core::metal::Fence']]], ['_7efilewriter_9',['~FileWriter',['../classmlx_1_1core_1_1io_1_1_file_writer.html#ac325f51cd22050b6359056290e8ef42c',1,'mlx::core::io::FileWriter']]], - ['_7eintracing_10',['~InTracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#a83d57d7fa63bcb0ff72080191d0f177a',1,'mlx::core::detail::InTracing']]], - ['_7eparallelfilereader_11',['~ParallelFileReader',['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#ae9e3fc1cc3e827dae4d3d107f6780817',1,'mlx::core::io::ParallelFileReader']]], - ['_7eprimitive_12',['~Primitive',['../classmlx_1_1core_1_1_primitive.html#a29f70eb2d3b7e6c5fe52779c03f03777',1,'mlx::core::Primitive']]], - ['_7ereader_13',['~Reader',['../classmlx_1_1core_1_1io_1_1_reader.html#a81cd4747d81311c87dc6753f2d0d8b16',1,'mlx::core::io::Reader']]], - ['_7eresidencyset_14',['~ResidencySet',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#a998f07776f489bea9a7dd3c290ea7a79',1,'mlx::core::metal::ResidencySet']]], - ['_7eretaingraph_15',['~RetainGraph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a6bd6dc2e1caf2f764f39856a72ff6cbc',1,'mlx::core::detail::RetainGraph']]], - ['_7escheduler_16',['~Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a6626c4a743a2b3004fc14042bc8b0edf',1,'mlx::core::scheduler::Scheduler']]], - ['_7estreamcontext_17',['~StreamContext',['../structmlx_1_1core_1_1_stream_context.html#ac5be1c576d22b3d0b0a6fcc7e6abe659',1,'mlx::core::StreamContext']]], - ['_7estreamthread_18',['~StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a71de50591388b6e2cc6c57827e1a1ad4',1,'mlx::core::scheduler::StreamThread']]], - ['_7ethread_5fpool_19',['~thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a68f71e79fa93379fcc5819e533b45472',1,'pocketfft::detail::threading::thread_pool']]], - ['_7ethreadpool_20',['~ThreadPool',['../class_thread_pool.html#a44d3d2ab618970605e684efc216655eb',1,'ThreadPool']]], - ['_7eunaryprimitive_21',['~UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html#ac0677ab99a5ca660ed6ab7902ea364de',1,'mlx::core::UnaryPrimitive']]], - ['_7ewriter_22',['~Writer',['../classmlx_1_1core_1_1io_1_1_writer.html#a0e42f93a64118e9f5ede54ffe1bda045',1,'mlx::core::io::Writer']]] + ['_7egroupimpl_10',['~GroupImpl',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a83f119b87210f53438c5ba675a78065e',1,'mlx::core::distributed::detail::GroupImpl']]], + ['_7eintracing_11',['~InTracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#a83d57d7fa63bcb0ff72080191d0f177a',1,'mlx::core::detail::InTracing']]], + ['_7eparallelfilereader_12',['~ParallelFileReader',['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#ae9e3fc1cc3e827dae4d3d107f6780817',1,'mlx::core::io::ParallelFileReader']]], + ['_7eprimitive_13',['~Primitive',['../classmlx_1_1core_1_1_primitive.html#a29f70eb2d3b7e6c5fe52779c03f03777',1,'mlx::core::Primitive']]], + ['_7ereader_14',['~Reader',['../classmlx_1_1core_1_1io_1_1_reader.html#a81cd4747d81311c87dc6753f2d0d8b16',1,'mlx::core::io::Reader']]], + ['_7eresidencyset_15',['~ResidencySet',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#a998f07776f489bea9a7dd3c290ea7a79',1,'mlx::core::metal::ResidencySet']]], + ['_7eretaingraph_16',['~RetainGraph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a6bd6dc2e1caf2f764f39856a72ff6cbc',1,'mlx::core::detail::RetainGraph']]], + ['_7escheduler_17',['~Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a6626c4a743a2b3004fc14042bc8b0edf',1,'mlx::core::scheduler::Scheduler']]], + ['_7estreamcontext_18',['~StreamContext',['../structmlx_1_1core_1_1_stream_context.html#ac5be1c576d22b3d0b0a6fcc7e6abe659',1,'mlx::core::StreamContext']]], + ['_7estreamthread_19',['~StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a71de50591388b6e2cc6c57827e1a1ad4',1,'mlx::core::scheduler::StreamThread']]], + ['_7ethread_5fpool_20',['~thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a68f71e79fa93379fcc5819e533b45472',1,'pocketfft::detail::threading::thread_pool']]], + ['_7ethreadpool_21',['~ThreadPool',['../class_thread_pool.html#a44d3d2ab618970605e684efc216655eb',1,'ThreadPool']]], + ['_7eunaryprimitive_22',['~UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html#ac0677ab99a5ca660ed6ab7902ea364de',1,'mlx::core::UnaryPrimitive']]], + ['_7ewriter_23',['~Writer',['../classmlx_1_1core_1_1io_1_1_writer.html#a0e42f93a64118e9f5ede54ffe1bda045',1,'mlx::core::io::Writer']]] ]; diff --git a/docs/build/html/search/all_2.js b/docs/build/html/search/all_2.js index 532f8efd9..fea2ee846 100644 --- a/docs/build/html/search/all_2.js +++ b/docs/build/html/search/all_2.js @@ -5,113 +5,119 @@ var searchData= ['b_5fstr_5fk_2',['B_str_k',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa71400922babd388177f228c2c82b211',1,'mlx::steel::BlockMMA']]], ['b_5fstr_5fn_3',['B_str_n',['../structmlx_1_1steel_1_1_block_m_m_a.html#a49538190209e522ddbef45fe95563d17',1,'mlx::steel::BlockMMA']]], ['backward_4',['BACKWARD',['../namespacepocketfft_1_1detail.html#a9d1eaa7469c018c39e745733eab9a9c3',1,'pocketfft::detail::BACKWARD'],['../namespacepocketfft.html#a9d1eaa7469c018c39e745733eab9a9c3',1,'pocketfft::BACKWARD']]], - ['base_5',['Base',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421',1,'mlx::core::Log']]], - ['base_5fwh_6',['base_wh',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aca37adba6f148579eb1cd0a7800a5cfe',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::base_wh'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6c46564bf1a96a02791dd432cc9c883e',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::base_wh']]], - ['base_5fww_7',['base_ww',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a32d020c6715d06f7de360877fcb7b6e4',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::base_ww'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a230f0e581f9b8227b9ee68760b3b1503',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::base_ww']]], - ['basemmafrag_8',['BaseMMAFrag',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]], - ['basemmafrag_3c_20accumtype_2c_20kfragsize_2c_20kfragsize_20_3e_9',['BaseMMAFrag< AccumType, kFragSize, kFragSize >',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]], - ['basemmafrag_3c_20t_2c_208_2c_208_20_3e_10',['BaseMMAFrag< T, 8, 8 >',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]], - ['batch_5fndim_11',['batch_ndim',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a640dc138a8bf7b2b5bed6a436b429c2f',1,'mlx::steel::GEMMParams']]], - ['batch_5fsize_12',['batch_size',['../struct_read_writer.html#a689f4890c1d2ce33fc6da7550beec735',1,'ReadWriter']]], - ['batch_5fstride_5fa_13',['batch_stride_a',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#aa9efc32581bba343b096123482507eee',1,'mlx::steel::GEMMParams']]], - ['batch_5fstride_5fb_14',['batch_stride_b',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a5fba5117664fbab81e22abf1e8c8fbc8',1,'mlx::steel::GEMMParams']]], - ['batch_5fstride_5fc_15',['batch_stride_c',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#ac977827a0de9650b7110df029324fd60',1,'mlx::steel::GEMMAddMMParams']]], - ['batch_5fstride_5fd_16',['batch_stride_d',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a4fc69951d02a9ce2d3e11b356557deb9',1,'mlx::steel::GEMMParams']]], - ['bcols_17',['BCOLS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a29fbeeacdf5b6feeb74815ced255fa5a',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ac3b40db720055350bba59d614ea1dd79',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a86519729ef0561686bb86e474c95b93d',1,'mlx::steel::Conv2DWeightBlockLoader::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9229d22e0a02d96825eb5a57c8cb95ac',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a8b6cf53a10514310d01f4d6459053a57',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3d6272d000f8ea79d9b3b5228bdca20f',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a1843921cd67926002bb0dcccf3048eb6',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BCOLS']]], - ['bcols_5fpacked_18',['BCOLS_PACKED',['../struct_quantized_block_loader.html#a1392a5278cf6e090ea80ebe7c4ac5fbb',1,'QuantizedBlockLoader']]], - ['begin_19',['begin',['../classmlx_1_1core_1_1array.html#a76b258b169d7d73419ebbf85340fb914',1,'mlx::core::array']]], - ['bernoulli_20',['bernoulli',['../namespacemlx_1_1core_1_1random.html#a1c601b637f60071dfc85cad19a841744',1,'mlx::core::random::bernoulli(const array &p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aaa49f6c2af5496822fa09435e54275cb',1,'mlx::core::random::bernoulli(const array &p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa9e360f9cb7bd23221352ed9e31d83c2',1,'mlx::core::random::bernoulli(T p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a2778876cd2318fec69cd1f3fc0955d68',1,'mlx::core::random::bernoulli(T p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#ad7eb4467e2f9d5f74a5607b29a935b6e',1,'mlx::core::random::bernoulli(const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['beta_21',['beta',['../structmlx_1_1steel_1_1_transform_axpby.html#a5fc726f085bafd1acbc391886f7fb8b6',1,'mlx::steel::TransformAxpby::beta'],['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#ac0ce4d8a6014f8adb29fd0a0bb23139f',1,'mlx::steel::GEMMAddMMParams::beta']]], - ['bf16_2eh_22',['bf16.h',['../backend_2metal_2kernels_2jit_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html',1,'(Global Namespace)'],['../types_2bf16_8h.html',1,'(Global Namespace)']]], - ['bf16_5fmath_2eh_23',['bf16_math.h',['../bf16__math_8h.html',1,'']]], - ['bfloat16_24',['bfloat16',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa444fe01f3a7a54d1809aef0912846a47',1,'mlx::core::Dtype::bfloat16'],['../namespacemlx_1_1core.html#a514cf8b4e6f0a6af3a867e752f4338f7',1,'mlx::core::bfloat16']]], - ['bfloat16_5ft_25',['bfloat16_t',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7782de82393104dd4ad754ce3b316e82',1,'bfloat16_t: bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a58e15a77da988b9104fee00cdf8b280e',1,'bfloat16_t: bf16.h'],['../namespacemlx_1_1core.html#acb5d16c9b83778c7621c38e522e0060b',1,'mlx::core::bfloat16_t']]], - ['bfloat16_5fto_5fuint16_26',['bfloat16_to_uint16',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h']]], - ['bfloat_5fbinop_27',['bfloat_binop',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7694892a131c0e31e5153c088cccb707',1,'bfloat_binop: bf16.h'],['../types_2bf16_8h.html#a7694892a131c0e31e5153c088cccb707',1,'bfloat_binop: bf16.h']]], - ['bfloat_5fbinop_5fbase_28',['bfloat_binop_base',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a78c92beda4436da9a2e520fa98c59f70',1,'bfloat_binop_base: bf16.h'],['../types_2bf16_8h.html#a78c92beda4436da9a2e520fa98c59f70',1,'bfloat_binop_base: bf16.h']]], - ['bfloat_5fbinop_5fhelper_29',['bfloat_binop_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac7ff36230dab17e8f17b7a7c80888594',1,'bfloat_binop_helper: bf16.h'],['../types_2bf16_8h.html#ac7ff36230dab17e8f17b7a7c80888594',1,'bfloat_binop_helper: bf16.h']]], - ['bfloat_5fbitop_30',['bfloat_bitop',['../types_2bf16_8h.html#aac9ba86d4bf05bcda1936494f9b9b4d3',1,'bf16.h']]], - ['bfloat_5fbits_5fto_5ffloat_31',['bfloat_bits_to_float',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3b33ae338dc4f223d0f3c748de07bad1',1,'bf16.h']]], - ['bfloat_5fcompop_32',['bfloat_compop',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a330a0883503cb640f1cf628a7ca50239',1,'bfloat_compop: bf16.h'],['../types_2bf16_8h.html#a330a0883503cb640f1cf628a7ca50239',1,'bfloat_compop: bf16.h']]], - ['bfloat_5finplace_5fbitop_33',['bfloat_inplace_bitop',['../types_2bf16_8h.html#af13b46bc58e6e6f675ae47aabec37711',1,'bf16.h']]], - ['bfloat_5finplace_5fop_34',['bfloat_inplace_op',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ac82467fbc674e990090f482b9c1e5c',1,'bfloat_inplace_op: bf16.h'],['../types_2bf16_8h.html#aee905053f51f76e0c1af94199714d514',1,'bfloat_inplace_op: bf16.h']]], - ['bfloat_5finplace_5fop_5faddr_5fspace_5fhelper_35',['bfloat_inplace_op_addr_space_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af30a2cbd2c3415516203b83bd21872f8',1,'bfloat_inplace_op_addr_space_helper: bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1457da931c28fa4e2500daa4e6441e8b',1,'bfloat_inplace_op_addr_space_helper: bf16.h']]], - ['bfloat_5finplace_5fop_5fhelper_36',['bfloat_inplace_op_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2846fd11b5e19b435e9f7ef0998c9b1d',1,'bfloat_inplace_op_helper: bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afe5988aa8147be2bafda6a5b7792fe15',1,'bfloat_inplace_op_helper: bf16.h']]], - ['bfloat_5for_5fbits_37',['bfloat_or_bits',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core::numeric_limits< bfloat16_t >']]], - ['bfs_5fmax_5fwidth_38',['bfs_max_width',['../namespacemlx_1_1core_1_1env.html#ac3266e1259a64c8b56bdc6c7029179f2',1,'mlx::core::env']]], - ['bi_39',['bi',['../struct_quantized_block_loader.html#a85041d72225a2095659c70509291a906',1,'QuantizedBlockLoader::bi'],['../structmlx_1_1steel_1_1_block_loader.html#a9ef13742bcdf07532d8f09394928a8af',1,'mlx::steel::BlockLoader::bi'],['../structmlx_1_1steel_1_1_block_loader_t.html#a6964273994b06d6cf8ef7e59fb10bb35',1,'mlx::steel::BlockLoaderT::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8e53b0a9951cb840d922cc285b257ee3',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ae3af75287f279d2cdeef189126740d4c',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::bi'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a8c5e74003600132954cb953616e1a026',1,'mlx::steel::Conv2DWeightBlockLoader::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9eb024e2fc6f07345f87fbf7141c0d16',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::bi'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae3b9f21f72e5e6c541c9978f55d354c7',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a32a3a91fa715b82f36e05ceb10933d09',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::bi'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a4c91f848856ab0872bdfd37c62d4b0ba',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::bi']]], - ['biases_40',['biases',['../struct_quantized_block_loader.html#a17d01a6aba0833b073586ef2c09d0fbd',1,'QuantizedBlockLoader']]], - ['binary_41',['binary',['../namespacemlx_1_1core_1_1metal.html#a269d591ec02e2f7c0f7a718fbfa37f73',1,'mlx::core::metal']]], - ['binary_2eh_42',['binary.h',['../common_2binary_8h.html',1,'(Global Namespace)'],['../metal_2binary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary_8h.html',1,'(Global Namespace)']]], - ['binary_5fg_43',['binary_g',['../metal_2kernels_2binary_8h.html#ab6b062acfb0497230e9476482d9dac20',1,'binary_g(device const T *a, device const T *b, device U *c, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a84dbd99589c7a18ff4d91056dc6fe17a',1,'binary_g(device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary_two.h']]], - ['binary_5fg_5fnd1_44',['binary_g_nd1',['../metal_2kernels_2binary_8h.html#aba4dd8bf59ed391789110e08ecfec6c2',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a1f2e8e7dc1998bab3864184878a75776',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary_two.h']]], - ['binary_5fg_5fnd2_45',['binary_g_nd2',['../metal_2kernels_2binary_8h.html#a73869f3771f16108f37120e485ce6e0b',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a0d31d8d6c8a1845d3329ef2e23e3fff6',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary_two.h']]], - ['binary_5fg_5fnd3_46',['binary_g_nd3',['../metal_2kernels_2binary_8h.html#ab6eb3c1f7349a52d5befff14e796320a',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#aae95e8f5fa772f780fc88054a2e35878',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary_two.h']]], - ['binary_5fop_5fgpu_47',['binary_op_gpu',['../namespacemlx_1_1core.html#ad884f4a36308b5b4f8a5d990d2e086df',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a094876ea5a2a2445ab64efc8222da202',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], - ['binary_5fop_5fgpu_5finplace_48',['binary_op_gpu_inplace',['../namespacemlx_1_1core.html#a8616c0b7b0fc118a75400bc86404c367',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a7e6af6624e322e7ad60a3873a66e18a3',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], - ['binary_5fops_49',['binary_ops',['../namespacemlx_1_1core_1_1metal.html#a8db7f9cc781d4bfb08423a401665f322',1,'mlx::core::metal']]], - ['binary_5fops_2eh_50',['binary_ops.h',['../binary__ops_8h.html',1,'']]], - ['binary_5fss_51',['binary_ss',['../metal_2kernels_2binary_8h.html#a242b8b29a852c255467e50628c6dccf5',1,'binary_ss(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#af8a791ac7ca88d32cd8f4e9ac0f9ab4f',1,'binary_ss(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fsv_52',['binary_sv',['../metal_2kernels_2binary_8h.html#a4116c35f2e4632366d1611d5a95ba141',1,'binary_sv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab18c6ecf5065275c93701efd095c916c',1,'binary_sv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fsv2_53',['binary_sv2',['../metal_2kernels_2binary_8h.html#aa8c48b1b21d8f5a181f5443de2346589',1,'binary_sv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a08822ff98ea6f61a98b49a9e9a38b891',1,'binary_sv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], - ['binary_5ftwo_54',['binary_two',['../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb',1,'mlx::core::metal']]], - ['binary_5ftwo_2eh_55',['binary_two.h',['../common_2binary__two_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__two_8h.html',1,'(Global Namespace)']]], - ['binary_5fvs_56',['binary_vs',['../metal_2kernels_2binary_8h.html#a649851d133358dd5832a73b1061b3313',1,'binary_vs(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca',1,'binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fvs2_57',['binary_vs2',['../metal_2kernels_2binary_8h.html#a48bd82eb10f9c623ce7d28daec4fa512',1,'binary_vs2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a273d2f31691f2c64623c2a97eab344be',1,'binary_vs2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], - ['binary_5fvv_58',['binary_vv',['../metal_2kernels_2binary_8h.html#add6a9aeee3cb0ba909574f27fa9ecd5b',1,'binary_vv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9',1,'binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fvv2_59',['binary_vv2',['../metal_2kernels_2binary_8h.html#a19dbbf8fea68b64bdd25dc8d36865171',1,'binary_vv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12e80730e43dfaa4c79ce8d5f99edc50',1,'binary_vv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], - ['bits_60',['bits',['../namespacemlx_1_1core_1_1random.html#ad7d1c0b530906538dd8fb31b17382f2b',1,'mlx::core::random::bits(const Shape &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a035d36774135faabad33d8f64a879df7',1,'mlx::core::random::bits(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['bits_5f_61',['bits_',['../struct___m_l_x___b_float16.html#a4113263b63e3757ea8334cc4f0f5c3c8',1,'_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aca48963f820065c3d8ecab24265ab3fc',1,'mlx::core::_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a5203fe52424fd32bce6eb7917dd9288b',1,'mlx::core::_MLX_Float16::bits_']]], - ['bits_5fto_5fbfloat_62',['bits_to_bfloat',['../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca',1,'_MLX_BFloat16']]], - ['bits_5fto_5fbfloat_5fstruct_63',['bits_to_bfloat_struct',['../struct___m_l_x___b_float16_1_1bits__to__bfloat__struct.html',1,'_MLX_BFloat16']]], - ['bitwise_5fand_64',['bitwise_and',['../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada',1,'mlx::core']]], - ['bitwise_5for_65',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]], - ['bitwise_5fxor_66',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]], - ['bitwiseand_67',['BitwiseAnd',['../struct_bitwise_and.html',1,'BitwiseAnd'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html',1,'mlx::core::detail::BitwiseAnd']]], - ['bitwisebinary_68',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html',1,'mlx::core::BitwiseBinary'],['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary::BitwiseBinary()']]], - ['bitwiseor_69',['BitwiseOr',['../struct_bitwise_or.html',1,'BitwiseOr'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html',1,'mlx::core::detail::BitwiseOr']]], - ['bitwisexor_70',['BitwiseXor',['../struct_bitwise_xor.html',1,'BitwiseXor'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html',1,'mlx::core::detail::BitwiseXor']]], - ['bj_71',['bj',['../struct_quantized_block_loader.html#ae2add92b2aaf3414e91f0470b9b0cc00',1,'QuantizedBlockLoader::bj'],['../structmlx_1_1steel_1_1_block_loader.html#a78c326e75ee35a484685771143047cd4',1,'mlx::steel::BlockLoader::bj'],['../structmlx_1_1steel_1_1_block_loader_t.html#aca83e49c31095badc8a46eb3c8e00957',1,'mlx::steel::BlockLoaderT::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a7ae9e41f50c0c63c35b63086a1c22cc3',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a6fd3dd7b74d91609fa9dd61c657a0e32',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a6f2fdcaf5a67567cca38ae3d8120ab37',1,'mlx::steel::Conv2DWeightBlockLoader::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a7cf448573d41fbc67f8dfc65b7aef2b2',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#adaa261fc2e8e694aedab4ebd60b52e5e',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#ace16704025bc6e6204c306a357f3a8b8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acec010e10d5733654963407af38d4f67',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::bj']]], - ['block_5fmasked_5fgemm_72',['block_masked_gemm',['../steel__gemm__masked_8h.html#a3642df61cba8eff47a08a20834137eac',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h'],['../steel__gemm__masked_8h.html#ac59086cee8cf396751da691fe80cda7b',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device bool *out_mask, const device bool *lhs_mask, const device bool *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h']]], - ['block_5fmasked_5fmm_73',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]], - ['block_5fmerge_5fsort_5ft_74',['block_merge_sort_t',['../struct_kernel_merge_sort.html#adae7850e057fc30d5328c7b3dcc998fa',1,'KernelMergeSort::block_merge_sort_t'],['../struct_kernel_multi_block_merge_sort.html#af27e9af4b58640c0aa620bc4efc68dff',1,'KernelMultiBlockMergeSort::block_merge_sort_t']]], - ['block_5fsort_75',['block_sort',['../struct_kernel_merge_sort.html#a56b644ec66f7fb5c01b280f124304be9',1,'KernelMergeSort::block_sort()'],['../struct_kernel_multi_block_merge_sort.html#a322ed2eac315a561e0fd90af2fd577eb',1,'KernelMultiBlockMergeSort::block_sort()'],['../sort_8h.html#a93f14092416169c4449141043ac45ffd',1,'block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid): sort.h']]], - ['block_5fsort_5fnc_76',['block_sort_nc',['../sort_8h.html#ad272ef4422c75565a79b7f5e751bb247',1,'sort.h']]], - ['blockloader_77',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)']]], - ['blockloader_3c_20t_2c_20transpose_5fa_20_3f_20bk_20_3abm_2c_20transpose_5fa_20_3f_20bm_20_3abk_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20_21transpose_5fa_2c_20tgp_5fsize_20_3e_78',['BlockLoader< T, transpose_a ? BK :BM, transpose_a ? BM :BK, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, !transpose_a, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], - ['blockloader_3c_20t_2c_20transpose_5fb_20_3f_20bn_20_3abk_2c_20transpose_5fb_20_3f_20bk_20_3abn_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20transpose_5fb_2c_20tgp_5fsize_20_3e_79',['BlockLoader< T, transpose_b ? BN :BK, transpose_b ? BK :BN, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, transpose_b, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], - ['blockloadert_80',['BlockLoaderT',['../structmlx_1_1steel_1_1_block_loader_t.html',1,'mlx::steel::BlockLoaderT< T, BROWS, BCOLS, kDstStrRow, kDstStrCol, reduction_dim, tgp_size, n_reads, TCOLS, TROWS >'],['../structmlx_1_1steel_1_1_block_loader_t.html#a076616a7c67ad1b847e0e6b046077ee2',1,'mlx::steel::BlockLoaderT::BlockLoaderT()']]], - ['blockm_81',['blockM',['../struct_g_e_m_v_kernel.html#a7281520100658811076400060663903c',1,'GEMVKernel::blockM'],['../struct_g_e_m_v_t_kernel.html#a2ae8ce535d59cccf453381b4485a77f0',1,'GEMVTKernel::blockM']]], - ['blockmaskedmm_82',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html',1,'mlx::core::BlockMaskedMM'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM::BlockMaskedMM()']]], - ['blockmergesort_83',['BlockMergeSort',['../struct_block_merge_sort.html',1,'']]], - ['blockmma_84',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)']]], - ['blockmma_3c_20t_2c_20u_2c_20bm_2c_20bn_2c_20bk_2c_20wm_2c_20wn_2c_20transpose_5fa_2c_20transpose_5fb_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20accumtype_2c_20epilogue_20_3e_85',['BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, AccumType, Epilogue >',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel']]], - ['blockn_86',['blockN',['../struct_g_e_m_v_kernel.html#a2fef17f9c9aa0bdf530ad3554fb0988b',1,'GEMVKernel::blockN'],['../struct_g_e_m_v_t_kernel.html#a60be87666006ba0bf88bc8e6902da42a',1,'GEMVTKernel::blockN']]], - ['blockswizzle_87',['BlockSwizzle',['../structmlx_1_1steel_1_1_block_swizzle.html',1,'mlx::steel']]], - ['bluestein_5ffft_88',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]], - ['bool4_5for_5fuint_89',['bool4_or_uint',['../unionbool4__or__uint.html',1,'']]], - ['bool_5f_90',['bool_',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa467afb5838aa377d55cce81f84c5512b',1,'mlx::core::Dtype::bool_'],['../namespacemlx_1_1core.html#a113d2bac7e4aa6a4cb4a5c3242527b82',1,'mlx::core::bool_']]], - ['bool_5fconstant_91',['bool_constant',['../namespacemlx_1_1steel.html#adbb34bcf0d2dca6b9fb803d591d00da9',1,'mlx::steel']]], - ['broadcast_92',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html',1,'mlx::core::Broadcast'],['../classmlx_1_1core_1_1_broadcast.html#a004cce3029c0427569830016f99648cb',1,'mlx::core::Broadcast::Broadcast()']]], - ['broadcast_5farrays_93',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]], - ['broadcast_5fshapes_94',['broadcast_shapes',['../namespacemlx_1_1core.html#aefa6a2ec6439e9619cafd227a1dc14ab',1,'mlx::core']]], - ['broadcast_5fto_95',['broadcast_to',['../group__ops.html#ga2fd5891f11593b7f09550e884f969013',1,'mlx::core']]], - ['broadcastaxes_96',['BroadcastAxes',['../classmlx_1_1core_1_1_broadcast_axes.html',1,'mlx::core::BroadcastAxes'],['../classmlx_1_1core_1_1_broadcast_axes.html#a5136f33489670cdc0802e46725288195',1,'mlx::core::BroadcastAxes::BroadcastAxes()']]], - ['brows_97',['BROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ac070c6bd5be85b1ae805e18890db4fd4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a10591ea957605a9c662f93d59ff3410d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ae9b86b05b23153ea1abaeead456c491c',1,'mlx::steel::Conv2DWeightBlockLoader::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a343984fb74ec579a4404278dbbc7e7b5',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acc8140aae84694f62e6324dbb6a614a4',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aba1e1c8012e4e50f0e9bcfb9486c1781',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a015a0c56de74a0c4d51953a7e94fbba8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BROWS']]], - ['bs_5foffset_98',['Bs_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a92f6aeee432f53638447eac842f43eca',1,'mlx::steel::BlockMMA']]], - ['bs_5fqmm_5fn_99',['bs_qmm_n',['../quantized_8h.html#a323d2d70799c8d9dffe4b64a4285a799',1,'quantized.h']]], - ['bs_5fqmm_5ft_100',['bs_qmm_t',['../quantized_8h.html#a693057a0c311a73ac0759e005b9806c1',1,'quantized.h']]], - ['bs_5fqmv_101',['bs_qmv',['../quantized_8h.html#aaf4fb9c4318c5cd27d118004dbdeba61',1,'quantized.h']]], - ['bs_5fqmv_5ffast_102',['bs_qmv_fast',['../quantized_8h.html#a359282a9f71e487e5d86d246896ab33d',1,'quantized.h']]], - ['bs_5fqvm_103',['bs_qvm',['../quantized_8h.html#a2c53419ba5019d4722c0f4c2026b1142',1,'quantized.h']]], - ['btile_104',['Btile',['../structmlx_1_1steel_1_1_block_m_m_a.html#a44fca27c821764317263047a780977b0',1,'mlx::steel::BlockMMA']]], - ['buf_105',['buf',['../struct_read_writer.html#a23bac3c96dd0265ddbee1f256be45ff5',1,'ReadWriter']]], - ['buffer_106',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html',1,'mlx::core::allocator::Buffer'],['../classmlx_1_1core_1_1metal_1_1_buffer.html',1,'mlx::core::metal::Buffer'],['../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::allocator::Buffer::Buffer()'],['../classmlx_1_1core_1_1metal_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::metal::Buffer::Buffer()']]], - ['buffer_107',['buffer',['../structmlx_1_1core_1_1array_1_1_data.html#a9a51e2d12ba505027cc0fca86bdd39ad',1,'mlx::core::array::Data::buffer'],['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a99183c92599edfeb75f7fa0f37e1d9eb',1,'mlx::core::metal::DeviceStream::buffer'],['../classmlx_1_1core_1_1array.html#ab3daf04c27c4593d9d73c397b8484a08',1,'mlx::core::array::buffer()'],['../classmlx_1_1core_1_1array.html#a634466ce661485394f2fdc3bd6796bcd',1,'mlx::core::array::buffer() const']]], - ['buffer_5fops_108',['buffer_ops',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#ab6048b329e65a59033834f3bdd351782',1,'mlx::core::metal::DeviceStream']]], - ['buffer_5fsize_109',['buffer_size',['../classmlx_1_1core_1_1array.html#a914577c63755b2e862d2da68bbf8e3dd',1,'mlx::core::array']]], - ['buffers_110',['buffers',['../struct_indices.html#ad705070a740579c07d109ae4f3d86e76',1,'Indices']]], - ['build_5fcommand_111',['build_command',['../classmlx_1_1core_1_1_jit_compiler.html#a10a5cde91ab929ccbdbdf4c4d940f156',1,'mlx::core::JitCompiler']]], - ['build_5flib_5fname_112',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]], - ['bytes_5fper_5fpack_113',['bytes_per_pack',['../struct_quantized_block_loader.html#ad00fe6d8bd395206a41693a8ed65d4db',1,'QuantizedBlockLoader']]] + ['barrier_5',['barrier',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a7f028c6ca48e75bf2c1806b5b8cfc90e',1,'mlx::core::metal::CommandEncoder::barrier()'],['../structmlx_1_1core_1_1_command_encoder.html#a7f028c6ca48e75bf2c1806b5b8cfc90e',1,'mlx::core::CommandEncoder::barrier()']]], + ['base_6',['Base',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421',1,'mlx::core::Log']]], + ['base_5fsimd_2eh_7',['base_simd.h',['../base__simd_8h.html',1,'']]], + ['base_5fwh_8',['base_wh',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aca37adba6f148579eb1cd0a7800a5cfe',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::base_wh'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6c46564bf1a96a02791dd432cc9c883e',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::base_wh']]], + ['base_5fww_9',['base_ww',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a32d020c6715d06f7de360877fcb7b6e4',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::base_ww'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a230f0e581f9b8227b9ee68760b3b1503',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::base_ww']]], + ['basemmafrag_10',['BaseMMAFrag',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]], + ['basemmafrag_3c_20accumtype_2c_20kfragsize_2c_20kfragsize_20_3e_11',['BaseMMAFrag< AccumType, kFragSize, kFragSize >',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]], + ['basemmafrag_3c_20t_2c_208_2c_208_20_3e_12',['BaseMMAFrag< T, 8, 8 >',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html',1,'mlx::steel']]], + ['batch_5fndim_13',['batch_ndim',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a640dc138a8bf7b2b5bed6a436b429c2f',1,'mlx::steel::GEMMParams']]], + ['batch_5fsize_14',['batch_size',['../struct_read_writer.html#a689f4890c1d2ce33fc6da7550beec735',1,'ReadWriter']]], + ['batch_5fstride_5fa_15',['batch_stride_a',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#aa9efc32581bba343b096123482507eee',1,'mlx::steel::GEMMParams']]], + ['batch_5fstride_5fb_16',['batch_stride_b',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a5fba5117664fbab81e22abf1e8c8fbc8',1,'mlx::steel::GEMMParams']]], + ['batch_5fstride_5fc_17',['batch_stride_c',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#ac977827a0de9650b7110df029324fd60',1,'mlx::steel::GEMMAddMMParams']]], + ['batch_5fstride_5fd_18',['batch_stride_d',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a4fc69951d02a9ce2d3e11b356557deb9',1,'mlx::steel::GEMMParams']]], + ['bcols_19',['BCOLS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a29fbeeacdf5b6feeb74815ced255fa5a',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ac3b40db720055350bba59d614ea1dd79',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a86519729ef0561686bb86e474c95b93d',1,'mlx::steel::Conv2DWeightBlockLoader::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9229d22e0a02d96825eb5a57c8cb95ac',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a8b6cf53a10514310d01f4d6459053a57',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3d6272d000f8ea79d9b3b5228bdca20f',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a1843921cd67926002bb0dcccf3048eb6',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BCOLS']]], + ['bcols_5fpacked_20',['BCOLS_PACKED',['../struct_quantized_block_loader.html#a1392a5278cf6e090ea80ebe7c4ac5fbb',1,'QuantizedBlockLoader']]], + ['begin_21',['begin',['../classmlx_1_1core_1_1array.html#a76b258b169d7d73419ebbf85340fb914',1,'mlx::core::array']]], + ['bernoulli_22',['bernoulli',['../namespacemlx_1_1core_1_1random.html#a1c601b637f60071dfc85cad19a841744',1,'mlx::core::random::bernoulli(const array &p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aaa49f6c2af5496822fa09435e54275cb',1,'mlx::core::random::bernoulli(const array &p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa9e360f9cb7bd23221352ed9e31d83c2',1,'mlx::core::random::bernoulli(T p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a2778876cd2318fec69cd1f3fc0955d68',1,'mlx::core::random::bernoulli(T p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#ad7eb4467e2f9d5f74a5607b29a935b6e',1,'mlx::core::random::bernoulli(const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['beta_23',['beta',['../structmlx_1_1steel_1_1_transform_axpby.html#a5fc726f085bafd1acbc391886f7fb8b6',1,'mlx::steel::TransformAxpby::beta'],['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#ac0ce4d8a6014f8adb29fd0a0bb23139f',1,'mlx::steel::GEMMAddMMParams::beta']]], + ['bf16_2eh_24',['bf16.h',['../backend_2metal_2kernels_2jit_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html',1,'(Global Namespace)'],['../types_2bf16_8h.html',1,'(Global Namespace)']]], + ['bf16_5fmath_2eh_25',['bf16_math.h',['../bf16__math_8h.html',1,'']]], + ['bfloat16_26',['bfloat16',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa444fe01f3a7a54d1809aef0912846a47',1,'mlx::core::Dtype::bfloat16'],['../namespacemlx_1_1core.html#a514cf8b4e6f0a6af3a867e752f4338f7',1,'mlx::core::bfloat16']]], + ['bfloat16_5ft_27',['bfloat16_t',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7782de82393104dd4ad754ce3b316e82',1,'bfloat16_t: bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a58e15a77da988b9104fee00cdf8b280e',1,'bfloat16_t: bf16.h'],['../namespacemlx_1_1core.html#acb5d16c9b83778c7621c38e522e0060b',1,'mlx::core::bfloat16_t']]], + ['bfloat16_5fto_5fuint16_28',['bfloat16_to_uint16',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h']]], + ['bfloat_5fbinop_29',['bfloat_binop',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7694892a131c0e31e5153c088cccb707',1,'bfloat_binop: bf16.h'],['../types_2bf16_8h.html#a7694892a131c0e31e5153c088cccb707',1,'bfloat_binop: bf16.h']]], + ['bfloat_5fbinop_5fbase_30',['bfloat_binop_base',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a78c92beda4436da9a2e520fa98c59f70',1,'bfloat_binop_base: bf16.h'],['../types_2bf16_8h.html#a78c92beda4436da9a2e520fa98c59f70',1,'bfloat_binop_base: bf16.h']]], + ['bfloat_5fbinop_5fhelper_31',['bfloat_binop_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac7ff36230dab17e8f17b7a7c80888594',1,'bfloat_binop_helper: bf16.h'],['../types_2bf16_8h.html#ac7ff36230dab17e8f17b7a7c80888594',1,'bfloat_binop_helper: bf16.h']]], + ['bfloat_5fbitop_32',['bfloat_bitop',['../types_2bf16_8h.html#aac9ba86d4bf05bcda1936494f9b9b4d3',1,'bf16.h']]], + ['bfloat_5fbits_5fto_5ffloat_33',['bfloat_bits_to_float',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3b33ae338dc4f223d0f3c748de07bad1',1,'bf16.h']]], + ['bfloat_5fcompop_34',['bfloat_compop',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a330a0883503cb640f1cf628a7ca50239',1,'bfloat_compop: bf16.h'],['../types_2bf16_8h.html#a330a0883503cb640f1cf628a7ca50239',1,'bfloat_compop: bf16.h']]], + ['bfloat_5finplace_5fbitop_35',['bfloat_inplace_bitop',['../types_2bf16_8h.html#af13b46bc58e6e6f675ae47aabec37711',1,'bf16.h']]], + ['bfloat_5finplace_5fop_36',['bfloat_inplace_op',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ac82467fbc674e990090f482b9c1e5c',1,'bfloat_inplace_op: bf16.h'],['../types_2bf16_8h.html#aee905053f51f76e0c1af94199714d514',1,'bfloat_inplace_op: bf16.h']]], + ['bfloat_5finplace_5fop_5faddr_5fspace_5fhelper_37',['bfloat_inplace_op_addr_space_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af30a2cbd2c3415516203b83bd21872f8',1,'bfloat_inplace_op_addr_space_helper: bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1457da931c28fa4e2500daa4e6441e8b',1,'bfloat_inplace_op_addr_space_helper: bf16.h']]], + ['bfloat_5finplace_5fop_5fhelper_38',['bfloat_inplace_op_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2846fd11b5e19b435e9f7ef0998c9b1d',1,'bfloat_inplace_op_helper: bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afe5988aa8147be2bafda6a5b7792fe15',1,'bfloat_inplace_op_helper: bf16.h']]], + ['bfs_5fmax_5fwidth_39',['bfs_max_width',['../namespacemlx_1_1core_1_1env.html#ac3266e1259a64c8b56bdc6c7029179f2',1,'mlx::core::env']]], + ['bi_40',['bi',['../struct_quantized_block_loader.html#a85041d72225a2095659c70509291a906',1,'QuantizedBlockLoader::bi'],['../structmlx_1_1steel_1_1_block_loader.html#a9ef13742bcdf07532d8f09394928a8af',1,'mlx::steel::BlockLoader::bi'],['../structmlx_1_1steel_1_1_block_loader_t.html#a6964273994b06d6cf8ef7e59fb10bb35',1,'mlx::steel::BlockLoaderT::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8e53b0a9951cb840d922cc285b257ee3',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ae3af75287f279d2cdeef189126740d4c',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::bi'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a8c5e74003600132954cb953616e1a026',1,'mlx::steel::Conv2DWeightBlockLoader::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9eb024e2fc6f07345f87fbf7141c0d16',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::bi'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae3b9f21f72e5e6c541c9978f55d354c7',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::bi'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a32a3a91fa715b82f36e05ceb10933d09',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::bi'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a4c91f848856ab0872bdfd37c62d4b0ba',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::bi']]], + ['biases_41',['biases',['../struct_quantized_block_loader.html#a17d01a6aba0833b073586ef2c09d0fbd',1,'QuantizedBlockLoader']]], + ['binary_42',['binary',['../namespacemlx_1_1core.html#ae374861abd45cf019c3e6be2026f3798',1,'mlx::core::binary()'],['../namespacemlx_1_1core_1_1metal.html#a269d591ec02e2f7c0f7a718fbfa37f73',1,'mlx::core::metal::binary()']]], + ['binary_2eh_43',['binary.h',['../common_2binary_8h.html',1,'(Global Namespace)'],['../cpu_2binary_8h.html',1,'(Global Namespace)'],['../metal_2binary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary_8h.html',1,'(Global Namespace)']]], + ['binary_5fg_44',['binary_g',['../metal_2kernels_2binary_8h.html#ab6b062acfb0497230e9476482d9dac20',1,'binary_g(device const T *a, device const T *b, device U *c, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a84dbd99589c7a18ff4d91056dc6fe17a',1,'binary_g(device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary_two.h']]], + ['binary_5fg_5fnd1_45',['binary_g_nd1',['../metal_2kernels_2binary_8h.html#aba4dd8bf59ed391789110e08ecfec6c2',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a1f2e8e7dc1998bab3864184878a75776',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary_two.h']]], + ['binary_5fg_5fnd2_46',['binary_g_nd2',['../metal_2kernels_2binary_8h.html#a73869f3771f16108f37120e485ce6e0b',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a0d31d8d6c8a1845d3329ef2e23e3fff6',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5fg_5fnd3_47',['binary_g_nd3',['../metal_2kernels_2binary_8h.html#ab6eb3c1f7349a52d5befff14e796320a',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#aae95e8f5fa772f780fc88054a2e35878',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary_two.h']]], + ['binary_5fop_48',['binary_op',['../namespacemlx_1_1core.html#a9c1c1fdf9a0840a16a4d10a8f74f761d',1,'mlx::core::binary_op(const array &a, const array &b, array &out, Op op)'],['../namespacemlx_1_1core.html#a2aca3458c56605a74d07ec39876549bc',1,'mlx::core::binary_op(const array &a, const array &b, array &out, Op op)']]], + ['binary_5fop_5fdims_49',['binary_op_dims',['../namespacemlx_1_1core.html#a7ca09ebf776fe32db580f9038587ec31',1,'mlx::core']]], + ['binary_5fop_5fdispatch_5fdims_50',['binary_op_dispatch_dims',['../namespacemlx_1_1core.html#a66c9ee5018168b9101de52e0122d9755',1,'mlx::core']]], + ['binary_5fop_5fgpu_51',['binary_op_gpu',['../namespacemlx_1_1core.html#ad884f4a36308b5b4f8a5d990d2e086df',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a094876ea5a2a2445ab64efc8222da202',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], + ['binary_5fop_5fgpu_5finplace_52',['binary_op_gpu_inplace',['../namespacemlx_1_1core.html#a8616c0b7b0fc118a75400bc86404c367',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a7e6af6624e322e7ad60a3873a66e18a3',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], + ['binary_5fops_53',['binary_ops',['../namespacemlx_1_1core_1_1metal.html#a8db7f9cc781d4bfb08423a401665f322',1,'mlx::core::metal']]], + ['binary_5fops_2eh_54',['binary_ops.h',['../cpu_2binary__ops_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__ops_8h.html',1,'(Global Namespace)']]], + ['binary_5fsingle_55',['BINARY_SINGLE',['../cpu_2binary__ops_8h.html#ad606fe7b9586413e760cfe461c9be119',1,'binary_ops.h']]], + ['binary_5fss_56',['binary_ss',['../metal_2kernels_2binary_8h.html#a242b8b29a852c255467e50628c6dccf5',1,'binary_ss(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#af8a791ac7ca88d32cd8f4e9ac0f9ab4f',1,'binary_ss(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fsv_57',['binary_sv',['../metal_2kernels_2binary_8h.html#a4116c35f2e4632366d1611d5a95ba141',1,'binary_sv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab18c6ecf5065275c93701efd095c916c',1,'binary_sv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fsv2_58',['binary_sv2',['../metal_2kernels_2binary_8h.html#aa8c48b1b21d8f5a181f5443de2346589',1,'binary_sv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a08822ff98ea6f61a98b49a9e9a38b891',1,'binary_sv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5ftwo_59',['binary_two',['../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb',1,'mlx::core::metal']]], + ['binary_5ftwo_2eh_60',['binary_two.h',['../cpu_2binary__two_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__two_8h.html',1,'(Global Namespace)']]], + ['binary_5fvs_61',['binary_vs',['../metal_2kernels_2binary_8h.html#a649851d133358dd5832a73b1061b3313',1,'binary_vs(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca',1,'binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fvs2_62',['binary_vs2',['../metal_2kernels_2binary_8h.html#a48bd82eb10f9c623ce7d28daec4fa512',1,'binary_vs2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a273d2f31691f2c64623c2a97eab344be',1,'binary_vs2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5fvv_63',['binary_vv',['../metal_2kernels_2binary_8h.html#add6a9aeee3cb0ba909574f27fa9ecd5b',1,'binary_vv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9',1,'binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fvv2_64',['binary_vv2',['../metal_2kernels_2binary_8h.html#a19dbbf8fea68b64bdd25dc8d36865171',1,'binary_vv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12e80730e43dfaa4c79ce8d5f99edc50',1,'binary_vv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['binaryoptype_65',['BinaryOpType',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6',1,'mlx::core']]], + ['bits_66',['bits',['../namespacemlx_1_1core_1_1random.html#ad7d1c0b530906538dd8fb31b17382f2b',1,'mlx::core::random::bits(const Shape &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a035d36774135faabad33d8f64a879df7',1,'mlx::core::random::bits(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['bits_5f_67',['bits_',['../struct___m_l_x___b_float16.html#a4113263b63e3757ea8334cc4f0f5c3c8',1,'_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aca48963f820065c3d8ecab24265ab3fc',1,'mlx::core::_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a5203fe52424fd32bce6eb7917dd9288b',1,'mlx::core::_MLX_Float16::bits_']]], + ['bits_5fto_5fbfloat_68',['bits_to_bfloat',['../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca',1,'_MLX_BFloat16']]], + ['bits_5fto_5fbfloat_5fstruct_69',['bits_to_bfloat_struct',['../struct___m_l_x___b_float16_1_1bits__to__bfloat__struct.html',1,'_MLX_BFloat16']]], + ['bitwise_5fand_70',['bitwise_and',['../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada',1,'mlx::core']]], + ['bitwise_5for_71',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]], + ['bitwise_5fxor_72',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]], + ['bitwiseand_73',['BitwiseAnd',['../struct_bitwise_and.html',1,'BitwiseAnd'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html',1,'mlx::core::detail::BitwiseAnd']]], + ['bitwisebinary_74',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html',1,'mlx::core::BitwiseBinary'],['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary::BitwiseBinary()']]], + ['bitwiseor_75',['BitwiseOr',['../struct_bitwise_or.html',1,'BitwiseOr'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html',1,'mlx::core::detail::BitwiseOr']]], + ['bitwisexor_76',['BitwiseXor',['../struct_bitwise_xor.html',1,'BitwiseXor'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html',1,'mlx::core::detail::BitwiseXor']]], + ['bj_77',['bj',['../struct_quantized_block_loader.html#ae2add92b2aaf3414e91f0470b9b0cc00',1,'QuantizedBlockLoader::bj'],['../structmlx_1_1steel_1_1_block_loader.html#a78c326e75ee35a484685771143047cd4',1,'mlx::steel::BlockLoader::bj'],['../structmlx_1_1steel_1_1_block_loader_t.html#aca83e49c31095badc8a46eb3c8e00957',1,'mlx::steel::BlockLoaderT::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a7ae9e41f50c0c63c35b63086a1c22cc3',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a6fd3dd7b74d91609fa9dd61c657a0e32',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a6f2fdcaf5a67567cca38ae3d8120ab37',1,'mlx::steel::Conv2DWeightBlockLoader::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a7cf448573d41fbc67f8dfc65b7aef2b2',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#adaa261fc2e8e694aedab4ebd60b52e5e',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#ace16704025bc6e6204c306a357f3a8b8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acec010e10d5733654963407af38d4f67',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::bj']]], + ['block_5fmasked_5fgemm_78',['block_masked_gemm',['../steel__gemm__masked_8h.html#a3642df61cba8eff47a08a20834137eac',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h'],['../steel__gemm__masked_8h.html#ac59086cee8cf396751da691fe80cda7b',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device bool *out_mask, const device bool *lhs_mask, const device bool *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h']]], + ['block_5fmasked_5fmm_79',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]], + ['block_5fmerge_5fsort_5ft_80',['block_merge_sort_t',['../struct_kernel_merge_sort.html#a37b94702630a4f97254d1da847d420c0',1,'KernelMergeSort::block_merge_sort_t'],['../struct_kernel_multi_block_merge_sort.html#aaf08efb1c540b276b0f1bbdeb514544d',1,'KernelMultiBlockMergeSort::block_merge_sort_t']]], + ['block_5fsort_81',['block_sort',['../struct_kernel_merge_sort.html#a90102e73c3f6e9652e9870d57b129e71',1,'KernelMergeSort::block_sort()'],['../struct_kernel_multi_block_merge_sort.html#a56b3d8bf726e5619325dd07752666f8a',1,'KernelMultiBlockMergeSort::block_sort()'],['../sort_8h.html#a93f14092416169c4449141043ac45ffd',1,'block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid): sort.h']]], + ['block_5fsort_5fnc_82',['block_sort_nc',['../sort_8h.html#ad272ef4422c75565a79b7f5e751bb247',1,'sort.h']]], + ['blockloader_83',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)']]], + ['blockloader_3c_20t_2c_20transpose_5fa_20_3f_20bk_20_3abm_2c_20transpose_5fa_20_3f_20bm_20_3abk_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20_21transpose_5fa_2c_20tgp_5fsize_20_3e_84',['BlockLoader< T, transpose_a ? BK :BM, transpose_a ? BM :BK, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, !transpose_a, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], + ['blockloader_3c_20t_2c_20transpose_5fb_20_3f_20bn_20_3abk_2c_20transpose_5fb_20_3f_20bk_20_3abn_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20transpose_5fb_2c_20tgp_5fsize_20_3e_85',['BlockLoader< T, transpose_b ? BN :BK, transpose_b ? BK :BN, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, transpose_b, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], + ['blockloadert_86',['BlockLoaderT',['../structmlx_1_1steel_1_1_block_loader_t.html',1,'mlx::steel::BlockLoaderT< T, BROWS, BCOLS, kDstStrRow, kDstStrCol, reduction_dim, tgp_size, n_reads, TCOLS, TROWS >'],['../structmlx_1_1steel_1_1_block_loader_t.html#a076616a7c67ad1b847e0e6b046077ee2',1,'mlx::steel::BlockLoaderT::BlockLoaderT()']]], + ['blockm_87',['blockM',['../struct_g_e_m_v_kernel.html#a7281520100658811076400060663903c',1,'GEMVKernel::blockM'],['../struct_g_e_m_v_t_kernel.html#a2ae8ce535d59cccf453381b4485a77f0',1,'GEMVTKernel::blockM']]], + ['blockmaskedmm_88',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html',1,'mlx::core::BlockMaskedMM'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM::BlockMaskedMM()']]], + ['blockmergesort_89',['BlockMergeSort',['../struct_block_merge_sort.html',1,'']]], + ['blockmma_90',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)']]], + ['blockmma_3c_20t_2c_20u_2c_20bm_2c_20bn_2c_20bk_2c_20wm_2c_20wn_2c_20transpose_5fa_2c_20transpose_5fb_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20accumtype_2c_20epilogue_20_3e_91',['BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, AccumType, Epilogue >',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel']]], + ['blockn_92',['blockN',['../struct_g_e_m_v_kernel.html#a2fef17f9c9aa0bdf530ad3554fb0988b',1,'GEMVKernel::blockN'],['../struct_g_e_m_v_t_kernel.html#a60be87666006ba0bf88bc8e6902da42a',1,'GEMVTKernel::blockN']]], + ['blockswizzle_93',['BlockSwizzle',['../structmlx_1_1steel_1_1_block_swizzle.html',1,'mlx::steel']]], + ['bluestein_5ffft_94',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]], + ['bool4_5for_5fuint_95',['bool4_or_uint',['../unionbool4__or__uint.html',1,'']]], + ['bool_5f_96',['bool_',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa467afb5838aa377d55cce81f84c5512b',1,'mlx::core::Dtype::bool_'],['../namespacemlx_1_1core.html#a113d2bac7e4aa6a4cb4a5c3242527b82',1,'mlx::core::bool_']]], + ['bool_5fconstant_97',['bool_constant',['../namespacemlx_1_1steel.html#adbb34bcf0d2dca6b9fb803d591d00da9',1,'mlx::steel']]], + ['broadcast_98',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html',1,'mlx::core::Broadcast'],['../classmlx_1_1core_1_1_broadcast.html#a004cce3029c0427569830016f99648cb',1,'mlx::core::Broadcast::Broadcast()']]], + ['broadcast_5farrays_99',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]], + ['broadcast_5fshapes_100',['broadcast_shapes',['../namespacemlx_1_1core.html#aefa6a2ec6439e9619cafd227a1dc14ab',1,'mlx::core']]], + ['broadcast_5fto_101',['broadcast_to',['../group__ops.html#ga2fd5891f11593b7f09550e884f969013',1,'mlx::core']]], + ['broadcastaxes_102',['BroadcastAxes',['../classmlx_1_1core_1_1_broadcast_axes.html',1,'mlx::core::BroadcastAxes'],['../classmlx_1_1core_1_1_broadcast_axes.html#a5136f33489670cdc0802e46725288195',1,'mlx::core::BroadcastAxes::BroadcastAxes()']]], + ['brows_103',['BROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ac070c6bd5be85b1ae805e18890db4fd4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a10591ea957605a9c662f93d59ff3410d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ae9b86b05b23153ea1abaeead456c491c',1,'mlx::steel::Conv2DWeightBlockLoader::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a343984fb74ec579a4404278dbbc7e7b5',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acc8140aae84694f62e6324dbb6a614a4',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aba1e1c8012e4e50f0e9bcfb9486c1781',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a015a0c56de74a0c4d51953a7e94fbba8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BROWS']]], + ['bs_5foffset_104',['Bs_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a92f6aeee432f53638447eac842f43eca',1,'mlx::steel::BlockMMA']]], + ['bs_5fqmm_5fn_105',['bs_qmm_n',['../quantized_8h.html#a323d2d70799c8d9dffe4b64a4285a799',1,'quantized.h']]], + ['bs_5fqmm_5ft_106',['bs_qmm_t',['../quantized_8h.html#a693057a0c311a73ac0759e005b9806c1',1,'quantized.h']]], + ['bs_5fqmv_107',['bs_qmv',['../quantized_8h.html#aaf4fb9c4318c5cd27d118004dbdeba61',1,'quantized.h']]], + ['bs_5fqmv_5ffast_108',['bs_qmv_fast',['../quantized_8h.html#a359282a9f71e487e5d86d246896ab33d',1,'quantized.h']]], + ['bs_5fqvm_109',['bs_qvm',['../quantized_8h.html#a2c53419ba5019d4722c0f4c2026b1142',1,'quantized.h']]], + ['btile_110',['Btile',['../structmlx_1_1steel_1_1_block_m_m_a.html#a44fca27c821764317263047a780977b0',1,'mlx::steel::BlockMMA']]], + ['buf_111',['buf',['../struct_read_writer.html#a23bac3c96dd0265ddbee1f256be45ff5',1,'ReadWriter']]], + ['buffer_112',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html',1,'mlx::core::allocator::Buffer'],['../classmlx_1_1core_1_1metal_1_1_buffer.html',1,'mlx::core::metal::Buffer'],['../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::allocator::Buffer::Buffer()'],['../classmlx_1_1core_1_1metal_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::metal::Buffer::Buffer()']]], + ['buffer_113',['buffer',['../structmlx_1_1core_1_1array_1_1_data.html#a9a51e2d12ba505027cc0fca86bdd39ad',1,'mlx::core::array::Data::buffer'],['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a99183c92599edfeb75f7fa0f37e1d9eb',1,'mlx::core::metal::DeviceStream::buffer'],['../classmlx_1_1core_1_1array.html#ab3daf04c27c4593d9d73c397b8484a08',1,'mlx::core::array::buffer()'],['../classmlx_1_1core_1_1array.html#a634466ce661485394f2fdc3bd6796bcd',1,'mlx::core::array::buffer() const']]], + ['buffer_5fops_114',['buffer_ops',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#ab6048b329e65a59033834f3bdd351782',1,'mlx::core::metal::DeviceStream']]], + ['buffer_5fsize_115',['buffer_size',['../classmlx_1_1core_1_1array.html#a914577c63755b2e862d2da68bbf8e3dd',1,'mlx::core::array']]], + ['buffers_116',['buffers',['../struct_indices.html#ad705070a740579c07d109ae4f3d86e76',1,'Indices']]], + ['build_5fcommand_117',['build_command',['../classmlx_1_1core_1_1_jit_compiler.html#a10a5cde91ab929ccbdbdf4c4d940f156',1,'mlx::core::JitCompiler']]], + ['build_5flib_5fname_118',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]], + ['bytes_5fper_5fpack_119',['bytes_per_pack',['../struct_quantized_block_loader.html#ad00fe6d8bd395206a41693a8ed65d4db',1,'QuantizedBlockLoader']]] ]; diff --git a/docs/build/html/search/all_3.js b/docs/build/html/search/all_3.js index b193ec064..6cda0df89 100644 --- a/docs/build/html/search/all_3.js +++ b/docs/build/html/search/all_3.js @@ -13,156 +13,158 @@ var searchData= ['categorical_10',['categorical',['../namespacemlx_1_1core_1_1random.html#a1ede4c7f5c86e7b7e834953853ffae9a',1,'mlx::core::random::categorical(const array &logits, int axis, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa7104c436b3972a2480cfeb54554855f',1,'mlx::core::random::categorical(const array &logits_, int axis, int num_samples, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa336e774783543705dffe2ad5b2c49c1',1,'mlx::core::random::categorical(const array &logits, int axis=-1, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], ['category_11',['Category',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2d',1,'mlx::core::Dtype']]], ['ceil_12',['Ceil',['../struct_ceil.html',1,'Ceil'],['../classmlx_1_1core_1_1_ceil.html',1,'mlx::core::Ceil'],['../structmlx_1_1core_1_1detail_1_1_ceil.html',1,'mlx::core::detail::Ceil'],['../classmlx_1_1core_1_1_ceil.html#aede38610ca25429f229301546bc9b682',1,'mlx::core::Ceil::Ceil()']]], - ['ceil_13',['ceil',['../namespacemetal.html#ad63204d38bc01df6ffc64583f7886b3c',1,'metal::ceil()'],['../namespacemetal_1_1fast.html#a97b0bbd79f1f45d9d3104d712914e6b8',1,'metal::fast::ceil()'],['../namespacemetal_1_1precise.html#a8577549a1afeea206dd9a2004af2868d',1,'metal::precise::ceil()'],['../group__ops.html#ga1404ecceff83fd9b9139b7520f55e096',1,'mlx::core::ceil()']]], + ['ceil_13',['ceil',['../namespacemlx_1_1core_1_1simd.html#ae1d5460c58c507a0104d8dfa90343f12',1,'mlx::core::simd::ceil(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a4e54bd4ceb51ec41b0f95ebabe558713',1,'mlx::core::simd::ceil(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#ae5714693df24c8e26384fe5b5888376d',1,'mlx::core::simd::ceil(Simd< float16_t, N > a)'],['../namespacemetal.html#ad63204d38bc01df6ffc64583f7886b3c',1,'metal::ceil()'],['../namespacemetal_1_1fast.html#a97b0bbd79f1f45d9d3104d712914e6b8',1,'metal::fast::ceil()'],['../namespacemetal_1_1precise.html#a8577549a1afeea206dd9a2004af2868d',1,'metal::precise::ceil()'],['../group__ops.html#ga1404ecceff83fd9b9139b7520f55e096',1,'mlx::core::ceil()']]], ['ceildiv_14',['ceildiv',['../backend_2metal_2kernels_2utils_8h.html#a8e5a4b0fb5d018d7b078d147efe4f1e3',1,'utils.h']]], ['cfftp_15',['cfftp',['../classpocketfft_1_1detail_1_1cfftp.html',1,'pocketfft::detail::cfftp< T0 >'],['../classpocketfft_1_1detail_1_1cfftp.html#a121dcd8d4db180061d09fc1c5eb7da27',1,'pocketfft::detail::cfftp::cfftp()']]], ['channelhelper_16',['ChannelHelper',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], - ['channelhelper_3c_201_20_3e_17',['ChannelHelper< 1 >',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], - ['channelhelper_3c_202_20_3e_18',['ChannelHelper< 2 >',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], - ['channelhelper_3c_203_20_3e_19',['ChannelHelper< 3 >',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], - ['channelhelper_3c_204_20_3e_20',['ChannelHelper< 4 >',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], + ['channelhelper_3c_201_20_3e_17',['ChannelHelper< 1 >',['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html',1,'mlx::steel']]], + ['channelhelper_3c_202_20_3e_18',['ChannelHelper< 2 >',['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html',1,'mlx::steel']]], + ['channelhelper_3c_203_20_3e_19',['ChannelHelper< 3 >',['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html',1,'mlx::steel']]], + ['channelhelper_3c_204_20_3e_20',['ChannelHelper< 4 >',['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html',1,'mlx::steel']]], ['check_5fcontiguity_21',['check_contiguity',['../namespacemlx_1_1core.html#ab2b50a44a9d3a06282be4611f5fc7447',1,'mlx::core']]], ['checkpoint_22',['checkpoint',['../namespacemlx_1_1core.html#a42fa813d72c15132f76ef5fd1213ed71',1,'mlx::core']]], ['cholesky_23',['Cholesky',['../classmlx_1_1core_1_1_cholesky.html',1,'mlx::core::Cholesky'],['../classmlx_1_1core_1_1_cholesky.html#a6ae2e30b85f99f4f0d7f14c7949818ab',1,'mlx::core::Cholesky::Cholesky()']]], ['cholesky_24',['cholesky',['../namespacemlx_1_1core_1_1linalg.html#a46c8a4f806f0a97a4323e91189aa512b',1,'mlx::core::linalg']]], ['cholesky_5finv_25',['cholesky_inv',['../namespacemlx_1_1core_1_1linalg.html#aef0fe4894c5cf98792d59859c6d20511',1,'mlx::core::linalg']]], - ['clear_26',['clear',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa97a98e423827a889c13a92217626ec7',1,'mlx::steel::MMATile::clear()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa97a98e423827a889c13a92217626ec7',1,'mlx::steel::MMATile::clear()']]], - ['clear_5fcache_27',['clear_cache',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a447c1eb38c00d2e8e521675297f4a9b1',1,'mlx::core::metal::MetalAllocator::clear_cache()'],['../namespacemlx_1_1core_1_1metal.html#a22b3384ebd17f2fca198f81b9f1b6dc3',1,'mlx::core::metal::clear_cache()']]], - ['clip_28',['clip',['../group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8',1,'mlx::core']]], - ['close_29',['close',['../structmlx_1_1core_1_1_function_exporter.html#a43454277d21709d2f9b23a7ead0e674f',1,'mlx::core::FunctionExporter']]], - ['cmplx_30',['cmplx',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail::cmplx< T >'],['../structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c',1,'pocketfft::detail::cmplx::cmplx()'],['../structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a',1,'pocketfft::detail::cmplx::cmplx(T r_, T i_)']]], - ['cmplx_3c_20t0_20_3e_31',['cmplx< T0 >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], - ['cmplx_3c_20thigh_20_3e_32',['cmplx< Thigh >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], - ['cmplx_3c_20vtype_5ft_3c_20t_20_3e_20_3e_33',['cmplx< vtype_t< T > >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], - ['cndarr_34',['cndarr',['../classpocketfft_1_1detail_1_1cndarr.html',1,'pocketfft::detail::cndarr< T >'],['../classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129',1,'pocketfft::detail::cndarr::cndarr()']]], - ['col_5fcontiguous_35',['col_contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#ae24709026598d635e6b5c24a15f8a802',1,'mlx::core::array::Flags']]], - ['col_5freduce_5f2pass_36',['col_reduce_2pass',['../reduce__col_8h.html#a5a3d4847e4ae940d3a532d790c0b9b27',1,'reduce_col.h']]], - ['col_5freduce_5flongcolumn_37',['col_reduce_longcolumn',['../reduce__col_8h.html#a00bb1ed154aef8c1d2cb15d0c4231a51',1,'reduce_col.h']]], - ['col_5freduce_5flooped_38',['col_reduce_looped',['../reduce__col_8h.html#a59be8d8c4bca2217f62ea868550393c0',1,'reduce_col.h']]], - ['col_5freduce_5fsmall_39',['col_reduce_small',['../reduce__col_8h.html#a674f4b6075bab1b89778e10ab24c557e',1,'reduce_col.h']]], - ['collapse_5fcontiguous_5fdims_40',['collapse_contiguous_dims',['../namespacemlx_1_1core.html#a4d594bb84abeff4619d1abb77b20123e',1,'mlx::core::collapse_contiguous_dims(const Shape &shape, const std::vector< Strides > &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#a977c7c84de79ad67055ae2a89b7f6869',1,'mlx::core::collapse_contiguous_dims(const std::vector< array > &xs, size_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#ac813412cce77fc1340dcfefc6e099276',1,'mlx::core::collapse_contiguous_dims(Arrays &&... xs)'],['../namespacemlx_1_1core.html#a79acfa8bc30c1f213bf893b5983eb666',1,'mlx::core::collapse_contiguous_dims(const Shape &shape, const Strides &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#ab607cd6974ca6606826e785807156d6a',1,'mlx::core::collapse_contiguous_dims(const array &a, int64_t size_cap=std::numeric_limits< int32_t >::max())']]], - ['commandencoder_41',['CommandEncoder',['../structmlx_1_1core_1_1_command_encoder.html',1,'mlx::core::CommandEncoder'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html',1,'mlx::core::metal::CommandEncoder'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::metal::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::metal::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete'],['../structmlx_1_1core_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete']]], - ['commit_5fcommand_5fbuffer_42',['commit_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a95248f1387824067fd4fed23ace5ac0c',1,'mlx::core::metal::Device']]], - ['commonallocator_43',['CommonAllocator',['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html',1,'mlx::core::allocator']]], - ['communication_5fstream_44',['communication_stream',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6',1,'mlx::core::distributed::detail']]], - ['compile_45',['compile',['../namespacemlx_1_1core.html#a55933c6665de9f81059120d6b0de1c87',1,'mlx::core::compile(std::function< std::vector< array >(const std::vector< array > &)> fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#a56de948c2dffbf35c88035bc1561db53',1,'mlx::core::compile(std::vector< array >(fun)(const std::vector< array > &), bool shapeless=false)'],['../namespacemlx_1_1core.html#ace67713d269595f5f2265e46728a6f9c',1,'mlx::core::compile(F &&f, bool shapeless=false)'],['../namespacemlx_1_1core_1_1detail.html#af556c7576658b2e2498ead70339d95e5',1,'mlx::core::detail::compile()']]], - ['compile_2eh_46',['compile.h',['../compile_8h.html',1,'']]], - ['compile_5favailable_5ffor_5fdevice_47',['compile_available_for_device',['../namespacemlx_1_1core_1_1detail.html#aeeff2ba6ec3d9d4ed090de6d2681dbc2',1,'mlx::core::detail']]], - ['compile_5fclear_5fcache_48',['compile_clear_cache',['../namespacemlx_1_1core_1_1detail.html#a3fb927c209b946aefebb195993fbe4cf',1,'mlx::core::detail']]], - ['compile_5fdfs_49',['compile_dfs',['../namespacemlx_1_1core_1_1detail.html#a545fccdb5dc365b154cf4f0a2ca4753b',1,'mlx::core::detail']]], - ['compile_5ferase_50',['compile_erase',['../namespacemlx_1_1core_1_1detail.html#a69eb76a14f845ca000f1ccb2edda0175',1,'mlx::core::detail']]], - ['compile_5fimpl_2eh_51',['compile_impl.h',['../compile__impl_8h.html',1,'']]], - ['compile_5freplace_52',['compile_replace',['../namespacemlx_1_1core_1_1detail.html#a56fc01df6ba4c508d1da8b366b1328ac',1,'mlx::core::detail']]], - ['compile_5fsimplify_53',['compile_simplify',['../namespacemlx_1_1core_1_1detail.html#a33c878c900ca06f35d479f99c57b9e39',1,'mlx::core::detail']]], - ['compile_5ftrace_54',['compile_trace',['../namespacemlx_1_1core_1_1detail.html#ac2163a401119bb6edecfeb43373ef0dd',1,'mlx::core::detail']]], - ['compile_5fvalidate_5fshapeless_55',['compile_validate_shapeless',['../namespacemlx_1_1core_1_1detail.html#a10d612cb45a17fa17b704a357a902a68',1,'mlx::core::detail']]], - ['compiled_56',['Compiled',['../classmlx_1_1core_1_1_compiled.html',1,'mlx::core::Compiled'],['../classmlx_1_1core_1_1_compiled.html#a2d8cefff835c419a48a077d306b8e051',1,'mlx::core::Compiled::Compiled()']]], - ['compiled_2eh_57',['compiled.h',['../compiled_8h.html',1,'']]], - ['compiled_5fallocate_5foutputs_58',['compiled_allocate_outputs',['../namespacemlx_1_1core.html#ab8c3c4fc05745f586de922c8266f4fce',1,'mlx::core']]], - ['compiled_5fcheck_5fcontiguity_59',['compiled_check_contiguity',['../namespacemlx_1_1core.html#a562040f4a03f2c0a5d50eb9c8f14a8be',1,'mlx::core']]], - ['compiled_5fpreamble_2eh_60',['compiled_preamble.h',['../compiled__preamble_8h.html',1,'']]], - ['compilemode_61',['CompileMode',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4',1,'mlx::core']]], - ['complex_2eh_62',['complex.h',['../backend_2metal_2kernels_2complex_8h.html',1,'(Global Namespace)'],['../types_2complex_8h.html',1,'(Global Namespace)']]], - ['complex128_5ft_63',['complex128_t',['../structmlx_1_1core_1_1complex128__t.html',1,'mlx::core::complex128_t'],['../structmlx_1_1core_1_1complex128__t.html#aa15d0b805f8790f7c7b76fc7b9d677e0',1,'mlx::core::complex128_t::complex128_t(double v, double u)'],['../structmlx_1_1core_1_1complex128__t.html#abf2842253b874f9f13f39ea68a89e5b6',1,'mlx::core::complex128_t::complex128_t(std::complex< double > v)'],['../structmlx_1_1core_1_1complex128__t.html#a526fba96d7e815360cb4226af085a1bf',1,'mlx::core::complex128_t::complex128_t(T x)']]], - ['complex64_64',['complex64',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa8c022579455bcd2c681f007e84f4e2cf',1,'mlx::core::Dtype::complex64'],['../namespacemlx_1_1core.html#af99db87e0078bfcdb383f5689bc874d4',1,'mlx::core::complex64']]], - ['complex64_5ft_65',['complex64_t',['../structcomplex64__t.html',1,'complex64_t'],['../structmlx_1_1core_1_1complex64__t.html',1,'mlx::core::complex64_t'],['../structcomplex64__t.html#adbd392a5e92d31997380ad0a38be4be8',1,'complex64_t::complex64_t(float real, float imag)'],['../structcomplex64__t.html#a29782289bb90d6294099667b86509cd3',1,'complex64_t::complex64_t()'],['../structcomplex64__t.html#a905b048d70eb8d748a62454268242291',1,'complex64_t::complex64_t() threadgroup'],['../structcomplex64__t.html#a33a2452eb33b5ed53655773539c357a5',1,'complex64_t::complex64_t(T x) thread'],['../structcomplex64__t.html#a89b65ace8588b7bf215355f705eb23d9',1,'complex64_t::complex64_t(T x) threadgroup'],['../structcomplex64__t.html#ac81b486f642fb3b26c5d659917bdbcd0',1,'complex64_t::complex64_t(T x) device'],['../structcomplex64__t.html#a0a27a41206400f1e62b60ceb56960c93',1,'complex64_t::complex64_t(T x) const ant'],['../structmlx_1_1core_1_1complex64__t.html#a697cc973ae27d63c8e00d830e780bd8c',1,'mlx::core::complex64_t::complex64_t(float v, float u)'],['../structmlx_1_1core_1_1complex64__t.html#ae065e39938f9c4374b4116f4c67d4d09',1,'mlx::core::complex64_t::complex64_t(std::complex< float > v)'],['../structmlx_1_1core_1_1complex64__t.html#a2232cbbe591a9d2bc228cb23fac38b50',1,'mlx::core::complex64_t::complex64_t(T x)']]], - ['complex_5fbinop_66',['complex_binop',['../types_2complex_8h.html#a9c7995d495359894e1b30c0f1678d6bd',1,'complex.h']]], - ['complex_5fbinop_5fhelper_67',['complex_binop_helper',['../types_2complex_8h.html#ac6890f9852de12339b09b65757ebc8c4',1,'complex.h']]], - ['complex_5fmul_68',['complex_mul',['../radix_8h.html#a5bfc53b531214c9ce277bebc18aa67d6',1,'radix.h']]], - ['complex_5fmul_5fconj_69',['complex_mul_conj',['../radix_8h.html#a0e2dfd3d1dda09f47ccc64eec35629f3',1,'radix.h']]], - ['complexfloating_70',['complexfloating',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2dafb203630099d501ff7c255a574bc4812',1,'mlx::core::Dtype::complexfloating'],['../namespacemlx_1_1core.html#a70b8e88c9df750af984757105af33423',1,'mlx::core::complexfloating']]], - ['compute_5fstrided_5findices_71',['compute_strided_indices',['../struct_read_writer.html#a7c903fbb8b85a856ba5564d7df537cdf',1,'ReadWriter']]], - ['concatenate_72',['Concatenate',['../classmlx_1_1core_1_1_concatenate.html',1,'mlx::core::Concatenate'],['../classmlx_1_1core_1_1_concatenate.html#acff07853de2d31faeec7c4ca40ce0888',1,'mlx::core::Concatenate::Concatenate()']]], - ['concatenate_73',['concatenate',['../namespacemlx_1_1core.html#a76a2e310857f60f5ea6f1388d45b964d',1,'mlx::core::concatenate(std::string &acc, T first)'],['../namespacemlx_1_1core.html#aaf51544472fa87fa974686eacdd2a4a6',1,'mlx::core::concatenate(std::string &acc, T first, Args... args)'],['../group__ops.html#ga52838af566948b1b96e7aa00832071b3',1,'mlx::core::concatenate(std::vector< array > arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga666ac69778984fafdc2f51d296270468',1,'mlx::core::concatenate(std::vector< array > arrays, StreamOrDevice s={})']]], - ['concatenate_5fgpu_74',['concatenate_gpu',['../namespacemlx_1_1core.html#a050299d0d366ca5c9d09d1004dcc3e7d',1,'mlx::core']]], - ['concurrent_5fqueue_75',['concurrent_queue',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html',1,'pocketfft::detail::threading']]], - ['concurrent_5fqueue_3c_20std_3a_3afunction_3c_20void_28_29_3e_20_3e_76',['concurrent_queue< std::function< void()> >',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html',1,'pocketfft::detail::threading']]], - ['concurrentcontext_77',['ConcurrentContext',['../structmlx_1_1core_1_1_command_encoder_1_1_concurrent_context.html',1,'mlx::core::CommandEncoder::ConcurrentContext'],['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html',1,'mlx::core::metal::CommandEncoder::ConcurrentContext'],['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::metal::CommandEncoder::ConcurrentContext::ConcurrentContext()'],['../structmlx_1_1core_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::CommandEncoder::ConcurrentContext::ConcurrentContext()']]], - ['cond_78',['cond',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4ffd524d6a5bedd1a303b63bdde6701c',1,'mlx::core::scheduler::StreamThread']]], - ['conditionaltype_79',['ConditionalType',['../struct_conditional_type.html',1,'']]], - ['conditionaltype_3c_20true_2c_20t_2c_20u_20_3e_80',['ConditionalType< true, T, U >',['../struct_conditional_type.html',1,'']]], - ['conj_81',['conj',['../namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3',1,'pocketfft::detail']]], - ['conjugate_82',['Conjugate',['../struct_conjugate.html',1,'Conjugate'],['../classmlx_1_1core_1_1_conjugate.html',1,'mlx::core::Conjugate'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html',1,'mlx::core::detail::Conjugate'],['../classmlx_1_1core_1_1_conjugate.html#a627f9e6a8729fb3ffb3ca3228d007c87',1,'mlx::core::Conjugate::Conjugate()']]], - ['conjugate_83',['conjugate',['../group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23',1,'mlx::core']]], - ['contiguous_84',['Contiguous',['../classmlx_1_1core_1_1_contiguous.html',1,'mlx::core::Contiguous'],['../classmlx_1_1core_1_1_contiguous.html#a3e83f414c02ae0b92a50b6f8e402e1c0',1,'mlx::core::Contiguous::Contiguous()']]], - ['contiguous_85',['contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#afd0ab11e7a486a2a8e50ee84b971ac8a',1,'mlx::core::array::Flags::contiguous'],['../group__ops.html#ga8ab10aa6c41416d739791164a52b25d5',1,'mlx::core::contiguous()']]], - ['contiguous_5fscan_86',['contiguous_scan',['../scan_8h.html#a60d279b9add7d56639bb209408f09d79',1,'scan.h']]], - ['contiguousallreduce_87',['ContiguousAllReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ae4e34c7154eb8dc47aa8503209730424',1,'mlx::core']]], - ['contiguousiterator_88',['ContiguousIterator',['../structmlx_1_1core_1_1_contiguous_iterator.html',1,'mlx::core::ContiguousIterator'],['../structmlx_1_1core_1_1_contiguous_iterator.html#a727442ddff5fd3c3ebe09b000a01c9d3',1,'mlx::core::ContiguousIterator::ContiguousIterator()'],['../structmlx_1_1core_1_1_contiguous_iterator.html#aa82bec516eb54656c74fdaa74de1d735',1,'mlx::core::ContiguousIterator::ContiguousIterator(const array &a)'],['../structmlx_1_1core_1_1_contiguous_iterator.html#a8760380bff7462a886e7a4edd2955375',1,'mlx::core::ContiguousIterator::ContiguousIterator(const Shape &shape, const Strides &strides, int dims)']]], - ['contiguousreduce_89',['ContiguousReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ad2547f25dffe8d8936dbec25601cfc84',1,'mlx::core']]], - ['contiguousstridedreduce_90',['ContiguousStridedReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ab48dac7508a2c790de1bdc33f29177ed',1,'mlx::core']]], - ['conv_91',['conv',['../namespacemlx_1_1core_1_1metal.html#ab1704e853394c725668c06752ebb5c24',1,'mlx::core::metal']]], - ['conv_2eh_92',['conv.h',['../conv_8h.html',1,'']]], - ['conv1d_93',['conv1d',['../group__ops.html#ga30d47e08093c03a3676f235f9f559411',1,'mlx::core']]], - ['conv2d_94',['conv2d',['../group__ops.html#ga73b02833229678786e7f302d458d5a83',1,'mlx::core']]], - ['conv2dgeneralbaseinfo_95',['Conv2DGeneralBaseInfo',['../structmlx_1_1steel_1_1_conv2_d_general_base_info.html',1,'mlx::steel']]], - ['conv2dgeneraljumpparams_96',['Conv2DGeneralJumpParams',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html',1,'mlx::steel']]], - ['conv2dinputblockloadergeneral_97',['Conv2DInputBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html',1,'mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1d83af561a483432bf8dcb42e734b23b',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::Conv2DInputBlockLoaderGeneral()']]], - ['conv2dinputblockloaderlargefilter_98',['Conv2DInputBlockLoaderLargeFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8755116a535539744e4947bc69f9c50f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::Conv2DInputBlockLoaderLargeFilter()']]], - ['conv2dinputblockloadersmallchannels_99',['Conv2DInputBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ab9fd3fdeab94470dde3326f1dd5c455a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::Conv2DInputBlockLoaderSmallChannels()']]], - ['conv2dinputblockloadersmallfilter_100',['Conv2DInputBlockLoaderSmallFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0a2cbf57c51cd928722e3f06aafcf933',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::Conv2DInputBlockLoaderSmallFilter()']]], - ['conv2dweightblockloader_101',['Conv2DWeightBlockLoader',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html',1,'mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a9a7dca3512b64cffb6eac305d795831c',1,'mlx::steel::Conv2DWeightBlockLoader::Conv2DWeightBlockLoader()']]], - ['conv2dweightblockloadergeneral_102',['Conv2DWeightBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ad0550fabbdc9297559381a5b488e9af1',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::Conv2DWeightBlockLoaderGeneral()']]], - ['conv2dweightblockloadersmallchannels_103',['Conv2DWeightBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae1806ea1c19713819dee83a38ab35fa6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::Conv2DWeightBlockLoaderSmallChannels()']]], - ['conv3d_104',['conv3d',['../group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca',1,'mlx::core']]], - ['conv_5fgeneral_105',['conv_general',['../group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e',1,'mlx::core::conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})'],['../group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d',1,'mlx::core::conv_general(const array &input, const array &weight, std::vector< int > stride={}, std::vector< int > padding={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})']]], - ['conv_5ftranspose1d_106',['conv_transpose1d',['../group__ops.html#gaa30bf1adcd78d1c2595d07b215731714',1,'mlx::core']]], - ['conv_5ftranspose2d_107',['conv_transpose2d',['../group__ops.html#gaebb59971cb9bc45005dc1d398e4f0a3d',1,'mlx::core']]], - ['conv_5ftranspose3d_108',['conv_transpose3d',['../group__ops.html#ga8db814da631d9cd32a8d6563bf4ac530',1,'mlx::core']]], - ['convolution_109',['Convolution',['../classmlx_1_1core_1_1_convolution.html',1,'mlx::core::Convolution'],['../classmlx_1_1core_1_1_convolution.html#a6f1de77b719bb13217b0d8c64cabb8ef',1,'mlx::core::Convolution::Convolution()']]], - ['copy_110',['Copy',['../classmlx_1_1core_1_1_copy.html',1,'mlx::core::Copy'],['../classmlx_1_1core_1_1_copy.html#a6243e044af119105ffaaed7d405cd584',1,'mlx::core::Copy::Copy()']]], - ['copy_111',['copy',['../namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2',1,'mlx::core::copy()'],['../namespacemlx_1_1core_1_1metal.html#aa215e631e2680f04a591b88d91571719',1,'mlx::core::metal::copy()'],['../group__ops.html#gae306e93af12f774bd80bad6c231b09d6',1,'mlx::core::copy()']]], - ['copy_2eh_112',['copy.h',['../common_2copy_8h.html',1,'(Global Namespace)'],['../metal_2copy_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2copy_8h.html',1,'(Global Namespace)']]], - ['copy_5fg_113',['copy_g',['../metal_2kernels_2copy_8h.html#a71e4103db4689d90ef6f9d5ba93604cf',1,'copy.h']]], - ['copy_5fg_5fnd1_114',['copy_g_nd1',['../metal_2kernels_2copy_8h.html#a232c5c6b8386cf8ecbf4cdadb6e4176e',1,'copy.h']]], - ['copy_5fg_5fnd2_115',['copy_g_nd2',['../metal_2kernels_2copy_8h.html#a39ec5b7b8351e4332b842982a2ee6260',1,'copy.h']]], - ['copy_5fg_5fnd3_116',['copy_g_nd3',['../metal_2kernels_2copy_8h.html#aab82689380897ff4716b5eafd6ef3ecc',1,'copy.h']]], - ['copy_5fgg_117',['copy_gg',['../metal_2kernels_2copy_8h.html#ade9a9eea9b8262a854a11721fe2bb9fa',1,'copy.h']]], - ['copy_5fgg_5fdynamic_118',['copy_gg_dynamic',['../metal_2kernels_2copy_8h.html#ad0f05a73165d4ee38c9f02c705ea6ca8',1,'copy.h']]], - ['copy_5fgg_5fdynamic_5fnd1_119',['copy_gg_dynamic_nd1',['../metal_2kernels_2copy_8h.html#a8548ea41cac179084ddd33d26921576f',1,'copy.h']]], - ['copy_5fgg_5fdynamic_5fnd2_120',['copy_gg_dynamic_nd2',['../metal_2kernels_2copy_8h.html#a9b9266ee25a4dbcbe4fde883b40170f1',1,'copy.h']]], - ['copy_5fgg_5fdynamic_5fnd3_121',['copy_gg_dynamic_nd3',['../metal_2kernels_2copy_8h.html#af33ccc02f10bcb5c19ea7b1dd0af4956',1,'copy.h']]], - ['copy_5fgg_5fnd1_122',['copy_gg_nd1',['../metal_2kernels_2copy_8h.html#a370d7bbba1a4b0d64da873bafd29a78b',1,'copy.h']]], - ['copy_5fgg_5fnd2_123',['copy_gg_nd2',['../metal_2kernels_2copy_8h.html#af0b06ac3a96852a64fa4274a94b58301',1,'copy.h']]], - ['copy_5fgg_5fnd3_124',['copy_gg_nd3',['../metal_2kernels_2copy_8h.html#a3f3836ad0b6545ec9b9e1864224f7a13',1,'copy.h']]], - ['copy_5fgpu_125',['copy_gpu',['../namespacemlx_1_1core.html#addaa46a13ac2deb1d9ce621338320e0e',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a6a6f4e46c8fc44fdc74c50ace02bcf38',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype)']]], - ['copy_5fgpu_5finplace_126',['copy_gpu_inplace',['../namespacemlx_1_1core.html#a473fb602368f6c73d9105c9a151c4c82',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s, const std::optional< array > &dynamic_i_offset=std::nullopt, const std::optional< array > &dynamic_o_offset=std::nullopt)'],['../namespacemlx_1_1core.html#a58ef0842dd1b8f79159d5fb6777d30a1',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a49fc043a981925b9be79e37fc415d966',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const Strides &i_strides, int64_t i_offset, CopyType ctype, const Stream &s)']]], - ['copy_5fhartley_127',['copy_hartley',['../namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#ae7b44d2773d9d06a9787aff01d66b3ed',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], - ['copy_5finplace_128',['copy_inplace',['../namespacemlx_1_1core.html#a98495894a796b2cc6d022e7a03432c64',1,'mlx::core::copy_inplace(const array &src, array &dst, CopyType ctype)'],['../namespacemlx_1_1core.html#ae85bafda5ab0b4b2289591260cf07685',1,'mlx::core::copy_inplace(const array &src, array &dst, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)']]], - ['copy_5finput_129',['copy_input',['../namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< cmplx< T > > &src, cmplx< vtype_t< T > > *dst)'],['../namespacepocketfft_1_1detail.html#a30fc708f9d8f9cfa74194925c7863c0a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, vtype_t< T > *dst)'],['../namespacepocketfft_1_1detail.html#a3387bd35f237870e42b8461769e6aec4',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, T *dst)']]], - ['copy_5foutput_130',['copy_output',['../namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const cmplx< vtype_t< T > > *src, ndarr< cmplx< T > > &dst)'],['../namespacepocketfft_1_1detail.html#a21980853aca4d92ed06e3dcffe7ef660',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#a310481c334e46674710ba794ad7403c0',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], - ['copy_5fs_131',['copy_s',['../metal_2kernels_2copy_8h.html#aef09f9b9475345b1bba121d037d222ea',1,'copy.h']]], - ['copy_5fs2_132',['copy_s2',['../metal_2kernels_2copy_8h.html#a8023e9335cc5334847a8d315042be3a3',1,'copy.h']]], - ['copy_5fshared_5fbuffer_133',['copy_shared_buffer',['../classmlx_1_1core_1_1array.html#ad2814dbffa5ad174d9c97a10bf4cf26b',1,'mlx::core::array::copy_shared_buffer(const array &other, const Strides &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a92974c656c35a972ad241f80584bbd29',1,'mlx::core::array::copy_shared_buffer(const array &other)']]], - ['copy_5fv_134',['copy_v',['../metal_2kernels_2copy_8h.html#ae26a13e0c8e6c15f7b10078e65970659',1,'copy.h']]], - ['copy_5fv2_135',['copy_v2',['../metal_2kernels_2copy_8h.html#aee14a5326f53d9b30b0b38e27d180ef3',1,'copy.h']]], - ['copytype_136',['CopyType',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337',1,'mlx::core']]], - ['core_20array_20operations_137',['Core array operations',['../group__ops.html',1,'']]], - ['cos_138',['Cos',['../struct_cos.html',1,'Cos'],['../classmlx_1_1core_1_1_cos.html',1,'mlx::core::Cos'],['../structmlx_1_1core_1_1detail_1_1_cos.html',1,'mlx::core::detail::Cos'],['../classmlx_1_1core_1_1_cos.html#a2acb9fcf0901462189c476756fd99995',1,'mlx::core::Cos::Cos()']]], - ['cos_139',['cos',['../namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d',1,'pocketfft::detail::cos()'],['../namespacemetal.html#a2fa4778a6fe2fa43253ea724e5a608a3',1,'metal::cos()'],['../namespacemetal_1_1fast.html#a75b6bb32fa3870eda46a7bfc9f481f88',1,'metal::fast::cos()'],['../namespacemetal_1_1precise.html#ac4941f62e7d8ab9d7cabbd967aa9f220',1,'metal::precise::cos()'],['../group__ops.html#ga39dfdf72b556012aa35ff27a94116e74',1,'mlx::core::cos()']]], - ['cosh_140',['Cosh',['../struct_cosh.html',1,'Cosh'],['../classmlx_1_1core_1_1_cosh.html',1,'mlx::core::Cosh'],['../structmlx_1_1core_1_1detail_1_1_cosh.html',1,'mlx::core::detail::Cosh'],['../classmlx_1_1core_1_1_cosh.html#a44e8ac2e09a55ec32e9dc6641eedc8f1',1,'mlx::core::Cosh::Cosh()']]], - ['cosh_141',['cosh',['../namespacemetal.html#a8a68a88cc110830d057dbd71431b93c0',1,'metal::cosh()'],['../namespacemetal_1_1fast.html#a31544ad9de28012a4ddda86e3966a77e',1,'metal::fast::cosh()'],['../namespacemetal_1_1precise.html#a72d86d508300a9b58f4ccbbe70da4fbc',1,'metal::precise::cosh()'],['../group__ops.html#ga2181b71cda88007a3092be4795ff0715',1,'mlx::core::cosh()']]], - ['cosine_142',['cosine',['../structpocketfft_1_1detail_1_1_exec_dcst.html#a185023fc1e386cc8f233b79c49c1fd8a',1,'pocketfft::detail::ExecDcst']]], - ['cospi_143',['cospi',['../namespacemetal.html#a5c2f37939ad705ddea4409d3bedb8ce1',1,'metal::cospi()'],['../namespacemetal_1_1fast.html#a9906b41f75319b384ffb570cc94d67ce',1,'metal::fast::cospi()'],['../namespacemetal_1_1precise.html#a2392b78bd196efdbbac65901c4ab20e7',1,'metal::precise::cospi()']]], - ['cost_5fguess_144',['cost_guess',['../structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0',1,'pocketfft::detail::util']]], - ['count_5fdown_145',['count_down',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342',1,'pocketfft::detail::threading::latch']]], - ['cpu_146',['cpu',['../structmlx_1_1core_1_1_device.html#a69ee81924251dec96f1945c9d91506fd',1,'mlx::core::Device::cpu'],['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdbad9747e2da342bdb995f6389533ad1a3d',1,'mlx::core::Device::cpu']]], - ['cross_147',['cross',['../namespacemlx_1_1core_1_1linalg.html#abcda3fbda45183c21e7f27aa0dde64e6',1,'mlx::core::linalg']]], - ['cshape_148',['CShape',['../structmlx_1_1steel_1_1_c_shape.html',1,'mlx::steel']]], - ['ctile_149',['Ctile',['../structmlx_1_1steel_1_1_block_m_m_a.html#a21b0c40d16eced109bd3196186170bc6',1,'mlx::steel::BlockMMA']]], - ['cummax_150',['CumMax',['../struct_cum_max.html',1,'']]], - ['cummax_151',['cummax',['../group__ops.html#gaee37cac8476e8f8d666bcded5bc59143',1,'mlx::core']]], - ['cummin_152',['CumMin',['../struct_cum_min.html',1,'']]], - ['cummin_153',['cummin',['../group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8',1,'mlx::core']]], - ['cumprod_154',['CumProd',['../struct_cum_prod.html',1,'']]], - ['cumprod_155',['cumprod',['../group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f',1,'mlx::core']]], - ['cumprod_3c_20bool_20_3e_156',['CumProd< bool >',['../struct_cum_prod.html',1,'']]], - ['cumsum_157',['CumSum',['../struct_cum_sum.html',1,'']]], - ['cumsum_158',['cumsum',['../group__ops.html#gaddc825a5c173e195ab0fda83ad630420',1,'mlx::core']]], - ['custom_159',['Custom',['../classmlx_1_1core_1_1fast_1_1_custom.html',1,'mlx::core::fast::Custom'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a4186fea23f7156c38960426821fca313',1,'mlx::core::fast::Custom::Custom()']]], - ['custom_5ffunction_160',['custom_function',['../namespacemlx_1_1core.html#a8d3ca5fbaecdb995660c24cde5aeebaf',1,'mlx::core']]], - ['custom_5fvjp_161',['custom_vjp',['../namespacemlx_1_1core.html#a9290596250fa308df4c69b44483bb8aa',1,'mlx::core']]], - ['customkernel_162',['CustomKernel',['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html',1,'mlx::core::fast::CustomKernel'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a954893e07f0d36715b4e1e414b6f2153',1,'mlx::core::fast::CustomKernel::CustomKernel()']]], - ['customkernelshapeinfo_163',['CustomKernelShapeInfo',['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html',1,'mlx::core::fast']]], - ['customtransforms_164',['CustomTransforms',['../classmlx_1_1core_1_1_custom_transforms.html',1,'mlx::core::CustomTransforms'],['../classmlx_1_1core_1_1_custom_transforms.html#ab52abadb9c6f6db83d087c7b751be488',1,'mlx::core::CustomTransforms::CustomTransforms()']]] + ['clamp_26',['clamp',['../namespacemlx_1_1core_1_1simd.html#ae4be4d88cd8eba7a8c1784fd53b86edb',1,'mlx::core::simd::clamp(Simd< T, N > v, Simd< T, N > min, Simd< T, N > max)'],['../namespacemlx_1_1core_1_1simd.html#a567c06bf988af03988478679055a6c45',1,'mlx::core::simd::clamp(Simd< T, 1 > v, Simd< T, 1 > min, Simd< T, 1 > max)'],['../namespacemlx_1_1core_1_1simd.html#a146d2a834c936a381c1f86caffa822d7',1,'mlx::core::simd::clamp(Simd< float16_t, N > v, Simd< float16_t, N > min, Simd< float16_t, N > max)']]], + ['clear_27',['clear',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa97a98e423827a889c13a92217626ec7',1,'mlx::steel::MMATile::clear()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa97a98e423827a889c13a92217626ec7',1,'mlx::steel::MMATile::clear()']]], + ['clear_5fcache_28',['clear_cache',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a447c1eb38c00d2e8e521675297f4a9b1',1,'mlx::core::metal::MetalAllocator::clear_cache()'],['../namespacemlx_1_1core_1_1metal.html#a22b3384ebd17f2fca198f81b9f1b6dc3',1,'mlx::core::metal::clear_cache()']]], + ['clip_29',['clip',['../group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8',1,'mlx::core']]], + ['close_30',['close',['../structmlx_1_1core_1_1_function_exporter.html#a43454277d21709d2f9b23a7ead0e674f',1,'mlx::core::FunctionExporter']]], + ['cmplx_31',['cmplx',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail::cmplx< T >'],['../structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c',1,'pocketfft::detail::cmplx::cmplx()'],['../structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a',1,'pocketfft::detail::cmplx::cmplx(T r_, T i_)']]], + ['cmplx_3c_20t0_20_3e_32',['cmplx< T0 >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], + ['cmplx_3c_20thigh_20_3e_33',['cmplx< Thigh >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], + ['cmplx_3c_20vtype_5ft_3c_20t_20_3e_20_3e_34',['cmplx< vtype_t< T > >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], + ['cndarr_35',['cndarr',['../classpocketfft_1_1detail_1_1cndarr.html',1,'pocketfft::detail::cndarr< T >'],['../classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129',1,'pocketfft::detail::cndarr::cndarr()']]], + ['col_5fcontiguous_36',['col_contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#ae24709026598d635e6b5c24a15f8a802',1,'mlx::core::array::Flags']]], + ['col_5ffrag_5ftype_37',['col_frag_type',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#adbb262a3c872e26533b68a39db16459e',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >']]], + ['col_5freduce_5f2pass_38',['col_reduce_2pass',['../reduce__col_8h.html#a5a3d4847e4ae940d3a532d790c0b9b27',1,'reduce_col.h']]], + ['col_5freduce_5flongcolumn_39',['col_reduce_longcolumn',['../reduce__col_8h.html#a00bb1ed154aef8c1d2cb15d0c4231a51',1,'reduce_col.h']]], + ['col_5freduce_5flooped_40',['col_reduce_looped',['../reduce__col_8h.html#a59be8d8c4bca2217f62ea868550393c0',1,'reduce_col.h']]], + ['col_5freduce_5fsmall_41',['col_reduce_small',['../reduce__col_8h.html#a674f4b6075bab1b89778e10ab24c557e',1,'reduce_col.h']]], + ['collapse_5fcontiguous_5fdims_42',['collapse_contiguous_dims',['../namespacemlx_1_1core.html#a4d594bb84abeff4619d1abb77b20123e',1,'mlx::core::collapse_contiguous_dims(const Shape &shape, const std::vector< Strides > &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#a977c7c84de79ad67055ae2a89b7f6869',1,'mlx::core::collapse_contiguous_dims(const std::vector< array > &xs, size_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#ac813412cce77fc1340dcfefc6e099276',1,'mlx::core::collapse_contiguous_dims(Arrays &&... xs)'],['../namespacemlx_1_1core.html#a79acfa8bc30c1f213bf893b5983eb666',1,'mlx::core::collapse_contiguous_dims(const Shape &shape, const Strides &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#ab607cd6974ca6606826e785807156d6a',1,'mlx::core::collapse_contiguous_dims(const array &a, int64_t size_cap=std::numeric_limits< int32_t >::max())']]], + ['commandencoder_43',['CommandEncoder',['../structmlx_1_1core_1_1_command_encoder.html',1,'mlx::core::CommandEncoder'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html',1,'mlx::core::metal::CommandEncoder'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::metal::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::metal::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete'],['../structmlx_1_1core_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete']]], + ['commit_5fcommand_5fbuffer_44',['commit_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a95248f1387824067fd4fed23ace5ac0c',1,'mlx::core::metal::Device']]], + ['commonallocator_45',['CommonAllocator',['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html',1,'mlx::core::allocator']]], + ['communication_5fstream_46',['communication_stream',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6',1,'mlx::core::distributed::detail']]], + ['compile_47',['compile',['../namespacemlx_1_1core.html#a55933c6665de9f81059120d6b0de1c87',1,'mlx::core::compile(std::function< std::vector< array >(const std::vector< array > &)> fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#a56de948c2dffbf35c88035bc1561db53',1,'mlx::core::compile(std::vector< array >(fun)(const std::vector< array > &), bool shapeless=false)'],['../namespacemlx_1_1core.html#ace67713d269595f5f2265e46728a6f9c',1,'mlx::core::compile(F &&f, bool shapeless=false)'],['../namespacemlx_1_1core_1_1detail.html#af556c7576658b2e2498ead70339d95e5',1,'mlx::core::detail::compile()']]], + ['compile_2eh_48',['compile.h',['../compile_8h.html',1,'']]], + ['compile_5favailable_5ffor_5fdevice_49',['compile_available_for_device',['../namespacemlx_1_1core_1_1detail.html#aeeff2ba6ec3d9d4ed090de6d2681dbc2',1,'mlx::core::detail']]], + ['compile_5fclear_5fcache_50',['compile_clear_cache',['../namespacemlx_1_1core_1_1detail.html#a3fb927c209b946aefebb195993fbe4cf',1,'mlx::core::detail']]], + ['compile_5fdfs_51',['compile_dfs',['../namespacemlx_1_1core_1_1detail.html#a545fccdb5dc365b154cf4f0a2ca4753b',1,'mlx::core::detail']]], + ['compile_5ferase_52',['compile_erase',['../namespacemlx_1_1core_1_1detail.html#a69eb76a14f845ca000f1ccb2edda0175',1,'mlx::core::detail']]], + ['compile_5fimpl_2eh_53',['compile_impl.h',['../compile__impl_8h.html',1,'']]], + ['compile_5freplace_54',['compile_replace',['../namespacemlx_1_1core_1_1detail.html#a56fc01df6ba4c508d1da8b366b1328ac',1,'mlx::core::detail']]], + ['compile_5fsimplify_55',['compile_simplify',['../namespacemlx_1_1core_1_1detail.html#a33c878c900ca06f35d479f99c57b9e39',1,'mlx::core::detail']]], + ['compile_5ftrace_56',['compile_trace',['../namespacemlx_1_1core_1_1detail.html#ac2163a401119bb6edecfeb43373ef0dd',1,'mlx::core::detail']]], + ['compile_5fvalidate_5fshapeless_57',['compile_validate_shapeless',['../namespacemlx_1_1core_1_1detail.html#a10d612cb45a17fa17b704a357a902a68',1,'mlx::core::detail']]], + ['compiled_58',['Compiled',['../classmlx_1_1core_1_1_compiled.html',1,'mlx::core::Compiled'],['../classmlx_1_1core_1_1_compiled.html#a2d8cefff835c419a48a077d306b8e051',1,'mlx::core::Compiled::Compiled()']]], + ['compiled_2eh_59',['compiled.h',['../compiled_8h.html',1,'']]], + ['compiled_5fallocate_5foutputs_60',['compiled_allocate_outputs',['../namespacemlx_1_1core.html#ab8c3c4fc05745f586de922c8266f4fce',1,'mlx::core']]], + ['compiled_5fcheck_5fcontiguity_61',['compiled_check_contiguity',['../namespacemlx_1_1core.html#a562040f4a03f2c0a5d50eb9c8f14a8be',1,'mlx::core']]], + ['compiled_5fpreamble_2eh_62',['compiled_preamble.h',['../compiled__preamble_8h.html',1,'']]], + ['compilemode_63',['CompileMode',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4',1,'mlx::core']]], + ['complex_2eh_64',['complex.h',['../backend_2metal_2kernels_2complex_8h.html',1,'(Global Namespace)'],['../types_2complex_8h.html',1,'(Global Namespace)']]], + ['complex128_5ft_65',['complex128_t',['../structmlx_1_1core_1_1complex128__t.html',1,'mlx::core::complex128_t'],['../structmlx_1_1core_1_1complex128__t.html#a4330d04587f3282bcd650e36532da178',1,'mlx::core::complex128_t::complex128_t()'],['../structmlx_1_1core_1_1complex128__t.html#aa15d0b805f8790f7c7b76fc7b9d677e0',1,'mlx::core::complex128_t::complex128_t(double v, double u)'],['../structmlx_1_1core_1_1complex128__t.html#abf2842253b874f9f13f39ea68a89e5b6',1,'mlx::core::complex128_t::complex128_t(std::complex< double > v)'],['../structmlx_1_1core_1_1complex128__t.html#a526fba96d7e815360cb4226af085a1bf',1,'mlx::core::complex128_t::complex128_t(T x)']]], + ['complex64_66',['complex64',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa8c022579455bcd2c681f007e84f4e2cf',1,'mlx::core::Dtype::complex64'],['../namespacemlx_1_1core.html#af99db87e0078bfcdb383f5689bc874d4',1,'mlx::core::complex64']]], + ['complex64_5ft_67',['complex64_t',['../structcomplex64__t.html',1,'complex64_t'],['../structmlx_1_1core_1_1complex64__t.html',1,'mlx::core::complex64_t'],['../structcomplex64__t.html#adbd392a5e92d31997380ad0a38be4be8',1,'complex64_t::complex64_t(float real, float imag)'],['../structcomplex64__t.html#a29782289bb90d6294099667b86509cd3',1,'complex64_t::complex64_t()'],['../structcomplex64__t.html#a905b048d70eb8d748a62454268242291',1,'complex64_t::complex64_t() threadgroup'],['../structcomplex64__t.html#a33a2452eb33b5ed53655773539c357a5',1,'complex64_t::complex64_t(T x) thread'],['../structcomplex64__t.html#a89b65ace8588b7bf215355f705eb23d9',1,'complex64_t::complex64_t(T x) threadgroup'],['../structcomplex64__t.html#ac81b486f642fb3b26c5d659917bdbcd0',1,'complex64_t::complex64_t(T x) device'],['../structcomplex64__t.html#a0a27a41206400f1e62b60ceb56960c93',1,'complex64_t::complex64_t(T x) const ant'],['../structmlx_1_1core_1_1complex64__t.html#ad27bed7d6b7966bfcf563af06bedddf3',1,'mlx::core::complex64_t::complex64_t()'],['../structmlx_1_1core_1_1complex64__t.html#a697cc973ae27d63c8e00d830e780bd8c',1,'mlx::core::complex64_t::complex64_t(float v, float u)'],['../structmlx_1_1core_1_1complex64__t.html#ae065e39938f9c4374b4116f4c67d4d09',1,'mlx::core::complex64_t::complex64_t(std::complex< float > v)'],['../structmlx_1_1core_1_1complex64__t.html#a2232cbbe591a9d2bc228cb23fac38b50',1,'mlx::core::complex64_t::complex64_t(T x)']]], + ['complex_5fbinop_68',['complex_binop',['../types_2complex_8h.html#a9c7995d495359894e1b30c0f1678d6bd',1,'complex.h']]], + ['complex_5fbinop_5fhelper_69',['complex_binop_helper',['../types_2complex_8h.html#ac6890f9852de12339b09b65757ebc8c4',1,'complex.h']]], + ['complex_5fmul_70',['complex_mul',['../radix_8h.html#a5bfc53b531214c9ce277bebc18aa67d6',1,'radix.h']]], + ['complex_5fmul_5fconj_71',['complex_mul_conj',['../radix_8h.html#a0e2dfd3d1dda09f47ccc64eec35629f3',1,'radix.h']]], + ['complexfloating_72',['complexfloating',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2dafb203630099d501ff7c255a574bc4812',1,'mlx::core::Dtype::complexfloating'],['../namespacemlx_1_1core.html#a70b8e88c9df750af984757105af33423',1,'mlx::core::complexfloating']]], + ['compute_5fstrided_5findices_73',['compute_strided_indices',['../struct_read_writer.html#a7c903fbb8b85a856ba5564d7df537cdf',1,'ReadWriter']]], + ['concatenate_74',['Concatenate',['../classmlx_1_1core_1_1_concatenate.html',1,'mlx::core::Concatenate'],['../classmlx_1_1core_1_1_concatenate.html#acff07853de2d31faeec7c4ca40ce0888',1,'mlx::core::Concatenate::Concatenate()']]], + ['concatenate_75',['concatenate',['../namespacemlx_1_1core.html#a76a2e310857f60f5ea6f1388d45b964d',1,'mlx::core::concatenate(std::string &acc, T first)'],['../namespacemlx_1_1core.html#aaf51544472fa87fa974686eacdd2a4a6',1,'mlx::core::concatenate(std::string &acc, T first, Args... args)'],['../group__ops.html#ga52838af566948b1b96e7aa00832071b3',1,'mlx::core::concatenate(std::vector< array > arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga666ac69778984fafdc2f51d296270468',1,'mlx::core::concatenate(std::vector< array > arrays, StreamOrDevice s={})']]], + ['concatenate_5fgpu_76',['concatenate_gpu',['../namespacemlx_1_1core.html#a050299d0d366ca5c9d09d1004dcc3e7d',1,'mlx::core']]], + ['concurrent_5fqueue_77',['concurrent_queue',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html',1,'pocketfft::detail::threading']]], + ['concurrent_5fqueue_3c_20std_3a_3afunction_3c_20void_28_29_3e_20_3e_78',['concurrent_queue< std::function< void()> >',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html',1,'pocketfft::detail::threading']]], + ['concurrentcontext_79',['ConcurrentContext',['../structmlx_1_1core_1_1_command_encoder_1_1_concurrent_context.html',1,'mlx::core::CommandEncoder::ConcurrentContext'],['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html',1,'mlx::core::metal::CommandEncoder::ConcurrentContext'],['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::metal::CommandEncoder::ConcurrentContext::ConcurrentContext()'],['../structmlx_1_1core_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::CommandEncoder::ConcurrentContext::ConcurrentContext()']]], + ['cond_80',['cond',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4ffd524d6a5bedd1a303b63bdde6701c',1,'mlx::core::scheduler::StreamThread']]], + ['conditionaltype_81',['ConditionalType',['../struct_conditional_type.html',1,'']]], + ['conditionaltype_3c_20true_2c_20t_2c_20u_20_3e_82',['ConditionalType< true, T, U >',['../struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.html',1,'']]], + ['conj_83',['conj',['../namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3',1,'pocketfft::detail::conj()'],['../namespacemlx_1_1core_1_1simd.html#a660b79a51fb439f4aba91e2aea276300',1,'mlx::core::simd::conj()']]], + ['conjugate_84',['Conjugate',['../struct_conjugate.html',1,'Conjugate'],['../classmlx_1_1core_1_1_conjugate.html',1,'mlx::core::Conjugate'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html',1,'mlx::core::detail::Conjugate'],['../classmlx_1_1core_1_1_conjugate.html#a627f9e6a8729fb3ffb3ca3228d007c87',1,'mlx::core::Conjugate::Conjugate()']]], + ['conjugate_85',['conjugate',['../group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23',1,'mlx::core']]], + ['contiguous_86',['Contiguous',['../classmlx_1_1core_1_1_contiguous.html',1,'mlx::core::Contiguous'],['../classmlx_1_1core_1_1_contiguous.html#a3e83f414c02ae0b92a50b6f8e402e1c0',1,'mlx::core::Contiguous::Contiguous()']]], + ['contiguous_87',['contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#afd0ab11e7a486a2a8e50ee84b971ac8a',1,'mlx::core::array::Flags::contiguous'],['../group__ops.html#ga8ab10aa6c41416d739791164a52b25d5',1,'mlx::core::contiguous()']]], + ['contiguous_5fscan_88',['contiguous_scan',['../scan_8h.html#a60d279b9add7d56639bb209408f09d79',1,'scan.h']]], + ['contiguousallreduce_89',['ContiguousAllReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ae4e34c7154eb8dc47aa8503209730424',1,'mlx::core']]], + ['contiguousiterator_90',['ContiguousIterator',['../structmlx_1_1core_1_1_contiguous_iterator.html',1,'mlx::core::ContiguousIterator'],['../structmlx_1_1core_1_1_contiguous_iterator.html#a727442ddff5fd3c3ebe09b000a01c9d3',1,'mlx::core::ContiguousIterator::ContiguousIterator()'],['../structmlx_1_1core_1_1_contiguous_iterator.html#aa82bec516eb54656c74fdaa74de1d735',1,'mlx::core::ContiguousIterator::ContiguousIterator(const array &a)'],['../structmlx_1_1core_1_1_contiguous_iterator.html#a8760380bff7462a886e7a4edd2955375',1,'mlx::core::ContiguousIterator::ContiguousIterator(const Shape &shape, const Strides &strides, int dims)']]], + ['contiguousreduce_91',['ContiguousReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ad2547f25dffe8d8936dbec25601cfc84',1,'mlx::core']]], + ['contiguousstridedreduce_92',['ContiguousStridedReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ab48dac7508a2c790de1bdc33f29177ed',1,'mlx::core']]], + ['conv_93',['conv',['../namespacemlx_1_1core_1_1metal.html#ab1704e853394c725668c06752ebb5c24',1,'mlx::core::metal']]], + ['conv_2eh_94',['conv.h',['../conv_8h.html',1,'']]], + ['conv1d_95',['conv1d',['../group__ops.html#ga30d47e08093c03a3676f235f9f559411',1,'mlx::core']]], + ['conv2d_96',['conv2d',['../group__ops.html#ga73b02833229678786e7f302d458d5a83',1,'mlx::core']]], + ['conv2dgeneralbaseinfo_97',['Conv2DGeneralBaseInfo',['../structmlx_1_1steel_1_1_conv2_d_general_base_info.html',1,'mlx::steel']]], + ['conv2dgeneraljumpparams_98',['Conv2DGeneralJumpParams',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html',1,'mlx::steel']]], + ['conv2dinputblockloadergeneral_99',['Conv2DInputBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html',1,'mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1d83af561a483432bf8dcb42e734b23b',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::Conv2DInputBlockLoaderGeneral()']]], + ['conv2dinputblockloaderlargefilter_100',['Conv2DInputBlockLoaderLargeFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8755116a535539744e4947bc69f9c50f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::Conv2DInputBlockLoaderLargeFilter()']]], + ['conv2dinputblockloadersmallchannels_101',['Conv2DInputBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ab9fd3fdeab94470dde3326f1dd5c455a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::Conv2DInputBlockLoaderSmallChannels()']]], + ['conv2dinputblockloadersmallfilter_102',['Conv2DInputBlockLoaderSmallFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0a2cbf57c51cd928722e3f06aafcf933',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::Conv2DInputBlockLoaderSmallFilter()']]], + ['conv2dweightblockloader_103',['Conv2DWeightBlockLoader',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html',1,'mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a9a7dca3512b64cffb6eac305d795831c',1,'mlx::steel::Conv2DWeightBlockLoader::Conv2DWeightBlockLoader()']]], + ['conv2dweightblockloadergeneral_104',['Conv2DWeightBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ad0550fabbdc9297559381a5b488e9af1',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::Conv2DWeightBlockLoaderGeneral()']]], + ['conv2dweightblockloadersmallchannels_105',['Conv2DWeightBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae1806ea1c19713819dee83a38ab35fa6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::Conv2DWeightBlockLoaderSmallChannels()']]], + ['conv3d_106',['conv3d',['../group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca',1,'mlx::core']]], + ['conv_5fgeneral_107',['conv_general',['../group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e',1,'mlx::core::conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})'],['../group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d',1,'mlx::core::conv_general(const array &input, const array &weight, std::vector< int > stride={}, std::vector< int > padding={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})']]], + ['conv_5ftranspose1d_108',['conv_transpose1d',['../group__ops.html#gaa30bf1adcd78d1c2595d07b215731714',1,'mlx::core']]], + ['conv_5ftranspose2d_109',['conv_transpose2d',['../group__ops.html#gaebb59971cb9bc45005dc1d398e4f0a3d',1,'mlx::core']]], + ['conv_5ftranspose3d_110',['conv_transpose3d',['../group__ops.html#ga8db814da631d9cd32a8d6563bf4ac530',1,'mlx::core']]], + ['convolution_111',['Convolution',['../classmlx_1_1core_1_1_convolution.html',1,'mlx::core::Convolution'],['../classmlx_1_1core_1_1_convolution.html#a6f1de77b719bb13217b0d8c64cabb8ef',1,'mlx::core::Convolution::Convolution()']]], + ['copy_112',['Copy',['../classmlx_1_1core_1_1_copy.html',1,'mlx::core::Copy'],['../classmlx_1_1core_1_1_copy.html#a6243e044af119105ffaaed7d405cd584',1,'mlx::core::Copy::Copy()']]], + ['copy_113',['copy',['../namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2',1,'mlx::core::copy()'],['../namespacemlx_1_1core_1_1metal.html#aa215e631e2680f04a591b88d91571719',1,'mlx::core::metal::copy()'],['../group__ops.html#gae306e93af12f774bd80bad6c231b09d6',1,'mlx::core::copy()']]], + ['copy_2eh_114',['copy.h',['../common_2copy_8h.html',1,'(Global Namespace)'],['../cpu_2copy_8h.html',1,'(Global Namespace)'],['../metal_2copy_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2copy_8h.html',1,'(Global Namespace)']]], + ['copy_5fg_115',['copy_g',['../metal_2kernels_2copy_8h.html#a71e4103db4689d90ef6f9d5ba93604cf',1,'copy.h']]], + ['copy_5fg_5fnd1_116',['copy_g_nd1',['../metal_2kernels_2copy_8h.html#a232c5c6b8386cf8ecbf4cdadb6e4176e',1,'copy.h']]], + ['copy_5fg_5fnd2_117',['copy_g_nd2',['../metal_2kernels_2copy_8h.html#a39ec5b7b8351e4332b842982a2ee6260',1,'copy.h']]], + ['copy_5fg_5fnd3_118',['copy_g_nd3',['../metal_2kernels_2copy_8h.html#aab82689380897ff4716b5eafd6ef3ecc',1,'copy.h']]], + ['copy_5fgg_119',['copy_gg',['../metal_2kernels_2copy_8h.html#ade9a9eea9b8262a854a11721fe2bb9fa',1,'copy.h']]], + ['copy_5fgg_5fdynamic_120',['copy_gg_dynamic',['../metal_2kernels_2copy_8h.html#ad0f05a73165d4ee38c9f02c705ea6ca8',1,'copy.h']]], + ['copy_5fgg_5fdynamic_5fnd1_121',['copy_gg_dynamic_nd1',['../metal_2kernels_2copy_8h.html#a8548ea41cac179084ddd33d26921576f',1,'copy.h']]], + ['copy_5fgg_5fdynamic_5fnd2_122',['copy_gg_dynamic_nd2',['../metal_2kernels_2copy_8h.html#a9b9266ee25a4dbcbe4fde883b40170f1',1,'copy.h']]], + ['copy_5fgg_5fdynamic_5fnd3_123',['copy_gg_dynamic_nd3',['../metal_2kernels_2copy_8h.html#af33ccc02f10bcb5c19ea7b1dd0af4956',1,'copy.h']]], + ['copy_5fgg_5fnd1_124',['copy_gg_nd1',['../metal_2kernels_2copy_8h.html#a370d7bbba1a4b0d64da873bafd29a78b',1,'copy.h']]], + ['copy_5fgg_5fnd2_125',['copy_gg_nd2',['../metal_2kernels_2copy_8h.html#af0b06ac3a96852a64fa4274a94b58301',1,'copy.h']]], + ['copy_5fgg_5fnd3_126',['copy_gg_nd3',['../metal_2kernels_2copy_8h.html#a3f3836ad0b6545ec9b9e1864224f7a13',1,'copy.h']]], + ['copy_5fgpu_127',['copy_gpu',['../namespacemlx_1_1core.html#addaa46a13ac2deb1d9ce621338320e0e',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a6a6f4e46c8fc44fdc74c50ace02bcf38',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype)']]], + ['copy_5fgpu_5finplace_128',['copy_gpu_inplace',['../namespacemlx_1_1core.html#a473fb602368f6c73d9105c9a151c4c82',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s, const std::optional< array > &dynamic_i_offset=std::nullopt, const std::optional< array > &dynamic_o_offset=std::nullopt)'],['../namespacemlx_1_1core.html#a58ef0842dd1b8f79159d5fb6777d30a1',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a49fc043a981925b9be79e37fc415d966',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const Strides &i_strides, int64_t i_offset, CopyType ctype, const Stream &s)']]], + ['copy_5fhartley_129',['copy_hartley',['../namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#ae7b44d2773d9d06a9787aff01d66b3ed',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], + ['copy_5finplace_130',['copy_inplace',['../namespacemlx_1_1core.html#a98495894a796b2cc6d022e7a03432c64',1,'mlx::core::copy_inplace(const array &src, array &dst, CopyType ctype)'],['../namespacemlx_1_1core.html#ae85bafda5ab0b4b2289591260cf07685',1,'mlx::core::copy_inplace(const array &src, array &dst, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)']]], + ['copy_5finput_131',['copy_input',['../namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< cmplx< T > > &src, cmplx< vtype_t< T > > *dst)'],['../namespacepocketfft_1_1detail.html#a30fc708f9d8f9cfa74194925c7863c0a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, vtype_t< T > *dst)'],['../namespacepocketfft_1_1detail.html#a3387bd35f237870e42b8461769e6aec4',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, T *dst)']]], + ['copy_5foutput_132',['copy_output',['../namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const cmplx< vtype_t< T > > *src, ndarr< cmplx< T > > &dst)'],['../namespacepocketfft_1_1detail.html#a21980853aca4d92ed06e3dcffe7ef660',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#a310481c334e46674710ba794ad7403c0',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], + ['copy_5fs_133',['copy_s',['../metal_2kernels_2copy_8h.html#aef09f9b9475345b1bba121d037d222ea',1,'copy.h']]], + ['copy_5fs2_134',['copy_s2',['../metal_2kernels_2copy_8h.html#a8023e9335cc5334847a8d315042be3a3',1,'copy.h']]], + ['copy_5fshared_5fbuffer_135',['copy_shared_buffer',['../classmlx_1_1core_1_1array.html#ad2814dbffa5ad174d9c97a10bf4cf26b',1,'mlx::core::array::copy_shared_buffer(const array &other, const Strides &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a92974c656c35a972ad241f80584bbd29',1,'mlx::core::array::copy_shared_buffer(const array &other)']]], + ['copy_5fv_136',['copy_v',['../metal_2kernels_2copy_8h.html#ae26a13e0c8e6c15f7b10078e65970659',1,'copy.h']]], + ['copy_5fv2_137',['copy_v2',['../metal_2kernels_2copy_8h.html#aee14a5326f53d9b30b0b38e27d180ef3',1,'copy.h']]], + ['copytype_138',['CopyType',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337',1,'mlx::core']]], + ['core_20array_20operations_139',['Core array operations',['../group__ops.html',1,'']]], + ['cos_140',['Cos',['../struct_cos.html',1,'Cos'],['../classmlx_1_1core_1_1_cos.html',1,'mlx::core::Cos'],['../structmlx_1_1core_1_1detail_1_1_cos.html',1,'mlx::core::detail::Cos'],['../classmlx_1_1core_1_1_cos.html#a2acb9fcf0901462189c476756fd99995',1,'mlx::core::Cos::Cos()']]], + ['cos_141',['cos',['../namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d',1,'pocketfft::detail::cos()'],['../namespacemlx_1_1core_1_1simd.html#ab179f429e34cd6d5c37050ea7e7c54ad',1,'mlx::core::simd::cos()'],['../namespacemetal.html#a2fa4778a6fe2fa43253ea724e5a608a3',1,'metal::cos()'],['../namespacemetal_1_1fast.html#a75b6bb32fa3870eda46a7bfc9f481f88',1,'metal::fast::cos()'],['../namespacemetal_1_1precise.html#ac4941f62e7d8ab9d7cabbd967aa9f220',1,'metal::precise::cos()'],['../group__ops.html#ga39dfdf72b556012aa35ff27a94116e74',1,'mlx::core::cos()']]], + ['cosh_142',['Cosh',['../struct_cosh.html',1,'Cosh'],['../classmlx_1_1core_1_1_cosh.html',1,'mlx::core::Cosh'],['../structmlx_1_1core_1_1detail_1_1_cosh.html',1,'mlx::core::detail::Cosh'],['../classmlx_1_1core_1_1_cosh.html#a44e8ac2e09a55ec32e9dc6641eedc8f1',1,'mlx::core::Cosh::Cosh()']]], + ['cosh_143',['cosh',['../namespacemlx_1_1core_1_1simd.html#aedc18b6fdb820cce9125c977c02833aa',1,'mlx::core::simd::cosh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#aa5b4f7d3b776e8d16907e15a11800f01',1,'mlx::core::simd::cosh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ae1265896d855818d20f2de2a9ebb684a',1,'mlx::core::simd::cosh(Simd< T, 1 > in)'],['../namespacemetal.html#a8a68a88cc110830d057dbd71431b93c0',1,'metal::cosh()'],['../namespacemetal_1_1fast.html#a31544ad9de28012a4ddda86e3966a77e',1,'metal::fast::cosh()'],['../namespacemetal_1_1precise.html#a72d86d508300a9b58f4ccbbe70da4fbc',1,'metal::precise::cosh()'],['../group__ops.html#ga2181b71cda88007a3092be4795ff0715',1,'mlx::core::cosh()']]], + ['cosine_144',['cosine',['../structpocketfft_1_1detail_1_1_exec_dcst.html#a185023fc1e386cc8f233b79c49c1fd8a',1,'pocketfft::detail::ExecDcst']]], + ['cospi_145',['cospi',['../namespacemetal.html#a5c2f37939ad705ddea4409d3bedb8ce1',1,'metal::cospi()'],['../namespacemetal_1_1fast.html#a9906b41f75319b384ffb570cc94d67ce',1,'metal::fast::cospi()'],['../namespacemetal_1_1precise.html#a2392b78bd196efdbbac65901c4ab20e7',1,'metal::precise::cospi()']]], + ['cost_5fguess_146',['cost_guess',['../structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0',1,'pocketfft::detail::util']]], + ['count_5fdown_147',['count_down',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342',1,'pocketfft::detail::threading::latch']]], + ['cpu_148',['cpu',['../structmlx_1_1core_1_1_device.html#a69ee81924251dec96f1945c9d91506fd',1,'mlx::core::Device::cpu'],['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdbad9747e2da342bdb995f6389533ad1a3d',1,'mlx::core::Device::cpu']]], + ['cross_149',['cross',['../namespacemlx_1_1core_1_1linalg.html#abcda3fbda45183c21e7f27aa0dde64e6',1,'mlx::core::linalg']]], + ['cshape_150',['CShape',['../structmlx_1_1steel_1_1_c_shape.html',1,'mlx::steel']]], + ['ctile_151',['Ctile',['../structmlx_1_1steel_1_1_block_m_m_a.html#a21b0c40d16eced109bd3196186170bc6',1,'mlx::steel::BlockMMA']]], + ['cummax_152',['CumMax',['../struct_cum_max.html',1,'']]], + ['cummax_153',['cummax',['../group__ops.html#gaee37cac8476e8f8d666bcded5bc59143',1,'mlx::core']]], + ['cummin_154',['CumMin',['../struct_cum_min.html',1,'']]], + ['cummin_155',['cummin',['../group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8',1,'mlx::core']]], + ['cumprod_156',['CumProd',['../struct_cum_prod.html',1,'']]], + ['cumprod_157',['cumprod',['../group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f',1,'mlx::core']]], + ['cumprod_3c_20bool_20_3e_158',['CumProd< bool >',['../struct_cum_prod_3_01bool_01_4.html',1,'']]], + ['cumsum_159',['CumSum',['../struct_cum_sum.html',1,'']]], + ['cumsum_160',['cumsum',['../group__ops.html#gaddc825a5c173e195ab0fda83ad630420',1,'mlx::core']]], + ['custom_161',['Custom',['../classmlx_1_1core_1_1fast_1_1_custom.html',1,'mlx::core::fast::Custom'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a4186fea23f7156c38960426821fca313',1,'mlx::core::fast::Custom::Custom()']]], + ['custom_5ffunction_162',['custom_function',['../namespacemlx_1_1core.html#a8d3ca5fbaecdb995660c24cde5aeebaf',1,'mlx::core']]], + ['custom_5fvjp_163',['custom_vjp',['../namespacemlx_1_1core.html#a9290596250fa308df4c69b44483bb8aa',1,'mlx::core']]], + ['customkernel_164',['CustomKernel',['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html',1,'mlx::core::fast::CustomKernel'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a954893e07f0d36715b4e1e414b6f2153',1,'mlx::core::fast::CustomKernel::CustomKernel()']]], + ['customkernelshapeinfo_165',['CustomKernelShapeInfo',['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html',1,'mlx::core::fast']]], + ['customtransforms_166',['CustomTransforms',['../classmlx_1_1core_1_1_custom_transforms.html',1,'mlx::core::CustomTransforms'],['../classmlx_1_1core_1_1_custom_transforms.html#ab52abadb9c6f6db83d087c7b751be488',1,'mlx::core::CustomTransforms::CustomTransforms()']]] ]; diff --git a/docs/build/html/search/all_4.js b/docs/build/html/search/all_4.js index f983c80a6..ff3e543ed 100644 --- a/docs/build/html/search/all_4.js +++ b/docs/build/html/search/all_4.js @@ -12,58 +12,66 @@ var searchData= ['debug_5fset_5fstream_5fqueue_5flabel_9',['debug_set_stream_queue_label',['../namespacemlx_1_1core.html#a79817d2432e782e596c9c49a08b93be2',1,'mlx::core']]], ['decompose_5fhadamard_10',['decompose_hadamard',['../namespacemlx_1_1core.html#a3a8fe7ba84714dbb5fdc81e93a07abc8',1,'mlx::core']]], ['default_5f_11',['default_',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#ab5993daeed822c6b970caddab7e3fd90',1,'mlx::core::random::KeySequence']]], - ['default_5fdevice_12',['default_device',['../namespacemlx_1_1core.html#a0196171cfe6ee2953113abce597dc815',1,'mlx::core']]], - ['default_5fstream_13',['default_stream',['../namespacemlx_1_1core.html#ac198b7e282957c724c84a435e8f1215e',1,'mlx::core']]], - ['defaultcontiguousreduce_14',['DefaultContiguousReduce',['../structmlx_1_1core_1_1_default_contiguous_reduce.html',1,'mlx::core::DefaultContiguousReduce< T, U, Op >'],['../structmlx_1_1core_1_1_default_contiguous_reduce.html#aeb4fb7fa1a4c8e7d1da1f450ce95c57f',1,'mlx::core::DefaultContiguousReduce::DefaultContiguousReduce()']]], - ['defaultstridedreduce_15',['DefaultStridedReduce',['../structmlx_1_1core_1_1_default_strided_reduce.html',1,'mlx::core::DefaultStridedReduce< T, U, Op >'],['../structmlx_1_1core_1_1_default_strided_reduce.html#a477e5dd0dd33071e48769d11d19a13fb',1,'mlx::core::DefaultStridedReduce::DefaultStridedReduce()']]], - ['define_5fdefault_5fis_5fequivalent_16',['DEFINE_DEFAULT_IS_EQUIVALENT',['../primitives_8h.html#a0fb9d19207dc4869aca35abfbdf4d70a',1,'primitives.h']]], - ['define_5fgrads_17',['DEFINE_GRADS',['../primitives_8h.html#a77abdcb55bc2eb0f9a45edc5ee639bf6',1,'primitives.h']]], - ['define_5finput_5foutput_5fshape_18',['DEFINE_INPUT_OUTPUT_SHAPE',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a2965dbda1bed67128e97c3c5d864c82f',1,'mlx::core::fast::RMSNorm::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#afd0818925ffea79f4e3dda0dd8cf0366',1,'mlx::core::fast::LayerNorm::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a361cc8e0e56ff45ec98dbf81ed8eff2c',1,'mlx::core::fast::RoPE::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a64d2ce4b46b529a6a9ef068947bc623e',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_INPUT_OUTPUT_SHAPE()'],['../primitives_8h.html#a649a06267b75e007224ea4ddefedb999',1,'DEFINE_INPUT_OUTPUT_SHAPE: primitives.h']]], - ['define_5fprint_19',['DEFINE_PRINT',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a8af1e90d4aa56f31ec40ad152ebd2421',1,'mlx::core::distributed::AllGather::DEFINE_PRINT()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a31bf76e24cf3836cf1fd26da30712e31',1,'mlx::core::distributed::Send::DEFINE_PRINT()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a7a0cad13da7cf8e565934318a2bc34f1',1,'mlx::core::distributed::Recv::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae6eea81b5e3789c2f6f376cc07f0a47c',1,'mlx::core::fast::RMSNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a9895733eab845e11484d86cf6ecedced',1,'mlx::core::fast::RMSNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a467fcf02b3ddf1d8b6d476b244ae3568',1,'mlx::core::fast::LayerNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a5ab3eb5402c7e8060916056eb2b7887f',1,'mlx::core::fast::LayerNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a2b06fe64fa8feca65140632087065e16',1,'mlx::core::fast::RoPE::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a6cc2092fa5b8e7585921b8e0f3ec3db7',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a4b8f1b1f633002c8ca6fa8f0ef4dd587',1,'mlx::core::fast::AffineQuantize::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a116ecf31c8672c94e5ea06c1d43e9534',1,'mlx::core::fast::CustomKernel::DEFINE_PRINT()'],['../primitives_8h.html#a1d3a37af519e16f6a703b1e9ebd0f592',1,'DEFINE_PRINT: primitives.h']]], - ['define_5fsimd_5fexclusive_5fscan_20',['DEFINE_SIMD_EXCLUSIVE_SCAN',['../scan_8h.html#a185f66aac8c5317587e6abd43f3013fc',1,'scan.h']]], - ['define_5fsimd_5freduce_21',['DEFINE_SIMD_REDUCE',['../backend_2metal_2kernels_2reduction_2ops_8h.html#acacf99e0ba629ed062ccc3c2eba89b05',1,'ops.h']]], - ['define_5fsimd_5fscan_22',['DEFINE_SIMD_SCAN',['../scan_8h.html#a0d8d6a9b0f3a1263629380bda8eca7bc',1,'scan.h']]], - ['define_5fvmap_23',['DEFINE_VMAP',['../primitives_8h.html#adc0fbd79fe0d1114dc85da4ed99798bd',1,'primitives.h']]], - ['defines_2eh_24',['defines.h',['../defines_8h.html',1,'(Global Namespace)'],['../steel_2defines_8h.html',1,'(Global Namespace)']]], - ['degrees_25',['degrees',['../group__ops.html#ga3a70569b50e1083c5ded199d73fb960c',1,'mlx::core']]], - ['deleter_26',['Deleter',['../namespacemlx_1_1core.html#af834c1e18d6f11c4f233a2e1ce814a4b',1,'mlx::core']]], - ['denorm_5fmin_27',['denorm_min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a6a9dbcba4dd79cad50876dda506b9eed',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['depends_28',['Depends',['../classmlx_1_1core_1_1_depends.html',1,'mlx::core::Depends'],['../classmlx_1_1core_1_1_depends.html#a4ccb792c99f5d8d133d3fac29f7d3f62',1,'mlx::core::Depends::Depends()']]], - ['depends_29',['depends',['../group__ops.html#gac4a51a68fbe1725436b026d2fbb95759',1,'mlx::core']]], - ['dequantize_30',['dequantize',['../quantized_8h.html#aecff265b63566d0d5689cfc4e5b037d2',1,'dequantize(): quantized.h'],['../group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077',1,'mlx::core::dequantize()']]], - ['detach_31',['detach',['../classmlx_1_1core_1_1array.html#a84948c29df8c957904919c8602692bd2',1,'mlx::core::array']]], - ['device_32',['Device',['../structmlx_1_1core_1_1_device.html',1,'mlx::core::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html',1,'mlx::core::metal::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html#ae0db74570eb4b19d8cf19774db91bfd6',1,'mlx::core::metal::Device::Device()'],['../classmlx_1_1core_1_1metal_1_1_device.html#abf59a4addb5473f9e814e3651ba85f06',1,'mlx::core::metal::Device::Device(const Device &)=delete'],['../structmlx_1_1core_1_1_device.html#a481ccfb94d689994396bd353e966b489',1,'mlx::core::Device::Device()']]], - ['device_33',['device',['../structmlx_1_1core_1_1_stream.html#a406b1b0162287a4162fab1f70e2ff3bb',1,'mlx::core::Stream::device'],['../classmlx_1_1core_1_1_primitive.html#a8ae61e3289c4134232a69295268f8261',1,'mlx::core::Primitive::device()'],['../namespacemlx_1_1core_1_1metal.html#a910797b74824e6ee576fbb533dee8b57',1,'mlx::core::metal::device()']]], - ['device_2eh_34',['device.h',['../backend_2metal_2device_8h.html',1,'(Global Namespace)'],['../device_8h.html',1,'(Global Namespace)']]], - ['device_5finfo_35',['device_info',['../namespacemlx_1_1core_1_1metal.html#a6ad19c44efabb7423f973407926ead61',1,'mlx::core::metal']]], - ['devicestream_36',['DeviceStream',['../structmlx_1_1core_1_1metal_1_1_device_stream.html',1,'mlx::core::metal::DeviceStream'],['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a573326bc8b48e39076850c7bf52ad0d7',1,'mlx::core::metal::DeviceStream::DeviceStream()']]], - ['devicetype_37',['DeviceType',['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdb',1,'mlx::core::Device']]], - ['diag_38',['diag',['../group__ops.html#ga11af511875640e1fa88e0ca87e199344',1,'mlx::core']]], - ['diagonal_39',['diagonal',['../group__ops.html#ga9236b085a88ead3128ed8079d009cac6',1,'mlx::core']]], - ['difference_5ftype_40',['difference_type',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#adcee44c77980fc2370a2c31e203aead5',1,'mlx::core::array::ArrayIterator']]], - ['digits_41',['digits',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#af6a681edff230c8d734a1feefb8d1879',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['digits10_42',['digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a0f48dd0c8a2d2dfa825067fb212b2e6b',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['dim_43',['dim',['../struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364',1,'LoopedElemToLoc']]], - ['disable_5fcompile_44',['disable_compile',['../namespacemlx_1_1core.html#a5f5fea955057bb3842b271b037909e66',1,'mlx::core']]], - ['disabled_45',['disabled',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a075ae3d2fc31640504f814f60e5ef713',1,'mlx::core']]], - ['dispatch_5fthreadgroups_46',['dispatch_threadgroups',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a85796b2bf41dbf347ae0978d4660600d',1,'mlx::core::metal::CommandEncoder::dispatch_threadgroups()'],['../structmlx_1_1core_1_1_command_encoder.html#a85796b2bf41dbf347ae0978d4660600d',1,'mlx::core::CommandEncoder::dispatch_threadgroups()']]], - ['dispatch_5fthreads_47',['dispatch_threads',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a0a8501b940e5a347475fa4bc38fb4c05',1,'mlx::core::metal::CommandEncoder::dispatch_threads()'],['../structmlx_1_1core_1_1_command_encoder.html#a0a8501b940e5a347475fa4bc38fb4c05',1,'mlx::core::CommandEncoder::dispatch_threads()']]], - ['distprimitive_48',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html',1,'mlx::core::distributed::DistPrimitive'],['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8c54166951522c2a52ef39fce8c87f8f',1,'mlx::core::distributed::DistPrimitive::DistPrimitive()']]], - ['distributed_2eh_49',['distributed.h',['../distributed_8h.html',1,'']]], - ['distributed_5fimpl_2eh_50',['distributed_impl.h',['../distributed__impl_8h.html',1,'']]], - ['divide_51',['Divide',['../struct_divide.html',1,'Divide'],['../structmlx_1_1core_1_1detail_1_1_divide.html',1,'mlx::core::detail::Divide'],['../classmlx_1_1core_1_1_divide.html',1,'mlx::core::Divide'],['../classmlx_1_1core_1_1_divide.html#a62fc71e8998be65ff18285dbbd21eedb',1,'mlx::core::Divide::Divide()']]], - ['divide_52',['divide',['../namespacemetal.html#a2aea493fc1a874970b77ed0031e965df',1,'metal::divide()'],['../namespacemetal_1_1fast.html#ae70bc2185e4649369cf7b15f5e1d48be',1,'metal::fast::divide()'],['../namespacemetal_1_1precise.html#aec0982cdb96a08b61f51129150d82e9d',1,'metal::precise::divide()'],['../group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f',1,'mlx::core::divide()']]], - ['divmod_53',['DivMod',['../struct_div_mod.html',1,'DivMod'],['../classmlx_1_1core_1_1_div_mod.html',1,'mlx::core::DivMod'],['../classmlx_1_1core_1_1_div_mod.html#a859e3b6149cdceab1c7ccfd2246fb826',1,'mlx::core::DivMod::DivMod()']]], - ['divmod_54',['divmod',['../group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894',1,'mlx::core']]], - ['divop_55',['DivOp',['../struct_div_op.html',1,'']]], - ['do_5faxpby_56',['do_axpby',['../steel__gemm__fused_8h.html#a703f06c849c89c37af7b1d27b0804a29',1,'steel_gemm_fused.h']]], - ['do_5fgather_57',['do_gather',['../steel__gemm__fused_8h.html#a60efac3ac3b7cd64d096bbae38a3ac69',1,'steel_gemm_fused.h']]], - ['do_5fread_58',['do_read',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a13eb86acf6abe288c19645935a47d2ad',1,'mlx::steel::Conv2DWeightBlockLoader::do_read'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a640155880483e1042ec5f647b9adaac6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::do_read']]], - ['dst_59',['dst',['../struct_quantized_block_loader.html#a9857214690fe6abad0e19d1045152f83',1,'QuantizedBlockLoader::dst'],['../structmlx_1_1steel_1_1_block_loader.html#af1c6c35a42e9da4408c1013ff1741bc2',1,'mlx::steel::BlockLoader::dst'],['../structmlx_1_1steel_1_1_block_loader_t.html#a6eb4e566b687395e27f290da288362db',1,'mlx::steel::BlockLoaderT::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ae048eb79f8b8d98f0fe8805c30fbb09f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8598bf23a2bce6af13c876cbfa76449f',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aea6494838175225d02cbc7768a646ec7',1,'mlx::steel::Conv2DWeightBlockLoader::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a59a4fffc1dc2f3fadfb3fdd1b886da70',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a24e20e4c1dd1ebf9534bfa2b3e050ed3',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aa84c4ad43a5defb83ba1a5f49a7adb2a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8474daf268013e138a84fc1c4bff7352',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::dst'],['../namespacepocketfft_1_1detail.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::detail::dst()'],['../namespacepocketfft.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::dst()']]], - ['dst_5fld_60',['dst_ld',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a91192d512e7a18c2d16a139065000959',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a9e59da7e4436e61b2d3c3f982355910b',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a0ff5a6d503e0bbac4634030a75ab818d',1,'mlx::steel::Conv2DWeightBlockLoader::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ae71570942c7b0ad8e67c62662b336c4a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ac18eeebea26cc6da434ead6eb4397350',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a07c85eab8cbf7b02c60df29cf32031ef',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aae121ca6016fc6c7255027b3641f3a09',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::dst_ld']]], - ['dtype_61',['Dtype',['../structmlx_1_1core_1_1_dtype.html',1,'mlx::core::Dtype'],['../structmlx_1_1core_1_1_dtype.html#aec17f0a4a51729e5ac40b62f0aa765d1',1,'mlx::core::Dtype::Dtype()']]], - ['dtype_62',['dtype',['../structmlx_1_1core_1_1finfo.html#a4edcbcfae55c1ef3cb8e61d427ac9124',1,'mlx::core::finfo::dtype'],['../classmlx_1_1core_1_1array.html#ae29e7d6fbfbea1e5e321a8d1ea3cfacd',1,'mlx::core::array::dtype()']]], - ['dtype_2eh_63',['dtype.h',['../dtype_8h.html',1,'']]], - ['dynamicslice_64',['DynamicSlice',['../classmlx_1_1core_1_1_dynamic_slice.html',1,'mlx::core::DynamicSlice'],['../classmlx_1_1core_1_1_dynamic_slice.html#a97f23f7d45b69219dee1a208d9a3063b',1,'mlx::core::DynamicSlice::DynamicSlice()']]], - ['dynamicsliceupdate_65',['DynamicSliceUpdate',['../classmlx_1_1core_1_1_dynamic_slice_update.html',1,'mlx::core::DynamicSliceUpdate'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a16bbd8d756598cf620e3b3c95dd23213',1,'mlx::core::DynamicSliceUpdate::DynamicSliceUpdate()']]] + ['default_5fbinary_12',['DEFAULT_BINARY',['../base__simd_8h.html#a1d01fad7230543d56f4facf17dcc4047',1,'base_simd.h']]], + ['default_5fbinary_5fop_13',['DEFAULT_BINARY_OP',['../cpu_2binary__ops_8h.html#a3f97e0a58306201b54eb73fac0affc8a',1,'binary_ops.h']]], + ['default_5fbool_5fop_14',['DEFAULT_BOOL_OP',['../cpu_2binary__ops_8h.html#a98fb0ee78e9fd26ad6d0a2b365f7caa7',1,'binary_ops.h']]], + ['default_5fcomparisons_15',['DEFAULT_COMPARISONS',['../base__simd_8h.html#af5153f51cdd18d56d1de24b5da9f4366',1,'base_simd.h']]], + ['default_5fdevice_16',['default_device',['../namespacemlx_1_1core.html#a0196171cfe6ee2953113abce597dc815',1,'mlx::core']]], + ['default_5fop_17',['DEFAULT_OP',['../cpu_2unary__ops_8h.html#a83702f31e6dbd79c339a6aad67319f64',1,'unary_ops.h']]], + ['default_5freduction_18',['DEFAULT_REDUCTION',['../base__simd_8h.html#a0adf6d25084019eff671abc59031573e',1,'base_simd.h']]], + ['default_5fstream_19',['default_stream',['../namespacemlx_1_1core.html#ac198b7e282957c724c84a435e8f1215e',1,'mlx::core']]], + ['default_5funary_20',['DEFAULT_UNARY',['../base__simd_8h.html#a122d6a2fa4dcfe78b221e52155419124',1,'base_simd.h']]], + ['define_5fdefault_5fis_5fequivalent_21',['DEFINE_DEFAULT_IS_EQUIVALENT',['../primitives_8h.html#a0fb9d19207dc4869aca35abfbdf4d70a',1,'primitives.h']]], + ['define_5fgrads_22',['DEFINE_GRADS',['../primitives_8h.html#a77abdcb55bc2eb0f9a45edc5ee639bf6',1,'primitives.h']]], + ['define_5finput_5foutput_5fshape_23',['DEFINE_INPUT_OUTPUT_SHAPE',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a2965dbda1bed67128e97c3c5d864c82f',1,'mlx::core::fast::RMSNorm::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#afd0818925ffea79f4e3dda0dd8cf0366',1,'mlx::core::fast::LayerNorm::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a361cc8e0e56ff45ec98dbf81ed8eff2c',1,'mlx::core::fast::RoPE::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a64d2ce4b46b529a6a9ef068947bc623e',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_INPUT_OUTPUT_SHAPE()'],['../primitives_8h.html#a649a06267b75e007224ea4ddefedb999',1,'DEFINE_INPUT_OUTPUT_SHAPE: primitives.h']]], + ['define_5fneon_5fbinary_5fop_24',['DEFINE_NEON_BINARY_OP',['../neon__fp16__simd_8h.html#a63ba6e115ef14e134081310f7d4579a5',1,'neon_fp16_simd.h']]], + ['define_5fneon_5fcomparison_25',['DEFINE_NEON_COMPARISON',['../neon__fp16__simd_8h.html#a9fcd94c7369a6b4437f9c310a805c79d',1,'neon_fp16_simd.h']]], + ['define_5fneon_5funary_5fop_26',['DEFINE_NEON_UNARY_OP',['../neon__fp16__simd_8h.html#af8138a463be93b9e0c9b685e94a1fd00',1,'neon_fp16_simd.h']]], + ['define_5fprint_27',['DEFINE_PRINT',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a8af1e90d4aa56f31ec40ad152ebd2421',1,'mlx::core::distributed::AllGather::DEFINE_PRINT()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a31bf76e24cf3836cf1fd26da30712e31',1,'mlx::core::distributed::Send::DEFINE_PRINT()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a7a0cad13da7cf8e565934318a2bc34f1',1,'mlx::core::distributed::Recv::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae6eea81b5e3789c2f6f376cc07f0a47c',1,'mlx::core::fast::RMSNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a9895733eab845e11484d86cf6ecedced',1,'mlx::core::fast::RMSNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a467fcf02b3ddf1d8b6d476b244ae3568',1,'mlx::core::fast::LayerNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a5ab3eb5402c7e8060916056eb2b7887f',1,'mlx::core::fast::LayerNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a2b06fe64fa8feca65140632087065e16',1,'mlx::core::fast::RoPE::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a6cc2092fa5b8e7585921b8e0f3ec3db7',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a4b8f1b1f633002c8ca6fa8f0ef4dd587',1,'mlx::core::fast::AffineQuantize::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a116ecf31c8672c94e5ea06c1d43e9534',1,'mlx::core::fast::CustomKernel::DEFINE_PRINT()'],['../primitives_8h.html#a1d3a37af519e16f6a703b1e9ebd0f592',1,'DEFINE_PRINT: primitives.h']]], + ['define_5fsimd_5fexclusive_5fscan_28',['DEFINE_SIMD_EXCLUSIVE_SCAN',['../scan_8h.html#a185f66aac8c5317587e6abd43f3013fc',1,'scan.h']]], + ['define_5fsimd_5freduce_29',['DEFINE_SIMD_REDUCE',['../backend_2metal_2kernels_2reduction_2ops_8h.html#acacf99e0ba629ed062ccc3c2eba89b05',1,'ops.h']]], + ['define_5fsimd_5fscan_30',['DEFINE_SIMD_SCAN',['../scan_8h.html#a0d8d6a9b0f3a1263629380bda8eca7bc',1,'scan.h']]], + ['define_5fvmap_31',['DEFINE_VMAP',['../primitives_8h.html#adc0fbd79fe0d1114dc85da4ed99798bd',1,'primitives.h']]], + ['defines_2eh_32',['defines.h',['../defines_8h.html',1,'(Global Namespace)'],['../steel_2defines_8h.html',1,'(Global Namespace)']]], + ['degrees_33',['degrees',['../group__ops.html#ga3a70569b50e1083c5ded199d73fb960c',1,'mlx::core']]], + ['deleter_34',['Deleter',['../namespacemlx_1_1core.html#af834c1e18d6f11c4f233a2e1ce814a4b',1,'mlx::core']]], + ['denorm_5fmin_35',['denorm_min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a6a9dbcba4dd79cad50876dda506b9eed',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['depends_36',['Depends',['../classmlx_1_1core_1_1_depends.html',1,'mlx::core::Depends'],['../classmlx_1_1core_1_1_depends.html#a4ccb792c99f5d8d133d3fac29f7d3f62',1,'mlx::core::Depends::Depends()']]], + ['depends_37',['depends',['../group__ops.html#gac4a51a68fbe1725436b026d2fbb95759',1,'mlx::core']]], + ['dequantize_38',['dequantize',['../quantized_8h.html#aecff265b63566d0d5689cfc4e5b037d2',1,'dequantize(): quantized.h'],['../group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077',1,'mlx::core::dequantize()']]], + ['detach_39',['detach',['../classmlx_1_1core_1_1array.html#a84948c29df8c957904919c8602692bd2',1,'mlx::core::array']]], + ['device_40',['Device',['../structmlx_1_1core_1_1_device.html',1,'mlx::core::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html',1,'mlx::core::metal::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html#ae0db74570eb4b19d8cf19774db91bfd6',1,'mlx::core::metal::Device::Device()'],['../classmlx_1_1core_1_1metal_1_1_device.html#abf59a4addb5473f9e814e3651ba85f06',1,'mlx::core::metal::Device::Device(const Device &)=delete'],['../structmlx_1_1core_1_1_device.html#a481ccfb94d689994396bd353e966b489',1,'mlx::core::Device::Device()']]], + ['device_41',['device',['../structmlx_1_1core_1_1_stream.html#a406b1b0162287a4162fab1f70e2ff3bb',1,'mlx::core::Stream::device'],['../classmlx_1_1core_1_1_primitive.html#a8ae61e3289c4134232a69295268f8261',1,'mlx::core::Primitive::device()'],['../namespacemlx_1_1core_1_1metal.html#a910797b74824e6ee576fbb533dee8b57',1,'mlx::core::metal::device()']]], + ['device_2eh_42',['device.h',['../backend_2metal_2device_8h.html',1,'(Global Namespace)'],['../device_8h.html',1,'(Global Namespace)']]], + ['device_5finfo_43',['device_info',['../namespacemlx_1_1core_1_1metal.html#a6ad19c44efabb7423f973407926ead61',1,'mlx::core::metal']]], + ['devicestream_44',['DeviceStream',['../structmlx_1_1core_1_1metal_1_1_device_stream.html',1,'mlx::core::metal::DeviceStream'],['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a573326bc8b48e39076850c7bf52ad0d7',1,'mlx::core::metal::DeviceStream::DeviceStream()']]], + ['devicetype_45',['DeviceType',['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdb',1,'mlx::core::Device']]], + ['diag_46',['diag',['../group__ops.html#ga11af511875640e1fa88e0ca87e199344',1,'mlx::core']]], + ['diagonal_47',['diagonal',['../group__ops.html#ga9236b085a88ead3128ed8079d009cac6',1,'mlx::core']]], + ['difference_5ftype_48',['difference_type',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#adcee44c77980fc2370a2c31e203aead5',1,'mlx::core::array::ArrayIterator']]], + ['digits_49',['digits',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#af6a681edff230c8d734a1feefb8d1879',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['digits10_50',['digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a0f48dd0c8a2d2dfa825067fb212b2e6b',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['dim_51',['dim',['../struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364',1,'LoopedElemToLoc::dim'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a7be6bf560080472d61e74b522979ef1e',1,'LoopedElemToLoc< 1, OffsetT, true >::dim'],['../struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364',1,'LoopedElemToLoc< 1, OffsetT, false >::dim'],['../struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364',1,'LoopedElemToLoc< 1, OffsetT, true >::dim']]], + ['disable_5fcompile_52',['disable_compile',['../namespacemlx_1_1core.html#a5f5fea955057bb3842b271b037909e66',1,'mlx::core']]], + ['disabled_53',['disabled',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a075ae3d2fc31640504f814f60e5ef713',1,'mlx::core']]], + ['dispatch_5fthreadgroups_54',['dispatch_threadgroups',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a85796b2bf41dbf347ae0978d4660600d',1,'mlx::core::metal::CommandEncoder::dispatch_threadgroups()'],['../structmlx_1_1core_1_1_command_encoder.html#a85796b2bf41dbf347ae0978d4660600d',1,'mlx::core::CommandEncoder::dispatch_threadgroups()']]], + ['dispatch_5fthreads_55',['dispatch_threads',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a0a8501b940e5a347475fa4bc38fb4c05',1,'mlx::core::metal::CommandEncoder::dispatch_threads()'],['../structmlx_1_1core_1_1_command_encoder.html#a0a8501b940e5a347475fa4bc38fb4c05',1,'mlx::core::CommandEncoder::dispatch_threads()']]], + ['distprimitive_56',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html',1,'mlx::core::distributed::DistPrimitive'],['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8c54166951522c2a52ef39fce8c87f8f',1,'mlx::core::distributed::DistPrimitive::DistPrimitive()']]], + ['distributed_2eh_57',['distributed.h',['../distributed_8h.html',1,'']]], + ['distributed_5fimpl_2eh_58',['distributed_impl.h',['../distributed__impl_8h.html',1,'']]], + ['divide_59',['Divide',['../struct_divide.html',1,'Divide'],['../structmlx_1_1core_1_1detail_1_1_divide.html',1,'mlx::core::detail::Divide'],['../classmlx_1_1core_1_1_divide.html',1,'mlx::core::Divide'],['../classmlx_1_1core_1_1_divide.html#a62fc71e8998be65ff18285dbbd21eedb',1,'mlx::core::Divide::Divide()']]], + ['divide_60',['divide',['../namespacemetal.html#a2aea493fc1a874970b77ed0031e965df',1,'metal::divide()'],['../namespacemetal_1_1fast.html#ae70bc2185e4649369cf7b15f5e1d48be',1,'metal::fast::divide()'],['../namespacemetal_1_1precise.html#aec0982cdb96a08b61f51129150d82e9d',1,'metal::precise::divide()'],['../group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f',1,'mlx::core::divide()']]], + ['divmod_61',['DivMod',['../struct_div_mod.html',1,'DivMod'],['../classmlx_1_1core_1_1_div_mod.html',1,'mlx::core::DivMod'],['../classmlx_1_1core_1_1_div_mod.html#a859e3b6149cdceab1c7ccfd2246fb826',1,'mlx::core::DivMod::DivMod()']]], + ['divmod_62',['divmod',['../group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894',1,'mlx::core']]], + ['divop_63',['DivOp',['../struct_div_op.html',1,'']]], + ['do_5faxpby_64',['do_axpby',['../steel__gemm__fused_8h.html#a703f06c849c89c37af7b1d27b0804a29',1,'steel_gemm_fused.h']]], + ['do_5fgather_65',['do_gather',['../steel__gemm__fused_8h.html#a60efac3ac3b7cd64d096bbae38a3ac69',1,'steel_gemm_fused.h']]], + ['do_5fread_66',['do_read',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a13eb86acf6abe288c19645935a47d2ad',1,'mlx::steel::Conv2DWeightBlockLoader::do_read'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a640155880483e1042ec5f647b9adaac6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::do_read']]], + ['dst_67',['dst',['../struct_quantized_block_loader.html#a9857214690fe6abad0e19d1045152f83',1,'QuantizedBlockLoader::dst'],['../structmlx_1_1steel_1_1_block_loader.html#af1c6c35a42e9da4408c1013ff1741bc2',1,'mlx::steel::BlockLoader::dst'],['../structmlx_1_1steel_1_1_block_loader_t.html#a6eb4e566b687395e27f290da288362db',1,'mlx::steel::BlockLoaderT::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ae048eb79f8b8d98f0fe8805c30fbb09f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8598bf23a2bce6af13c876cbfa76449f',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aea6494838175225d02cbc7768a646ec7',1,'mlx::steel::Conv2DWeightBlockLoader::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a59a4fffc1dc2f3fadfb3fdd1b886da70',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a24e20e4c1dd1ebf9534bfa2b3e050ed3',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aa84c4ad43a5defb83ba1a5f49a7adb2a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8474daf268013e138a84fc1c4bff7352',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::dst'],['../namespacepocketfft_1_1detail.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::detail::dst()'],['../namespacepocketfft.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::dst()']]], + ['dst_5fld_68',['dst_ld',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a91192d512e7a18c2d16a139065000959',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a9e59da7e4436e61b2d3c3f982355910b',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a0ff5a6d503e0bbac4634030a75ab818d',1,'mlx::steel::Conv2DWeightBlockLoader::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ae71570942c7b0ad8e67c62662b336c4a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ac18eeebea26cc6da434ead6eb4397350',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a07c85eab8cbf7b02c60df29cf32031ef',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aae121ca6016fc6c7255027b3641f3a09',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::dst_ld']]], + ['dtype_69',['Dtype',['../structmlx_1_1core_1_1_dtype.html',1,'mlx::core::Dtype'],['../structmlx_1_1core_1_1_dtype.html#aec17f0a4a51729e5ac40b62f0aa765d1',1,'mlx::core::Dtype::Dtype()']]], + ['dtype_70',['dtype',['../structmlx_1_1core_1_1finfo.html#a4edcbcfae55c1ef3cb8e61d427ac9124',1,'mlx::core::finfo::dtype'],['../classmlx_1_1core_1_1array.html#ae29e7d6fbfbea1e5e321a8d1ea3cfacd',1,'mlx::core::array::dtype()']]], + ['dtype_2eh_71',['dtype.h',['../dtype_8h.html',1,'']]], + ['dynamicslice_72',['DynamicSlice',['../classmlx_1_1core_1_1_dynamic_slice.html',1,'mlx::core::DynamicSlice'],['../classmlx_1_1core_1_1_dynamic_slice.html#a97f23f7d45b69219dee1a208d9a3063b',1,'mlx::core::DynamicSlice::DynamicSlice()']]], + ['dynamicsliceupdate_73',['DynamicSliceUpdate',['../classmlx_1_1core_1_1_dynamic_slice_update.html',1,'mlx::core::DynamicSliceUpdate'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a16bbd8d756598cf620e3b3c95dd23213',1,'mlx::core::DynamicSliceUpdate::DynamicSliceUpdate()']]] ]; diff --git a/docs/build/html/search/all_5.js b/docs/build/html/search/all_5.js index 5c00a32f0..0ce15da40 100644 --- a/docs/build/html/search/all_5.js +++ b/docs/build/html/search/all_5.js @@ -23,49 +23,51 @@ var searchData= ['enable_5fcompile_20',['enable_compile',['../namespacemlx_1_1core.html#a1983a2466bff3bae4d23cf34bd0946c9',1,'mlx::core']]], ['enable_5ffor_5farrays_5ft_21',['enable_for_arrays_t',['../namespacemlx_1_1core.html#af89751d79339f3e4d9318ea97d64d114',1,'mlx::core']]], ['enabled_22',['enabled',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4aa10311459433adf322f2590a4987c423',1,'mlx::core']]], - ['encoder_23',['encoder',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a58e435217b9922f882507ebf48bfbbdd',1,'mlx::core::metal::DeviceStream']]], - ['end_24',['end',['../classmlx_1_1core_1_1array.html#a5daf64552fb450825c9b382f3a5fa2d4',1,'mlx::core::array']]], - ['end_5fencoding_25',['end_encoding',['../classmlx_1_1core_1_1metal_1_1_device.html#a60689f97347811b27e8c5ca23e0372bf',1,'mlx::core::metal::Device']]], - ['enqueue_26',['enqueue',['../class_thread_pool.html#a375fa2d63197282277be640b54e8a196',1,'ThreadPool::enqueue()'],['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4918720319cf224a1b4208568964c286',1,'mlx::core::scheduler::StreamThread::enqueue()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a03809c783bd1866362dc7cb9118abbcc',1,'mlx::core::scheduler::Scheduler::enqueue()'],['../namespacemlx_1_1core_1_1scheduler.html#aa2d4eacf5d5cbc778a51aafd4fd8e4d7',1,'mlx::core::scheduler::enqueue()']]], - ['epsilon_27',['epsilon',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a96c4197e3076f0aa9065370b8ece49ca',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['equal_28',['Equal',['../struct_equal.html',1,'Equal'],['../structmlx_1_1core_1_1detail_1_1_equal.html',1,'mlx::core::detail::Equal'],['../classmlx_1_1core_1_1_equal.html',1,'mlx::core::Equal'],['../classmlx_1_1core_1_1_equal.html#a4af81cf2dd071db5bbf8ce1df95fdf36',1,'mlx::core::Equal::Equal()']]], - ['equal_29',['equal',['../group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde',1,'mlx::core']]], - ['erase_30',['erase',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#ae136ad270522210c85c13cacf5165238',1,'mlx::core::metal::ResidencySet']]], - ['erf_31',['Erf',['../struct_erf.html',1,'Erf'],['../structmlx_1_1core_1_1detail_1_1_erf.html',1,'mlx::core::detail::Erf'],['../classmlx_1_1core_1_1_erf.html',1,'mlx::core::Erf'],['../classmlx_1_1core_1_1_erf.html#a702f76f848928d8d7d3d0881ac6e4c82',1,'mlx::core::Erf::Erf()']]], - ['erf_32',['erf',['../erf_8h.html#a6ce199ee56105c67adbf8c48c019a8b2',1,'erf(): erf.h'],['../group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0',1,'mlx::core::erf()']]], - ['erf_2eh_33',['erf.h',['../erf_8h.html',1,'']]], - ['erfinv_34',['ErfInv',['../struct_erf_inv.html',1,'ErfInv'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html',1,'mlx::core::detail::ErfInv'],['../classmlx_1_1core_1_1_erf_inv.html',1,'mlx::core::ErfInv'],['../classmlx_1_1core_1_1_erf_inv.html#a5d0279247b67da4592311559f04e1478',1,'mlx::core::ErfInv::ErfInv()']]], - ['erfinv_35',['erfinv',['../erf_8h.html#a1846e0d683c7aff826bb32addcc3b885',1,'erfinv(): erf.h'],['../group__ops.html#ga76fb9062c64264e34d2e07013390557c',1,'mlx::core::erfinv()']]], - ['eval_36',['eval',['../classmlx_1_1core_1_1array.html#a2820c45188071a22175e9fa42e10a49a',1,'mlx::core::array::eval()'],['../namespacemlx_1_1core.html#a7d6e097d8effed52f4713672e471f299',1,'mlx::core::eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#adb14f689c9f75f7901edb196c2bfb971',1,'mlx::core::eval(Arrays &&... outputs)']]], - ['eval_5fcpu_37',['eval_cpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#acdc1965ad64ee9ee6328fe150a97902e',1,'mlx::core::distributed::AllReduce::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ab721fe0072fffbddbc3c4334dd033ba5',1,'mlx::core::distributed::AllGather::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#af2620837bfc1b97217d006ed6e374051',1,'mlx::core::distributed::Send::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a3be84b08122a939edd6062d26261358a',1,'mlx::core::distributed::Recv::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a7da6e0cfd630958d9633b2e2bd97a54f',1,'mlx::core::fast::RMSNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#adfc1d52bc266466ab29ee45fd8fab439',1,'mlx::core::fast::RMSNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a5d7a4c1c9ee84e327d1c371733108c05',1,'mlx::core::fast::LayerNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a0d8c4c6e7462befc38f7e08244fa1c2b',1,'mlx::core::fast::LayerNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a05a7d595c6b9dadf7ddfd6e3fd402f0e',1,'mlx::core::fast::RoPE::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ae20851e002f7fcb6d4f97817596f6328',1,'mlx::core::fast::ScaledDotProductAttention::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a3b5d628628d245b38911118d4a0ff9fd',1,'mlx::core::fast::AffineQuantize::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a4ad1b7a9919753c759093f3e21a15bad',1,'mlx::core::fast::CustomKernel::eval_cpu()'],['../classmlx_1_1core_1_1_primitive.html#a1596dc50b910538eae14878e98f07575',1,'mlx::core::Primitive::eval_cpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a7e8f6f5d6ae0a33f6abc0f5a46e0b132',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#aa0ed6e32c36200a3ff9bc592c9b300db',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0d3e697496ef8e842d21195cb3c14e60',1,'mlx::core::Abs::eval_cpu()'],['../classmlx_1_1core_1_1_add.html#a5bacfc51dfa2a5a931bad2dd7bdc7a5f',1,'mlx::core::Add::eval_cpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a15694e3bf2ed5c193237b2b9ca00867c',1,'mlx::core::AddMM::eval_cpu()'],['../classmlx_1_1core_1_1_arange.html#aba44432491cbd599bf72712f5f4267a1',1,'mlx::core::Arange::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a58dcba9e706cb12bab062bb7fa5fa006',1,'mlx::core::ArcCos::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#a0f6d989bcbbc38f15ef17a136879a9c9',1,'mlx::core::ArcCosh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sin.html#ab3542492c14021329788de8f2a9be1e4',1,'mlx::core::ArcSin::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a52574b24d8d16839c58673f51f8ac066',1,'mlx::core::ArcSinh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a1211bc31241227528f04435239ddb9a3',1,'mlx::core::ArcTan::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a13094e6b702769928ca0da468f5ce45c',1,'mlx::core::ArcTan2::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a5af9224e1f1ffec412b0baa0af7e1ecd',1,'mlx::core::ArcTanh::eval_cpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a896f75c5325798ac3f9093f6a4581828',1,'mlx::core::ArgPartition::eval_cpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#ad8d48725623ede1ff654fa13eccf2287',1,'mlx::core::ArgReduce::eval_cpu()'],['../classmlx_1_1core_1_1_arg_sort.html#a022079683774bfeb531b3a002cff16fa',1,'mlx::core::ArgSort::eval_cpu()'],['../classmlx_1_1core_1_1_as_type.html#aa89dbf4d73b00c6a44cffd04d5bb228d',1,'mlx::core::AsType::eval_cpu()'],['../classmlx_1_1core_1_1_as_strided.html#acdd4705e4503ff0b124215c4676b4193',1,'mlx::core::AsStrided::eval_cpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a2194bf585213bda1b2966aa02d2fe283',1,'mlx::core::BitwiseBinary::eval_cpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aa85da478cdc6d4a97be06e5d4abee1f2',1,'mlx::core::BlockMaskedMM::eval_cpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#a62352074a480df0e1f879b0bae425730',1,'mlx::core::GatherMM::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a6423095cd28b2f90893c03166257a568',1,'mlx::core::BroadcastAxes::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast.html#a53d48d9778e2d4c24a124cd767900780',1,'mlx::core::Broadcast::eval_cpu()'],['../classmlx_1_1core_1_1_ceil.html#a9791801fff3f8b79944e15ac2a45a035',1,'mlx::core::Ceil::eval_cpu()'],['../classmlx_1_1core_1_1_compiled.html#ac45b1d0fedd85feefbff7ce7e168b151',1,'mlx::core::Compiled::eval_cpu()'],['../classmlx_1_1core_1_1_concatenate.html#a609e76bede7fc5581ec84ddcb727a258',1,'mlx::core::Concatenate::eval_cpu()'],['../classmlx_1_1core_1_1_conjugate.html#ae39643e2178f442ffba05139f8609d61',1,'mlx::core::Conjugate::eval_cpu()'],['../classmlx_1_1core_1_1_contiguous.html#a742de24e6c0310cd85a606dec0cd8336',1,'mlx::core::Contiguous::eval_cpu()'],['../classmlx_1_1core_1_1_convolution.html#ac74256068da01730629109fa4fa8432b',1,'mlx::core::Convolution::eval_cpu()'],['../classmlx_1_1core_1_1_copy.html#af4a0ebec423e84ffe8083a5e9ed0d70c',1,'mlx::core::Copy::eval_cpu()'],['../classmlx_1_1core_1_1_cos.html#a061fc446268fe56237ae6b20ccf78152',1,'mlx::core::Cos::eval_cpu()'],['../classmlx_1_1core_1_1_cosh.html#ae8702df7e8f0e20cbeccb2a548961d3d',1,'mlx::core::Cosh::eval_cpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#adba1c40c77a2138df6b5f75483f62184',1,'mlx::core::CustomTransforms::eval_cpu()'],['../classmlx_1_1core_1_1_depends.html#a0c7ea6db97337591fa53c6e6bde41e5e',1,'mlx::core::Depends::eval_cpu()'],['../classmlx_1_1core_1_1_divide.html#a823443c2a8e8b81bbcaeee6ddbcdbf49',1,'mlx::core::Divide::eval_cpu()'],['../classmlx_1_1core_1_1_div_mod.html#ae350b7b93ad128e3133ee14f247193b3',1,'mlx::core::DivMod::eval_cpu()'],['../classmlx_1_1core_1_1_select.html#aa51aa36e0adbd69e0d23d7c7adf88de2',1,'mlx::core::Select::eval_cpu()'],['../classmlx_1_1core_1_1_remainder.html#ac6c6c86a0bf02e6e529eb87f6e617ccc',1,'mlx::core::Remainder::eval_cpu()'],['../classmlx_1_1core_1_1_equal.html#aabb8aa61fa581defddcdca1274b1b454',1,'mlx::core::Equal::eval_cpu()'],['../classmlx_1_1core_1_1_erf.html#a84ea16e43d5b7f83bbc2d5ece78a3fb6',1,'mlx::core::Erf::eval_cpu()'],['../classmlx_1_1core_1_1_erf_inv.html#af579627402af3249565134884701d39e',1,'mlx::core::ErfInv::eval_cpu()'],['../classmlx_1_1core_1_1_exp.html#a47934c5a5023bc7ae7ae89bff45ebb2c',1,'mlx::core::Exp::eval_cpu()'],['../classmlx_1_1core_1_1_expm1.html#ab9c8b7aa50fe4592d55f8957baac647a',1,'mlx::core::Expm1::eval_cpu()'],['../classmlx_1_1core_1_1_expand_dims.html#a34058a87582a6ab2e5d82a75bc713030',1,'mlx::core::ExpandDims::eval_cpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a6bc262a0c2b5d4fe655e3e2e0ff28635',1,'mlx::core::FFT::eval_cpu()'],['../classmlx_1_1core_1_1_flatten.html#a72ade7d22386b349712f6c7c1f619842',1,'mlx::core::Flatten::eval_cpu()'],['../classmlx_1_1core_1_1_floor.html#a1a7dc5f571b7b73e7ef3cbdc1dd1fcf7',1,'mlx::core::Floor::eval_cpu()'],['../classmlx_1_1core_1_1_full.html#a3dccd3756599d7fd018b2af0093b082c',1,'mlx::core::Full::eval_cpu()'],['../classmlx_1_1core_1_1_gather.html#a9ed5587f0d04b59a2b9186c0aac21290',1,'mlx::core::Gather::eval_cpu()'],['../classmlx_1_1core_1_1_greater.html#abe1c03f311d0e0b610f3392a6566f2ae',1,'mlx::core::Greater::eval_cpu()'],['../classmlx_1_1core_1_1_greater_equal.html#a15469125b9bea89b64bfeac01590c075',1,'mlx::core::GreaterEqual::eval_cpu()'],['../classmlx_1_1core_1_1_hadamard.html#ab27d6a9df42b3aab41ace3073a4c880d',1,'mlx::core::Hadamard::eval_cpu()'],['../classmlx_1_1core_1_1_imag.html#a17d1f1f9f8528668fcdf39b636720829',1,'mlx::core::Imag::eval_cpu()'],['../classmlx_1_1core_1_1_less.html#a32624124ffece066f496b3299056bcef',1,'mlx::core::Less::eval_cpu()'],['../classmlx_1_1core_1_1_less_equal.html#a55d1352b0e97841a92503bc57c19ed16',1,'mlx::core::LessEqual::eval_cpu()'],['../classmlx_1_1core_1_1_load.html#ada026ac30566f3109d8182e35d307c0a',1,'mlx::core::Load::eval_cpu()'],['../classmlx_1_1core_1_1_log.html#aadc7bb4cb24f3ecbbb9ed54a699ab74f',1,'mlx::core::Log::eval_cpu()'],['../classmlx_1_1core_1_1_log1p.html#a8192e5438de99c4cda056987935cba23',1,'mlx::core::Log1p::eval_cpu()'],['../classmlx_1_1core_1_1_logical_not.html#acf3f7b3b20ca69533536e0e0a05725b3',1,'mlx::core::LogicalNot::eval_cpu()'],['../classmlx_1_1core_1_1_logical_and.html#adbe1c1785af1a8b827289d22b0d170b3',1,'mlx::core::LogicalAnd::eval_cpu()'],['../classmlx_1_1core_1_1_logical_or.html#a13cd4cbf26589287e85aeaaca42d7f62',1,'mlx::core::LogicalOr::eval_cpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#abef17fb590b1a8d356f2a580e45d41f0',1,'mlx::core::LogAddExp::eval_cpu()'],['../classmlx_1_1core_1_1_matmul.html#a357a7f57a2a220a91977f810a69413fc',1,'mlx::core::Matmul::eval_cpu()'],['../classmlx_1_1core_1_1_maximum.html#a62b38fbe5f96db58c2b60165ac4eadcf',1,'mlx::core::Maximum::eval_cpu()'],['../classmlx_1_1core_1_1_minimum.html#a6b93f493ee87089943a8085fe59dfc6e',1,'mlx::core::Minimum::eval_cpu()'],['../classmlx_1_1core_1_1_multiply.html#a624fce06c047cdc4dfdbdcaaddb25f34',1,'mlx::core::Multiply::eval_cpu()'],['../classmlx_1_1core_1_1_negative.html#af43553dc418c8ebe75fa9cdcba103c3b',1,'mlx::core::Negative::eval_cpu()'],['../classmlx_1_1core_1_1_not_equal.html#a8f95f8b5873850b875b1641df8196047',1,'mlx::core::NotEqual::eval_cpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#acc328321cf5300874ee884367cbede3f',1,'mlx::core::NumberOfElements::eval_cpu()'],['../classmlx_1_1core_1_1_pad.html#aaf82dd163cd536fbf97304f8b29080cb',1,'mlx::core::Pad::eval_cpu()'],['../classmlx_1_1core_1_1_partition.html#a784596ab567f9f3cb4fe1a69466523d8',1,'mlx::core::Partition::eval_cpu()'],['../classmlx_1_1core_1_1_power.html#a6783da16fb6ff393aaa57737f1973206',1,'mlx::core::Power::eval_cpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ab3dfa73b74d8f4f2e9ab4f0eb016b0e3',1,'mlx::core::QuantizedMatmul::eval_cpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a89aae98bfbdd6563df44ef7d70f0bf8c',1,'mlx::core::GatherQMM::eval_cpu()'],['../classmlx_1_1core_1_1_random_bits.html#a5752d051cd16cf5f8d4754c0a656f0d2',1,'mlx::core::RandomBits::eval_cpu()'],['../classmlx_1_1core_1_1_real.html#a365d046caac91b521f0f5a5518037934',1,'mlx::core::Real::eval_cpu()'],['../classmlx_1_1core_1_1_reshape.html#a658de2c5f710991b48e14b2bd19b229f',1,'mlx::core::Reshape::eval_cpu()'],['../classmlx_1_1core_1_1_reduce.html#aeb8a58b560c0a09ae3a695df7829acfa',1,'mlx::core::Reduce::eval_cpu()'],['../classmlx_1_1core_1_1_round.html#ad066b0944b437f64ab546025efa00007',1,'mlx::core::Round::eval_cpu()'],['../classmlx_1_1core_1_1_scan.html#a15676d9fd066e935782a923fba3e940b',1,'mlx::core::Scan::eval_cpu()'],['../classmlx_1_1core_1_1_scatter.html#a7623f590f8b77167b5ebb4f14bc9dc97',1,'mlx::core::Scatter::eval_cpu()'],['../classmlx_1_1core_1_1_sigmoid.html#aa930ce05734cca529ebcb8d0ca8e1255',1,'mlx::core::Sigmoid::eval_cpu()'],['../classmlx_1_1core_1_1_sign.html#a7498ec993b66879be30c5d9762c45a97',1,'mlx::core::Sign::eval_cpu()'],['../classmlx_1_1core_1_1_sin.html#ab34f9cebc2aed55a0b6ab4c991f02eb5',1,'mlx::core::Sin::eval_cpu()'],['../classmlx_1_1core_1_1_sinh.html#ab6d5f6f40d177f6435f6a51c71b939dd',1,'mlx::core::Sinh::eval_cpu()'],['../classmlx_1_1core_1_1_slice.html#a4b13503f5b2f5c6a90d394b020f9b3f2',1,'mlx::core::Slice::eval_cpu()'],['../classmlx_1_1core_1_1_slice_update.html#ad82ca0e3ab88a0e086431050deea831b',1,'mlx::core::SliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a4e8c22c24a587ea0648ce89f461ed1ee',1,'mlx::core::DynamicSlice::eval_cpu()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a379185914db0326a5d4839839fe4fc83',1,'mlx::core::DynamicSliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_softmax.html#ac9ebc2eab1683b682e689ed8f4622b79',1,'mlx::core::Softmax::eval_cpu()'],['../classmlx_1_1core_1_1_sort.html#a459769a0241b2620e55bedaba19827cd',1,'mlx::core::Sort::eval_cpu()'],['../classmlx_1_1core_1_1_split.html#aff2889cb9074f0fda53edf8fa40b1fd4',1,'mlx::core::Split::eval_cpu()'],['../classmlx_1_1core_1_1_square.html#a1f4d327a705950616da63b83c2829e59',1,'mlx::core::Square::eval_cpu()'],['../classmlx_1_1core_1_1_sqrt.html#a5a64ecc4eef1e30a2963435dca7cefd5',1,'mlx::core::Sqrt::eval_cpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a56207714d374b08f60e4d9cdbc7340b2',1,'mlx::core::StopGradient::eval_cpu()'],['../classmlx_1_1core_1_1_subtract.html#a47574258b6c95f8ad260c114d6d36a12',1,'mlx::core::Subtract::eval_cpu()'],['../classmlx_1_1core_1_1_squeeze.html#a9bcb7476041020f59ef816196ddb81cb',1,'mlx::core::Squeeze::eval_cpu()'],['../classmlx_1_1core_1_1_tan.html#a9c9a731158fa60eef30067fe0da9f3e9',1,'mlx::core::Tan::eval_cpu()'],['../classmlx_1_1core_1_1_tanh.html#af7ed4345f622da069e5b0284067923f5',1,'mlx::core::Tanh::eval_cpu()'],['../classmlx_1_1core_1_1_unflatten.html#a507c22306b7afcdd5970cfaa32188f0a',1,'mlx::core::Unflatten::eval_cpu()'],['../classmlx_1_1core_1_1_view.html#a0ad6deb11914a242f10e8039fcb02497',1,'mlx::core::View::eval_cpu()'],['../classmlx_1_1core_1_1_transpose.html#a1fbcfcca43f9ec06c63a3c14708c30f8',1,'mlx::core::Transpose::eval_cpu()'],['../classmlx_1_1core_1_1_q_r_f.html#a48493887395d65a27f04de1804d277d2',1,'mlx::core::QRF::eval_cpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a637f5c39fa8b10722c04a066f6c1ada6',1,'mlx::core::SVD::eval_cpu()'],['../classmlx_1_1core_1_1_inverse.html#aeb1d8dc9bc4052a616023f65b3c7bb81',1,'mlx::core::Inverse::eval_cpu()'],['../classmlx_1_1core_1_1_cholesky.html#a4bdec36c1cc99aadf9a4a39d4c57bea5',1,'mlx::core::Cholesky::eval_cpu()'],['../classmlx_1_1core_1_1_eigh.html#a894b32e17229394f6a43b4a0655fd8be',1,'mlx::core::Eigh::eval_cpu()']]], - ['eval_5fgpu_38',['eval_gpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a52df7155f56b8450581b2fd2747cad20',1,'mlx::core::distributed::AllReduce::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a4251ce0f2db2045226b66210b828af7a',1,'mlx::core::distributed::AllGather::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a0c8dbd2a912be91be04ec701e29fba3d',1,'mlx::core::distributed::Send::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a932e39624bc3d234a7489c3decc4749e',1,'mlx::core::distributed::Recv::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae7955e8d43c097eecae264e804b4d8ca',1,'mlx::core::fast::RMSNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a48efb8fa84c4ba6cc9fb560ebbe01560',1,'mlx::core::fast::RMSNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a77abda7f47bffa2c037a5d60cccc1528',1,'mlx::core::fast::LayerNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a954a003a4a27c8c4c60a5a14142a9cc3',1,'mlx::core::fast::LayerNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a913b6b00fc518b25ac3947e4e15790f2',1,'mlx::core::fast::RoPE::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a505f38ba93a3499895f5312e0112e73d',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ad51666e69f670e286293aff96eb435a9',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, array &out)'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a63812b2abaf26ad7e7fa4c9e82db1628',1,'mlx::core::fast::AffineQuantize::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a2ed2a16b23053f8195068386a99fd6db',1,'mlx::core::fast::CustomKernel::eval_gpu()'],['../classmlx_1_1core_1_1_primitive.html#ad217376dcf5eff691d731566faec2ba2',1,'mlx::core::Primitive::eval_gpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a6b7f80abaf038d53ec6ffbb0dfac6adb',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#a971fe9ad47f6569118879ce1d0f41447',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0a976e636dd8505b473fbdddf949f514',1,'mlx::core::Abs::eval_gpu()'],['../classmlx_1_1core_1_1_add.html#aa0aacbc1e26b95a2f040f62aa4f69c3d',1,'mlx::core::Add::eval_gpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a5f933be14baebc32a0be0f9a69148aa9',1,'mlx::core::AddMM::eval_gpu()'],['../classmlx_1_1core_1_1_arange.html#a7a2e9787c6c3a78b4a6df91206974031',1,'mlx::core::Arange::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a46f72d4af89b0a0f5f203783fb44589c',1,'mlx::core::ArcCos::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#aa6a2587485a0e015ac2d5211d7d045fc',1,'mlx::core::ArcCosh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sin.html#a7fa4ae7a85bc8bed97ea258ae30762f3',1,'mlx::core::ArcSin::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79f648a86de4c10386a1ce3b5e38e8ac',1,'mlx::core::ArcSinh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a77866feb27028865d844070447c9a254',1,'mlx::core::ArcTan::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a76d3f0c29e0ff4642b8d39dac90d3f50',1,'mlx::core::ArcTan2::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a10566b9d3b2c7d090895b46d9040bc1d',1,'mlx::core::ArcTanh::eval_gpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a9a60995eaf85f63c877e86b23cbc15fc',1,'mlx::core::ArgPartition::eval_gpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#aafa982ce2abc0cd9e81e43aa2c823d29',1,'mlx::core::ArgReduce::eval_gpu()'],['../classmlx_1_1core_1_1_arg_sort.html#abc2d730850ec4ee8d7968b7417911709',1,'mlx::core::ArgSort::eval_gpu()'],['../classmlx_1_1core_1_1_as_type.html#a5b111b9d74c60d27b4a7ebaa49f96e0b',1,'mlx::core::AsType::eval_gpu()'],['../classmlx_1_1core_1_1_as_strided.html#ab6771a208323994927ca162ba7bb10ed',1,'mlx::core::AsStrided::eval_gpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#ac831a29fc46701b00bbe63ee33832afd',1,'mlx::core::BitwiseBinary::eval_gpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ab372b6df4de00a33795a052a23bb1df9',1,'mlx::core::BlockMaskedMM::eval_gpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#ad754c35f460a055cc383ad93a5f72da1',1,'mlx::core::GatherMM::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a56d16e75a0df867d2f1ba4e5198f15cb',1,'mlx::core::BroadcastAxes::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast.html#ab9bd9dbcedcefc9b29c84911b5ce69fe',1,'mlx::core::Broadcast::eval_gpu()'],['../classmlx_1_1core_1_1_ceil.html#abe178e0058e44b6618be414215e96887',1,'mlx::core::Ceil::eval_gpu()'],['../classmlx_1_1core_1_1_compiled.html#aa3d5ff0f2b3554ad48fbbf2a0f3336d5',1,'mlx::core::Compiled::eval_gpu()'],['../classmlx_1_1core_1_1_concatenate.html#a309a1c50e97f9925866433ee2841c474',1,'mlx::core::Concatenate::eval_gpu()'],['../classmlx_1_1core_1_1_conjugate.html#aff0a802166e3724db88ab5d3feb2d3de',1,'mlx::core::Conjugate::eval_gpu()'],['../classmlx_1_1core_1_1_contiguous.html#a519cd16fd0c55b371ea7625fbb37c70f',1,'mlx::core::Contiguous::eval_gpu()'],['../classmlx_1_1core_1_1_convolution.html#a30b64109eeb1778f002b99447dff9dd2',1,'mlx::core::Convolution::eval_gpu()'],['../classmlx_1_1core_1_1_copy.html#a1eda7b2ea771a168f67421f0d384b3a1',1,'mlx::core::Copy::eval_gpu()'],['../classmlx_1_1core_1_1_cos.html#a5ef41aafad595f6cdd8c535e36e12060',1,'mlx::core::Cos::eval_gpu()'],['../classmlx_1_1core_1_1_cosh.html#a23f71b43792934c3ec0ebe9b74f32559',1,'mlx::core::Cosh::eval_gpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#a7b3538681acbb20af3ed37b0877f6667',1,'mlx::core::CustomTransforms::eval_gpu()'],['../classmlx_1_1core_1_1_depends.html#ae5057f65e69490ad0add8eeda2b75e28',1,'mlx::core::Depends::eval_gpu()'],['../classmlx_1_1core_1_1_divide.html#abffda0ce37221ddc28dc9eea794f6bc7',1,'mlx::core::Divide::eval_gpu()'],['../classmlx_1_1core_1_1_div_mod.html#a003117c9ecf3c06a27248f72a76348dc',1,'mlx::core::DivMod::eval_gpu()'],['../classmlx_1_1core_1_1_select.html#a2a82b6cba4c386b2b87f225a4b08ea9b',1,'mlx::core::Select::eval_gpu()'],['../classmlx_1_1core_1_1_remainder.html#a7919ea9b84e42522d51bf0d5a396e161',1,'mlx::core::Remainder::eval_gpu()'],['../classmlx_1_1core_1_1_equal.html#ac3757001fec42ceb5ece2954df42161c',1,'mlx::core::Equal::eval_gpu()'],['../classmlx_1_1core_1_1_erf.html#ad8551be664d767dccc3c0d8cc1eca008',1,'mlx::core::Erf::eval_gpu()'],['../classmlx_1_1core_1_1_erf_inv.html#a4a2413d0634db1f3dae1806ddfa632db',1,'mlx::core::ErfInv::eval_gpu()'],['../classmlx_1_1core_1_1_exp.html#a7d63695a97a14760fd33b5d4e6590822',1,'mlx::core::Exp::eval_gpu()'],['../classmlx_1_1core_1_1_expm1.html#a82930071f4b77d883b300f77966aff5f',1,'mlx::core::Expm1::eval_gpu()'],['../classmlx_1_1core_1_1_expand_dims.html#ad350ede3abecc55371ddeb89fbba2b90',1,'mlx::core::ExpandDims::eval_gpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a1c21b26d1e9ad7c4da78ae845721b2dd',1,'mlx::core::FFT::eval_gpu()'],['../classmlx_1_1core_1_1_flatten.html#acb2219cc122d218b273af2cb9a882e7f',1,'mlx::core::Flatten::eval_gpu()'],['../classmlx_1_1core_1_1_floor.html#aaa29c83538099eb8f951c95a41f2eb65',1,'mlx::core::Floor::eval_gpu()'],['../classmlx_1_1core_1_1_full.html#aa54f99bb4cba12a551392dea56003872',1,'mlx::core::Full::eval_gpu()'],['../classmlx_1_1core_1_1_gather.html#aec48ee529cb2449915a7b27a3c4361e8',1,'mlx::core::Gather::eval_gpu()'],['../classmlx_1_1core_1_1_greater.html#ae8957cccf4c924d941f57a1bb751c878',1,'mlx::core::Greater::eval_gpu()'],['../classmlx_1_1core_1_1_greater_equal.html#ac246263b4548126c3d4ab7e392575d24',1,'mlx::core::GreaterEqual::eval_gpu()'],['../classmlx_1_1core_1_1_hadamard.html#a2470feb690f5463138490763c38b5733',1,'mlx::core::Hadamard::eval_gpu()'],['../classmlx_1_1core_1_1_imag.html#a247a4d059b0a99678c6be8c15e42c1e6',1,'mlx::core::Imag::eval_gpu()'],['../classmlx_1_1core_1_1_less.html#a353335ce06ddbe8498d86d129c835917',1,'mlx::core::Less::eval_gpu()'],['../classmlx_1_1core_1_1_less_equal.html#acf035a82b11e6f63742143ea540fedac',1,'mlx::core::LessEqual::eval_gpu()'],['../classmlx_1_1core_1_1_load.html#a06933e887ea94a4d01d81195c5e07a3d',1,'mlx::core::Load::eval_gpu()'],['../classmlx_1_1core_1_1_log.html#aaaa49e9455f3a197bc319646b5ca6390',1,'mlx::core::Log::eval_gpu()'],['../classmlx_1_1core_1_1_log1p.html#a1b97decae7338d46874e736c95fa7431',1,'mlx::core::Log1p::eval_gpu()'],['../classmlx_1_1core_1_1_logical_not.html#a1d0d2bc93f935eca6c85ef7bf67f2d6a',1,'mlx::core::LogicalNot::eval_gpu()'],['../classmlx_1_1core_1_1_logical_and.html#a132b2eedaa3978de5a5350da3c2ca40f',1,'mlx::core::LogicalAnd::eval_gpu()'],['../classmlx_1_1core_1_1_logical_or.html#a3be1da328f0f8620de2e4fc1d22a077a',1,'mlx::core::LogicalOr::eval_gpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#acace355b62ec00df649f9f99e8f2eb7a',1,'mlx::core::LogAddExp::eval_gpu()'],['../classmlx_1_1core_1_1_matmul.html#a8707a4e9b75c769e8f1dbca15c6a1ae7',1,'mlx::core::Matmul::eval_gpu()'],['../classmlx_1_1core_1_1_maximum.html#ade0f721b10a6b3a12bdadd34c48f72a7',1,'mlx::core::Maximum::eval_gpu()'],['../classmlx_1_1core_1_1_minimum.html#aadc68afa0afbe2103f19d161f5e0a2ba',1,'mlx::core::Minimum::eval_gpu()'],['../classmlx_1_1core_1_1_multiply.html#a634fcb4e981d8d3f4d94252caf25bee0',1,'mlx::core::Multiply::eval_gpu()'],['../classmlx_1_1core_1_1_negative.html#a97f1b316eace0c6d9e576d766940c75b',1,'mlx::core::Negative::eval_gpu()'],['../classmlx_1_1core_1_1_not_equal.html#a61179747e34e203150e9c660dfddb5f2',1,'mlx::core::NotEqual::eval_gpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#a2c98c42915fb2bfe12f5c99ea553eff5',1,'mlx::core::NumberOfElements::eval_gpu()'],['../classmlx_1_1core_1_1_pad.html#aefd4d3a5bd8b6b35b266c9e558ada153',1,'mlx::core::Pad::eval_gpu()'],['../classmlx_1_1core_1_1_partition.html#a8eca1be21ae9ccfda46e6f3e85f506ef',1,'mlx::core::Partition::eval_gpu()'],['../classmlx_1_1core_1_1_power.html#a80577d4c0853c24027777c90a1ec7e11',1,'mlx::core::Power::eval_gpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a2812ad007d695ed1aaf9cf706fb9c4b3',1,'mlx::core::QuantizedMatmul::eval_gpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a86eb048afc95646b2e96ec5493e3d887',1,'mlx::core::GatherQMM::eval_gpu()'],['../classmlx_1_1core_1_1_random_bits.html#a578756866665358577418e4cdd94aa3a',1,'mlx::core::RandomBits::eval_gpu()'],['../classmlx_1_1core_1_1_real.html#a1e209e88a43bdd1eea43ad0b03f9a7f2',1,'mlx::core::Real::eval_gpu()'],['../classmlx_1_1core_1_1_reshape.html#aa1e85f28471875750c47351520b56059',1,'mlx::core::Reshape::eval_gpu()'],['../classmlx_1_1core_1_1_reduce.html#ae9caaf42edadfe73ea208d98f526890f',1,'mlx::core::Reduce::eval_gpu()'],['../classmlx_1_1core_1_1_round.html#af7fe5ff8f3db166c203b4be4b07f13ec',1,'mlx::core::Round::eval_gpu()'],['../classmlx_1_1core_1_1_scan.html#aef22c6fc2b2cb2a907cd8965c7413dde',1,'mlx::core::Scan::eval_gpu()'],['../classmlx_1_1core_1_1_scatter.html#ab304345db3d8cfeea15e27461ae2e678',1,'mlx::core::Scatter::eval_gpu()'],['../classmlx_1_1core_1_1_sigmoid.html#a7a6bd0222d51d7f25f2719a91ccdfeca',1,'mlx::core::Sigmoid::eval_gpu()'],['../classmlx_1_1core_1_1_sign.html#afa2b48b99a194106006b44af69ffda8b',1,'mlx::core::Sign::eval_gpu()'],['../classmlx_1_1core_1_1_sin.html#a6b59f1156cf8bdad8d45acd1d825cb5e',1,'mlx::core::Sin::eval_gpu()'],['../classmlx_1_1core_1_1_sinh.html#a5a1af2399f166d5b228b5e83a1837c75',1,'mlx::core::Sinh::eval_gpu()'],['../classmlx_1_1core_1_1_slice.html#aa53c21ff06a7c659e889af6b97d10a4a',1,'mlx::core::Slice::eval_gpu()'],['../classmlx_1_1core_1_1_slice_update.html#aac1a1d122e5697be057d63552141032b',1,'mlx::core::SliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_dynamic_slice.html#ab0a2e31c03f02a4f25700e240cf18e3e',1,'mlx::core::DynamicSlice::eval_gpu()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a249dab28690c45203c3995698de0cab7',1,'mlx::core::DynamicSliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_softmax.html#a35dac69ddcc7e2ec0e1a76fe93db85af',1,'mlx::core::Softmax::eval_gpu()'],['../classmlx_1_1core_1_1_sort.html#a4141c48f0e8670c728663f3722675382',1,'mlx::core::Sort::eval_gpu()'],['../classmlx_1_1core_1_1_split.html#a78ddda89c4daee73c74cfbc1e44656df',1,'mlx::core::Split::eval_gpu()'],['../classmlx_1_1core_1_1_square.html#a0ea2a78a5bb52daa4103263bf2f98045',1,'mlx::core::Square::eval_gpu()'],['../classmlx_1_1core_1_1_sqrt.html#a6d205e679a593d1ba20206c5c47ba501',1,'mlx::core::Sqrt::eval_gpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a907b96f0a1ce608e211d87ccf2b9ca89',1,'mlx::core::StopGradient::eval_gpu()'],['../classmlx_1_1core_1_1_subtract.html#a69021b23daf061764d97fabbc0f4f06c',1,'mlx::core::Subtract::eval_gpu()'],['../classmlx_1_1core_1_1_squeeze.html#a18d382c8bc59d60b38e9fd1cb70660fd',1,'mlx::core::Squeeze::eval_gpu()'],['../classmlx_1_1core_1_1_tan.html#aca7dbb4836507005a2032ac957a04d3f',1,'mlx::core::Tan::eval_gpu()'],['../classmlx_1_1core_1_1_tanh.html#a48df896599ae93dbce84a5c0f50cf761',1,'mlx::core::Tanh::eval_gpu()'],['../classmlx_1_1core_1_1_unflatten.html#adfbb8208355f9c3cb2e4cb1fd4fe788f',1,'mlx::core::Unflatten::eval_gpu()'],['../classmlx_1_1core_1_1_view.html#add6e12ff1e476fe1db7718b14f21b075',1,'mlx::core::View::eval_gpu()'],['../classmlx_1_1core_1_1_transpose.html#a38d25739c08aa594a6775015a1d7d92e',1,'mlx::core::Transpose::eval_gpu()'],['../classmlx_1_1core_1_1_q_r_f.html#ae5fa3482192f4713605cd07e7fc1c6c9',1,'mlx::core::QRF::eval_gpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a7067b2207f826a25549d571856b94e83',1,'mlx::core::SVD::eval_gpu()'],['../classmlx_1_1core_1_1_inverse.html#a086fbbc947ad232e01686ad063a78ed2',1,'mlx::core::Inverse::eval_gpu()'],['../classmlx_1_1core_1_1_cholesky.html#a8c918594bf129888044ef37fcae56795',1,'mlx::core::Cholesky::eval_gpu()'],['../classmlx_1_1core_1_1_eigh.html#a67775b41c0a15e356f08d51d9736baa2',1,'mlx::core::Eigh::eval_gpu()']]], - ['evaluated_39',['evaluated',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078a6fc3d7595445dd877584495f47535268',1,'mlx::core::array']]], - ['event_40',['Event',['../classmlx_1_1core_1_1_event.html',1,'mlx::core::Event'],['../classmlx_1_1core_1_1_event.html#a833506419b2110ad1abd89b2dd238b4d',1,'mlx::core::Event::Event()=default'],['../classmlx_1_1core_1_1_event.html#a13e4835f2ffb2cc22e29148a448ea184',1,'mlx::core::Event::Event(const Stream &steam)']]], - ['event_41',['event',['../classmlx_1_1core_1_1array.html#a0a8e4d6e67e739a712876bb36f88f9bf',1,'mlx::core::array']]], - ['event_2eh_42',['event.h',['../event_8h.html',1,'']]], - ['excess_43',['excess',['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper']]], - ['exec_44',['exec',['../classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2',1,'pocketfft::detail::cfftp::exec()'],['../classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac',1,'pocketfft::detail::rfftp::exec()'],['../classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c',1,'pocketfft::detail::fftblue::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d',1,'pocketfft::detail::pocketfft_c::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33',1,'pocketfft::detail::pocketfft_r::exec()'],['../classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058',1,'pocketfft::detail::T_dct1::exec()'],['../classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26',1,'pocketfft::detail::T_dst1::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac',1,'pocketfft::detail::T_dcst23::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708',1,'pocketfft::detail::T_dcst4::exec()'],['../classmlx_1_1core_1_1_jit_compiler.html#adcf98f940e1919388eaab907ea17a540',1,'mlx::core::JitCompiler::exec()']]], - ['exec_5fr_45',['exec_r',['../classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00',1,'pocketfft::detail::fftblue']]], - ['execc2c_46',['ExecC2C',['../structpocketfft_1_1detail_1_1_exec_c2_c.html',1,'pocketfft::detail']]], - ['execdcst_47',['ExecDcst',['../structpocketfft_1_1detail_1_1_exec_dcst.html',1,'pocketfft::detail']]], - ['exechartley_48',['ExecHartley',['../structpocketfft_1_1detail_1_1_exec_hartley.html',1,'pocketfft::detail']]], - ['execr2r_49',['ExecR2R',['../structpocketfft_1_1detail_1_1_exec_r2_r.html',1,'pocketfft::detail']]], - ['exp_50',['Exp',['../struct_exp.html',1,'Exp'],['../structmlx_1_1core_1_1detail_1_1_exp.html',1,'mlx::core::detail::Exp'],['../classmlx_1_1core_1_1_exp.html',1,'mlx::core::Exp'],['../classmlx_1_1core_1_1_exp.html#a1d0a618cbb91ab29ef53b57ff6ed6e06',1,'mlx::core::Exp::Exp()']]], - ['exp_51',['exp',['../namespacemetal.html#ac2a0b3618d922ac014baac8189d44650',1,'metal::exp()'],['../namespacemetal_1_1fast.html#ad3dbd387b63373c29e3449609f763ede',1,'metal::fast::exp()'],['../namespacemetal_1_1precise.html#a8d8d2d5700ce432b33cf47cf22528e8f',1,'metal::precise::exp()'],['../group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125',1,'mlx::core::exp()']]], - ['exp10_52',['exp10',['../namespacemetal.html#a4c63707d13c89364496a48906631c204',1,'metal::exp10()'],['../namespacemetal_1_1fast.html#a453122f982485cbb4e471b3ac282ee5e',1,'metal::fast::exp10()'],['../namespacemetal_1_1precise.html#af9addb343c967da3a83e9e123a8521fd',1,'metal::precise::exp10()']]], - ['exp2_53',['exp2',['../namespacemetal.html#a228201c20777848804a4d0589c1d33e7',1,'metal::exp2()'],['../namespacemetal_1_1fast.html#ac092b65a46720adaf22f6266671d2d71',1,'metal::fast::exp2()'],['../namespacemetal_1_1precise.html#a92a880bd2197efc0da0f8f0f7ec1e4c9',1,'metal::precise::exp2()']]], - ['expand_5fdims_54',['expand_dims',['../group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133',1,'mlx::core::expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a',1,'mlx::core::expand_dims(const array &a, int axis, StreamOrDevice s={})']]], - ['expanddims_55',['ExpandDims',['../classmlx_1_1core_1_1_expand_dims.html',1,'mlx::core::ExpandDims'],['../classmlx_1_1core_1_1_expand_dims.html#aea2479ea4dd93941eb83a22e087983a8',1,'mlx::core::ExpandDims::ExpandDims()']]], - ['expm1_56',['Expm1',['../struct_expm1.html',1,'Expm1'],['../structmlx_1_1core_1_1detail_1_1_expm1.html',1,'mlx::core::detail::Expm1'],['../classmlx_1_1core_1_1_expm1.html',1,'mlx::core::Expm1'],['../classmlx_1_1core_1_1_expm1.html#a47c2a1b2a4ef6bb07ba77c55ddddaec2',1,'mlx::core::Expm1::Expm1()']]], - ['expm1_57',['expm1',['../group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0',1,'mlx::core']]], - ['expm1f_58',['expm1f',['../expm1f_8h.html#a87f66d30e185950f42ce3641783cdc40',1,'expm1f.h']]], - ['expm1f_2eh_59',['expm1f.h',['../expm1f_8h.html',1,'']]], - ['expm1f_5fscaled_5funchecked_60',['expm1f_scaled_unchecked',['../expm1f_8h.html#adf20e03405fba634ca8d01acac24592e',1,'expm1f.h']]], - ['export_2eh_61',['export.h',['../export_8h.html',1,'']]], - ['export_5ffunction_62',['export_function',['../namespacemlx_1_1core.html#a2afa4ea816ac9317200fd5c964fc89dc',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, const Args &args, bool shapeless=false)'],['../namespacemlx_1_1core.html#ace51644e2aa72f8d56b86eaa0e1a68b7',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, const Kwargs &kwargs, bool shapeless=false)'],['../namespacemlx_1_1core.html#a9692d7bb6de3456abc535d0f4bac7a94',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, const Args &args, const Kwargs &kwargs, bool shapeless=false)']]], - ['export_5fimpl_2eh_63',['export_impl.h',['../export__impl_8h.html',1,'']]], - ['export_5fto_5fdot_64',['export_to_dot',['../namespacemlx_1_1core.html#a30338cb7d259334e46dc7a4819716fa6',1,'mlx::core::export_to_dot(std::ostream &os, NodeNamer namer, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a57395bdf43d9c5c134e610c169222cca',1,'mlx::core::export_to_dot(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a839f94dbad44f0d37333006fc876b42e',1,'mlx::core::export_to_dot(std::ostream &os, Arrays &&... outputs)'],['../namespacemlx_1_1core.html#a12faa564e0d85a1daa77c08babb75261',1,'mlx::core::export_to_dot(std::ostream &os, NodeNamer namer, Arrays &&... outputs)']]], - ['exporter_65',['exporter',['../structmlx_1_1core_1_1_function_exporter.html#a3921e0f41f795708c33bda7c50a72055',1,'mlx::core::FunctionExporter::exporter(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, bool shapeless=false)'],['../structmlx_1_1core_1_1_function_exporter.html#ae53b863ba9fea62cc7c1b9eb55993269',1,'mlx::core::FunctionExporter::exporter(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, bool shapeless=false)'],['../structmlx_1_1core_1_1_function_exporter.html#a82eb4ca466592b97225e8252f1cdb2e4',1,'mlx::core::FunctionExporter::exporter(const std::string &path, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#a0303e26b737c9fd197ed9caa90fd21a7',1,'mlx::core::exporter(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#af2735df8513ecce88456585f5aea50f5',1,'mlx::core::exporter(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#af38d5718f517e50a590fdb3d63a90df1',1,'mlx::core::exporter(const std::string &path, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, bool shapeless=false)']]], - ['expsubop_66',['ExpSubOp',['../struct_exp_sub_op.html',1,'']]], - ['eye_67',['eye',['../group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3',1,'mlx::core::eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd',1,'mlx::core::eye(int n, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga61657db78ef35d41112d362c869c25d2',1,'mlx::core::eye(int n, int m, StreamOrDevice s={})'],['../group__ops.html#ga908a15b42834be498a46856c99dfc779',1,'mlx::core::eye(int n, int m, int k, StreamOrDevice s={})'],['../group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f',1,'mlx::core::eye(int n, StreamOrDevice s={})']]] + ['encode_5fsignal_23',['encode_signal',['../namespacemlx_1_1core.html#a6d452306f0f046a7d021bd94f8713a89',1,'mlx::core']]], + ['encode_5fwait_24',['encode_wait',['../namespacemlx_1_1core.html#a2874ba55b73057b76c23a7429fdd2d6e',1,'mlx::core']]], + ['encoder_25',['encoder',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a58e435217b9922f882507ebf48bfbbdd',1,'mlx::core::metal::DeviceStream']]], + ['end_26',['end',['../classmlx_1_1core_1_1array.html#a5daf64552fb450825c9b382f3a5fa2d4',1,'mlx::core::array']]], + ['end_5fencoding_27',['end_encoding',['../classmlx_1_1core_1_1metal_1_1_device.html#a60689f97347811b27e8c5ca23e0372bf',1,'mlx::core::metal::Device']]], + ['enqueue_28',['enqueue',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4918720319cf224a1b4208568964c286',1,'mlx::core::scheduler::StreamThread::enqueue()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a03809c783bd1866362dc7cb9118abbcc',1,'mlx::core::scheduler::Scheduler::enqueue()'],['../class_thread_pool.html#a375fa2d63197282277be640b54e8a196',1,'ThreadPool::enqueue()'],['../namespacemlx_1_1core_1_1scheduler.html#aa2d4eacf5d5cbc778a51aafd4fd8e4d7',1,'mlx::core::scheduler::enqueue()']]], + ['epsilon_29',['epsilon',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a96c4197e3076f0aa9065370b8ece49ca',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['equal_30',['Equal',['../struct_equal.html',1,'Equal'],['../structmlx_1_1core_1_1detail_1_1_equal.html',1,'mlx::core::detail::Equal'],['../classmlx_1_1core_1_1_equal.html',1,'mlx::core::Equal'],['../classmlx_1_1core_1_1_equal.html#a4af81cf2dd071db5bbf8ce1df95fdf36',1,'mlx::core::Equal::Equal()']]], + ['equal_31',['equal',['../group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde',1,'mlx::core']]], + ['erase_32',['erase',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#ae136ad270522210c85c13cacf5165238',1,'mlx::core::metal::ResidencySet']]], + ['erf_33',['Erf',['../struct_erf.html',1,'Erf'],['../structmlx_1_1core_1_1detail_1_1_erf.html',1,'mlx::core::detail::Erf'],['../classmlx_1_1core_1_1_erf.html',1,'mlx::core::Erf'],['../classmlx_1_1core_1_1_erf.html#a702f76f848928d8d7d3d0881ac6e4c82',1,'mlx::core::Erf::Erf()']]], + ['erf_34',['erf',['../namespacemlx_1_1core_1_1simd.html#a60e33ebb16d9bab375a64aec8015a5c2',1,'mlx::core::simd::erf()'],['../erf_8h.html#a6ce199ee56105c67adbf8c48c019a8b2',1,'erf(): erf.h'],['../group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0',1,'mlx::core::erf()']]], + ['erf_2eh_35',['erf.h',['../erf_8h.html',1,'']]], + ['erfinv_36',['ErfInv',['../struct_erf_inv.html',1,'ErfInv'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html',1,'mlx::core::detail::ErfInv'],['../classmlx_1_1core_1_1_erf_inv.html',1,'mlx::core::ErfInv'],['../classmlx_1_1core_1_1_erf_inv.html#a5d0279247b67da4592311559f04e1478',1,'mlx::core::ErfInv::ErfInv()']]], + ['erfinv_37',['erfinv',['../namespacemlx_1_1core_1_1simd.html#a7687f3d14077b51fb421f0efb5b626db',1,'mlx::core::simd::erfinv()'],['../erf_8h.html#a1846e0d683c7aff826bb32addcc3b885',1,'erfinv(): erf.h'],['../group__ops.html#ga76fb9062c64264e34d2e07013390557c',1,'mlx::core::erfinv()']]], + ['eval_38',['eval',['../classmlx_1_1core_1_1array.html#a2820c45188071a22175e9fa42e10a49a',1,'mlx::core::array::eval()'],['../namespacemlx_1_1core.html#a7d6e097d8effed52f4713672e471f299',1,'mlx::core::eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#adb14f689c9f75f7901edb196c2bfb971',1,'mlx::core::eval(Arrays &&... outputs)']]], + ['eval_5fcpu_39',['eval_cpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#acdc1965ad64ee9ee6328fe150a97902e',1,'mlx::core::distributed::AllReduce::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ab721fe0072fffbddbc3c4334dd033ba5',1,'mlx::core::distributed::AllGather::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#af2620837bfc1b97217d006ed6e374051',1,'mlx::core::distributed::Send::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a3be84b08122a939edd6062d26261358a',1,'mlx::core::distributed::Recv::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a7da6e0cfd630958d9633b2e2bd97a54f',1,'mlx::core::fast::RMSNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#adfc1d52bc266466ab29ee45fd8fab439',1,'mlx::core::fast::RMSNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a5d7a4c1c9ee84e327d1c371733108c05',1,'mlx::core::fast::LayerNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a0d8c4c6e7462befc38f7e08244fa1c2b',1,'mlx::core::fast::LayerNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a05a7d595c6b9dadf7ddfd6e3fd402f0e',1,'mlx::core::fast::RoPE::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ae20851e002f7fcb6d4f97817596f6328',1,'mlx::core::fast::ScaledDotProductAttention::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a3b5d628628d245b38911118d4a0ff9fd',1,'mlx::core::fast::AffineQuantize::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a4ad1b7a9919753c759093f3e21a15bad',1,'mlx::core::fast::CustomKernel::eval_cpu()'],['../classmlx_1_1core_1_1_primitive.html#a1596dc50b910538eae14878e98f07575',1,'mlx::core::Primitive::eval_cpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a7e8f6f5d6ae0a33f6abc0f5a46e0b132',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#aa0ed6e32c36200a3ff9bc592c9b300db',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0d3e697496ef8e842d21195cb3c14e60',1,'mlx::core::Abs::eval_cpu()'],['../classmlx_1_1core_1_1_add.html#a5bacfc51dfa2a5a931bad2dd7bdc7a5f',1,'mlx::core::Add::eval_cpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a15694e3bf2ed5c193237b2b9ca00867c',1,'mlx::core::AddMM::eval_cpu()'],['../classmlx_1_1core_1_1_arange.html#aba44432491cbd599bf72712f5f4267a1',1,'mlx::core::Arange::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a58dcba9e706cb12bab062bb7fa5fa006',1,'mlx::core::ArcCos::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#a0f6d989bcbbc38f15ef17a136879a9c9',1,'mlx::core::ArcCosh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sin.html#ab3542492c14021329788de8f2a9be1e4',1,'mlx::core::ArcSin::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a52574b24d8d16839c58673f51f8ac066',1,'mlx::core::ArcSinh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a1211bc31241227528f04435239ddb9a3',1,'mlx::core::ArcTan::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a13094e6b702769928ca0da468f5ce45c',1,'mlx::core::ArcTan2::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a5af9224e1f1ffec412b0baa0af7e1ecd',1,'mlx::core::ArcTanh::eval_cpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a896f75c5325798ac3f9093f6a4581828',1,'mlx::core::ArgPartition::eval_cpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#ad8d48725623ede1ff654fa13eccf2287',1,'mlx::core::ArgReduce::eval_cpu()'],['../classmlx_1_1core_1_1_arg_sort.html#a022079683774bfeb531b3a002cff16fa',1,'mlx::core::ArgSort::eval_cpu()'],['../classmlx_1_1core_1_1_as_type.html#aa89dbf4d73b00c6a44cffd04d5bb228d',1,'mlx::core::AsType::eval_cpu()'],['../classmlx_1_1core_1_1_as_strided.html#acdd4705e4503ff0b124215c4676b4193',1,'mlx::core::AsStrided::eval_cpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a2194bf585213bda1b2966aa02d2fe283',1,'mlx::core::BitwiseBinary::eval_cpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aa85da478cdc6d4a97be06e5d4abee1f2',1,'mlx::core::BlockMaskedMM::eval_cpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#a62352074a480df0e1f879b0bae425730',1,'mlx::core::GatherMM::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a6423095cd28b2f90893c03166257a568',1,'mlx::core::BroadcastAxes::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast.html#a53d48d9778e2d4c24a124cd767900780',1,'mlx::core::Broadcast::eval_cpu()'],['../classmlx_1_1core_1_1_ceil.html#a9791801fff3f8b79944e15ac2a45a035',1,'mlx::core::Ceil::eval_cpu()'],['../classmlx_1_1core_1_1_compiled.html#ac45b1d0fedd85feefbff7ce7e168b151',1,'mlx::core::Compiled::eval_cpu()'],['../classmlx_1_1core_1_1_concatenate.html#a609e76bede7fc5581ec84ddcb727a258',1,'mlx::core::Concatenate::eval_cpu()'],['../classmlx_1_1core_1_1_conjugate.html#ae39643e2178f442ffba05139f8609d61',1,'mlx::core::Conjugate::eval_cpu()'],['../classmlx_1_1core_1_1_contiguous.html#a742de24e6c0310cd85a606dec0cd8336',1,'mlx::core::Contiguous::eval_cpu()'],['../classmlx_1_1core_1_1_convolution.html#ac74256068da01730629109fa4fa8432b',1,'mlx::core::Convolution::eval_cpu()'],['../classmlx_1_1core_1_1_copy.html#af4a0ebec423e84ffe8083a5e9ed0d70c',1,'mlx::core::Copy::eval_cpu()'],['../classmlx_1_1core_1_1_cos.html#a061fc446268fe56237ae6b20ccf78152',1,'mlx::core::Cos::eval_cpu()'],['../classmlx_1_1core_1_1_cosh.html#ae8702df7e8f0e20cbeccb2a548961d3d',1,'mlx::core::Cosh::eval_cpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#adba1c40c77a2138df6b5f75483f62184',1,'mlx::core::CustomTransforms::eval_cpu()'],['../classmlx_1_1core_1_1_depends.html#a0c7ea6db97337591fa53c6e6bde41e5e',1,'mlx::core::Depends::eval_cpu()'],['../classmlx_1_1core_1_1_divide.html#a823443c2a8e8b81bbcaeee6ddbcdbf49',1,'mlx::core::Divide::eval_cpu()'],['../classmlx_1_1core_1_1_div_mod.html#ae350b7b93ad128e3133ee14f247193b3',1,'mlx::core::DivMod::eval_cpu()'],['../classmlx_1_1core_1_1_select.html#aa51aa36e0adbd69e0d23d7c7adf88de2',1,'mlx::core::Select::eval_cpu()'],['../classmlx_1_1core_1_1_remainder.html#ac6c6c86a0bf02e6e529eb87f6e617ccc',1,'mlx::core::Remainder::eval_cpu()'],['../classmlx_1_1core_1_1_equal.html#aabb8aa61fa581defddcdca1274b1b454',1,'mlx::core::Equal::eval_cpu()'],['../classmlx_1_1core_1_1_erf.html#a84ea16e43d5b7f83bbc2d5ece78a3fb6',1,'mlx::core::Erf::eval_cpu()'],['../classmlx_1_1core_1_1_erf_inv.html#af579627402af3249565134884701d39e',1,'mlx::core::ErfInv::eval_cpu()'],['../classmlx_1_1core_1_1_exp.html#a47934c5a5023bc7ae7ae89bff45ebb2c',1,'mlx::core::Exp::eval_cpu()'],['../classmlx_1_1core_1_1_expm1.html#ab9c8b7aa50fe4592d55f8957baac647a',1,'mlx::core::Expm1::eval_cpu()'],['../classmlx_1_1core_1_1_expand_dims.html#a34058a87582a6ab2e5d82a75bc713030',1,'mlx::core::ExpandDims::eval_cpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a6bc262a0c2b5d4fe655e3e2e0ff28635',1,'mlx::core::FFT::eval_cpu()'],['../classmlx_1_1core_1_1_flatten.html#a72ade7d22386b349712f6c7c1f619842',1,'mlx::core::Flatten::eval_cpu()'],['../classmlx_1_1core_1_1_floor.html#a1a7dc5f571b7b73e7ef3cbdc1dd1fcf7',1,'mlx::core::Floor::eval_cpu()'],['../classmlx_1_1core_1_1_full.html#a3dccd3756599d7fd018b2af0093b082c',1,'mlx::core::Full::eval_cpu()'],['../classmlx_1_1core_1_1_gather.html#a9ed5587f0d04b59a2b9186c0aac21290',1,'mlx::core::Gather::eval_cpu()'],['../classmlx_1_1core_1_1_gather_axis.html#a474eae1d024e676e668318bf10928e2a',1,'mlx::core::GatherAxis::eval_cpu()'],['../classmlx_1_1core_1_1_greater.html#abe1c03f311d0e0b610f3392a6566f2ae',1,'mlx::core::Greater::eval_cpu()'],['../classmlx_1_1core_1_1_greater_equal.html#a15469125b9bea89b64bfeac01590c075',1,'mlx::core::GreaterEqual::eval_cpu()'],['../classmlx_1_1core_1_1_hadamard.html#ab27d6a9df42b3aab41ace3073a4c880d',1,'mlx::core::Hadamard::eval_cpu()'],['../classmlx_1_1core_1_1_imag.html#a17d1f1f9f8528668fcdf39b636720829',1,'mlx::core::Imag::eval_cpu()'],['../classmlx_1_1core_1_1_less.html#a32624124ffece066f496b3299056bcef',1,'mlx::core::Less::eval_cpu()'],['../classmlx_1_1core_1_1_less_equal.html#a55d1352b0e97841a92503bc57c19ed16',1,'mlx::core::LessEqual::eval_cpu()'],['../classmlx_1_1core_1_1_load.html#ada026ac30566f3109d8182e35d307c0a',1,'mlx::core::Load::eval_cpu()'],['../classmlx_1_1core_1_1_log.html#aadc7bb4cb24f3ecbbb9ed54a699ab74f',1,'mlx::core::Log::eval_cpu()'],['../classmlx_1_1core_1_1_log1p.html#a8192e5438de99c4cda056987935cba23',1,'mlx::core::Log1p::eval_cpu()'],['../classmlx_1_1core_1_1_logical_not.html#acf3f7b3b20ca69533536e0e0a05725b3',1,'mlx::core::LogicalNot::eval_cpu()'],['../classmlx_1_1core_1_1_logical_and.html#adbe1c1785af1a8b827289d22b0d170b3',1,'mlx::core::LogicalAnd::eval_cpu()'],['../classmlx_1_1core_1_1_logical_or.html#a13cd4cbf26589287e85aeaaca42d7f62',1,'mlx::core::LogicalOr::eval_cpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#abef17fb590b1a8d356f2a580e45d41f0',1,'mlx::core::LogAddExp::eval_cpu()'],['../classmlx_1_1core_1_1_matmul.html#a357a7f57a2a220a91977f810a69413fc',1,'mlx::core::Matmul::eval_cpu()'],['../classmlx_1_1core_1_1_maximum.html#a62b38fbe5f96db58c2b60165ac4eadcf',1,'mlx::core::Maximum::eval_cpu()'],['../classmlx_1_1core_1_1_minimum.html#a6b93f493ee87089943a8085fe59dfc6e',1,'mlx::core::Minimum::eval_cpu()'],['../classmlx_1_1core_1_1_multiply.html#a624fce06c047cdc4dfdbdcaaddb25f34',1,'mlx::core::Multiply::eval_cpu()'],['../classmlx_1_1core_1_1_negative.html#af43553dc418c8ebe75fa9cdcba103c3b',1,'mlx::core::Negative::eval_cpu()'],['../classmlx_1_1core_1_1_not_equal.html#a8f95f8b5873850b875b1641df8196047',1,'mlx::core::NotEqual::eval_cpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#acc328321cf5300874ee884367cbede3f',1,'mlx::core::NumberOfElements::eval_cpu()'],['../classmlx_1_1core_1_1_pad.html#aaf82dd163cd536fbf97304f8b29080cb',1,'mlx::core::Pad::eval_cpu()'],['../classmlx_1_1core_1_1_partition.html#a784596ab567f9f3cb4fe1a69466523d8',1,'mlx::core::Partition::eval_cpu()'],['../classmlx_1_1core_1_1_power.html#a6783da16fb6ff393aaa57737f1973206',1,'mlx::core::Power::eval_cpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ab3dfa73b74d8f4f2e9ab4f0eb016b0e3',1,'mlx::core::QuantizedMatmul::eval_cpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a89aae98bfbdd6563df44ef7d70f0bf8c',1,'mlx::core::GatherQMM::eval_cpu()'],['../classmlx_1_1core_1_1_random_bits.html#a5752d051cd16cf5f8d4754c0a656f0d2',1,'mlx::core::RandomBits::eval_cpu()'],['../classmlx_1_1core_1_1_real.html#a365d046caac91b521f0f5a5518037934',1,'mlx::core::Real::eval_cpu()'],['../classmlx_1_1core_1_1_reshape.html#a658de2c5f710991b48e14b2bd19b229f',1,'mlx::core::Reshape::eval_cpu()'],['../classmlx_1_1core_1_1_reduce.html#aeb8a58b560c0a09ae3a695df7829acfa',1,'mlx::core::Reduce::eval_cpu()'],['../classmlx_1_1core_1_1_round.html#ad066b0944b437f64ab546025efa00007',1,'mlx::core::Round::eval_cpu()'],['../classmlx_1_1core_1_1_scan.html#a15676d9fd066e935782a923fba3e940b',1,'mlx::core::Scan::eval_cpu()'],['../classmlx_1_1core_1_1_scatter.html#a7623f590f8b77167b5ebb4f14bc9dc97',1,'mlx::core::Scatter::eval_cpu()'],['../classmlx_1_1core_1_1_scatter_axis.html#abf9d24565abdd7e1034daacac603cc54',1,'mlx::core::ScatterAxis::eval_cpu()'],['../classmlx_1_1core_1_1_sigmoid.html#aa930ce05734cca529ebcb8d0ca8e1255',1,'mlx::core::Sigmoid::eval_cpu()'],['../classmlx_1_1core_1_1_sign.html#a7498ec993b66879be30c5d9762c45a97',1,'mlx::core::Sign::eval_cpu()'],['../classmlx_1_1core_1_1_sin.html#ab34f9cebc2aed55a0b6ab4c991f02eb5',1,'mlx::core::Sin::eval_cpu()'],['../classmlx_1_1core_1_1_sinh.html#ab6d5f6f40d177f6435f6a51c71b939dd',1,'mlx::core::Sinh::eval_cpu()'],['../classmlx_1_1core_1_1_slice.html#a4b13503f5b2f5c6a90d394b020f9b3f2',1,'mlx::core::Slice::eval_cpu()'],['../classmlx_1_1core_1_1_slice_update.html#ad82ca0e3ab88a0e086431050deea831b',1,'mlx::core::SliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a4e8c22c24a587ea0648ce89f461ed1ee',1,'mlx::core::DynamicSlice::eval_cpu()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a379185914db0326a5d4839839fe4fc83',1,'mlx::core::DynamicSliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_softmax.html#ac9ebc2eab1683b682e689ed8f4622b79',1,'mlx::core::Softmax::eval_cpu()'],['../classmlx_1_1core_1_1_sort.html#a459769a0241b2620e55bedaba19827cd',1,'mlx::core::Sort::eval_cpu()'],['../classmlx_1_1core_1_1_split.html#aff2889cb9074f0fda53edf8fa40b1fd4',1,'mlx::core::Split::eval_cpu()'],['../classmlx_1_1core_1_1_square.html#a1f4d327a705950616da63b83c2829e59',1,'mlx::core::Square::eval_cpu()'],['../classmlx_1_1core_1_1_sqrt.html#a5a64ecc4eef1e30a2963435dca7cefd5',1,'mlx::core::Sqrt::eval_cpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a56207714d374b08f60e4d9cdbc7340b2',1,'mlx::core::StopGradient::eval_cpu()'],['../classmlx_1_1core_1_1_subtract.html#a47574258b6c95f8ad260c114d6d36a12',1,'mlx::core::Subtract::eval_cpu()'],['../classmlx_1_1core_1_1_squeeze.html#a9bcb7476041020f59ef816196ddb81cb',1,'mlx::core::Squeeze::eval_cpu()'],['../classmlx_1_1core_1_1_tan.html#a9c9a731158fa60eef30067fe0da9f3e9',1,'mlx::core::Tan::eval_cpu()'],['../classmlx_1_1core_1_1_tanh.html#af7ed4345f622da069e5b0284067923f5',1,'mlx::core::Tanh::eval_cpu()'],['../classmlx_1_1core_1_1_unflatten.html#a507c22306b7afcdd5970cfaa32188f0a',1,'mlx::core::Unflatten::eval_cpu()'],['../classmlx_1_1core_1_1_view.html#a0ad6deb11914a242f10e8039fcb02497',1,'mlx::core::View::eval_cpu()'],['../classmlx_1_1core_1_1_transpose.html#a1fbcfcca43f9ec06c63a3c14708c30f8',1,'mlx::core::Transpose::eval_cpu()'],['../classmlx_1_1core_1_1_q_r_f.html#a48493887395d65a27f04de1804d277d2',1,'mlx::core::QRF::eval_cpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a637f5c39fa8b10722c04a066f6c1ada6',1,'mlx::core::SVD::eval_cpu()'],['../classmlx_1_1core_1_1_inverse.html#aeb1d8dc9bc4052a616023f65b3c7bb81',1,'mlx::core::Inverse::eval_cpu()'],['../classmlx_1_1core_1_1_cholesky.html#a4bdec36c1cc99aadf9a4a39d4c57bea5',1,'mlx::core::Cholesky::eval_cpu()'],['../classmlx_1_1core_1_1_eigh.html#a894b32e17229394f6a43b4a0655fd8be',1,'mlx::core::Eigh::eval_cpu()']]], + ['eval_5fgpu_40',['eval_gpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a52df7155f56b8450581b2fd2747cad20',1,'mlx::core::distributed::AllReduce::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a4251ce0f2db2045226b66210b828af7a',1,'mlx::core::distributed::AllGather::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a0c8dbd2a912be91be04ec701e29fba3d',1,'mlx::core::distributed::Send::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a932e39624bc3d234a7489c3decc4749e',1,'mlx::core::distributed::Recv::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae7955e8d43c097eecae264e804b4d8ca',1,'mlx::core::fast::RMSNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a48efb8fa84c4ba6cc9fb560ebbe01560',1,'mlx::core::fast::RMSNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a77abda7f47bffa2c037a5d60cccc1528',1,'mlx::core::fast::LayerNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a954a003a4a27c8c4c60a5a14142a9cc3',1,'mlx::core::fast::LayerNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a913b6b00fc518b25ac3947e4e15790f2',1,'mlx::core::fast::RoPE::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a505f38ba93a3499895f5312e0112e73d',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ad51666e69f670e286293aff96eb435a9',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, array &out)'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a63812b2abaf26ad7e7fa4c9e82db1628',1,'mlx::core::fast::AffineQuantize::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a2ed2a16b23053f8195068386a99fd6db',1,'mlx::core::fast::CustomKernel::eval_gpu()'],['../classmlx_1_1core_1_1_primitive.html#ad217376dcf5eff691d731566faec2ba2',1,'mlx::core::Primitive::eval_gpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a6b7f80abaf038d53ec6ffbb0dfac6adb',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#a971fe9ad47f6569118879ce1d0f41447',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0a976e636dd8505b473fbdddf949f514',1,'mlx::core::Abs::eval_gpu()'],['../classmlx_1_1core_1_1_add.html#aa0aacbc1e26b95a2f040f62aa4f69c3d',1,'mlx::core::Add::eval_gpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a5f933be14baebc32a0be0f9a69148aa9',1,'mlx::core::AddMM::eval_gpu()'],['../classmlx_1_1core_1_1_arange.html#a7a2e9787c6c3a78b4a6df91206974031',1,'mlx::core::Arange::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a46f72d4af89b0a0f5f203783fb44589c',1,'mlx::core::ArcCos::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#aa6a2587485a0e015ac2d5211d7d045fc',1,'mlx::core::ArcCosh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sin.html#a7fa4ae7a85bc8bed97ea258ae30762f3',1,'mlx::core::ArcSin::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79f648a86de4c10386a1ce3b5e38e8ac',1,'mlx::core::ArcSinh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a77866feb27028865d844070447c9a254',1,'mlx::core::ArcTan::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a76d3f0c29e0ff4642b8d39dac90d3f50',1,'mlx::core::ArcTan2::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a10566b9d3b2c7d090895b46d9040bc1d',1,'mlx::core::ArcTanh::eval_gpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a9a60995eaf85f63c877e86b23cbc15fc',1,'mlx::core::ArgPartition::eval_gpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#aafa982ce2abc0cd9e81e43aa2c823d29',1,'mlx::core::ArgReduce::eval_gpu()'],['../classmlx_1_1core_1_1_arg_sort.html#abc2d730850ec4ee8d7968b7417911709',1,'mlx::core::ArgSort::eval_gpu()'],['../classmlx_1_1core_1_1_as_type.html#a5b111b9d74c60d27b4a7ebaa49f96e0b',1,'mlx::core::AsType::eval_gpu()'],['../classmlx_1_1core_1_1_as_strided.html#ab6771a208323994927ca162ba7bb10ed',1,'mlx::core::AsStrided::eval_gpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#ac831a29fc46701b00bbe63ee33832afd',1,'mlx::core::BitwiseBinary::eval_gpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ab372b6df4de00a33795a052a23bb1df9',1,'mlx::core::BlockMaskedMM::eval_gpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#ad754c35f460a055cc383ad93a5f72da1',1,'mlx::core::GatherMM::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a56d16e75a0df867d2f1ba4e5198f15cb',1,'mlx::core::BroadcastAxes::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast.html#ab9bd9dbcedcefc9b29c84911b5ce69fe',1,'mlx::core::Broadcast::eval_gpu()'],['../classmlx_1_1core_1_1_ceil.html#abe178e0058e44b6618be414215e96887',1,'mlx::core::Ceil::eval_gpu()'],['../classmlx_1_1core_1_1_compiled.html#aa3d5ff0f2b3554ad48fbbf2a0f3336d5',1,'mlx::core::Compiled::eval_gpu()'],['../classmlx_1_1core_1_1_concatenate.html#a309a1c50e97f9925866433ee2841c474',1,'mlx::core::Concatenate::eval_gpu()'],['../classmlx_1_1core_1_1_conjugate.html#aff0a802166e3724db88ab5d3feb2d3de',1,'mlx::core::Conjugate::eval_gpu()'],['../classmlx_1_1core_1_1_contiguous.html#a519cd16fd0c55b371ea7625fbb37c70f',1,'mlx::core::Contiguous::eval_gpu()'],['../classmlx_1_1core_1_1_convolution.html#a30b64109eeb1778f002b99447dff9dd2',1,'mlx::core::Convolution::eval_gpu()'],['../classmlx_1_1core_1_1_copy.html#a1eda7b2ea771a168f67421f0d384b3a1',1,'mlx::core::Copy::eval_gpu()'],['../classmlx_1_1core_1_1_cos.html#a5ef41aafad595f6cdd8c535e36e12060',1,'mlx::core::Cos::eval_gpu()'],['../classmlx_1_1core_1_1_cosh.html#a23f71b43792934c3ec0ebe9b74f32559',1,'mlx::core::Cosh::eval_gpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#a7b3538681acbb20af3ed37b0877f6667',1,'mlx::core::CustomTransforms::eval_gpu()'],['../classmlx_1_1core_1_1_depends.html#ae5057f65e69490ad0add8eeda2b75e28',1,'mlx::core::Depends::eval_gpu()'],['../classmlx_1_1core_1_1_divide.html#abffda0ce37221ddc28dc9eea794f6bc7',1,'mlx::core::Divide::eval_gpu()'],['../classmlx_1_1core_1_1_div_mod.html#a003117c9ecf3c06a27248f72a76348dc',1,'mlx::core::DivMod::eval_gpu()'],['../classmlx_1_1core_1_1_select.html#a2a82b6cba4c386b2b87f225a4b08ea9b',1,'mlx::core::Select::eval_gpu()'],['../classmlx_1_1core_1_1_remainder.html#a7919ea9b84e42522d51bf0d5a396e161',1,'mlx::core::Remainder::eval_gpu()'],['../classmlx_1_1core_1_1_equal.html#ac3757001fec42ceb5ece2954df42161c',1,'mlx::core::Equal::eval_gpu()'],['../classmlx_1_1core_1_1_erf.html#ad8551be664d767dccc3c0d8cc1eca008',1,'mlx::core::Erf::eval_gpu()'],['../classmlx_1_1core_1_1_erf_inv.html#a4a2413d0634db1f3dae1806ddfa632db',1,'mlx::core::ErfInv::eval_gpu()'],['../classmlx_1_1core_1_1_exp.html#a7d63695a97a14760fd33b5d4e6590822',1,'mlx::core::Exp::eval_gpu()'],['../classmlx_1_1core_1_1_expm1.html#a82930071f4b77d883b300f77966aff5f',1,'mlx::core::Expm1::eval_gpu()'],['../classmlx_1_1core_1_1_expand_dims.html#ad350ede3abecc55371ddeb89fbba2b90',1,'mlx::core::ExpandDims::eval_gpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a1c21b26d1e9ad7c4da78ae845721b2dd',1,'mlx::core::FFT::eval_gpu()'],['../classmlx_1_1core_1_1_flatten.html#acb2219cc122d218b273af2cb9a882e7f',1,'mlx::core::Flatten::eval_gpu()'],['../classmlx_1_1core_1_1_floor.html#aaa29c83538099eb8f951c95a41f2eb65',1,'mlx::core::Floor::eval_gpu()'],['../classmlx_1_1core_1_1_full.html#aa54f99bb4cba12a551392dea56003872',1,'mlx::core::Full::eval_gpu()'],['../classmlx_1_1core_1_1_gather.html#aec48ee529cb2449915a7b27a3c4361e8',1,'mlx::core::Gather::eval_gpu()'],['../classmlx_1_1core_1_1_gather_axis.html#a1344749d33e4ea2cb80b69a5a4a21afc',1,'mlx::core::GatherAxis::eval_gpu()'],['../classmlx_1_1core_1_1_greater.html#ae8957cccf4c924d941f57a1bb751c878',1,'mlx::core::Greater::eval_gpu()'],['../classmlx_1_1core_1_1_greater_equal.html#ac246263b4548126c3d4ab7e392575d24',1,'mlx::core::GreaterEqual::eval_gpu()'],['../classmlx_1_1core_1_1_hadamard.html#a2470feb690f5463138490763c38b5733',1,'mlx::core::Hadamard::eval_gpu()'],['../classmlx_1_1core_1_1_imag.html#a247a4d059b0a99678c6be8c15e42c1e6',1,'mlx::core::Imag::eval_gpu()'],['../classmlx_1_1core_1_1_less.html#a353335ce06ddbe8498d86d129c835917',1,'mlx::core::Less::eval_gpu()'],['../classmlx_1_1core_1_1_less_equal.html#acf035a82b11e6f63742143ea540fedac',1,'mlx::core::LessEqual::eval_gpu()'],['../classmlx_1_1core_1_1_load.html#a06933e887ea94a4d01d81195c5e07a3d',1,'mlx::core::Load::eval_gpu()'],['../classmlx_1_1core_1_1_log.html#aaaa49e9455f3a197bc319646b5ca6390',1,'mlx::core::Log::eval_gpu()'],['../classmlx_1_1core_1_1_log1p.html#a1b97decae7338d46874e736c95fa7431',1,'mlx::core::Log1p::eval_gpu()'],['../classmlx_1_1core_1_1_logical_not.html#a1d0d2bc93f935eca6c85ef7bf67f2d6a',1,'mlx::core::LogicalNot::eval_gpu()'],['../classmlx_1_1core_1_1_logical_and.html#a132b2eedaa3978de5a5350da3c2ca40f',1,'mlx::core::LogicalAnd::eval_gpu()'],['../classmlx_1_1core_1_1_logical_or.html#a3be1da328f0f8620de2e4fc1d22a077a',1,'mlx::core::LogicalOr::eval_gpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#acace355b62ec00df649f9f99e8f2eb7a',1,'mlx::core::LogAddExp::eval_gpu()'],['../classmlx_1_1core_1_1_matmul.html#a8707a4e9b75c769e8f1dbca15c6a1ae7',1,'mlx::core::Matmul::eval_gpu()'],['../classmlx_1_1core_1_1_maximum.html#ade0f721b10a6b3a12bdadd34c48f72a7',1,'mlx::core::Maximum::eval_gpu()'],['../classmlx_1_1core_1_1_minimum.html#aadc68afa0afbe2103f19d161f5e0a2ba',1,'mlx::core::Minimum::eval_gpu()'],['../classmlx_1_1core_1_1_multiply.html#a634fcb4e981d8d3f4d94252caf25bee0',1,'mlx::core::Multiply::eval_gpu()'],['../classmlx_1_1core_1_1_negative.html#a97f1b316eace0c6d9e576d766940c75b',1,'mlx::core::Negative::eval_gpu()'],['../classmlx_1_1core_1_1_not_equal.html#a61179747e34e203150e9c660dfddb5f2',1,'mlx::core::NotEqual::eval_gpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#a2c98c42915fb2bfe12f5c99ea553eff5',1,'mlx::core::NumberOfElements::eval_gpu()'],['../classmlx_1_1core_1_1_pad.html#aefd4d3a5bd8b6b35b266c9e558ada153',1,'mlx::core::Pad::eval_gpu()'],['../classmlx_1_1core_1_1_partition.html#a8eca1be21ae9ccfda46e6f3e85f506ef',1,'mlx::core::Partition::eval_gpu()'],['../classmlx_1_1core_1_1_power.html#a80577d4c0853c24027777c90a1ec7e11',1,'mlx::core::Power::eval_gpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a2812ad007d695ed1aaf9cf706fb9c4b3',1,'mlx::core::QuantizedMatmul::eval_gpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a86eb048afc95646b2e96ec5493e3d887',1,'mlx::core::GatherQMM::eval_gpu()'],['../classmlx_1_1core_1_1_random_bits.html#a578756866665358577418e4cdd94aa3a',1,'mlx::core::RandomBits::eval_gpu()'],['../classmlx_1_1core_1_1_real.html#a1e209e88a43bdd1eea43ad0b03f9a7f2',1,'mlx::core::Real::eval_gpu()'],['../classmlx_1_1core_1_1_reshape.html#aa1e85f28471875750c47351520b56059',1,'mlx::core::Reshape::eval_gpu()'],['../classmlx_1_1core_1_1_reduce.html#ae9caaf42edadfe73ea208d98f526890f',1,'mlx::core::Reduce::eval_gpu()'],['../classmlx_1_1core_1_1_round.html#af7fe5ff8f3db166c203b4be4b07f13ec',1,'mlx::core::Round::eval_gpu()'],['../classmlx_1_1core_1_1_scan.html#aef22c6fc2b2cb2a907cd8965c7413dde',1,'mlx::core::Scan::eval_gpu()'],['../classmlx_1_1core_1_1_scatter.html#ab304345db3d8cfeea15e27461ae2e678',1,'mlx::core::Scatter::eval_gpu()'],['../classmlx_1_1core_1_1_scatter_axis.html#a715c3b959dc904faefb16edbb11f29d7',1,'mlx::core::ScatterAxis::eval_gpu()'],['../classmlx_1_1core_1_1_sigmoid.html#a7a6bd0222d51d7f25f2719a91ccdfeca',1,'mlx::core::Sigmoid::eval_gpu()'],['../classmlx_1_1core_1_1_sign.html#afa2b48b99a194106006b44af69ffda8b',1,'mlx::core::Sign::eval_gpu()'],['../classmlx_1_1core_1_1_sin.html#a6b59f1156cf8bdad8d45acd1d825cb5e',1,'mlx::core::Sin::eval_gpu()'],['../classmlx_1_1core_1_1_sinh.html#a5a1af2399f166d5b228b5e83a1837c75',1,'mlx::core::Sinh::eval_gpu()'],['../classmlx_1_1core_1_1_slice.html#aa53c21ff06a7c659e889af6b97d10a4a',1,'mlx::core::Slice::eval_gpu()'],['../classmlx_1_1core_1_1_slice_update.html#aac1a1d122e5697be057d63552141032b',1,'mlx::core::SliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_dynamic_slice.html#ab0a2e31c03f02a4f25700e240cf18e3e',1,'mlx::core::DynamicSlice::eval_gpu()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a249dab28690c45203c3995698de0cab7',1,'mlx::core::DynamicSliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_softmax.html#a35dac69ddcc7e2ec0e1a76fe93db85af',1,'mlx::core::Softmax::eval_gpu()'],['../classmlx_1_1core_1_1_sort.html#a4141c48f0e8670c728663f3722675382',1,'mlx::core::Sort::eval_gpu()'],['../classmlx_1_1core_1_1_split.html#a78ddda89c4daee73c74cfbc1e44656df',1,'mlx::core::Split::eval_gpu()'],['../classmlx_1_1core_1_1_square.html#a0ea2a78a5bb52daa4103263bf2f98045',1,'mlx::core::Square::eval_gpu()'],['../classmlx_1_1core_1_1_sqrt.html#a6d205e679a593d1ba20206c5c47ba501',1,'mlx::core::Sqrt::eval_gpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a907b96f0a1ce608e211d87ccf2b9ca89',1,'mlx::core::StopGradient::eval_gpu()'],['../classmlx_1_1core_1_1_subtract.html#a69021b23daf061764d97fabbc0f4f06c',1,'mlx::core::Subtract::eval_gpu()'],['../classmlx_1_1core_1_1_squeeze.html#a18d382c8bc59d60b38e9fd1cb70660fd',1,'mlx::core::Squeeze::eval_gpu()'],['../classmlx_1_1core_1_1_tan.html#aca7dbb4836507005a2032ac957a04d3f',1,'mlx::core::Tan::eval_gpu()'],['../classmlx_1_1core_1_1_tanh.html#a48df896599ae93dbce84a5c0f50cf761',1,'mlx::core::Tanh::eval_gpu()'],['../classmlx_1_1core_1_1_unflatten.html#adfbb8208355f9c3cb2e4cb1fd4fe788f',1,'mlx::core::Unflatten::eval_gpu()'],['../classmlx_1_1core_1_1_view.html#add6e12ff1e476fe1db7718b14f21b075',1,'mlx::core::View::eval_gpu()'],['../classmlx_1_1core_1_1_transpose.html#a38d25739c08aa594a6775015a1d7d92e',1,'mlx::core::Transpose::eval_gpu()'],['../classmlx_1_1core_1_1_q_r_f.html#ae5fa3482192f4713605cd07e7fc1c6c9',1,'mlx::core::QRF::eval_gpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a7067b2207f826a25549d571856b94e83',1,'mlx::core::SVD::eval_gpu()'],['../classmlx_1_1core_1_1_inverse.html#a086fbbc947ad232e01686ad063a78ed2',1,'mlx::core::Inverse::eval_gpu()'],['../classmlx_1_1core_1_1_cholesky.html#a8c918594bf129888044ef37fcae56795',1,'mlx::core::Cholesky::eval_gpu()'],['../classmlx_1_1core_1_1_eigh.html#a67775b41c0a15e356f08d51d9736baa2',1,'mlx::core::Eigh::eval_gpu()']]], + ['evaluated_41',['evaluated',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078a6fc3d7595445dd877584495f47535268',1,'mlx::core::array']]], + ['event_42',['Event',['../classmlx_1_1core_1_1_event.html',1,'mlx::core::Event'],['../classmlx_1_1core_1_1_event.html#a833506419b2110ad1abd89b2dd238b4d',1,'mlx::core::Event::Event()=default'],['../classmlx_1_1core_1_1_event.html#a13e4835f2ffb2cc22e29148a448ea184',1,'mlx::core::Event::Event(const Stream &steam)']]], + ['event_43',['event',['../classmlx_1_1core_1_1array.html#a0a8e4d6e67e739a712876bb36f88f9bf',1,'mlx::core::array']]], + ['event_2eh_44',['event.h',['../backend_2metal_2event_8h.html',1,'(Global Namespace)'],['../event_8h.html',1,'(Global Namespace)']]], + ['excess_45',['excess',['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#ada22a8bd8a89078cfa28874055c8e753',1,'mlx::steel::ChannelHelper< 1 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#acc490f3999230aa592c61bbed7eb7cfe',1,'mlx::steel::ChannelHelper< 2 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#aae404674763f3dc73c5ab29169f8b80f',1,'mlx::steel::ChannelHelper< 3 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#aecdd8331fec703d739a6f07b9b901ac8',1,'mlx::steel::ChannelHelper< 4 >::excess'],['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper< 1 >::excess'],['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper< 2 >::excess'],['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper< 3 >::excess'],['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper< 4 >::excess']]], + ['exec_46',['exec',['../classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2',1,'pocketfft::detail::cfftp::exec()'],['../classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac',1,'pocketfft::detail::rfftp::exec()'],['../classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c',1,'pocketfft::detail::fftblue::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d',1,'pocketfft::detail::pocketfft_c::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33',1,'pocketfft::detail::pocketfft_r::exec()'],['../classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058',1,'pocketfft::detail::T_dct1::exec()'],['../classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26',1,'pocketfft::detail::T_dst1::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac',1,'pocketfft::detail::T_dcst23::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708',1,'pocketfft::detail::T_dcst4::exec()'],['../classmlx_1_1core_1_1_jit_compiler.html#adcf98f940e1919388eaab907ea17a540',1,'mlx::core::JitCompiler::exec()']]], + ['exec_5fr_47',['exec_r',['../classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00',1,'pocketfft::detail::fftblue']]], + ['execc2c_48',['ExecC2C',['../structpocketfft_1_1detail_1_1_exec_c2_c.html',1,'pocketfft::detail']]], + ['execdcst_49',['ExecDcst',['../structpocketfft_1_1detail_1_1_exec_dcst.html',1,'pocketfft::detail']]], + ['exechartley_50',['ExecHartley',['../structpocketfft_1_1detail_1_1_exec_hartley.html',1,'pocketfft::detail']]], + ['execr2r_51',['ExecR2R',['../structpocketfft_1_1detail_1_1_exec_r2_r.html',1,'pocketfft::detail']]], + ['exp_52',['Exp',['../struct_exp.html',1,'Exp'],['../structmlx_1_1core_1_1detail_1_1_exp.html',1,'mlx::core::detail::Exp'],['../classmlx_1_1core_1_1_exp.html',1,'mlx::core::Exp'],['../classmlx_1_1core_1_1_exp.html#a1d0a618cbb91ab29ef53b57ff6ed6e06',1,'mlx::core::Exp::Exp()']]], + ['exp_53',['exp',['../namespacemlx_1_1core_1_1simd.html#a835d71dd0bb2f9494a397d9939696ec2',1,'mlx::core::simd::exp()'],['../namespacemetal.html#ac2a0b3618d922ac014baac8189d44650',1,'metal::exp()'],['../namespacemetal_1_1fast.html#ad3dbd387b63373c29e3449609f763ede',1,'metal::fast::exp()'],['../namespacemetal_1_1precise.html#a8d8d2d5700ce432b33cf47cf22528e8f',1,'metal::precise::exp()'],['../group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125',1,'mlx::core::exp()']]], + ['exp10_54',['exp10',['../namespacemetal.html#a4c63707d13c89364496a48906631c204',1,'metal::exp10()'],['../namespacemetal_1_1fast.html#a453122f982485cbb4e471b3ac282ee5e',1,'metal::fast::exp10()'],['../namespacemetal_1_1precise.html#af9addb343c967da3a83e9e123a8521fd',1,'metal::precise::exp10()']]], + ['exp2_55',['exp2',['../namespacemetal.html#a228201c20777848804a4d0589c1d33e7',1,'metal::exp2()'],['../namespacemetal_1_1fast.html#ac092b65a46720adaf22f6266671d2d71',1,'metal::fast::exp2()'],['../namespacemetal_1_1precise.html#a92a880bd2197efc0da0f8f0f7ec1e4c9',1,'metal::precise::exp2()']]], + ['expand_5fdims_56',['expand_dims',['../group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133',1,'mlx::core::expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a',1,'mlx::core::expand_dims(const array &a, int axis, StreamOrDevice s={})']]], + ['expanddims_57',['ExpandDims',['../classmlx_1_1core_1_1_expand_dims.html',1,'mlx::core::ExpandDims'],['../classmlx_1_1core_1_1_expand_dims.html#aea2479ea4dd93941eb83a22e087983a8',1,'mlx::core::ExpandDims::ExpandDims()']]], + ['expm1_58',['Expm1',['../struct_expm1.html',1,'Expm1'],['../structmlx_1_1core_1_1detail_1_1_expm1.html',1,'mlx::core::detail::Expm1'],['../classmlx_1_1core_1_1_expm1.html',1,'mlx::core::Expm1'],['../classmlx_1_1core_1_1_expm1.html#a47c2a1b2a4ef6bb07ba77c55ddddaec2',1,'mlx::core::Expm1::Expm1()']]], + ['expm1_59',['expm1',['../namespacemlx_1_1core_1_1simd.html#a9407980793ecff5d5eb19c9a2cbda1eb',1,'mlx::core::simd::expm1(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a464687a8809d0180035acc9af2943a94',1,'mlx::core::simd::expm1(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a8f73d1dac82177e0aeadaeda349c4f96',1,'mlx::core::simd::expm1(Simd< T, 1 > in)'],['../group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0',1,'mlx::core::expm1()']]], + ['expm1f_60',['expm1f',['../expm1f_8h.html#a87f66d30e185950f42ce3641783cdc40',1,'expm1f.h']]], + ['expm1f_2eh_61',['expm1f.h',['../expm1f_8h.html',1,'']]], + ['expm1f_5fscaled_5funchecked_62',['expm1f_scaled_unchecked',['../expm1f_8h.html#adf20e03405fba634ca8d01acac24592e',1,'expm1f.h']]], + ['export_2eh_63',['export.h',['../export_8h.html',1,'']]], + ['export_5ffunction_64',['export_function',['../namespacemlx_1_1core.html#a2afa4ea816ac9317200fd5c964fc89dc',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, const Args &args, bool shapeless=false)'],['../namespacemlx_1_1core.html#ace51644e2aa72f8d56b86eaa0e1a68b7',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, const Kwargs &kwargs, bool shapeless=false)'],['../namespacemlx_1_1core.html#a9692d7bb6de3456abc535d0f4bac7a94',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, const Args &args, const Kwargs &kwargs, bool shapeless=false)']]], + ['export_5fimpl_2eh_65',['export_impl.h',['../export__impl_8h.html',1,'']]], + ['export_5fto_5fdot_66',['export_to_dot',['../namespacemlx_1_1core.html#a30338cb7d259334e46dc7a4819716fa6',1,'mlx::core::export_to_dot(std::ostream &os, NodeNamer namer, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a57395bdf43d9c5c134e610c169222cca',1,'mlx::core::export_to_dot(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a839f94dbad44f0d37333006fc876b42e',1,'mlx::core::export_to_dot(std::ostream &os, Arrays &&... outputs)'],['../namespacemlx_1_1core.html#a12faa564e0d85a1daa77c08babb75261',1,'mlx::core::export_to_dot(std::ostream &os, NodeNamer namer, Arrays &&... outputs)']]], + ['exporter_67',['exporter',['../structmlx_1_1core_1_1_function_exporter.html#a3921e0f41f795708c33bda7c50a72055',1,'mlx::core::FunctionExporter::exporter(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, bool shapeless=false)'],['../structmlx_1_1core_1_1_function_exporter.html#ae53b863ba9fea62cc7c1b9eb55993269',1,'mlx::core::FunctionExporter::exporter(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, bool shapeless=false)'],['../structmlx_1_1core_1_1_function_exporter.html#a82eb4ca466592b97225e8252f1cdb2e4',1,'mlx::core::FunctionExporter::exporter(const std::string &path, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#a0303e26b737c9fd197ed9caa90fd21a7',1,'mlx::core::exporter(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#af2735df8513ecce88456585f5aea50f5',1,'mlx::core::exporter(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#af38d5718f517e50a590fdb3d63a90df1',1,'mlx::core::exporter(const std::string &path, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, bool shapeless=false)']]], + ['expsubop_68',['ExpSubOp',['../struct_exp_sub_op.html',1,'']]], + ['eye_69',['eye',['../group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3',1,'mlx::core::eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd',1,'mlx::core::eye(int n, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga61657db78ef35d41112d362c869c25d2',1,'mlx::core::eye(int n, int m, StreamOrDevice s={})'],['../group__ops.html#ga908a15b42834be498a46856c99dfc779',1,'mlx::core::eye(int n, int m, int k, StreamOrDevice s={})'],['../group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f',1,'mlx::core::eye(int n, StreamOrDevice s={})']]] ]; diff --git a/docs/build/html/search/all_6.js b/docs/build/html/search/all_6.js index 1d0b2acbe..2ae3f9a82 100644 --- a/docs/build/html/search/all_6.js +++ b/docs/build/html/search/all_6.js @@ -1,6 +1,6 @@ var searchData= [ - ['f_0',['f',['../unionmlx_1_1core_1_1detail_1_1_int_or_float.html#a90bd738e3b6b96a1a67685e8ab75b7ec',1,'mlx::core::detail::IntOrFloat::f'],['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a8fa14cdd754f91cc6554c9e71929cce7',1,'mlx::core::Dtype::f']]], + ['f_0',['f',['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a8fa14cdd754f91cc6554c9e71929cce7',1,'mlx::core::Dtype']]], ['f_5fout_5fjump_5fh_1',['f_out_jump_h',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a78d48b55cf182f000abece0e5e7fadcb',1,'mlx::steel::Conv2DGeneralJumpParams']]], ['f_5fout_5fjump_5fw_2',['f_out_jump_w',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a0fd755691482cb03ea4534b4a556c197',1,'mlx::steel::Conv2DGeneralJumpParams']]], ['f_5fwgt_5fjump_5fh_3',['f_wgt_jump_h',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a198ba0c2740ab4ded99345edf58917a7',1,'mlx::steel::Conv2DGeneralJumpParams']]], @@ -8,58 +8,56 @@ var searchData= ['fabs_5',['fabs',['../namespacemetal.html#a487eba718144be1325abcf66e109bb21',1,'metal::fabs()'],['../namespacemetal_1_1fast.html#a129fbd68c9df1a437e8959a25187f554',1,'metal::fast::fabs()'],['../namespacemetal_1_1precise.html#ae4c71d8bc8ef291036a7aaa05f8be3d1',1,'metal::precise::fabs()']]], ['false_5ftype_6',['false_type',['../namespacemlx_1_1steel.html#ab0ef721cedc2b5a97f60d76b765aff2e',1,'mlx::steel']]], ['fast_2eh_7',['fast.h',['../fast_8h.html',1,'']]], - ['fast_5ferf_8',['fast_erf',['../namespacemlx_1_1core_1_1detail.html#a90c9f6149af5adf4e2a95608d5f7b790',1,'mlx::core::detail']]], - ['fast_5ferfinv_9',['fast_erfinv',['../namespacemlx_1_1core_1_1detail.html#ad0ff3975e4c96317df1a2de0f7a30c1d',1,'mlx::core::detail']]], - ['fast_5fexp_10',['fast_exp',['../namespacemlx_1_1core_1_1detail.html#a2726436fc72d4a3f0030c89579b4d374',1,'mlx::core::detail']]], - ['fast_5fprimitives_2eh_11',['fast_primitives.h',['../fast__primitives_8h.html',1,'']]], - ['fdc_12',['fdc',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#a42efa2a1fddc11f71987377b9048f953',1,'mlx::steel::GEMMAddMMParams']]], - ['fdim_13',['fdim',['../namespacemetal.html#a85a560794be56d8116889c1ee2d78761',1,'metal::fdim()'],['../namespacemetal_1_1fast.html#a667df76100d5ea0ce5860ddae3e5a00b',1,'metal::fast::fdim()'],['../namespacemetal_1_1precise.html#af693e7c93de446e80dd1377f5e9e7260',1,'metal::precise::fdim()']]], - ['fence_14',['Fence',['../structmlx_1_1core_1_1metal_1_1_fence.html',1,'mlx::core::metal::Fence'],['../structmlx_1_1core_1_1metal_1_1_fence.html#a30bee4957ae595e04922952a8010fc79',1,'mlx::core::metal::Fence::Fence(MTL::Fence *fence)']]], - ['fence_15',['fence',['../structmlx_1_1core_1_1metal_1_1_fence.html#aeccd8f2b81418ae9fc446ae2b6e15b87',1,'mlx::core::metal::Fence::fence'],['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a876199de8da1efa9a362451029638499',1,'mlx::core::metal::DeviceStream::fence']]], - ['fence_5fmtx_16',['fence_mtx',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a6fa08cca881fc3798ae45994a11a4fcd',1,'mlx::core::metal::DeviceStream']]], - ['fft_17',['FFT',['../classmlx_1_1core_1_1_f_f_t.html',1,'mlx::core::FFT'],['../classmlx_1_1core_1_1_f_f_t.html#a0cdce626ed2c8eeeecc6949418437839',1,'mlx::core::FFT::FFT()']]], - ['fft_18',['fft',['../namespacemlx_1_1core_1_1metal.html#a39f43360d9e916fcf7e86c919b419554',1,'mlx::core::metal::fft()'],['../backend_2metal_2kernels_2fft_8h.html#a4010b0e151e5f01e610e9c32234458c7',1,'fft(): fft.h'],['../namespacemlx_1_1core_1_1fft.html#ad672de5ca029a6925b05f03bbebe5ad3',1,'mlx::core::fft::fft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a3fe55b7b6eba32c4c8b2d206036216e0',1,'mlx::core::fft::fft(const array &a, int axis=-1, StreamOrDevice s={})']]], - ['fft_2eh_19',['fft.h',['../backend_2metal_2kernels_2fft_8h.html',1,'(Global Namespace)'],['../fft_8h.html',1,'(Global Namespace)']]], - ['fft2_20',['fft2',['../namespacemlx_1_1core_1_1fft.html#a7a318ed0ab6a600cd7cba96ecbd72a1d',1,'mlx::core::fft::fft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a6eb0c5f8b33694ddb56748a97d17e8b7',1,'mlx::core::fft::fft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], - ['fftblue_21',['fftblue',['../classpocketfft_1_1detail_1_1fftblue.html',1,'pocketfft::detail::fftblue< T0 >'],['../classpocketfft_1_1detail_1_1fftblue.html#a8613b2131e7843bbca9e1cd4cc15bc01',1,'pocketfft::detail::fftblue::fftblue()']]], - ['fftn_22',['fftn',['../namespacemlx_1_1core_1_1fft.html#a2c6685806eef1cae8d2da53567d23e5c',1,'mlx::core::fft::fftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#aaa116429c2cb5bab20b464be890252c8',1,'mlx::core::fft::fftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a039a44197ad299a15a5847639292800c',1,'mlx::core::fft::fftn(const array &a, StreamOrDevice s={})']]], - ['filewriter_23',['FileWriter',['../classmlx_1_1core_1_1io_1_1_file_writer.html',1,'mlx::core::io::FileWriter'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a40b241ad540ee4aadc3a19a6b1ccfb4d',1,'mlx::core::io::FileWriter::FileWriter(std::string file_path)'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#aee57db8516361f17de3cf2087d9a87d9',1,'mlx::core::io::FileWriter::FileWriter(const FileWriter &)=delete'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a12b148df8a52136628728b508ee9c55e',1,'mlx::core::io::FileWriter::FileWriter(FileWriter &&other)']]], - ['fill_5fgpu_24',['fill_gpu',['../namespacemlx_1_1core.html#ae789dbda2a0f4e21aa0984f6a5dc986c',1,'mlx::core']]], - ['finfo_25',['finfo',['../structmlx_1_1core_1_1finfo.html',1,'mlx::core::finfo'],['../structmlx_1_1core_1_1finfo.html#a00dee158d75d12768d02a3e7b6709109',1,'mlx::core::finfo::finfo()']]], - ['finite_5fmax_26',['finite_max',['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits']]], - ['finite_5fmin_27',['finite_min',['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits']]], - ['flags_28',['Flags',['../structmlx_1_1core_1_1array_1_1_flags.html',1,'mlx::core::array']]], - ['flags_29',['flags',['../classmlx_1_1core_1_1array.html#a0a20a6065ae71b64c1e3aa22a45fd8a1',1,'mlx::core::array']]], - ['flatten_30',['Flatten',['../classmlx_1_1core_1_1_flatten.html',1,'mlx::core::Flatten'],['../classmlx_1_1core_1_1_flatten.html#ab9f72c6a90640b91f35a2bcc8dac8780',1,'mlx::core::Flatten::Flatten()']]], - ['flatten_31',['flatten',['../group__ops.html#ga50aa98754b412bb57c083f6e3e95061f',1,'mlx::core::flatten(const array &a, int start_axis, int end_axis=-1, StreamOrDevice s={})'],['../group__ops.html#gaa6adbc9c86f0ab27d8810a02e9e719fd',1,'mlx::core::flatten(const array &a, StreamOrDevice s={})']]], - ['flip_32',['flip',['../struct_m_l_x_conv_params.html#a8b30cda15eda20f84f12db868f21d0ef',1,'MLXConvParams']]], - ['float16_33',['float16',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa098e7844282e240fdee28a9dac11c1c6',1,'mlx::core::Dtype::float16'],['../namespacemlx_1_1core.html#abf228ee9d8ec48c03bb15adcc4e1f3ec',1,'mlx::core::float16']]], - ['float16_5ft_34',['float16_t',['../backend_2metal_2kernels_2utils_8h.html#acb8ddf4a29129846b673c50ba7078773',1,'float16_t: utils.h'],['../namespacemlx_1_1core.html#afbd2769c30e721afc85a7b9fb55b8e52',1,'mlx::core::float16_t']]], - ['float32_35',['float32',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daad33ec2b0bbea6d471a4706cea030e1e3',1,'mlx::core::Dtype::float32'],['../namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319',1,'mlx::core::float32']]], - ['float_5fto_5fbfloat_5fbits_36',['float_to_bfloat_bits',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31ce5e8e860295fa236e0d4b0befeae1',1,'bf16.h']]], - ['floating_37',['floating',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da374515b23d6f106696387776a6077d17',1,'mlx::core::Dtype::floating'],['../namespacemlx_1_1core.html#ac9f9ea13cf0661e671569d37d14a128a',1,'mlx::core::floating']]], - ['floor_38',['Floor',['../struct_floor.html',1,'Floor'],['../structmlx_1_1core_1_1detail_1_1_floor.html',1,'mlx::core::detail::Floor'],['../classmlx_1_1core_1_1_floor.html',1,'mlx::core::Floor'],['../classmlx_1_1core_1_1_floor.html#ada4e979b784b732696313d7094e91340',1,'mlx::core::Floor::Floor()']]], - ['floor_39',['floor',['../namespacemetal.html#a020790f30c28a9982c4a83deaa258277',1,'metal::floor()'],['../namespacemetal_1_1fast.html#ac012ce1701c2339914f15cce9f2c632f',1,'metal::fast::floor()'],['../namespacemetal_1_1precise.html#a66e02b028e3cecfe7c80773460dc7925',1,'metal::precise::floor()'],['../group__ops.html#ga8d656904aa2690b60955ae745aecfc30',1,'mlx::core::floor(const array &a, StreamOrDevice s={})']]], - ['floor_5fdivide_40',['floor_divide',['../group__ops.html#ga05b4c6054d028107869511f927da01cd',1,'mlx::core']]], - ['floordivide_41',['FloorDivide',['../struct_floor_divide.html',1,'']]], - ['fma_42',['fma',['../namespacemetal.html#a6301a78d69ff14a06194ca85a0c7d326',1,'metal::fma()'],['../namespacemetal_1_1fast.html#aebcd6e951da6f7157ec219eb7a8f1ddd',1,'metal::fast::fma()'],['../namespacemetal_1_1precise.html#a49391a64d6b66fe3a212516b316a2144',1,'metal::precise::fma()']]], - ['fmax_43',['fmax',['../namespacemetal.html#a0558e56fdb94b456deea6a4eb53964ed',1,'metal::fmax()'],['../namespacemetal_1_1fast.html#a26e3257cf877154f8a0d434be0bdb034',1,'metal::fast::fmax()'],['../namespacemetal_1_1precise.html#ac7d49f921c2883caf9eec66efc4de1cd',1,'metal::precise::fmax()']]], - ['fmax3_44',['fmax3',['../namespacemetal.html#ae0c1a7ba1a7449adc64d00b2a29e67f6',1,'metal::fmax3()'],['../namespacemetal_1_1fast.html#a5c6a3a389f348e1f92e8392b765a32c7',1,'metal::fast::fmax3()'],['../namespacemetal_1_1precise.html#adf750e51bd83d569994d0967029e3bdc',1,'metal::precise::fmax3()']]], - ['fmedian3_45',['fmedian3',['../namespacemetal.html#aa35227450d943fb88cf43162aa9d8c49',1,'metal::fmedian3()'],['../namespacemetal_1_1fast.html#a923869181c3f576f2d86fba5bfa85633',1,'metal::fast::fmedian3()'],['../namespacemetal_1_1precise.html#a48d1d0be889de4043b775bb6b030a989',1,'metal::precise::fmedian3()']]], - ['fmin_46',['fmin',['../namespacemetal.html#a66ac19825ea79b8294e243ae6d0b3d3c',1,'metal::fmin()'],['../namespacemetal_1_1fast.html#a7e202ec52bf12bfabdf2265b300acbfa',1,'metal::fast::fmin()'],['../namespacemetal_1_1precise.html#a18df8eb481dfa56c92ad31b5bab8e069',1,'metal::precise::fmin()']]], - ['fmin3_47',['fmin3',['../namespacemetal.html#ae2acd25f2241f00aaf89ff48f132a879',1,'metal::fmin3()'],['../namespacemetal_1_1fast.html#a9531c6a4a520927523961e6eb6b94c1a',1,'metal::fast::fmin3()'],['../namespacemetal_1_1precise.html#a5bb710e6742996d32225a8f54a0f116c',1,'metal::precise::fmin3()']]], - ['fmod_48',['fmod',['../namespacemetal.html#a2ff952d4d596a7969b2a3035fc2fda58',1,'metal::fmod()'],['../namespacemetal_1_1fast.html#adbec09f18a89f773d7e368ef04a69526',1,'metal::fast::fmod()'],['../namespacemetal_1_1precise.html#aa99937178a1fc8158054e328eeeae648',1,'metal::precise::fmod()']]], - ['forward_49',['FORWARD',['../namespacepocketfft_1_1detail.html#aecc5444a333360628be65a6f91ceb824',1,'pocketfft::detail::FORWARD'],['../namespacepocketfft.html#aecc5444a333360628be65a6f91ceb824',1,'pocketfft::FORWARD']]], - ['forward_50',['forward',['../structpocketfft_1_1detail_1_1_exec_c2_c.html#a63e27292b327597674deede9debe1c43',1,'pocketfft::detail::ExecC2C::forward'],['../structpocketfft_1_1detail_1_1_exec_r2_r.html#a5ec66ebb2ccd079f62b068ddd1fc7bdf',1,'pocketfft::detail::ExecR2R::forward']]], - ['four_5fstep_5ffft_51',['four_step_fft',['../backend_2metal_2kernels_2fft_8h.html#a6558a8205ee4c3e4767bafa93f7606de',1,'fft.h']]], - ['fp16_2eh_52',['fp16.h',['../fp16_8h.html',1,'']]], - ['fp16_5fbf16_5fbinop_5fhelper_53',['fp16_bf16_binop_helper',['../half__types_8h.html#a1f0d5d395d403bde764fffe4846617f9',1,'half_types.h']]], - ['fract_54',['fract',['../namespacemetal.html#a6b1c15d251aeaacb1f4338a5e152ae78',1,'metal::fract()'],['../namespacemetal_1_1fast.html#aa8bb448827503e485eb649eb3edb2d4c',1,'metal::fast::fract()'],['../namespacemetal_1_1precise.html#a0f21c19332a90df1a8ff507a813b5757',1,'metal::precise::fract()']]], - ['frag_5fat_55',['frag_at',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1a6b1446e8c8da46885bbaa8e8fdc7e4',1,'mlx::steel::MMATile::frag_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad476e1d9a12178fb35c207312339e485',1,'mlx::steel::MMATile::frag_at(const short i, const short j) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1a6b1446e8c8da46885bbaa8e8fdc7e4',1,'mlx::steel::MMATile::frag_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad476e1d9a12178fb35c207312339e485',1,'mlx::steel::MMATile::frag_at(const short i, const short j) const']]], - ['frag_5ftype_56',['frag_type',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aac25cd0a9bdf24aa2af809c95f0bd171',1,'mlx::steel::MMATile::frag_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aac25cd0a9bdf24aa2af809c95f0bd171',1,'mlx::steel::MMATile::frag_type']]], - ['free_57',['free',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#ae963d551be646ae0e13df2c16f2beefb',1,'mlx::core::allocator::Allocator::free()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#a84b50d1a3cbffa12c1a6cf0ed8c71079',1,'mlx::core::allocator::CommonAllocator::free()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a109a0a37fb0b3be381a62dc3b1a54bf0',1,'mlx::core::metal::MetalAllocator::free()'],['../namespacemlx_1_1core_1_1allocator.html#a77f0a1215be242db6485612bcb273af5',1,'mlx::core::allocator::free()']]], - ['frexp_58',['frexp',['../namespacemetal.html#ac89d4ef524d21a301da6c37dbd95ff9f',1,'metal::frexp()'],['../namespacemetal_1_1fast.html#a23902df22aeaa859ef673a36381387c2',1,'metal::fast::frexp()'],['../namespacemetal_1_1precise.html#a0fbb1624c308b97380f894f92fd858b4',1,'metal::precise::frexp()']]], - ['full_59',['Full',['../classmlx_1_1core_1_1_full.html',1,'mlx::core::Full'],['../classmlx_1_1core_1_1_full.html#aafcb86a2e41353853ec48c717e0c54d6',1,'mlx::core::Full::Full()']]], - ['full_60',['full',['../group__ops.html#ga1cf232308668fe3f4214c8b895ed4aee',1,'mlx::core::full(Shape shape, array vals, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga59f6c844cbb173e108c3eeb11801f8c6',1,'mlx::core::full(Shape shape, array vals, StreamOrDevice s={})'],['../group__ops.html#gaf073760b7b51fe35932da0d81c531a55',1,'mlx::core::full(Shape shape, T val, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaf6f2cce92aff9b71756a3cc3c961fd5a',1,'mlx::core::full(Shape shape, T val, StreamOrDevice s={})']]], - ['functionexporter_61',['FunctionExporter',['../structmlx_1_1core_1_1_function_exporter.html',1,'mlx::core::FunctionExporter'],['../structmlx_1_1core_1_1_function_exporter.html#a97ff954496a084d96e73a9c520c9dc0c',1,'mlx::core::FunctionExporter::FunctionExporter(const FunctionExporter &)=delete'],['../structmlx_1_1core_1_1_function_exporter.html#ac317e349139f8a6cd70d63ef65368fc2',1,'mlx::core::FunctionExporter::FunctionExporter(FunctionExporter &&other)=default']]] + ['fast_5fprimitives_2eh_8',['fast_primitives.h',['../fast__primitives_8h.html',1,'']]], + ['fdc_9',['fdc',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#a42efa2a1fddc11f71987377b9048f953',1,'mlx::steel::GEMMAddMMParams']]], + ['fdim_10',['fdim',['../namespacemetal.html#a85a560794be56d8116889c1ee2d78761',1,'metal::fdim()'],['../namespacemetal_1_1fast.html#a667df76100d5ea0ce5860ddae3e5a00b',1,'metal::fast::fdim()'],['../namespacemetal_1_1precise.html#af693e7c93de446e80dd1377f5e9e7260',1,'metal::precise::fdim()']]], + ['fence_11',['Fence',['../classmlx_1_1core_1_1_fence.html',1,'mlx::core::Fence'],['../structmlx_1_1core_1_1metal_1_1_fence.html',1,'mlx::core::metal::Fence'],['../structmlx_1_1core_1_1metal_1_1_fence.html#a30bee4957ae595e04922952a8010fc79',1,'mlx::core::metal::Fence::Fence()'],['../classmlx_1_1core_1_1_fence.html#a3e3ed08eb6a1025b051b5a435f6f95f7',1,'mlx::core::Fence::Fence()']]], + ['fence_12',['fence',['../structmlx_1_1core_1_1metal_1_1_fence.html#aeccd8f2b81418ae9fc446ae2b6e15b87',1,'mlx::core::metal::Fence::fence'],['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a876199de8da1efa9a362451029638499',1,'mlx::core::metal::DeviceStream::fence']]], + ['fence_2eh_13',['fence.h',['../fence_8h.html',1,'']]], + ['fence_5fmtx_14',['fence_mtx',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a6fa08cca881fc3798ae45994a11a4fcd',1,'mlx::core::metal::DeviceStream']]], + ['fft_15',['FFT',['../classmlx_1_1core_1_1_f_f_t.html',1,'mlx::core::FFT'],['../classmlx_1_1core_1_1_f_f_t.html#a0cdce626ed2c8eeeecc6949418437839',1,'mlx::core::FFT::FFT()']]], + ['fft_16',['fft',['../namespacemlx_1_1core_1_1metal.html#a39f43360d9e916fcf7e86c919b419554',1,'mlx::core::metal::fft()'],['../backend_2metal_2kernels_2fft_8h.html#a4010b0e151e5f01e610e9c32234458c7',1,'fft(): fft.h'],['../namespacemlx_1_1core_1_1fft.html#ad672de5ca029a6925b05f03bbebe5ad3',1,'mlx::core::fft::fft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a3fe55b7b6eba32c4c8b2d206036216e0',1,'mlx::core::fft::fft(const array &a, int axis=-1, StreamOrDevice s={})']]], + ['fft_2eh_17',['fft.h',['../backend_2metal_2kernels_2fft_8h.html',1,'(Global Namespace)'],['../fft_8h.html',1,'(Global Namespace)']]], + ['fft2_18',['fft2',['../namespacemlx_1_1core_1_1fft.html#a7a318ed0ab6a600cd7cba96ecbd72a1d',1,'mlx::core::fft::fft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a6eb0c5f8b33694ddb56748a97d17e8b7',1,'mlx::core::fft::fft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], + ['fftblue_19',['fftblue',['../classpocketfft_1_1detail_1_1fftblue.html',1,'pocketfft::detail::fftblue< T0 >'],['../classpocketfft_1_1detail_1_1fftblue.html#a8613b2131e7843bbca9e1cd4cc15bc01',1,'pocketfft::detail::fftblue::fftblue()']]], + ['fftn_20',['fftn',['../namespacemlx_1_1core_1_1fft.html#a2c6685806eef1cae8d2da53567d23e5c',1,'mlx::core::fft::fftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#aaa116429c2cb5bab20b464be890252c8',1,'mlx::core::fft::fftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a039a44197ad299a15a5847639292800c',1,'mlx::core::fft::fftn(const array &a, StreamOrDevice s={})']]], + ['filewriter_21',['FileWriter',['../classmlx_1_1core_1_1io_1_1_file_writer.html',1,'mlx::core::io::FileWriter'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a40b241ad540ee4aadc3a19a6b1ccfb4d',1,'mlx::core::io::FileWriter::FileWriter(std::string file_path)'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#aee57db8516361f17de3cf2087d9a87d9',1,'mlx::core::io::FileWriter::FileWriter(const FileWriter &)=delete'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a12b148df8a52136628728b508ee9c55e',1,'mlx::core::io::FileWriter::FileWriter(FileWriter &&other)']]], + ['fill_5fgpu_22',['fill_gpu',['../namespacemlx_1_1core.html#ae789dbda2a0f4e21aa0984f6a5dc986c',1,'mlx::core']]], + ['finfo_23',['finfo',['../structmlx_1_1core_1_1finfo.html',1,'mlx::core::finfo'],['../structmlx_1_1core_1_1finfo.html#a00dee158d75d12768d02a3e7b6709109',1,'mlx::core::finfo::finfo()']]], + ['finite_5fmax_24',['finite_max',['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits::finite_max'],['../struct_limits_3_01uint8__t_01_4.html#a55f48b89033e8c8683f8540ec6b23f02',1,'Limits< uint8_t >::finite_max'],['../struct_limits_3_01uint16__t_01_4.html#a9d517d8265ea1898b6b16e91b8595146',1,'Limits< uint16_t >::finite_max'],['../struct_limits_3_01uint32__t_01_4.html#a0698139f3fe440d7aa08ac5029d72235',1,'Limits< uint32_t >::finite_max'],['../struct_limits_3_01uint64__t_01_4.html#aff101ff38be5ccdbb9790aecb3069071',1,'Limits< uint64_t >::finite_max'],['../struct_limits_3_01int8__t_01_4.html#a24cdab873e0fb778393c69f1dc9ecf73',1,'Limits< int8_t >::finite_max'],['../struct_limits_3_01int16__t_01_4.html#acb2936d1cdbf347a9a014c8e036a5782',1,'Limits< int16_t >::finite_max'],['../struct_limits_3_01int32__t_01_4.html#aa9ed9f0e8c7400d8fc92e1cba9588794',1,'Limits< int32_t >::finite_max'],['../struct_limits_3_01int64__t_01_4.html#a6c7254b641878fa0fb9538814c45457a',1,'Limits< int64_t >::finite_max'],['../struct_limits_3_01half_01_4.html#aedaf0190aabf23da20510e558e2690b4',1,'Limits< half >::finite_max'],['../struct_limits_3_01float_01_4.html#a291eea590113fc1858b7f83f2e0c977d',1,'Limits< float >::finite_max'],['../struct_limits_3_01bfloat16__t_01_4.html#a6337dc35207b3f6f7185cd73eabac211',1,'Limits< bfloat16_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< bfloat16_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< bool >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< complex64_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< float >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< half >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< int16_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< int32_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< int64_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< int8_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< uint16_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< uint32_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< uint64_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< uint8_t >::finite_max']]], + ['finite_5fmin_25',['finite_min',['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits::finite_min'],['../struct_limits_3_01uint8__t_01_4.html#a60cea662971b09f78ef19f1da4760b73',1,'Limits< uint8_t >::finite_min'],['../struct_limits_3_01uint16__t_01_4.html#a1a7c029eccba4ab89743abdfaabfa7b4',1,'Limits< uint16_t >::finite_min'],['../struct_limits_3_01uint32__t_01_4.html#ad5d811fce62f44488190ff01d9e7608b',1,'Limits< uint32_t >::finite_min'],['../struct_limits_3_01uint64__t_01_4.html#a6556e7de6e0670da8f768bbc4479deae',1,'Limits< uint64_t >::finite_min'],['../struct_limits_3_01int8__t_01_4.html#a592797ce82cc2f7e27b0c477165b3452',1,'Limits< int8_t >::finite_min'],['../struct_limits_3_01int16__t_01_4.html#a158c4dbc9333939691b1637478e28e39',1,'Limits< int16_t >::finite_min'],['../struct_limits_3_01int32__t_01_4.html#ad9777dc6a84dcb9c63b598189ff0a4ff',1,'Limits< int32_t >::finite_min'],['../struct_limits_3_01int64__t_01_4.html#af80726162b44a741aae679f1fe85142a',1,'Limits< int64_t >::finite_min'],['../struct_limits_3_01half_01_4.html#a98d153748be68dbb428c50df3c0285ab',1,'Limits< half >::finite_min'],['../struct_limits_3_01float_01_4.html#afaa5162a47083447c5ac758d6dc02a8b',1,'Limits< float >::finite_min'],['../struct_limits_3_01bfloat16__t_01_4.html#ae4132a37154707cc31bbc1734636cf36',1,'Limits< bfloat16_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< bfloat16_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< bool >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< complex64_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< float >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< half >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< int16_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< int32_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< int64_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< int8_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< uint16_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< uint32_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< uint64_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< uint8_t >::finite_min']]], + ['flags_26',['Flags',['../structmlx_1_1core_1_1array_1_1_flags.html',1,'mlx::core::array']]], + ['flags_27',['flags',['../classmlx_1_1core_1_1array.html#a0a20a6065ae71b64c1e3aa22a45fd8a1',1,'mlx::core::array']]], + ['flatten_28',['Flatten',['../classmlx_1_1core_1_1_flatten.html',1,'mlx::core::Flatten'],['../classmlx_1_1core_1_1_flatten.html#ab9f72c6a90640b91f35a2bcc8dac8780',1,'mlx::core::Flatten::Flatten()']]], + ['flatten_29',['flatten',['../group__ops.html#ga50aa98754b412bb57c083f6e3e95061f',1,'mlx::core::flatten(const array &a, int start_axis, int end_axis=-1, StreamOrDevice s={})'],['../group__ops.html#gaa6adbc9c86f0ab27d8810a02e9e719fd',1,'mlx::core::flatten(const array &a, StreamOrDevice s={})']]], + ['flip_30',['flip',['../struct_m_l_x_conv_params.html#a8b30cda15eda20f84f12db868f21d0ef',1,'MLXConvParams']]], + ['float16_31',['float16',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa098e7844282e240fdee28a9dac11c1c6',1,'mlx::core::Dtype::float16'],['../namespacemlx_1_1core.html#abf228ee9d8ec48c03bb15adcc4e1f3ec',1,'mlx::core::float16']]], + ['float16_5ft_32',['float16_t',['../backend_2metal_2kernels_2utils_8h.html#acb8ddf4a29129846b673c50ba7078773',1,'float16_t: utils.h'],['../namespacemlx_1_1core.html#afbd2769c30e721afc85a7b9fb55b8e52',1,'mlx::core::float16_t']]], + ['float32_33',['float32',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daad33ec2b0bbea6d471a4706cea030e1e3',1,'mlx::core::Dtype::float32'],['../namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319',1,'mlx::core::float32']]], + ['float_5fto_5fbfloat_5fbits_34',['float_to_bfloat_bits',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31ce5e8e860295fa236e0d4b0befeae1',1,'bf16.h']]], + ['floating_35',['floating',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da374515b23d6f106696387776a6077d17',1,'mlx::core::Dtype::floating'],['../namespacemlx_1_1core.html#ac9f9ea13cf0661e671569d37d14a128a',1,'mlx::core::floating']]], + ['floor_36',['Floor',['../struct_floor.html',1,'Floor'],['../structmlx_1_1core_1_1detail_1_1_floor.html',1,'mlx::core::detail::Floor'],['../classmlx_1_1core_1_1_floor.html',1,'mlx::core::Floor'],['../classmlx_1_1core_1_1_floor.html#ada4e979b784b732696313d7094e91340',1,'mlx::core::Floor::Floor()']]], + ['floor_37',['floor',['../namespacemlx_1_1core_1_1simd.html#a8e22c484298d9af10b6604c835e52052',1,'mlx::core::simd::floor(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#aa396efa6e9c94f4ac1f8381d5e07f069',1,'mlx::core::simd::floor(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#ad6b89aecafefe57b6ce69bec143ccd6e',1,'mlx::core::simd::floor(Simd< float16_t, N > a)'],['../namespacemetal.html#a020790f30c28a9982c4a83deaa258277',1,'metal::floor()'],['../namespacemetal_1_1fast.html#ac012ce1701c2339914f15cce9f2c632f',1,'metal::fast::floor()'],['../namespacemetal_1_1precise.html#a66e02b028e3cecfe7c80773460dc7925',1,'metal::precise::floor()'],['../group__ops.html#ga8d656904aa2690b60955ae745aecfc30',1,'mlx::core::floor(const array &a, StreamOrDevice s={})']]], + ['floor_5fdivide_38',['floor_divide',['../group__ops.html#ga05b4c6054d028107869511f927da01cd',1,'mlx::core']]], + ['floordivide_39',['FloorDivide',['../struct_floor_divide.html',1,'']]], + ['fma_40',['fma',['../namespacemlx_1_1core_1_1simd.html#a9ddc7f119cc1dc04372ec1adcaf55f70',1,'mlx::core::simd::fma(Simd< T, N > x, Simd< T, N > y, U z)'],['../namespacemlx_1_1core_1_1simd.html#a8aa81ebff4c26f21cae2253d885fd87a',1,'mlx::core::simd::fma(Simd< T, 1 > x, Simd< T, 1 > y, U z)'],['../namespacemlx_1_1core_1_1simd.html#a99099c338377518773b55d4042f9410d',1,'mlx::core::simd::fma(Simd< float16_t, N > x, Simd< float16_t, N > y, T z)'],['../namespacemetal.html#a6301a78d69ff14a06194ca85a0c7d326',1,'metal::fma()'],['../namespacemetal_1_1fast.html#aebcd6e951da6f7157ec219eb7a8f1ddd',1,'metal::fast::fma()'],['../namespacemetal_1_1precise.html#a49391a64d6b66fe3a212516b316a2144',1,'metal::precise::fma()']]], + ['fmax_41',['fmax',['../namespacemetal.html#a0558e56fdb94b456deea6a4eb53964ed',1,'metal::fmax()'],['../namespacemetal_1_1fast.html#a26e3257cf877154f8a0d434be0bdb034',1,'metal::fast::fmax()'],['../namespacemetal_1_1precise.html#ac7d49f921c2883caf9eec66efc4de1cd',1,'metal::precise::fmax()']]], + ['fmax3_42',['fmax3',['../namespacemetal.html#ae0c1a7ba1a7449adc64d00b2a29e67f6',1,'metal::fmax3()'],['../namespacemetal_1_1fast.html#a5c6a3a389f348e1f92e8392b765a32c7',1,'metal::fast::fmax3()'],['../namespacemetal_1_1precise.html#adf750e51bd83d569994d0967029e3bdc',1,'metal::precise::fmax3()']]], + ['fmedian3_43',['fmedian3',['../namespacemetal.html#aa35227450d943fb88cf43162aa9d8c49',1,'metal::fmedian3()'],['../namespacemetal_1_1fast.html#a923869181c3f576f2d86fba5bfa85633',1,'metal::fast::fmedian3()'],['../namespacemetal_1_1precise.html#a48d1d0be889de4043b775bb6b030a989',1,'metal::precise::fmedian3()']]], + ['fmin_44',['fmin',['../namespacemetal.html#a66ac19825ea79b8294e243ae6d0b3d3c',1,'metal::fmin()'],['../namespacemetal_1_1fast.html#a7e202ec52bf12bfabdf2265b300acbfa',1,'metal::fast::fmin()'],['../namespacemetal_1_1precise.html#a18df8eb481dfa56c92ad31b5bab8e069',1,'metal::precise::fmin()']]], + ['fmin3_45',['fmin3',['../namespacemetal.html#ae2acd25f2241f00aaf89ff48f132a879',1,'metal::fmin3()'],['../namespacemetal_1_1fast.html#a9531c6a4a520927523961e6eb6b94c1a',1,'metal::fast::fmin3()'],['../namespacemetal_1_1precise.html#a5bb710e6742996d32225a8f54a0f116c',1,'metal::precise::fmin3()']]], + ['fmod_46',['fmod',['../namespacemetal.html#a2ff952d4d596a7969b2a3035fc2fda58',1,'metal::fmod()'],['../namespacemetal_1_1fast.html#adbec09f18a89f773d7e368ef04a69526',1,'metal::fast::fmod()'],['../namespacemetal_1_1precise.html#aa99937178a1fc8158054e328eeeae648',1,'metal::precise::fmod()']]], + ['forward_47',['FORWARD',['../namespacepocketfft_1_1detail.html#aecc5444a333360628be65a6f91ceb824',1,'pocketfft::detail::FORWARD'],['../namespacepocketfft.html#aecc5444a333360628be65a6f91ceb824',1,'pocketfft::FORWARD']]], + ['forward_48',['forward',['../structpocketfft_1_1detail_1_1_exec_c2_c.html#a63e27292b327597674deede9debe1c43',1,'pocketfft::detail::ExecC2C::forward'],['../structpocketfft_1_1detail_1_1_exec_r2_r.html#a5ec66ebb2ccd079f62b068ddd1fc7bdf',1,'pocketfft::detail::ExecR2R::forward']]], + ['four_5fstep_5ffft_49',['four_step_fft',['../backend_2metal_2kernels_2fft_8h.html#a6558a8205ee4c3e4767bafa93f7606de',1,'fft.h']]], + ['fp16_2eh_50',['fp16.h',['../fp16_8h.html',1,'']]], + ['fp16_5fbf16_5fbinop_5fhelper_51',['fp16_bf16_binop_helper',['../half__types_8h.html#a1f0d5d395d403bde764fffe4846617f9',1,'half_types.h']]], + ['fract_52',['fract',['../namespacemetal.html#a6b1c15d251aeaacb1f4338a5e152ae78',1,'metal::fract()'],['../namespacemetal_1_1fast.html#aa8bb448827503e485eb649eb3edb2d4c',1,'metal::fast::fract()'],['../namespacemetal_1_1precise.html#a0f21c19332a90df1a8ff507a813b5757',1,'metal::precise::fract()']]], + ['frag_5fat_53',['frag_at',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1a6b1446e8c8da46885bbaa8e8fdc7e4',1,'mlx::steel::MMATile::frag_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad476e1d9a12178fb35c207312339e485',1,'mlx::steel::MMATile::frag_at(const short i, const short j) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1a6b1446e8c8da46885bbaa8e8fdc7e4',1,'mlx::steel::MMATile::frag_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad476e1d9a12178fb35c207312339e485',1,'mlx::steel::MMATile::frag_at(const short i, const short j) const']]], + ['frag_5ftype_54',['frag_type',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8536bfaa108031c2ea3e9ccdc766ee5b',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::frag_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aac25cd0a9bdf24aa2af809c95f0bd171',1,'mlx::steel::MMATile::frag_type'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8536bfaa108031c2ea3e9ccdc766ee5b',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::frag_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aac25cd0a9bdf24aa2af809c95f0bd171',1,'mlx::steel::MMATile::frag_type']]], + ['free_55',['free',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#ae963d551be646ae0e13df2c16f2beefb',1,'mlx::core::allocator::Allocator::free()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#a84b50d1a3cbffa12c1a6cf0ed8c71079',1,'mlx::core::allocator::CommonAllocator::free()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a109a0a37fb0b3be381a62dc3b1a54bf0',1,'mlx::core::metal::MetalAllocator::free()'],['../namespacemlx_1_1core_1_1allocator.html#a77f0a1215be242db6485612bcb273af5',1,'mlx::core::allocator::free()']]], + ['frexp_56',['frexp',['../namespacemetal.html#ac89d4ef524d21a301da6c37dbd95ff9f',1,'metal::frexp()'],['../namespacemetal_1_1fast.html#a23902df22aeaa859ef673a36381387c2',1,'metal::fast::frexp()'],['../namespacemetal_1_1precise.html#a0fbb1624c308b97380f894f92fd858b4',1,'metal::precise::frexp()']]], + ['full_57',['Full',['../classmlx_1_1core_1_1_full.html',1,'mlx::core::Full'],['../classmlx_1_1core_1_1_full.html#aafcb86a2e41353853ec48c717e0c54d6',1,'mlx::core::Full::Full()']]], + ['full_58',['full',['../group__ops.html#ga1cf232308668fe3f4214c8b895ed4aee',1,'mlx::core::full(Shape shape, array vals, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga59f6c844cbb173e108c3eeb11801f8c6',1,'mlx::core::full(Shape shape, array vals, StreamOrDevice s={})'],['../group__ops.html#gaf073760b7b51fe35932da0d81c531a55',1,'mlx::core::full(Shape shape, T val, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaf6f2cce92aff9b71756a3cc3c961fd5a',1,'mlx::core::full(Shape shape, T val, StreamOrDevice s={})']]], + ['functionexporter_59',['FunctionExporter',['../structmlx_1_1core_1_1_function_exporter.html',1,'mlx::core::FunctionExporter'],['../structmlx_1_1core_1_1_function_exporter.html#a97ff954496a084d96e73a9c520c9dc0c',1,'mlx::core::FunctionExporter::FunctionExporter(const FunctionExporter &)=delete'],['../structmlx_1_1core_1_1_function_exporter.html#ac317e349139f8a6cd70d63ef65368fc2',1,'mlx::core::FunctionExporter::FunctionExporter(FunctionExporter &&other)=default']]] ]; diff --git a/docs/build/html/search/all_7.js b/docs/build/html/search/all_7.js index b0b953297..290f59457 100644 --- a/docs/build/html/search/all_7.js +++ b/docs/build/html/search/all_7.js @@ -3,112 +3,116 @@ var searchData= ['gather_0',['Gather',['../classmlx_1_1core_1_1_gather.html',1,'mlx::core::Gather'],['../classmlx_1_1core_1_1_gather.html#af24220fde798f2ad17cdce297c0dbc43',1,'mlx::core::Gather::Gather()']]], ['gather_1',['gather',['../namespacemlx_1_1core_1_1metal.html#a545de371fefba1feec2e70b7e9f4187c',1,'mlx::core::metal::gather()'],['../group__ops.html#ga8fcc3ad0677c834c36b72d5b2ebba6d0',1,'mlx::core::gather(const array &a, const std::vector< array > &indices, const std::vector< int > &axes, const Shape &slice_sizes, StreamOrDevice s={})'],['../group__ops.html#gafe2bd174c9953ed7f12664f7abaca0e6',1,'mlx::core::gather(const array &a, const array &indices, int axis, const Shape &slice_sizes, StreamOrDevice s={})']]], ['gather_2eh_2',['gather.h',['../gather_8h.html',1,'']]], - ['gather_5fbias_3',['gather_bias',['../steel__gemm__fused_8h.html#aaaf17233201156be684f858bfd0f1b67',1,'steel_gemm_fused.h']]], - ['gather_5fimpl_4',['gather_impl',['../gather_8h.html#a5cf4ff9849edd2d99ff24ab4a6c9cd5d',1,'gather.h']]], - ['gather_5fkernels_5',['gather_kernels',['../jit_2indexing_8h.html#a1a03318128191891a84707602b57b3cf',1,'indexing.h']]], - ['gather_5fmm_6',['gather_mm',['../group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7',1,'mlx::core']]], - ['gather_5fqmm_7',['gather_qmm',['../group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0',1,'mlx::core']]], - ['gathermm_8',['GatherMM',['../classmlx_1_1core_1_1_gather_m_m.html',1,'mlx::core::GatherMM'],['../classmlx_1_1core_1_1_gather_m_m.html#afd9bbc08138181b80e2fb86536ff3f2a',1,'mlx::core::GatherMM::GatherMM()']]], - ['gatherqmm_9',['GatherQMM',['../classmlx_1_1core_1_1_gather_q_m_m.html',1,'mlx::core::GatherQMM'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a60ed2ade7f10dd9c9314913a810f9360',1,'mlx::core::GatherQMM::GatherQMM()']]], - ['gemm_10',['gemm',['../namespacemlx_1_1core_1_1metal.html#ac46fd23516a61fc56d997910e4144281',1,'mlx::core::metal::gemm()'],['../steel__gemm__fused_8h.html#a70d5b54bfae6471242d4e85675c929fd',1,'gemm(const device T *A, const device T *B, const device T *C, device T *D, const constant GEMMParams *params, const constant GEMMAddMMParams *addmm_params, const constant int *batch_shape, const constant int64_t *batch_strides, const constant uint32_t *lhs_indices, const constant uint32_t *rhs_indices, const constant uint32_t *C_indices, const constant int *operand_shape, const constant int64_t *operand_strides, const constant packed_int3 &operand_batch_ndim, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_fused.h']]], - ['gemm_2eh_11',['gemm.h',['../gemm_8h.html',1,'']]], - ['gemm_5fk_5fiterations_12',['gemm_k_iterations',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a8b50863e4e2d3481c154be6c3629bf51',1,'mlx::steel::ImplicitGemmConv2DParams']]], - ['gemm_5fk_5fiterations_5faligned_13',['gemm_k_iterations_aligned',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0d7f419ba265805b418e93ce1ca2e0f9',1,'mlx::steel::GEMMParams::gemm_k_iterations_aligned'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#aa37e05a03ac8b34ec7dc31ca42f68998',1,'mlx::steel::GEMMSpiltKParams::gemm_k_iterations_aligned']]], - ['gemm_5floop_14',['gemm_loop',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a756d7bbcc96e2919cd65eec4bc135780',1,'mlx::steel::GEMMKernel::gemm_loop(threadgroup T *As, threadgroup T *Bs, const int gemm_k_iterations, thread loader_a_t &loader_a, thread loader_b_t &loader_b, thread mma_t &mma_op, thread const short &tgp_bm, thread const short &tgp_bn, thread const short &lbk, LoopAlignment< M_aligned, N_aligned, K_aligned_ > l={})'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a756d7bbcc96e2919cd65eec4bc135780',1,'mlx::steel::GEMMKernel::gemm_loop(threadgroup T *As, threadgroup T *Bs, const int gemm_k_iterations, thread loader_a_t &loader_a, thread loader_b_t &loader_b, thread mma_t &mma_op, thread const short &tgp_bm, thread const short &tgp_bn, thread const short &lbk, LoopAlignment< M_aligned, N_aligned, K_aligned_ > l={})']]], - ['gemm_5fparams_15',['gemm_params',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ab0724eb3ef52ee773b6607f6433b9f2c',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::gemm_params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#acc778b3c0b7ec38a43e8ea943df8704c',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::gemm_params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af59f9d356c4c3ec5627dc5a263d239d4',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::gemm_params']]], - ['gemm_5fsplitk_16',['gemm_splitk',['../steel__gemm__splitk_8h.html#a3be6e095a0a026d3ecf57a3e67f76188',1,'steel_gemm_splitk.h']]], - ['gemm_5fsplitk_5faccum_17',['gemm_splitk_accum',['../steel__gemm__splitk_8h.html#abeb921bf1dc7941125188ddd390b0907',1,'steel_gemm_splitk.h']]], - ['gemm_5fsplitk_5faccum_5faxpby_18',['gemm_splitk_accum_axpby',['../steel__gemm__splitk_8h.html#acc33fdfaaf3eb3a0629b3d52c7043dc1',1,'steel_gemm_splitk.h']]], - ['gemmaddmmparams_19',['GEMMAddMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html',1,'mlx::steel']]], - ['gemmkernel_20',['GEMMKernel',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html',1,'mlx::steel']]], - ['gemmparams_21',['GEMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_params.html',1,'mlx::steel']]], - ['gemmspiltkparams_22',['GEMMSpiltKParams',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html',1,'mlx::steel']]], - ['gemv_5fmasked_23',['gemv_masked',['../namespacemlx_1_1core_1_1metal.html#abc055b75e6a059618f279c35f8de36e7',1,'mlx::core::metal::gemv_masked()'],['../kernels_2gemv__masked_8h.html#af890b6ac155165f8ee0c600363938341',1,'gemv_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant int64_t *vector_batch_stride, const constant int64_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant int64_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid): gemv_masked.h']]], - ['gemv_5fmasked_2eh_24',['gemv_masked.h',['../jit_2gemv__masked_8h.html',1,'(Global Namespace)'],['../kernels_2gemv__masked_8h.html',1,'(Global Namespace)']]], - ['gemv_5fmasked_5fkernel_25',['gemv_masked_kernel',['../jit_2gemv__masked_8h.html#a933f06c211f86c37673dee329ed6901f',1,'gemv_masked.h']]], - ['gemv_5ft_5fmasked_26',['gemv_t_masked',['../kernels_2gemv__masked_8h.html#ae5b4a5124ddf92a984258a0be1ff0f4f',1,'gemv_masked.h']]], - ['gemvkernel_27',['GEMVKernel',['../struct_g_e_m_v_kernel.html',1,'']]], - ['gemvtkernel_28',['GEMVTKernel',['../struct_g_e_m_v_t_kernel.html',1,'']]], - ['general_29',['General',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a0db377921f4ce762c62526131097968f',1,'mlx::core']]], - ['general_5fc2r_30',['general_c2r',['../namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc',1,'pocketfft::detail']]], - ['general_5fnd_31',['general_nd',['../namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0',1,'pocketfft::detail']]], - ['general_5fr2c_32',['general_r2c',['../namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed',1,'pocketfft::detail']]], - ['generalcontiguousreduce_33',['GeneralContiguousReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a540cf31fe6858115a02e789938297cdb',1,'mlx::core']]], - ['generalgeneral_34',['GeneralGeneral',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a6fe62e8ce1fae1e70cb9eeaa67d29dab',1,'mlx::core']]], - ['generalreduce_35',['GeneralReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a3622f95ed0ec99657f9ad8ef39ec2184',1,'mlx::core']]], - ['generalstridedreduce_36',['GeneralStridedReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ad999b1a8ae1d7436efb5ffdfafb1dd3d',1,'mlx::core']]], - ['generic_37',['generic',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da3d517f8924ac7fd03699a29d97dc52d9',1,'mlx::core::Dtype::generic'],['../namespacemlx_1_1core.html#a34d69c4d46aa9b2a4a79dba7aba093d2',1,'mlx::core::generic']]], - ['get_5f2d_5fgrid_5fdims_38',['get_2d_grid_dims',['../namespacemlx_1_1core.html#aa9692de582995fd3ce19493b45ab7144',1,'mlx::core::get_2d_grid_dims(const Shape &shape, const Strides &strides)'],['../namespacemlx_1_1core.html#a6ec5cdf3253a9f20ca5ea7a1590fb386',1,'mlx::core::get_2d_grid_dims(const Shape &shape, const Strides &strides, size_t divisor)']]], - ['get_5factive_5fmemory_39',['get_active_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a7a3ad4e33d57a47474c98e2f88e775d7',1,'mlx::core::metal::MetalAllocator::get_active_memory()'],['../namespacemlx_1_1core_1_1metal.html#a7b75c2639016ac4d350fa6c9da386667',1,'mlx::core::metal::get_active_memory()']]], - ['get_5farange_5fkernel_40',['get_arange_kernel',['../namespacemlx_1_1core.html#a76f614e9956a6ca05a9be4db5a483446',1,'mlx::core']]], - ['get_5farchitecture_41',['get_architecture',['../classmlx_1_1core_1_1metal_1_1_device.html#a65f64dd8bafdc704d871fc5be5e7bc0b',1,'mlx::core::metal::Device']]], - ['get_5fbinary_5fkernel_42',['get_binary_kernel',['../namespacemlx_1_1core.html#a4decd4a07d91487e6903f6e3c8b7513a',1,'mlx::core']]], - ['get_5fbinary_5ftwo_5fkernel_43',['get_binary_two_kernel',['../namespacemlx_1_1core.html#a4e809746f48e5dcf7fa63215d3f5e33e',1,'mlx::core']]], - ['get_5fblock_5fdims_44',['get_block_dims',['../namespacemlx_1_1core.html#a0f0f59d3ffe2d16a684e5fc093302e15',1,'mlx::core']]], - ['get_5fcache_5fmemory_45',['get_cache_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ad3cabbe638917ca4114eb74dcabe381f',1,'mlx::core::metal::MetalAllocator::get_cache_memory()'],['../namespacemlx_1_1core_1_1metal.html#a43307654f62ed7c58e014be7fb03909c',1,'mlx::core::metal::get_cache_memory()']]], - ['get_5fcolocated_5fmtllib_5fpath_46',['get_colocated_mtllib_path',['../namespacemlx_1_1core_1_1metal.html#a5fd6ba2040e53a254b9d71ae7ebd315f',1,'mlx::core::metal']]], - ['get_5fcommand_5fbuffer_47',['get_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a5fe3970fbe92ccc55fce4241ffbe5210',1,'mlx::core::metal::Device']]], - ['get_5fcommand_5fbuffer_5fops_48',['get_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a064e1cb6a16de7a0619f6447622350f8',1,'mlx::core::metal::Device']]], - ['get_5fcommand_5fencoder_49',['get_command_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#affa682ef612def4890f5152f81ffb7e6',1,'mlx::core::metal::Device']]], - ['get_5fcoord_50',['get_coord',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a7331fff1d12f2f8b72b0006a3ad0dd83',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord(ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a7331fff1d12f2f8b72b0006a3ad0dd83',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord(ushort simd_lane_id)']]], - ['get_5fcopy_5fkernel_51',['get_copy_kernel',['../namespacemlx_1_1core.html#a05a220cff45f12439fde775983c6df78',1,'mlx::core']]], - ['get_5fdefault_5fstream_52',['get_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a2366c7b888e433608e203752edc92282',1,'mlx::core::scheduler::Scheduler']]], - ['get_5fdynamic_5fcopy_5fkernel_53',['get_dynamic_copy_kernel',['../namespacemlx_1_1core.html#addb29b3e6771875f2aecd035ed560461',1,'mlx::core']]], - ['get_5ffft_5fkernel_54',['get_fft_kernel',['../namespacemlx_1_1core.html#a1d4cffc3c78067b3d9a62d64f3fb686f',1,'mlx::core']]], - ['get_5fgemv_5fmasked_5fkernel_55',['get_gemv_masked_kernel',['../namespacemlx_1_1core.html#a90c24e0d0b99b68fad9deefcf4d3e818',1,'mlx::core']]], - ['get_5fglobal_5fformatter_56',['get_global_formatter',['../namespacemlx_1_1core.html#afc71e62dc5757564486cea5ebb12500e',1,'mlx::core']]], - ['get_5fkernel_57',['get_kernel',['../classmlx_1_1core_1_1metal_1_1_device.html#a6810c4dcbcfbf93fc51d42aa5ff0fc3a',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, MTL::Library *mtl_lib, const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#afa0cac9d800c21a8a7f6cb224256abaf',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, const std::string &lib_name="mlx", const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})']]], - ['get_5fkernel_5fpreamble_58',['get_kernel_preamble',['../compiled__preamble_8h.html#a1dfa17a0369fb90ff615c7461f5013f3',1,'compiled_preamble.h']]], - ['get_5flibrary_59',['get_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a75ed55e73baf48013028796518723ff0',1,'mlx::core::metal::Device']]], - ['get_5fmb_5fsort_5fkernel_60',['get_mb_sort_kernel',['../namespacemlx_1_1core.html#afb57825bb763050cc9a9d194aa41ac36',1,'mlx::core']]], - ['get_5fname_61',['get_name',['../structmlx_1_1core_1_1_node_namer.html#a1690dd38de288c0aee2bb53156eb770e',1,'mlx::core::NodeNamer']]], - ['get_5fpeak_5fmemory_62',['get_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ac7972a3fe58e69489de775a0f152da17',1,'mlx::core::metal::MetalAllocator::get_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#a4b67d680cefa95f0ed5801f0e14e48ce',1,'mlx::core::metal::get_peak_memory()']]], - ['get_5fplan_63',['get_plan',['../namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b',1,'pocketfft::detail']]], - ['get_5fpool_64',['get_pool',['../namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a',1,'pocketfft::detail::threading']]], - ['get_5fprimitive_5fstring_65',['get_primitive_string',['../namespacemlx_1_1core.html#ad4be35b310a252edd80d9cf04f094a60',1,'mlx::core']]], - ['get_5fquantized_5fkernel_66',['get_quantized_kernel',['../namespacemlx_1_1core.html#aa3faeae5378bfaafe3ce3432a051e43e',1,'mlx::core']]], - ['get_5freduce_5finit_5fkernel_67',['get_reduce_init_kernel',['../namespacemlx_1_1core.html#ae0470605dc819efeb6510183619f0299',1,'mlx::core']]], - ['get_5freduce_5fkernel_68',['get_reduce_kernel',['../namespacemlx_1_1core.html#a1be32ba7d67137dde7ac191dfe83ff49',1,'mlx::core']]], - ['get_5freduction_5fplan_69',['get_reduction_plan',['../namespacemlx_1_1core.html#ac97b5a6f009ca3d99854ce9512c20dba',1,'mlx::core']]], - ['get_5fscan_5fkernel_70',['get_scan_kernel',['../namespacemlx_1_1core.html#aeefaff208444d3fa61ecc0946fe1de5f',1,'mlx::core']]], - ['get_5fshape_71',['get_shape',['../namespacemlx_1_1core.html#a30fb38e05feeee19ae2b87e62bff3acf',1,'mlx::core']]], - ['get_5fsoftmax_5fkernel_72',['get_softmax_kernel',['../namespacemlx_1_1core.html#a35a412f688d79eb47e42d20a7c8650ee',1,'mlx::core']]], - ['get_5fsort_5fkernel_73',['get_sort_kernel',['../namespacemlx_1_1core.html#a84ebe6275218070f0ea320f126f64e22',1,'mlx::core']]], - ['get_5fsteel_5fconv_5fgeneral_5fkernel_74',['get_steel_conv_general_kernel',['../namespacemlx_1_1core.html#abce2b67044ee06a7bbe7a91ec7c8c48d',1,'mlx::core']]], - ['get_5fsteel_5fconv_5fkernel_75',['get_steel_conv_kernel',['../namespacemlx_1_1core.html#adce79d220672f5f3c65cc31d145ca9c4',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5ffused_5fkernel_76',['get_steel_gemm_fused_kernel',['../namespacemlx_1_1core.html#a84fa8e0aee321a9d614433a0b933103b',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fmasked_5fkernel_77',['get_steel_gemm_masked_kernel',['../namespacemlx_1_1core.html#ab5f60614e965144b451930fdf935e08d',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fsplitk_5faccum_5fkernel_78',['get_steel_gemm_splitk_accum_kernel',['../namespacemlx_1_1core.html#a195b86cad5bb99aa1bcd23952305af6b',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fsplitk_5fkernel_79',['get_steel_gemm_splitk_kernel',['../namespacemlx_1_1core.html#af48c6f2f72b61dbd6766e4f5fea85df5',1,'mlx::core']]], - ['get_5fstream_80',['get_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#aa6726858b362c7cd1f8a846a63085dbc',1,'mlx::core::scheduler::Scheduler::get_stream()'],['../namespacemlx_1_1core.html#ab436b8c08be2be32ef61bd72f7df63cd',1,'mlx::core::get_stream(int index)']]], - ['get_5ftemplate_5fdefinition_81',['get_template_definition',['../namespacemlx_1_1core.html#aae0d19f0acdef2accd2428fb84c8a032',1,'mlx::core']]], - ['get_5fternary_5fkernel_82',['get_ternary_kernel',['../namespacemlx_1_1core.html#a54eb3b65375022428aab5f810e40624b',1,'mlx::core']]], - ['get_5ftwiddle_83',['get_twiddle',['../radix_8h.html#ac5cf950316b9445296ee9ecfc56a56bd',1,'radix.h']]], - ['get_5ftype_5fstring_84',['get_type_string',['../namespacemlx_1_1core.html#af776fd91dd60594dcfebbafd17f19068',1,'mlx::core']]], - ['get_5funary_5fkernel_85',['get_unary_kernel',['../namespacemlx_1_1core.html#afbb085188b563a54606d84f87a9bf5a6',1,'mlx::core']]], - ['get_5fvar_86',['get_var',['../namespacemlx_1_1core_1_1env.html#a0efecbf9efe695adafad12b5a4945df3',1,'mlx::core::env']]], - ['gguf_2eh_87',['gguf.h',['../gguf_8h.html',1,'']]], - ['gguf_5fload_5fquantized_88',['gguf_load_quantized',['../namespacemlx_1_1core.html#a65dd68163bdaef3631e3724327782498',1,'mlx::core']]], - ['ggufload_89',['GGUFLoad',['../namespacemlx_1_1core.html#aa5b0f7f13a941e1f41c411194e9033c7',1,'mlx::core']]], - ['ggufmetadata_90',['GGUFMetaData',['../namespacemlx_1_1core.html#a8c2c1b9a37aadfb48f4c3a7e806e32e3',1,'mlx::core']]], - ['good_91',['good',['../classmlx_1_1core_1_1io_1_1_reader.html#a005d0b52c1f34866f7412b7f41dabec3',1,'mlx::core::io::Reader::good()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a0b050c2c27487007e250e2e19560ffe4',1,'mlx::core::io::Writer::good()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#ac54a2c693acc3d9e6e942412148ffcc9',1,'mlx::core::io::ParallelFileReader::good()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9ec4934b26fb358d699ddce1482b2d54',1,'mlx::core::io::FileWriter::good()']]], - ['good_5fsize_5fcmplx_92',['good_size_cmplx',['../structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624',1,'pocketfft::detail::util']]], - ['good_5fsize_5freal_93',['good_size_real',['../structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa',1,'pocketfft::detail::util']]], - ['gpu_94',['gpu',['../structmlx_1_1core_1_1_device.html#a45ed081b56ae5d4ddd39c83a5d8a1616',1,'mlx::core::Device::gpu'],['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdba0aa0be2a866411d9ff03515227454947',1,'mlx::core::Device::gpu']]], - ['gqa_5ffactor_95',['gqa_factor',['../structmlx_1_1steel_1_1_attn_params.html#a3b3e18cb993ab24819c852bc64288841',1,'mlx::steel::AttnParams']]], - ['grad_96',['grad',['../namespacemlx_1_1core.html#a3d2b2929ed4636e9e2b86e125b2e57d9',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#af482f6c64acd77c57ef5bb4b7be9726c',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a64bc619876b0f8cc81a2637ca81c99f7',1,'mlx::core::grad(const std::function< array(const array &)> &fun)']]], - ['graph_5futils_2eh_97',['graph_utils.h',['../graph__utils_8h.html',1,'']]], - ['greater_98',['Greater',['../struct_greater.html',1,'Greater'],['../structmlx_1_1core_1_1detail_1_1_greater.html',1,'mlx::core::detail::Greater'],['../classmlx_1_1core_1_1_greater.html',1,'mlx::core::Greater'],['../classmlx_1_1core_1_1_greater.html#a1d5992a66c020cd97a70e8e3d8cd1a1b',1,'mlx::core::Greater::Greater()']]], - ['greater_99',['greater',['../group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d',1,'mlx::core']]], - ['greater_5fequal_100',['greater_equal',['../group__ops.html#ga7153071bcfff6faad21332163fb9a430',1,'mlx::core']]], - ['greaterequal_101',['GreaterEqual',['../struct_greater_equal.html',1,'GreaterEqual'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html',1,'mlx::core::detail::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html',1,'mlx::core::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html#a19a3c49d5a9b40e17da0e56ef6908527',1,'mlx::core::GreaterEqual::GreaterEqual()']]], - ['grid_102',['grid',['../struct_read_writer.html#ac7a957f99873d3797081f5d620f3b2c8',1,'ReadWriter']]], - ['group_103',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html',1,'mlx::core::distributed::Group'],['../structmlx_1_1core_1_1distributed_1_1_group.html#a32e6e085a427b41ca3529c5e5db30a1b',1,'mlx::core::distributed::Group::Group()']]], - ['group_104',['group',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8831cb61ac633431b78b5fb99c0ea9ff',1,'mlx::core::distributed::DistPrimitive']]], - ['group_5fstep_5fcnt_105',['group_step_cnt',['../struct_quantized_block_loader.html#a234feacde36a4afc0d740332a3769fb6',1,'QuantizedBlockLoader']]], - ['group_5fsteps_106',['group_steps',['../struct_quantized_block_loader.html#a31e14175f3d4902d9fe5ab5a219f61ba',1,'QuantizedBlockLoader']]], - ['group_5fstride_107',['group_stride',['../struct_quantized_block_loader.html#a0ace7e3762ecfa5a4106e7dee7e1b6ab',1,'QuantizedBlockLoader']]], - ['groupimpl_108',['GroupImpl',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html',1,'mlx::core::distributed::detail::GroupImpl'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#ab9a91276b3c84ea63f1d1831ef4079dd',1,'mlx::core::distributed::mpi::GroupImpl']]], - ['groups_109',['groups',['../struct_m_l_x_conv_params.html#af7a5590ac0974c7841c7f8b9fda0cbed',1,'MLXConvParams']]], - ['gumbel_110',['gumbel',['../namespacemlx_1_1core_1_1random.html#a76f81f8f9468039a0b941513b46cb825',1,'mlx::core::random']]] + ['gather_5faxis_3',['gather_axis',['../namespacemlx_1_1core_1_1metal.html#adc66b1b48b51ac2d2f1f5bfac1b95ee3',1,'mlx::core::metal::gather_axis()'],['../gather__axis_8h.html#a0d7d3866afe2007e280781502dc767c6',1,'gather_axis(const device T *src, const device IdxT *indices, device T *out, const constant int *shape, const constant int64_t *src_strides, const constant int64_t *idx_strides, const constant size_t &ndim, const constant int &axis, const constant int &axis_size, const constant size_t &src_ax_stride, const constant size_t &idx_ax_stride, uint3 index, uint3 grid_dim): gather_axis.h']]], + ['gather_5faxis_2eh_4',['gather_axis.h',['../gather__axis_8h.html',1,'']]], + ['gather_5fbias_5',['gather_bias',['../steel__gemm__fused_8h.html#aaaf17233201156be684f858bfd0f1b67',1,'steel_gemm_fused.h']]], + ['gather_5fimpl_6',['gather_impl',['../gather_8h.html#a5cf4ff9849edd2d99ff24ab4a6c9cd5d',1,'gather.h']]], + ['gather_5fkernels_7',['gather_kernels',['../jit_2indexing_8h.html#a1a03318128191891a84707602b57b3cf',1,'indexing.h']]], + ['gather_5fmm_8',['gather_mm',['../group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7',1,'mlx::core']]], + ['gather_5fqmm_9',['gather_qmm',['../group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0',1,'mlx::core']]], + ['gatheraxis_10',['GatherAxis',['../classmlx_1_1core_1_1_gather_axis.html',1,'mlx::core::GatherAxis'],['../classmlx_1_1core_1_1_gather_axis.html#ad8fc6400954c52079f0a2f2b711df060',1,'mlx::core::GatherAxis::GatherAxis()']]], + ['gathermm_11',['GatherMM',['../classmlx_1_1core_1_1_gather_m_m.html',1,'mlx::core::GatherMM'],['../classmlx_1_1core_1_1_gather_m_m.html#afd9bbc08138181b80e2fb86536ff3f2a',1,'mlx::core::GatherMM::GatherMM()']]], + ['gatherqmm_12',['GatherQMM',['../classmlx_1_1core_1_1_gather_q_m_m.html',1,'mlx::core::GatherQMM'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a60ed2ade7f10dd9c9314913a810f9360',1,'mlx::core::GatherQMM::GatherQMM()']]], + ['gemm_13',['gemm',['../namespacemlx_1_1core_1_1metal.html#ac46fd23516a61fc56d997910e4144281',1,'mlx::core::metal::gemm()'],['../steel__gemm__fused_8h.html#a70d5b54bfae6471242d4e85675c929fd',1,'gemm(const device T *A, const device T *B, const device T *C, device T *D, const constant GEMMParams *params, const constant GEMMAddMMParams *addmm_params, const constant int *batch_shape, const constant int64_t *batch_strides, const constant uint32_t *lhs_indices, const constant uint32_t *rhs_indices, const constant uint32_t *C_indices, const constant int *operand_shape, const constant int64_t *operand_strides, const constant packed_int3 &operand_batch_ndim, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_fused.h']]], + ['gemm_2eh_14',['gemm.h',['../cpu_2gemm_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2steel_2gemm_2gemm_8h.html',1,'(Global Namespace)']]], + ['gemm_5fk_5fiterations_15',['gemm_k_iterations',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a8b50863e4e2d3481c154be6c3629bf51',1,'mlx::steel::ImplicitGemmConv2DParams']]], + ['gemm_5fk_5fiterations_5faligned_16',['gemm_k_iterations_aligned',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0d7f419ba265805b418e93ce1ca2e0f9',1,'mlx::steel::GEMMParams::gemm_k_iterations_aligned'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#aa37e05a03ac8b34ec7dc31ca42f68998',1,'mlx::steel::GEMMSpiltKParams::gemm_k_iterations_aligned']]], + ['gemm_5floop_17',['gemm_loop',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a756d7bbcc96e2919cd65eec4bc135780',1,'mlx::steel::GEMMKernel::gemm_loop(threadgroup T *As, threadgroup T *Bs, const int gemm_k_iterations, thread loader_a_t &loader_a, thread loader_b_t &loader_b, thread mma_t &mma_op, thread const short &tgp_bm, thread const short &tgp_bn, thread const short &lbk, LoopAlignment< M_aligned, N_aligned, K_aligned_ > l={})'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a756d7bbcc96e2919cd65eec4bc135780',1,'mlx::steel::GEMMKernel::gemm_loop(threadgroup T *As, threadgroup T *Bs, const int gemm_k_iterations, thread loader_a_t &loader_a, thread loader_b_t &loader_b, thread mma_t &mma_op, thread const short &tgp_bm, thread const short &tgp_bn, thread const short &lbk, LoopAlignment< M_aligned, N_aligned, K_aligned_ > l={})']]], + ['gemm_5fparams_18',['gemm_params',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ab0724eb3ef52ee773b6607f6433b9f2c',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::gemm_params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#acc778b3c0b7ec38a43e8ea943df8704c',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::gemm_params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af59f9d356c4c3ec5627dc5a263d239d4',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::gemm_params']]], + ['gemm_5fsplitk_19',['gemm_splitk',['../steel__gemm__splitk_8h.html#a3be6e095a0a026d3ecf57a3e67f76188',1,'steel_gemm_splitk.h']]], + ['gemm_5fsplitk_5faccum_20',['gemm_splitk_accum',['../steel__gemm__splitk_8h.html#abeb921bf1dc7941125188ddd390b0907',1,'steel_gemm_splitk.h']]], + ['gemm_5fsplitk_5faccum_5faxpby_21',['gemm_splitk_accum_axpby',['../steel__gemm__splitk_8h.html#acc33fdfaaf3eb3a0629b3d52c7043dc1',1,'steel_gemm_splitk.h']]], + ['gemmaddmmparams_22',['GEMMAddMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html',1,'mlx::steel']]], + ['gemmkernel_23',['GEMMKernel',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html',1,'mlx::steel']]], + ['gemmparams_24',['GEMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_params.html',1,'mlx::steel']]], + ['gemmspiltkparams_25',['GEMMSpiltKParams',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html',1,'mlx::steel']]], + ['gemv_5fmasked_26',['gemv_masked',['../namespacemlx_1_1core_1_1metal.html#abc055b75e6a059618f279c35f8de36e7',1,'mlx::core::metal::gemv_masked()'],['../gemv__masked_8h.html#af890b6ac155165f8ee0c600363938341',1,'gemv_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant int64_t *vector_batch_stride, const constant int64_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant int64_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid): gemv_masked.h']]], + ['gemv_5fmasked_2eh_27',['gemv_masked.h',['../gemv__masked_8h.html',1,'']]], + ['gemv_5ft_5fmasked_28',['gemv_t_masked',['../gemv__masked_8h.html#ae5b4a5124ddf92a984258a0be1ff0f4f',1,'gemv_masked.h']]], + ['gemvkernel_29',['GEMVKernel',['../struct_g_e_m_v_kernel.html',1,'']]], + ['gemvtkernel_30',['GEMVTKernel',['../struct_g_e_m_v_t_kernel.html',1,'']]], + ['general_31',['General',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6a0db377921f4ce762c62526131097968f',1,'mlx::core::General'],['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a0db377921f4ce762c62526131097968f',1,'mlx::core::General'],['../namespacemlx_1_1core.html#ac2b8997537c7f25dd2b244d4c0a865a1a0db377921f4ce762c62526131097968f',1,'mlx::core::General']]], + ['general_5fc2r_32',['general_c2r',['../namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc',1,'pocketfft::detail']]], + ['general_5fnd_33',['general_nd',['../namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0',1,'pocketfft::detail']]], + ['general_5fr2c_34',['general_r2c',['../namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed',1,'pocketfft::detail']]], + ['generalcontiguousreduce_35',['GeneralContiguousReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a540cf31fe6858115a02e789938297cdb',1,'mlx::core']]], + ['generalgeneral_36',['GeneralGeneral',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a6fe62e8ce1fae1e70cb9eeaa67d29dab',1,'mlx::core']]], + ['generalreduce_37',['GeneralReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a3622f95ed0ec99657f9ad8ef39ec2184',1,'mlx::core']]], + ['generalstridedreduce_38',['GeneralStridedReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ad999b1a8ae1d7436efb5ffdfafb1dd3d',1,'mlx::core']]], + ['generic_39',['generic',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da3d517f8924ac7fd03699a29d97dc52d9',1,'mlx::core::Dtype::generic'],['../namespacemlx_1_1core.html#a34d69c4d46aa9b2a4a79dba7aba093d2',1,'mlx::core::generic']]], + ['get_5f2d_5fgrid_5fdims_40',['get_2d_grid_dims',['../namespacemlx_1_1core.html#aa9692de582995fd3ce19493b45ab7144',1,'mlx::core::get_2d_grid_dims(const Shape &shape, const Strides &strides)'],['../namespacemlx_1_1core.html#a6ec5cdf3253a9f20ca5ea7a1590fb386',1,'mlx::core::get_2d_grid_dims(const Shape &shape, const Strides &strides, size_t divisor)']]], + ['get_5factive_5fmemory_41',['get_active_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a7a3ad4e33d57a47474c98e2f88e775d7',1,'mlx::core::metal::MetalAllocator::get_active_memory()'],['../namespacemlx_1_1core_1_1metal.html#a7b75c2639016ac4d350fa6c9da386667',1,'mlx::core::metal::get_active_memory()']]], + ['get_5farange_5fkernel_42',['get_arange_kernel',['../namespacemlx_1_1core.html#a76f614e9956a6ca05a9be4db5a483446',1,'mlx::core']]], + ['get_5farchitecture_43',['get_architecture',['../classmlx_1_1core_1_1metal_1_1_device.html#a65f64dd8bafdc704d871fc5be5e7bc0b',1,'mlx::core::metal::Device']]], + ['get_5fbinary_5fkernel_44',['get_binary_kernel',['../namespacemlx_1_1core.html#a4decd4a07d91487e6903f6e3c8b7513a',1,'mlx::core']]], + ['get_5fbinary_5fop_5ftype_45',['get_binary_op_type',['../namespacemlx_1_1core.html#a24eef9908f164adeece3be7c6924919a',1,'mlx::core']]], + ['get_5fbinary_5ftwo_5fkernel_46',['get_binary_two_kernel',['../namespacemlx_1_1core.html#a4e809746f48e5dcf7fa63215d3f5e33e',1,'mlx::core']]], + ['get_5fblock_5fdims_47',['get_block_dims',['../namespacemlx_1_1core.html#a0f0f59d3ffe2d16a684e5fc093302e15',1,'mlx::core']]], + ['get_5fcache_5fmemory_48',['get_cache_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ad3cabbe638917ca4114eb74dcabe381f',1,'mlx::core::metal::MetalAllocator::get_cache_memory()'],['../namespacemlx_1_1core_1_1metal.html#a43307654f62ed7c58e014be7fb03909c',1,'mlx::core::metal::get_cache_memory()']]], + ['get_5fcolocated_5fmtllib_5fpath_49',['get_colocated_mtllib_path',['../namespacemlx_1_1core_1_1metal.html#a5fd6ba2040e53a254b9d71ae7ebd315f',1,'mlx::core::metal']]], + ['get_5fcommand_5fbuffer_50',['get_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a5fe3970fbe92ccc55fce4241ffbe5210',1,'mlx::core::metal::Device']]], + ['get_5fcommand_5fbuffer_5fops_51',['get_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a064e1cb6a16de7a0619f6447622350f8',1,'mlx::core::metal::Device']]], + ['get_5fcommand_5fencoder_52',['get_command_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#affa682ef612def4890f5152f81ffb7e6',1,'mlx::core::metal::Device']]], + ['get_5fcoord_53',['get_coord',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a7331fff1d12f2f8b72b0006a3ad0dd83',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord(ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a7331fff1d12f2f8b72b0006a3ad0dd83',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord(ushort simd_lane_id)']]], + ['get_5fcopy_5fkernel_54',['get_copy_kernel',['../namespacemlx_1_1core.html#a05a220cff45f12439fde775983c6df78',1,'mlx::core']]], + ['get_5fdefault_5fstream_55',['get_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a2366c7b888e433608e203752edc92282',1,'mlx::core::scheduler::Scheduler']]], + ['get_5fdynamic_5fcopy_5fkernel_56',['get_dynamic_copy_kernel',['../namespacemlx_1_1core.html#addb29b3e6771875f2aecd035ed560461',1,'mlx::core']]], + ['get_5ffft_5fkernel_57',['get_fft_kernel',['../namespacemlx_1_1core.html#a1d4cffc3c78067b3d9a62d64f3fb686f',1,'mlx::core']]], + ['get_5fgemv_5fmasked_5fkernel_58',['get_gemv_masked_kernel',['../namespacemlx_1_1core.html#a90c24e0d0b99b68fad9deefcf4d3e818',1,'mlx::core']]], + ['get_5fglobal_5fformatter_59',['get_global_formatter',['../namespacemlx_1_1core.html#afc71e62dc5757564486cea5ebb12500e',1,'mlx::core']]], + ['get_5fkernel_60',['get_kernel',['../classmlx_1_1core_1_1metal_1_1_device.html#a6810c4dcbcfbf93fc51d42aa5ff0fc3a',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, MTL::Library *mtl_lib, const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#afa0cac9d800c21a8a7f6cb224256abaf',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, const std::string &lib_name="mlx", const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})']]], + ['get_5fkernel_5fpreamble_61',['get_kernel_preamble',['../compiled__preamble_8h.html#a1dfa17a0369fb90ff615c7461f5013f3',1,'compiled_preamble.h']]], + ['get_5flibrary_62',['get_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a75ed55e73baf48013028796518723ff0',1,'mlx::core::metal::Device']]], + ['get_5fmb_5fsort_5fkernel_63',['get_mb_sort_kernel',['../namespacemlx_1_1core.html#afb57825bb763050cc9a9d194aa41ac36',1,'mlx::core']]], + ['get_5fname_64',['get_name',['../structmlx_1_1core_1_1_node_namer.html#a1690dd38de288c0aee2bb53156eb770e',1,'mlx::core::NodeNamer']]], + ['get_5fpeak_5fmemory_65',['get_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ac7972a3fe58e69489de775a0f152da17',1,'mlx::core::metal::MetalAllocator::get_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#a4b67d680cefa95f0ed5801f0e14e48ce',1,'mlx::core::metal::get_peak_memory()']]], + ['get_5fplan_66',['get_plan',['../namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b',1,'pocketfft::detail']]], + ['get_5fpool_67',['get_pool',['../namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a',1,'pocketfft::detail::threading']]], + ['get_5fprimitive_5fstring_68',['get_primitive_string',['../namespacemlx_1_1core.html#ad4be35b310a252edd80d9cf04f094a60',1,'mlx::core']]], + ['get_5fquantized_5fkernel_69',['get_quantized_kernel',['../namespacemlx_1_1core.html#aa3faeae5378bfaafe3ce3432a051e43e',1,'mlx::core']]], + ['get_5freduce_5finit_5fkernel_70',['get_reduce_init_kernel',['../namespacemlx_1_1core.html#ae0470605dc819efeb6510183619f0299',1,'mlx::core']]], + ['get_5freduce_5fkernel_71',['get_reduce_kernel',['../namespacemlx_1_1core.html#a1be32ba7d67137dde7ac191dfe83ff49',1,'mlx::core']]], + ['get_5freduction_5fplan_72',['get_reduction_plan',['../namespacemlx_1_1core.html#ac97b5a6f009ca3d99854ce9512c20dba',1,'mlx::core']]], + ['get_5fscan_5fkernel_73',['get_scan_kernel',['../namespacemlx_1_1core.html#aeefaff208444d3fa61ecc0946fe1de5f',1,'mlx::core']]], + ['get_5fshape_74',['get_shape',['../namespacemlx_1_1core.html#a30fb38e05feeee19ae2b87e62bff3acf',1,'mlx::core']]], + ['get_5fsoftmax_5fkernel_75',['get_softmax_kernel',['../namespacemlx_1_1core.html#a35a412f688d79eb47e42d20a7c8650ee',1,'mlx::core']]], + ['get_5fsort_5fkernel_76',['get_sort_kernel',['../namespacemlx_1_1core.html#a84ebe6275218070f0ea320f126f64e22',1,'mlx::core']]], + ['get_5fsteel_5fconv_5fgeneral_5fkernel_77',['get_steel_conv_general_kernel',['../namespacemlx_1_1core.html#abce2b67044ee06a7bbe7a91ec7c8c48d',1,'mlx::core']]], + ['get_5fsteel_5fconv_5fkernel_78',['get_steel_conv_kernel',['../namespacemlx_1_1core.html#adce79d220672f5f3c65cc31d145ca9c4',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5ffused_5fkernel_79',['get_steel_gemm_fused_kernel',['../namespacemlx_1_1core.html#a84fa8e0aee321a9d614433a0b933103b',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fmasked_5fkernel_80',['get_steel_gemm_masked_kernel',['../namespacemlx_1_1core.html#ab5f60614e965144b451930fdf935e08d',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fsplitk_5faccum_5fkernel_81',['get_steel_gemm_splitk_accum_kernel',['../namespacemlx_1_1core.html#a195b86cad5bb99aa1bcd23952305af6b',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fsplitk_5fkernel_82',['get_steel_gemm_splitk_kernel',['../namespacemlx_1_1core.html#af48c6f2f72b61dbd6766e4f5fea85df5',1,'mlx::core']]], + ['get_5fstream_83',['get_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#aa6726858b362c7cd1f8a846a63085dbc',1,'mlx::core::scheduler::Scheduler::get_stream()'],['../namespacemlx_1_1core.html#ab436b8c08be2be32ef61bd72f7df63cd',1,'mlx::core::get_stream(int index)']]], + ['get_5ftemplate_5fdefinition_84',['get_template_definition',['../namespacemlx_1_1core.html#aae0d19f0acdef2accd2428fb84c8a032',1,'mlx::core']]], + ['get_5fternary_5fkernel_85',['get_ternary_kernel',['../namespacemlx_1_1core.html#a54eb3b65375022428aab5f810e40624b',1,'mlx::core']]], + ['get_5fternary_5fop_5ftype_86',['get_ternary_op_type',['../namespacemlx_1_1core.html#a548b6f4a39e639c18896e50b1702c830',1,'mlx::core']]], + ['get_5ftwiddle_87',['get_twiddle',['../radix_8h.html#ac5cf950316b9445296ee9ecfc56a56bd',1,'radix.h']]], + ['get_5ftype_5fstring_88',['get_type_string',['../namespacemlx_1_1core.html#af776fd91dd60594dcfebbafd17f19068',1,'mlx::core']]], + ['get_5funary_5fkernel_89',['get_unary_kernel',['../namespacemlx_1_1core.html#afbb085188b563a54606d84f87a9bf5a6',1,'mlx::core']]], + ['get_5fvar_90',['get_var',['../namespacemlx_1_1core_1_1env.html#a0efecbf9efe695adafad12b5a4945df3',1,'mlx::core::env']]], + ['gguf_2eh_91',['gguf.h',['../gguf_8h.html',1,'']]], + ['gguf_5fload_5fquantized_92',['gguf_load_quantized',['../namespacemlx_1_1core.html#a65dd68163bdaef3631e3724327782498',1,'mlx::core']]], + ['ggufload_93',['GGUFLoad',['../namespacemlx_1_1core.html#aa5b0f7f13a941e1f41c411194e9033c7',1,'mlx::core']]], + ['ggufmetadata_94',['GGUFMetaData',['../namespacemlx_1_1core.html#a8c2c1b9a37aadfb48f4c3a7e806e32e3',1,'mlx::core']]], + ['good_95',['good',['../classmlx_1_1core_1_1io_1_1_reader.html#a005d0b52c1f34866f7412b7f41dabec3',1,'mlx::core::io::Reader::good()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a0b050c2c27487007e250e2e19560ffe4',1,'mlx::core::io::Writer::good()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#ac54a2c693acc3d9e6e942412148ffcc9',1,'mlx::core::io::ParallelFileReader::good()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9ec4934b26fb358d699ddce1482b2d54',1,'mlx::core::io::FileWriter::good()']]], + ['good_5fsize_5fcmplx_96',['good_size_cmplx',['../structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624',1,'pocketfft::detail::util']]], + ['good_5fsize_5freal_97',['good_size_real',['../structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa',1,'pocketfft::detail::util']]], + ['gpu_98',['gpu',['../structmlx_1_1core_1_1_device.html#a45ed081b56ae5d4ddd39c83a5d8a1616',1,'mlx::core::Device::gpu'],['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdba0aa0be2a866411d9ff03515227454947',1,'mlx::core::Device::gpu']]], + ['gqa_5ffactor_99',['gqa_factor',['../structmlx_1_1steel_1_1_attn_params.html#a3b3e18cb993ab24819c852bc64288841',1,'mlx::steel::AttnParams']]], + ['grad_100',['grad',['../namespacemlx_1_1core.html#a3d2b2929ed4636e9e2b86e125b2e57d9',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#af482f6c64acd77c57ef5bb4b7be9726c',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a64bc619876b0f8cc81a2637ca81c99f7',1,'mlx::core::grad(const std::function< array(const array &)> &fun)']]], + ['graph_5futils_2eh_101',['graph_utils.h',['../graph__utils_8h.html',1,'']]], + ['greater_102',['Greater',['../struct_greater.html',1,'Greater'],['../structmlx_1_1core_1_1detail_1_1_greater.html',1,'mlx::core::detail::Greater'],['../classmlx_1_1core_1_1_greater.html',1,'mlx::core::Greater'],['../classmlx_1_1core_1_1_greater.html#a1d5992a66c020cd97a70e8e3d8cd1a1b',1,'mlx::core::Greater::Greater()']]], + ['greater_103',['greater',['../group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d',1,'mlx::core']]], + ['greater_5fequal_104',['greater_equal',['../group__ops.html#ga7153071bcfff6faad21332163fb9a430',1,'mlx::core']]], + ['greaterequal_105',['GreaterEqual',['../struct_greater_equal.html',1,'GreaterEqual'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html',1,'mlx::core::detail::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html',1,'mlx::core::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html#a19a3c49d5a9b40e17da0e56ef6908527',1,'mlx::core::GreaterEqual::GreaterEqual()']]], + ['grid_106',['grid',['../struct_read_writer.html#ac7a957f99873d3797081f5d620f3b2c8',1,'ReadWriter']]], + ['group_107',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html',1,'mlx::core::distributed::Group'],['../structmlx_1_1core_1_1distributed_1_1_group.html#a32e6e085a427b41ca3529c5e5db30a1b',1,'mlx::core::distributed::Group::Group()']]], + ['group_108',['group',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8831cb61ac633431b78b5fb99c0ea9ff',1,'mlx::core::distributed::DistPrimitive']]], + ['group_5fstep_5fcnt_109',['group_step_cnt',['../struct_quantized_block_loader.html#a234feacde36a4afc0d740332a3769fb6',1,'QuantizedBlockLoader']]], + ['group_5fsteps_110',['group_steps',['../struct_quantized_block_loader.html#a31e14175f3d4902d9fe5ab5a219f61ba',1,'QuantizedBlockLoader']]], + ['group_5fstride_111',['group_stride',['../struct_quantized_block_loader.html#a0ace7e3762ecfa5a4106e7dee7e1b6ab',1,'QuantizedBlockLoader']]], + ['groupimpl_112',['GroupImpl',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html',1,'mlx::core::distributed::detail::GroupImpl'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#ab9a91276b3c84ea63f1d1831ef4079dd',1,'mlx::core::distributed::mpi::GroupImpl'],['../namespacemlx_1_1core_1_1distributed_1_1ring.html#a6e4d590e07f0cf3cc2d15f258f9438ed',1,'mlx::core::distributed::ring::GroupImpl']]], + ['groups_113',['groups',['../struct_m_l_x_conv_params.html#af7a5590ac0974c7841c7f8b9fda0cbed',1,'MLXConvParams']]], + ['gumbel_114',['gumbel',['../namespacemlx_1_1core_1_1random.html#a76f81f8f9468039a0b941513b46cb825',1,'mlx::core::random']]] ]; diff --git a/docs/build/html/search/all_8.js b/docs/build/html/search/all_8.js index 5aee85a16..8bb335aae 100644 --- a/docs/build/html/search/all_8.js +++ b/docs/build/html/search/all_8.js @@ -18,13 +18,12 @@ var searchData= ['half_5fcompop_15',['half_compop',['../fp16_8h.html#acec0b85a9974cbde7b270a121f382405',1,'fp16.h']]], ['half_5finplace_5fbitop_16',['half_inplace_bitop',['../fp16_8h.html#a378e011e994bf62a961c3c1cd6f7c290',1,'fp16.h']]], ['half_5finplace_5fop_17',['half_inplace_op',['../fp16_8h.html#a6348c00d31a50b2df1b47d18af49c4b8',1,'fp16.h']]], - ['half_5for_5fbits_18',['half_or_bits',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core::numeric_limits< float16_t >']]], - ['half_5ftypes_2eh_19',['half_types.h',['../half__types_8h.html',1,'']]], - ['has_5fbatch_20',['has_batch',['../steel__gemm__fused_8h.html#adffcdc900c19ff97f1523e43f1a5a6cc',1,'steel_gemm_fused.h']]], - ['has_5fmask_21',['has_mask',['../sdpa__vector_8h.html#a6ed0dd113fe7d471fc0b869b8c028c81',1,'sdpa_vector.h']]], - ['has_5fmul_5foperand_5fmask_22',['has_mul_operand_mask',['../struct_g_e_m_v_kernel.html#ad47223ee49b3cb7bf3746a2cec45f883',1,'GEMVKernel::has_mul_operand_mask'],['../struct_g_e_m_v_t_kernel.html#a8db6f01f96a36b216acd801c34a96ef5',1,'GEMVTKernel::has_mul_operand_mask']]], - ['has_5fmul_5foutput_5fmask_23',['has_mul_output_mask',['../struct_g_e_m_v_kernel.html#a0edbf2dd6a6563e7afa6dab6b670615c',1,'GEMVKernel::has_mul_output_mask'],['../struct_g_e_m_v_t_kernel.html#a8eb06f6569e4042e24fee220b11fa10d',1,'GEMVTKernel::has_mul_output_mask']]], - ['has_5foperand_5fmask_24',['has_operand_mask',['../struct_g_e_m_v_kernel.html#ab00784dff1512a7b0919fcb4cfa5d50e',1,'GEMVKernel::has_operand_mask'],['../struct_g_e_m_v_t_kernel.html#a6729d6e63e76a1e9c7c8e78d9aac4869',1,'GEMVTKernel::has_operand_mask']]], - ['has_5foutput_5fmask_25',['has_output_mask',['../struct_g_e_m_v_kernel.html#ab8b64c94f4c8f6f09c0777415589b487',1,'GEMVKernel::has_output_mask'],['../struct_g_e_m_v_t_kernel.html#aaefdf8f023da255bbb70a0c3e3408626',1,'GEMVTKernel::has_output_mask']]], - ['has_5fprimitive_26',['has_primitive',['../classmlx_1_1core_1_1array.html#aa5aceab15241e7826cbaf8b8a41440c1',1,'mlx::core::array']]] + ['half_5ftypes_2eh_18',['half_types.h',['../half__types_8h.html',1,'']]], + ['has_5fbatch_19',['has_batch',['../steel__gemm__fused_8h.html#adffcdc900c19ff97f1523e43f1a5a6cc',1,'steel_gemm_fused.h']]], + ['has_5fmask_20',['has_mask',['../sdpa__vector_8h.html#a6ed0dd113fe7d471fc0b869b8c028c81',1,'sdpa_vector.h']]], + ['has_5fmul_5foperand_5fmask_21',['has_mul_operand_mask',['../struct_g_e_m_v_kernel.html#ad47223ee49b3cb7bf3746a2cec45f883',1,'GEMVKernel::has_mul_operand_mask'],['../struct_g_e_m_v_t_kernel.html#a8db6f01f96a36b216acd801c34a96ef5',1,'GEMVTKernel::has_mul_operand_mask']]], + ['has_5fmul_5foutput_5fmask_22',['has_mul_output_mask',['../struct_g_e_m_v_kernel.html#a0edbf2dd6a6563e7afa6dab6b670615c',1,'GEMVKernel::has_mul_output_mask'],['../struct_g_e_m_v_t_kernel.html#a8eb06f6569e4042e24fee220b11fa10d',1,'GEMVTKernel::has_mul_output_mask']]], + ['has_5foperand_5fmask_23',['has_operand_mask',['../struct_g_e_m_v_kernel.html#ab00784dff1512a7b0919fcb4cfa5d50e',1,'GEMVKernel::has_operand_mask'],['../struct_g_e_m_v_t_kernel.html#a6729d6e63e76a1e9c7c8e78d9aac4869',1,'GEMVTKernel::has_operand_mask']]], + ['has_5foutput_5fmask_24',['has_output_mask',['../struct_g_e_m_v_kernel.html#ab8b64c94f4c8f6f09c0777415589b487',1,'GEMVKernel::has_output_mask'],['../struct_g_e_m_v_t_kernel.html#aaefdf8f023da255bbb70a0c3e3408626',1,'GEMVTKernel::has_output_mask']]], + ['has_5fprimitive_25',['has_primitive',['../classmlx_1_1core_1_1array.html#aa5aceab15241e7826cbaf8b8a41440c1',1,'mlx::core::array']]] ]; diff --git a/docs/build/html/search/all_9.js b/docs/build/html/search/all_9.js index 66a43f03e..4362e440d 100644 --- a/docs/build/html/search/all_9.js +++ b/docs/build/html/search/all_9.js @@ -1,16 +1,16 @@ var searchData= [ - ['i_0',['i',['../structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d',1,'pocketfft::detail::cmplx::i'],['../unionmlx_1_1core_1_1detail_1_1_int_or_float.html#aa6d4b02e99df90a321e0eeeb98e71b49',1,'mlx::core::detail::IntOrFloat::i'],['../unionbool4__or__uint.html#ab24d95aaf4203ddf3e6b1ed19397ced7',1,'bool4_or_uint::i'],['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a865c0c0b4ab0e063e5caa3387c1a8741',1,'mlx::core::Dtype::i']]], + ['i_0',['i',['../structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d',1,'pocketfft::detail::cmplx::i'],['../unionbool4__or__uint.html#ab24d95aaf4203ddf3e6b1ed19397ced7',1,'bool4_or_uint::i'],['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a865c0c0b4ab0e063e5caa3387c1a8741',1,'mlx::core::Dtype::i']]], ['id_1',['id',['../classmlx_1_1core_1_1array.html#adfa53f3f26bb0f942fb1c67ec8cd5380',1,'mlx::core::array']]], ['identity_2',['identity',['../group__ops.html#ga484eaa10d5e19a4ca46d3a9cd9fab600',1,'mlx::core::identity(int n, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gad994d65ac6019c26b5ad6c41179d3424',1,'mlx::core::identity(int n, StreamOrDevice s={})']]], ['idil_3',['idil',['../struct_m_l_x_conv_params.html#af900fdb4c4d4ea35eed02940dee8d4d1',1,'MLXConvParams']]], - ['idx_5ft_4',['idx_t',['../struct_kernel_merge_sort.html#a0df65b709ae7f153a2bf381179d55e00',1,'KernelMergeSort']]], + ['idxt_4',['IdxT',['../struct_kernel_merge_sort.html#a587b32e6748cffd2ace5a70b8c59dee4',1,'KernelMergeSort']]], ['ifft_5',['ifft',['../namespacemlx_1_1core_1_1fft.html#a893a667b85d6bef9b27fb40b591352b3',1,'mlx::core::fft::ifft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a700e1659e101bc0f806de712079d9273',1,'mlx::core::fft::ifft(const array &a, int axis=-1, StreamOrDevice s={})']]], ['ifft2_6',['ifft2',['../namespacemlx_1_1core_1_1fft.html#a9c83e5b0498c63cac9c882494b6b3eb6',1,'mlx::core::fft::ifft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a865adcb7d7fe35541ad8c21f963905e0',1,'mlx::core::fft::ifft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], ['ifftn_7',['ifftn',['../namespacemlx_1_1core_1_1fft.html#ae29b70549a1106cf7e70a31a0f80e66a',1,'mlx::core::fft::ifftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a3794c67262e4ab28d35fa89abfdfd063',1,'mlx::core::fft::ifftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a8adeca9b76277676390ec7d04dc0620b',1,'mlx::core::fft::ifftn(const array &a, StreamOrDevice s={})']]], ['imag_8',['Imag',['../struct_imag.html',1,'Imag'],['../structmlx_1_1core_1_1detail_1_1_imag.html',1,'mlx::core::detail::Imag'],['../classmlx_1_1core_1_1_imag.html',1,'mlx::core::Imag'],['../classmlx_1_1core_1_1_imag.html#a284b7de34a316110fdc98e7b753372b2',1,'mlx::core::Imag::Imag()']]], - ['imag_9',['imag',['../structcomplex64__t.html#a94037c0cf8451aaff7cb4d154a8426de',1,'complex64_t::imag'],['../group__ops.html#ga7ff592a64d528f0cf4f3d098465da029',1,'mlx::core::imag()']]], - ['implicit_5fgemm_5fconv_5f2d_10',['implicit_gemm_conv_2d',['../kernels_2steel_2conv_2kernels_2steel__conv_8h.html#a5728711d1c2ee4038457babb7ac12888',1,'steel_conv.h']]], + ['imag_9',['imag',['../structcomplex64__t.html#a94037c0cf8451aaff7cb4d154a8426de',1,'complex64_t::imag'],['../namespacemlx_1_1core_1_1simd.html#a97c69b04852ccba242f1348fda17ca20',1,'mlx::core::simd::imag()'],['../group__ops.html#ga7ff592a64d528f0cf4f3d098465da029',1,'mlx::core::imag()']]], + ['implicit_5fgemm_5fconv_5f2d_10',['implicit_gemm_conv_2d',['../steel__conv_8h.html#a5728711d1c2ee4038457babb7ac12888',1,'steel_conv.h']]], ['implicit_5fgemm_5fconv_5f2d_5fgeneral_11',['implicit_gemm_conv_2d_general',['../steel__conv__general_8h.html#ad620c0656ae92e36dcb4a285d9e790ee',1,'steel_conv_general.h']]], ['implicitgemmconv2dparams_12',['ImplicitGemmConv2DParams',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html',1,'mlx::steel']]], ['import_5ffunction_13',['import_function',['../structmlx_1_1core_1_1_imported_function.html#a65f83405a292831a6fb2073a7db6fb90',1,'mlx::core::ImportedFunction::import_function()'],['../namespacemlx_1_1core.html#a4fbb29691ee1ff22c3ee2a67cbc053d5',1,'mlx::core::import_function(const std::string &file)']]], @@ -21,40 +21,40 @@ var searchData= ['in_5ftracing_18',['in_tracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#ac52b8e2c3f808d3076c4e1ebaf9dc63d',1,'mlx::core::detail::InTracing::in_tracing()'],['../namespacemlx_1_1core_1_1detail.html#af974e1a6f06acfc949e67a330898ac11',1,'mlx::core::detail::in_tracing()']]], ['includes_2eh_19',['includes.h',['../includes_8h.html',1,'']]], ['increment_5fcommand_5fbuffer_5fops_20',['increment_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a7a33d4d601423a3d3c23d5ad7072abb6',1,'mlx::core::metal::Device']]], - ['index_21',['index',['../struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333',1,'LoopedElemToLoc::index'],['../structmlx_1_1core_1_1_device.html#a5e345748fe318a267833ab7398b364ac',1,'mlx::core::Device::index'],['../structmlx_1_1core_1_1_stream.html#a9d0dafc1899333e1176eb2bbc0a8b626',1,'mlx::core::Stream::index']]], + ['index_21',['index',['../struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333',1,'LoopedElemToLoc::index'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a2cd3b616739b3d5b41e5b46ae335957d',1,'LoopedElemToLoc< 1, OffsetT, true >::index'],['../structmlx_1_1core_1_1_device.html#a5e345748fe318a267833ab7398b364ac',1,'mlx::core::Device::index'],['../structmlx_1_1core_1_1_stream.html#a9d0dafc1899333e1176eb2bbc0a8b626',1,'mlx::core::Stream::index'],['../struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333',1,'LoopedElemToLoc< 1, OffsetT, false >::index'],['../struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333',1,'LoopedElemToLoc< 1, OffsetT, true >::index']]], ['indexing_2eh_22',['indexing.h',['../jit_2indexing_8h.html',1,'(Global Namespace)'],['../kernels_2indexing_8h.html',1,'(Global Namespace)']]], ['indices_23',['Indices',['../struct_indices.html',1,'']]], ['inexact_24',['inexact',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2dae03b116564cd944b048fde87dbd4d5c9',1,'mlx::core::Dtype::inexact'],['../namespacemlx_1_1core.html#a54c6fae21b7f2fea8e6f80011ef38534',1,'mlx::core::inexact']]], - ['infinity_25',['infinity',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61eb741e7af49046beb863abf023b206',1,'metal::_numeric_limits_impl< bfloat16_t >::infinity()'],['../structmlx_1_1core_1_1numeric__limits.html#a2a8f3f489b47b7e8398bec9895ae0c27',1,'mlx::core::numeric_limits< float16_t >::infinity()'],['../structmlx_1_1core_1_1numeric__limits.html#a974982399d0211786599526abdb843b8',1,'mlx::core::numeric_limits< bfloat16_t >::infinity()']]], - ['init_26',['init',['../struct_cum_max.html#a16480052a2eeb4340e546838aab59cc4',1,'CumMax::init'],['../struct_cum_min.html#a8b67f739c620d0cc194b533190990ab9',1,'CumMin::init'],['../struct_less_than.html#abf97a6b0163048e4ba96460939dbd3a3',1,'LessThan::init'],['../namespacemlx_1_1core_1_1distributed.html#a33633c058c7ec82cca4f237243c6810d',1,'mlx::core::distributed::init()'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#ab40a34a8837956e24fb9b9661104c8f9',1,'mlx::core::distributed::mpi::init()']]], - ['init_5freduce_27',['init_reduce',['../reduce__init_8h.html#a0088604ac2eaa6940689ff12c4ba5fc2',1,'reduce_init.h']]], - ['inner_28',['inner',['../group__ops.html#ga654fec16a9746b390916697a2ab2546e',1,'mlx::core']]], - ['inner_5flooper_29',['inner_looper',['../struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40',1,'LoopedElemToLoc']]], - ['inp_5fjump_5fc_30',['inp_jump_c',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a78d30e843d65d1829623afb0b607f0a5',1,'mlx::steel::ImplicitGemmConv2DParams']]], - ['inp_5fjump_5fh_31',['inp_jump_h',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a03685a4066cdb11ffb647408e2c5b122',1,'mlx::steel::ImplicitGemmConv2DParams']]], - ['inp_5fjump_5fw_32',['inp_jump_w',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#acf168c72f4a86b72b8f5f386f07c9d8c',1,'mlx::steel::ImplicitGemmConv2DParams']]], - ['inputs_33',['inputs',['../classmlx_1_1core_1_1array.html#a2913abcdf71826827c8457f529825fff',1,'mlx::core::array::inputs() const'],['../classmlx_1_1core_1_1array.html#acffb082177f9b78f0c52e406adff972f',1,'mlx::core::array::inputs()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a27ded7e54bc1712063c874646b445509',1,'mlx::core::metal::CommandEncoder::inputs()'],['../structmlx_1_1core_1_1_command_encoder.html#a27ded7e54bc1712063c874646b445509',1,'mlx::core::CommandEncoder::inputs()']]], - ['insert_34',['insert',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aaafe1a4305a107d4bcdd4f35d3df09b3',1,'mlx::core::metal::ResidencySet']]], - ['instantiate_5fdefault_5flimit_35',['instantiate_default_limit',['../backend_2metal_2kernels_2utils_8h.html#abedffa358e7ba7782cc78d6772064c7c',1,'utils.h']]], - ['instantiate_5ffloat_5flimit_36',['instantiate_float_limit',['../backend_2metal_2kernels_2utils_8h.html#a1e520e23f58ca645dea1ac20998d987a',1,'utils.h']]], - ['instantiate_5fkernel_37',['instantiate_kernel',['../defines_8h.html#a0cc4a821c1090d4183ff3a31da7e9f7b',1,'defines.h']]], - ['instantiate_5fmetal_5fmath_5ffuncs_38',['instantiate_metal_math_funcs',['../bf16__math_8h.html#a9e9f0fdd6e304522bc88acd22c576842',1,'bf16_math.h']]], - ['instantiate_5fmetal_5fsimd_5fcomm_5ffuncs_39',['instantiate_metal_simd_comm_funcs',['../bf16__math_8h.html#aecc11cb898846d01bfc9faa109fcf791',1,'bf16_math.h']]], - ['instantiate_5fmetal_5fsimd_5freduction_5ffuncs_40',['instantiate_metal_simd_reduction_funcs',['../bf16__math_8h.html#a51688bc24fc9292aaec5f54a58eaa2d0',1,'bf16_math.h']]], - ['int_41',['Int',['../namespacemlx_1_1steel.html#afe36ddf6725498d273e5eef4f1579891',1,'mlx::steel']]], - ['int16_42',['int16',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daace80d5ec65b1d2a2f1049eadc100db23',1,'mlx::core::Dtype::int16'],['../namespacemlx_1_1core.html#a71ebba4ad1afa730962f0692c4f42f07',1,'mlx::core::int16']]], - ['int32_43',['int32',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa0241adbbd83925f051b694d40f02747f',1,'mlx::core::Dtype::int32'],['../namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3',1,'mlx::core::int32']]], - ['int64_44',['int64',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daaff9b3f96d37353c528517bc3656a00a8',1,'mlx::core::Dtype::int64'],['../namespacemlx_1_1core.html#a9019bdc191054ada0a502c7c34cef5b8',1,'mlx::core::int64']]], - ['int8_45',['int8',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa27c006cc56b1ba88f960cf8b5144fcac',1,'mlx::core::Dtype::int8'],['../namespacemlx_1_1core.html#a892e934e146dd938d144cee8813ca672',1,'mlx::core::int8']]], - ['integer_46',['integer',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da157db7df530023575515d366c9b672e8',1,'mlx::core::Dtype::integer'],['../namespacemlx_1_1core.html#a074d000f25ae3ed77450e6a5fec4b38b',1,'mlx::core::integer']]], - ['integral_5fconst_5fbinop_47',['integral_const_binop',['../integral__constant_8h.html#ab28d2705f6fd4f54faccbb78fd5ddfb6',1,'integral_constant.h']]], - ['integral_5fconstant_48',['integral_constant',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], - ['integral_5fconstant_2eh_49',['integral_constant.h',['../integral__constant_8h.html',1,'']]], - ['integral_5fconstant_3c_20bool_2c_20b_20_3e_50',['integral_constant< bool, B >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], - ['integral_5fconstant_3c_20bool_2c_20b_20_3e_3c_20false_20_3e_51',['integral_constant< bool, B >< false >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], - ['integral_5fconstant_3c_20bool_2c_20b_20_3e_3c_20true_20_3e_52',['integral_constant< bool, B >< true >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], - ['integral_5fconstant_3c_20int_2c_20val_20_3e_53',['integral_constant< int, val >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], - ['intorfloat_54',['IntOrFloat',['../unionmlx_1_1core_1_1detail_1_1_int_or_float.html',1,'mlx::core::detail']]], + ['inf_25',['inf',['../namespacemlx_1_1core_1_1simd.html#a771b6597803beb800ff5e7560c41e341',1,'mlx::core::simd']]], + ['infinity_26',['infinity',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61eb741e7af49046beb863abf023b206',1,'metal::_numeric_limits_impl< bfloat16_t >::infinity()'],['../structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html#a2a8f3f489b47b7e8398bec9895ae0c27',1,'mlx::core::numeric_limits< float16_t >::infinity()'],['../structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html#a974982399d0211786599526abdb843b8',1,'mlx::core::numeric_limits< bfloat16_t >::infinity()']]], + ['init_27',['init',['../struct_cum_prod_3_01bool_01_4.html#ae7a8b0ba9e6898356b87b18766e76d2c',1,'CumProd< bool >::init'],['../struct_cum_max.html#a16480052a2eeb4340e546838aab59cc4',1,'CumMax::init'],['../struct_cum_min.html#a8b67f739c620d0cc194b533190990ab9',1,'CumMin::init'],['../struct_less_than.html#abf97a6b0163048e4ba96460939dbd3a3',1,'LessThan::init'],['../namespacemlx_1_1core_1_1distributed.html#a24cdcd2aa23a3410a8973753ade3f772',1,'mlx::core::distributed::init()'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#ab40a34a8837956e24fb9b9661104c8f9',1,'mlx::core::distributed::mpi::init()'],['../namespacemlx_1_1core_1_1distributed_1_1ring.html#a81a13abe6f334d2f6b058b39a2221e67',1,'mlx::core::distributed::ring::init()']]], + ['init_5freduce_28',['init_reduce',['../reduce__init_8h.html#a0088604ac2eaa6940689ff12c4ba5fc2',1,'reduce_init.h']]], + ['inner_29',['inner',['../group__ops.html#ga654fec16a9746b390916697a2ab2546e',1,'mlx::core']]], + ['inner_5flooper_30',['inner_looper',['../struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40',1,'LoopedElemToLoc::inner_looper'],['../struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40',1,'LoopedElemToLoc< 1, OffsetT, false >::inner_looper'],['../struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40',1,'LoopedElemToLoc< 1, OffsetT, true >::inner_looper']]], + ['inp_5fjump_5fc_31',['inp_jump_c',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a78d30e843d65d1829623afb0b607f0a5',1,'mlx::steel::ImplicitGemmConv2DParams']]], + ['inp_5fjump_5fh_32',['inp_jump_h',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a03685a4066cdb11ffb647408e2c5b122',1,'mlx::steel::ImplicitGemmConv2DParams']]], + ['inp_5fjump_5fw_33',['inp_jump_w',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#acf168c72f4a86b72b8f5f386f07c9d8c',1,'mlx::steel::ImplicitGemmConv2DParams']]], + ['inputs_34',['inputs',['../classmlx_1_1core_1_1array.html#a2913abcdf71826827c8457f529825fff',1,'mlx::core::array::inputs() const'],['../classmlx_1_1core_1_1array.html#acffb082177f9b78f0c52e406adff972f',1,'mlx::core::array::inputs()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a27ded7e54bc1712063c874646b445509',1,'mlx::core::metal::CommandEncoder::inputs()'],['../structmlx_1_1core_1_1_command_encoder.html#a27ded7e54bc1712063c874646b445509',1,'mlx::core::CommandEncoder::inputs()']]], + ['insert_35',['insert',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aaafe1a4305a107d4bcdd4f35d3df09b3',1,'mlx::core::metal::ResidencySet']]], + ['instantiate_5fdefault_5flimit_36',['instantiate_default_limit',['../backend_2metal_2kernels_2utils_8h.html#abedffa358e7ba7782cc78d6772064c7c',1,'utils.h']]], + ['instantiate_5ffloat_5flimit_37',['instantiate_float_limit',['../backend_2metal_2kernels_2utils_8h.html#a1e520e23f58ca645dea1ac20998d987a',1,'utils.h']]], + ['instantiate_5fkernel_38',['instantiate_kernel',['../defines_8h.html#a0cc4a821c1090d4183ff3a31da7e9f7b',1,'defines.h']]], + ['instantiate_5fmetal_5fmath_5ffuncs_39',['instantiate_metal_math_funcs',['../bf16__math_8h.html#a9e9f0fdd6e304522bc88acd22c576842',1,'bf16_math.h']]], + ['instantiate_5fmetal_5fsimd_5fcomm_5ffuncs_40',['instantiate_metal_simd_comm_funcs',['../bf16__math_8h.html#aecc11cb898846d01bfc9faa109fcf791',1,'bf16_math.h']]], + ['instantiate_5fmetal_5fsimd_5freduction_5ffuncs_41',['instantiate_metal_simd_reduction_funcs',['../bf16__math_8h.html#a51688bc24fc9292aaec5f54a58eaa2d0',1,'bf16_math.h']]], + ['int_42',['Int',['../namespacemlx_1_1steel.html#afe36ddf6725498d273e5eef4f1579891',1,'mlx::steel']]], + ['int16_43',['int16',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daace80d5ec65b1d2a2f1049eadc100db23',1,'mlx::core::Dtype::int16'],['../namespacemlx_1_1core.html#a71ebba4ad1afa730962f0692c4f42f07',1,'mlx::core::int16']]], + ['int32_44',['int32',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa0241adbbd83925f051b694d40f02747f',1,'mlx::core::Dtype::int32'],['../namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3',1,'mlx::core::int32']]], + ['int64_45',['int64',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daaff9b3f96d37353c528517bc3656a00a8',1,'mlx::core::Dtype::int64'],['../namespacemlx_1_1core.html#a9019bdc191054ada0a502c7c34cef5b8',1,'mlx::core::int64']]], + ['int8_46',['int8',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa27c006cc56b1ba88f960cf8b5144fcac',1,'mlx::core::Dtype::int8'],['../namespacemlx_1_1core.html#a892e934e146dd938d144cee8813ca672',1,'mlx::core::int8']]], + ['integer_47',['integer',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da157db7df530023575515d366c9b672e8',1,'mlx::core::Dtype::integer'],['../namespacemlx_1_1core.html#a074d000f25ae3ed77450e6a5fec4b38b',1,'mlx::core::integer']]], + ['integral_5fconst_5fbinop_48',['integral_const_binop',['../integral__constant_8h.html#ab28d2705f6fd4f54faccbb78fd5ddfb6',1,'integral_constant.h']]], + ['integral_5fconstant_49',['integral_constant',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], + ['integral_5fconstant_2eh_50',['integral_constant.h',['../integral__constant_8h.html',1,'']]], + ['integral_5fconstant_3c_20bool_2c_20b_20_3e_51',['integral_constant< bool, B >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], + ['integral_5fconstant_3c_20bool_2c_20b_20_3e_3c_20false_20_3e_52',['integral_constant< bool, B >< false >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], + ['integral_5fconstant_3c_20bool_2c_20b_20_3e_3c_20true_20_3e_53',['integral_constant< bool, B >< true >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], + ['integral_5fconstant_3c_20int_2c_20val_20_3e_54',['integral_constant< int, val >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], ['intracing_55',['InTracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html',1,'mlx::core::detail::InTracing'],['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#a6beb74f83bde21734ab46b8d999b3b0b',1,'mlx::core::detail::InTracing::InTracing()']]], ['inv_56',['inv',['../struct_read_writer.html#a773fa8524515bdc2ff8b0e2060741813',1,'ReadWriter::inv'],['../namespacemlx_1_1core_1_1linalg.html#ad966a0b6bff176c9f933534ed62389a2',1,'mlx::core::linalg::inv()']]], ['inv_5f_57',['inv_',['../backend_2metal_2kernels_2fft_8h.html#a7a83318497519ff3ff0141b7d511ed38',1,'fft.h']]], @@ -67,31 +67,33 @@ var searchData= ['is_64',['iS',['../struct_m_l_x_conv_params.html#a72e1c3b4da0f70622cf18036bbf97fe6',1,'MLXConvParams']]], ['is_5farray_5fv_65',['is_array_v',['../namespacemlx_1_1core.html#a01b0d64a75dfa2e95d6c7b5c53d708af',1,'mlx::core']]], ['is_5farrays_5fv_66',['is_arrays_v',['../namespacemlx_1_1core.html#a94c1057929b390e5613304afa16dfbda',1,'mlx::core']]], - ['is_5favailable_67',['is_available',['../classmlx_1_1core_1_1array.html#aebed1f37c19197be76105161102a8a40',1,'mlx::core::array::is_available()'],['../namespacemlx_1_1core_1_1metal.html#a0cdf2c08c7bc0927a86070adc206987f',1,'mlx::core::metal::is_available()'],['../namespacemlx_1_1core_1_1distributed.html#a95655473cd0032c06e5fe3fca85aeef3',1,'mlx::core::distributed::is_available()'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#a86d8a52e75b15bae8fb0992b418a41c7',1,'mlx::core::distributed::mpi::is_available()']]], - ['is_5fdonatable_68',['is_donatable',['../classmlx_1_1core_1_1array.html#a4677a404b5d191af20b52649225de087',1,'mlx::core::array::is_donatable()'],['../namespacemlx_1_1core.html#af650e831ce21759da1ac103037d08d84',1,'mlx::core::is_donatable()']]], - ['is_5fempty_69',['is_empty',['../structmetal_1_1is__empty.html',1,'metal']]], - ['is_5fequivalent_70',['is_equivalent',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#af08b1294f3f93505a96fdfa85b1edd62',1,'mlx::core::fast::ScaledDotProductAttention::is_equivalent()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a5936175e5923aec272d6f718785f57a1',1,'mlx::core::fast::AffineQuantize::is_equivalent()'],['../classmlx_1_1core_1_1_primitive.html#a6140a502af4c2bbbc776ab26e9afebcd',1,'mlx::core::Primitive::is_equivalent()'],['../classmlx_1_1core_1_1_abs.html#ab6f0ec56bc7c048382297e12dabadc67',1,'mlx::core::Abs::is_equivalent()'],['../classmlx_1_1core_1_1_add.html#aba0a35410c3aac53d0f7a0c283d9ee3f',1,'mlx::core::Add::is_equivalent()'],['../classmlx_1_1core_1_1_add_m_m.html#a6e37c6882dba995a63fb6d8dfb01754f',1,'mlx::core::AddMM::is_equivalent()'],['../classmlx_1_1core_1_1_arange.html#a7b6a45cf9c4b109d4e0373f3fe576c35',1,'mlx::core::Arange::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cos.html#a39557461e3235801886675a9b7d25bf5',1,'mlx::core::ArcCos::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6928e827b9ac2e86e7d5b02b78150eee',1,'mlx::core::ArcCosh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sin.html#a13b5e39eeccaf32d94b8eb85b3b753ab',1,'mlx::core::ArcSin::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sinh.html#a63c7a765c7906242dc3371deec094f0f',1,'mlx::core::ArcSinh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan.html#a0e5b5fc7218143ecd0a8666d9137c34c',1,'mlx::core::ArcTan::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan2.html#aeaee58cd803d3ebf0b76574a409682cc',1,'mlx::core::ArcTan2::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tanh.html#ac8ecdd640043dab0461d49d7650679a2',1,'mlx::core::ArcTanh::is_equivalent()'],['../classmlx_1_1core_1_1_arg_partition.html#ad87509ce70b51fb75dfb9c3a05a5b31a',1,'mlx::core::ArgPartition::is_equivalent()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03b81a670dcb1e39bf7279e4d4583b97',1,'mlx::core::ArgReduce::is_equivalent()'],['../classmlx_1_1core_1_1_arg_sort.html#a048cd09c557d29d1111726f97010a845',1,'mlx::core::ArgSort::is_equivalent()'],['../classmlx_1_1core_1_1_as_type.html#a8e6c8b2428ab15c4fb43f2e3a8fb38af',1,'mlx::core::AsType::is_equivalent()'],['../classmlx_1_1core_1_1_as_strided.html#a1738c6aa0a3a3eb68530f0d5b436e094',1,'mlx::core::AsStrided::is_equivalent()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8cd6b916b4838a6c329cf4df8530c3b8',1,'mlx::core::BitwiseBinary::is_equivalent()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aef1c303955f9b8f445296372cf181160',1,'mlx::core::BlockMaskedMM::is_equivalent()'],['../classmlx_1_1core_1_1_gather_m_m.html#a163f17f6ce2c002f22e81b302777342b',1,'mlx::core::GatherMM::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a44d14b22b995e643cb04cc43654d7b16',1,'mlx::core::BroadcastAxes::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast.html#a0e27692b0090ec451954649a36042616',1,'mlx::core::Broadcast::is_equivalent()'],['../classmlx_1_1core_1_1_ceil.html#aacd90acb56eb0649c1cef807aa21df52',1,'mlx::core::Ceil::is_equivalent()'],['../classmlx_1_1core_1_1_compiled.html#a63e5016458887813b4a59dee5a0a3f10',1,'mlx::core::Compiled::is_equivalent()'],['../classmlx_1_1core_1_1_concatenate.html#aaf8a72a0c30114460caf519580cc35d2',1,'mlx::core::Concatenate::is_equivalent()'],['../classmlx_1_1core_1_1_conjugate.html#af42f00a790c6bc5572bd8fe9e5b36c5e',1,'mlx::core::Conjugate::is_equivalent()'],['../classmlx_1_1core_1_1_contiguous.html#aa5d273a461fc6e64f3c9a67c24cb3372',1,'mlx::core::Contiguous::is_equivalent()'],['../classmlx_1_1core_1_1_convolution.html#afb87708a5e3aab2e9e663daa9d8863de',1,'mlx::core::Convolution::is_equivalent()'],['../classmlx_1_1core_1_1_copy.html#afcfa39465015f638e294aa954ea0f3da',1,'mlx::core::Copy::is_equivalent()'],['../classmlx_1_1core_1_1_cos.html#ab611ca38c987915659f7ffcce0370417',1,'mlx::core::Cos::is_equivalent()'],['../classmlx_1_1core_1_1_cosh.html#ae0bacccaf501f5349db0c13cca776ff9',1,'mlx::core::Cosh::is_equivalent()'],['../classmlx_1_1core_1_1_divide.html#a3dda091f05c4164c29bb8129e9712650',1,'mlx::core::Divide::is_equivalent()'],['../classmlx_1_1core_1_1_div_mod.html#af5fcf8ec8515d46844cbeeab6dafb38a',1,'mlx::core::DivMod::is_equivalent()'],['../classmlx_1_1core_1_1_select.html#afc3c333fac7f902c98839921ef2874c8',1,'mlx::core::Select::is_equivalent()'],['../classmlx_1_1core_1_1_remainder.html#a802039faaa2ed7b763ec3d7debcce814',1,'mlx::core::Remainder::is_equivalent()'],['../classmlx_1_1core_1_1_equal.html#a58c1c5003e43f47dc0788c1851deaa02',1,'mlx::core::Equal::is_equivalent()'],['../classmlx_1_1core_1_1_erf.html#abe99dfbc2954c3a7d5dec56ab165ee82',1,'mlx::core::Erf::is_equivalent()'],['../classmlx_1_1core_1_1_erf_inv.html#aaac9e3b454ba564f9c6e804ab6562832',1,'mlx::core::ErfInv::is_equivalent()'],['../classmlx_1_1core_1_1_exp.html#ac6e44bffe7a643ab4ca51e74c7328357',1,'mlx::core::Exp::is_equivalent()'],['../classmlx_1_1core_1_1_expand_dims.html#aef468da4027527afec7b24161ce1e1f3',1,'mlx::core::ExpandDims::is_equivalent()'],['../classmlx_1_1core_1_1_f_f_t.html#a0ede3bc8b6d77d560c0a750b68fddc06',1,'mlx::core::FFT::is_equivalent()'],['../classmlx_1_1core_1_1_flatten.html#a42499e796aac751fceb4628317cc58f4',1,'mlx::core::Flatten::is_equivalent()'],['../classmlx_1_1core_1_1_floor.html#a24b64feb026c4fcd02fc481cffdb1c94',1,'mlx::core::Floor::is_equivalent()'],['../classmlx_1_1core_1_1_full.html#afafcbcae1e28597fe8f7fde289105792',1,'mlx::core::Full::is_equivalent()'],['../classmlx_1_1core_1_1_gather.html#a23ff1406dbf0c770e75ad47440b467aa',1,'mlx::core::Gather::is_equivalent()'],['../classmlx_1_1core_1_1_greater.html#a6877a6888614a618dc64296763ccabb1',1,'mlx::core::Greater::is_equivalent()'],['../classmlx_1_1core_1_1_greater_equal.html#a3daef8596b963026b602019bc56fc5fc',1,'mlx::core::GreaterEqual::is_equivalent()'],['../classmlx_1_1core_1_1_hadamard.html#a8a528d8d69a7343bdfd704a3e74230b8',1,'mlx::core::Hadamard::is_equivalent()'],['../classmlx_1_1core_1_1_imag.html#a51c15ae82855edebba2ba779516465f5',1,'mlx::core::Imag::is_equivalent()'],['../classmlx_1_1core_1_1_less.html#a7d6ed6353a0dcefebd008026dbd3cd63',1,'mlx::core::Less::is_equivalent()'],['../classmlx_1_1core_1_1_less_equal.html#a76ee1438cf4bd109eae4e0b3472b26af',1,'mlx::core::LessEqual::is_equivalent()'],['../classmlx_1_1core_1_1_log.html#a2fc58ea4ca744db493b947d1136d05f8',1,'mlx::core::Log::is_equivalent()'],['../classmlx_1_1core_1_1_logical_not.html#aba53675da351cd9b71a73d475b4bbe99',1,'mlx::core::LogicalNot::is_equivalent()'],['../classmlx_1_1core_1_1_logical_and.html#a9572c35f72e0db2f7f86bbf42438a6be',1,'mlx::core::LogicalAnd::is_equivalent()'],['../classmlx_1_1core_1_1_logical_or.html#a9c8b10a5cf5c69fdc2362390197e4e71',1,'mlx::core::LogicalOr::is_equivalent()'],['../classmlx_1_1core_1_1_log_add_exp.html#a3cf9a202c05aff39919d713d6e2b32e4',1,'mlx::core::LogAddExp::is_equivalent()'],['../classmlx_1_1core_1_1_matmul.html#aab372b59eae0840fc4f75ef5719a2630',1,'mlx::core::Matmul::is_equivalent()'],['../classmlx_1_1core_1_1_maximum.html#a21fe93fbd7799682f481260aee8bdb46',1,'mlx::core::Maximum::is_equivalent()'],['../classmlx_1_1core_1_1_minimum.html#a56c54ee3293cc2cd84462b9ec7ac36b4',1,'mlx::core::Minimum::is_equivalent()'],['../classmlx_1_1core_1_1_multiply.html#ae288159fa2d6d35087a85aca8eafa9b2',1,'mlx::core::Multiply::is_equivalent()'],['../classmlx_1_1core_1_1_negative.html#ac2a4d8159c548639d6289980c8975823',1,'mlx::core::Negative::is_equivalent()'],['../classmlx_1_1core_1_1_not_equal.html#ac12fd6b3e2f2e7e4e622b59badf2c73d',1,'mlx::core::NotEqual::is_equivalent()'],['../classmlx_1_1core_1_1_number_of_elements.html#ad6a32565ccc64499e368e15bba0b438f',1,'mlx::core::NumberOfElements::is_equivalent()'],['../classmlx_1_1core_1_1_pad.html#aad7c3bfecafe435d6a8e807de4c7ea9b',1,'mlx::core::Pad::is_equivalent()'],['../classmlx_1_1core_1_1_partition.html#aabdf6ef4f2159b2bfe93e0e87d4772f8',1,'mlx::core::Partition::is_equivalent()'],['../classmlx_1_1core_1_1_power.html#a76b4ec9d1ff07f06189e414480453d68',1,'mlx::core::Power::is_equivalent()'],['../classmlx_1_1core_1_1_quantized_matmul.html#af28b36e3f40ea41785387800326cc8e1',1,'mlx::core::QuantizedMatmul::is_equivalent()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a6a7da6bcf657fcdb157c45bf35fdec11',1,'mlx::core::GatherQMM::is_equivalent()'],['../classmlx_1_1core_1_1_random_bits.html#a72ec915debf5823e7c0463045b2894e6',1,'mlx::core::RandomBits::is_equivalent()'],['../classmlx_1_1core_1_1_real.html#a6d9bed396862a9e9d6abfbdcd8d8d239',1,'mlx::core::Real::is_equivalent()'],['../classmlx_1_1core_1_1_reshape.html#abd07c53af476777a04307e0423784cf3',1,'mlx::core::Reshape::is_equivalent()'],['../classmlx_1_1core_1_1_reduce.html#abe8f3327d617d0dd7438f066497ae08e',1,'mlx::core::Reduce::is_equivalent()'],['../classmlx_1_1core_1_1_round.html#aeb3d8607bbba7345a3142d4cbd4e6927',1,'mlx::core::Round::is_equivalent()'],['../classmlx_1_1core_1_1_scan.html#a54445a4d677ca4fe2a58d08eb5223ac6',1,'mlx::core::Scan::is_equivalent()'],['../classmlx_1_1core_1_1_scatter.html#a0208172562abdc90472e6eb5f84c987f',1,'mlx::core::Scatter::is_equivalent()'],['../classmlx_1_1core_1_1_sigmoid.html#a04814ba1b0edf8299d5ca1bcb8749d8e',1,'mlx::core::Sigmoid::is_equivalent()'],['../classmlx_1_1core_1_1_sign.html#a8c0934acbcc4b146e5aacd35a8c445bb',1,'mlx::core::Sign::is_equivalent()'],['../classmlx_1_1core_1_1_sin.html#af00b0e5516f884996ce7a97e6c1e3e6a',1,'mlx::core::Sin::is_equivalent()'],['../classmlx_1_1core_1_1_sinh.html#adcb1878996fd4902cd550042dd6ad70d',1,'mlx::core::Sinh::is_equivalent()'],['../classmlx_1_1core_1_1_slice.html#a43202c3b8966ae1db9ab82072e4918b0',1,'mlx::core::Slice::is_equivalent()'],['../classmlx_1_1core_1_1_slice_update.html#a60f588acced42391e6e5615ae8d16119',1,'mlx::core::SliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a0325271def8d9ea9ed21eb27e51994b4',1,'mlx::core::DynamicSlice::is_equivalent()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ae6292d2b1f3221a7c8ef6b77cb466481',1,'mlx::core::DynamicSliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_softmax.html#a9215ed7bd36bc11276c58dfb9808d728',1,'mlx::core::Softmax::is_equivalent()'],['../classmlx_1_1core_1_1_sort.html#ae48f07cf641d54234fc4fb6529a33511',1,'mlx::core::Sort::is_equivalent()'],['../classmlx_1_1core_1_1_split.html#af25a0cc259573b9dce60d285eee18345',1,'mlx::core::Split::is_equivalent()'],['../classmlx_1_1core_1_1_square.html#a6abc881d44071019aa15481e5ea75ab2',1,'mlx::core::Square::is_equivalent()'],['../classmlx_1_1core_1_1_sqrt.html#ab871c2b8ab4a27a3f782a005d0e87c46',1,'mlx::core::Sqrt::is_equivalent()'],['../classmlx_1_1core_1_1_stop_gradient.html#a327539298b21d800d26482b94fce41b3',1,'mlx::core::StopGradient::is_equivalent()'],['../classmlx_1_1core_1_1_subtract.html#af1c05e1e3f703ba916d54f8ccbbd102b',1,'mlx::core::Subtract::is_equivalent()'],['../classmlx_1_1core_1_1_squeeze.html#a08f35991d36e30fa4c05a5c9e91feb93',1,'mlx::core::Squeeze::is_equivalent()'],['../classmlx_1_1core_1_1_tan.html#afdf46288e7f60ea7f878688347dff7e4',1,'mlx::core::Tan::is_equivalent()'],['../classmlx_1_1core_1_1_tanh.html#a0692a1de2373b86eb394252ed4fecfda',1,'mlx::core::Tanh::is_equivalent()'],['../classmlx_1_1core_1_1_unflatten.html#a6a89fc709aae0fb3e17035e39b5ccd58',1,'mlx::core::Unflatten::is_equivalent()'],['../classmlx_1_1core_1_1_view.html#a7cb8403a96a47cb258caac4e3b850f64',1,'mlx::core::View::is_equivalent()'],['../classmlx_1_1core_1_1_transpose.html#a799ec3c3fa9f1b9e6177c755252a3eab',1,'mlx::core::Transpose::is_equivalent()'],['../classmlx_1_1core_1_1_eigh.html#a09414e3fe88a952408d164d6dd0af381',1,'mlx::core::Eigh::is_equivalent()']]], - ['is_5fintegral_71',['is_integral',['../structmlx_1_1steel_1_1is__integral.html',1,'mlx::steel']]], - ['is_5fintegral_3c_20integral_5fconstant_3c_20t_2c_20v_20_3e_20_3e_72',['is_integral< integral_constant< T, v > >',['../structmlx_1_1steel_1_1is__integral.html',1,'mlx::steel']]], - ['is_5fintegral_5fv_73',['is_integral_v',['../namespacemlx_1_1steel.html#a92a3465716ea7fd682d22cecc08d45fd',1,'mlx::steel']]], - ['is_5fmetal_5fatomic_74',['is_metal_atomic',['../atomic_8h.html#a91a8bdcae647947a83c6689d7f252d24',1,'atomic.h']]], - ['is_5fopen_75',['is_open',['../classmlx_1_1core_1_1io_1_1_reader.html#a780f504058bd9c80cb3d105046a9f985',1,'mlx::core::io::Reader::is_open()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a85aa36bdb0dbfb8c5b6cfd955b03417a',1,'mlx::core::io::Writer::is_open()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a653009adbcbce8248bc666df502fdbde',1,'mlx::core::io::ParallelFileReader::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#ad5d2ee671a81700cb1658c41309d6676',1,'mlx::core::io::FileWriter::is_open()']]], - ['is_5fpower_5fof_5f2_76',['is_power_of_2',['../namespacemlx_1_1core.html#adacbc4526e8964b267a8ec3eb1bc1a32',1,'mlx::core']]], - ['is_5fpower_5fof_5f2_5f_77',['is_power_of_2_',['../backend_2metal_2kernels_2fft_8h.html#a2a4df90e329b84ee6c1890ba7c265c9c',1,'fft.h']]], - ['is_5fready_78',['is_ready',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3',1,'pocketfft::detail::threading::latch']]], - ['is_5fscalar_79',['is_scalar',['../namespacemlx_1_1core.html#a985c60929757190e0b4ec51f57c767d0',1,'mlx::core']]], - ['is_5fsignaled_80',['is_signaled',['../classmlx_1_1core_1_1_event.html#a05a9a3de88185b4a89e154242b4e770a',1,'mlx::core::Event']]], - ['is_5fstatic_81',['is_static',['../structmetal_1_1is__static.html',1,'metal']]], - ['is_5fstatic_5fcast_82',['is_static_cast',['../namespacemlx_1_1core.html#afd9e740e567f9d7c28e00113caf46d5f',1,'mlx::core']]], - ['is_5ftracer_83',['is_tracer',['../classmlx_1_1core_1_1array.html#af9acb115019b995354d366c4ac6b968c',1,'mlx::core::array']]], - ['isclose_84',['isclose',['../group__ops.html#ga51eac95c04400921c54716de14b52491',1,'mlx::core']]], - ['isfinite_85',['isfinite',['../group__ops.html#ga725ff0789f934b1fdd54ee29e47022ff',1,'mlx::core']]], - ['isinf_86',['isinf',['../group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844',1,'mlx::core']]], - ['isnan_87',['isnan',['../namespacemetal.html#a83320ba983d90dd1fa5847b6940dc0bb',1,'metal::isnan()'],['../group__ops.html#ga175592792471b0ffb45196dca4711ba6',1,'mlx::core::isnan(const array &a, StreamOrDevice s={})']]], - ['isneginf_88',['isneginf',['../group__ops.html#ga1940523da381ed7be50656a3bc465ff3',1,'mlx::core']]], - ['isposinf_89',['isposinf',['../group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993',1,'mlx::core']]], - ['issubdtype_90',['issubdtype',['../namespacemlx_1_1core.html#aca9e69b06f4212eba44bf0ce6711d5f7',1,'mlx::core::issubdtype(const Dtype &a, const Dtype &b)'],['../namespacemlx_1_1core.html#aca39f224c1d17bde35dfcb9088430704',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype &b)'],['../namespacemlx_1_1core.html#ae9ee4a7c205df061c1caa7e62b7504e8',1,'mlx::core::issubdtype(const Dtype &a, const Dtype::Category &b)'],['../namespacemlx_1_1core.html#ab5b1a5a3d545a5de00c3117f76d71a1d',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype::Category &b)']]], - ['item_91',['item',['../classmlx_1_1core_1_1array.html#a90c5afddc2fa3028c0f8099bd64c8a99',1,'mlx::core::array::item()'],['../classmlx_1_1core_1_1array.html#a8650a99a6b7549bc823b03ad92590ff7',1,'mlx::core::array::item() const']]], - ['itemsize_92',['itemsize',['../classmlx_1_1core_1_1array.html#af329d9432c92de87cbaa2de8454eefc0',1,'mlx::core::array']]], - ['iterator_5fcategory_93',['iterator_category',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a2cbf481e39164245668b3be6cbcc614d',1,'mlx::core::array::ArrayIterator']]] + ['is_5favailable_67',['is_available',['../classmlx_1_1core_1_1array.html#aebed1f37c19197be76105161102a8a40',1,'mlx::core::array::is_available()'],['../namespacemlx_1_1core_1_1metal.html#a0cdf2c08c7bc0927a86070adc206987f',1,'mlx::core::metal::is_available()'],['../namespacemlx_1_1core_1_1distributed.html#a95655473cd0032c06e5fe3fca85aeef3',1,'mlx::core::distributed::is_available()'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#a86d8a52e75b15bae8fb0992b418a41c7',1,'mlx::core::distributed::mpi::is_available()'],['../namespacemlx_1_1core_1_1distributed_1_1ring.html#a1238e89ee95ba016741f0abe91b540ac',1,'mlx::core::distributed::ring::is_available()']]], + ['is_5fcomplex_68',['is_complex',['../namespacemlx_1_1core_1_1simd.html#a64e80f096a8baf99ba8d396414473cc7',1,'mlx::core::simd']]], + ['is_5fcomplex_3c_20t_2c_20std_3a_3avoid_5ft_3c_20decltype_28std_3a_3adeclval_3c_20t_20_3e_28_29_2ereal_28_29_29_3e_20_3e_69',['is_complex< T, std::void_t< decltype(std::declval< T >().real())> >',['../namespacemlx_1_1core_1_1simd.html#a29fe8445e54a61f6bccc8d50f142ca54',1,'mlx::core::simd']]], + ['is_5fdonatable_70',['is_donatable',['../classmlx_1_1core_1_1array.html#a4677a404b5d191af20b52649225de087',1,'mlx::core::array::is_donatable()'],['../namespacemlx_1_1core.html#af650e831ce21759da1ac103037d08d84',1,'mlx::core::is_donatable()']]], + ['is_5fempty_71',['is_empty',['../structmetal_1_1is__empty.html',1,'metal']]], + ['is_5fequivalent_72',['is_equivalent',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#af08b1294f3f93505a96fdfa85b1edd62',1,'mlx::core::fast::ScaledDotProductAttention::is_equivalent()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a5936175e5923aec272d6f718785f57a1',1,'mlx::core::fast::AffineQuantize::is_equivalent()'],['../classmlx_1_1core_1_1_primitive.html#a6140a502af4c2bbbc776ab26e9afebcd',1,'mlx::core::Primitive::is_equivalent()'],['../classmlx_1_1core_1_1_abs.html#ab6f0ec56bc7c048382297e12dabadc67',1,'mlx::core::Abs::is_equivalent()'],['../classmlx_1_1core_1_1_add.html#aba0a35410c3aac53d0f7a0c283d9ee3f',1,'mlx::core::Add::is_equivalent()'],['../classmlx_1_1core_1_1_add_m_m.html#a6e37c6882dba995a63fb6d8dfb01754f',1,'mlx::core::AddMM::is_equivalent()'],['../classmlx_1_1core_1_1_arange.html#a7b6a45cf9c4b109d4e0373f3fe576c35',1,'mlx::core::Arange::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cos.html#a39557461e3235801886675a9b7d25bf5',1,'mlx::core::ArcCos::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6928e827b9ac2e86e7d5b02b78150eee',1,'mlx::core::ArcCosh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sin.html#a13b5e39eeccaf32d94b8eb85b3b753ab',1,'mlx::core::ArcSin::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sinh.html#a63c7a765c7906242dc3371deec094f0f',1,'mlx::core::ArcSinh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan.html#a0e5b5fc7218143ecd0a8666d9137c34c',1,'mlx::core::ArcTan::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan2.html#aeaee58cd803d3ebf0b76574a409682cc',1,'mlx::core::ArcTan2::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tanh.html#ac8ecdd640043dab0461d49d7650679a2',1,'mlx::core::ArcTanh::is_equivalent()'],['../classmlx_1_1core_1_1_arg_partition.html#ad87509ce70b51fb75dfb9c3a05a5b31a',1,'mlx::core::ArgPartition::is_equivalent()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03b81a670dcb1e39bf7279e4d4583b97',1,'mlx::core::ArgReduce::is_equivalent()'],['../classmlx_1_1core_1_1_arg_sort.html#a048cd09c557d29d1111726f97010a845',1,'mlx::core::ArgSort::is_equivalent()'],['../classmlx_1_1core_1_1_as_type.html#a8e6c8b2428ab15c4fb43f2e3a8fb38af',1,'mlx::core::AsType::is_equivalent()'],['../classmlx_1_1core_1_1_as_strided.html#a1738c6aa0a3a3eb68530f0d5b436e094',1,'mlx::core::AsStrided::is_equivalent()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8cd6b916b4838a6c329cf4df8530c3b8',1,'mlx::core::BitwiseBinary::is_equivalent()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aef1c303955f9b8f445296372cf181160',1,'mlx::core::BlockMaskedMM::is_equivalent()'],['../classmlx_1_1core_1_1_gather_m_m.html#a163f17f6ce2c002f22e81b302777342b',1,'mlx::core::GatherMM::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a44d14b22b995e643cb04cc43654d7b16',1,'mlx::core::BroadcastAxes::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast.html#a0e27692b0090ec451954649a36042616',1,'mlx::core::Broadcast::is_equivalent()'],['../classmlx_1_1core_1_1_ceil.html#aacd90acb56eb0649c1cef807aa21df52',1,'mlx::core::Ceil::is_equivalent()'],['../classmlx_1_1core_1_1_compiled.html#a63e5016458887813b4a59dee5a0a3f10',1,'mlx::core::Compiled::is_equivalent()'],['../classmlx_1_1core_1_1_concatenate.html#aaf8a72a0c30114460caf519580cc35d2',1,'mlx::core::Concatenate::is_equivalent()'],['../classmlx_1_1core_1_1_conjugate.html#af42f00a790c6bc5572bd8fe9e5b36c5e',1,'mlx::core::Conjugate::is_equivalent()'],['../classmlx_1_1core_1_1_contiguous.html#aa5d273a461fc6e64f3c9a67c24cb3372',1,'mlx::core::Contiguous::is_equivalent()'],['../classmlx_1_1core_1_1_convolution.html#afb87708a5e3aab2e9e663daa9d8863de',1,'mlx::core::Convolution::is_equivalent()'],['../classmlx_1_1core_1_1_copy.html#afcfa39465015f638e294aa954ea0f3da',1,'mlx::core::Copy::is_equivalent()'],['../classmlx_1_1core_1_1_cos.html#ab611ca38c987915659f7ffcce0370417',1,'mlx::core::Cos::is_equivalent()'],['../classmlx_1_1core_1_1_cosh.html#ae0bacccaf501f5349db0c13cca776ff9',1,'mlx::core::Cosh::is_equivalent()'],['../classmlx_1_1core_1_1_divide.html#a3dda091f05c4164c29bb8129e9712650',1,'mlx::core::Divide::is_equivalent()'],['../classmlx_1_1core_1_1_div_mod.html#af5fcf8ec8515d46844cbeeab6dafb38a',1,'mlx::core::DivMod::is_equivalent()'],['../classmlx_1_1core_1_1_select.html#afc3c333fac7f902c98839921ef2874c8',1,'mlx::core::Select::is_equivalent()'],['../classmlx_1_1core_1_1_remainder.html#a802039faaa2ed7b763ec3d7debcce814',1,'mlx::core::Remainder::is_equivalent()'],['../classmlx_1_1core_1_1_equal.html#a58c1c5003e43f47dc0788c1851deaa02',1,'mlx::core::Equal::is_equivalent()'],['../classmlx_1_1core_1_1_erf.html#abe99dfbc2954c3a7d5dec56ab165ee82',1,'mlx::core::Erf::is_equivalent()'],['../classmlx_1_1core_1_1_erf_inv.html#aaac9e3b454ba564f9c6e804ab6562832',1,'mlx::core::ErfInv::is_equivalent()'],['../classmlx_1_1core_1_1_exp.html#ac6e44bffe7a643ab4ca51e74c7328357',1,'mlx::core::Exp::is_equivalent()'],['../classmlx_1_1core_1_1_expand_dims.html#aef468da4027527afec7b24161ce1e1f3',1,'mlx::core::ExpandDims::is_equivalent()'],['../classmlx_1_1core_1_1_f_f_t.html#a0ede3bc8b6d77d560c0a750b68fddc06',1,'mlx::core::FFT::is_equivalent()'],['../classmlx_1_1core_1_1_flatten.html#a42499e796aac751fceb4628317cc58f4',1,'mlx::core::Flatten::is_equivalent()'],['../classmlx_1_1core_1_1_floor.html#a24b64feb026c4fcd02fc481cffdb1c94',1,'mlx::core::Floor::is_equivalent()'],['../classmlx_1_1core_1_1_full.html#afafcbcae1e28597fe8f7fde289105792',1,'mlx::core::Full::is_equivalent()'],['../classmlx_1_1core_1_1_gather.html#a23ff1406dbf0c770e75ad47440b467aa',1,'mlx::core::Gather::is_equivalent()'],['../classmlx_1_1core_1_1_gather_axis.html#a8f603c5c46d566654bd8a615d24c1089',1,'mlx::core::GatherAxis::is_equivalent()'],['../classmlx_1_1core_1_1_greater.html#a6877a6888614a618dc64296763ccabb1',1,'mlx::core::Greater::is_equivalent()'],['../classmlx_1_1core_1_1_greater_equal.html#a3daef8596b963026b602019bc56fc5fc',1,'mlx::core::GreaterEqual::is_equivalent()'],['../classmlx_1_1core_1_1_hadamard.html#a8a528d8d69a7343bdfd704a3e74230b8',1,'mlx::core::Hadamard::is_equivalent()'],['../classmlx_1_1core_1_1_imag.html#a51c15ae82855edebba2ba779516465f5',1,'mlx::core::Imag::is_equivalent()'],['../classmlx_1_1core_1_1_less.html#a7d6ed6353a0dcefebd008026dbd3cd63',1,'mlx::core::Less::is_equivalent()'],['../classmlx_1_1core_1_1_less_equal.html#a76ee1438cf4bd109eae4e0b3472b26af',1,'mlx::core::LessEqual::is_equivalent()'],['../classmlx_1_1core_1_1_log.html#a2fc58ea4ca744db493b947d1136d05f8',1,'mlx::core::Log::is_equivalent()'],['../classmlx_1_1core_1_1_logical_not.html#aba53675da351cd9b71a73d475b4bbe99',1,'mlx::core::LogicalNot::is_equivalent()'],['../classmlx_1_1core_1_1_logical_and.html#a9572c35f72e0db2f7f86bbf42438a6be',1,'mlx::core::LogicalAnd::is_equivalent()'],['../classmlx_1_1core_1_1_logical_or.html#a9c8b10a5cf5c69fdc2362390197e4e71',1,'mlx::core::LogicalOr::is_equivalent()'],['../classmlx_1_1core_1_1_log_add_exp.html#a3cf9a202c05aff39919d713d6e2b32e4',1,'mlx::core::LogAddExp::is_equivalent()'],['../classmlx_1_1core_1_1_matmul.html#aab372b59eae0840fc4f75ef5719a2630',1,'mlx::core::Matmul::is_equivalent()'],['../classmlx_1_1core_1_1_maximum.html#a21fe93fbd7799682f481260aee8bdb46',1,'mlx::core::Maximum::is_equivalent()'],['../classmlx_1_1core_1_1_minimum.html#a56c54ee3293cc2cd84462b9ec7ac36b4',1,'mlx::core::Minimum::is_equivalent()'],['../classmlx_1_1core_1_1_multiply.html#ae288159fa2d6d35087a85aca8eafa9b2',1,'mlx::core::Multiply::is_equivalent()'],['../classmlx_1_1core_1_1_negative.html#ac2a4d8159c548639d6289980c8975823',1,'mlx::core::Negative::is_equivalent()'],['../classmlx_1_1core_1_1_not_equal.html#ac12fd6b3e2f2e7e4e622b59badf2c73d',1,'mlx::core::NotEqual::is_equivalent()'],['../classmlx_1_1core_1_1_number_of_elements.html#ad6a32565ccc64499e368e15bba0b438f',1,'mlx::core::NumberOfElements::is_equivalent()'],['../classmlx_1_1core_1_1_pad.html#aad7c3bfecafe435d6a8e807de4c7ea9b',1,'mlx::core::Pad::is_equivalent()'],['../classmlx_1_1core_1_1_partition.html#aabdf6ef4f2159b2bfe93e0e87d4772f8',1,'mlx::core::Partition::is_equivalent()'],['../classmlx_1_1core_1_1_power.html#a76b4ec9d1ff07f06189e414480453d68',1,'mlx::core::Power::is_equivalent()'],['../classmlx_1_1core_1_1_quantized_matmul.html#af28b36e3f40ea41785387800326cc8e1',1,'mlx::core::QuantizedMatmul::is_equivalent()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a6a7da6bcf657fcdb157c45bf35fdec11',1,'mlx::core::GatherQMM::is_equivalent()'],['../classmlx_1_1core_1_1_random_bits.html#a72ec915debf5823e7c0463045b2894e6',1,'mlx::core::RandomBits::is_equivalent()'],['../classmlx_1_1core_1_1_real.html#a6d9bed396862a9e9d6abfbdcd8d8d239',1,'mlx::core::Real::is_equivalent()'],['../classmlx_1_1core_1_1_reshape.html#abd07c53af476777a04307e0423784cf3',1,'mlx::core::Reshape::is_equivalent()'],['../classmlx_1_1core_1_1_reduce.html#abe8f3327d617d0dd7438f066497ae08e',1,'mlx::core::Reduce::is_equivalent()'],['../classmlx_1_1core_1_1_round.html#aeb3d8607bbba7345a3142d4cbd4e6927',1,'mlx::core::Round::is_equivalent()'],['../classmlx_1_1core_1_1_scan.html#a54445a4d677ca4fe2a58d08eb5223ac6',1,'mlx::core::Scan::is_equivalent()'],['../classmlx_1_1core_1_1_scatter.html#a0208172562abdc90472e6eb5f84c987f',1,'mlx::core::Scatter::is_equivalent()'],['../classmlx_1_1core_1_1_scatter_axis.html#af511c39926d5b85ca59558d64e0608fb',1,'mlx::core::ScatterAxis::is_equivalent()'],['../classmlx_1_1core_1_1_sigmoid.html#a04814ba1b0edf8299d5ca1bcb8749d8e',1,'mlx::core::Sigmoid::is_equivalent()'],['../classmlx_1_1core_1_1_sign.html#a8c0934acbcc4b146e5aacd35a8c445bb',1,'mlx::core::Sign::is_equivalent()'],['../classmlx_1_1core_1_1_sin.html#af00b0e5516f884996ce7a97e6c1e3e6a',1,'mlx::core::Sin::is_equivalent()'],['../classmlx_1_1core_1_1_sinh.html#adcb1878996fd4902cd550042dd6ad70d',1,'mlx::core::Sinh::is_equivalent()'],['../classmlx_1_1core_1_1_slice.html#a43202c3b8966ae1db9ab82072e4918b0',1,'mlx::core::Slice::is_equivalent()'],['../classmlx_1_1core_1_1_slice_update.html#a60f588acced42391e6e5615ae8d16119',1,'mlx::core::SliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a0325271def8d9ea9ed21eb27e51994b4',1,'mlx::core::DynamicSlice::is_equivalent()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ae6292d2b1f3221a7c8ef6b77cb466481',1,'mlx::core::DynamicSliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_softmax.html#a9215ed7bd36bc11276c58dfb9808d728',1,'mlx::core::Softmax::is_equivalent()'],['../classmlx_1_1core_1_1_sort.html#ae48f07cf641d54234fc4fb6529a33511',1,'mlx::core::Sort::is_equivalent()'],['../classmlx_1_1core_1_1_split.html#af25a0cc259573b9dce60d285eee18345',1,'mlx::core::Split::is_equivalent()'],['../classmlx_1_1core_1_1_square.html#a6abc881d44071019aa15481e5ea75ab2',1,'mlx::core::Square::is_equivalent()'],['../classmlx_1_1core_1_1_sqrt.html#ab871c2b8ab4a27a3f782a005d0e87c46',1,'mlx::core::Sqrt::is_equivalent()'],['../classmlx_1_1core_1_1_stop_gradient.html#a327539298b21d800d26482b94fce41b3',1,'mlx::core::StopGradient::is_equivalent()'],['../classmlx_1_1core_1_1_subtract.html#af1c05e1e3f703ba916d54f8ccbbd102b',1,'mlx::core::Subtract::is_equivalent()'],['../classmlx_1_1core_1_1_squeeze.html#a08f35991d36e30fa4c05a5c9e91feb93',1,'mlx::core::Squeeze::is_equivalent()'],['../classmlx_1_1core_1_1_tan.html#afdf46288e7f60ea7f878688347dff7e4',1,'mlx::core::Tan::is_equivalent()'],['../classmlx_1_1core_1_1_tanh.html#a0692a1de2373b86eb394252ed4fecfda',1,'mlx::core::Tanh::is_equivalent()'],['../classmlx_1_1core_1_1_unflatten.html#a6a89fc709aae0fb3e17035e39b5ccd58',1,'mlx::core::Unflatten::is_equivalent()'],['../classmlx_1_1core_1_1_view.html#a7cb8403a96a47cb258caac4e3b850f64',1,'mlx::core::View::is_equivalent()'],['../classmlx_1_1core_1_1_transpose.html#a799ec3c3fa9f1b9e6177c755252a3eab',1,'mlx::core::Transpose::is_equivalent()'],['../classmlx_1_1core_1_1_eigh.html#a09414e3fe88a952408d164d6dd0af381',1,'mlx::core::Eigh::is_equivalent()']]], + ['is_5fintegral_73',['is_integral',['../structmlx_1_1steel_1_1is__integral.html',1,'mlx::steel']]], + ['is_5fintegral_3c_20integral_5fconstant_3c_20t_2c_20v_20_3e_20_3e_74',['is_integral< integral_constant< T, v > >',['../structmlx_1_1steel_1_1is__integral_3_01integral__constant_3_01_t_00_01v_01_4_01_4.html',1,'mlx::steel']]], + ['is_5fintegral_5fv_75',['is_integral_v',['../namespacemlx_1_1steel.html#a92a3465716ea7fd682d22cecc08d45fd',1,'mlx::steel']]], + ['is_5fmetal_5fatomic_76',['is_metal_atomic',['../atomic_8h.html#a91a8bdcae647947a83c6689d7f252d24',1,'atomic.h']]], + ['is_5fopen_77',['is_open',['../classmlx_1_1core_1_1io_1_1_reader.html#a780f504058bd9c80cb3d105046a9f985',1,'mlx::core::io::Reader::is_open()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a85aa36bdb0dbfb8c5b6cfd955b03417a',1,'mlx::core::io::Writer::is_open()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a653009adbcbce8248bc666df502fdbde',1,'mlx::core::io::ParallelFileReader::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#ad5d2ee671a81700cb1658c41309d6676',1,'mlx::core::io::FileWriter::is_open()']]], + ['is_5fpower_5fof_5f2_78',['is_power_of_2',['../namespacemlx_1_1core.html#adacbc4526e8964b267a8ec3eb1bc1a32',1,'mlx::core']]], + ['is_5fpower_5fof_5f2_5f_79',['is_power_of_2_',['../backend_2metal_2kernels_2fft_8h.html#a2a4df90e329b84ee6c1890ba7c265c9c',1,'fft.h']]], + ['is_5fready_80',['is_ready',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3',1,'pocketfft::detail::threading::latch']]], + ['is_5fscalar_81',['is_scalar',['../namespacemlx_1_1core.html#a985c60929757190e0b4ec51f57c767d0',1,'mlx::core']]], + ['is_5fsignaled_82',['is_signaled',['../classmlx_1_1core_1_1_event.html#a05a9a3de88185b4a89e154242b4e770a',1,'mlx::core::Event']]], + ['is_5fstatic_83',['is_static',['../structmetal_1_1is__static.html',1,'metal']]], + ['is_5fstatic_5fcast_84',['is_static_cast',['../namespacemlx_1_1core.html#afd9e740e567f9d7c28e00113caf46d5f',1,'mlx::core']]], + ['is_5ftracer_85',['is_tracer',['../classmlx_1_1core_1_1array.html#af9acb115019b995354d366c4ac6b968c',1,'mlx::core::array']]], + ['isclose_86',['isclose',['../group__ops.html#ga51eac95c04400921c54716de14b52491',1,'mlx::core']]], + ['isfinite_87',['isfinite',['../group__ops.html#ga725ff0789f934b1fdd54ee29e47022ff',1,'mlx::core']]], + ['isinf_88',['isinf',['../group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844',1,'mlx::core']]], + ['isnan_89',['isnan',['../namespacemlx_1_1core_1_1simd.html#a05f4422a037c3bef343fb11f71363b65',1,'mlx::core::simd::isnan(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ac1c6c9b8bc7f3cd32ae39fa84975194d',1,'mlx::core::simd::isnan(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a20ffdefe25beda96860a1dc9a6f4aa02',1,'mlx::core::simd::isnan(Simd< float16_t, N > v)'],['../namespacemetal.html#a83320ba983d90dd1fa5847b6940dc0bb',1,'metal::isnan()'],['../group__ops.html#ga175592792471b0ffb45196dca4711ba6',1,'mlx::core::isnan(const array &a, StreamOrDevice s={})']]], + ['isneginf_90',['isneginf',['../group__ops.html#ga1940523da381ed7be50656a3bc465ff3',1,'mlx::core']]], + ['isposinf_91',['isposinf',['../group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993',1,'mlx::core']]], + ['issubdtype_92',['issubdtype',['../namespacemlx_1_1core.html#aca9e69b06f4212eba44bf0ce6711d5f7',1,'mlx::core::issubdtype(const Dtype &a, const Dtype &b)'],['../namespacemlx_1_1core.html#aca39f224c1d17bde35dfcb9088430704',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype &b)'],['../namespacemlx_1_1core.html#ae9ee4a7c205df061c1caa7e62b7504e8',1,'mlx::core::issubdtype(const Dtype &a, const Dtype::Category &b)'],['../namespacemlx_1_1core.html#ab5b1a5a3d545a5de00c3117f76d71a1d',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype::Category &b)']]], + ['item_93',['item',['../classmlx_1_1core_1_1array.html#a90c5afddc2fa3028c0f8099bd64c8a99',1,'mlx::core::array::item()'],['../classmlx_1_1core_1_1array.html#a8650a99a6b7549bc823b03ad92590ff7',1,'mlx::core::array::item() const']]], + ['itemsize_94',['itemsize',['../classmlx_1_1core_1_1array.html#af329d9432c92de87cbaa2de8454eefc0',1,'mlx::core::array']]], + ['iterator_5fcategory_95',['iterator_category',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a2cbf481e39164245668b3be6cbcc614d',1,'mlx::core::array::ArrayIterator']]] ]; diff --git a/docs/build/html/search/all_a.js b/docs/build/html/search/all_a.js index c81f478ad..d46c01d18 100644 --- a/docs/build/html/search/all_a.js +++ b/docs/build/html/search/all_a.js @@ -6,5 +6,5 @@ var searchData= ['jit_5fif_3',['jit_if',['../backend_2metal_2kernels_2jit_2bf16_8h.html#aaf5bb88c2349054a6c4c2aefee63d3d2',1,'bf16.h']]], ['jitcompiler_4',['JitCompiler',['../classmlx_1_1core_1_1_jit_compiler.html',1,'mlx::core']]], ['jump_5fparams_5',['jump_params',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a21b9ee9168dad4af84a611f861519e77',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::jump_params'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aa5611e9a84bebaee966d2b339c214ff5',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::jump_params']]], - ['jvp_6',['jvp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#aeaf6f2b5955e7417cd1e36db42c45a80',1,'mlx::core::distributed::AllReduce::jvp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a96f08a4ea8453d0b4b737c7b07972913',1,'mlx::core::distributed::AllGather::jvp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#ac77b28702654df8e7d882a49357a9584',1,'mlx::core::fast::Custom::jvp()'],['../classmlx_1_1core_1_1_primitive.html#a9fecf38f53da08ba1947543c2b3158c2',1,'mlx::core::Primitive::jvp()'],['../classmlx_1_1core_1_1_abs.html#a6c1e6eeaf4f5e63898c3487106e88e11',1,'mlx::core::Abs::jvp()'],['../classmlx_1_1core_1_1_add.html#a77230069f76fe60a2fe1007822a277b7',1,'mlx::core::Add::jvp()'],['../classmlx_1_1core_1_1_arc_cos.html#a240079c616f1a1f127aa783308096fe9',1,'mlx::core::ArcCos::jvp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a80fcb790649219c30260af903b76a1d7',1,'mlx::core::ArcCosh::jvp()'],['../classmlx_1_1core_1_1_arc_sin.html#a37affc8c5e84e5c54e73a71fc0821ea4',1,'mlx::core::ArcSin::jvp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79ebf2f6dfecbfbb93170fdd1ca87bf4',1,'mlx::core::ArcSinh::jvp()'],['../classmlx_1_1core_1_1_arc_tan.html#a0f5590a2297fc133b4b0a15f9dd0c760',1,'mlx::core::ArcTan::jvp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a01675433f2a4fa466b2f48272dbca738',1,'mlx::core::ArcTan2::jvp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a534ebdbfe77241884630d25021274c4a',1,'mlx::core::ArcTanh::jvp()'],['../classmlx_1_1core_1_1_arg_partition.html#aedea4b47f947a6fe358dd1238cdfb595',1,'mlx::core::ArgPartition::jvp()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03bb925e1b488c560bc3d67ce62ba6fa',1,'mlx::core::ArgReduce::jvp()'],['../classmlx_1_1core_1_1_as_type.html#a213400967150c57da35795e1c9f65ca0',1,'mlx::core::AsType::jvp()'],['../classmlx_1_1core_1_1_as_strided.html#a8ff0a398c47b42e08bc1122e07a02b53',1,'mlx::core::AsStrided::jvp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a1dae6ce5dc0498d20530403fe5c5531d',1,'mlx::core::BitwiseBinary::jvp()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a830bae1f3f9078bd5b422ce6e46685a7',1,'mlx::core::BroadcastAxes::jvp()'],['../classmlx_1_1core_1_1_broadcast.html#ae2fc3851a117079244708864be770ece',1,'mlx::core::Broadcast::jvp()'],['../classmlx_1_1core_1_1_ceil.html#a7ad74b27d9f26c886c2af516b845f066',1,'mlx::core::Ceil::jvp()'],['../classmlx_1_1core_1_1_compiled.html#aa385fe28626856ca5f57161b47a3c205',1,'mlx::core::Compiled::jvp()'],['../classmlx_1_1core_1_1_concatenate.html#a9f9e7a9dc3a00e02b84c94e1868baff1',1,'mlx::core::Concatenate::jvp()'],['../classmlx_1_1core_1_1_contiguous.html#a1f9fcae7235e0ae9217825b78cb0f991',1,'mlx::core::Contiguous::jvp()'],['../classmlx_1_1core_1_1_copy.html#a5acf02aa360cbefd86749fe9877b29cc',1,'mlx::core::Copy::jvp()'],['../classmlx_1_1core_1_1_cos.html#a99dd0b7e4aa2c838b77736f1fd539ee1',1,'mlx::core::Cos::jvp()'],['../classmlx_1_1core_1_1_cosh.html#a79facb0882443533f36a0a18407f5863',1,'mlx::core::Cosh::jvp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa9f695100170d5cae999b3da138ce720',1,'mlx::core::CustomTransforms::jvp()'],['../classmlx_1_1core_1_1_divide.html#ae1f408c447b17b3c84fe7f951d95559c',1,'mlx::core::Divide::jvp()'],['../classmlx_1_1core_1_1_div_mod.html#a1267401f25f25847888dd0a00b3fe3b9',1,'mlx::core::DivMod::jvp()'],['../classmlx_1_1core_1_1_select.html#a172df6812c2ea3e9d3c3fc5d527548d6',1,'mlx::core::Select::jvp()'],['../classmlx_1_1core_1_1_remainder.html#a972002173fc00ee86029d12bf1a9ba79',1,'mlx::core::Remainder::jvp()'],['../classmlx_1_1core_1_1_equal.html#a659d484589d7cd96d038922a1a98730f',1,'mlx::core::Equal::jvp()'],['../classmlx_1_1core_1_1_erf.html#ac733d605d80277d613954794eb8c46fe',1,'mlx::core::Erf::jvp()'],['../classmlx_1_1core_1_1_erf_inv.html#aa52710297ab6f7cd6826418c303e64be',1,'mlx::core::ErfInv::jvp()'],['../classmlx_1_1core_1_1_exp.html#aef6721832fcc283b082e35a7d436fa59',1,'mlx::core::Exp::jvp()'],['../classmlx_1_1core_1_1_expm1.html#ad463730632a00945d3a8addfdaec67b1',1,'mlx::core::Expm1::jvp()'],['../classmlx_1_1core_1_1_expand_dims.html#a2ccfe836a715dd8fa908731523f3ce5f',1,'mlx::core::ExpandDims::jvp()'],['../classmlx_1_1core_1_1_f_f_t.html#a34578814b6576f7b7b447541984ecba6',1,'mlx::core::FFT::jvp()'],['../classmlx_1_1core_1_1_flatten.html#ae0351b8d8088baa09db927ac8546186b',1,'mlx::core::Flatten::jvp()'],['../classmlx_1_1core_1_1_floor.html#aa47bc360ec563b6e7d93e8b50626d8af',1,'mlx::core::Floor::jvp()'],['../classmlx_1_1core_1_1_full.html#a281a865d0664596ac8d05ea8e7f26407',1,'mlx::core::Full::jvp()'],['../classmlx_1_1core_1_1_gather.html#ac54ef8fac92ab190f1793f3dd95b9e8d',1,'mlx::core::Gather::jvp()'],['../classmlx_1_1core_1_1_greater.html#aa47a9f80f45daf6a405e34f6dc7c99c1',1,'mlx::core::Greater::jvp()'],['../classmlx_1_1core_1_1_greater_equal.html#ac7346080aaaa01d52896127f383f9d20',1,'mlx::core::GreaterEqual::jvp()'],['../classmlx_1_1core_1_1_hadamard.html#a22b9d55ae3ba5eef63505124696e712a',1,'mlx::core::Hadamard::jvp()'],['../classmlx_1_1core_1_1_imag.html#ac01c5ed9b886983450ed9f049ddac55a',1,'mlx::core::Imag::jvp()'],['../classmlx_1_1core_1_1_less.html#af1493d566f6d940b8f674aac17f5dfce',1,'mlx::core::Less::jvp()'],['../classmlx_1_1core_1_1_less_equal.html#addfe62d3557d216f8307bdf1cbff6a8f',1,'mlx::core::LessEqual::jvp()'],['../classmlx_1_1core_1_1_log.html#ac646d4155322c34f58183d97301e3832',1,'mlx::core::Log::jvp()'],['../classmlx_1_1core_1_1_log1p.html#a537e44c7c993daf48698082e75e71ba2',1,'mlx::core::Log1p::jvp()'],['../classmlx_1_1core_1_1_logical_not.html#a4838c483ced707cfda3d6cd24bf4667c',1,'mlx::core::LogicalNot::jvp()'],['../classmlx_1_1core_1_1_logical_and.html#a78d3be71da224ea19158cf9e8c4cf434',1,'mlx::core::LogicalAnd::jvp()'],['../classmlx_1_1core_1_1_logical_or.html#a292de6001c551214c8152a7a5b0e6bd4',1,'mlx::core::LogicalOr::jvp()'],['../classmlx_1_1core_1_1_log_add_exp.html#aea2d1d58794e86f3488219ed3fa14329',1,'mlx::core::LogAddExp::jvp()'],['../classmlx_1_1core_1_1_maximum.html#a25ac5d5b453e571bf7240aa8de103c39',1,'mlx::core::Maximum::jvp()'],['../classmlx_1_1core_1_1_minimum.html#a10acf4fef35eed7ca55d131b5ae2d038',1,'mlx::core::Minimum::jvp()'],['../classmlx_1_1core_1_1_multiply.html#a79f7f0bb70de2e3e41a66c96285325b4',1,'mlx::core::Multiply::jvp()'],['../classmlx_1_1core_1_1_negative.html#a7d918f9b26b8fb7b047a27d85ebab979',1,'mlx::core::Negative::jvp()'],['../classmlx_1_1core_1_1_not_equal.html#ae2d3e5776efaefed7f4c73f679b02f17',1,'mlx::core::NotEqual::jvp()'],['../classmlx_1_1core_1_1_pad.html#a6e43a42032ef11497e8d91290574ec72',1,'mlx::core::Pad::jvp()'],['../classmlx_1_1core_1_1_partition.html#a310f569a163958940ed02cf52079746a',1,'mlx::core::Partition::jvp()'],['../classmlx_1_1core_1_1_power.html#a3e78b06453faa4fd149fd19c0e7a300a',1,'mlx::core::Power::jvp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ae51fdd0b81dd26c6687577567c126e23',1,'mlx::core::QuantizedMatmul::jvp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#adc579058752b927c71b45a962d4869e0',1,'mlx::core::GatherQMM::jvp()'],['../classmlx_1_1core_1_1_real.html#adff418a54970e2344bd3c2885aae5526',1,'mlx::core::Real::jvp()'],['../classmlx_1_1core_1_1_reshape.html#ab8fc28748991017cc3e29f93c91087a5',1,'mlx::core::Reshape::jvp()'],['../classmlx_1_1core_1_1_round.html#a032075a7d0dde2dba6189636d216c5e7',1,'mlx::core::Round::jvp()'],['../classmlx_1_1core_1_1_scan.html#a6f9c862f4fbc7eaf430a361cdd8933ee',1,'mlx::core::Scan::jvp()'],['../classmlx_1_1core_1_1_scatter.html#a270fa8ccf36ce4bbbc23875139223934',1,'mlx::core::Scatter::jvp()'],['../classmlx_1_1core_1_1_sigmoid.html#a62ca1c440896e32958c77af3340847db',1,'mlx::core::Sigmoid::jvp()'],['../classmlx_1_1core_1_1_sign.html#a957992c7aa0e86cf06f861a94372086b',1,'mlx::core::Sign::jvp()'],['../classmlx_1_1core_1_1_sin.html#af662d10180967399820496477ff050de',1,'mlx::core::Sin::jvp()'],['../classmlx_1_1core_1_1_sinh.html#a86e2b37823daf20a4c74c9f273215f9c',1,'mlx::core::Sinh::jvp()'],['../classmlx_1_1core_1_1_slice.html#a8288324045ab21d6c97b1695ce86ef36',1,'mlx::core::Slice::jvp()'],['../classmlx_1_1core_1_1_slice_update.html#a0ce3248cc61dae2b51d7aa8ee4197611',1,'mlx::core::SliceUpdate::jvp()'],['../classmlx_1_1core_1_1_dynamic_slice.html#acd0d2d6d83d4112e9e6fdd9ca8072ac3',1,'mlx::core::DynamicSlice::jvp()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ad1eae28869ebc2ecad87a9a01e314d56',1,'mlx::core::DynamicSliceUpdate::jvp()'],['../classmlx_1_1core_1_1_softmax.html#af96172634a24332b0fc8d7ca7e73f19f',1,'mlx::core::Softmax::jvp()'],['../classmlx_1_1core_1_1_sort.html#af113ac983473433eec851c8fddfcba62',1,'mlx::core::Sort::jvp()'],['../classmlx_1_1core_1_1_split.html#ab8a8d30fd1ebf0891f41f3c32eabe282',1,'mlx::core::Split::jvp()'],['../classmlx_1_1core_1_1_square.html#a822629b93b91e2bef29959431d95e22d',1,'mlx::core::Square::jvp()'],['../classmlx_1_1core_1_1_sqrt.html#a78544b1fb5da0c14bce3051ffd177818',1,'mlx::core::Sqrt::jvp()'],['../classmlx_1_1core_1_1_subtract.html#a8100081a99df5166f02efc76d6641220',1,'mlx::core::Subtract::jvp()'],['../classmlx_1_1core_1_1_squeeze.html#a65ac5f63f98d85453ad884e9fa6e8083',1,'mlx::core::Squeeze::jvp()'],['../classmlx_1_1core_1_1_tan.html#a5d7c76122d63619df17b0e45450bc8f2',1,'mlx::core::Tan::jvp()'],['../classmlx_1_1core_1_1_tanh.html#ae0fbb5370dc1c3a4fb0dd02ca28a832a',1,'mlx::core::Tanh::jvp()'],['../classmlx_1_1core_1_1_unflatten.html#aa3da5fc9920581931d6f9d4236a6d8e5',1,'mlx::core::Unflatten::jvp()'],['../classmlx_1_1core_1_1_transpose.html#ac1a523e25ab7fd9df4da363a922afbe1',1,'mlx::core::Transpose::jvp()'],['../namespacemlx_1_1core.html#a179a632200366c223d6ab56d3e032592',1,'mlx::core::jvp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &tangents)'],['../namespacemlx_1_1core.html#af38e7582db29519bb39326f6fa531d20',1,'mlx::core::jvp(const std::function< array(const array &)> &fun, const array &primal, const array &tangent)']]] + ['jvp_6',['jvp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#aeaf6f2b5955e7417cd1e36db42c45a80',1,'mlx::core::distributed::AllReduce::jvp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a96f08a4ea8453d0b4b737c7b07972913',1,'mlx::core::distributed::AllGather::jvp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#ac77b28702654df8e7d882a49357a9584',1,'mlx::core::fast::Custom::jvp()'],['../classmlx_1_1core_1_1_primitive.html#a9fecf38f53da08ba1947543c2b3158c2',1,'mlx::core::Primitive::jvp()'],['../classmlx_1_1core_1_1_abs.html#a6c1e6eeaf4f5e63898c3487106e88e11',1,'mlx::core::Abs::jvp()'],['../classmlx_1_1core_1_1_add.html#a77230069f76fe60a2fe1007822a277b7',1,'mlx::core::Add::jvp()'],['../classmlx_1_1core_1_1_add_m_m.html#a6572a4ffdd09ab857d3c09d9c5103f66',1,'mlx::core::AddMM::jvp()'],['../classmlx_1_1core_1_1_arc_cos.html#a240079c616f1a1f127aa783308096fe9',1,'mlx::core::ArcCos::jvp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a80fcb790649219c30260af903b76a1d7',1,'mlx::core::ArcCosh::jvp()'],['../classmlx_1_1core_1_1_arc_sin.html#a37affc8c5e84e5c54e73a71fc0821ea4',1,'mlx::core::ArcSin::jvp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79ebf2f6dfecbfbb93170fdd1ca87bf4',1,'mlx::core::ArcSinh::jvp()'],['../classmlx_1_1core_1_1_arc_tan.html#a0f5590a2297fc133b4b0a15f9dd0c760',1,'mlx::core::ArcTan::jvp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a01675433f2a4fa466b2f48272dbca738',1,'mlx::core::ArcTan2::jvp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a534ebdbfe77241884630d25021274c4a',1,'mlx::core::ArcTanh::jvp()'],['../classmlx_1_1core_1_1_arg_partition.html#aedea4b47f947a6fe358dd1238cdfb595',1,'mlx::core::ArgPartition::jvp()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03bb925e1b488c560bc3d67ce62ba6fa',1,'mlx::core::ArgReduce::jvp()'],['../classmlx_1_1core_1_1_as_type.html#a213400967150c57da35795e1c9f65ca0',1,'mlx::core::AsType::jvp()'],['../classmlx_1_1core_1_1_as_strided.html#a8ff0a398c47b42e08bc1122e07a02b53',1,'mlx::core::AsStrided::jvp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a1dae6ce5dc0498d20530403fe5c5531d',1,'mlx::core::BitwiseBinary::jvp()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a830bae1f3f9078bd5b422ce6e46685a7',1,'mlx::core::BroadcastAxes::jvp()'],['../classmlx_1_1core_1_1_broadcast.html#ae2fc3851a117079244708864be770ece',1,'mlx::core::Broadcast::jvp()'],['../classmlx_1_1core_1_1_ceil.html#a7ad74b27d9f26c886c2af516b845f066',1,'mlx::core::Ceil::jvp()'],['../classmlx_1_1core_1_1_compiled.html#aa385fe28626856ca5f57161b47a3c205',1,'mlx::core::Compiled::jvp()'],['../classmlx_1_1core_1_1_concatenate.html#a9f9e7a9dc3a00e02b84c94e1868baff1',1,'mlx::core::Concatenate::jvp()'],['../classmlx_1_1core_1_1_contiguous.html#a1f9fcae7235e0ae9217825b78cb0f991',1,'mlx::core::Contiguous::jvp()'],['../classmlx_1_1core_1_1_copy.html#a5acf02aa360cbefd86749fe9877b29cc',1,'mlx::core::Copy::jvp()'],['../classmlx_1_1core_1_1_cos.html#a99dd0b7e4aa2c838b77736f1fd539ee1',1,'mlx::core::Cos::jvp()'],['../classmlx_1_1core_1_1_cosh.html#a79facb0882443533f36a0a18407f5863',1,'mlx::core::Cosh::jvp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa9f695100170d5cae999b3da138ce720',1,'mlx::core::CustomTransforms::jvp()'],['../classmlx_1_1core_1_1_divide.html#ae1f408c447b17b3c84fe7f951d95559c',1,'mlx::core::Divide::jvp()'],['../classmlx_1_1core_1_1_div_mod.html#a1267401f25f25847888dd0a00b3fe3b9',1,'mlx::core::DivMod::jvp()'],['../classmlx_1_1core_1_1_select.html#a172df6812c2ea3e9d3c3fc5d527548d6',1,'mlx::core::Select::jvp()'],['../classmlx_1_1core_1_1_remainder.html#a972002173fc00ee86029d12bf1a9ba79',1,'mlx::core::Remainder::jvp()'],['../classmlx_1_1core_1_1_equal.html#a659d484589d7cd96d038922a1a98730f',1,'mlx::core::Equal::jvp()'],['../classmlx_1_1core_1_1_erf.html#ac733d605d80277d613954794eb8c46fe',1,'mlx::core::Erf::jvp()'],['../classmlx_1_1core_1_1_erf_inv.html#aa52710297ab6f7cd6826418c303e64be',1,'mlx::core::ErfInv::jvp()'],['../classmlx_1_1core_1_1_exp.html#aef6721832fcc283b082e35a7d436fa59',1,'mlx::core::Exp::jvp()'],['../classmlx_1_1core_1_1_expm1.html#ad463730632a00945d3a8addfdaec67b1',1,'mlx::core::Expm1::jvp()'],['../classmlx_1_1core_1_1_expand_dims.html#a2ccfe836a715dd8fa908731523f3ce5f',1,'mlx::core::ExpandDims::jvp()'],['../classmlx_1_1core_1_1_f_f_t.html#a34578814b6576f7b7b447541984ecba6',1,'mlx::core::FFT::jvp()'],['../classmlx_1_1core_1_1_flatten.html#ae0351b8d8088baa09db927ac8546186b',1,'mlx::core::Flatten::jvp()'],['../classmlx_1_1core_1_1_floor.html#aa47bc360ec563b6e7d93e8b50626d8af',1,'mlx::core::Floor::jvp()'],['../classmlx_1_1core_1_1_full.html#a281a865d0664596ac8d05ea8e7f26407',1,'mlx::core::Full::jvp()'],['../classmlx_1_1core_1_1_gather.html#ac54ef8fac92ab190f1793f3dd95b9e8d',1,'mlx::core::Gather::jvp()'],['../classmlx_1_1core_1_1_gather_axis.html#a4f6015bf2c9bb8773118eb51be45b378',1,'mlx::core::GatherAxis::jvp()'],['../classmlx_1_1core_1_1_greater.html#aa47a9f80f45daf6a405e34f6dc7c99c1',1,'mlx::core::Greater::jvp()'],['../classmlx_1_1core_1_1_greater_equal.html#ac7346080aaaa01d52896127f383f9d20',1,'mlx::core::GreaterEqual::jvp()'],['../classmlx_1_1core_1_1_hadamard.html#a22b9d55ae3ba5eef63505124696e712a',1,'mlx::core::Hadamard::jvp()'],['../classmlx_1_1core_1_1_imag.html#ac01c5ed9b886983450ed9f049ddac55a',1,'mlx::core::Imag::jvp()'],['../classmlx_1_1core_1_1_less.html#af1493d566f6d940b8f674aac17f5dfce',1,'mlx::core::Less::jvp()'],['../classmlx_1_1core_1_1_less_equal.html#addfe62d3557d216f8307bdf1cbff6a8f',1,'mlx::core::LessEqual::jvp()'],['../classmlx_1_1core_1_1_log.html#ac646d4155322c34f58183d97301e3832',1,'mlx::core::Log::jvp()'],['../classmlx_1_1core_1_1_log1p.html#a537e44c7c993daf48698082e75e71ba2',1,'mlx::core::Log1p::jvp()'],['../classmlx_1_1core_1_1_logical_not.html#a4838c483ced707cfda3d6cd24bf4667c',1,'mlx::core::LogicalNot::jvp()'],['../classmlx_1_1core_1_1_logical_and.html#a78d3be71da224ea19158cf9e8c4cf434',1,'mlx::core::LogicalAnd::jvp()'],['../classmlx_1_1core_1_1_logical_or.html#a292de6001c551214c8152a7a5b0e6bd4',1,'mlx::core::LogicalOr::jvp()'],['../classmlx_1_1core_1_1_log_add_exp.html#aea2d1d58794e86f3488219ed3fa14329',1,'mlx::core::LogAddExp::jvp()'],['../classmlx_1_1core_1_1_matmul.html#a6d949d8ab0fab0395532706c174686d5',1,'mlx::core::Matmul::jvp()'],['../classmlx_1_1core_1_1_maximum.html#a25ac5d5b453e571bf7240aa8de103c39',1,'mlx::core::Maximum::jvp()'],['../classmlx_1_1core_1_1_minimum.html#a10acf4fef35eed7ca55d131b5ae2d038',1,'mlx::core::Minimum::jvp()'],['../classmlx_1_1core_1_1_multiply.html#a79f7f0bb70de2e3e41a66c96285325b4',1,'mlx::core::Multiply::jvp()'],['../classmlx_1_1core_1_1_negative.html#a7d918f9b26b8fb7b047a27d85ebab979',1,'mlx::core::Negative::jvp()'],['../classmlx_1_1core_1_1_not_equal.html#ae2d3e5776efaefed7f4c73f679b02f17',1,'mlx::core::NotEqual::jvp()'],['../classmlx_1_1core_1_1_pad.html#a6e43a42032ef11497e8d91290574ec72',1,'mlx::core::Pad::jvp()'],['../classmlx_1_1core_1_1_partition.html#a310f569a163958940ed02cf52079746a',1,'mlx::core::Partition::jvp()'],['../classmlx_1_1core_1_1_power.html#a3e78b06453faa4fd149fd19c0e7a300a',1,'mlx::core::Power::jvp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ae51fdd0b81dd26c6687577567c126e23',1,'mlx::core::QuantizedMatmul::jvp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#adc579058752b927c71b45a962d4869e0',1,'mlx::core::GatherQMM::jvp()'],['../classmlx_1_1core_1_1_real.html#adff418a54970e2344bd3c2885aae5526',1,'mlx::core::Real::jvp()'],['../classmlx_1_1core_1_1_reshape.html#ab8fc28748991017cc3e29f93c91087a5',1,'mlx::core::Reshape::jvp()'],['../classmlx_1_1core_1_1_round.html#a032075a7d0dde2dba6189636d216c5e7',1,'mlx::core::Round::jvp()'],['../classmlx_1_1core_1_1_scan.html#a6f9c862f4fbc7eaf430a361cdd8933ee',1,'mlx::core::Scan::jvp()'],['../classmlx_1_1core_1_1_scatter.html#a270fa8ccf36ce4bbbc23875139223934',1,'mlx::core::Scatter::jvp()'],['../classmlx_1_1core_1_1_scatter_axis.html#a77129b601e5ca9d97669a8b0fdc69805',1,'mlx::core::ScatterAxis::jvp()'],['../classmlx_1_1core_1_1_sigmoid.html#a62ca1c440896e32958c77af3340847db',1,'mlx::core::Sigmoid::jvp()'],['../classmlx_1_1core_1_1_sign.html#a957992c7aa0e86cf06f861a94372086b',1,'mlx::core::Sign::jvp()'],['../classmlx_1_1core_1_1_sin.html#af662d10180967399820496477ff050de',1,'mlx::core::Sin::jvp()'],['../classmlx_1_1core_1_1_sinh.html#a86e2b37823daf20a4c74c9f273215f9c',1,'mlx::core::Sinh::jvp()'],['../classmlx_1_1core_1_1_slice.html#a8288324045ab21d6c97b1695ce86ef36',1,'mlx::core::Slice::jvp()'],['../classmlx_1_1core_1_1_slice_update.html#a0ce3248cc61dae2b51d7aa8ee4197611',1,'mlx::core::SliceUpdate::jvp()'],['../classmlx_1_1core_1_1_dynamic_slice.html#acd0d2d6d83d4112e9e6fdd9ca8072ac3',1,'mlx::core::DynamicSlice::jvp()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ad1eae28869ebc2ecad87a9a01e314d56',1,'mlx::core::DynamicSliceUpdate::jvp()'],['../classmlx_1_1core_1_1_softmax.html#af96172634a24332b0fc8d7ca7e73f19f',1,'mlx::core::Softmax::jvp()'],['../classmlx_1_1core_1_1_sort.html#af113ac983473433eec851c8fddfcba62',1,'mlx::core::Sort::jvp()'],['../classmlx_1_1core_1_1_split.html#ab8a8d30fd1ebf0891f41f3c32eabe282',1,'mlx::core::Split::jvp()'],['../classmlx_1_1core_1_1_square.html#a822629b93b91e2bef29959431d95e22d',1,'mlx::core::Square::jvp()'],['../classmlx_1_1core_1_1_sqrt.html#a78544b1fb5da0c14bce3051ffd177818',1,'mlx::core::Sqrt::jvp()'],['../classmlx_1_1core_1_1_subtract.html#a8100081a99df5166f02efc76d6641220',1,'mlx::core::Subtract::jvp()'],['../classmlx_1_1core_1_1_squeeze.html#a65ac5f63f98d85453ad884e9fa6e8083',1,'mlx::core::Squeeze::jvp()'],['../classmlx_1_1core_1_1_tan.html#a5d7c76122d63619df17b0e45450bc8f2',1,'mlx::core::Tan::jvp()'],['../classmlx_1_1core_1_1_tanh.html#ae0fbb5370dc1c3a4fb0dd02ca28a832a',1,'mlx::core::Tanh::jvp()'],['../classmlx_1_1core_1_1_unflatten.html#aa3da5fc9920581931d6f9d4236a6d8e5',1,'mlx::core::Unflatten::jvp()'],['../classmlx_1_1core_1_1_transpose.html#ac1a523e25ab7fd9df4da363a922afbe1',1,'mlx::core::Transpose::jvp()'],['../namespacemlx_1_1core.html#a179a632200366c223d6ab56d3e032592',1,'mlx::core::jvp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &tangents)'],['../namespacemlx_1_1core.html#af38e7582db29519bb39326f6fa531d20',1,'mlx::core::jvp(const std::function< array(const array &)> &fun, const array &primal, const array &tangent)']]] ]; diff --git a/docs/build/html/search/all_b.js b/docs/build/html/search/all_b.js index 701e328e8..efc4da08b 100644 --- a/docs/build/html/search/all_b.js +++ b/docs/build/html/search/all_b.js @@ -5,24 +5,26 @@ var searchData= ['kcols_2',['kCols',['../structmlx_1_1steel_1_1_c_shape.html#a01b09227356b6a682a0694523a8e6901',1,'mlx::steel::CShape::kCols'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a948784652e93830887ee8ad506ec3257',1,'mlx::steel::MMATile::kCols']]], ['kcolsperthread_3',['kColsPerThread',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1ea49efd92696b15302ee4b52ecd548c',1,'mlx::steel::MMATile']]], ['kdil_4',['kdil',['../struct_m_l_x_conv_params.html#a7611db8f1621c7e09fc685ed44073b14',1,'MLXConvParams']]], - ['kelemsperfrag_5',['kElemsPerFrag',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aef0ea2387e1ff5767bff8563b2d36bd6',1,'mlx::steel::MMATile']]], - ['kelemspertile_6',['kElemsPerTile',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a98357339ec98f804a1b12597937b318f',1,'mlx::steel::MMATile']]], - ['kernelmergesort_7',['KernelMergeSort',['../struct_kernel_merge_sort.html',1,'']]], - ['kernelmultiblockmergesort_8',['KernelMultiBlockMergeSort',['../struct_kernel_multi_block_merge_sort.html',1,'']]], - ['kernels_2eh_9',['kernels.h',['../kernels_8h.html',1,'']]], - ['key_10',['key',['../namespacemlx_1_1core_1_1random.html#acf04b6f42de11383e86dcc7f98c67bd8',1,'mlx::core::random']]], - ['keysequence_11',['KeySequence',['../classmlx_1_1core_1_1random_1_1_key_sequence.html',1,'mlx::core::random::KeySequence'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a196eb6ce5ba1eb37cc8c67d6d1332bfe',1,'mlx::core::random::KeySequence::KeySequence()']]], - ['kfragcols_12',['kFragCols',['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad095371db98e7c335ec41ca77c10f906',1,'mlx::steel::MMATile']]], - ['kfragrows_13',['kFragRows',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a594142f957ffb99296a243f7af7b59e7',1,'mlx::steel::MMATile']]], - ['kfragsize_14',['kFragSize',['../structmlx_1_1steel_1_1_block_m_m_a.html#aee8caec45c1f9e4428586effbfe6137d',1,'mlx::steel::BlockMMA']]], - ['kind_15',['Kind',['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715',1,'mlx::core::Dtype']]], - ['kindof_16',['kindof',['../namespacemlx_1_1core.html#ad527b86818823db040195785efd7d724',1,'mlx::core']]], - ['kl_17',['kL',['../structmlx_1_1steel_1_1_attn_params.html#a497b7404bcd25b535c3589c61f269f63',1,'mlx::steel::AttnParams']]], - ['knumfrags_18',['kNumFrags',['../structmlx_1_1steel_1_1_m_m_a_tile.html#ae326e7693eb77c22d5a6e3e9219019d3',1,'mlx::steel::MMATile']]], - ['kron_19',['kron',['../group__ops.html#ga6df16248cb68bc73644cdb1698967c19',1,'mlx::core']]], - ['krows_20',['kRows',['../structmlx_1_1steel_1_1_c_shape.html#a5caf36cb9acf9f90ba59a9b0b4197993',1,'mlx::steel::CShape::kRows'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a60ea6b8ff2923b7fe6f598e74ac54323',1,'mlx::steel::MMATile::kRows']]], - ['krowsperthread_21',['kRowsPerThread',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a5b1d1c85a5046108a4e38bdc5a0ea74e',1,'mlx::steel::MMATile']]], - ['ktilecols_22',['kTileCols',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a46324d40f8ad61cade08a1ebad6d9ad4',1,'mlx::steel::MMATile']]], - ['ktilerows_23',['kTileRows',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1d126b14910385ab644e224ac1d0307a',1,'mlx::steel::MMATile']]], - ['kwargs_24',['Kwargs',['../namespacemlx_1_1core.html#af3efb38b31c0bc08754a4edfda656b83',1,'mlx::core']]] + ['kelemcols_5',['kElemCols',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a7c212200d86b4e93f274d99addf668bd',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >']]], + ['kelemrows_6',['kElemRows',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a76aa5aa690dbcc954e957d767fad661f',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >']]], + ['kelemsperfrag_7',['kElemsPerFrag',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a3c34dfdc944db110f4735f1b25307cf0',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::kElemsPerFrag'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aef0ea2387e1ff5767bff8563b2d36bd6',1,'mlx::steel::MMATile::kElemsPerFrag']]], + ['kelemspertile_8',['kElemsPerTile',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a98357339ec98f804a1b12597937b318f',1,'mlx::steel::MMATile']]], + ['kernelmergesort_9',['KernelMergeSort',['../struct_kernel_merge_sort.html',1,'']]], + ['kernelmultiblockmergesort_10',['KernelMultiBlockMergeSort',['../struct_kernel_multi_block_merge_sort.html',1,'']]], + ['kernels_2eh_11',['kernels.h',['../kernels_8h.html',1,'']]], + ['key_12',['key',['../namespacemlx_1_1core_1_1random.html#acf04b6f42de11383e86dcc7f98c67bd8',1,'mlx::core::random']]], + ['keysequence_13',['KeySequence',['../classmlx_1_1core_1_1random_1_1_key_sequence.html',1,'mlx::core::random::KeySequence'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a196eb6ce5ba1eb37cc8c67d6d1332bfe',1,'mlx::core::random::KeySequence::KeySequence()']]], + ['kfragcols_14',['kFragCols',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a211102315e2afbcfcd2e2c201b638e9f',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::kFragCols'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad095371db98e7c335ec41ca77c10f906',1,'mlx::steel::MMATile::kFragCols']]], + ['kfragrows_15',['kFragRows',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a2fe53db449c692226f23f6b99fb2c0d4',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::kFragRows'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a594142f957ffb99296a243f7af7b59e7',1,'mlx::steel::MMATile::kFragRows']]], + ['kfragsize_16',['kFragSize',['../structmlx_1_1steel_1_1_block_m_m_a.html#aee8caec45c1f9e4428586effbfe6137d',1,'mlx::steel::BlockMMA']]], + ['kind_17',['Kind',['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715',1,'mlx::core::Dtype']]], + ['kindof_18',['kindof',['../namespacemlx_1_1core.html#ad527b86818823db040195785efd7d724',1,'mlx::core']]], + ['kl_19',['kL',['../structmlx_1_1steel_1_1_attn_params.html#a497b7404bcd25b535c3589c61f269f63',1,'mlx::steel::AttnParams']]], + ['knumfrags_20',['kNumFrags',['../structmlx_1_1steel_1_1_m_m_a_tile.html#ae326e7693eb77c22d5a6e3e9219019d3',1,'mlx::steel::MMATile']]], + ['kron_21',['kron',['../group__ops.html#ga6df16248cb68bc73644cdb1698967c19',1,'mlx::core']]], + ['krows_22',['kRows',['../structmlx_1_1steel_1_1_c_shape.html#a5caf36cb9acf9f90ba59a9b0b4197993',1,'mlx::steel::CShape::kRows'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a60ea6b8ff2923b7fe6f598e74ac54323',1,'mlx::steel::MMATile::kRows']]], + ['krowsperthread_23',['kRowsPerThread',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a5b1d1c85a5046108a4e38bdc5a0ea74e',1,'mlx::steel::MMATile']]], + ['ktilecols_24',['kTileCols',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a46324d40f8ad61cade08a1ebad6d9ad4',1,'mlx::steel::MMATile']]], + ['ktilerows_25',['kTileRows',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1d126b14910385ab644e224ac1d0307a',1,'mlx::steel::MMATile']]], + ['kwargs_26',['Kwargs',['../namespacemlx_1_1core.html#af3efb38b31c0bc08754a4edfda656b83',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/all_c.js b/docs/build/html/search/all_c.js index 0ab11e33e..f2ceefdeb 100644 --- a/docs/build/html/search/all_c.js +++ b/docs/build/html/search/all_c.js @@ -28,27 +28,27 @@ var searchData= ['lib_5fname_25',['lib_name',['../classmlx_1_1core_1_1_compiled.html#ae5c16cb91ac31b97e7652cc526c07439',1,'mlx::core::Compiled']]], ['limits_26',['Limits',['../struct_limits.html',1,'']]], ['limits_2eh_27',['limits.h',['../limits_8h.html',1,'']]], - ['limits_3c_20bfloat16_5ft_20_3e_28',['Limits< bfloat16_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20bool_20_3e_29',['Limits< bool >',['../struct_limits.html',1,'']]], - ['limits_3c_20complex64_5ft_20_3e_30',['Limits< complex64_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20float_20_3e_31',['Limits< float >',['../struct_limits.html',1,'']]], - ['limits_3c_20half_20_3e_32',['Limits< half >',['../struct_limits.html',1,'']]], - ['limits_3c_20int16_5ft_20_3e_33',['Limits< int16_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20int32_5ft_20_3e_34',['Limits< int32_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20int64_5ft_20_3e_35',['Limits< int64_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20int8_5ft_20_3e_36',['Limits< int8_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20uint16_5ft_20_3e_37',['Limits< uint16_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20uint32_5ft_20_3e_38',['Limits< uint32_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20uint64_5ft_20_3e_39',['Limits< uint64_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20uint8_5ft_20_3e_40',['Limits< uint8_t >',['../struct_limits.html',1,'']]], + ['limits_3c_20bfloat16_5ft_20_3e_28',['Limits< bfloat16_t >',['../struct_limits_3_01bfloat16__t_01_4.html',1,'']]], + ['limits_3c_20bool_20_3e_29',['Limits< bool >',['../struct_limits_3_01bool_01_4.html',1,'']]], + ['limits_3c_20complex64_5ft_20_3e_30',['Limits< complex64_t >',['../struct_limits_3_01complex64__t_01_4.html',1,'']]], + ['limits_3c_20float_20_3e_31',['Limits< float >',['../struct_limits_3_01float_01_4.html',1,'']]], + ['limits_3c_20half_20_3e_32',['Limits< half >',['../struct_limits_3_01half_01_4.html',1,'']]], + ['limits_3c_20int16_5ft_20_3e_33',['Limits< int16_t >',['../struct_limits_3_01int16__t_01_4.html',1,'']]], + ['limits_3c_20int32_5ft_20_3e_34',['Limits< int32_t >',['../struct_limits_3_01int32__t_01_4.html',1,'']]], + ['limits_3c_20int64_5ft_20_3e_35',['Limits< int64_t >',['../struct_limits_3_01int64__t_01_4.html',1,'']]], + ['limits_3c_20int8_5ft_20_3e_36',['Limits< int8_t >',['../struct_limits_3_01int8__t_01_4.html',1,'']]], + ['limits_3c_20uint16_5ft_20_3e_37',['Limits< uint16_t >',['../struct_limits_3_01uint16__t_01_4.html',1,'']]], + ['limits_3c_20uint32_5ft_20_3e_38',['Limits< uint32_t >',['../struct_limits_3_01uint32__t_01_4.html',1,'']]], + ['limits_3c_20uint64_5ft_20_3e_39',['Limits< uint64_t >',['../struct_limits_3_01uint64__t_01_4.html',1,'']]], + ['limits_3c_20uint8_5ft_20_3e_40',['Limits< uint8_t >',['../struct_limits_3_01uint8__t_01_4.html',1,'']]], ['linalg_2eh_41',['linalg.h',['../linalg_8h.html',1,'']]], ['linspace_42',['linspace',['../group__ops.html#ga968bcabed902311dcfbd903b0fb886ec',1,'mlx::core']]], ['load_43',['Load',['../classmlx_1_1core_1_1_load.html',1,'mlx::core::Load'],['../classmlx_1_1core_1_1_load.html#a3aa8a537cd90bab048df47dca1ed526a',1,'mlx::core::Load::Load()']]], - ['load_44',['load',['../struct_read_writer.html#a120eaf4b5f32e80972a18d14e82a2d75',1,'ReadWriter::load()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#ac73006b36fc710feda3a7c796e21415c',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa5426c6beabfb3ee41b58f01b3392a96',1,'mlx::steel::MMATile::load(const threadgroup U *src)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa9e484d8cae936503898d5b772c573f9',1,'mlx::steel::MMATile::load(const device U *src, const int ld)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#ac73006b36fc710feda3a7c796e21415c',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa5426c6beabfb3ee41b58f01b3392a96',1,'mlx::steel::MMATile::load(const threadgroup U *src)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa9e484d8cae936503898d5b772c573f9',1,'mlx::steel::MMATile::load(const device U *src, const int ld)'],['../struct_read_writer.html#a8a97ba42db5692898ef7391db08d8fd0',1,'ReadWriter::load() const'],['../struct_read_writer.html#a2506ee61be67826ac9494efb12a81900',1,'ReadWriter::load() const'],['../namespacemlx_1_1core.html#a954de19249da7c1fa39b89bdc47368aa',1,'mlx::core::load(array &out, size_t offset, const std::shared_ptr< io::Reader > &reader, bool swap_endianess)'],['../namespacemlx_1_1core.html#abada9bfa834d7423959362386720f3db',1,'mlx::core::load(std::shared_ptr< io::Reader > in_stream, StreamOrDevice s={})'],['../namespacemlx_1_1core.html#ac71a08bf4c052ae3c77e9e89cbea071d',1,'mlx::core::load(std::string file, StreamOrDevice s={})']]], + ['load_44',['load',['../struct_read_writer.html#a120eaf4b5f32e80972a18d14e82a2d75',1,'ReadWriter::load()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ac73006b36fc710feda3a7c796e21415c',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa5426c6beabfb3ee41b58f01b3392a96',1,'mlx::steel::MMATile::load(const threadgroup U *src)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa9e484d8cae936503898d5b772c573f9',1,'mlx::steel::MMATile::load(const device U *src, const int ld)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ac73006b36fc710feda3a7c796e21415c',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa5426c6beabfb3ee41b58f01b3392a96',1,'mlx::steel::MMATile::load(const threadgroup U *src)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa9e484d8cae936503898d5b772c573f9',1,'mlx::steel::MMATile::load(const device U *src, const int ld)'],['../struct_read_writer.html#a8a97ba42db5692898ef7391db08d8fd0',1,'ReadWriter::load() const'],['../struct_read_writer.html#a2506ee61be67826ac9494efb12a81900',1,'ReadWriter::load() const'],['../namespacemlx_1_1core.html#a954de19249da7c1fa39b89bdc47368aa',1,'mlx::core::load()'],['../namespacemlx_1_1core_1_1simd.html#a4041676517d96870293e5448c7e2b5a4',1,'mlx::core::simd::load()'],['../namespacemlx_1_1core.html#abada9bfa834d7423959362386720f3db',1,'mlx::core::load(std::shared_ptr< io::Reader > in_stream, StreamOrDevice s={})'],['../namespacemlx_1_1core.html#ac71a08bf4c052ae3c77e9e89cbea071d',1,'mlx::core::load(std::string file, StreamOrDevice s={})']]], ['load_2eh_45',['load.h',['../backend_2common_2load_8h.html',1,'(Global Namespace)'],['../io_2load_8h.html',1,'(Global Namespace)']]], ['load_5fgguf_46',['load_gguf',['../namespacemlx_1_1core.html#a2aa12b351ce559deb14cda0a5292c2ce',1,'mlx::core']]], ['load_5fpadded_47',['load_padded',['../struct_read_writer.html#add5bd3f647793a5a19d63197a19df73c',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#af3ce6bbb1a8dfb3bab1ae18d3eb45bc0',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#ab116f4569bb9dc6eaef0d8d08472e239',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const']]], - ['load_5fsafe_48',['load_safe',['../struct_g_e_m_v_kernel.html#a04bb72da9a93d6d1eba468fa311bbba7',1,'GEMVKernel::load_safe()'],['../struct_quantized_block_loader.html#a699dc9aa284b8fbf870310bbb224465b',1,'QuantizedBlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader_t.html#ac2d95e35ba39e0984e6f1e58ca935f7d',1,'mlx::steel::BlockLoaderT::load_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#ad22aaee4a2938cbdd315b39eda84e07d',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3a4af67813908109da08ce7352f82da',1,'mlx::steel::MMATile::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#ad22aaee4a2938cbdd315b39eda84e07d',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3a4af67813908109da08ce7352f82da',1,'mlx::steel::MMATile::load_safe()'],['../scan_8h.html#ae8eb101e538b85f8a4bcf451489ae0ac',1,'load_safe(): scan.h']]], + ['load_5fsafe_48',['load_safe',['../struct_g_e_m_v_kernel.html#a04bb72da9a93d6d1eba468fa311bbba7',1,'GEMVKernel::load_safe()'],['../struct_quantized_block_loader.html#a699dc9aa284b8fbf870310bbb224465b',1,'QuantizedBlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader_t.html#ac2d95e35ba39e0984e6f1e58ca935f7d',1,'mlx::steel::BlockLoaderT::load_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ad22aaee4a2938cbdd315b39eda84e07d',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3a4af67813908109da08ce7352f82da',1,'mlx::steel::MMATile::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ad22aaee4a2938cbdd315b39eda84e07d',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3a4af67813908109da08ce7352f82da',1,'mlx::steel::MMATile::load_safe()'],['../scan_8h.html#ae8eb101e538b85f8a4bcf451489ae0ac',1,'load_safe(): scan.h']]], ['load_5fsafetensors_49',['load_safetensors',['../namespacemlx_1_1core.html#a96cc40e1af8c4626c813ce4859f70a5c',1,'mlx::core::load_safetensors(std::shared_ptr< io::Reader > in_stream, StreamOrDevice s={})'],['../namespacemlx_1_1core.html#af7eea1682a38d363c56a066321e6d526',1,'mlx::core::load_safetensors(const std::string &file, StreamOrDevice s={})']]], ['load_5fstrided_50',['load_strided',['../struct_read_writer.html#a998ef484bade81f726b9edfc6b878197',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a3d9c8cbc582cad6b5218339d0f721559',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a795a71a8e1f154a5af415ebe1b3f0713',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a0935b946b8bf2e769427fcbf2da2f7be',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a7d45368c74a8b7c632659504b3273a13',1,'ReadWriter::load_strided(int stride, int overall_n)']]], ['load_5funsafe_51',['load_unsafe',['../struct_g_e_m_v_kernel.html#a6013e9c5b2f72fa1311dd038172df0ce',1,'GEMVKernel::load_unsafe()'],['../struct_quantized_block_loader.html#a86009527cb4b53e4c21fd6b1f78cfefc',1,'QuantizedBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader.html#a6c9e27f11f48b34580ed2c7e9cad9a27',1,'mlx::steel::BlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader_t.html#acb743f32146fdc7986264b7beb35fb38',1,'mlx::steel::BlockLoaderT::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a961836be363409744e48e595d5e0c2ec',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8034abc10483487fc94313e3674d1111',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a69e2f7c9814d1cc1c5c267be8618dc55',1,'mlx::steel::Conv2DWeightBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#aa11d1a142bc868df462f48a7102147f3',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a0e262b003ac0e7ee6272585eac921704',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3859ca11b5991ef6ee9b99afdc3ea30a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8f078982186421f5b484c0b53af9c655',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader.html#a6c9e27f11f48b34580ed2c7e9cad9a27',1,'mlx::steel::BlockLoader::load_unsafe()'],['../scan_8h.html#a9c415d07921f3961bad0a00a34f4a9a3',1,'load_unsafe(U values[N_READS], const device T *input): scan.h']]], @@ -61,15 +61,15 @@ var searchData= ['loader_5fchannel_5fn_2eh_58',['loader_channel_n.h',['../loader__channel__n_8h.html',1,'']]], ['loader_5fgeneral_2eh_59',['loader_general.h',['../loader__general_8h.html',1,'']]], ['loc_60',['loc',['../structmlx_1_1core_1_1_contiguous_iterator.html#a5ea4f0e40900e8c7e0830e1fb561af1a',1,'mlx::core::ContiguousIterator']]], - ['location_61',['location',['../struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e',1,'LoopedElemToLoc::location()'],['../struct_looped_elem_to_loc.html#a66b84b12f6c1494e5908989ed2849a9f',1,'LoopedElemToLoc< 1, OffsetT, true >::location()'],['../struct_looped_elem_to_loc.html#a89d9ec4dc2f2f0d77e27aa0c05f261ef',1,'LoopedElemToLoc< 1, OffsetT, false >::location()']]], + ['location_61',['location',['../struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e',1,'LoopedElemToLoc::location()'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a66b84b12f6c1494e5908989ed2849a9f',1,'LoopedElemToLoc< 1, OffsetT, true >::location()'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a89d9ec4dc2f2f0d77e27aa0c05f261ef',1,'LoopedElemToLoc< 1, OffsetT, false >::location()'],['../struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e',1,'LoopedElemToLoc< 1, OffsetT, false >::location()'],['../struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e',1,'LoopedElemToLoc< 1, OffsetT, true >::location()']]], ['log_62',['Log',['../struct_log.html',1,'Log'],['../structmlx_1_1core_1_1detail_1_1_log.html',1,'mlx::core::detail::Log'],['../classmlx_1_1core_1_1_log.html',1,'mlx::core::Log'],['../classmlx_1_1core_1_1_log.html#a663e54790c60b56eb0ff09f4f6635fb9',1,'mlx::core::Log::Log()']]], - ['log_63',['log',['../namespacemetal.html#a423a9f4f2fc7ef5ec7eda061277b51b6',1,'metal::log()'],['../namespacemetal_1_1fast.html#aef942e7f9e5c2e58c58644ab1bdd58d1',1,'metal::fast::log()'],['../namespacemetal_1_1precise.html#a341c2b8c27d1bed860f85f8b355023d4',1,'metal::precise::log()'],['../group__ops.html#ga6fb22d4926133573e430fcc92f4eef31',1,'mlx::core::log()']]], + ['log_63',['log',['../namespacemlx_1_1core_1_1simd.html#a8cec82f4fb15bfd31d7554c6c09ceed4',1,'mlx::core::simd::log(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a97a8ca857fe0edd84c68dc0f3dc2c6c4',1,'mlx::core::simd::log(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#abd09d3f5989558ce5156549a94d0fb04',1,'mlx::core::simd::log(Simd< T, 1 > in)'],['../namespacemetal.html#a423a9f4f2fc7ef5ec7eda061277b51b6',1,'metal::log()'],['../namespacemetal_1_1fast.html#aef942e7f9e5c2e58c58644ab1bdd58d1',1,'metal::fast::log()'],['../namespacemetal_1_1precise.html#a341c2b8c27d1bed860f85f8b355023d4',1,'metal::precise::log()'],['../group__ops.html#ga6fb22d4926133573e430fcc92f4eef31',1,'mlx::core::log()']]], ['log10_64',['Log10',['../struct_log10.html',1,'Log10'],['../structmlx_1_1core_1_1detail_1_1_log10.html',1,'mlx::core::detail::Log10']]], - ['log10_65',['log10',['../namespacemetal.html#a042b98827baa910e9d726227cec55a80',1,'metal::log10()'],['../namespacemetal_1_1fast.html#a0d1150cf2deee5100a7ea2988b3bb39e',1,'metal::fast::log10()'],['../namespacemetal_1_1precise.html#a44239067e8e9248b1574353f98e94d72',1,'metal::precise::log10()'],['../group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e',1,'mlx::core::log10()']]], + ['log10_65',['log10',['../namespacemlx_1_1core_1_1simd.html#a3cb6ea94836e999c07329b34c501ed85',1,'mlx::core::simd::log10(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#aba81b735e8f99cedf8b4846b2ab4e236',1,'mlx::core::simd::log10(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#adf754ade6cc1dd0e0bae0e31c7b513a2',1,'mlx::core::simd::log10(Simd< T, 1 > in)'],['../namespacemetal.html#a042b98827baa910e9d726227cec55a80',1,'metal::log10()'],['../namespacemetal_1_1fast.html#a0d1150cf2deee5100a7ea2988b3bb39e',1,'metal::fast::log10()'],['../namespacemetal_1_1precise.html#a44239067e8e9248b1574353f98e94d72',1,'metal::precise::log10()'],['../group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e',1,'mlx::core::log10()']]], ['log1p_66',['Log1p',['../struct_log1p.html',1,'Log1p'],['../structmlx_1_1core_1_1detail_1_1_log1p.html',1,'mlx::core::detail::Log1p'],['../classmlx_1_1core_1_1_log1p.html',1,'mlx::core::Log1p'],['../classmlx_1_1core_1_1_log1p.html#ab0d6eb90c6f98775fce56f3446ff127a',1,'mlx::core::Log1p::Log1p()']]], - ['log1p_67',['log1p',['../backend_2metal_2kernels_2utils_8h.html#a27c03f2f90ab56db2e4d59559a3d2e9a',1,'log1p(float x): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a3501b665c8837eabf9789ea27a7d6946',1,'log1p(bfloat16_t x): utils.h'],['../group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604',1,'mlx::core::log1p()']]], + ['log1p_67',['log1p',['../namespacemlx_1_1core_1_1simd.html#ad06680bbc041e76efe2dbff4e11b9a13',1,'mlx::core::simd::log1p(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a830591eb3007fef5d87dc296f5615108',1,'mlx::core::simd::log1p(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a92fcc8037ddb767bff517814ab55c259',1,'mlx::core::simd::log1p(Simd< T, 1 > in)'],['../backend_2metal_2kernels_2utils_8h.html#a27c03f2f90ab56db2e4d59559a3d2e9a',1,'log1p(float x): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a3501b665c8837eabf9789ea27a7d6946',1,'log1p(bfloat16_t x): utils.h'],['../group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604',1,'mlx::core::log1p()']]], ['log2_68',['Log2',['../struct_log2.html',1,'Log2'],['../structmlx_1_1core_1_1detail_1_1_log2.html',1,'mlx::core::detail::Log2']]], - ['log2_69',['log2',['../namespacemetal.html#ae894dd5fc13799f120b55cab6267c89c',1,'metal::log2()'],['../namespacemetal_1_1fast.html#a986ef245dd433ae62af864f5cbb07118',1,'metal::fast::log2()'],['../namespacemetal_1_1precise.html#a632dbbdcc1a465cf4739a14306147573',1,'metal::precise::log2()'],['../group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b',1,'mlx::core::log2()']]], + ['log2_69',['log2',['../namespacemlx_1_1core_1_1simd.html#a66426c28a4324b9f617b7018d9354ea1',1,'mlx::core::simd::log2(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#acb1c49b90d029bc4a7eed257ec52791d',1,'mlx::core::simd::log2(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#af9eafa15692dec783860ddae3dd8c072',1,'mlx::core::simd::log2(Simd< T, 1 > in)'],['../namespacemetal.html#ae894dd5fc13799f120b55cab6267c89c',1,'metal::log2()'],['../namespacemetal_1_1fast.html#a986ef245dd433ae62af864f5cbb07118',1,'metal::fast::log2()'],['../namespacemetal_1_1precise.html#a632dbbdcc1a465cf4739a14306147573',1,'metal::precise::log2()'],['../group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b',1,'mlx::core::log2()']]], ['logaddexp_70',['LogAddExp',['../struct_log_add_exp.html',1,'LogAddExp'],['../structmlx_1_1core_1_1detail_1_1_log_add_exp.html',1,'mlx::core::detail::LogAddExp'],['../classmlx_1_1core_1_1_log_add_exp.html',1,'mlx::core::LogAddExp'],['../classmlx_1_1core_1_1_log_add_exp.html#ad8938ca90ccf1a3259973fc68902975a',1,'mlx::core::LogAddExp::LogAddExp()']]], ['logaddexp_71',['logaddexp',['../group__ops.html#gaf985df6609c6bd75a14a844655d89eaa',1,'mlx::core']]], ['logical_5fand_72',['logical_and',['../group__ops.html#ga768977cda8d68cf23f464a6af9907876',1,'mlx::core']]], @@ -80,8 +80,8 @@ var searchData= ['logicalor_77',['LogicalOr',['../struct_logical_or.html',1,'LogicalOr'],['../structmlx_1_1core_1_1detail_1_1_logical_or.html',1,'mlx::core::detail::LogicalOr'],['../classmlx_1_1core_1_1_logical_or.html',1,'mlx::core::LogicalOr'],['../classmlx_1_1core_1_1_logical_or.html#a269c22daca1c15ad010bb860bce93918',1,'mlx::core::LogicalOr::LogicalOr()']]], ['logsumexp_78',['logsumexp',['../group__ops.html#gacff4eb57c085d571e722083680267ac5',1,'mlx::core::logsumexp(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga59be50b4e92f1dc20b53460cefa3910d',1,'mlx::core::logsumexp(const array &a, StreamOrDevice s={})'],['../group__ops.html#gae3969c7bd24c4f3ab97831df28239689',1,'mlx::core::logsumexp(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafef5cb2159c16a60a95470cc823bdd44',1,'mlx::core::logsumexp(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], ['loopalignment_79',['LoopAlignment',['../structmlx_1_1steel_1_1_loop_alignment.html',1,'mlx::steel']]], - ['loopedelemtoloc_80',['LoopedElemToLoc',['../struct_looped_elem_to_loc.html',1,'LoopedElemToLoc< DIM, OffsetT, General >'],['../struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b',1,'LoopedElemToLoc::LoopedElemToLoc()'],['../struct_looped_elem_to_loc.html#abf536c7162d36af7367e390789944c86',1,'LoopedElemToLoc< 1, OffsetT, true >::LoopedElemToLoc()'],['../struct_looped_elem_to_loc.html#a0e21977d9f23b6994773e8e4f3ee70de',1,'LoopedElemToLoc< 1, OffsetT, false >::LoopedElemToLoc()']]], - ['loopedelemtoloc_3c_201_2c_20offsett_2c_20false_20_3e_81',['LoopedElemToLoc< 1, OffsetT, false >',['../struct_looped_elem_to_loc.html',1,'']]], - ['loopedelemtoloc_3c_201_2c_20offsett_2c_20true_20_3e_82',['LoopedElemToLoc< 1, OffsetT, true >',['../struct_looped_elem_to_loc.html',1,'']]], - ['lowest_83',['lowest',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ae81c58b8223e504965183c99d19a2116',1,'metal::_numeric_limits_impl< bfloat16_t >::lowest()'],['../structmlx_1_1core_1_1numeric__limits.html#abc2d9cd8d0a90219f7eb6fd05b98e4ac',1,'mlx::core::numeric_limits< float16_t >::lowest()'],['../structmlx_1_1core_1_1numeric__limits.html#a3623580fbfd92ceb69cdd8e329e18fa8',1,'mlx::core::numeric_limits< bfloat16_t >::lowest()']]] + ['loopedelemtoloc_80',['LoopedElemToLoc',['../struct_looped_elem_to_loc.html',1,'LoopedElemToLoc< DIM, OffsetT, General >'],['../struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b',1,'LoopedElemToLoc::LoopedElemToLoc()'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#abf536c7162d36af7367e390789944c86',1,'LoopedElemToLoc< 1, OffsetT, true >::LoopedElemToLoc()'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a0e21977d9f23b6994773e8e4f3ee70de',1,'LoopedElemToLoc< 1, OffsetT, false >::LoopedElemToLoc(int)'],['../struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b',1,'LoopedElemToLoc< 1, OffsetT, false >::LoopedElemToLoc(int dim)'],['../struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b',1,'LoopedElemToLoc< 1, OffsetT, true >::LoopedElemToLoc()']]], + ['loopedelemtoloc_3c_201_2c_20offsett_2c_20false_20_3e_81',['LoopedElemToLoc< 1, OffsetT, false >',['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html',1,'']]], + ['loopedelemtoloc_3c_201_2c_20offsett_2c_20true_20_3e_82',['LoopedElemToLoc< 1, OffsetT, true >',['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html',1,'']]], + ['lowest_83',['lowest',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ae81c58b8223e504965183c99d19a2116',1,'metal::_numeric_limits_impl< bfloat16_t >::lowest()'],['../structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html#abc2d9cd8d0a90219f7eb6fd05b98e4ac',1,'mlx::core::numeric_limits< float16_t >::lowest()'],['../structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html#a3623580fbfd92ceb69cdd8e329e18fa8',1,'mlx::core::numeric_limits< bfloat16_t >::lowest()']]] ]; diff --git a/docs/build/html/search/all_d.js b/docs/build/html/search/all_d.js index 62e531a4e..82dc3ae91 100644 --- a/docs/build/html/search/all_d.js +++ b/docs/build/html/search/all_d.js @@ -13,106 +13,123 @@ var searchData= ['mask_5ft_10',['mask_t',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a270ab3da7c98a12525a59952742cc97d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], ['mask_5fw_11',['mask_w',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a19ddba7259c3c2c02ed90f3f635557be',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], ['mat_5fat_12',['mat_at',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a323a4f38cd0693bf333832bb4258b28e',1,'mlx::steel::MMATile::mat_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a323a4f38cd0693bf333832bb4258b28e',1,'mlx::steel::MMATile::mat_at(const short i, const short j)']]], - ['mat_5ftype_13',['mat_type',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1eeb197c9bdf4db42892a39cdb9bd73a',1,'mlx::steel::MMATile::mat_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1eeb197c9bdf4db42892a39cdb9bd73a',1,'mlx::steel::MMATile::mat_type']]], - ['matmul_14',['Matmul',['../classmlx_1_1core_1_1_matmul.html',1,'mlx::core::Matmul'],['../classmlx_1_1core_1_1_matmul.html#adef92f30ab35e540ccb316ea6b94e6f7',1,'mlx::core::Matmul::Matmul()']]], - ['matmul_15',['matmul',['../group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a',1,'mlx::core']]], - ['matmul_2eh_16',['matmul.h',['../matmul_8h.html',1,'']]], - ['max_17',['Max',['../struct_max.html',1,'Max< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924ac00cf69bbba24f7ab08d3ad618705988',1,'mlx::core::distributed::AllReduce::Max'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a3d11c500ea4f7f639e20dd0755d39260',1,'mlx::core::Reduce::Max'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ad54b2905015a390708f79bae6cdac56d',1,'mlx::core::Scan::Max'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca1c2da7b96d743296fe660f5fc4072f16',1,'mlx::core::Scatter::Max']]], - ['max_18',['max',['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits::max'],['../structmlx_1_1core_1_1finfo.html#a9975908d99d25e9225d0c70bead0abc2',1,'mlx::core::finfo::max'],['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a92320d40a58218e40cc414986ac95c50',1,'metal::_numeric_limits_impl< bfloat16_t >::max()'],['../structmlx_1_1core_1_1numeric__limits.html#a6dd1fadd4cc7c2cec6223977c238c334',1,'mlx::core::numeric_limits< float16_t >::max()'],['../structmlx_1_1core_1_1numeric__limits.html#a01712fcb04266320225c168a0e6f619a',1,'mlx::core::numeric_limits< bfloat16_t >::max()'],['../namespacemetal.html#a853c80479ab2264d9c4587c7bcac767b',1,'metal::max()'],['../namespacemetal_1_1fast.html#a747e2e58092a27fb8b4dd3d16934fb52',1,'metal::fast::max()'],['../namespacemetal_1_1precise.html#a6a954a4e4e3753303d1dc734855a185f',1,'metal::precise::max()'],['../group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7',1,'mlx::core::max(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc',1,'mlx::core::max(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f',1,'mlx::core::max(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga7b638050e03a93f2896c981bc2850a47',1,'mlx::core::max(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['max3_19',['max3',['../namespacemetal.html#a00f9c0ad66d969794614f56912eed9c9',1,'metal::max3()'],['../namespacemetal_1_1fast.html#a6fc2cf18ffa8149561864c86dba0f803',1,'metal::fast::max3()'],['../namespacemetal_1_1precise.html#ac490e8614ebd2c9343af1ae6c0d4e82c',1,'metal::precise::max3()']]], - ['max_5fdigits10_20',['max_digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a8d3905e6f158379a0c52682266e8d0e2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['max_5fexponent_21',['max_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61bb136f819fa392c50bdf3c38f3aad2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['max_5fexponent10_22',['max_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a76bfb2deb0e0afc011f77bf5a6d0ed94',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['max_5fops_5fper_5fbuffer_23',['max_ops_per_buffer',['../namespacemlx_1_1core_1_1env.html#aedbf4e739553024c33dd0094dd9107aa',1,'mlx::core::env']]], - ['max_5foutput_5fsize_24',['MAX_OUTPUT_SIZE',['../backend_2metal_2kernels_2fft_8h.html#a28d683cf067736d76f867f30c066317e',1,'fft.h']]], - ['max_5fradix_25',['MAX_RADIX',['../backend_2metal_2kernels_2fft_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: fft.h'],['../readwrite_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: readwrite.h']]], - ['max_5freduce_5fspecialized_5fdims_26',['MAX_REDUCE_SPECIALIZED_DIMS',['../defines_8h.html#a15629f1b81a2b6f1cca26d07a2734623',1,'defines.h']]], - ['max_5fthreads_27',['max_threads',['../namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70',1,'pocketfft::detail::threading']]], - ['maximum_28',['Maximum',['../struct_maximum.html',1,'Maximum'],['../structmlx_1_1core_1_1detail_1_1_maximum.html',1,'mlx::core::detail::Maximum'],['../classmlx_1_1core_1_1_maximum.html',1,'mlx::core::Maximum'],['../classmlx_1_1core_1_1_maximum.html#a28389307e385efe1b2955b86b115e816',1,'mlx::core::Maximum::Maximum()']]], - ['maximum_29',['maximum',['../group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d',1,'mlx::core']]], - ['maxop_30',['MaxOp',['../struct_max_op.html',1,'']]], - ['maybeinsertbarrier_31',['maybeInsertBarrier',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ad538ae88f90560063f9ba502e2795991',1,'mlx::core::metal::CommandEncoder::maybeInsertBarrier()'],['../structmlx_1_1core_1_1_command_encoder.html#ad538ae88f90560063f9ba502e2795991',1,'mlx::core::CommandEncoder::maybeInsertBarrier()']]], - ['mb_5fblock_5fmerge_32',['mb_block_merge',['../sort_8h.html#ab381cd57f344bc7304ab580bfdc78807',1,'sort.h']]], - ['mb_5fblock_5fpartition_33',['mb_block_partition',['../sort_8h.html#a32cbe4163b8b0f5cb2c97b256119a4b2',1,'sort.h']]], - ['mb_5fblock_5fsort_34',['mb_block_sort',['../sort_8h.html#a7b5b1180af2b8c6ab814eadd701ce4ba',1,'sort.h']]], - ['mean_35',['mean',['../group__ops.html#gade46e768fd46b8b640eb16f26abeecef',1,'mlx::core::mean(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga52b59fdd8e8430538e564f5bbcfa31e6',1,'mlx::core::mean(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga066161f3d3e395a1d76c638cb680d444',1,'mlx::core::mean(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga45fba73eab0e3b6e128ed3ce2f43a5da',1,'mlx::core::mean(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['median3_36',['median3',['../namespacemetal.html#aa3ff49457ce3c93fc1c0897fd1525157',1,'metal::median3()'],['../namespacemetal_1_1fast.html#a742b55f1e4369921ee7f60d70185bfbc',1,'metal::fast::median3()'],['../namespacemetal_1_1precise.html#a14555ff99c4388493fec48e070144ae2',1,'metal::precise::median3()']]], - ['merge_5fpartition_37',['merge_partition',['../struct_block_merge_sort.html#ab2300cbecb23f3433bad888924c831ca',1,'BlockMergeSort::merge_partition()'],['../struct_kernel_multi_block_merge_sort.html#ab15895b4233aba0e279cc44a07a201fe',1,'KernelMultiBlockMergeSort::merge_partition()']]], - ['merge_5fstep_38',['merge_step',['../struct_block_merge_sort.html#ab65f190edf1851b37c39ad49ce99a43c',1,'BlockMergeSort']]], - ['meshgrid_39',['meshgrid',['../group__ops.html#ga577c911618575314de63d1060656a26e',1,'mlx::core']]], - ['metal_40',['metal',['../namespacemetal.html',1,'']]], - ['metal_2eh_41',['metal.h',['../metal_8h.html',1,'']]], - ['metal_3a_3afast_42',['fast',['../namespacemetal_1_1fast.html',1,'metal']]], - ['metal_3a_3aprecise_43',['precise',['../namespacemetal_1_1precise.html',1,'metal']]], - ['metal_5fimpl_2eh_44',['metal_impl.h',['../metal__impl_8h.html',1,'']]], - ['metal_5fkernel_45',['metal_kernel',['../namespacemlx_1_1core_1_1fast.html#ab16436b465dc10ce472193d541d8426e',1,'mlx::core::fast']]], - ['metalallocator_46',['MetalAllocator',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html',1,'mlx::core::metal']]], - ['metalkernelfunction_47',['MetalKernelFunction',['../namespacemlx_1_1core_1_1fast.html#aa45bf61e7a5c4ad0114b82ed80ae0dbd',1,'mlx::core::fast']]], - ['min_48',['Min',['../struct_min.html',1,'Min< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a4f685dcd48e6614d6bb2ccda4f2686ef',1,'mlx::core::distributed::AllReduce::Min'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a0d3d1f5c94725bdc42fa692e2c074418',1,'mlx::core::Reduce::Min'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1a7d2ee8f14f2e70a9d47170fecc6da898',1,'mlx::core::Scan::Min'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613cad914e4c3475ce9858f2de4bf35dcfdbf',1,'mlx::core::Scatter::Min']]], - ['min_49',['min',['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits::min'],['../structmlx_1_1core_1_1finfo.html#a91332dcbb502799b6cc8e1c6933c042b',1,'mlx::core::finfo::min'],['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#adaed80031f5ca0ff69d30ec4c5d0c98f',1,'metal::_numeric_limits_impl< bfloat16_t >::min()'],['../namespacemetal.html#a6653b28c9473087141eddce39878d4d3',1,'metal::min()'],['../namespacemetal_1_1fast.html#a3e958e56a4712687c381a0b64d123e61',1,'metal::fast::min()'],['../namespacemetal_1_1precise.html#afed0da2f7df3505b5dffa2389c3cb36e',1,'metal::precise::min()'],['../group__ops.html#gab27599802617a4c8f9964ab5f4ffee12',1,'mlx::core::min(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga0140b91e9cdfc3fef0da8e332f65a9e8',1,'mlx::core::min(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga6efb83cd46436678c8f8c4af15cc00f5',1,'mlx::core::min(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga36fa315eef677f4143868f552cd26d03',1,'mlx::core::min(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['min3_50',['min3',['../namespacemetal.html#a005510c8c0f964ce2b8aad3ba76a7a3f',1,'metal::min3()'],['../namespacemetal_1_1fast.html#a606a4c1b34ce05ea89ca5af81724036f',1,'metal::fast::min3()'],['../namespacemetal_1_1precise.html#a4d37ce31c3549ca4772a4ee29798e231',1,'metal::precise::min3()']]], - ['min_5fexponent_51',['min_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a13829f8c7a7c0efdc8946eff5d3c9470',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['min_5fexponent10_52',['min_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#aeaed172780720e06b8731cef3177e277',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['minimum_53',['Minimum',['../struct_minimum.html',1,'Minimum'],['../structmlx_1_1core_1_1detail_1_1_minimum.html',1,'mlx::core::detail::Minimum'],['../classmlx_1_1core_1_1_minimum.html',1,'mlx::core::Minimum'],['../classmlx_1_1core_1_1_minimum.html#ab0f2ce17108df44b82cff68886b0f6f5',1,'mlx::core::Minimum::Minimum()']]], - ['minimum_54',['minimum',['../group__ops.html#ga49ba00c090f81f331c91b0c97040bce0',1,'mlx::core']]], - ['mlx_55',['mlx',['../namespacemlx.html',1,'']]], - ['mlx_2eh_56',['mlx.h',['../mlx_8h.html',1,'']]], - ['mlx_3a_3acore_57',['core',['../namespacemlx_1_1core.html',1,'mlx']]], - ['mlx_3a_3acore_3a_3aallocator_58',['allocator',['../namespacemlx_1_1core_1_1allocator.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3adetail_59',['detail',['../namespacemlx_1_1core_1_1detail.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3adistributed_60',['distributed',['../namespacemlx_1_1core_1_1distributed.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3adistributed_3a_3adetail_61',['detail',['../namespacemlx_1_1core_1_1distributed_1_1detail.html',1,'mlx::core::distributed']]], - ['mlx_3a_3acore_3a_3adistributed_3a_3ampi_62',['mpi',['../namespacemlx_1_1core_1_1distributed_1_1mpi.html',1,'mlx::core::distributed']]], - ['mlx_3a_3acore_3a_3aenv_63',['env',['../namespacemlx_1_1core_1_1env.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3afast_64',['fast',['../namespacemlx_1_1core_1_1fast.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3afft_65',['fft',['../namespacemlx_1_1core_1_1fft.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3aio_66',['io',['../namespacemlx_1_1core_1_1io.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3alinalg_67',['linalg',['../namespacemlx_1_1core_1_1linalg.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3ametal_68',['metal',['../namespacemlx_1_1core_1_1metal.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3arandom_69',['random',['../namespacemlx_1_1core_1_1random.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3ascheduler_70',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html',1,'mlx::core']]], - ['mlx_3a_3asteel_71',['steel',['../namespacemlx_1_1steel.html',1,'mlx']]], - ['mlx_5fatomic_72',['mlx_atomic',['../structmlx__atomic.html',1,'']]], - ['mlx_5fatomic_3c_20t_2c_20enable_5fif_5ft_3c_20is_5fmetal_5fatomic_3c_20t_20_3e_20_3e_20_3e_73',['mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >',['../structmlx__atomic.html',1,'']]], - ['mlx_5fatomic_5fcompare_5fexchange_5fweak_5fexplicit_74',['mlx_atomic_compare_exchange_weak_explicit',['../atomic_8h.html#ad7f32327ff66354cfa2f0cfdac79316f',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread T *expected, T val, size_t offset): atomic.h'],['../atomic_8h.html#aa8f47b2e9b95d4b00ad51f08b070deb5',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread uint *expected, uint val, size_t offset): atomic.h']]], - ['mlx_5fatomic_5ffetch_5fadd_5fexplicit_75',['mlx_atomic_fetch_add_explicit',['../atomic_8h.html#aad448d9e06e001700b65ca8317216a3b',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fand_5fexplicit_76',['mlx_atomic_fetch_and_explicit',['../atomic_8h.html#a253e3c870c0ddc7c28ab2f6ca2c3eae5',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_77',['mlx_atomic_fetch_max_explicit',['../atomic_8h.html#ac480f2b459a8ad9095cee353e152d00c',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_3c_20float_20_3e_78',['mlx_atomic_fetch_max_explicit< float >',['../atomic_8h.html#a1dce2abfa16417122c4d2bf261129ae4',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_79',['mlx_atomic_fetch_min_explicit',['../atomic_8h.html#a2ec33dca0039bd944d73d1c2b378cc19',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_3c_20float_20_3e_80',['mlx_atomic_fetch_min_explicit< float >',['../atomic_8h.html#ab7d1dc49f319f239b7ee0b7c72976dd0',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmul_5fexplicit_81',['mlx_atomic_fetch_mul_explicit',['../atomic_8h.html#adfdbea60436f14f1af9ce36e2a0a77a3',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5for_5fexplicit_82',['mlx_atomic_fetch_or_explicit',['../atomic_8h.html#ab7391f197001471e4788312bdb6ab37a',1,'atomic.h']]], - ['mlx_5fatomic_5fload_5fexplicit_83',['mlx_atomic_load_explicit',['../atomic_8h.html#a253a4e8c2c5768a069e2791b627dfc99',1,'atomic.h']]], - ['mlx_5fatomic_5fstore_5fexplicit_84',['mlx_atomic_store_explicit',['../atomic_8h.html#a0ae453140b0819a4c02f265334de98c0',1,'atomic.h']]], - ['mlx_5flapack_5ffunc_85',['MLX_LAPACK_FUNC',['../lapack_8h.html#ae22db9704827bf013a0a61f21a47464b',1,'lapack.h']]], - ['mlx_5fmtl_5fconst_86',['MLX_MTL_CONST',['../kernels_2gemv__masked_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: gemv_masked.h'],['../quantized_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: quantized.h'],['../sort_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: sort.h']]], - ['mlx_5fmtl_5floop_5funroll_87',['MLX_MTL_LOOP_UNROLL',['../sort_8h.html#ad34b622323cebef136669fedd7229515',1,'sort.h']]], - ['mlx_5fmtl_5fpragma_5funroll_88',['MLX_MTL_PRAGMA_UNROLL',['../kernels_2gemv__masked_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: gemv_masked.h'],['../backend_2metal_2kernels_2utils_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: utils.h']]], - ['mlxconvparams_89',['MLXConvParams',['../struct_m_l_x_conv_params.html',1,'']]], - ['mlxconvparams_3c_202_20_3e_90',['MLXConvParams< 2 >',['../struct_m_l_x_conv_params.html',1,'']]], - ['mma_91',['mma',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a8028512f5a3d2b6acaf966be529627a3',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a1868f57d57c8adedab2c58492ec76946',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA::mma()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a8028512f5a3d2b6acaf966be529627a3',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a1868f57d57c8adedab2c58492ec76946',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA::mma()']]], - ['mma_2eh_92',['mma.h',['../attn_2mma_8h.html',1,'(Global Namespace)'],['../gemm_2mma_8h.html',1,'(Global Namespace)']]], - ['mma_5ft_93',['mma_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel::mma_t'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel::mma_t']]], - ['mmafrag_5facc_5ft_94',['MMAFrag_acc_t',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae2c42cb6d0dde785859164c195f4d13c',1,'mlx::steel::BlockMMA::MMAFrag_acc_t'],['../structmlx_1_1steel_1_1_block_m_m_a.html#ae2c42cb6d0dde785859164c195f4d13c',1,'mlx::steel::BlockMMA::MMAFrag_acc_t']]], - ['mmafrag_5ft_95',['MMAFrag_t',['../structmlx_1_1steel_1_1_m_m_a_tile.html#abe33de70e34300745bad9aa822fd0382',1,'mlx::steel::MMATile::MMAFrag_t'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#abe33de70e34300745bad9aa822fd0382',1,'mlx::steel::MMATile::MMAFrag_t']]], - ['mmatile_96',['MMATile',['../structmlx_1_1steel_1_1_m_m_a_tile.html',1,'mlx::steel::MMATile< T, kTileRows_, kTileCols_, MMAFrag_ >'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3fb310dd08ec23c334511f7b316d1b6',1,'mlx::steel::MMATile::MMATile() thread'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3fb310dd08ec23c334511f7b316d1b6',1,'mlx::steel::MMATile::MMATile() thread']]], - ['mmatile_3c_20float_2c_201_2c_20tn_2c_20mlx_3a_3asteel_3a_3abasemmafrag_3c_20float_2c_20kfragsize_2c_20kfragsize_20_3e_20_3e_97',['MMATile< float, 1, TN, mlx::steel::BaseMMAFrag< float, kFragSize, kFragSize > >',['../structmlx_1_1steel_1_1_m_m_a_tile.html',1,'mlx::steel']]], - ['mmatile_3c_20float_2c_20tm_2c_201_2c_20mlx_3a_3asteel_3a_3abasemmafrag_3c_20float_2c_20kfragsize_2c_20kfragsize_20_3e_20_3e_98',['MMATile< float, TM, 1, mlx::steel::BaseMMAFrag< float, kFragSize, kFragSize > >',['../structmlx_1_1steel_1_1_m_m_a_tile.html',1,'mlx::steel']]], - ['mmatile_3c_20float_2c_20tm_2c_20tn_2c_20mlx_3a_3asteel_3a_3abasemmafrag_3c_20float_2c_20kfragsize_2c_20kfragsize_20_3e_20_3e_99',['MMATile< float, TM, TN, mlx::steel::BaseMMAFrag< float, kFragSize, kFragSize > >',['../structmlx_1_1steel_1_1_m_m_a_tile.html',1,'mlx::steel']]], - ['move_5for_5fcopy_100',['move_or_copy',['../namespacemlx_1_1core.html#a830a47d8a317dffb0c88e5a7afe6aee2',1,'mlx::core::move_or_copy(const array &in, array &out)'],['../namespacemlx_1_1core.html#a9fcb3711b150cb65c7778a35c51284b2',1,'mlx::core::move_or_copy(const array &in, array &out, const Strides &strides, array::Flags flags, size_t data_size, size_t offset=0)']]], - ['move_5fshared_5fbuffer_101',['move_shared_buffer',['../classmlx_1_1core_1_1array.html#ad41cc5e7aebfcad849ad15d697584cf8',1,'mlx::core::array::move_shared_buffer(array other, const Strides &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a38d7ad605f8282e5e49d0c09e0555c78',1,'mlx::core::array::move_shared_buffer(array other)']]], - ['moveaxis_102',['moveaxis',['../group__ops.html#ga24067d10a842db2c9d509ea48135a2c3',1,'mlx::core']]], - ['mpi_2eh_103',['mpi.h',['../mpi_8h.html',1,'']]], - ['mpinplace_104',['MPINPLACE',['../namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2',1,'pocketfft::detail']]], - ['mtl_5fconst_105',['MTL_CONST',['../defines_8h.html#a767ed9f2604de22b259cee02c4ce1d22',1,'defines.h']]], - ['mtl_5fdevice_106',['mtl_device',['../classmlx_1_1core_1_1metal_1_1_device.html#a31dba377f2be44a746db10d1b9367653',1,'mlx::core::metal::Device']]], - ['mtl_5fresidency_5fset_107',['mtl_residency_set',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#ac4bfe5ef5e2eaebc458a1ed1953d15e9',1,'mlx::core::metal::ResidencySet']]], - ['mtlfclist_108',['MTLFCList',['../namespacemlx_1_1core_1_1metal.html#a616e09a1ef321d527770721cef264c54',1,'mlx::core::metal']]], - ['mtx_109',['mtx',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a70410c9e612f871663929f1e8441a976',1,'mlx::core::scheduler::StreamThread']]], - ['mulop_110',['MulOp',['../struct_mul_op.html',1,'']]], - ['multi_5fiter_111',['multi_iter',['../classpocketfft_1_1detail_1_1multi__iter.html',1,'pocketfft::detail::multi_iter< N >'],['../classpocketfft_1_1detail_1_1multi__iter.html#a9be43bb18840202da6d17988fccc64b9',1,'pocketfft::detail::multi_iter::multi_iter()']]], - ['multiply_112',['Multiply',['../structmlx_1_1core_1_1detail_1_1_multiply.html',1,'mlx::core::detail::Multiply'],['../classmlx_1_1core_1_1_multiply.html',1,'mlx::core::Multiply'],['../struct_multiply.html',1,'Multiply'],['../classmlx_1_1core_1_1_multiply.html#aca5c50f900321f3eb4d6fbcbc225c00c',1,'mlx::core::Multiply::Multiply()']]], - ['multiply_113',['multiply',['../group__ops.html#gaf57392e641640b5d06e4c99518391c38',1,'mlx::core']]], - ['multivariate_5fnormal_114',['multivariate_normal',['../namespacemlx_1_1core_1_1random.html#ae6a8407fbca0817a4b8c94e02952f77d',1,'mlx::core::random']]] + ['mat_5ftype_13',['mat_type',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a25675ae18947a97c6e04157b540103a9',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mat_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1eeb197c9bdf4db42892a39cdb9bd73a',1,'mlx::steel::MMATile::mat_type'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a25675ae18947a97c6e04157b540103a9',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mat_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1eeb197c9bdf4db42892a39cdb9bd73a',1,'mlx::steel::MMATile::mat_type']]], + ['math_2eh_14',['math.h',['../math_8h.html',1,'']]], + ['matmul_15',['Matmul',['../classmlx_1_1core_1_1_matmul.html',1,'mlx::core::Matmul'],['../classmlx_1_1core_1_1_matmul.html#adef92f30ab35e540ccb316ea6b94e6f7',1,'mlx::core::Matmul::Matmul()']]], + ['matmul_16',['matmul',['../namespacemlx_1_1core.html#aaacf0afe13d77a5c49ce96f1e833eb2d',1,'mlx::core::matmul(const array &a, const array &b, array &out, bool a_transposed, bool b_transposed, size_t lda, size_t ldb, float alpha, float beta)'],['../group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a',1,'mlx::core::matmul(const array &a, const array &b, StreamOrDevice s={})']]], + ['matmul_2eh_17',['matmul.h',['../matmul_8h.html',1,'']]], + ['max_18',['Max',['../struct_max.html',1,'Max< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924ac00cf69bbba24f7ab08d3ad618705988',1,'mlx::core::distributed::AllReduce::Max'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a3d11c500ea4f7f639e20dd0755d39260',1,'mlx::core::Reduce::Max'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ad54b2905015a390708f79bae6cdac56d',1,'mlx::core::Scan::Max'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca1c2da7b96d743296fe660f5fc4072f16',1,'mlx::core::Scatter::Max']]], + ['max_19',['max',['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits::max'],['../struct_limits_3_01uint8__t_01_4.html#a1570fb640e2e41f96776db5ca08d500c',1,'Limits< uint8_t >::max'],['../struct_limits_3_01uint16__t_01_4.html#a228b33556ba4cb7e6137ab6258628488',1,'Limits< uint16_t >::max'],['../struct_limits_3_01uint32__t_01_4.html#a91fa8f7214ec936976a8324c7431c651',1,'Limits< uint32_t >::max'],['../struct_limits_3_01uint64__t_01_4.html#aa8c2257881a4e1fa8596fa07dba5e107',1,'Limits< uint64_t >::max'],['../struct_limits_3_01int8__t_01_4.html#a96fed01fa9249226be69760652643289',1,'Limits< int8_t >::max'],['../struct_limits_3_01int16__t_01_4.html#a12d64c398ca7609b7c906f3cf1a6f678',1,'Limits< int16_t >::max'],['../struct_limits_3_01int32__t_01_4.html#af756344b31e84222dd73d3445dcd5640',1,'Limits< int32_t >::max'],['../struct_limits_3_01int64__t_01_4.html#ac9c420604c0f3d237ddfb2b8a2439224',1,'Limits< int64_t >::max'],['../struct_limits_3_01half_01_4.html#a4f9515dbf2a622074f121bea39a7b175',1,'Limits< half >::max'],['../struct_limits_3_01float_01_4.html#aba172b22b388190aa3969ef16885d8a6',1,'Limits< float >::max'],['../struct_limits_3_01bfloat16__t_01_4.html#a0ead3618da6718629ea9fa4670b5005f',1,'Limits< bfloat16_t >::max'],['../struct_limits_3_01bool_01_4.html#acbd2132145888d51220558a101ffcff4',1,'Limits< bool >::max'],['../struct_limits_3_01complex64__t_01_4.html#ac01c274b224b90f5210b675a484f4607',1,'Limits< complex64_t >::max'],['../structmlx_1_1core_1_1finfo.html#a9975908d99d25e9225d0c70bead0abc2',1,'mlx::core::finfo::max'],['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a92320d40a58218e40cc414986ac95c50',1,'metal::_numeric_limits_impl< bfloat16_t >::max()'],['../structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html#a6dd1fadd4cc7c2cec6223977c238c334',1,'mlx::core::numeric_limits< float16_t >::max()'],['../structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html#a01712fcb04266320225c168a0e6f619a',1,'mlx::core::numeric_limits< bfloat16_t >::max()'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< bfloat16_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< bool >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< complex64_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< float >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< half >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< int16_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< int32_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< int64_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< int8_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< uint16_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< uint32_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< uint64_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< uint8_t >::max'],['../namespacemlx_1_1core_1_1simd.html#a6fcea259041cecfd042d0c4e6afc4b8f',1,'mlx::core::simd::max(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#a312ecd0ae1c38d32147cee71fd8539d7',1,'mlx::core::simd::max(Simd< T, 1 > x)'],['../namespacemlx_1_1core_1_1simd.html#a995da0f1b4ca8077abbbc6f6a6dfd663',1,'mlx::core::simd::max(Simd< float16_t, N > x)'],['../namespacemetal.html#a853c80479ab2264d9c4587c7bcac767b',1,'metal::max()'],['../namespacemetal_1_1fast.html#a747e2e58092a27fb8b4dd3d16934fb52',1,'metal::fast::max()'],['../namespacemetal_1_1precise.html#a6a954a4e4e3753303d1dc734855a185f',1,'metal::precise::max()'],['../group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7',1,'mlx::core::max(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc',1,'mlx::core::max(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f',1,'mlx::core::max(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga7b638050e03a93f2896c981bc2850a47',1,'mlx::core::max(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['max3_20',['max3',['../namespacemetal.html#a00f9c0ad66d969794614f56912eed9c9',1,'metal::max3()'],['../namespacemetal_1_1fast.html#a6fc2cf18ffa8149561864c86dba0f803',1,'metal::fast::max3()'],['../namespacemetal_1_1precise.html#ac490e8614ebd2c9343af1ae6c0d4e82c',1,'metal::precise::max3()']]], + ['max_5fdigits10_21',['max_digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a8d3905e6f158379a0c52682266e8d0e2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['max_5fexponent_22',['max_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61bb136f819fa392c50bdf3c38f3aad2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['max_5fexponent10_23',['max_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a76bfb2deb0e0afc011f77bf5a6d0ed94',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['max_5fops_5fper_5fbuffer_24',['max_ops_per_buffer',['../namespacemlx_1_1core_1_1env.html#aedbf4e739553024c33dd0094dd9107aa',1,'mlx::core::env']]], + ['max_5foutput_5fsize_25',['MAX_OUTPUT_SIZE',['../backend_2metal_2kernels_2fft_8h.html#a28d683cf067736d76f867f30c066317e',1,'fft.h']]], + ['max_5fradix_26',['MAX_RADIX',['../backend_2metal_2kernels_2fft_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: fft.h'],['../readwrite_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: readwrite.h']]], + ['max_5freduce_5fspecialized_5fdims_27',['MAX_REDUCE_SPECIALIZED_DIMS',['../defines_8h.html#a15629f1b81a2b6f1cca26d07a2734623',1,'defines.h']]], + ['max_5fsize_28',['max_size',['../namespacemlx_1_1core_1_1simd.html#ac91bd36c7caafd3c7ff176e7e2f81887',1,'mlx::core::simd']]], + ['max_5fsize_3c_20double_20_3e_29',['max_size< double >',['../namespacemlx_1_1core_1_1simd.html#a3fa3d1f571027c5cdd1dce5d2cd041e3',1,'mlx::core::simd']]], + ['max_5fsize_3c_20float_20_3e_30',['max_size< float >',['../namespacemlx_1_1core_1_1simd.html#ae745e117cacfe455df39aa4569c34c11',1,'mlx::core::simd']]], + ['max_5fsize_3c_20float16_5ft_20_3e_31',['max_size< float16_t >',['../namespacemlx_1_1core_1_1simd.html#a155df1de3c26e1a3725b63e9e97c0b53',1,'mlx::core::simd']]], + ['max_5fsize_3c_20int_20_3e_32',['max_size< int >',['../namespacemlx_1_1core_1_1simd.html#ab25fc96fa6f00d0a8c335b8da293fbbb',1,'mlx::core::simd']]], + ['max_5fsize_3c_20int16_5ft_20_3e_33',['max_size< int16_t >',['../namespacemlx_1_1core_1_1simd.html#a7e63a5eb08898b84fd4000dadc460fd9',1,'mlx::core::simd']]], + ['max_5fsize_3c_20int64_5ft_20_3e_34',['max_size< int64_t >',['../namespacemlx_1_1core_1_1simd.html#a7913cb2854ffc37efcf26635a097f0a9',1,'mlx::core::simd']]], + ['max_5fsize_3c_20int8_5ft_20_3e_35',['max_size< int8_t >',['../namespacemlx_1_1core_1_1simd.html#ac368e4701363cfece4935e57f3c709b1',1,'mlx::core::simd']]], + ['max_5fsize_3c_20uint16_5ft_20_3e_36',['max_size< uint16_t >',['../namespacemlx_1_1core_1_1simd.html#a0cc9ca2925c25d2eb225af9125bd6bc4',1,'mlx::core::simd']]], + ['max_5fsize_3c_20uint32_5ft_20_3e_37',['max_size< uint32_t >',['../namespacemlx_1_1core_1_1simd.html#a06cb29f91deeaec69471058044abd2aa',1,'mlx::core::simd']]], + ['max_5fsize_3c_20uint64_5ft_20_3e_38',['max_size< uint64_t >',['../namespacemlx_1_1core_1_1simd.html#ab367b9b65be2fda4830a56fc9cc0cd2f',1,'mlx::core::simd']]], + ['max_5fsize_3c_20uint8_5ft_20_3e_39',['max_size< uint8_t >',['../namespacemlx_1_1core_1_1simd.html#a8f731e5a287c714dfc92879fe37503d5',1,'mlx::core::simd']]], + ['max_5fthreads_40',['max_threads',['../namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70',1,'pocketfft::detail::threading']]], + ['maximum_41',['Maximum',['../struct_maximum.html',1,'Maximum'],['../structmlx_1_1core_1_1detail_1_1_maximum.html',1,'mlx::core::detail::Maximum'],['../classmlx_1_1core_1_1_maximum.html',1,'mlx::core::Maximum'],['../classmlx_1_1core_1_1_maximum.html#a28389307e385efe1b2955b86b115e816',1,'mlx::core::Maximum::Maximum()']]], + ['maximum_42',['maximum',['../namespacemlx_1_1core_1_1simd.html#a7f7a298284e71ddbd2ba0bb6d98b0d16',1,'mlx::core::simd::maximum(Simd< T, N > a, Simd< T, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ab54ff0f073be504e8428912f8e21effd',1,'mlx::core::simd::maximum(Simd< T, 1 > a_, Simd< T, 1 > b_)'],['../namespacemlx_1_1core_1_1simd.html#ae1f11d9c2c15ebecf001d11b3fca5da2',1,'mlx::core::simd::maximum(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aa78385c9cf0b87aabc377b1b47b2929d',1,'mlx::core::simd::maximum(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a0ff63db5f193a57ef3b1fffa374eb15a',1,'mlx::core::simd::maximum(T a, Simd< float16_t, N > b)'],['../group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d',1,'mlx::core::maximum()']]], + ['maxop_43',['MaxOp',['../struct_max_op.html',1,'']]], + ['maybeinsertbarrier_44',['maybeInsertBarrier',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ad538ae88f90560063f9ba502e2795991',1,'mlx::core::metal::CommandEncoder::maybeInsertBarrier()'],['../structmlx_1_1core_1_1_command_encoder.html#ad538ae88f90560063f9ba502e2795991',1,'mlx::core::CommandEncoder::maybeInsertBarrier()']]], + ['mb_5fblock_5fmerge_45',['mb_block_merge',['../sort_8h.html#a9cd2751d251acde874a95330d35fac5f',1,'sort.h']]], + ['mb_5fblock_5fpartition_46',['mb_block_partition',['../sort_8h.html#a812f19ed1db562026edc24e29185fe8c',1,'sort.h']]], + ['mb_5fblock_5fsort_47',['mb_block_sort',['../sort_8h.html#ad1ebc6ed8452f970c37c8aad5414551f',1,'sort.h']]], + ['mean_48',['mean',['../group__ops.html#gade46e768fd46b8b640eb16f26abeecef',1,'mlx::core::mean(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga52b59fdd8e8430538e564f5bbcfa31e6',1,'mlx::core::mean(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga066161f3d3e395a1d76c638cb680d444',1,'mlx::core::mean(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga45fba73eab0e3b6e128ed3ce2f43a5da',1,'mlx::core::mean(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['median3_49',['median3',['../namespacemetal.html#aa3ff49457ce3c93fc1c0897fd1525157',1,'metal::median3()'],['../namespacemetal_1_1fast.html#a742b55f1e4369921ee7f60d70185bfbc',1,'metal::fast::median3()'],['../namespacemetal_1_1precise.html#a14555ff99c4388493fec48e070144ae2',1,'metal::precise::median3()']]], + ['merge_5fpartition_50',['merge_partition',['../struct_block_merge_sort.html#ad5bd0d853e9b4352ecfd902a706d7178',1,'BlockMergeSort::merge_partition()'],['../struct_kernel_multi_block_merge_sort.html#a811e72376de254af2bf5303133562a9a',1,'KernelMultiBlockMergeSort::merge_partition()']]], + ['merge_5fstep_51',['merge_step',['../struct_block_merge_sort.html#a0386ce33d7bcfd12dbb17558d26da1bb',1,'BlockMergeSort']]], + ['meshgrid_52',['meshgrid',['../group__ops.html#ga5ecddb74ba7861eb82eca8653501d5dc',1,'mlx::core']]], + ['metal_53',['metal',['../namespacemetal.html',1,'']]], + ['metal_2eh_54',['metal.h',['../metal_8h.html',1,'']]], + ['metal_3a_3afast_55',['fast',['../namespacemetal_1_1fast.html',1,'metal']]], + ['metal_3a_3aprecise_56',['precise',['../namespacemetal_1_1precise.html',1,'metal']]], + ['metal_5ffast_5fsynch_57',['metal_fast_synch',['../namespacemlx_1_1core_1_1env.html#afa1ecf087fe0c633d5460ddb2c31c945',1,'mlx::core::env']]], + ['metal_5fimpl_2eh_58',['metal_impl.h',['../metal__impl_8h.html',1,'']]], + ['metal_5fkernel_59',['metal_kernel',['../namespacemlx_1_1core_1_1fast.html#ab16436b465dc10ce472193d541d8426e',1,'mlx::core::fast']]], + ['metalallocator_60',['MetalAllocator',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html',1,'mlx::core::metal']]], + ['metalkernelfunction_61',['MetalKernelFunction',['../namespacemlx_1_1core_1_1fast.html#aa45bf61e7a5c4ad0114b82ed80ae0dbd',1,'mlx::core::fast']]], + ['min_62',['Min',['../struct_min.html',1,'Min< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a4f685dcd48e6614d6bb2ccda4f2686ef',1,'mlx::core::distributed::AllReduce::Min'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a0d3d1f5c94725bdc42fa692e2c074418',1,'mlx::core::Reduce::Min'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1a7d2ee8f14f2e70a9d47170fecc6da898',1,'mlx::core::Scan::Min'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613cad914e4c3475ce9858f2de4bf35dcfdbf',1,'mlx::core::Scatter::Min']]], + ['min_63',['min',['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits::min'],['../struct_limits_3_01uint8__t_01_4.html#a408bd5a337e7292f06e63da81193629a',1,'Limits< uint8_t >::min'],['../struct_limits_3_01uint16__t_01_4.html#ae173984c3be8b6750f27daed581805fe',1,'Limits< uint16_t >::min'],['../struct_limits_3_01uint32__t_01_4.html#ab0c3975e02053b234c7b606ababa66e1',1,'Limits< uint32_t >::min'],['../struct_limits_3_01uint64__t_01_4.html#a80627f39e951398283942cefa48f4dd0',1,'Limits< uint64_t >::min'],['../struct_limits_3_01int8__t_01_4.html#a7a809307d2bba80382f0645d277eaa4b',1,'Limits< int8_t >::min'],['../struct_limits_3_01int16__t_01_4.html#adca7139647801e223c35b0abc7da5240',1,'Limits< int16_t >::min'],['../struct_limits_3_01int32__t_01_4.html#af336a1b22a8ed6a83a4cfb5bf8869771',1,'Limits< int32_t >::min'],['../struct_limits_3_01int64__t_01_4.html#a1c90fb96af515badaccaa835b08f7428',1,'Limits< int64_t >::min'],['../struct_limits_3_01half_01_4.html#aca7b036c257878bf1b80912fb5d4516d',1,'Limits< half >::min'],['../struct_limits_3_01float_01_4.html#a3225e334d372ee86128c89a440d8648f',1,'Limits< float >::min'],['../struct_limits_3_01bfloat16__t_01_4.html#a2fd1811b9f615b2b897904bc27d1cb49',1,'Limits< bfloat16_t >::min'],['../struct_limits_3_01bool_01_4.html#a139f787b57536d455490b8ef801d37cc',1,'Limits< bool >::min'],['../struct_limits_3_01complex64__t_01_4.html#aa67b04aa7abcd67f7af0808737ab8e14',1,'Limits< complex64_t >::min'],['../structmlx_1_1core_1_1finfo.html#a91332dcbb502799b6cc8e1c6933c042b',1,'mlx::core::finfo::min'],['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#adaed80031f5ca0ff69d30ec4c5d0c98f',1,'metal::_numeric_limits_impl< bfloat16_t >::min()'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< bfloat16_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< bool >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< complex64_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< float >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< half >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< int16_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< int32_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< int64_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< int8_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< uint16_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< uint32_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< uint64_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< uint8_t >::min'],['../namespacemlx_1_1core_1_1simd.html#acd4196d0c66204cfae70b064c305e146',1,'mlx::core::simd::min(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#a96db878d780a8da6abad19ac772d08ca',1,'mlx::core::simd::min(Simd< T, 1 > x)'],['../namespacemlx_1_1core_1_1simd.html#a160075943b92d541f2e7f7472eaa5167',1,'mlx::core::simd::min(Simd< float16_t, N > x)'],['../namespacemetal.html#a6653b28c9473087141eddce39878d4d3',1,'metal::min()'],['../namespacemetal_1_1fast.html#a3e958e56a4712687c381a0b64d123e61',1,'metal::fast::min()'],['../namespacemetal_1_1precise.html#afed0da2f7df3505b5dffa2389c3cb36e',1,'metal::precise::min()'],['../group__ops.html#gab27599802617a4c8f9964ab5f4ffee12',1,'mlx::core::min(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga0140b91e9cdfc3fef0da8e332f65a9e8',1,'mlx::core::min(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga6efb83cd46436678c8f8c4af15cc00f5',1,'mlx::core::min(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga36fa315eef677f4143868f552cd26d03',1,'mlx::core::min(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['min3_64',['min3',['../namespacemetal.html#a005510c8c0f964ce2b8aad3ba76a7a3f',1,'metal::min3()'],['../namespacemetal_1_1fast.html#a606a4c1b34ce05ea89ca5af81724036f',1,'metal::fast::min3()'],['../namespacemetal_1_1precise.html#a4d37ce31c3549ca4772a4ee29798e231',1,'metal::precise::min3()']]], + ['min_5fexponent_65',['min_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a13829f8c7a7c0efdc8946eff5d3c9470',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['min_5fexponent10_66',['min_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#aeaed172780720e06b8731cef3177e277',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['minimum_67',['Minimum',['../struct_minimum.html',1,'Minimum'],['../structmlx_1_1core_1_1detail_1_1_minimum.html',1,'mlx::core::detail::Minimum'],['../classmlx_1_1core_1_1_minimum.html',1,'mlx::core::Minimum'],['../classmlx_1_1core_1_1_minimum.html#ab0f2ce17108df44b82cff68886b0f6f5',1,'mlx::core::Minimum::Minimum()']]], + ['minimum_68',['minimum',['../namespacemlx_1_1core_1_1simd.html#a1996e77a8c3c24b1ba706113ed9028c4',1,'mlx::core::simd::minimum(Simd< T, N > a, Simd< T, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ac836568622a3e5957c275e115e2fcaf3',1,'mlx::core::simd::minimum(Simd< T, 1 > a_, Simd< T, 1 > b_)'],['../namespacemlx_1_1core_1_1simd.html#abaa09259e92f0fe758dc979d54c327e8',1,'mlx::core::simd::minimum(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ae9ce2f34c97aba7b99223792a86d5c83',1,'mlx::core::simd::minimum(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a17f7baec6300f2ff96ec53fb1943cb49',1,'mlx::core::simd::minimum(T a, Simd< float16_t, N > b)'],['../group__ops.html#ga49ba00c090f81f331c91b0c97040bce0',1,'mlx::core::minimum()']]], + ['mlx_69',['mlx',['../namespacemlx.html',1,'']]], + ['mlx_2eh_70',['mlx.h',['../mlx_8h.html',1,'']]], + ['mlx_3a_3acore_71',['core',['../namespacemlx_1_1core.html',1,'mlx']]], + ['mlx_3a_3acore_3a_3aallocator_72',['allocator',['../namespacemlx_1_1core_1_1allocator.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3adetail_73',['detail',['../namespacemlx_1_1core_1_1detail.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3adistributed_74',['distributed',['../namespacemlx_1_1core_1_1distributed.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3adistributed_3a_3adetail_75',['detail',['../namespacemlx_1_1core_1_1distributed_1_1detail.html',1,'mlx::core::distributed']]], + ['mlx_3a_3acore_3a_3adistributed_3a_3ampi_76',['mpi',['../namespacemlx_1_1core_1_1distributed_1_1mpi.html',1,'mlx::core::distributed']]], + ['mlx_3a_3acore_3a_3adistributed_3a_3aring_77',['ring',['../namespacemlx_1_1core_1_1distributed_1_1ring.html',1,'mlx::core::distributed']]], + ['mlx_3a_3acore_3a_3aenv_78',['env',['../namespacemlx_1_1core_1_1env.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3afast_79',['fast',['../namespacemlx_1_1core_1_1fast.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3afft_80',['fft',['../namespacemlx_1_1core_1_1fft.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3aio_81',['io',['../namespacemlx_1_1core_1_1io.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3alinalg_82',['linalg',['../namespacemlx_1_1core_1_1linalg.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3ametal_83',['metal',['../namespacemlx_1_1core_1_1metal.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3arandom_84',['random',['../namespacemlx_1_1core_1_1random.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3ascheduler_85',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3asimd_86',['simd',['../namespacemlx_1_1core_1_1simd.html',1,'mlx::core']]], + ['mlx_3a_3asteel_87',['steel',['../namespacemlx_1_1steel.html',1,'mlx']]], + ['mlx_5fatomic_88',['mlx_atomic',['../structmlx__atomic.html',1,'']]], + ['mlx_5fatomic_3c_20t_2c_20enable_5fif_5ft_3c_20is_5fmetal_5fatomic_3c_20t_20_3e_20_3e_20_3e_89',['mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >',['../structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html',1,'']]], + ['mlx_5fatomic_5fcompare_5fexchange_5fweak_5fexplicit_90',['mlx_atomic_compare_exchange_weak_explicit',['../atomic_8h.html#ad7f32327ff66354cfa2f0cfdac79316f',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread T *expected, T val, size_t offset): atomic.h'],['../atomic_8h.html#aa8f47b2e9b95d4b00ad51f08b070deb5',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread uint *expected, uint val, size_t offset): atomic.h']]], + ['mlx_5fatomic_5ffetch_5fadd_5fexplicit_91',['mlx_atomic_fetch_add_explicit',['../atomic_8h.html#aad448d9e06e001700b65ca8317216a3b',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fand_5fexplicit_92',['mlx_atomic_fetch_and_explicit',['../atomic_8h.html#a253e3c870c0ddc7c28ab2f6ca2c3eae5',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_93',['mlx_atomic_fetch_max_explicit',['../atomic_8h.html#ac480f2b459a8ad9095cee353e152d00c',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_3c_20float_20_3e_94',['mlx_atomic_fetch_max_explicit< float >',['../atomic_8h.html#a1dce2abfa16417122c4d2bf261129ae4',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_95',['mlx_atomic_fetch_min_explicit',['../atomic_8h.html#a2ec33dca0039bd944d73d1c2b378cc19',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_3c_20float_20_3e_96',['mlx_atomic_fetch_min_explicit< float >',['../atomic_8h.html#ab7d1dc49f319f239b7ee0b7c72976dd0',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmul_5fexplicit_97',['mlx_atomic_fetch_mul_explicit',['../atomic_8h.html#adfdbea60436f14f1af9ce36e2a0a77a3',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5for_5fexplicit_98',['mlx_atomic_fetch_or_explicit',['../atomic_8h.html#ab7391f197001471e4788312bdb6ab37a',1,'atomic.h']]], + ['mlx_5fatomic_5fload_5fexplicit_99',['mlx_atomic_load_explicit',['../atomic_8h.html#a253a4e8c2c5768a069e2791b627dfc99',1,'atomic.h']]], + ['mlx_5fatomic_5fstore_5fexplicit_100',['mlx_atomic_store_explicit',['../atomic_8h.html#a0ae453140b0819a4c02f265334de98c0',1,'atomic.h']]], + ['mlx_5flapack_5ffunc_101',['MLX_LAPACK_FUNC',['../lapack_8h.html#ae22db9704827bf013a0a61f21a47464b',1,'lapack.h']]], + ['mlx_5fmtl_5fconst_102',['MLX_MTL_CONST',['../gemv__masked_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: gemv_masked.h'],['../quantized_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: quantized.h'],['../sort_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: sort.h']]], + ['mlx_5fmtl_5floop_5funroll_103',['MLX_MTL_LOOP_UNROLL',['../sort_8h.html#ad34b622323cebef136669fedd7229515',1,'sort.h']]], + ['mlx_5fmtl_5fpragma_5funroll_104',['MLX_MTL_PRAGMA_UNROLL',['../gemv__masked_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: gemv_masked.h'],['../backend_2metal_2kernels_2utils_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: utils.h']]], + ['mlx_5fsimd_5flibrary_5fversion_105',['MLX_SIMD_LIBRARY_VERSION',['../accelerate__simd_8h.html#a628420ebe2f37d352c8c4b0dd0db59d0',1,'accelerate_simd.h']]], + ['mlxconvparams_106',['MLXConvParams',['../struct_m_l_x_conv_params.html',1,'']]], + ['mlxconvparams_3c_202_20_3e_107',['MLXConvParams< 2 >',['../struct_m_l_x_conv_params.html',1,'']]], + ['mma_108',['mma',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8028512f5a3d2b6acaf966be529627a3',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1868f57d57c8adedab2c58492ec76946',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA::mma()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8028512f5a3d2b6acaf966be529627a3',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1868f57d57c8adedab2c58492ec76946',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA::mma()']]], + ['mma_2eh_109',['mma.h',['../attn_2mma_8h.html',1,'(Global Namespace)'],['../gemm_2mma_8h.html',1,'(Global Namespace)']]], + ['mma_5ft_110',['mma_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel::mma_t'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel::mma_t']]], + ['mmafrag_5facc_5ft_111',['MMAFrag_acc_t',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae2c42cb6d0dde785859164c195f4d13c',1,'mlx::steel::BlockMMA::MMAFrag_acc_t'],['../structmlx_1_1steel_1_1_block_m_m_a.html#ae2c42cb6d0dde785859164c195f4d13c',1,'mlx::steel::BlockMMA::MMAFrag_acc_t']]], + ['mmafrag_5ft_112',['MMAFrag_t',['../structmlx_1_1steel_1_1_m_m_a_tile.html#abe33de70e34300745bad9aa822fd0382',1,'mlx::steel::MMATile::MMAFrag_t'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#abe33de70e34300745bad9aa822fd0382',1,'mlx::steel::MMATile::MMAFrag_t']]], + ['mmatile_113',['MMATile',['../structmlx_1_1steel_1_1_m_m_a_tile.html',1,'mlx::steel::MMATile< T, kTileRows_, kTileCols_, MMAFrag_ >'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3fb310dd08ec23c334511f7b316d1b6',1,'mlx::steel::MMATile::MMATile() thread'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3fb310dd08ec23c334511f7b316d1b6',1,'mlx::steel::MMATile::MMATile() thread']]], + ['mmatile_3c_20float_2c_201_2c_20tn_2c_20mlx_3a_3asteel_3a_3abasemmafrag_3c_20float_2c_20kfragsize_2c_20kfragsize_20_3e_20_3e_114',['MMATile< float, 1, TN, mlx::steel::BaseMMAFrag< float, kFragSize, kFragSize > >',['../structmlx_1_1steel_1_1_m_m_a_tile.html',1,'mlx::steel']]], + ['mmatile_3c_20float_2c_20tm_2c_201_2c_20mlx_3a_3asteel_3a_3abasemmafrag_3c_20float_2c_20kfragsize_2c_20kfragsize_20_3e_20_3e_115',['MMATile< float, TM, 1, mlx::steel::BaseMMAFrag< float, kFragSize, kFragSize > >',['../structmlx_1_1steel_1_1_m_m_a_tile.html',1,'mlx::steel']]], + ['mmatile_3c_20float_2c_20tm_2c_20tn_2c_20mlx_3a_3asteel_3a_3abasemmafrag_3c_20float_2c_20kfragsize_2c_20kfragsize_20_3e_20_3e_116',['MMATile< float, TM, TN, mlx::steel::BaseMMAFrag< float, kFragSize, kFragSize > >',['../structmlx_1_1steel_1_1_m_m_a_tile.html',1,'mlx::steel']]], + ['move_5for_5fcopy_117',['move_or_copy',['../namespacemlx_1_1core.html#a830a47d8a317dffb0c88e5a7afe6aee2',1,'mlx::core::move_or_copy(const array &in, array &out)'],['../namespacemlx_1_1core.html#a9fcb3711b150cb65c7778a35c51284b2',1,'mlx::core::move_or_copy(const array &in, array &out, const Strides &strides, array::Flags flags, size_t data_size, size_t offset=0)']]], + ['move_5fshared_5fbuffer_118',['move_shared_buffer',['../classmlx_1_1core_1_1array.html#ad41cc5e7aebfcad849ad15d697584cf8',1,'mlx::core::array::move_shared_buffer(array other, const Strides &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a38d7ad605f8282e5e49d0c09e0555c78',1,'mlx::core::array::move_shared_buffer(array other)']]], + ['moveaxis_119',['moveaxis',['../group__ops.html#ga24067d10a842db2c9d509ea48135a2c3',1,'mlx::core']]], + ['mpi_2eh_120',['mpi.h',['../mpi_8h.html',1,'']]], + ['mpinplace_121',['MPINPLACE',['../namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2',1,'pocketfft::detail']]], + ['mtl_5fconst_122',['MTL_CONST',['../defines_8h.html#a767ed9f2604de22b259cee02c4ce1d22',1,'defines.h']]], + ['mtl_5fdevice_123',['mtl_device',['../classmlx_1_1core_1_1metal_1_1_device.html#a31dba377f2be44a746db10d1b9367653',1,'mlx::core::metal::Device']]], + ['mtl_5fresidency_5fset_124',['mtl_residency_set',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#ac4bfe5ef5e2eaebc458a1ed1953d15e9',1,'mlx::core::metal::ResidencySet']]], + ['mtlfclist_125',['MTLFCList',['../namespacemlx_1_1core_1_1metal.html#a616e09a1ef321d527770721cef264c54',1,'mlx::core::metal']]], + ['mtx_126',['mtx',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a70410c9e612f871663929f1e8441a976',1,'mlx::core::scheduler::StreamThread']]], + ['mulop_127',['MulOp',['../struct_mul_op.html',1,'']]], + ['multi_5fiter_128',['multi_iter',['../classpocketfft_1_1detail_1_1multi__iter.html',1,'pocketfft::detail::multi_iter< N >'],['../classpocketfft_1_1detail_1_1multi__iter.html#a9be43bb18840202da6d17988fccc64b9',1,'pocketfft::detail::multi_iter::multi_iter()']]], + ['multiply_129',['Multiply',['../structmlx_1_1core_1_1detail_1_1_multiply.html',1,'mlx::core::detail::Multiply'],['../classmlx_1_1core_1_1_multiply.html',1,'mlx::core::Multiply'],['../struct_multiply.html',1,'Multiply'],['../classmlx_1_1core_1_1_multiply.html#aca5c50f900321f3eb4d6fbcbc225c00c',1,'mlx::core::Multiply::Multiply()']]], + ['multiply_130',['multiply',['../group__ops.html#gaf57392e641640b5d06e4c99518391c38',1,'mlx::core']]], + ['multivariate_5fnormal_131',['multivariate_normal',['../namespacemlx_1_1core_1_1random.html#ae6a8407fbca0817a4b8c94e02952f77d',1,'mlx::core::random']]] ]; diff --git a/docs/build/html/search/all_e.js b/docs/build/html/search/all_e.js index da87b517b..e0cbbed99 100644 --- a/docs/build/html/search/all_e.js +++ b/docs/build/html/search/all_e.js @@ -1,26 +1,26 @@ var searchData= [ - ['n_0',['N',['../struct_m_l_x_conv_params.html#ae6b7054dc3cffa8e6aedeb29fa7da932',1,'MLXConvParams::N'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a213f5ea4018120d8b61ab82754aaba83',1,'mlx::steel::ImplicitGemmConv2DParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a174626ab98515d89923b2841a664b9a1',1,'mlx::steel::GEMMParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a1103e79fb8962812b9a3c9d5c902ff86',1,'mlx::steel::GEMMSpiltKParams::N']]], + ['n_0',['N',['../struct_m_l_x_conv_params.html#ae6b7054dc3cffa8e6aedeb29fa7da932',1,'MLXConvParams::N'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a213f5ea4018120d8b61ab82754aaba83',1,'mlx::steel::ImplicitGemmConv2DParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a174626ab98515d89923b2841a664b9a1',1,'mlx::steel::GEMMParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a1103e79fb8962812b9a3c9d5c902ff86',1,'mlx::steel::GEMMSpiltKParams::N'],['../namespacemlx_1_1core_1_1simd.html#a12b1553495a0c99d52472bd2a6626ddb',1,'mlx::core::simd::N']]], ['n_1',['n',['../struct_read_writer.html#a655346c9ebfc33a69da3f1c1d4238dfb',1,'ReadWriter']]], ['n_5factive_5ftasks_2',['n_active_tasks',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a3c9fa21442974acba3409d49bb033131',1,'mlx::core::scheduler::Scheduler::n_active_tasks()'],['../namespacemlx_1_1core_1_1scheduler.html#a9bf641981df5fc16b0fb0dbacc0c3afd',1,'mlx::core::scheduler::n_active_tasks()']]], - ['n_5fchannels_3',['n_channels',['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper']]], - ['n_5fper_5fblock_4',['N_PER_BLOCK',['../struct_kernel_merge_sort.html#a959aaf5bfb70796a525fed318f7ae8ab',1,'KernelMergeSort::N_PER_BLOCK'],['../struct_kernel_multi_block_merge_sort.html#ae5113ca5852d11999ae932439af95a5c',1,'KernelMultiBlockMergeSort::N_PER_BLOCK']]], + ['n_5fchannels_3',['n_channels',['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#a06c2fb9c93660e8f6916228cd77f9494',1,'mlx::steel::ChannelHelper< 1 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#ac66ff37bc2cf78d96667192a6cca73b5',1,'mlx::steel::ChannelHelper< 2 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#a071c015713b7bab09930661165517eff',1,'mlx::steel::ChannelHelper< 3 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#a167b00a84adf93b60e3d7a943d5eb977',1,'mlx::steel::ChannelHelper< 4 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper< 1 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper< 2 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper< 3 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper< 4 >::n_channels']]], + ['n_5fper_5fblock_4',['N_PER_BLOCK',['../struct_kernel_merge_sort.html#a959aaf5bfb70796a525fed318f7ae8ab',1,'KernelMergeSort::N_PER_BLOCK'],['../struct_kernel_multi_block_merge_sort.html#a4a0c920980c5e56471558d22a727a514',1,'KernelMultiBlockMergeSort::N_PER_BLOCK']]], ['n_5freads_5',['n_reads',['../struct_quantized_block_loader.html#a6213479f7a6d9314d8879f8856b0b6fb',1,'QuantizedBlockLoader']]], ['n_5frows_6',['n_rows',['../structmlx_1_1steel_1_1_block_loader.html#a973804e5b1d418c98c90861cda1a6fb5',1,'mlx::steel::BlockLoader::n_rows'],['../structmlx_1_1steel_1_1_block_loader_t.html#a0ccc7caa93e6e709981a1a08159d41dc',1,'mlx::steel::BlockLoaderT::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a097c48a23e1bd7d8cf3e9d531397602f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a3ec8a92c9e6643c1d5bf8af278026fe8',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a593ec140370d53f8c968f6240116d38b',1,'mlx::steel::Conv2DWeightBlockLoader::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a8b6c0936c9ad2766242664f034d1115f',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae905e56c1129606e93dbbcd7baed8f0f',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#abff29c5d96645d9113314c9a997dd7a8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aaebb6da2cac9961f5edf52d16c18de7d',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::n_rows']]], ['names_7',['names',['../structmlx_1_1core_1_1_node_namer.html#a57823f9a2cdc60b2f06f857b36019277',1,'mlx::core::NodeNamer']]], ['nan_5fto_5fnum_8',['nan_to_num',['../group__ops.html#gab1467c6a9e675152e768afd6dcfb61de',1,'mlx::core']]], ['nanequal_9',['NaNEqual',['../structmlx_1_1core_1_1detail_1_1_na_n_equal.html',1,'mlx::core::detail::NaNEqual'],['../struct_na_n_equal.html',1,'NaNEqual']]], ['nbytes_10',['nbytes',['../classmlx_1_1core_1_1array.html#a387b67cd3ef5cfc1e749c371766c4a05',1,'mlx::core::array']]], - ['nd_5floop_11',['nd_loop',['../namespacemlx_1_1core.html#adbe9ca762e3e6cea8286c4636347c4b5',1,'mlx::core']]], - ['ndarr_12',['ndarr',['../classpocketfft_1_1detail_1_1ndarr.html',1,'pocketfft::detail::ndarr< T >'],['../classpocketfft_1_1detail_1_1ndarr.html#a8f0037a172d96cb1ad915a5069175fa2',1,'pocketfft::detail::ndarr::ndarr()']]], - ['ndim_13',['ndim',['../struct_indices.html#a7dec359e91d0eb2b64e5461b54308313',1,'Indices::ndim'],['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#ae605df33f449872e3da9777d97008051',1,'mlx::core::fast::CustomKernelShapeInfo::ndim'],['../classpocketfft_1_1detail_1_1arr__info.html#ac608c8af2a59a28a0012e308be7ee414',1,'pocketfft::detail::arr_info::ndim()'],['../classmlx_1_1core_1_1array.html#a53006e77d13d9d88b525ef577748939f',1,'mlx::core::array::ndim()']]], - ['needs_5ftgp_5freduction_14',['needs_tgp_reduction',['../struct_g_e_m_v_kernel.html#ae8113fddf6fb637acfd12efd978b704c',1,'GEMVKernel::needs_tgp_reduction'],['../struct_g_e_m_v_t_kernel.html#a67be7ec69c3791f02e97ccdb00ae0e03',1,'GEMVTKernel::needs_tgp_reduction']]], - ['negative_15',['Negative',['../structmlx_1_1core_1_1detail_1_1_negative.html',1,'mlx::core::detail::Negative'],['../classmlx_1_1core_1_1_negative.html',1,'mlx::core::Negative'],['../struct_negative.html',1,'Negative'],['../classmlx_1_1core_1_1_negative.html#aa3b73395d9fa5b7215dca488bc0d3c70',1,'mlx::core::Negative::Negative()']]], - ['negative_16',['negative',['../group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6',1,'mlx::core']]], + ['ndarr_11',['ndarr',['../classpocketfft_1_1detail_1_1ndarr.html',1,'pocketfft::detail::ndarr< T >'],['../classpocketfft_1_1detail_1_1ndarr.html#a8f0037a172d96cb1ad915a5069175fa2',1,'pocketfft::detail::ndarr::ndarr()']]], + ['ndim_12',['ndim',['../struct_indices.html#a7dec359e91d0eb2b64e5461b54308313',1,'Indices::ndim'],['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#ae605df33f449872e3da9777d97008051',1,'mlx::core::fast::CustomKernelShapeInfo::ndim'],['../classpocketfft_1_1detail_1_1arr__info.html#ac608c8af2a59a28a0012e308be7ee414',1,'pocketfft::detail::arr_info::ndim()'],['../classmlx_1_1core_1_1array.html#a53006e77d13d9d88b525ef577748939f',1,'mlx::core::array::ndim()']]], + ['needs_5ftgp_5freduction_13',['needs_tgp_reduction',['../struct_g_e_m_v_kernel.html#ae8113fddf6fb637acfd12efd978b704c',1,'GEMVKernel::needs_tgp_reduction'],['../struct_g_e_m_v_t_kernel.html#a67be7ec69c3791f02e97ccdb00ae0e03',1,'GEMVTKernel::needs_tgp_reduction']]], + ['negative_14',['Negative',['../structmlx_1_1core_1_1detail_1_1_negative.html',1,'mlx::core::detail::Negative'],['../classmlx_1_1core_1_1_negative.html',1,'mlx::core::Negative'],['../struct_negative.html',1,'Negative'],['../classmlx_1_1core_1_1_negative.html#aa3b73395d9fa5b7215dca488bc0d3c70',1,'mlx::core::Negative::Negative()']]], + ['negative_15',['negative',['../group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6',1,'mlx::core']]], + ['neon_5ffp16_5fsimd_2eh_16',['neon_fp16_simd.h',['../neon__fp16__simd_8h.html',1,'']]], ['new_5fqueue_17',['new_queue',['../classmlx_1_1core_1_1metal_1_1_device.html#a8135ae2a8c1e6f3861e84d4e60c28b67',1,'mlx::core::metal::Device']]], ['new_5fscoped_5fmemory_5fpool_18',['new_scoped_memory_pool',['../namespacemlx_1_1core_1_1metal.html#a46583a1aba89449fa72e6cb3a7090981',1,'mlx::core::metal']]], ['new_5fstream_19',['new_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a157c8da85fa1bddb8eacf8515a3cc879',1,'mlx::core::scheduler::Scheduler::new_stream()'],['../namespacemlx_1_1core_1_1metal.html#a8b4188f9a090a1da42d62b8a369bf106',1,'mlx::core::metal::new_stream()'],['../namespacemlx_1_1core.html#a6f7c63a9be10337b3b96d527e1db3c2f',1,'mlx::core::new_stream()']]], - ['next_20',['next',['../struct_quantized_block_loader.html#a674138ef7c43cc45586ea9f8fd6f6bd9',1,'QuantizedBlockLoader::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../structmlx_1_1steel_1_1_block_loader_t.html#a6008ef45ff980dbe1119da0630f6c697',1,'mlx::steel::BlockLoaderT::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a3b71f379ff9baf39830c92f4f1ecde52',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a78d2b0098311a278be8394edbd5fc731',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aae56c19bb562219770fec38e5666c6ce',1,'mlx::steel::Conv2DWeightBlockLoader::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af9ce1a767266664bea131a5437002c80',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a30b10bebde7f08b89d03bdd9ea0f48da',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3e5ee68ed0ee43f7e979dd4222f76a8c',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a11743cb1c108f42ccdc6e59204a5b3e8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531',1,'LoopedElemToLoc::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394',1,'LoopedElemToLoc::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#adc8113de3152d291331f219bfe5970c2',1,'LoopedElemToLoc< 1, OffsetT, true >::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#a70d92070097fe0622b836afc778bf8d2',1,'LoopedElemToLoc< 1, OffsetT, true >::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#a9d85f32e5460707827df3ba74b905c23',1,'LoopedElemToLoc< 1, OffsetT, false >::next(const constant int *, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#a634f4fa8f4b5f1fdd47a26754ef2eba7',1,'LoopedElemToLoc< 1, OffsetT, false >::next(int n, const constant int *, const constant int64_t *strides)'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a4193c5eac3ef093a740d5305b25d3e18',1,'mlx::core::random::KeySequence::next()']]], + ['next_20',['next',['../struct_quantized_block_loader.html#a674138ef7c43cc45586ea9f8fd6f6bd9',1,'QuantizedBlockLoader::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../structmlx_1_1steel_1_1_block_loader_t.html#a6008ef45ff980dbe1119da0630f6c697',1,'mlx::steel::BlockLoaderT::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a3b71f379ff9baf39830c92f4f1ecde52',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a78d2b0098311a278be8394edbd5fc731',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aae56c19bb562219770fec38e5666c6ce',1,'mlx::steel::Conv2DWeightBlockLoader::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af9ce1a767266664bea131a5437002c80',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a30b10bebde7f08b89d03bdd9ea0f48da',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3e5ee68ed0ee43f7e979dd4222f76a8c',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a11743cb1c108f42ccdc6e59204a5b3e8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531',1,'LoopedElemToLoc::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394',1,'LoopedElemToLoc::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#adc8113de3152d291331f219bfe5970c2',1,'LoopedElemToLoc< 1, OffsetT, true >::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a70d92070097fe0622b836afc778bf8d2',1,'LoopedElemToLoc< 1, OffsetT, true >::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a9d85f32e5460707827df3ba74b905c23',1,'LoopedElemToLoc< 1, OffsetT, false >::next(const constant int *, const constant int64_t *strides)'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a634f4fa8f4b5f1fdd47a26754ef2eba7',1,'LoopedElemToLoc< 1, OffsetT, false >::next(int n, const constant int *, const constant int64_t *strides)'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a4193c5eac3ef093a740d5305b25d3e18',1,'mlx::core::random::KeySequence::next()'],['../struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531',1,'LoopedElemToLoc< 1, OffsetT, false >::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394',1,'LoopedElemToLoc< 1, OffsetT, false >::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531',1,'LoopedElemToLoc< 1, OffsetT, true >::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394',1,'LoopedElemToLoc< 1, OffsetT, true >::next(int n, const constant int *shape, const constant int64_t *strides)']]], ['next_5fpower_5fof_5f2_21',['next_power_of_2',['../namespacemlx_1_1core.html#a685c0530e338aabc622325685846ce93',1,'mlx::core']]], ['nextafter_22',['nextafter',['../namespacemetal.html#a9547fd7b09164931986f6db4813bd72d',1,'metal::nextafter()'],['../namespacemetal_1_1fast.html#a4583e8be04fc0bd475b97b0934604f23',1,'metal::fast::nextafter()'],['../namespacemetal_1_1precise.html#ad012ceeb55b77f1533749b351331e026',1,'metal::precise::nextafter()']]], ['nk_23',['NK',['../structmlx_1_1steel_1_1_attn_params.html#a68a66e3fafa922dcfd1ab1f6bdc2375e',1,'mlx::steel::AttnParams']]], @@ -28,8 +28,8 @@ var searchData= ['no_5ffuse_25',['no_fuse',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4ada8df7fd43da6073fec4fe5666b03dbb',1,'mlx::core']]], ['no_5fsimplify_26',['no_simplify',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a8e5611dfddbae6e68624c59aa3e4e3e2',1,'mlx::core']]], ['nodenamer_27',['NodeNamer',['../structmlx_1_1core_1_1_node_namer.html',1,'mlx::core']]], - ['nomask_5ft_28',['nomask_t',['../kernels_2gemv__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'gemv_masked.h']]], - ['none_29',['None',['../struct_none.html',1,'None'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca87a6a1927de175b71d7d0b5c11b8665c',1,'mlx::core::Scatter::None']]], + ['nomask_5ft_28',['nomask_t',['../gemv__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'gemv_masked.h']]], + ['none_29',['None',['../struct_none.html',1,'None'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca87a6a1927de175b71d7d0b5c11b8665c',1,'mlx::core::Scatter::None'],['../classmlx_1_1core_1_1_scatter_axis.html#aa292e6cb2a4b32c42ad4f7a258b334f2a93146c4280504f1f67459e6ae0d25c38',1,'mlx::core::ScatterAxis::None']]], ['norm_30',['norm',['../namespacemlx_1_1core_1_1linalg.html#aba765b8e95e9a1d33d31f727a185919d',1,'mlx::core::linalg::norm(const array &a, const double ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#acaa85b4146821c268abecec2422c02d2',1,'mlx::core::linalg::norm(const array &a, const double ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#af1ebe0c6dcba9a1c49b5e397dddf3264',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#ae8da67e4c6e073f93889f1051203cd9e',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a229018071d5602e38d6248230f334a10',1,'mlx::core::linalg::norm(const array &a, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a44250cff34238f01471fd61e76036f03',1,'mlx::core::linalg::norm(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], ['normal_31',['normal',['../namespacemlx_1_1core_1_1random.html#a6ec2c08e63d379594b0744cfc7d8dc41',1,'mlx::core::random::normal(const Shape &shape, Dtype dtype, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a34c5efe846c68b62e65cdb26803d609c',1,'mlx::core::random::normal(const Shape &shape, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a980b9805dac939e80c2915f0046e6dee',1,'mlx::core::random::normal(const Shape &shape, const Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a1df350be44a32e2eefdd60bb21811246',1,'mlx::core::random::normal(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], ['normalize_5faxis_5findex_32',['normalize_axis_index',['../namespacemlx_1_1core.html#a1cc130b06d9cdd03dddc74a3b1db0167',1,'mlx::core']]], @@ -44,7 +44,7 @@ var searchData= ['number_5fof_5felements_41',['number_of_elements',['../group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636',1,'mlx::core']]], ['numberofelements_42',['NumberOfElements',['../classmlx_1_1core_1_1_number_of_elements.html',1,'mlx::core::NumberOfElements'],['../classmlx_1_1core_1_1_number_of_elements.html#ac64d7c40ae29d687f8b7d2fa33e13b06',1,'mlx::core::NumberOfElements::NumberOfElements()']]], ['numeric_5flimits_43',['numeric_limits',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core']]], - ['numeric_5flimits_3c_20bfloat16_5ft_20_3e_44',['numeric_limits< bfloat16_t >',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core']]], - ['numeric_5flimits_3c_20float_20_3e_45',['numeric_limits< float >',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core']]], - ['numeric_5flimits_3c_20float16_5ft_20_3e_46',['numeric_limits< float16_t >',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core']]] + ['numeric_5flimits_3c_20bfloat16_5ft_20_3e_44',['numeric_limits< bfloat16_t >',['../structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html',1,'mlx::core']]], + ['numeric_5flimits_3c_20float_20_3e_45',['numeric_limits< float >',['../structmlx_1_1core_1_1numeric__limits_3_01float_01_4.html',1,'mlx::core']]], + ['numeric_5flimits_3c_20float16_5ft_20_3e_46',['numeric_limits< float16_t >',['../structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/all_f.js b/docs/build/html/search/all_f.js index c30b6191a..57978c063 100644 --- a/docs/build/html/search/all_f.js +++ b/docs/build/html/search/all_f.js @@ -3,60 +3,63 @@ var searchData= ['o_0',['O',['../struct_m_l_x_conv_params.html#ad55ff586d30072d8154865f9dfe92d97',1,'MLXConvParams']]], ['o_5fbinary_1',['O_BINARY',['../io_2load_8h.html#a36fa9b2e726512bc17a7a6d3e39002be',1,'load.h']]], ['o_5fstrides_2',['O_strides',['../structmlx_1_1steel_1_1_attn_params.html#ab210f29dcc3a732aba34894cd5a42cf7',1,'mlx::steel::AttnParams']]], - ['offset_3',['offset',['../struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791',1,'LoopedElemToLoc']]], + ['offset_3',['offset',['../struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791',1,'LoopedElemToLoc::offset'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a3a18944c158e2747a6ddebb420299a3b',1,'LoopedElemToLoc< 1, OffsetT, true >::offset'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#af792b1fd4e8286f97b9b863c127a2d9a',1,'LoopedElemToLoc< 1, OffsetT, false >::offset'],['../struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791',1,'LoopedElemToLoc< 1, OffsetT, false >::offset'],['../struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791',1,'LoopedElemToLoc< 1, OffsetT, true >::offset']]], ['offset_5fneg_5fidx_4',['offset_neg_idx',['../kernels_2indexing_8h.html#a58a65ea6215999cd4ccb4fe757cc2dc8',1,'indexing.h']]], ['ofs_5',['ofs',['../classpocketfft_1_1detail_1_1simple__iter.html#ab59481ad9c8f04addb907c3ebb89f8fa',1,'pocketfft::detail::simple_iter::ofs()'],['../classpocketfft_1_1detail_1_1rev__iter.html#a78c3b4ad19edf9d20cab40ad109e9dd1',1,'pocketfft::detail::rev_iter::ofs()']]], ['ones_6',['ones',['../group__ops.html#gae0069146cf8c819b15ba29aa7231a3f0',1,'mlx::core::ones(const Shape &shape, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gace4cf016562af58991f9f961170e156f',1,'mlx::core::ones(const Shape &shape, StreamOrDevice s={})']]], ['ones_5flike_7',['ones_like',['../group__ops.html#ga94f8d3b1906fee99da9cbe39f7be7d42',1,'mlx::core']]], ['oofs_8',['oofs',['../classpocketfft_1_1detail_1_1multi__iter.html#aae63e67caac095d474ddd32daa5ffa34',1,'pocketfft::detail::multi_iter::oofs(size_t i) const'],['../classpocketfft_1_1detail_1_1multi__iter.html#a9236047e7419e5d21379cbf95eb3a78e',1,'pocketfft::detail::multi_iter::oofs(size_t j, size_t i) const']]], ['op_9',['Op',['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23d',1,'mlx::core::BitwiseBinary']]], - ['op_10',['op',['../structmlx_1_1core_1_1_default_strided_reduce.html#ac871f55a7ddd205574974cb4492a240b',1,'mlx::core::DefaultStridedReduce::op'],['../structmlx_1_1core_1_1_default_contiguous_reduce.html#a1928f07db988715cc177999e386f4830',1,'mlx::core::DefaultContiguousReduce::op']]], + ['op_10',['op',['../structmlx_1_1core_1_1_vector_scalar.html#a5fe1744adb58aaa845acca1e46725537',1,'mlx::core::VectorScalar::op'],['../structmlx_1_1core_1_1_scalar_vector.html#ac9c2214744bc972150740e169b603b9b',1,'mlx::core::ScalarVector::op'],['../structmlx_1_1core_1_1_vector_vector.html#a6d69d070c75cf0281e11e36e0717ab50',1,'mlx::core::VectorVector::op']]], ['operations_11',['Core array operations',['../group__ops.html',1,'']]], ['operator_20bool_12',['operator bool',['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant'],['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant']]], ['operator_20dtype_13',['operator Dtype',['../structmlx_1_1core_1_1_type_to_dtype.html#aefdd0fd6a5bbf0197a3996ccd4adea13',1,'mlx::core::TypeToDtype']]], ['operator_20float_14',['operator float',['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aaae72e5340ce91325f1925be36ba46cb',1,'mlx::core::_MLX_BFloat16::operator float()'],['../structmlx_1_1core_1_1complex128__t.html#a3e2faf180c0b785646a0e4296f709a5e',1,'mlx::core::complex128_t::operator float()'],['../structmlx_1_1core_1_1complex64__t.html#a90d224dd37308345086bb9cc882ef6fc',1,'mlx::core::complex64_t::operator float()'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a363de5054f3673bddc90293fc3c9bb99',1,'mlx::core::_MLX_Float16::operator float()']]], - ['operator_20t_15',['operator T',['../structcomplex64__t.html#a70e9b16031eeaff3baa601f400023fcd',1,'complex64_t::operator T() const thread'],['../structcomplex64__t.html#a4f3beea7ab6001189b782a74d1746b67',1,'complex64_t::operator T() const threadgroup'],['../structcomplex64__t.html#a9f4f7eca89ffe6c8d126a4145df6d9f2',1,'complex64_t::operator T() const device'],['../structcomplex64__t.html#ac33e2e5263fec76a4fb4418c6e1d8d14',1,'complex64_t::operator T() const constant'],['../struct___m_l_x___b_float16.html#aa7dfefdf0d15e102d2b8258c9ab01836',1,'_MLX_BFloat16::operator T() const thread'],['../struct___m_l_x___b_float16.html#a2546a8afa77e14ed5b3c5da79a281260',1,'_MLX_BFloat16::operator T() const threadgroup'],['../struct___m_l_x___b_float16.html#a1d523f87740fcb852db6ab57896c245a',1,'_MLX_BFloat16::operator T() const device'],['../struct___m_l_x___b_float16.html#a95acd29283024d7093a0bc58c9468a0a',1,'_MLX_BFloat16::operator T() const constant']]], - ['operator_20val_16',['operator Val',['../structmlx_1_1core_1_1_dtype.html#a3b3bc059be5836476da3cb88a4f5e9fd',1,'mlx::core::Dtype']]], - ['operator_20value_5ftype_17',['operator value_type',['../structmlx_1_1steel_1_1integral__constant.html#a0c11203bed44a6a2c387b365134dcd64',1,'mlx::steel::integral_constant']]], - ['operator_21_3d_18',['operator!=',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a971aa511ab2e7ae1caae09556643a0bd',1,'mlx::core::array::ArrayIterator::operator!=()'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afc6e4fc5589bbf30f978f34868dd4e55',1,'operator!=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6baa722c22d66c7510786bb275cb8cc2',1,'operator!=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa8d9f01582a0a9f01a666d110c74db2a',1,'operator!=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa504a474ab6e00ebe2b1b7ed2f7d1ffb',1,'operator!=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abf5f3040227f021a5b84cf2eda248b2f',1,'operator!=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a347c9bbf816bad2e9e5e91aa448f8b65',1,'operator!=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a33ea086b561c652f25833a5e1ded34dd',1,'operator!=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2bbdcece13148826d3fe33af727bb79b',1,'operator!=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeb1efa47c5f22cc0b35d49ccce73c406',1,'operator!=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa6b99cde403405df1865c989e4ce845a',1,'operator!=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a204d13a881ae8d337f6efbb98673790c',1,'operator!=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3602117b4c61d5cd4fd72fb8e5f68bd6',1,'operator!=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2721c088adfc9d73cde442d6badd2a6c',1,'operator!=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa4364eda56525cf7576ff00e550175e6',1,'mlx::steel::operator!=()'],['../namespacemlx_1_1core.html#a94d00a1b7f8a4717ab3f26f45e4da655',1,'mlx::core::operator!=(const Device &lhs, const Device &rhs)'],['../group__ops.html#ga0ac483d85f23252ca8757e9926d5a3c5',1,'mlx::core::operator!=(const array &a, const array &b)'],['../group__ops.html#ga3fecba9f3cb9a19afd8ca492cf509ce0',1,'mlx::core::operator!=(T a, const array &b)'],['../group__ops.html#gaebbf1cfde388c7480159a03c92c9a385',1,'mlx::core::operator!=(const array &a, T b)'],['../namespacemlx_1_1core.html#a164f109bc19c927b2b3bcc47a5021419',1,'mlx::core::operator!=(const Stream &lhs, const Stream &rhs)'],['../namespacemlx_1_1core.html#ad2f9e1c230ec35d5c406dd616e8f4dea',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af5899b4d5644682cb0ac2a488f630d55',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a72ac8edd190601d7a46782582cedecd8',1,'mlx::core::operator!=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a8084162ba2dd3f9b89195d2bebc3fbb0',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a514263e63f6825b490203ca586864687',1,'mlx::core::operator!=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a1c482bb3d9f9d4c62dee5865892c1f96',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a0030fe7ad09837c670cdfb7d51279519',1,'mlx::core::operator!=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ade3791bc723b8f10fbab22eadb0f705a',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ad78c664f242cd36247c13868547e3dd4',1,'mlx::core::operator!=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab0743a1a1dcb92d40f41ca42d36f242c',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ae7a0f810e546a166c7d05849b5d41f30',1,'mlx::core::operator!=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a676a40637a563f013c725d24fa33fdc8',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a9fcb662b1561e4136bac0106cfb63b6c',1,'mlx::core::operator!=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abcca7fd43590c4347e0f5df8f134030c',1,'mlx::core::operator!=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af3ede3688a2e3b3ba8cb2da180ffe151',1,'mlx::core::operator!=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a54f48469fabd1414bef5097bcded0002',1,'mlx::core::operator!=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af8c648e892cbc6973de535aa17dc2cfe',1,'mlx::core::operator!=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#abc855e1c0584b64d7d995e33211361ab',1,'mlx::core::operator!=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad3684d660d18a54505c759ab286bd936',1,'mlx::core::operator!=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a8afdda14b14262ab5ce0a00c7745d7e8',1,'mlx::core::operator!=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7ccc479be236f2bf3f7725729c5ba201',1,'mlx::core::operator!=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a26a721b8111fce3a1dec9bf724034cd4',1,'mlx::core::operator!=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad5f8c221a53a89e8095aa39fd1f61867',1,'mlx::core::operator!=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a017b52ecf30b33da4aa8da35ccc43220',1,'mlx::core::operator!=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a43c10ca5fb05ee7d0ee63ba56f8a08a3',1,'mlx::core::operator!=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a81284b6ac737f91a8d1ffbbbbf938fe5',1,'mlx::core::operator!=(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_25_19',['operator%',['../backend_2metal_2kernels_2complex_8h.html#aaf53122a07c8eca858b5a8e38ae280e0',1,'operator%(): complex.h'],['../group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde',1,'mlx::core::operator%(const array &a, const array &b)'],['../group__ops.html#ga50817666f0b82afcbf4a123486af9908',1,'mlx::core::operator%(T a, const array &b)'],['../group__ops.html#ga46c01daa07433542a477d216e13a8480',1,'mlx::core::operator%(const array &a, T b)'],['../namespacemlx_1_1core.html#a8723d145dd49021bfcb8e6c99e1c91a5',1,'mlx::core::operator%(complex64_t a, complex64_t b)']]], - ['operator_26_20',['operator&',['../group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff',1,'mlx::core::operator&(const array &a, const array &b)'],['../namespacemlx_1_1core.html#a9ee95f97bbd69262d99d7bea3bf77631',1,'mlx::core::operator&(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0fefc3ae4f1350ebe05ec6098fd6bae3',1,'mlx::core::operator&(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a1e4cb758ccfe5c267baed9aeb0044834',1,'mlx::core::operator&(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab9d0f9910070231695d61de08cadb930',1,'mlx::core::operator&(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a889d401f425db79d1868aa3beea4829b',1,'mlx::core::operator&(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a76dcd1fa3c68b386bc1d1d899a68a120',1,'mlx::core::operator&(uint16_t lhs, _MLX_Float16 rhs)']]], - ['operator_26_26_21',['operator&&',['../namespacemlx_1_1steel.html#a6353bf11881842e25c46b56f92b7044f',1,'mlx::steel::operator&&()'],['../group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3',1,'mlx::core::operator&&(const array &a, const array &b)']]], - ['operator_26_3d_22',['operator&=',['../namespacemlx_1_1core.html#a60c263ef46e552c3954688869734b513',1,'mlx::core::operator&=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af9670fc8088339669c54c68b3a320e25',1,'mlx::core::operator&=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ad1f96f0a02024f347b4c4431629407fc',1,'mlx::core::operator&=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae0540f16c4e7bd55d0e86a88495e4967',1,'mlx::core::operator&=(_MLX_Float16 &lhs, uint16_t rhs)']]], - ['operator_28_29_23',['operator()',['../structpocketfft_1_1detail_1_1_exec_c2_c.html#a4fd637f1a6d335826789af28ac089ecb',1,'pocketfft::detail::ExecC2C::operator()()'],['../structpocketfft_1_1detail_1_1_exec_hartley.html#a67c98b38d12440781053552b9a33bba1',1,'pocketfft::detail::ExecHartley::operator()()'],['../structpocketfft_1_1detail_1_1_exec_dcst.html#a67f4f56e3574c491695f8cb8a1e983d8',1,'pocketfft::detail::ExecDcst::operator()()'],['../structpocketfft_1_1detail_1_1_exec_r2_r.html#acdba1650962714e6afff51e9ca456970',1,'pocketfft::detail::ExecR2R::operator()()'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a0d657bc9a381dca1b5860b9a1b5a5702',1,'mlx::core::detail::Abs::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a564232db7d32811e2ae126c86de104f0',1,'mlx::core::detail::Abs::operator()(uint8_t x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a5fac7e6c8277d8706535a52820503c9d',1,'mlx::core::detail::Abs::operator()(uint16_t x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#af2c3723e648bd5ed2fe558cc20b7f5eb',1,'mlx::core::detail::Abs::operator()(uint32_t x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a57312cd8530dd0ede3b8037f9c401883',1,'mlx::core::detail::Abs::operator()(uint64_t x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#ab3b5e3853ed56bfbfa577d965c21112e',1,'mlx::core::detail::Abs::operator()(bool x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html#a04b4c9d1fc0160973aa28b1f809b9d51',1,'mlx::core::detail::ArcCos::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html#a767d354bec863942822ee0b9b6742a88',1,'mlx::core::detail::ArcCosh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html#ac69091929815e5317308b4088f5c2f46',1,'mlx::core::detail::ArcSin::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html#ac7bf9bac66fef917f75494b2345e6aaf',1,'mlx::core::detail::ArcSinh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html#aee87bf10c278a70ca788085d1b499afe',1,'mlx::core::detail::ArcTan::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html#a9040b7afcdb4969924aa782fa67f03ac',1,'mlx::core::detail::ArcTan2::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html#a601e8c52bb938eb3a616756a35419e8b',1,'mlx::core::detail::ArcTanh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a672f65e47d65e4e8d88be252bce0164b',1,'mlx::core::detail::Ceil::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a72a2cab2728fb5e1cc6329a539e5d573',1,'mlx::core::detail::Ceil::operator()(int8_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#aa34590f6a41331be92988558a90dc6fa',1,'mlx::core::detail::Ceil::operator()(int16_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#af14120f3dd98f6198ea257d75be223f7',1,'mlx::core::detail::Ceil::operator()(int32_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#af263ce7743cf7319387baba616c375b5',1,'mlx::core::detail::Ceil::operator()(int64_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a48f00affcd5c2ea1f81d821e019fec29',1,'mlx::core::detail::Ceil::operator()(uint8_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#ad4d24a44e8a328948393701dacb0ceac',1,'mlx::core::detail::Ceil::operator()(uint16_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a70da19b5c9c69f04b9f196bdf266f93c',1,'mlx::core::detail::Ceil::operator()(uint32_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#af0e7e806b73c664ada837476f9d4d43b',1,'mlx::core::detail::Ceil::operator()(uint64_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#acc1bfc84a9b91f6e9764234cbe3b9687',1,'mlx::core::detail::Ceil::operator()(bool x)'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html#a7e662d05c6998bd6ced8ad9c187324a5',1,'mlx::core::detail::Conjugate::operator()()'],['../structmlx_1_1core_1_1detail_1_1_cos.html#ad4caef573f9d9071f8945a8efed231ad',1,'mlx::core::detail::Cos::operator()()'],['../structmlx_1_1core_1_1detail_1_1_cosh.html#a63591f49776d9aadc02200036ae38317',1,'mlx::core::detail::Cosh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_erf.html#a168f8ccc6c8053b05dd1a48904ca8fd4',1,'mlx::core::detail::Erf::operator()()'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html#acc93c0511141404208b35f302f8c1fcb',1,'mlx::core::detail::ErfInv::operator()()'],['../structmlx_1_1core_1_1detail_1_1_exp.html#a0846300cee28315e5b42f74acafbd1a1',1,'mlx::core::detail::Exp::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_exp.html#af247c0d19d59f3310fd0a081eb92cf8b',1,'mlx::core::detail::Exp::operator()(complex64_t x)'],['../structmlx_1_1core_1_1detail_1_1_expm1.html#abf7e61b8387521e9d44334ce88d833a0',1,'mlx::core::detail::Expm1::operator()()'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a16c13cfe736098bffc81d655e172294a',1,'mlx::core::detail::Floor::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a9b6c4c34b6594b8c413abe31f34a73df',1,'mlx::core::detail::Floor::operator()(int8_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#aca4c71204b3ceeca6329f7ea2b041f4c',1,'mlx::core::detail::Floor::operator()(int16_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a3c3ab9e00d1fbd124802517e8c35fe02',1,'mlx::core::detail::Floor::operator()(int32_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a4b5954ffc59c741dd7b86bafda69d5cc',1,'mlx::core::detail::Floor::operator()(int64_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a2e33b10bd5b04551054a87c601440bc7',1,'mlx::core::detail::Floor::operator()(uint8_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a2500f971100919a694f78669a5e4f438',1,'mlx::core::detail::Floor::operator()(uint16_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a23df818301d68389e6e12f5a9ec1fbd7',1,'mlx::core::detail::Floor::operator()(uint32_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#ac988b4f265cf46c68609c9c8787c15fb',1,'mlx::core::detail::Floor::operator()(uint64_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a7f936e3fd53772bc189d845c73b53202',1,'mlx::core::detail::Floor::operator()(bool x)'],['../structmlx_1_1core_1_1detail_1_1_imag.html#a5bd82e2185f3779e398c179d42a3e782',1,'mlx::core::detail::Imag::operator()()'],['../structmlx_1_1core_1_1detail_1_1_log.html#a0012a4e1744dbe9a28c3b5652be6e1c6',1,'mlx::core::detail::Log::operator()()'],['../structmlx_1_1core_1_1detail_1_1_log2.html#a467bd4c995674721ff5fff6df33aead8',1,'mlx::core::detail::Log2::operator()()'],['../structmlx_1_1core_1_1detail_1_1_log10.html#a2633c5b772bbc9f8b66cffd4a3e01a3f',1,'mlx::core::detail::Log10::operator()()'],['../structmlx_1_1core_1_1detail_1_1_log1p.html#a3220de8c6090c44aa2070b1fbb2dc340',1,'mlx::core::detail::Log1p::operator()()'],['../structmlx_1_1core_1_1detail_1_1_logical_not.html#a79799668ea5c364b0b4e2bc330e76253',1,'mlx::core::detail::LogicalNot::operator()()'],['../structmlx_1_1core_1_1detail_1_1_negative.html#afc4595c70ef7196df374cf4b2cc5e526',1,'mlx::core::detail::Negative::operator()()'],['../structmlx_1_1core_1_1detail_1_1_real.html#ae84a939fdb5916257a7731cda66d4d61',1,'mlx::core::detail::Real::operator()()'],['../structmlx_1_1core_1_1detail_1_1_round.html#a653f29c059bbfa6192378732a8a23351',1,'mlx::core::detail::Round::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_round.html#a82a984f13568051009e257fe85227da6',1,'mlx::core::detail::Round::operator()(complex64_t x)'],['../structmlx_1_1core_1_1detail_1_1_sigmoid.html#a64b72561bfaf758632167f00648f4c89',1,'mlx::core::detail::Sigmoid::operator()()'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a64ed5013cee7ff18c7fe70bc04737e7b',1,'mlx::core::detail::Sign::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a7106ed1f2f98a365fcb3e6ee39084748',1,'mlx::core::detail::Sign::operator()(uint8_t x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a7163e8c068dcc460600ed04014dc9945',1,'mlx::core::detail::Sign::operator()(uint16_t x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#ae8f56c7134721c846240830169424c22',1,'mlx::core::detail::Sign::operator()(uint32_t x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a10ae519e9a74a327fc72c410e9ab2936',1,'mlx::core::detail::Sign::operator()(uint64_t x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a91be4e273f6c7ea5d44cfab380b77603',1,'mlx::core::detail::Sign::operator()(complex64_t x)'],['../structmlx_1_1core_1_1detail_1_1_sin.html#ae95671816529cc2188389af37a2f1a13',1,'mlx::core::detail::Sin::operator()()'],['../structmlx_1_1core_1_1detail_1_1_sinh.html#a9663ddf0fa4c0003576b48f3d5385f00',1,'mlx::core::detail::Sinh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_square.html#a54e9e3c0d0896e142289e8282eab1099',1,'mlx::core::detail::Square::operator()()'],['../structmlx_1_1core_1_1detail_1_1_sqrt.html#aa5a4830b3ef7efab20ea88a110667efd',1,'mlx::core::detail::Sqrt::operator()()'],['../structmlx_1_1core_1_1detail_1_1_rsqrt.html#a9af247be16bab83243038aac54446b79',1,'mlx::core::detail::Rsqrt::operator()()'],['../structmlx_1_1core_1_1detail_1_1_tan.html#aba397cd7ac05bbe06dfa9e3a64bdb05f',1,'mlx::core::detail::Tan::operator()()'],['../structmlx_1_1core_1_1detail_1_1_tanh.html#a1749ba1edfd53095ed7d45c0e53bab61',1,'mlx::core::detail::Tanh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_add.html#a2d6011c35768b5fcd2bb75747b944353',1,'mlx::core::detail::Add::operator()()'],['../structmlx_1_1core_1_1detail_1_1_divide.html#a5e0d22e2084c4ca81bec0d457a46c662',1,'mlx::core::detail::Divide::operator()()'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a3bdaf1095ad883ecc0fecc455f02cbf3',1,'mlx::core::detail::Remainder::operator()(T numerator, T denominator)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a52c3a2ba86fccb24d37d218ae8328954',1,'mlx::core::detail::Remainder::operator()(T numerator, T denominator)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a0b0dd6ef5b08585fdf8355770da8d747',1,'mlx::core::detail::Remainder::operator()(T numerator, T denominator)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a68fe542084fb94d9a5abd740fe07832b',1,'mlx::core::detail::Remainder::operator()(complex64_t numerator, complex64_t denominator)'],['../structmlx_1_1core_1_1detail_1_1_equal.html#a2994cf1884e7126e76d0a20b215fe3ab',1,'mlx::core::detail::Equal::operator()()'],['../structmlx_1_1core_1_1detail_1_1_na_n_equal.html#a073b20b0d8d41ec8364b7c477421b9bf',1,'mlx::core::detail::NaNEqual::operator()()'],['../structmlx_1_1core_1_1detail_1_1_greater.html#aa3844c2bae3c7a981739f642aa0dd094',1,'mlx::core::detail::Greater::operator()()'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html#a3b005f85522ad0e4b57044eed930ac30',1,'mlx::core::detail::GreaterEqual::operator()()'],['../structmlx_1_1core_1_1detail_1_1_less.html#a0b4032dff1ad2b387745cb000aabdcbb',1,'mlx::core::detail::Less::operator()()'],['../structmlx_1_1core_1_1detail_1_1_less_equal.html#a31e70f8830a07557697541301555a7a7',1,'mlx::core::detail::LessEqual::operator()()'],['../structmlx_1_1core_1_1detail_1_1_maximum.html#a3eb37abec8426ebc42b8c685075c523a',1,'mlx::core::detail::Maximum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_maximum.html#af99345c7c8bc95ccab1b22c0792ac6fd',1,'mlx::core::detail::Maximum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_minimum.html#afca0861556416a8547dd8574528feb69',1,'mlx::core::detail::Minimum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_minimum.html#a64b2eecfbc56aaef7deb939423bac3f8',1,'mlx::core::detail::Minimum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_log_add_exp.html#ad1663fd809acaa4038f90666436599e5',1,'mlx::core::detail::LogAddExp::operator()()'],['../structmlx_1_1core_1_1detail_1_1_multiply.html#a898b090966b047723513224b8d3b22f1',1,'mlx::core::detail::Multiply::operator()()'],['../structmlx_1_1core_1_1detail_1_1_not_equal.html#a23d662b5fd968dc17d3bee2595b5f99d',1,'mlx::core::detail::NotEqual::operator()()'],['../structmlx_1_1core_1_1detail_1_1_power.html#a2c047e1b488e6525447a224975a75db8',1,'mlx::core::detail::Power::operator()(T base, T exp)'],['../structmlx_1_1core_1_1detail_1_1_power.html#a9967db24b8f67d54b6aa3810e274f28c',1,'mlx::core::detail::Power::operator()(T base, T exp)'],['../structmlx_1_1core_1_1detail_1_1_subtract.html#a72ef05830615a2d5d9662926ed82672a',1,'mlx::core::detail::Subtract::operator()()'],['../structmlx_1_1core_1_1detail_1_1_logical_and.html#a046536c1f2f9367983f052a213d7b7d8',1,'mlx::core::detail::LogicalAnd::operator()()'],['../structmlx_1_1core_1_1detail_1_1_logical_or.html#afb134dbab79307d4ba597843c61d0b1a',1,'mlx::core::detail::LogicalOr::operator()()'],['../structmlx_1_1core_1_1detail_1_1_select.html#a930f9da2e6b3453e04f21382435a2cfb',1,'mlx::core::detail::Select::operator()()'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html#ae0bed77f95fe2b2f0b594addddd04700',1,'mlx::core::detail::BitwiseAnd::operator()()'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html#a5ab05734c5000b454975de6647a08d20',1,'mlx::core::detail::BitwiseOr::operator()()'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html#a0989e3bcd064ae06c33f660696a869a0',1,'mlx::core::detail::BitwiseXor::operator()()'],['../structmlx_1_1core_1_1detail_1_1_left_shift.html#a9385f580830a6ad163dd9bb8c4905e7a',1,'mlx::core::detail::LeftShift::operator()()'],['../structmlx_1_1core_1_1detail_1_1_right_shift.html#a154528ba50e89a4c532a181f135b1620',1,'mlx::core::detail::RightShift::operator()()'],['../structmlx_1_1core_1_1_default_strided_reduce.html#a024682ab93b84e544a07e3a9c3c51fba',1,'mlx::core::DefaultStridedReduce::operator()()'],['../structmlx_1_1core_1_1_default_contiguous_reduce.html#a08144c7a3cdf10af5e47f4575da3694f',1,'mlx::core::DefaultContiguousReduce::operator()()'],['../struct_add.html#ac5c66b63d63a222d3ae0ab8cc7c90eb5',1,'Add::operator()()'],['../struct_floor_divide.html#a2b328e4d768e718fa439f955c524666a',1,'FloorDivide::operator()(T x, T y)'],['../struct_floor_divide.html#afc16a2b2a745225e0bc95640f3fc0219',1,'FloorDivide::operator()(float x, float y)'],['../struct_floor_divide.html#ae91719a15f7e643d552129f476089c6a',1,'FloorDivide::operator()(half x, half y)'],['../struct_floor_divide.html#a4aa9f858626583e02bd79f747229bbca',1,'FloorDivide::operator()(bfloat16_t x, bfloat16_t y)'],['../struct_divide.html#a0a16b9194abc2ab7c61129f81a9bbb3d',1,'Divide::operator()()'],['../struct_remainder.html#ab7875512ff4341c580c6dc372e64fc58',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#a18150b5f4425e30b95ffabc6bb25cede',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#ab3b75f54b56fd357c9755daadb2cafc2',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#ae918ce0e246937d4fe04e2ea36e4b2c1',1,'Remainder::operator()(complex64_t x, complex64_t y)'],['../struct_equal.html#aa498087080900d4428ba428a6496a769',1,'Equal::operator()()'],['../struct_na_n_equal.html#a00220898e02db656d21dde9e9354a8dc',1,'NaNEqual::operator()(T x, T y)'],['../struct_na_n_equal.html#a6185e4554dce5b4659d21673c576be51',1,'NaNEqual::operator()(complex64_t x, complex64_t y)'],['../struct_greater.html#a98d7d8ee360cd0f469c6eb9a017560f5',1,'Greater::operator()()'],['../struct_greater_equal.html#ae69a3bccc567a46506cf0d296294ce80',1,'GreaterEqual::operator()()'],['../struct_less.html#a5ee0b31b2d9123dc4504f2979a5854d3',1,'Less::operator()()'],['../struct_less_equal.html#ae9f9a1b2eae548977139704f0044acfe',1,'LessEqual::operator()()'],['../struct_log_add_exp.html#ab32417f18e8ff68c15f78aceeb624edf',1,'LogAddExp::operator()()'],['../struct_maximum.html#a3ea0f42bc4cd80b68a98f189f9fa859c',1,'Maximum::operator()(T x, T y)'],['../struct_maximum.html#a0bc8fadc87f2c49fc440d625bfc97ca6',1,'Maximum::operator()(T x, T y)'],['../struct_maximum.html#a907e8793900be5927625377dab199644',1,'Maximum::operator()(complex64_t x, complex64_t y)'],['../struct_minimum.html#aa6113dfac3986c0f571fa53f65c5330e',1,'Minimum::operator()(T x, T y)'],['../struct_minimum.html#a0c939921de87ab9c6959238aac81a059',1,'Minimum::operator()(T x, T y)'],['../struct_minimum.html#a800fba087280f79c2f7e9aff75bed093',1,'Minimum::operator()(complex64_t x, complex64_t y)'],['../struct_multiply.html#a1327fc5a0713931afe997b0d4d2988e0',1,'Multiply::operator()()'],['../struct_not_equal.html#af008d73a5d9cde0b8309b7e8ee7438b2',1,'NotEqual::operator()(T x, T y)'],['../struct_not_equal.html#a14de494cea4e4869351202cad1149f17',1,'NotEqual::operator()(complex64_t x, complex64_t y)'],['../struct_power.html#a2b6df2a9e48155ff9734caca8504a79f',1,'Power::operator()(T base, T exp)'],['../struct_power.html#a36829163d42973034a1f8a7ecc57a1de',1,'Power::operator()(T base, T exp)'],['../struct_power.html#a27cdfb313c4e82b63bdcdaee923cbbef',1,'Power::operator()(complex64_t x, complex64_t y)'],['../struct_subtract.html#ae0856cd8d449074ca287baa7e460f68a',1,'Subtract::operator()()'],['../struct_logical_and.html#a8bc6bdabc0ea0678a46e2cf6217cb3a6',1,'LogicalAnd::operator()()'],['../struct_logical_or.html#ade6a931324a604a3119d2220d6f5460d',1,'LogicalOr::operator()()'],['../struct_bitwise_and.html#afb48af090b01dd0200963bc12d842e36',1,'BitwiseAnd::operator()()'],['../struct_bitwise_or.html#a41f847463daafa99ee56f4035578390f',1,'BitwiseOr::operator()()'],['../struct_bitwise_xor.html#a3a3e8a56caab739d40262d9349c9c485',1,'BitwiseXor::operator()()'],['../struct_left_shift.html#aa729747784c38bfdbba34794fcf5175b',1,'LeftShift::operator()()'],['../struct_right_shift.html#a2cc59b400c68342b0e43050431323c17',1,'RightShift::operator()()'],['../struct_arc_tan2.html#ac9b7729753e13be293ab700231d061ac',1,'ArcTan2::operator()()'],['../struct_div_mod.html#a8b5758f2ea18d4c903b462331b25abfe',1,'DivMod::operator()()'],['../struct_cum_prod.html#ad634be0b139d10ce6d21332eef0d936b',1,'CumProd< bool >::operator()()'],['../struct_cum_max.html#a781b9b955c5412466da6af6c70d73c06',1,'CumMax::operator()()'],['../struct_cum_min.html#ae0b8c3761e04fa538d304ca842281a66',1,'CumMin::operator()()'],['../struct_less_than.html#a2798eb377b411c93a4ed30cf35caade2',1,'LessThan::operator()()'],['../struct_select.html#adb51692aae3038de07dd745891bf9848',1,'Select::operator()()'],['../struct_abs.html#a9e7481dfcc162509769852026ff4a344',1,'Abs::operator()(T x)'],['../struct_abs.html#a0ca113fd036151c443df3f83cc667f28',1,'Abs::operator()(uint8_t x)'],['../struct_abs.html#adaeab32a7e377dc990077ab15f3dc4c2',1,'Abs::operator()(uint16_t x)'],['../struct_abs.html#a99d2a2f37a6cddd3168b0224f2a9b963',1,'Abs::operator()(uint32_t x)'],['../struct_abs.html#ac9cbc02422d930479303f240a7ea6c71',1,'Abs::operator()(uint64_t x)'],['../struct_abs.html#ac30835b27784d451bd2e4524c8eb9e11',1,'Abs::operator()(bool x)'],['../struct_abs.html#ab82917d6b30a2c579e7eb879d305c5fc',1,'Abs::operator()(complex64_t x)'],['../struct_arc_cos.html#a5553cecf58511e24e76ac97f2d90b9ac',1,'ArcCos::operator()()'],['../struct_arc_cosh.html#a5c9e7712c14c97298b23ec48e19abc58',1,'ArcCosh::operator()()'],['../struct_arc_sin.html#a0343872f2da93bae2bb0baadf49da022',1,'ArcSin::operator()()'],['../struct_arc_sinh.html#a3066fb7dc7c3180100fb55ff94af6a7a',1,'ArcSinh::operator()()'],['../struct_arc_tan.html#af3a0aec6acec8ae8f5e4c4d5cf8c91ba',1,'ArcTan::operator()()'],['../struct_arc_tanh.html#a37dc3e01ec2830de7e82ed6c6363ac88',1,'ArcTanh::operator()()'],['../struct_ceil.html#a5e2a4ef1b012f5d352064489156e5e44',1,'Ceil::operator()(T x)'],['../struct_ceil.html#a455cd8083ba859993077f2e078ae165b',1,'Ceil::operator()(int8_t x)'],['../struct_ceil.html#a2acb61bc658c7a216795e7f76ebcf98a',1,'Ceil::operator()(int16_t x)'],['../struct_ceil.html#aef8c37f7a8ee3fc80700d605a09891fb',1,'Ceil::operator()(int32_t x)'],['../struct_ceil.html#a93d0110511ad5dd200e12d37a3d7d6e3',1,'Ceil::operator()(int64_t x)'],['../struct_ceil.html#aa335b745fa26e0f443cdb36298105484',1,'Ceil::operator()(uint8_t x)'],['../struct_ceil.html#ade17e13b7f30f5c590fae1581a2013ac',1,'Ceil::operator()(uint16_t x)'],['../struct_ceil.html#a411c75cc35cdc088402e176a1defd22d',1,'Ceil::operator()(uint32_t x)'],['../struct_ceil.html#a9ac660ca29eef7a7429fceb7b917a68a',1,'Ceil::operator()(uint64_t x)'],['../struct_ceil.html#a40de367e62f06ebd7e1330afa93a9ad9',1,'Ceil::operator()(bool x)'],['../struct_cos.html#ae222f8710f6b8254c471ebd475aa5bda',1,'Cos::operator()(T x)'],['../struct_cos.html#a5f26feb1dcc4bec5f59a9ff511c5b163',1,'Cos::operator()(complex64_t x)'],['../struct_cosh.html#a5847ebeebb236fdc926798ddc16475ba',1,'Cosh::operator()(T x)'],['../struct_cosh.html#aefdd91298dac16d528d29ee47e2f7252',1,'Cosh::operator()(complex64_t x)'],['../struct_conjugate.html#acb0a2694285f1f57c7654b371ce8cbd8',1,'Conjugate::operator()()'],['../struct_erf.html#a80719402ad7f7d418859a6677d7b604d',1,'Erf::operator()()'],['../struct_erf_inv.html#afbf3668d1a512e889f093a0bc7673309',1,'ErfInv::operator()()'],['../struct_exp.html#a5ef395868e055348c0802fd5fe45669c',1,'Exp::operator()(T x)'],['../struct_exp.html#a2b341ac400c4d145397950eb60734336',1,'Exp::operator()(complex64_t x)'],['../struct_expm1.html#a4b834d42cf0b84daf03fec62c222091a',1,'Expm1::operator()()'],['../struct_floor.html#ace3551f28429081e9f3a3dab0c84212b',1,'Floor::operator()(T x)'],['../struct_floor.html#a10d7fd05b4c224c9f135451246d13014',1,'Floor::operator()(int8_t x)'],['../struct_floor.html#a2865a04a492e3590302f4bd3215a10d7',1,'Floor::operator()(int16_t x)'],['../struct_floor.html#a41012343ff0463ec44b4d06196f41182',1,'Floor::operator()(int32_t x)'],['../struct_floor.html#aae3181d15856796aa0628cf30c92aa2e',1,'Floor::operator()(int64_t x)'],['../struct_floor.html#ac6cf38d82c8e270911afdca4c69ad51b',1,'Floor::operator()(uint8_t x)'],['../struct_floor.html#a78969b9e2b53ae248e72a67259eea5d8',1,'Floor::operator()(uint16_t x)'],['../struct_floor.html#a959009320ed622ed45b39becab1d5b98',1,'Floor::operator()(uint32_t x)'],['../struct_floor.html#a7d04b83c3345cd867315cae2d7ff68ab',1,'Floor::operator()(uint64_t x)'],['../struct_floor.html#abea845fe5e8e6b93bd4bca8717337e0b',1,'Floor::operator()(bool x)'],['../struct_imag.html#a3b29e9f8a46c194d683f6a9938314400',1,'Imag::operator()()'],['../struct_log.html#a32a383cb6be06e616a75f23bf49089c3',1,'Log::operator()()'],['../struct_log2.html#ac1e067ecdcbdbffb6106e789c2b98b64',1,'Log2::operator()()'],['../struct_log10.html#ac596a74c1642a00f3eced07ee3334122',1,'Log10::operator()()'],['../struct_log1p.html#a4464c6e7bdbe55ffd7d961c695cd13ce',1,'Log1p::operator()()'],['../struct_logical_not.html#a8a620bac957ab8c09ac85adfddd96708',1,'LogicalNot::operator()()'],['../struct_negative.html#af6879b374314a559faa321e8cce3d710',1,'Negative::operator()()'],['../struct_real.html#a85b9c5b9e65297994fa26ff68e19e809',1,'Real::operator()()'],['../struct_round.html#aa06a0195867e2ceb679c403b6909a1c4',1,'Round::operator()(T x)'],['../struct_round.html#ad3a08f2276ff1033900bc0a7da812655',1,'Round::operator()(complex64_t x)'],['../struct_sigmoid.html#a75a24cd75cb4d4c9a072811b2d70ad55',1,'Sigmoid::operator()()'],['../struct_sign.html#aa3304c6b43bcad53061614b741d8403c',1,'Sign::operator()(T x)'],['../struct_sign.html#ac48992b675b8b28be1e27e1f2ec5d2f7',1,'Sign::operator()(uint32_t x)'],['../struct_sign.html#ae07a4249e1b61419a3b9ca6c337b7bb5',1,'Sign::operator()(complex64_t x)'],['../struct_sin.html#a7caf98c777521fa5d5c6ddaaa3b779fd',1,'Sin::operator()(T x)'],['../struct_sin.html#aa510cf4595b6d49065ab6b602d8fcb14',1,'Sin::operator()(complex64_t x)'],['../struct_sinh.html#a02cf32bcf560657b9ee34fb1affed8e2',1,'Sinh::operator()(T x)'],['../struct_sinh.html#a1f8ba1858d352ee68861cd6ea861af43',1,'Sinh::operator()(complex64_t x)'],['../struct_square.html#afde739fc544e45dd30964c02dca94310',1,'Square::operator()()'],['../struct_sqrt.html#ab9b16d2b9b03a1c54190f4479a56a4ad',1,'Sqrt::operator()()'],['../struct_rsqrt.html#ae16699fd829e40416436247a39233fda',1,'Rsqrt::operator()()'],['../struct_tan.html#a1e6fb8c691621c69cb9bd393de4f6e78',1,'Tan::operator()(T x)'],['../struct_tan.html#a2ef120c9f92b0d2e9cec8389eda05724',1,'Tan::operator()(complex64_t x)'],['../struct_tanh.html#adce11a7ad33226c6ecff34f46f5c45d7',1,'Tanh::operator()(T x)'],['../struct_tanh.html#aa8423b43c725bb4b88965a11e8cf20f6',1,'Tanh::operator()(complex64_t x)'],['../structmlx_1_1core_1_1_function_exporter.html#ada4e13daeb3ba0f5ebe20ec0663727b3',1,'mlx::core::FunctionExporter::operator()(const std::initializer_list< array > &args)'],['../structmlx_1_1core_1_1_function_exporter.html#a82aeb5fa32ef5638f42dc2372278427e',1,'mlx::core::FunctionExporter::operator()(const Args &args)'],['../structmlx_1_1core_1_1_function_exporter.html#ac8b8fa0a23d58a94e2e9b923dc7324e8',1,'mlx::core::FunctionExporter::operator()(const Kwargs &kwargs)'],['../structmlx_1_1core_1_1_function_exporter.html#a35a3c1d94249ce0fe0e82b0ea047d441',1,'mlx::core::FunctionExporter::operator()(const Args &args, const Kwargs &kwargs)'],['../structmlx_1_1core_1_1_imported_function.html#a3555db23026d30eaeee265fed99947b2',1,'mlx::core::ImportedFunction::operator()(const std::initializer_list< array > &args) const'],['../structmlx_1_1core_1_1_imported_function.html#a5953b3f47c094cc47bcbb0845379ca8d',1,'mlx::core::ImportedFunction::operator()(const Args &args) const'],['../structmlx_1_1core_1_1_imported_function.html#a10fec4eab5851ed825a9b46a31cedcc9',1,'mlx::core::ImportedFunction::operator()(const Kwargs &kwargs) const'],['../structmlx_1_1core_1_1_imported_function.html#a7d1accece61230eec256e0f70610776d',1,'mlx::core::ImportedFunction::operator()(const Args &args, const Kwargs &kwargs) const']]], - ['operator_2a_24',['operator*',['../structpocketfft_1_1detail_1_1cmplx.html#a26bf3d709a58f06228e502af6db8e5ac',1,'pocketfft::detail::cmplx::operator*(const T2 &other) const -> cmplx< decltype(r *other)>'],['../structpocketfft_1_1detail_1_1cmplx.html#ad9c591ef8ae976293f207937d273e9a1',1,'pocketfft::detail::cmplx::operator*(const cmplx< T2 > &other) const -> cmplx< decltype(r+other.r)>'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a153756072fda6d3e53bcca11b46a1238',1,'mlx::core::array::ArrayIterator::operator*()'],['../backend_2metal_2kernels_2complex_8h.html#a681d4fb076973f58f7dac894ec62a385',1,'operator*(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8f06316063fc91747533105f256b55b5',1,'operator*(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7b3bce3f6f17089d87e13e91f580a581',1,'operator*(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a54ae7216b82c5cea362f6b83e1df3a9b',1,'operator*(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a852689073c17596de4fb545bc046b380',1,'operator*(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a168300bbd04d8e97c5e4218cb14ae378',1,'operator*(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6278bd2e0e2805090b33ef666bf7f6bb',1,'operator*(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aecf703522d9ce32dfeefe1e6e903db06',1,'operator*(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7cd44d27fa9a4f13df39894c34fdb348',1,'operator*(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee64dc1890abb6d1035361cb8c751f96',1,'operator*(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1a559ab88dbbb4fd2c7509d2c94e55b',1,'operator*(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a495ae2d9be5d97c4c6448fc4e50a03e1',1,'operator*(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a87ab4b7a502430da664ccb8abd383058',1,'operator*(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5f997839cf49c24ab594a0dff486a7bc',1,'operator*(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa0c2d29950926ae579adf6337fbea64b',1,'mlx::steel::operator*()'],['../group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0',1,'mlx::core::operator*(const array &a, const array &b)'],['../group__ops.html#gac22a67f7de797b1ae59029843cbdcab6',1,'mlx::core::operator*(T a, const array &b)'],['../group__ops.html#ga6f2369ed5fae8ff9b1528670a004dde2',1,'mlx::core::operator*(const array &a, T b)'],['../namespacemlx_1_1core.html#a0cc824d6318f97f7058918ab64ddfc25',1,'mlx::core::operator*(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a81e1c727c3fc48910b030cb65a9e7afa',1,'mlx::core::operator*(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a861d948220d8f48d46c68d2ddb16a096',1,'mlx::core::operator*(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a13d16561812679b36e68185dc4b2d04d',1,'mlx::core::operator*(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a5287610200ff573730c9c92413f48881',1,'mlx::core::operator*(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a377ccc6b4ef36767abca102dca56dc10',1,'mlx::core::operator*(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a5d696b63635ce6967526d6a410f7f6b1',1,'mlx::core::operator*(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abe90e9527bfa3e1c813d41df4a2372e7',1,'mlx::core::operator*(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a5f14963c77f96bcb5a3bef5661a86ba4',1,'mlx::core::operator*(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#acfb06fe9f5fee01dbb5a2b23bccfd0d3',1,'mlx::core::operator*(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#afc9a87f1fccbac05242b91bfbb35c24d',1,'mlx::core::operator*(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0b9678af9b487900cacf6639a4693de0',1,'mlx::core::operator*(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ad5950619081389e6ed7512f38358d33d',1,'mlx::core::operator*(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a65d25d082374761c05b056e1046d1d4e',1,'mlx::core::operator*(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a759191fb984e7737f0ef529c2053ad73',1,'mlx::core::operator*(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3a52675c3d4552b319dd9707844abdec',1,'mlx::core::operator*(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a45d67f5d80fba4d42e34c682a8d22beb',1,'mlx::core::operator*(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ad25880c67bbcbfafbe54dc16418bf736',1,'mlx::core::operator*(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a63c836e1141e07ae72cee770bad01200',1,'mlx::core::operator*(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a265a37b8ee4a97390213e9ec49693e66',1,'mlx::core::operator*(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab5a457da04dcb157a0b5172c4b2244b6',1,'mlx::core::operator*(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#aa56a8bda08be9ef3711496e216a75c95',1,'mlx::core::operator*(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af89612098dd355b1eefb841c753b36ab',1,'mlx::core::operator*(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a4552687a0637f710b5d55bb6378fcabe',1,'mlx::core::operator*(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af69db7def588d7da430434a69456e29c',1,'mlx::core::operator*(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a00af6e5095888f00791ee0ab6d993ad6',1,'mlx::core::operator*(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab48feddc1aa304383e5493923506ad7a',1,'mlx::core::operator*(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a0367b582e85162b4180e086f725e49e9',1,'mlx::core::operator*(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a45f0479526fbccdb00bc73ea7f3b7625',1,'mlx::core::operator*(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a394797646010ba9ef2a1f9b9a4b8ddd9',1,'mlx::core::operator*(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acaaa86b59c7ceb2e092ac07f2a75225c',1,'mlx::core::operator*(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a067d47823a322b88043cce7ce4a3ec78',1,'mlx::core::operator*(bfloat16_t lhs, float16_t rhs)']]], - ['operator_2a_3d_25',['operator*=',['../structpocketfft_1_1detail_1_1cmplx.html#a683fd490182c9189fa2c05b1823edd93',1,'pocketfft::detail::cmplx::operator*=(T2 other)'],['../structpocketfft_1_1detail_1_1cmplx.html#a06f2c26c6fc4722e61b44da4c242ed87',1,'pocketfft::detail::cmplx::operator*=(const cmplx< T2 > &other)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7232b0a0e193b3c6172d6fc2578bf419',1,'operator*=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ade65ebca11e38d56408c512df89b99f4',1,'operator*=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af4348ce3425dd99d069e8fdf06e25a3c',1,'operator*=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2c3c5f793b3d957d7295d7f1faabebee',1,'operator*=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac66657077d55e94197b52b63acb50b7d',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a383165ea838cc3feeee4d9cf54aa77cc',1,'operator*=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab706af260b61f735b28464877d02137c',1,'operator*=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a979374b1dd4e0eaf602326fa901336d1',1,'operator*=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac815eec2c1b15a47b1c6ea6790e77d24',1,'operator*=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8110fae7bcc34a0de5927546b24aa935',1,'operator*=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae4acef3e7ae7dfe359422503f894e885',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc268cdbc30500f3009f5de2b2f0f67a',1,'operator*=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a81f65b04a87a25c7eb1a751d1be9fa55',1,'operator*=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08c1f916302eb9d48c93f8b7260538fe',1,'operator*=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc8e82b8f593b12c6d405e2250ab0f62',1,'operator*=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4611728172afea51860a77fdb06cafa0',1,'operator*=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0b8736e2ae24758b6e24ea72668df5b4',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad920df9579603f0b0ee2689eba330617',1,'operator*=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae97ab6c3ddcc2754b24f86319a5398be',1,'operator*=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3ff4ff59f411010ac8502cfabda4bd6f',1,'operator*=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abd3d82e2dec1847e97eb8fc3bab2985a',1,'operator*=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a738078eb7d5ff94ff48156a555d763a5',1,'operator*=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a435f2f4256aadb1b57fd62bb7f733cf7',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0e4377b120d6305335d296e031ee5b30',1,'operator*=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a917354f77eac26189da8a2f610a00074',1,'operator*=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af725f935bfa0405e5ff17ede3ac47283',1,'operator*=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7c56980c234a04260b8b19298085e526',1,'operator*=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab840ff9de0cdd0e9afffb8baa2a850a3',1,'operator*=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a73416a7415f3fe31525e33419e5e8aab',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a16978f4b16d954ef4d4cf0f32f6c0b94',1,'operator*=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a99aa4cc110d1c7aa3b4c8c5cbf9235b7',1,'operator*=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2179abbc91ce8763e96e39e1917bfa6e',1,'operator*=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab070ea4676d10a10ff3e9379a4068a57',1,'operator*=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0197e039d4c65bf49649a6f250c2d436',1,'operator*=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad3565cc6fd1e088d052b1108aa065851',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a711693988c437c2fb4d7da505982fe21',1,'operator*=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeff4c28986f98c23de1df17043edb0f5',1,'operator*=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7dbf0c75df4817cb4ef8b60c417a89d0',1,'operator*=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a323a80492cd17a49e2c3dd18f8c8b5cc',1,'operator*=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adb465776d3868bda0525d632ffc4d129',1,'operator*=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a12a98d71d670b409b8065e0d61672d55',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5d00eb2ec2b0e15b2753d100694c45ae',1,'operator*=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1a2a683ff40490226eb1371fb905023d',1,'operator*=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4126fb7ed5bbb27a2332c543cf56a337',1,'operator*=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab092d9790ef20fc0386707530aee89db',1,'operator*=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abff1fd2439e31e6e64a3d2fdee3c7821',1,'operator*=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a625dcb133f1f953f263e6200399866c6',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08b6071245513e1726ec68e3b63edc53',1,'operator*=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a13aa79165ec87710e977f33fe0361e91',1,'operator*=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3796dcf819adb1ef8152f57ba63ff6b1',1,'operator*=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aaab79d0b4c9e9bdc059ace6ec58c5b00',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a0dd3893abc8986901872c8365ab1509d',1,'mlx::core::operator*=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a3cc5c154e4ad9a83ad43da8513146fdc',1,'mlx::core::operator*=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a600e77dbc72e78207b5f5dbf4b298781',1,'mlx::core::operator*=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a54833be1d44bc3adfc9ea218fc3685bd',1,'mlx::core::operator*=(float &lhs, _MLX_Float16 rhs)']]], - ['operator_2b_26',['operator+',['../structpocketfft_1_1detail_1_1cmplx.html#a76447ef141c8732d57421749fc81b236',1,'pocketfft::detail::cmplx::operator+()'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ae2adde594b5a4853f6bc78263a957d85',1,'mlx::core::array::ArrayIterator::operator+()'],['../backend_2metal_2kernels_2complex_8h.html#ad6af5c6c5ed4898b49758618e5aee189',1,'operator+(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a09c1a797eb7f43742578680899932f50',1,'operator+(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a551b970f73bb4a3b287653021d000b60',1,'operator+(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a43a225e7e548bb041f3a5d844faaf0da',1,'operator+(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8b6c3fd9d068a2159084359df8b9b449',1,'operator+(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0a5bfe15d95ba540795f4c25ebfa4f07',1,'operator+(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa415ce182fe7582d885fe633fc3527ce',1,'operator+(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a62f891b7dbba0000749cf338f594bedb',1,'operator+(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab43932322f81bf322aa1b0deeee9a987',1,'operator+(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acd15d46ea5827a2a39898ccbb8352eb8',1,'operator+(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a006763fae6e0577fc168ec9446f0f747',1,'operator+(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a12a47e8ac0be788edff57ae0a96d7830',1,'operator+(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af87dfa2122e9c76042dc41fb7f338a87',1,'operator+(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af2737d09c887ee8cd43fdeabceddbe82',1,'operator+(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a12ff4f38aa8474bf76770c7b8e3e18cb',1,'mlx::steel::operator+()'],['../group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c',1,'mlx::core::operator+(const array &a, const array &b)'],['../group__ops.html#ga7d0ec8d01e7cefa6a6b25f11876761b5',1,'mlx::core::operator+(T a, const array &b)'],['../group__ops.html#ga7cc080a4f9d4a667f2099aa0dbfefadd',1,'mlx::core::operator+(const array &a, T b)'],['../namespacemlx_1_1core.html#ac14b984970cafd8fbe24d080949515cc',1,'mlx::core::operator+(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab076069c6f0047c548a8dc29d35dd36a',1,'mlx::core::operator+(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aab9d96b0a168f4d05146000a6212b5d8',1,'mlx::core::operator+(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac4e6f03d7e4ae701b4eefa784f36185b',1,'mlx::core::operator+(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a4cabd600a5271b0d416c91e8d31dd9c1',1,'mlx::core::operator+(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af26df9dc279d71b7cc10892c72162b58',1,'mlx::core::operator+(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#ac3b97eecec9bd8efb313f8f201560343',1,'mlx::core::operator+(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2e3bb121cbde30c2e6d806df0d41ff59',1,'mlx::core::operator+(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ac87ecce4b44b0826e666a169ddc6f878',1,'mlx::core::operator+(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aed3d9cd32698ef0fe65b1280f103b3f5',1,'mlx::core::operator+(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6fa13b9359cf3f575fbda5260e6e035d',1,'mlx::core::operator+(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af240a6471ff827819192808bffeb857a',1,'mlx::core::operator+(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ac25a05679f312b724c406d8b282803c9',1,'mlx::core::operator+(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a54863a54f258acf2b5c734950618e4e1',1,'mlx::core::operator+(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a9f81f5ea8909db9660197217612ee446',1,'mlx::core::operator+(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a13e26c38da0a4e332e0ae4eb0aed9cb8',1,'mlx::core::operator+(const std::complex< float > &x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a59bb13a0bb7f748c8de34415b248bc57',1,'mlx::core::operator+(const complex64_t &x, const std::complex< float > &y)'],['../namespacemlx_1_1core.html#a38a44c412c8be4c8b952d3082cc7db74',1,'mlx::core::operator+(const complex64_t &x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a011dbdbd2413e59e744cf82b05431340',1,'mlx::core::operator+(bool x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a230e3b7c479add1b171fa0aaa3a8b13c',1,'mlx::core::operator+(const complex64_t &x, bool y)'],['../namespacemlx_1_1core.html#a3a6f43c2485f0d42293184f1aecbeaee',1,'mlx::core::operator+(uint32_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a766157c5d5d00fdf3da95eb7cb2981b9',1,'mlx::core::operator+(const complex64_t &x, uint32_t y)'],['../namespacemlx_1_1core.html#a64dceec2bb03eee963a2a1bc1ac69284',1,'mlx::core::operator+(uint64_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#ae36badb78a17cd7d13663a69645fc328',1,'mlx::core::operator+(const complex64_t &x, uint64_t y)'],['../namespacemlx_1_1core.html#ac1afa5d4c856e4b58109eff086e70ffd',1,'mlx::core::operator+(int32_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a8978def3c2cfe2a96314d564613b80db',1,'mlx::core::operator+(const complex64_t &x, int32_t y)'],['../namespacemlx_1_1core.html#a5b8af5ca4c0e37aba0b7530542bd64c2',1,'mlx::core::operator+(int64_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a3eaa72850205c18450c3af9a01cda219',1,'mlx::core::operator+(const complex64_t &x, int64_t y)'],['../namespacemlx_1_1core.html#ad38b38a3faf050735d45eed4438ee27a',1,'mlx::core::operator+(float16_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a358e66ff205bda3e8542427b6d2edadc',1,'mlx::core::operator+(const complex64_t &x, float16_t y)'],['../namespacemlx_1_1core.html#af56d4b85e329e39a825c01a50e3a2522',1,'mlx::core::operator+(bfloat16_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a806a495a129ebaab69cc57ca7db831d6',1,'mlx::core::operator+(const complex64_t &x, bfloat16_t y)'],['../namespacemlx_1_1core.html#a09fc6ebda917969383783a112a8547e7',1,'mlx::core::operator+(float x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a7ed0e2cdb65612f54e67166762cb6408',1,'mlx::core::operator+(const complex64_t &x, float y)'],['../namespacemlx_1_1core.html#af7577c91b8c43682f0ebc9eb9758aae4',1,'mlx::core::operator+(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#abe36af9951afd8dd3ffe90ceedeb7f2b',1,'mlx::core::operator+(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#afb9f780dd056a4f975518f71a3b021ee',1,'mlx::core::operator+(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a6a8e093b24c4c789b7cd160f7e7f7de9',1,'mlx::core::operator+(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#af3a603690fd3de9e4f7f2035a4d25621',1,'mlx::core::operator+(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afa2a4bccfeea9688ac922cb638341511',1,'mlx::core::operator+(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a6111e94d51de12391e5d68b765f28fc3',1,'mlx::core::operator+(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7c7dd6d346e0cdf398a896f2c6958258',1,'mlx::core::operator+(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a00872a443f462b0ae0a30c84fb001bc0',1,'mlx::core::operator+(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4f5d80d03bae6d8d90455d3c47a8c116',1,'mlx::core::operator+(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a78f1f388f9d81ed93f60311f4645d8d0',1,'mlx::core::operator+(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aa43e1d6958c5d5a6fa9a625a1660e741',1,'mlx::core::operator+(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ae877e1d5e3cf57734da8b49535fe3fb3',1,'mlx::core::operator+(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a9a5ae769f67f886d59c8e292a8218550',1,'mlx::core::operator+(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a058878237ce50baa4c909d8d15448d7e',1,'mlx::core::operator+(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a95fd207028f125eefbafe9e0522407fe',1,'mlx::core::operator+(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#abc6425a3fbb386f5ea5964b42507e989',1,'mlx::core::operator+(bfloat16_t lhs, float16_t rhs)']]], - ['operator_2b_2b_27',['operator++',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a3efe69356a84d0d4438f033992fcbd9d',1,'mlx::core::array::ArrayIterator']]], - ['operator_2b_3d_28',['operator+=',['../structpocketfft_1_1detail_1_1cmplx.html#ad4e69dcd89bdb7764c9c5807168f911e',1,'pocketfft::detail::cmplx::operator+=(const cmplx &other)'],['../structpocketfft_1_1detail_1_1cmplx.html#affa618d8850a7c232793b7c61db6d184',1,'pocketfft::detail::cmplx::operator+=(const cmplx< T2 > &other)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab04f480aea9fbba0895068c7558dd400',1,'operator+=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a251780ac4592cc2b1a543e417ff57770',1,'operator+=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24381d991c2d570aa953694f396a69b5',1,'operator+=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7595740d4cc12924905d6bd1b99ee4da',1,'operator+=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac1498acb8c3623b5f412f70ab6a6528b',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abce5ab327110c164f054b43ed47f79a0',1,'operator+=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae0c70198e236ffe1a98f79987c686419',1,'operator+=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a13b3338935440ae51ecc4a356093efc5',1,'operator+=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a0cb8544b4ebd2906ba8e7f2868e8de',1,'operator+=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7b134429ea0c8493800ff8b465410f9c',1,'operator+=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4154f90ab7857ca856f9e15fe1bf5acf',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab9ae6a51e2027b02cac9966e05f3ba68',1,'operator+=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab93ce536eb7998bee00de4af868e31a9',1,'operator+=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad0ae9e2b4874f991a2c853e1c1fe735d',1,'operator+=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a194a6670cc25ade35a24b566f31af785',1,'operator+=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3d0d689516c99003659c5d026847bd2e',1,'operator+=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a007f58508b98bb79e5c323ed0dec89b6',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa7198e580e2a83c1fd01a4b6fdf86a80',1,'operator+=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a15573fefd880adefbba079b1c1bd8082',1,'operator+=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a104cf94cb9e359d1b6ef92ced2ce0c27',1,'operator+=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa04cfcb52191fd23205a1a3572b46ae0',1,'operator+=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad684bc2ae1a2a627cd3e4a4c641e2d77',1,'operator+=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1e28448e35f4934075b397c34ba3d66',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8ad16afd7f1711de83c0cec5af868f76',1,'operator+=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac45e9ca0c7155caebe3d0f7261518077',1,'operator+=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3c62ac679d6aa515144d40ebafe4a188',1,'operator+=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9ff5ab3aef1057fa083b53a65c8aba03',1,'operator+=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae74bb0a3c12cd1a23f3d29ce307d6fb1',1,'operator+=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac188bd19f236b098d603b0d8acd08921',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef9fa600d107b509f2e3df7d6b080e01',1,'operator+=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af5713afb3a62967a02c3c20661951ee4',1,'operator+=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7f1b84352a3ed6171444a43da1fc7e92',1,'operator+=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af1983edd26245e6e51c6e47354095e32',1,'operator+=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8cd55d1a579540eb450e12a8a8a950be',1,'operator+=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a588ef0f7e03f306758524d378278976f',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a74751abec7086f85f4f26ced44f1ca1f',1,'operator+=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4dd3cf0e5aa116ff330352a50c18cde7',1,'operator+=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afb9a0e18c0e40c77e6143fb7d84ebfba',1,'operator+=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adf0cfd9a608a6fb3d57933e32e7d81d2',1,'operator+=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4bd92db6c8b9b5dc96332c7ae3eff8c7',1,'operator+=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5d628a5bc4fa755610392f47a523a1f1',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7c790442f77f2437b482c4a55e224fc3',1,'operator+=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a77bab4481b41be50297b257e95058706',1,'operator+=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7816a97d16b1d2f8a90227bb1da2f6ac',1,'operator+=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac244d140c6149726ea44174d3e836ca3',1,'operator+=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af802541c4c65ee4442acd495de4d27fe',1,'operator+=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac06eb2fea47a09a8a8abdaa1aa9b4603',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5263b2463fecdc97f9521d00bffea059',1,'operator+=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24ca436ab299a710263d65302532dd3b',1,'operator+=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee1bdf0ab2e445293708b476e8cfde3b',1,'operator+=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a190e27077f0fba642a86f5c8f488bcc2',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a9f2c9d2f21fbf9fbbacd940c6967c9d1',1,'mlx::core::operator+=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a0b1b3c48afc0a785282e43435bba8418',1,'mlx::core::operator+=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7b763db8194e6fcb1b87eab143dfa47a',1,'mlx::core::operator+=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a827167f6a1ae55428fd218ddd51ec3b6',1,'mlx::core::operator+=(float &lhs, _MLX_Float16 rhs)']]], - ['operator_2d_29',['operator-',['../structpocketfft_1_1detail_1_1cmplx.html#a460da5db36d1c72fb1ed3496fd3abde4',1,'pocketfft::detail::cmplx::operator-()'],['../backend_2metal_2kernels_2complex_8h.html#a226cfd54d49f02e35c5aab3139c7596b',1,'operator-(complex64_t x): complex.h'],['../backend_2metal_2kernels_2complex_8h.html#af5608264cf920688607059b4e8cd3117',1,'operator-(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6aedc8d6d0980134ac69b96f22d9a855',1,'operator-(_MLX_BFloat16 x): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a333f67614dbf8027439a7e124052cb85',1,'operator-(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a891aa4bf46c20a26a55061736aba25f1',1,'operator-(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7ad7ff44a3200853711869f7a577d931',1,'operator-(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af69ef8f1d8ecae0e6f755bf1c46cf075',1,'operator-(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5bd875a54b79b2dcedf674807c3e53c5',1,'operator-(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab02f8646b47806e1d2038f248df03f06',1,'operator-(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab27b26182c7c6e08af37e6d511fd9253',1,'operator-(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5868c85c988ec3432cf86d7df40e464d',1,'operator-(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad03ef47e6cc7521bbfb45740dee20f88',1,'operator-(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab789f8a400512ff27e36b3373170f0c5',1,'operator-(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7f601b22ecc480132d82ad782e5363bf',1,'operator-(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a152366ab4e2ccc867e919af6c74ced91',1,'operator-(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a42bead8ef0beb9f3452128d64cd4df9d',1,'operator-(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aca8ef21c16984ccb329b3bd0c1e4be48',1,'mlx::steel::operator-()'],['../group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816',1,'mlx::core::operator-(const array &a)'],['../group__ops.html#ga0c7f3cb36d4ca516c7a33142f88b9181',1,'mlx::core::operator-(const array &a, const array &b)'],['../group__ops.html#gae68d3d0691ba951501218e98439f3465',1,'mlx::core::operator-(T a, const array &b)'],['../group__ops.html#gaf5e5d882c51ad0a0ea315c274d5439b2',1,'mlx::core::operator-(const array &a, T b)'],['../namespacemlx_1_1core.html#a622ce842fe44e4b6a95e03242341b459',1,'mlx::core::operator-(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af32a99d930d49e9b178472d7a65531ab',1,'mlx::core::operator-(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a3555a2b31fc0925850d3240e85e03ec5',1,'mlx::core::operator-(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a46080889fd9e5c3f9916508e97dff5ad',1,'mlx::core::operator-(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a9ca27fd1e512c8ed126342e565da12ae',1,'mlx::core::operator-(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3803f8d36558d32bb7dd6e580ea683b4',1,'mlx::core::operator-(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#af5d865528989ca66b3d357e5ce4e0300',1,'mlx::core::operator-(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#afb784b960f55aeb4edd7f567fa74d443',1,'mlx::core::operator-(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a29cbacf4b399c24728fb0808fad498f9',1,'mlx::core::operator-(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aececc0e451237aa6c0d1a2c3d828c86e',1,'mlx::core::operator-(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a383a26cc2689c98fd6c4435ade8dc669',1,'mlx::core::operator-(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad6311ef8df59bdfb212b5cf8169246b2',1,'mlx::core::operator-(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a23b7329bc1c93c8ac0a1f576565fefb0',1,'mlx::core::operator-(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad8d650bf63998abd716ee0ca28e1cbb9',1,'mlx::core::operator-(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a7339b33201254e9119d99d3a728ded72',1,'mlx::core::operator-(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a064318b7a16e5cb6d0a6407501b5c7dc',1,'mlx::core::operator-(_MLX_BFloat16 lhs)'],['../namespacemlx_1_1core.html#a7bae3ff296d9a60ff3c7e448f7fbc6bd',1,'mlx::core::operator-(const complex64_t &v)'],['../namespacemlx_1_1core.html#afb5069ecebdfd9d388c26f83df12c93c',1,'mlx::core::operator-(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a8d126e3f3fa9f8c1c1ae1b09f94df487',1,'mlx::core::operator-(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ad04f1ccd2cd7c487a2f2aaa055939f64',1,'mlx::core::operator-(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a15eb2ea76508ff823fa0591e811d0b7d',1,'mlx::core::operator-(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a96d9577db38d6809d022893e32feeda1',1,'mlx::core::operator-(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a5d9c02765c1672930757416411567bf2',1,'mlx::core::operator-(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a6105d3b5266666b7c6bb9469285a9ec3',1,'mlx::core::operator-(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a777aa772dfb205b25d26f3180d98a2f6',1,'mlx::core::operator-(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a085eb092f4ada47f8169de62886cff90',1,'mlx::core::operator-(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab25e5d211e2c8785b45c3a81a6282e2b',1,'mlx::core::operator-(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#abf5d09561a81b0f0b32d59d77e32e16f',1,'mlx::core::operator-(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4ce6867dbb4d1631d1870dac14022dbb',1,'mlx::core::operator-(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a8a049e646e0442064cfe9e202d7047c5',1,'mlx::core::operator-(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a78e2a1cfc65453185bcca13bd4f523cf',1,'mlx::core::operator-(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#af143cf68673e06390d4bb2ec2892bd22',1,'mlx::core::operator-(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a46d502dfe0b027955950d4e716c2eb26',1,'mlx::core::operator-(_MLX_Float16 lhs)'],['../namespacemlx_1_1core.html#a2631e78c6f0a602f6754ac577ec75f83',1,'mlx::core::operator-(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a73d79cbd75d543d0837b8a51bf103f9e',1,'mlx::core::operator-(bfloat16_t lhs, float16_t rhs)']]], - ['operator_2d_3d_30',['operator-=',['../structpocketfft_1_1detail_1_1cmplx.html#a12441ff423274bd1b54245933d69ad7e',1,'pocketfft::detail::cmplx::operator-=()'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab225043bd02bb423930bc98aae9c2bca',1,'operator-=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac2f1e1f2365cfa531b1519aa9ff67695',1,'operator-=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a513501355a5912a1263fd8b10864142b',1,'operator-=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab4f4ecd62c3d8b3363d02019573dc9f1',1,'operator-=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a92d1348f201d78fcd474f75d5b23ef68',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3eefe9a7f5fb226335ea687012f32d5c',1,'operator-=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef62c7e3e494b6a511a7833c0d942a60',1,'operator-=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad30726cc8b69fd300d33c2a46e123c28',1,'operator-=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8859b5b8dc241e4f58243c85d2630cc8',1,'operator-=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7003e1e5881e3d106257f22b6a3e59fe',1,'operator-=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3165e37d393be50c2cfa9ddcba153684',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a76f5bd895b7214cbc3cea3440992718a',1,'operator-=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7167343d90eb70e5a0d5fa9ec5398e94',1,'operator-=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9b31c363ebc93d592b6fa0e27b00335a',1,'operator-=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a967a1d7b5664f616e5b6f2d257367f0c',1,'operator-=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aff19193e1b2cee29a8737318e95cc74a',1,'operator-=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aede0cc4179507b739849948f1a2fed4b',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7e1a6056f9c96f3c89fe204dbf103be5',1,'operator-=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9d06cceea5c179bcc608452188bd7d6a',1,'operator-=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0aa9ffe056f49fda181bbacbd60556ea',1,'operator-=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ada5685d99c2d6708d1c4ef826d68e879',1,'operator-=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a726cecf778b8584b6f7c37db1b064576',1,'operator-=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3816a35f8468156d59c239256c12dcf3',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa332fae098e7c6dc23b98bc0026f1070',1,'operator-=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afb3cd302e0b78902c62111dce4494fe8',1,'operator-=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abb884888f14086cc674657677cb4b8bc',1,'operator-=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a38bb89f925eca4f9c042f6ee7a2c0193',1,'operator-=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac30c580713f354916088a7dc049ae4cd',1,'operator-=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a715c824ee8c87e0256114a85624d9949',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7bc91aaaf476a37063264d1d53d862cc',1,'operator-=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab155f418f15cabd86ff942c6f9472ddb',1,'operator-=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aaa66dc6d7b2c5efbfaa97ca9c7872bd8',1,'operator-=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a696978d9401e09200045b2d8aad045c2',1,'operator-=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae998d8f423a9fb73405cfbd4b836bc72',1,'operator-=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a279d09ab8542f1c1a8dc8173b65946b6',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a491dadfae957cd7cc0c36188d910f6f6',1,'operator-=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9a837c3b9c4e42f53d7cd1ed0d266e2f',1,'operator-=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acf7af2284269544064b68e807064bba4',1,'operator-=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a28d297705e29009197418546ef435393',1,'operator-=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a948579a4d9ba276523190b03b09578fb',1,'operator-=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a4b98a0a11db5b77cf9168df37c8bc7',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31a3d8f2ff8038f7e0d717845c039808',1,'operator-=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1dac193d9f1c8c0eb4473441895f8c58',1,'operator-=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad2817d53fdd4b112babfb6f0b38c8f39',1,'operator-=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa705d87cf4b78e9d7c6b07dd0c66cac6',1,'operator-=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a542affc376726840647a6e93acf2c1a7',1,'operator-=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#add18cfe4c0d38e95c6dff6bab3e7a932',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab1de7e7e7304ff3598925d2e69134764',1,'operator-=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0d3fb52437c677c5d0f1a3642384b15c',1,'operator-=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adda64cae388baac1f138b06dc8595237',1,'operator-=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af20874a61c6c3f4c3fd045a96e806644',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a8b8a55690df46d97fcfc2a60120783af',1,'mlx::core::operator-=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#ab03949b1f60fa035ce454a894cd73ae9',1,'mlx::core::operator-=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#adaf70bbfb3667df0d08fd3c99896e20a',1,'mlx::core::operator-=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a321c98e5a78621d3c9a3895f707f2f1c',1,'mlx::core::operator-=(float &lhs, _MLX_Float16 rhs)']]], - ['operator_2f_31',['operator/',['../backend_2metal_2kernels_2complex_8h.html#ae6a708f67d6fd9b0962aa8877cec6d35',1,'operator/(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9f16a44e1c9836ca57edc1d7b93b5d7c',1,'operator/(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aacaedf12f862c76457133336dd6fc446',1,'operator/(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a584a513596de20663dad951a5b81695e',1,'operator/(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad8f7b11669736fbd6ed2e28211d877d4',1,'operator/(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a59515695ebc48844345fa5120511aed1',1,'operator/(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8c8ac6736440fdca366ebdefe2a12b9f',1,'operator/(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad6859b04680d0d26d75fd6c4dd74ee24',1,'operator/(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4720cc79ab2b8e39952ea9ef20e51250',1,'operator/(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a72d10ec0e62949247da129eb3a83fb9b',1,'operator/(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad6399ba2b8708899739b4cdbb44add8d',1,'operator/(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a998b1ba877a606aedf722ab46b290403',1,'operator/(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa3277ae33976c70f7bd937ddff027b72',1,'operator/(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa708a970a200822c99c0489f389469fa',1,'operator/(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a6bde717aca2051499f73a3eee199bfdd',1,'mlx::steel::operator/()'],['../group__ops.html#gaeedf77f722b394429f1a7f6c367883bf',1,'mlx::core::operator/(const array &a, const array &b)'],['../group__ops.html#ga7366ec7f453be2a4dc449f0faa1bf554',1,'mlx::core::operator/(double a, const array &b)'],['../group__ops.html#gadfb324ae9b4feb2c7ea0ac6ade639f38',1,'mlx::core::operator/(const array &a, double b)'],['../namespacemlx_1_1core.html#a7573ac3b93ddecd69e9c88a26fc84ba9',1,'mlx::core::operator/(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a40e868dad70401d9aa9ee9c32235c315',1,'mlx::core::operator/(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a7587c28fbd2023b134e5fc12bb0dde23',1,'mlx::core::operator/(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a92cdd377c408becf4cf83c1ee9b7085d',1,'mlx::core::operator/(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aef89566301cb133d98c8e7bdd2b7bec6',1,'mlx::core::operator/(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a94e7b51185590492b46916685641276f',1,'mlx::core::operator/(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a04584788c08180835219d0ea1e2b97b1',1,'mlx::core::operator/(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad5af96e2ff09d207eb1e1980fe3e7c2d',1,'mlx::core::operator/(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ac2217bf760038cd011781158923149ed',1,'mlx::core::operator/(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aea414c04bddc4b9b609262e97398f1b4',1,'mlx::core::operator/(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a27fe23230cd082c0363b9451b731ce6b',1,'mlx::core::operator/(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abdd9bb8fb4411e5924f3eb7ef1bb52f8',1,'mlx::core::operator/(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a50bae338a7353f8b0ed3441071bb0cf6',1,'mlx::core::operator/(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aab26a3284dd3ac7d47c8b5b3a3290ce3',1,'mlx::core::operator/(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a749f48db01de38f259a0c6750a97fa77',1,'mlx::core::operator/(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a32a6a08a2a4652975b0a1bd1fcf3eafd',1,'mlx::core::operator/(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4b66fb38ddc5cc0c2489583d5c499602',1,'mlx::core::operator/(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a45726f1905b709cf8253e6efa046027b',1,'mlx::core::operator/(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afd4170c1e364384f30e6bae341146fa6',1,'mlx::core::operator/(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aef85739d150b9d5609973da8a3f1086a',1,'mlx::core::operator/(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af52a941f8ed9b25eec91402c7b9e281f',1,'mlx::core::operator/(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a477cade78296bc85894170f62db68870',1,'mlx::core::operator/(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a22f5a2257e11423fc2fe18e2dce91590',1,'mlx::core::operator/(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a640d3574dfe6ad934c720ae8bdd78bfa',1,'mlx::core::operator/(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a6f65d8fd0cdddc96fc01f6af95804873',1,'mlx::core::operator/(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a517019d42d4e426b7b98e1c719bb47ce',1,'mlx::core::operator/(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a0beb7a223c542015a4eff4aed814a9dd',1,'mlx::core::operator/(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#abc9b1bd5018d46514bc19d23db2e5063',1,'mlx::core::operator/(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af22937df654ddbd6e398ef12764d18c0',1,'mlx::core::operator/(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a775aed5f49b530c57e71cbac81404d45',1,'mlx::core::operator/(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a97efcd96d6be666e5608034ae77289ef',1,'mlx::core::operator/(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a899851f85dbddd96f9d36319b82542a0',1,'mlx::core::operator/(bfloat16_t lhs, float16_t rhs)']]], - ['operator_2f_3d_32',['operator/=',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5aa3b8c68a2b58d41ea33eaabbf83095',1,'operator/=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a90a1c5130db515db48624d8587edbb91',1,'operator/=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a65f30a2dc199134e35bc7c5d431b2263',1,'operator/=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7172d84db640e6c49dff0d08dd64b53e',1,'operator/=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acf7cb9927bf09022088401923f2e1916',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a86b2a001cbec0d3a8d762a3c7ff47b0b',1,'operator/=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a744f72ba83522fe3cc2a49a007b42543',1,'operator/=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a77c678665b34df7652dcde053ca73185',1,'operator/=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae0614b6b199d8a65ae95d4621b118b82',1,'operator/=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa846fde89c7d2d18b18ef180a8a9c8a3',1,'operator/=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08e778be18e4a291c108fcc528b981d3',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6b9e49ad9ea256d2d0220c0d81552602',1,'operator/=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab933bc3cdf9adfea10ab9dba5292c812',1,'operator/=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a25e7c5d2ecf3375756d59074f333858f',1,'operator/=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ae4a80fde67eea9a0a37b2803946544',1,'operator/=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a912393b7208fa45bd1e87f30b218b68b',1,'operator/=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a18963246f2b640874bef6dca7049f64d',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0e2c2c2cb50b3a55ff213f18978aca35',1,'operator/=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a64f1136b17006f168ef837e17240814f',1,'operator/=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae46d75b8046d557452d74513f1106710',1,'operator/=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08d2460e259b9106d90d889481ad60d5',1,'operator/=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f7fd418408806ef498745c6fdb2c062',1,'operator/=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac703495cb370b52526a5a2d36ae26038',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ca11d43174baf0a729f93b35eabcbea',1,'operator/=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9f835a0a80c411580c97b65fdc5bdfd3',1,'operator/=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a17f47ec9cff60f8e1b3477a2793b7ac0',1,'operator/=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5be23e296bbed3a885586a6424b1666e',1,'operator/=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afba39221eb54e272aae79910b3cd7ef5',1,'operator/=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac057d95a2bf087575584aa6f9a2c6bf5',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab986ae2cec780a1f494b7b4468b7ba11',1,'operator/=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a44522c2304c6396bbe6b9d32000f4b6f',1,'operator/=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef8e7e499ea9d432aa743d83c076f945',1,'operator/=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3a0a3edbf1ba2314551454059c3f422b',1,'operator/=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acb9f0aef9fbdfde8a4f46e33b0d6c52f',1,'operator/=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a303dfcc81ffd355f866f863d7d9f0fa5',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a359edd4bcb8776861ceb26a3005624c0',1,'operator/=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc9f32cc6f40768df4285fba2e4783c7',1,'operator/=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae71f66d814a03f6377c9d86cf0a2b5d7',1,'operator/=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad0125b6baba3065a87a174ec27aa9a61',1,'operator/=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5cc74ad3e522d7104e6e2117751151ad',1,'operator/=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab3b594321fb42b0c2da99954d1e0976c',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4a0023e2fd08875156cd6ef747fbb5cd',1,'operator/=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4358ee606e66ba2081fcf94f9c3b5915',1,'operator/=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1e7ef6f065695d4b1d017547b60ef62',1,'operator/=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a284dfc702f0f67b9c233b87162eeabdd',1,'operator/=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab8f211ea896fc5190004f3ad6ad8932f',1,'operator/=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7e1bcf3bc06cbcbc304c0cdf729802bc',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abbe42648a46092137b303ccd08f7df86',1,'operator/=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af1a12a1efb618a57da6dd41ae18cb53c',1,'operator/=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a94686039356dfa9aa45608a8b0562fdc',1,'operator/=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa251d6483d3b099d1b5311fbe6f0bce2',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a045ff27257cb6d8ab7a94771ba5a17e6',1,'mlx::core::operator/=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a58112951a56a0f9f8c90b60fe74f9508',1,'mlx::core::operator/=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae736defc89a04fbaf7627ad2695bb838',1,'mlx::core::operator/=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#ab1f260710251256ef737dd59be9e143c',1,'mlx::core::operator/=(float &lhs, _MLX_Float16 rhs)']]], - ['operator_3c_33',['operator<',['../backend_2metal_2kernels_2complex_8h.html#a67674e32596a9dae2258bb8e0e6a2058',1,'operator<(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9ef6a57b7185e9ca49e255fec1a44e25',1,'operator<(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aab02c65bc38ea66335b2192ead4095a8',1,'operator<(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae91686513e284bcc9635833744bbdda1',1,'operator<(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2486f3b5de85b0d57f458d8f21f82b42',1,'operator<(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a435a2aec4c777b4b184ff5d24992e8a1',1,'operator<(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abdd04257e6a73883b5f56f1186d0e906',1,'operator<(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a69984aaa05ae1d4fccccf7f57e8ecb4a',1,'operator<(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a501cc01d5bf15d9f03aa28545f9624ea',1,'operator<(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1b029e4ca72125a5f9471f582c819705',1,'operator<(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0736a76f56578d26ba1422dc8b744a18',1,'operator<(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24b1fa8998c892f90f8dde7c34fb10a5',1,'operator<(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af80ff2020ec2c4b406c5fdae3fe55e63',1,'operator<(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac03f6eefb836373d37dc280b0d813d78',1,'operator<(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#adb5f24b57d98214fc215a06475f21412',1,'mlx::steel::operator<()'],['../group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3',1,'mlx::core::operator<(const array &a, const array &b)'],['../group__ops.html#ga1ef8ea11cf15ce628c54201fa42748ef',1,'mlx::core::operator<(T a, const array &b)'],['../group__ops.html#ga95e72226dc7a79c40b3d16f990922050',1,'mlx::core::operator<(const array &a, T b)'],['../namespacemlx_1_1core.html#a987d631e1508e8df55d98ddd57e4d086',1,'mlx::core::operator<(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad3fb46370cd8f0992866fad9e2c64a3c',1,'mlx::core::operator<(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a3026691bf7ee5095243a8611bf3411aa',1,'mlx::core::operator<(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0d42d6c1d5f77a96e2f296b8ebd79ee6',1,'mlx::core::operator<(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ab5ce08a7de0a0ca00d61f7a7f8ea3ab4',1,'mlx::core::operator<(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abce8b7f24b61e5ec0f9a3afe20845caf',1,'mlx::core::operator<(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#aff97612627ae1ed260c43c0a7af0d306',1,'mlx::core::operator<(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a9119e518234df7923cae2b3802d59bf2',1,'mlx::core::operator<(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#aefb9b05ce8864ada99a920ab32017b89',1,'mlx::core::operator<(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abc55f3676c2d112a6e9ab276bd6b1796',1,'mlx::core::operator<(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#afe6581a2c45f24d7fab1e4006c1e3c70',1,'mlx::core::operator<(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aca1d50cdd9506481dcc4cd1ad4a4f734',1,'mlx::core::operator<(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a310720f513b6a2490e9df80c65f1bfb3',1,'mlx::core::operator<(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a29e457a170b6cefb6ba1e394c96c6f7b',1,'mlx::core::operator<(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#afd4519985b6b207ec41ad8530d1036df',1,'mlx::core::operator<(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae1e41ca94022e43a00cdfc5845102daa',1,'mlx::core::operator<(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ac80f4022bffd95b57526685ce8e1cbc1',1,'mlx::core::operator<(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a3a8f6f0af477788c4f0aa98abfc5f1ab',1,'mlx::core::operator<(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a3728ed9b6cbd152bf675251a0501b466',1,'mlx::core::operator<(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a5b9ad811a5e1358100c5423dd70ea387',1,'mlx::core::operator<(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a5c77e1db83995d3e06a8a26265bce5d6',1,'mlx::core::operator<(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab8a0a3f70664049b35ce1887bd8ff5c2',1,'mlx::core::operator<(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6652d93bfb2d426e261a1712a181a4d2',1,'mlx::core::operator<(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a03758b8d13da2de07cc4f4fc45d2854b',1,'mlx::core::operator<(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a325161b81a9ff179fd37d949780a17ba',1,'mlx::core::operator<(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a92eca79fce8233e4299343eee3996511',1,'mlx::core::operator<(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#adb016662b8f7eb680abfe1a421eabe72',1,'mlx::core::operator<(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3c_3c_34',['operator<<',['../group__ops.html#gad656c30f9fd7d9467e405657b325aa7e',1,'mlx::core::operator<<(const array &a, const array &b)'],['../namespacemlx_1_1core.html#a1e5c30e316afa30c14bc48b92afdb794',1,'mlx::core::operator<<(std::ostream &os, const Device &d)'],['../namespacemlx_1_1core.html#a4ddd07021b36c848d6fb1dd9ac276822',1,'mlx::core::operator<<(std::ostream &os, const Stream &s)'],['../namespacemlx_1_1core.html#a0023c267cf81345fad65e7a797954cd3',1,'mlx::core::operator<<(std::ostream &os, const Dtype &d)'],['../namespacemlx_1_1core.html#a1fd58658474fb842d648dcf8f7d9f078',1,'mlx::core::operator<<(std::ostream &os, const Dtype::Kind &k)'],['../namespacemlx_1_1core.html#a123331f01188bd76e37623b63b6b4340',1,'mlx::core::operator<<(std::ostream &os, array a)'],['../namespacemlx_1_1core.html#a4e733bba89760abed32393e085812b22',1,'mlx::core::operator<<(std::ostream &os, const std::vector< int > &v)'],['../namespacemlx_1_1core.html#a5e5bd5c57b1cf19776bdb41e732861d9',1,'mlx::core::operator<<(std::ostream &os, const std::vector< int64_t > &v)'],['../namespacemlx_1_1core.html#a42a19c8442b173606e714364227e7d45',1,'mlx::core::operator<<(std::ostream &os, const complex64_t &v)'],['../namespacemlx_1_1core.html#a57eb97a5eba99a846ac429795e407574',1,'mlx::core::operator<<(std::ostream &os, const float16_t &v)'],['../namespacemlx_1_1core.html#a7db909d54cf07375e89424c32c07a29c',1,'mlx::core::operator<<(std::ostream &os, const bfloat16_t &v)']]], - ['operator_3c_3d_35',['operator<=',['../backend_2metal_2kernels_2complex_8h.html#aee04c9a63c6716a99a027418354debb0',1,'operator<=(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af469c58cffeab488c681f4b33f02cd05',1,'operator<=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a81eae168dfafd299c2b94e3e8558cf',1,'operator<=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f486bf02c6ad5b9b6a96d3450f03e47',1,'operator<=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acba9efe192d22b7781b4622103c7a944',1,'operator<=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aff100489cc40ad276c2d5d67a9df67db',1,'operator<=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7eac96f64ca42991caf819c8e8c8d2bc',1,'operator<=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a88c11cd37600de5480570da3d2ae5732',1,'operator<=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08c7d12a0d16565fbf052dba2db8b22d',1,'operator<=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2b9de9624c0a507b4ead85f898ad9daf',1,'operator<=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a28f8d21c5eef047c701cf690ce9c2ef0',1,'operator<=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a14b56c687053ee2432398a25663c068f',1,'operator<=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f360806708b95a3be400af0b8871b57',1,'operator<=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a80d288f22cadfdf5e904410349e616a1',1,'operator<=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a6cc3bab5e7f6e7c719c82afa90ad2827',1,'mlx::steel::operator<=()'],['../group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6',1,'mlx::core::operator<=(const array &a, const array &b)'],['../group__ops.html#ga150a9be467c9f91482a6d6fc13504bc4',1,'mlx::core::operator<=(T a, const array &b)'],['../group__ops.html#ga624eeccef0cc4b130e1325abfea057cb',1,'mlx::core::operator<=(const array &a, T b)'],['../namespacemlx_1_1core.html#a0066a47cb21223ddebc77992ee874fb9',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2593dbace3ce50e7146d9514726a543f',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a88654bcf6c9728517a2933ca2e29a7c1',1,'mlx::core::operator<=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a5d4f449e9c1699b99fcf894dd15e8af3',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a6b678bea8fdcda1f11c6691b56a15211',1,'mlx::core::operator<=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae8aacc606ea16f018a90eae758830a35',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a25668dea4ffb51c7c00eeecb9530d1d8',1,'mlx::core::operator<=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a084558b6a5487549799c49c37c9e9652',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ade2e2a0daa79d5c52f278f85f03dde2e',1,'mlx::core::operator<=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a750a2d2b4976ad94b08994d081f83445',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ade5a175ff45347689ac4c798d04c8ffc',1,'mlx::core::operator<=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae25e0c01b46612f039313a4825ba6428',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a5c90f16d8f6edf4b75c96b945b9fa591',1,'mlx::core::operator<=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a8cd6583fa0fc9957f993e00b2ec01d91',1,'mlx::core::operator<=(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a012130a0458cbc30b88365e0e0eab232',1,'mlx::core::operator<=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae8c890bdcffadee8c5dab85c907f57eb',1,'mlx::core::operator<=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a43cb070553c1f2fffb32ef6670e30980',1,'mlx::core::operator<=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ac759b7798d668a99535e59e26d6ba192',1,'mlx::core::operator<=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a70e528a789b5660d98e783b045aaa379',1,'mlx::core::operator<=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a40bd8abb8a4d989ddabbb298518bd7f5',1,'mlx::core::operator<=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a4155d4b0c76f37ab5e0b54f9cd683f35',1,'mlx::core::operator<=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad8bb648d0603a206e0392990c911ca0b',1,'mlx::core::operator<=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ace72a5853f2afd6510dcb97d54fa650d',1,'mlx::core::operator<=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab38f7a0d3c0809071ff5d3af859018d6',1,'mlx::core::operator<=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a7904b886d7b535a6af0a885d00597323',1,'mlx::core::operator<=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a57952168bd0b54c2677204d4ab1cb6e5',1,'mlx::core::operator<=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a6235dc5f4db517618bb3449b08c96e8b',1,'mlx::core::operator<=(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3d_36',['operator=',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a027b84cddc8d476f736ac1f1a9991fe4',1,'mlx::core::allocator::Allocator::operator=(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a2e971b47339b1d0849a334a902a9df3c',1,'mlx::core::allocator::Allocator::operator=(Allocator &&other)=delete'],['../classmlx_1_1core_1_1array.html#a8acf2b4c75f9b7f79da6675dbc36cf36',1,'mlx::core::array::operator=(const array &other) &&=delete'],['../classmlx_1_1core_1_1array.html#a5c89c2406a610b32943955f9a5060fbd',1,'mlx::core::array::operator=(array &&other) &&=delete'],['../classmlx_1_1core_1_1array.html#ad3277ff68f1336aa217f9cbe40181479',1,'mlx::core::array::operator=(array &&other) &=default'],['../classmlx_1_1core_1_1array.html#a5da41aabecf4c8055b7515341bf57147',1,'mlx::core::array::operator=(const array &other) &'],['../structmlx_1_1core_1_1array_1_1_data.html#a68e9417954fe811b5e41e6317a526748',1,'mlx::core::array::Data::operator=()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a3f42a1362b4a513fa89e7b3dcc570a8e',1,'mlx::core::metal::CommandEncoder::operator=()'],['../classmlx_1_1core_1_1metal_1_1_device.html#ad1d6382fd18a46b1906e1b43e0bd2e73',1,'mlx::core::metal::Device::operator=()'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aef97dbbc755940789f99a26164591c45',1,'mlx::core::metal::ResidencySet::operator=()'],['../structmlx_1_1core_1_1_function_exporter.html#a7ec0f53eb2783d5b1953be612e36d5c7',1,'mlx::core::FunctionExporter::operator=()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a957211656a13b4c0d126989a9aba3e25',1,'mlx::core::io::FileWriter::operator=()'],['../classmlx_1_1core_1_1_primitive.html#a6b1be7ea92f3a7bb19875c70259dad6b',1,'mlx::core::Primitive::operator=(const Primitive &other)=delete'],['../classmlx_1_1core_1_1_primitive.html#a50bbddd43e1ba0cf5f127cd7aa756a9e',1,'mlx::core::Primitive::operator=(Primitive &&other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a0a859309a4f192f2679e07f2e4ff4d22',1,'mlx::core::UnaryPrimitive::operator=(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#ab90b2ea80f1d914be03cf44def5db5a5',1,'mlx::core::UnaryPrimitive::operator=(UnaryPrimitive &&other)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ab170dbd2ce34c51e2eeebf5d08e7e2db',1,'mlx::core::scheduler::Scheduler::operator=(const Scheduler &)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a035ea35f4dd8ee985973080f14029379',1,'mlx::core::scheduler::Scheduler::operator=(Scheduler &&)=delete'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#a0f65b0523b8ddd989f338da6cb2860e3',1,'mlx::core::_MLX_BFloat16::operator=(std::vector< bool >::reference x)'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#abb8cd44ee22b17c55333ff2eb4e13a14',1,'mlx::core::_MLX_BFloat16::operator=(const float &x)'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a608a099bf7116ee608dcfd31ea3ade2c',1,'mlx::core::_MLX_Float16::operator=(std::vector< bool >::reference x)'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a35543c3653d477c46350697fb808373d',1,'mlx::core::_MLX_Float16::operator=(const float &x)'],['../structmlx_1_1core_1_1_command_encoder.html#a3f42a1362b4a513fa89e7b3dcc570a8e',1,'mlx::core::CommandEncoder::operator=()']]], - ['operator_3d_3d_37',['operator==',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a1afd6d2a19a2b0d712063f221ab4eba7',1,'mlx::core::array::ArrayIterator::operator==()'],['../backend_2metal_2kernels_2complex_8h.html#abfc19f03616441245dfc7726b278f190',1,'operator==(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a49a13b06a325ed3cca4004b6a0cde065',1,'operator==(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0aa3bfcfab53700488e5f386e6de60d5',1,'operator==(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3936148781ab1c4f33f58d12c116f370',1,'operator==(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae753526b669fba27771089dc809abd66',1,'operator==(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a05a4f197a71d0f16879032f44492bb79',1,'operator==(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae86f5917847b1ec9f313996250f2e0be',1,'operator==(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aab74ec4d33a64b92b908717d500f1ecf',1,'operator==(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac30a2c1fa6f172af903fdeb6a8632606',1,'operator==(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab4e9ad547aa23daa351075e0ecc58fa2',1,'operator==(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa5fa1a8f2b39c3508fe38205469756d1',1,'operator==(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeadc1f36c6bdc219294ce9341d80afa5',1,'operator==(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3ae2091ada1e39e857fbc53c97bdb79f',1,'operator==(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac7b4d295f3c7b1e09964f24f306422da',1,'operator==(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#abcc797f27e87e857b41c1a8d33ee2c78',1,'mlx::steel::operator==()'],['../namespacemlx_1_1core.html#a937503d72b66c661bf3f5fdcd98ef97c',1,'mlx::core::operator==(const Device &lhs, const Device &rhs)'],['../group__ops.html#gaa30cf69f3d22f65615f5e1696dd5703f',1,'mlx::core::operator==(const array &a, const array &b)'],['../group__ops.html#gaf115782d009ac2a547fcca395c9ec797',1,'mlx::core::operator==(T a, const array &b)'],['../group__ops.html#ga3ad3ed7aece2650943a35082dbe3a0a5',1,'mlx::core::operator==(const array &a, T b)'],['../namespacemlx_1_1core.html#ac470f937a379d6356c8f567c97cd7481',1,'mlx::core::operator==(const Stream &lhs, const Stream &rhs)'],['../namespacemlx_1_1core.html#aec63a0472cb943fe39f31e7678555572',1,'mlx::core::operator==(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad05311ca8e2f19ffe5849e963837cec7',1,'mlx::core::operator==(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aaaf591cb2188381e6cbd857132d04eb7',1,'mlx::core::operator==(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7ef33c33509ccccf1ab217500e8b3c1a',1,'mlx::core::operator==(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#abec4200a718b7c5ed80b7abcc4447260',1,'mlx::core::operator==(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad853981b1c5ba69b07d54c7b77055d22',1,'mlx::core::operator==(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a752d6cb4172a9cb91e5da19582329c6d',1,'mlx::core::operator==(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0175beb3de139faa08479a88215b35ea',1,'mlx::core::operator==(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a61da2851cb3beeef28049228346c28b5',1,'mlx::core::operator==(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aa24713cb9e39bacb516c992eb03d2b2b',1,'mlx::core::operator==(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a6d565dd93c46259f9486d9fdf0969589',1,'mlx::core::operator==(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a24e79a82557861de64dad66d36e6ff30',1,'mlx::core::operator==(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#af27d515ac390d62bd852b73ea759a947',1,'mlx::core::operator==(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae3e1e8b7a5410e0edf35f31f74295e2f',1,'mlx::core::operator==(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aaa22230a66b15c3e774d8ce45783a746',1,'mlx::core::operator==(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ae2a0bcdc171d7e9745d33e1d9aac4f8a',1,'mlx::core::operator==(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a331ec62442a8d3eb8ccba7b4de5168d1',1,'mlx::core::operator==(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#acfcaefe0990eb3533e2b11a6f2657492',1,'mlx::core::operator==(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a8d48dbd49cccff07777affb2a412058c',1,'mlx::core::operator==(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a88eae27edd22fa4418776672023cb276',1,'mlx::core::operator==(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a188b363f633ea360407b3f9cf4e1f1a6',1,'mlx::core::operator==(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ae065fe5c42c1a333d7858d19f6434fa9',1,'mlx::core::operator==(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a2f98db199deb6d7a82551fa4afec655a',1,'mlx::core::operator==(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a85f83add412cb320b5cd1c3da6aadbd5',1,'mlx::core::operator==(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7e2cee66c3ca1b56f4f3d7fd1d6e0be1',1,'mlx::core::operator==(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#ad436557da5c7fea71fc58182a876cfe5',1,'mlx::core::operator==(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3e_38',['operator>',['../backend_2metal_2kernels_2complex_8h.html#a032a8d3eec2384c9f03066f7fd945995',1,'operator>(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae394c0a10e47d1d047854a888402eb57',1,'operator>(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab9cd098786d2f4c855c42e4a6f30ab3e',1,'operator>(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a55600f3b9859e2891e0e0b5690867b72',1,'operator>(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afd7cdb8ed2a9820efe9cf322c06f188c',1,'operator>(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31bbdbe0b62b90a4d6ea4bb0a7db586b',1,'operator>(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a68125e66f74eaffe5ea9267638ce870d',1,'operator>(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac89eb6b29edad8cca63727ab97171c29',1,'operator>(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a74e477567c9477c2cf0684f81ef4498f',1,'operator>(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2d37130b6fd79b425f5ba92b65e36bed',1,'operator>(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a41d55d167e9dc63bf29d15e0ff004869',1,'operator>(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa95f9ebfdab3c5f524775651362ce914',1,'operator>(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2826bd301bb5393473ccd363f2052c0d',1,'operator>(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a62a512d0edd894759c69f724b970fbdb',1,'operator>(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a7512eadda6160e4c9d9e6aa4049fac20',1,'mlx::steel::operator>()'],['../group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb',1,'mlx::core::operator>(const array &a, const array &b)'],['../group__ops.html#ga32e106e794e2c32e4e7decee2df2477f',1,'mlx::core::operator>(T a, const array &b)'],['../group__ops.html#ga96552b90e89923c5d2064cc427775ec5',1,'mlx::core::operator>(const array &a, T b)'],['../namespacemlx_1_1core.html#aedc4e9df4bf71c0ac34fcfae60cdf550',1,'mlx::core::operator>(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a14c188303d09b97867bcfd34519aa4a6',1,'mlx::core::operator>(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ac97736fadafa7efa201624d0e1128ee8',1,'mlx::core::operator>(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3c41a304126bc225bdc68062d1eb6e7e',1,'mlx::core::operator>(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ab594f3ae1ee13227fae940fef0d00cb9',1,'mlx::core::operator>(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a01dabc077a872c115a9a9ccd95f1acec',1,'mlx::core::operator>(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#adabbd8768d216873617768249473a5c7',1,'mlx::core::operator>(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#adae1b14669d27ce1fe0c214771c07b77',1,'mlx::core::operator>(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ab03a22961d99fa12d3e74b3116e94e8f',1,'mlx::core::operator>(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a42011a27a3d23a60be5be44ee7cac87c',1,'mlx::core::operator>(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a50f6a94bb36d89cf28817aff88ab89c8',1,'mlx::core::operator>(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac173de50ee57b1b066d49363ba978c53',1,'mlx::core::operator>(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#ab09f1b4879aa3190c2f66c9bd1224021',1,'mlx::core::operator>(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a91eb6ca854217424129a55ae95a123b5',1,'mlx::core::operator>(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a58d5795d8312599d101ae16f194e4a2a',1,'mlx::core::operator>(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aafa3bbeda78610c4285f3e57042268f3',1,'mlx::core::operator>(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a8a928d76a6fbf3d336296401e14617a4',1,'mlx::core::operator>(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ade2f9222fd433cd4d673c6182f256235',1,'mlx::core::operator>(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ae24c337810c841ff23e327efde7045e1',1,'mlx::core::operator>(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acf401ede354fcc998b13ea6442994d7e',1,'mlx::core::operator>(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a2bb28a9a0894a73ae1b27e7f4da0841a',1,'mlx::core::operator>(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a09d631e8a85fd7ae72e1a868b8f9b9cb',1,'mlx::core::operator>(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a49421ea65b5a98df080d75b1636b2157',1,'mlx::core::operator>(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a692ce931b660415e17f92d18a8e0d446',1,'mlx::core::operator>(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a579bb87b3ede5663d7cd68c7c0f6fb9e',1,'mlx::core::operator>(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af810587a17e692f4eec256d3c3cd27de',1,'mlx::core::operator>(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a50f4177d3ca03a95fc2614e100c7391d',1,'mlx::core::operator>(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3e_3d_39',['operator>=',['../backend_2metal_2kernels_2complex_8h.html#aafbd686c180398c98b33d7643f893a46',1,'operator>=(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a430dd11fbf4c6f39bc1506ab43b2341f',1,'operator>=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a64f6787a96386246f83a8981d274150e',1,'operator>=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1a788f82212afad30e4c2ee40f1c313c',1,'operator>=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae88617c4a012c5dc12781a349a28c886',1,'operator>=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a467a88531150a4d9d30fce07c49c126e',1,'operator>=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9e21c5ea9dd724dc2ca8c54ad908f09c',1,'operator>=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2f6286d222e2176bcbdc824c5d598100',1,'operator>=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abec53064aa96265385ecc57de5fbc74c',1,'operator>=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac766839f8f9e4863e8e18418c342c875',1,'operator>=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2807fa6862b0f9689c81199b1e695ed8',1,'operator>=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee3ae0d0d1f941463b06eca0bf041b2b',1,'operator>=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a523eda93c809733368e2b45382d2add6',1,'operator>=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1f4e90909ac1c7280f4c7d1977c55fb7',1,'operator>=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa3c95c60cf69603705bb4636de547bcb',1,'mlx::steel::operator>=()'],['../group__ops.html#ga3a41895f25ed083a36994d95fa102546',1,'mlx::core::operator>=(const array &a, const array &b)'],['../group__ops.html#gaf509f2cb3b18963232f20d6c3bd229b2',1,'mlx::core::operator>=(T a, const array &b)'],['../group__ops.html#gafa0eb25d5978674bfc9e59d4145ec590',1,'mlx::core::operator>=(const array &a, T b)'],['../namespacemlx_1_1core.html#a8494764f5c686743ede66dc76d85d955',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a019df48807b506d9995856684bf7797a',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a96ab6405430efb887cdb5c828cb67d6e',1,'mlx::core::operator>=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac18be72269b1bcfb0249cc00a0600681',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aeb879815228efbd2c8f80986e1c8d41f',1,'mlx::core::operator>=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0051156f6a568f58cd54850f746fb507',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ae93556906e115625ed1b62d36cf21b70',1,'mlx::core::operator>=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab81ad16e3be591dfc9e42ac3c19b055f',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6cfe9b03e7c5f1eb9374208a552c3cc9',1,'mlx::core::operator>=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2f5add83812fb137dd9226c6c01e45d5',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ad1014a836e7ce9301de8588eef1e89ee',1,'mlx::core::operator>=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a17791561434dc995de9f268d145c0ed1',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a3755925b24a903045937464be117de2f',1,'mlx::core::operator>=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a6262aeb513d27fc8313293b261e72abb',1,'mlx::core::operator>=(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a6feb4b3ea511b0eda4d1ec9725f3fb4c',1,'mlx::core::operator>=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a03b3f7fcb755ec075985ab26336926f0',1,'mlx::core::operator>=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aecfbf5ef4872ae447eb4a374e4db28e4',1,'mlx::core::operator>=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae4690f349b2483f5d1a4b75aba67399f',1,'mlx::core::operator>=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a667e95146dd5199e67bcb121b984b1f0',1,'mlx::core::operator>=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a3375f1562f148bdc07451f2b6e54e6df',1,'mlx::core::operator>=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ae83df12368cb07ccb1c10c1117ff3922',1,'mlx::core::operator>=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad41251938cf852b5560c1180944ebb49',1,'mlx::core::operator>=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a4ddb5ef0b88929086f9b09729fda0dde',1,'mlx::core::operator>=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a0908a61ab261aff726922b33fa6ed159',1,'mlx::core::operator>=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a0fdadf87edd8a0a57c63953fb0ebe053',1,'mlx::core::operator>=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a47c82778e43032c0bbf5d59407e81dc9',1,'mlx::core::operator>=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a14e6c43b924eacca1b2dac1d5d00ca2b',1,'mlx::core::operator>=(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3e_3e_40',['operator>>',['../group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a',1,'mlx::core']]], - ['operator_5b_5d_41',['operator[]',['../classpocketfft_1_1detail_1_1arr.html#aea0bd899b19e03f54dfd6c188727061a',1,'pocketfft::detail::arr::operator[](size_t idx)'],['../classpocketfft_1_1detail_1_1arr.html#a99c54f96bc79c7cdd8925c1663462842',1,'pocketfft::detail::arr::operator[](size_t idx) const'],['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a71b02f67c47b24adb296eafd2c7a3598',1,'pocketfft::detail::sincos_2pibyn::operator[]()'],['../classpocketfft_1_1detail_1_1cndarr.html#ae4852d1fe936a5d61832b507816c7054',1,'pocketfft::detail::cndarr::operator[]()'],['../classpocketfft_1_1detail_1_1ndarr.html#a2b2c4e205e8b5c32c9fe55dfd7b8c8d8',1,'pocketfft::detail::ndarr::operator[]()']]], - ['operator_5e_42',['operator^',['../group__ops.html#gac3a6fe18694e84b3d63458e9553ac181',1,'mlx::core::operator^(const array &a, const array &b)'],['../namespacemlx_1_1core.html#ae36ea40b8477bfa12d41aae8245225c9',1,'mlx::core::operator^(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a03fc96696f5c6d9411841889d05f4670',1,'mlx::core::operator^(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a55130edf926366db0d6207989e609b7c',1,'mlx::core::operator^(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0b75198f364d742a1c25dd13e398f2c2',1,'mlx::core::operator^(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7f205f1b10b23180a23bf2be4bb726b1',1,'mlx::core::operator^(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a9edfe65f3c6da583c7b109290ec94b22',1,'mlx::core::operator^(uint16_t lhs, _MLX_Float16 rhs)']]], - ['operator_5e_3d_43',['operator^=',['../namespacemlx_1_1core.html#a97cb7d3eac404a442e84656cefe7cfb4',1,'mlx::core::operator^=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abcfd2d9615c96561fd44dfb9c341cf8e',1,'mlx::core::operator^=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ae78083d766b9cf6f87cded341bbcd63e',1,'mlx::core::operator^=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acf36c10779fbf1efbe1e6a7fd41176cd',1,'mlx::core::operator^=(_MLX_Float16 &lhs, uint16_t rhs)']]], - ['operator_7c_44',['operator|',['../group__ops.html#ga52392a2a98f09a80da8d338c4908bd02',1,'mlx::core::operator|(const array &a, const array &b)'],['../namespacemlx_1_1core.html#af84ed854132c1514dca5a524fdb7ed05',1,'mlx::core::operator|(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7423aac70f9f2e3fb6a5c9a3fc96f703',1,'mlx::core::operator|(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a19805f505cb7ac72bfab66c339ea7900',1,'mlx::core::operator|(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2d933573edf4ed305fddd8a0caef1ee8',1,'mlx::core::operator|(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afab3d4eb1b36a276922879ce6e44b7f5',1,'mlx::core::operator|(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ab132729fa6912d22a8e402057eb4ba12',1,'mlx::core::operator|(uint16_t lhs, _MLX_Float16 rhs)']]], - ['operator_7c_3d_45',['operator|=',['../namespacemlx_1_1core.html#a8e1d21375ae4b89b3cbea3a46d262abd',1,'mlx::core::operator|=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a28d6c2f89e73b7b874dd1f67f853a96f',1,'mlx::core::operator|=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a2d8470b69cbbeefece08d3ffd46c0082',1,'mlx::core::operator|=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a359c6257097a304c00d41d64296ef4c9',1,'mlx::core::operator|=(_MLX_Float16 &lhs, uint16_t rhs)']]], - ['operator_7c_7c_46',['operator||',['../namespacemlx_1_1steel.html#a1bb3ac5061a04e407fc4cdcc9f6ea03f',1,'mlx::steel::operator||()'],['../group__ops.html#ga27af56a98270d4d76d139f0f9171b83a',1,'mlx::core::operator||()']]], - ['ops_2eh_47',['ops.h',['../backend_2common_2ops_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2reduction_2ops_8h.html',1,'(Global Namespace)'],['../distributed_2ops_8h.html',1,'(Global Namespace)'],['../ops_8h.html',1,'(Global Namespace)']]], - ['or_48',['Or',['../struct_or.html',1,'Or< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a7a959bb7b33f410a03b3c887173fd7ed',1,'mlx::core::distributed::AllReduce::Or'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23da51065a44e7f9a76a6dab6de637c6db22',1,'mlx::core::BitwiseBinary::Or'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a2e53e38f8b906ed4def9a5653aeb51fe',1,'mlx::core::Reduce::Or']]], - ['ortho_49',['ortho',['../structpocketfft_1_1detail_1_1_exec_dcst.html#aea17551a49acaca5e7808dc181d38b7f',1,'pocketfft::detail::ExecDcst']]], - ['os_50',['oS',['../struct_m_l_x_conv_params.html#a19ccb9fecfccdc18b6a7f0cc43adbc6e',1,'MLXConvParams']]], - ['out_51',['out',['../struct_read_writer.html#abea3b913c952c505d0ca4e529c7316ef',1,'ReadWriter']]], - ['out_5fof_5fbounds_52',['out_of_bounds',['../struct_read_writer.html#a08e10626fbc789b6dff9172fd6c36f7c',1,'ReadWriter::out_of_bounds() const'],['../struct_read_writer.html#a6f946aea5452109dca7fc70ed39c6efe',1,'ReadWriter::out_of_bounds() const'],['../struct_read_writer.html#a8f40d7f343d32134fe27a694abfde6bf',1,'ReadWriter::out_of_bounds() const']]], - ['out_5fstrides_53',['out_strides',['../struct_m_l_x_conv_params.html#adfca77f9a3c2b4c74752f90636ff5667',1,'MLXConvParams']]], - ['outer_54',['outer',['../group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d',1,'mlx::core']]], - ['output_5fshape_55',['output_shape',['../classmlx_1_1core_1_1_broadcast_axes.html#aaa495110c16fbbc642fbb224ef8dfae6',1,'mlx::core::BroadcastAxes::output_shape()'],['../classmlx_1_1core_1_1_broadcast.html#a00c39c113fe3e698771e2e6b595c32cd',1,'mlx::core::Broadcast::output_shape()'],['../classmlx_1_1core_1_1_expand_dims.html#a3814ad4697eccb75fdb9275017a3fd67',1,'mlx::core::ExpandDims::output_shape()'],['../classmlx_1_1core_1_1_flatten.html#a2f8e1defb9c33af2dec29ff8697132aa',1,'mlx::core::Flatten::output_shape()'],['../classmlx_1_1core_1_1_reshape.html#aa15020d7d844d714d42bc60b44aeefc1',1,'mlx::core::Reshape::output_shape()'],['../classmlx_1_1core_1_1_squeeze.html#aadf1d3b85839390a2ec560603aeed04a',1,'mlx::core::Squeeze::output_shape()'],['../classmlx_1_1core_1_1_unflatten.html#a4c760c8fe981fd2ac17a31ff9faff10a',1,'mlx::core::Unflatten::output_shape()']]], - ['output_5fshapes_56',['output_shapes',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a50934862ccdb16a3dcce6626c5727080',1,'mlx::core::fast::AffineQuantize::output_shapes()'],['../classmlx_1_1core_1_1_primitive.html#aa5b443d71db1c7ed31a5ae6e31b7fe29',1,'mlx::core::Primitive::output_shapes()'],['../classmlx_1_1core_1_1_abs.html#ac9d55481e5490423e4aaf02b95cafc75',1,'mlx::core::Abs::output_shapes()'],['../classmlx_1_1core_1_1_add.html#a50877893083fd78b31aa25152f750418',1,'mlx::core::Add::output_shapes()'],['../classmlx_1_1core_1_1_arange.html#a447083a1403d3d42a7ad9c307a666946',1,'mlx::core::Arange::output_shapes()'],['../classmlx_1_1core_1_1_arc_cos.html#a67a5025f8d7e5bac22888ad4bf813679',1,'mlx::core::ArcCos::output_shapes()'],['../classmlx_1_1core_1_1_arc_cosh.html#a3ab82e9f0452faea735338abccb5f0ac',1,'mlx::core::ArcCosh::output_shapes()'],['../classmlx_1_1core_1_1_arc_sin.html#a0217b9a4e18196ed65ba96b4ad096ecd',1,'mlx::core::ArcSin::output_shapes()'],['../classmlx_1_1core_1_1_arc_sinh.html#a2f668f230d93c7b90e62200a0b7cb6f6',1,'mlx::core::ArcSinh::output_shapes()'],['../classmlx_1_1core_1_1_arc_tan.html#a2ebabfd1c2963199df0d7610b7ddf422',1,'mlx::core::ArcTan::output_shapes()'],['../classmlx_1_1core_1_1_arc_tan2.html#acb8e5cf85c4bc58f909ce2e8b83c3619',1,'mlx::core::ArcTan2::output_shapes()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6806f04142d850f107a18a71900759c6',1,'mlx::core::ArcTanh::output_shapes()'],['../classmlx_1_1core_1_1_arg_partition.html#a587ce69b0639683ba646652f887d0239',1,'mlx::core::ArgPartition::output_shapes()'],['../classmlx_1_1core_1_1_arg_reduce.html#a81a70885480c1d436329025091b2fa4c',1,'mlx::core::ArgReduce::output_shapes()'],['../classmlx_1_1core_1_1_arg_sort.html#a219ce04a811397a900c3235d8e6aef5c',1,'mlx::core::ArgSort::output_shapes()'],['../classmlx_1_1core_1_1_as_type.html#a3975b31cfd86d6eb33dc73554b357b88',1,'mlx::core::AsType::output_shapes()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a49c9d2688d3cca8abf5698a250d57d56',1,'mlx::core::BitwiseBinary::output_shapes()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a42c4385e65851d58e4411a4afe73f58e',1,'mlx::core::BroadcastAxes::output_shapes()'],['../classmlx_1_1core_1_1_broadcast.html#adef65b1ec75efbe43e5574ec81b7c0ac',1,'mlx::core::Broadcast::output_shapes()'],['../classmlx_1_1core_1_1_ceil.html#a3bf7db5178ed26e23d9ba360ba34ab85',1,'mlx::core::Ceil::output_shapes()'],['../classmlx_1_1core_1_1_compiled.html#a15cb081590ee024ba11476494581a4d4',1,'mlx::core::Compiled::output_shapes()'],['../classmlx_1_1core_1_1_concatenate.html#af8415a2fe28804a1437d0876ba15615f',1,'mlx::core::Concatenate::output_shapes()'],['../classmlx_1_1core_1_1_conjugate.html#afd68332463d12e69c47388f6b81ae96c',1,'mlx::core::Conjugate::output_shapes()'],['../classmlx_1_1core_1_1_contiguous.html#afff58fbf61f0c26b3606208dd2fa2072',1,'mlx::core::Contiguous::output_shapes()'],['../classmlx_1_1core_1_1_copy.html#a6bbe5fd9ce3cb5a39853b316106d2674',1,'mlx::core::Copy::output_shapes()'],['../classmlx_1_1core_1_1_cos.html#a923312e71c5a003a38b37ab67ec82580',1,'mlx::core::Cos::output_shapes()'],['../classmlx_1_1core_1_1_cosh.html#adf58c7e24b5059e66007132bc16dfe49',1,'mlx::core::Cosh::output_shapes()'],['../classmlx_1_1core_1_1_divide.html#a9563d9ee243204cfdaac6aca34853cd7',1,'mlx::core::Divide::output_shapes()'],['../classmlx_1_1core_1_1_div_mod.html#a1b7f104346cb5423ac15371b45c7ef86',1,'mlx::core::DivMod::output_shapes()'],['../classmlx_1_1core_1_1_select.html#a10e837a391542b364186288a87e11513',1,'mlx::core::Select::output_shapes()'],['../classmlx_1_1core_1_1_remainder.html#ab4de49818d1fdea8cdfef502f519b255',1,'mlx::core::Remainder::output_shapes()'],['../classmlx_1_1core_1_1_equal.html#ae714c2b0641fc9c339a2f8483bb4e257',1,'mlx::core::Equal::output_shapes()'],['../classmlx_1_1core_1_1_erf.html#ace70b96c48419e29243982ed697f6411',1,'mlx::core::Erf::output_shapes()'],['../classmlx_1_1core_1_1_erf_inv.html#a067cac7a7244b4dae6629c7e4466589f',1,'mlx::core::ErfInv::output_shapes()'],['../classmlx_1_1core_1_1_exp.html#aef2b3c24dba3ca3a63a210d3bd8e39b6',1,'mlx::core::Exp::output_shapes()'],['../classmlx_1_1core_1_1_expm1.html#ae78f03a204687f16164ed702cfc0d5cc',1,'mlx::core::Expm1::output_shapes()'],['../classmlx_1_1core_1_1_expand_dims.html#af64bd4bc2cc5f5c58869f34cd974bb3c',1,'mlx::core::ExpandDims::output_shapes()'],['../classmlx_1_1core_1_1_flatten.html#a5069a73ba1e7b52b7b051f692db6d0d2',1,'mlx::core::Flatten::output_shapes()'],['../classmlx_1_1core_1_1_floor.html#a0a62dee6df6a82fcd955bf7670be2cd5',1,'mlx::core::Floor::output_shapes()'],['../classmlx_1_1core_1_1_gather.html#a53d89a6c4ebb634bc208bd85aa2fcda1',1,'mlx::core::Gather::output_shapes()'],['../classmlx_1_1core_1_1_greater.html#af798a7cd704a2a9a8b3ecb6ef49583b0',1,'mlx::core::Greater::output_shapes()'],['../classmlx_1_1core_1_1_greater_equal.html#a1a77c18d89ee227171ff38efef6cacf6',1,'mlx::core::GreaterEqual::output_shapes()'],['../classmlx_1_1core_1_1_hadamard.html#aa709166de3c493308689769579d665e8',1,'mlx::core::Hadamard::output_shapes()'],['../classmlx_1_1core_1_1_imag.html#ad4f847483ba07d20aba5b927c2689be8',1,'mlx::core::Imag::output_shapes()'],['../classmlx_1_1core_1_1_less.html#ad7604a75b79260d263ac0c7d959cadd5',1,'mlx::core::Less::output_shapes()'],['../classmlx_1_1core_1_1_less_equal.html#a5598c700e881673098928e47b4da9ff8',1,'mlx::core::LessEqual::output_shapes()'],['../classmlx_1_1core_1_1_log.html#ab2cae6889352ca0674f6463f8f52d77d',1,'mlx::core::Log::output_shapes()'],['../classmlx_1_1core_1_1_log1p.html#a73a02ddf0f125fff83462d97146a0a08',1,'mlx::core::Log1p::output_shapes()'],['../classmlx_1_1core_1_1_logical_not.html#ad3889969521c6a040aa2f26caee219b7',1,'mlx::core::LogicalNot::output_shapes()'],['../classmlx_1_1core_1_1_logical_and.html#a266f1eaced19b8b11e273de9219cf9ed',1,'mlx::core::LogicalAnd::output_shapes()'],['../classmlx_1_1core_1_1_logical_or.html#a931b98fca3e19085af9fa97a43db8ced',1,'mlx::core::LogicalOr::output_shapes()'],['../classmlx_1_1core_1_1_log_add_exp.html#a234f8c8ea5f5bf2fb7e371588fea98b9',1,'mlx::core::LogAddExp::output_shapes()'],['../classmlx_1_1core_1_1_matmul.html#abfabe69f428f7f125bf5665713a0eb5c',1,'mlx::core::Matmul::output_shapes()'],['../classmlx_1_1core_1_1_maximum.html#a888a69fb68726c3c18973f3ea38cfd2b',1,'mlx::core::Maximum::output_shapes()'],['../classmlx_1_1core_1_1_minimum.html#af921b5202ebf9716972bcf0e3056742a',1,'mlx::core::Minimum::output_shapes()'],['../classmlx_1_1core_1_1_multiply.html#adfd4c7f89660b42ab58e088b1ae19435',1,'mlx::core::Multiply::output_shapes()'],['../classmlx_1_1core_1_1_negative.html#a606fb13a48d10c88707f1a2c41bee9e8',1,'mlx::core::Negative::output_shapes()'],['../classmlx_1_1core_1_1_not_equal.html#ad1e8a577dc103d96f1ab65bf3b389d35',1,'mlx::core::NotEqual::output_shapes()'],['../classmlx_1_1core_1_1_number_of_elements.html#a6cdf307348ba22b3dc8f90f1fb1e0757',1,'mlx::core::NumberOfElements::output_shapes()'],['../classmlx_1_1core_1_1_partition.html#a5e62aa0109e53fb4acb861ef39787b4a',1,'mlx::core::Partition::output_shapes()'],['../classmlx_1_1core_1_1_power.html#af23ed795bdcdc4c3f91f0d4c1bb1d928',1,'mlx::core::Power::output_shapes()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a7d57a31d41c58e1bd88ffe9c6b0dbf52',1,'mlx::core::QuantizedMatmul::output_shapes()'],['../classmlx_1_1core_1_1_real.html#a75999bd0b97d97a5675b9cdbab27dcff',1,'mlx::core::Real::output_shapes()'],['../classmlx_1_1core_1_1_reshape.html#aed3a83606d6917b2c344607101a2c43d',1,'mlx::core::Reshape::output_shapes()'],['../classmlx_1_1core_1_1_reduce.html#aaf3da1c98cdf530803118b382c5f58bc',1,'mlx::core::Reduce::output_shapes()'],['../classmlx_1_1core_1_1_round.html#a61821399e177e142723fc986e437d459',1,'mlx::core::Round::output_shapes()'],['../classmlx_1_1core_1_1_sigmoid.html#aff024a3309584724c9842f172a4e440b',1,'mlx::core::Sigmoid::output_shapes()'],['../classmlx_1_1core_1_1_sign.html#a2260f2e8e081010192eb8a6f90acde6e',1,'mlx::core::Sign::output_shapes()'],['../classmlx_1_1core_1_1_sin.html#abdd433ecbb54898161b43aa9e14ec7f1',1,'mlx::core::Sin::output_shapes()'],['../classmlx_1_1core_1_1_sinh.html#ae04d8f6175c691a8f0d2a9fdd15af0ad',1,'mlx::core::Sinh::output_shapes()'],['../classmlx_1_1core_1_1_slice_update.html#abb6376f13c4269bd9e739e131893da53',1,'mlx::core::SliceUpdate::output_shapes()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a920dc4d1ee4976065e6d91fe3ecfbbf3',1,'mlx::core::DynamicSlice::output_shapes()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a804c03c745fc563e209a7bfb3d425a91',1,'mlx::core::DynamicSliceUpdate::output_shapes()'],['../classmlx_1_1core_1_1_softmax.html#a1a798a4dcd62486362d4b58582357490',1,'mlx::core::Softmax::output_shapes()'],['../classmlx_1_1core_1_1_sort.html#acc0a3f078b3f4c83e6e1137cb81ee62c',1,'mlx::core::Sort::output_shapes()'],['../classmlx_1_1core_1_1_square.html#a0513541766bb997ed166643fe95a6d38',1,'mlx::core::Square::output_shapes()'],['../classmlx_1_1core_1_1_sqrt.html#ae45215d61e2e99749d9a0bae291edd45',1,'mlx::core::Sqrt::output_shapes()'],['../classmlx_1_1core_1_1_stop_gradient.html#a8af7641d478505d1dc39c75ba7d5a3cf',1,'mlx::core::StopGradient::output_shapes()'],['../classmlx_1_1core_1_1_subtract.html#aaaff4872bde70ad40cf90e6131ea0489',1,'mlx::core::Subtract::output_shapes()'],['../classmlx_1_1core_1_1_squeeze.html#a839d9d72ac0a19e1146b5b470292a174',1,'mlx::core::Squeeze::output_shapes()'],['../classmlx_1_1core_1_1_tan.html#a9e4bba311bb24617dbb5ca591bc2868e',1,'mlx::core::Tan::output_shapes()'],['../classmlx_1_1core_1_1_tanh.html#a8873286b69b805486fa83c4806843f3d',1,'mlx::core::Tanh::output_shapes()'],['../classmlx_1_1core_1_1_unflatten.html#a068cf053b5b0612fafd4a2d53d42f9fa',1,'mlx::core::Unflatten::output_shapes()'],['../classmlx_1_1core_1_1_transpose.html#ac9328f43900bedec555909d09202ccd7',1,'mlx::core::Transpose::output_shapes()'],['../classmlx_1_1core_1_1_eigh.html#a9892f5b72dec19a5a2f7af5efcf2a952',1,'mlx::core::Eigh::output_shapes()']]], - ['outputs_57',['outputs',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a55a7a92c6abad369c99a5ede7a2521b9',1,'mlx::core::metal::DeviceStream::outputs'],['../classmlx_1_1core_1_1array.html#a2c186fd527f984f0589d4183b4976289',1,'mlx::core::array::outputs()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aefa48740fdee884f02e2d379bca4e78f',1,'mlx::core::metal::CommandEncoder::outputs()'],['../structmlx_1_1core_1_1_command_encoder.html#aefa48740fdee884f02e2d379bca4e78f',1,'mlx::core::CommandEncoder::outputs()']]], - ['overwrite_5fdescriptor_58',['overwrite_descriptor',['../classmlx_1_1core_1_1array.html#a95e6b156c8e05439f076b85c05079387',1,'mlx::core::array']]] + ['operator_20simd_3c_20float_2c_20n_20_3e_15',['operator Simd< float, N >',['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a98affc184d83627d8654e3530ab52d75',1,'mlx::core::simd::Simd< float16_t, N >']]], + ['operator_20simd_3c_20int16_5ft_2c_20n_20_3e_16',['operator Simd< int16_t, N >',['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a823af21442333505114fd3fdac9f24de',1,'mlx::core::simd::Simd< float16_t, N >']]], + ['operator_20t_17',['operator T',['../structcomplex64__t.html#a70e9b16031eeaff3baa601f400023fcd',1,'complex64_t::operator T() const thread'],['../structcomplex64__t.html#a4f3beea7ab6001189b782a74d1746b67',1,'complex64_t::operator T() const threadgroup'],['../structcomplex64__t.html#a9f4f7eca89ffe6c8d126a4145df6d9f2',1,'complex64_t::operator T() const device'],['../structcomplex64__t.html#ac33e2e5263fec76a4fb4418c6e1d8d14',1,'complex64_t::operator T() const constant'],['../struct___m_l_x___b_float16.html#aa7dfefdf0d15e102d2b8258c9ab01836',1,'_MLX_BFloat16::operator T() const thread'],['../struct___m_l_x___b_float16.html#a2546a8afa77e14ed5b3c5da79a281260',1,'_MLX_BFloat16::operator T() const threadgroup'],['../struct___m_l_x___b_float16.html#a1d523f87740fcb852db6ab57896c245a',1,'_MLX_BFloat16::operator T() const device'],['../struct___m_l_x___b_float16.html#a95acd29283024d7093a0bc58c9468a0a',1,'_MLX_BFloat16::operator T() const constant']]], + ['operator_20val_18',['operator Val',['../structmlx_1_1core_1_1_dtype.html#a3b3bc059be5836476da3cb88a4f5e9fd',1,'mlx::core::Dtype']]], + ['operator_20value_5ftype_19',['operator value_type',['../structmlx_1_1steel_1_1integral__constant.html#a0c11203bed44a6a2c387b365134dcd64',1,'mlx::steel::integral_constant']]], + ['operator_21_20',['operator!',['../namespacemlx_1_1core_1_1simd.html#a745e05627c77152ec13d8d90c19cc9bf',1,'mlx::core::simd::operator!(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#afaa6ce61de4d80a4b7e9b2ab7454fff4',1,'mlx::core::simd::operator!(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#aadb0ed44c238d8d643c056298d5b20ca',1,'mlx::core::simd::operator!(Simd< float16_t, N > v)']]], + ['operator_21_3d_21',['operator!=',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a971aa511ab2e7ae1caae09556643a0bd',1,'mlx::core::array::ArrayIterator::operator!=()'],['../namespacemlx_1_1core_1_1simd.html#a4971bfe7f9f9319f859b3040c18f39ca',1,'mlx::core::simd::operator!=(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#a5c49123bf2647a5ca4f0579a54f3e53a',1,'mlx::core::simd::operator!=(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a125cbaa7c5dd0931b0abd11003ab584a',1,'mlx::core::simd::operator!=(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a757838b9d56e132e797a381d3bb0dc86',1,'mlx::core::simd::operator!=(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#ae8ca6615d51866d876b5efb3425600ed',1,'mlx::core::simd::operator!=(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a7f1cebaff9cb88df59b5ec7557b5d167',1,'mlx::core::simd::operator!=(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#a6cce6db46c391a5d06dcb262e21b81fc',1,'mlx::core::simd::operator!=(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a3699410174385f5e597cfccad57fc736',1,'mlx::core::simd::operator!=(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#abc6a26b6e28d3d532fc356f96c97df1d',1,'mlx::core::simd::operator!=(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afc6e4fc5589bbf30f978f34868dd4e55',1,'operator!=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6baa722c22d66c7510786bb275cb8cc2',1,'operator!=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa8d9f01582a0a9f01a666d110c74db2a',1,'operator!=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa504a474ab6e00ebe2b1b7ed2f7d1ffb',1,'operator!=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abf5f3040227f021a5b84cf2eda248b2f',1,'operator!=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a347c9bbf816bad2e9e5e91aa448f8b65',1,'operator!=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a33ea086b561c652f25833a5e1ded34dd',1,'operator!=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2bbdcece13148826d3fe33af727bb79b',1,'operator!=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeb1efa47c5f22cc0b35d49ccce73c406',1,'operator!=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa6b99cde403405df1865c989e4ce845a',1,'operator!=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a204d13a881ae8d337f6efbb98673790c',1,'operator!=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3602117b4c61d5cd4fd72fb8e5f68bd6',1,'operator!=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2721c088adfc9d73cde442d6badd2a6c',1,'operator!=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa4364eda56525cf7576ff00e550175e6',1,'mlx::steel::operator!=()'],['../namespacemlx_1_1core.html#a94d00a1b7f8a4717ab3f26f45e4da655',1,'mlx::core::operator!=(const Device &lhs, const Device &rhs)'],['../group__ops.html#ga0ac483d85f23252ca8757e9926d5a3c5',1,'mlx::core::operator!=(const array &a, const array &b)'],['../group__ops.html#ga3fecba9f3cb9a19afd8ca492cf509ce0',1,'mlx::core::operator!=(T a, const array &b)'],['../group__ops.html#gaebbf1cfde388c7480159a03c92c9a385',1,'mlx::core::operator!=(const array &a, T b)'],['../namespacemlx_1_1core.html#a164f109bc19c927b2b3bcc47a5021419',1,'mlx::core::operator!=(const Stream &lhs, const Stream &rhs)'],['../namespacemlx_1_1core.html#ad2f9e1c230ec35d5c406dd616e8f4dea',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af5899b4d5644682cb0ac2a488f630d55',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a72ac8edd190601d7a46782582cedecd8',1,'mlx::core::operator!=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a8084162ba2dd3f9b89195d2bebc3fbb0',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a514263e63f6825b490203ca586864687',1,'mlx::core::operator!=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a1c482bb3d9f9d4c62dee5865892c1f96',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a0030fe7ad09837c670cdfb7d51279519',1,'mlx::core::operator!=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ade3791bc723b8f10fbab22eadb0f705a',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ad78c664f242cd36247c13868547e3dd4',1,'mlx::core::operator!=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab0743a1a1dcb92d40f41ca42d36f242c',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ae7a0f810e546a166c7d05849b5d41f30',1,'mlx::core::operator!=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a676a40637a563f013c725d24fa33fdc8',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a9fcb662b1561e4136bac0106cfb63b6c',1,'mlx::core::operator!=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abcca7fd43590c4347e0f5df8f134030c',1,'mlx::core::operator!=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af3ede3688a2e3b3ba8cb2da180ffe151',1,'mlx::core::operator!=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a54f48469fabd1414bef5097bcded0002',1,'mlx::core::operator!=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af8c648e892cbc6973de535aa17dc2cfe',1,'mlx::core::operator!=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#abc855e1c0584b64d7d995e33211361ab',1,'mlx::core::operator!=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad3684d660d18a54505c759ab286bd936',1,'mlx::core::operator!=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a8afdda14b14262ab5ce0a00c7745d7e8',1,'mlx::core::operator!=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7ccc479be236f2bf3f7725729c5ba201',1,'mlx::core::operator!=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a26a721b8111fce3a1dec9bf724034cd4',1,'mlx::core::operator!=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad5f8c221a53a89e8095aa39fd1f61867',1,'mlx::core::operator!=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a017b52ecf30b33da4aa8da35ccc43220',1,'mlx::core::operator!=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a43c10ca5fb05ee7d0ee63ba56f8a08a3',1,'mlx::core::operator!=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a81284b6ac737f91a8d1ffbbbbf938fe5',1,'mlx::core::operator!=(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_25_22',['operator%',['../backend_2metal_2kernels_2complex_8h.html#aaf53122a07c8eca858b5a8e38ae280e0',1,'operator%(): complex.h'],['../group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde',1,'mlx::core::operator%(const array &a, const array &b)'],['../group__ops.html#ga50817666f0b82afcbf4a123486af9908',1,'mlx::core::operator%(T a, const array &b)'],['../group__ops.html#ga46c01daa07433542a477d216e13a8480',1,'mlx::core::operator%(const array &a, T b)'],['../namespacemlx_1_1core.html#a8723d145dd49021bfcb8e6c99e1c91a5',1,'mlx::core::operator%(complex64_t a, complex64_t b)']]], + ['operator_26_23',['operator&',['../namespacemlx_1_1core_1_1simd.html#a0727c897502944659b3e32b3cde9ee9b',1,'mlx::core::simd::operator&(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a832bbc02ed5589e70106c831c04500f1',1,'mlx::core::simd::operator&(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ac790406f4cf51cbc40d750d377dd741b',1,'mlx::core::simd::operator&(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a3c42ac1dc74f6c0bb934dfa45986875b',1,'mlx::core::simd::operator&(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value &b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a8beb567724ab9735b616afb777b93abd',1,'mlx::core::simd::operator&(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a &b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a3a060a225b6ead483ca93247c9ad8e4d',1,'mlx::core::simd::operator&(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value &b), 1 >'],['../group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff',1,'mlx::core::operator&(const array &a, const array &b)'],['../namespacemlx_1_1core.html#a9ee95f97bbd69262d99d7bea3bf77631',1,'mlx::core::operator&(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0fefc3ae4f1350ebe05ec6098fd6bae3',1,'mlx::core::operator&(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a1e4cb758ccfe5c267baed9aeb0044834',1,'mlx::core::operator&(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab9d0f9910070231695d61de08cadb930',1,'mlx::core::operator&(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a889d401f425db79d1868aa3beea4829b',1,'mlx::core::operator&(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a76dcd1fa3c68b386bc1d1d899a68a120',1,'mlx::core::operator&(uint16_t lhs, _MLX_Float16 rhs)']]], + ['operator_26_26_24',['operator&&',['../namespacemlx_1_1core_1_1simd.html#a85c23e7ed6fe0ec6dfe4c61f7412a362',1,'mlx::core::simd::operator&&(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a8a2c8aea209236b06c594c8451017ecb',1,'mlx::core::simd::operator&&(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a070f1fa094cf2da5ab7d6baecbbf4f56',1,'mlx::core::simd::operator&&(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a82676bd32059d1172296f8074a841de6',1,'mlx::core::simd::operator&&(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value &&b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#af97917ef704103c6ea1d0e44f22ec0d3',1,'mlx::core::simd::operator&&(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a &&b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a1eca7cf07b2a238307459c28204319fb',1,'mlx::core::simd::operator&&(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value &&b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a2a381e5ec89406074b8d1921304238bb',1,'mlx::core::simd::operator&&(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#af9d5f107ce0c40c3b6a2f176cbb70cd7',1,'mlx::core::simd::operator&&(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#af8f245dfc5154c04c0865a208ab1cfe9',1,'mlx::core::simd::operator&&(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1steel.html#a6353bf11881842e25c46b56f92b7044f',1,'mlx::steel::operator&&()'],['../group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3',1,'mlx::core::operator&&(const array &a, const array &b)']]], + ['operator_26_3d_25',['operator&=',['../namespacemlx_1_1core.html#a60c263ef46e552c3954688869734b513',1,'mlx::core::operator&=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af9670fc8088339669c54c68b3a320e25',1,'mlx::core::operator&=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ad1f96f0a02024f347b4c4431629407fc',1,'mlx::core::operator&=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae0540f16c4e7bd55d0e86a88495e4967',1,'mlx::core::operator&=(_MLX_Float16 &lhs, uint16_t rhs)']]], + ['operator_28_29_26',['operator()',['../structpocketfft_1_1detail_1_1_exec_c2_c.html#a4fd637f1a6d335826789af28ac089ecb',1,'pocketfft::detail::ExecC2C::operator()()'],['../structpocketfft_1_1detail_1_1_exec_hartley.html#a67c98b38d12440781053552b9a33bba1',1,'pocketfft::detail::ExecHartley::operator()()'],['../structpocketfft_1_1detail_1_1_exec_dcst.html#a67f4f56e3574c491695f8cb8a1e983d8',1,'pocketfft::detail::ExecDcst::operator()()'],['../structpocketfft_1_1detail_1_1_exec_r2_r.html#acdba1650962714e6afff51e9ca456970',1,'pocketfft::detail::ExecR2R::operator()()'],['../structmlx_1_1core_1_1_vector_scalar.html#a1af3ff644ce023a7e4f92a7c3634c44f',1,'mlx::core::VectorScalar::operator()()'],['../structmlx_1_1core_1_1_scalar_vector.html#ab174fe55970fb4ee1c6a2b7628a24df1',1,'mlx::core::ScalarVector::operator()()'],['../structmlx_1_1core_1_1_vector_vector.html#a97a0bed419933d7685238a962f2e4215',1,'mlx::core::VectorVector::operator()()'],['../structmlx_1_1core_1_1detail_1_1_add.html#a95cf053f89883d82f31ec53154b430a0',1,'mlx::core::detail::Add::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_add.html#a2d6011c35768b5fcd2bb75747b944353',1,'mlx::core::detail::Add::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html#a01da277adf65232bd67b252a31baedd7',1,'mlx::core::detail::ArcTan2::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html#af0cfd2ea4d541379b9c427fd4054828d',1,'mlx::core::detail::ArcTan2::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_divide.html#a9a3eab9eaf77b5a94ede2db8c7cef9f2',1,'mlx::core::detail::Divide::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_divide.html#a5e0d22e2084c4ca81bec0d457a46c662',1,'mlx::core::detail::Divide::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_multiply.html#a9dda09d0bf0f4153abf37ba894df37d4',1,'mlx::core::detail::Multiply::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_multiply.html#a898b090966b047723513224b8d3b22f1',1,'mlx::core::detail::Multiply::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_subtract.html#a48913052e0a051648b7a69376ec3e3e1',1,'mlx::core::detail::Subtract::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_subtract.html#a72ef05830615a2d5d9662926ed82672a',1,'mlx::core::detail::Subtract::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_logical_and.html#a5fb547e51ea53517deb54d89c76b4860',1,'mlx::core::detail::LogicalAnd::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_logical_and.html#a046536c1f2f9367983f052a213d7b7d8',1,'mlx::core::detail::LogicalAnd::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_logical_or.html#a4701821e656931d808815753ee529bad',1,'mlx::core::detail::LogicalOr::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_logical_or.html#afb134dbab79307d4ba597843c61d0b1a',1,'mlx::core::detail::LogicalOr::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html#a91cff5472e47b13fd9d291b17d2e877b',1,'mlx::core::detail::BitwiseAnd::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html#ae0bed77f95fe2b2f0b594addddd04700',1,'mlx::core::detail::BitwiseAnd::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html#abd39ee9af548b16e3fabe4ae956b6f1c',1,'mlx::core::detail::BitwiseOr::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html#a5ab05734c5000b454975de6647a08d20',1,'mlx::core::detail::BitwiseOr::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html#a8ed25d90a73141938a71ddddfd40b83d',1,'mlx::core::detail::BitwiseXor::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html#a0989e3bcd064ae06c33f660696a869a0',1,'mlx::core::detail::BitwiseXor::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_left_shift.html#a50bcbc53e2278483d9063decf7ad78d8',1,'mlx::core::detail::LeftShift::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_left_shift.html#a9385f580830a6ad163dd9bb8c4905e7a',1,'mlx::core::detail::LeftShift::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_right_shift.html#aa86d02e4ca59bc7ffacdc342841a0ea9',1,'mlx::core::detail::RightShift::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_right_shift.html#a154528ba50e89a4c532a181f135b1620',1,'mlx::core::detail::RightShift::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a8b672df71eea3f31f5e2aa50662f3b19',1,'mlx::core::detail::Remainder::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#ac1bcf314046fa1c76e5491336cf68e02',1,'mlx::core::detail::Remainder::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_maximum.html#a1edfed0e0b33227b67c7709691f846c7',1,'mlx::core::detail::Maximum::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_maximum.html#a1a3bd09f6c4e61982ebf1a9bfaa38059',1,'mlx::core::detail::Maximum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_minimum.html#a28b51060b9345fb2021d5176cd607778',1,'mlx::core::detail::Minimum::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_minimum.html#a5cdc82cc78adbc9854aa9b1c4417d6d3',1,'mlx::core::detail::Minimum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_power.html#ad047c7d25e1b0f32dc17a03d826cf0a0',1,'mlx::core::detail::Power::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_power.html#a5d3c31365fcf2de52f78c3695da83152',1,'mlx::core::detail::Power::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_equal.html#a5d3f7423078444e5d690fb6d50fcce23',1,'mlx::core::detail::Equal::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_equal.html#a2994cf1884e7126e76d0a20b215fe3ab',1,'mlx::core::detail::Equal::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_greater.html#a9186b3e29c84700ea93ca9470556b0b3',1,'mlx::core::detail::Greater::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_greater.html#aa3844c2bae3c7a981739f642aa0dd094',1,'mlx::core::detail::Greater::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html#a8da40f79562ef8ffbd30ddcf40d83e0f',1,'mlx::core::detail::GreaterEqual::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html#a3b005f85522ad0e4b57044eed930ac30',1,'mlx::core::detail::GreaterEqual::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_less.html#a8e9c159887284420b1161421e58a0bda',1,'mlx::core::detail::Less::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_less.html#a0b4032dff1ad2b387745cb000aabdcbb',1,'mlx::core::detail::Less::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_less_equal.html#a5f7f700be5fdf4629a96ab271caf5440',1,'mlx::core::detail::LessEqual::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_less_equal.html#a31e70f8830a07557697541301555a7a7',1,'mlx::core::detail::LessEqual::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_not_equal.html#a99d16a3d7f637901869bf650b1ea6e13',1,'mlx::core::detail::NotEqual::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_not_equal.html#a23d662b5fd968dc17d3bee2595b5f99d',1,'mlx::core::detail::NotEqual::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_na_n_equal.html#a441e5e8552be45ced34001b465d251e1',1,'mlx::core::detail::NaNEqual::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_na_n_equal.html#a073b20b0d8d41ec8364b7c477421b9bf',1,'mlx::core::detail::NaNEqual::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_log_add_exp.html#a434da15bcb95dc979c73ec795cfec339',1,'mlx::core::detail::LogAddExp::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_log_add_exp.html#ad1663fd809acaa4038f90666436599e5',1,'mlx::core::detail::LogAddExp::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_select.html#a930f9da2e6b3453e04f21382435a2cfb',1,'mlx::core::detail::Select::operator()(bool condition, T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_select.html#a8c5135e3098cfd2521a2a266ba08f1e4',1,'mlx::core::detail::Select::operator()(Simd< bool, N > condition, Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#acb9168d40f09d73a2243f75f13bbadc2',1,'mlx::core::detail::Abs::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a0d657bc9a381dca1b5860b9a1b5a5702',1,'mlx::core::detail::Abs::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html#a1b927a97bbef1478c768bb85cb764c94',1,'mlx::core::detail::ArcCos::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html#a04b4c9d1fc0160973aa28b1f809b9d51',1,'mlx::core::detail::ArcCos::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html#a4436be0278ceaced10ef98eb6f30f789',1,'mlx::core::detail::ArcCosh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html#a767d354bec863942822ee0b9b6742a88',1,'mlx::core::detail::ArcCosh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html#ab1ad6339c662305bd682b14f8d8afd6c',1,'mlx::core::detail::ArcSin::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html#ac69091929815e5317308b4088f5c2f46',1,'mlx::core::detail::ArcSin::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html#ac6e45e41f931f556697c060a2a858816',1,'mlx::core::detail::ArcSinh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html#ac7bf9bac66fef917f75494b2345e6aaf',1,'mlx::core::detail::ArcSinh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html#a697b7f12f30d642ee5f0c54aaf86a8ec',1,'mlx::core::detail::ArcTan::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html#aee87bf10c278a70ca788085d1b499afe',1,'mlx::core::detail::ArcTan::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html#a93a660ea073526e1f75b2d3c4ac6c366',1,'mlx::core::detail::ArcTanh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html#a601e8c52bb938eb3a616756a35419e8b',1,'mlx::core::detail::ArcTanh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a2354e9fa1502d1743834b98cdec17653',1,'mlx::core::detail::Ceil::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a672f65e47d65e4e8d88be252bce0164b',1,'mlx::core::detail::Ceil::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html#a33bbfcc195781eb33df0a4efc50569ed',1,'mlx::core::detail::Conjugate::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html#a386b583d24a2cf1ba8dcc3ba52c226f5',1,'mlx::core::detail::Conjugate::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_cos.html#a663065fd41e5d85e8f044e9f81070568',1,'mlx::core::detail::Cos::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_cos.html#ad4caef573f9d9071f8945a8efed231ad',1,'mlx::core::detail::Cos::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_cosh.html#ae94b6da9ceb47e9d4aaf61451126f58d',1,'mlx::core::detail::Cosh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_cosh.html#a63591f49776d9aadc02200036ae38317',1,'mlx::core::detail::Cosh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_erf.html#a4f5986391863d30e0e7b17bd1996a5f6',1,'mlx::core::detail::Erf::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_erf.html#a168f8ccc6c8053b05dd1a48904ca8fd4',1,'mlx::core::detail::Erf::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html#a0cdd8d6e71222695d0f148b9ad048429',1,'mlx::core::detail::ErfInv::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html#acc93c0511141404208b35f302f8c1fcb',1,'mlx::core::detail::ErfInv::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_exp.html#aad7fb8de7561479c7aa3c741322a3101',1,'mlx::core::detail::Exp::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_exp.html#a0846300cee28315e5b42f74acafbd1a1',1,'mlx::core::detail::Exp::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_expm1.html#a2c78a15f0dd01d13f3a78ac45347ed3e',1,'mlx::core::detail::Expm1::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_expm1.html#abf7e61b8387521e9d44334ce88d833a0',1,'mlx::core::detail::Expm1::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a5c41fb72ec3da9289c24b92802e28f2e',1,'mlx::core::detail::Floor::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a16c13cfe736098bffc81d655e172294a',1,'mlx::core::detail::Floor::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_log.html#a0041795bfd063a9769a3747bd7a91d61',1,'mlx::core::detail::Log::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_log.html#a0012a4e1744dbe9a28c3b5652be6e1c6',1,'mlx::core::detail::Log::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_log2.html#a83258d8a3fe12e082d0b317fcfafb28b',1,'mlx::core::detail::Log2::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_log2.html#a467bd4c995674721ff5fff6df33aead8',1,'mlx::core::detail::Log2::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_log10.html#ade464425f69e5b76bf61b5ba3da75089',1,'mlx::core::detail::Log10::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_log10.html#a2633c5b772bbc9f8b66cffd4a3e01a3f',1,'mlx::core::detail::Log10::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_log1p.html#abed96d56b07c6a96666b770c9711e52e',1,'mlx::core::detail::Log1p::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_log1p.html#a3220de8c6090c44aa2070b1fbb2dc340',1,'mlx::core::detail::Log1p::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_logical_not.html#a4978cc3a63e70a1a4fee6470764ae9d9',1,'mlx::core::detail::LogicalNot::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_logical_not.html#a79799668ea5c364b0b4e2bc330e76253',1,'mlx::core::detail::LogicalNot::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_negative.html#a93a1dfb47eba54aff44b2945d131c97e',1,'mlx::core::detail::Negative::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_negative.html#afc4595c70ef7196df374cf4b2cc5e526',1,'mlx::core::detail::Negative::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_round.html#acd099ba81c8c281e9660cf8c0fed0cd1',1,'mlx::core::detail::Round::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_round.html#a653f29c059bbfa6192378732a8a23351',1,'mlx::core::detail::Round::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sin.html#a07c357c49dbf6b0579b1e771c6eb5766',1,'mlx::core::detail::Sin::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sin.html#ae95671816529cc2188389af37a2f1a13',1,'mlx::core::detail::Sin::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sinh.html#a1e299cd64bc0c7aaa1ceeac35dfe7831',1,'mlx::core::detail::Sinh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sinh.html#a9663ddf0fa4c0003576b48f3d5385f00',1,'mlx::core::detail::Sinh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sqrt.html#acac518e8e7cf3dd103f4f72f22b23221',1,'mlx::core::detail::Sqrt::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sqrt.html#aa5a4830b3ef7efab20ea88a110667efd',1,'mlx::core::detail::Sqrt::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_rsqrt.html#ac6720a6270393152ab2924a77bfb17b2',1,'mlx::core::detail::Rsqrt::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_rsqrt.html#a9af247be16bab83243038aac54446b79',1,'mlx::core::detail::Rsqrt::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_tan.html#a9c8d3570a1e4daa054bb41999043d9e9',1,'mlx::core::detail::Tan::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_tan.html#aba397cd7ac05bbe06dfa9e3a64bdb05f',1,'mlx::core::detail::Tan::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_tanh.html#a79eeba686f3dd5dce097ff5b9b27dd7c',1,'mlx::core::detail::Tanh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_tanh.html#a1749ba1edfd53095ed7d45c0e53bab61',1,'mlx::core::detail::Tanh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_imag.html#a070cf43bc4e30871f8f32d4b84be05c8',1,'mlx::core::detail::Imag::operator()(Simd< complex64_t, N > x)'],['../structmlx_1_1core_1_1detail_1_1_imag.html#a5bd82e2185f3779e398c179d42a3e782',1,'mlx::core::detail::Imag::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_real.html#a7c6c6c188d611e2084dba66b7489c21f',1,'mlx::core::detail::Real::operator()(Simd< complex64_t, N > x)'],['../structmlx_1_1core_1_1detail_1_1_real.html#ae84a939fdb5916257a7731cda66d4d61',1,'mlx::core::detail::Real::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sigmoid.html#a12a3d53f0fd797b5cdd9d04d048ce1a4',1,'mlx::core::detail::Sigmoid::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sigmoid.html#a64b72561bfaf758632167f00648f4c89',1,'mlx::core::detail::Sigmoid::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a913c095e25668c8a6bb6e3243e150606',1,'mlx::core::detail::Sign::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a64ed5013cee7ff18c7fe70bc04737e7b',1,'mlx::core::detail::Sign::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_square.html#abab2378a94c4c38dffeb06a74b0f81ee',1,'mlx::core::detail::Square::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_square.html#a54e9e3c0d0896e142289e8282eab1099',1,'mlx::core::detail::Square::operator()(T x)'],['../struct_add.html#ac5c66b63d63a222d3ae0ab8cc7c90eb5',1,'Add::operator()()'],['../struct_floor_divide.html#a2b328e4d768e718fa439f955c524666a',1,'FloorDivide::operator()(T x, T y)'],['../struct_floor_divide.html#afc16a2b2a745225e0bc95640f3fc0219',1,'FloorDivide::operator()(float x, float y)'],['../struct_floor_divide.html#ae91719a15f7e643d552129f476089c6a',1,'FloorDivide::operator()(half x, half y)'],['../struct_floor_divide.html#a4aa9f858626583e02bd79f747229bbca',1,'FloorDivide::operator()(bfloat16_t x, bfloat16_t y)'],['../struct_divide.html#a0a16b9194abc2ab7c61129f81a9bbb3d',1,'Divide::operator()()'],['../struct_remainder.html#ab7875512ff4341c580c6dc372e64fc58',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#a18150b5f4425e30b95ffabc6bb25cede',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#ab3b75f54b56fd357c9755daadb2cafc2',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#ae918ce0e246937d4fe04e2ea36e4b2c1',1,'Remainder::operator()(complex64_t x, complex64_t y)'],['../struct_equal.html#aa498087080900d4428ba428a6496a769',1,'Equal::operator()()'],['../struct_na_n_equal.html#a00220898e02db656d21dde9e9354a8dc',1,'NaNEqual::operator()(T x, T y)'],['../struct_na_n_equal.html#a6185e4554dce5b4659d21673c576be51',1,'NaNEqual::operator()(complex64_t x, complex64_t y)'],['../struct_greater.html#a98d7d8ee360cd0f469c6eb9a017560f5',1,'Greater::operator()()'],['../struct_greater_equal.html#ae69a3bccc567a46506cf0d296294ce80',1,'GreaterEqual::operator()()'],['../struct_less.html#a5ee0b31b2d9123dc4504f2979a5854d3',1,'Less::operator()()'],['../struct_less_equal.html#ae9f9a1b2eae548977139704f0044acfe',1,'LessEqual::operator()()'],['../struct_log_add_exp.html#ab32417f18e8ff68c15f78aceeb624edf',1,'LogAddExp::operator()()'],['../struct_maximum.html#a3ea0f42bc4cd80b68a98f189f9fa859c',1,'Maximum::operator()(T x, T y)'],['../struct_maximum.html#a0bc8fadc87f2c49fc440d625bfc97ca6',1,'Maximum::operator()(T x, T y)'],['../struct_maximum.html#a907e8793900be5927625377dab199644',1,'Maximum::operator()(complex64_t x, complex64_t y)'],['../struct_minimum.html#aa6113dfac3986c0f571fa53f65c5330e',1,'Minimum::operator()(T x, T y)'],['../struct_minimum.html#a0c939921de87ab9c6959238aac81a059',1,'Minimum::operator()(T x, T y)'],['../struct_minimum.html#a800fba087280f79c2f7e9aff75bed093',1,'Minimum::operator()(complex64_t x, complex64_t y)'],['../struct_multiply.html#a1327fc5a0713931afe997b0d4d2988e0',1,'Multiply::operator()()'],['../struct_not_equal.html#af008d73a5d9cde0b8309b7e8ee7438b2',1,'NotEqual::operator()(T x, T y)'],['../struct_not_equal.html#a14de494cea4e4869351202cad1149f17',1,'NotEqual::operator()(complex64_t x, complex64_t y)'],['../struct_power.html#a2b6df2a9e48155ff9734caca8504a79f',1,'Power::operator()(T base, T exp)'],['../struct_power.html#a36829163d42973034a1f8a7ecc57a1de',1,'Power::operator()(T base, T exp)'],['../struct_power.html#a27cdfb313c4e82b63bdcdaee923cbbef',1,'Power::operator()(complex64_t x, complex64_t y)'],['../struct_subtract.html#ae0856cd8d449074ca287baa7e460f68a',1,'Subtract::operator()()'],['../struct_logical_and.html#a8bc6bdabc0ea0678a46e2cf6217cb3a6',1,'LogicalAnd::operator()()'],['../struct_logical_or.html#ade6a931324a604a3119d2220d6f5460d',1,'LogicalOr::operator()()'],['../struct_bitwise_and.html#afb48af090b01dd0200963bc12d842e36',1,'BitwiseAnd::operator()()'],['../struct_bitwise_or.html#a41f847463daafa99ee56f4035578390f',1,'BitwiseOr::operator()()'],['../struct_bitwise_xor.html#a3a3e8a56caab739d40262d9349c9c485',1,'BitwiseXor::operator()()'],['../struct_left_shift.html#aa729747784c38bfdbba34794fcf5175b',1,'LeftShift::operator()()'],['../struct_right_shift.html#a2cc59b400c68342b0e43050431323c17',1,'RightShift::operator()()'],['../struct_arc_tan2.html#ac9b7729753e13be293ab700231d061ac',1,'ArcTan2::operator()()'],['../struct_div_mod.html#a8b5758f2ea18d4c903b462331b25abfe',1,'DivMod::operator()()'],['../struct_cum_prod_3_01bool_01_4.html#ad634be0b139d10ce6d21332eef0d936b',1,'CumProd< bool >::operator()()'],['../struct_cum_max.html#a781b9b955c5412466da6af6c70d73c06',1,'CumMax::operator()()'],['../struct_cum_min.html#ae0b8c3761e04fa538d304ca842281a66',1,'CumMin::operator()()'],['../struct_less_than.html#a2798eb377b411c93a4ed30cf35caade2',1,'LessThan::operator()()'],['../struct_select.html#adb51692aae3038de07dd745891bf9848',1,'Select::operator()()'],['../struct_abs.html#a9e7481dfcc162509769852026ff4a344',1,'Abs::operator()(T x)'],['../struct_abs.html#a0ca113fd036151c443df3f83cc667f28',1,'Abs::operator()(uint8_t x)'],['../struct_abs.html#adaeab32a7e377dc990077ab15f3dc4c2',1,'Abs::operator()(uint16_t x)'],['../struct_abs.html#a99d2a2f37a6cddd3168b0224f2a9b963',1,'Abs::operator()(uint32_t x)'],['../struct_abs.html#ac9cbc02422d930479303f240a7ea6c71',1,'Abs::operator()(uint64_t x)'],['../struct_abs.html#ac30835b27784d451bd2e4524c8eb9e11',1,'Abs::operator()(bool x)'],['../struct_abs.html#ab82917d6b30a2c579e7eb879d305c5fc',1,'Abs::operator()(complex64_t x)'],['../struct_arc_cos.html#a5553cecf58511e24e76ac97f2d90b9ac',1,'ArcCos::operator()()'],['../struct_arc_cosh.html#a5c9e7712c14c97298b23ec48e19abc58',1,'ArcCosh::operator()()'],['../struct_arc_sin.html#a0343872f2da93bae2bb0baadf49da022',1,'ArcSin::operator()()'],['../struct_arc_sinh.html#a3066fb7dc7c3180100fb55ff94af6a7a',1,'ArcSinh::operator()()'],['../struct_arc_tan.html#af3a0aec6acec8ae8f5e4c4d5cf8c91ba',1,'ArcTan::operator()()'],['../struct_arc_tanh.html#a37dc3e01ec2830de7e82ed6c6363ac88',1,'ArcTanh::operator()()'],['../struct_ceil.html#a5e2a4ef1b012f5d352064489156e5e44',1,'Ceil::operator()(T x)'],['../struct_ceil.html#a455cd8083ba859993077f2e078ae165b',1,'Ceil::operator()(int8_t x)'],['../struct_ceil.html#a2acb61bc658c7a216795e7f76ebcf98a',1,'Ceil::operator()(int16_t x)'],['../struct_ceil.html#aef8c37f7a8ee3fc80700d605a09891fb',1,'Ceil::operator()(int32_t x)'],['../struct_ceil.html#a93d0110511ad5dd200e12d37a3d7d6e3',1,'Ceil::operator()(int64_t x)'],['../struct_ceil.html#aa335b745fa26e0f443cdb36298105484',1,'Ceil::operator()(uint8_t x)'],['../struct_ceil.html#ade17e13b7f30f5c590fae1581a2013ac',1,'Ceil::operator()(uint16_t x)'],['../struct_ceil.html#a411c75cc35cdc088402e176a1defd22d',1,'Ceil::operator()(uint32_t x)'],['../struct_ceil.html#a9ac660ca29eef7a7429fceb7b917a68a',1,'Ceil::operator()(uint64_t x)'],['../struct_ceil.html#a40de367e62f06ebd7e1330afa93a9ad9',1,'Ceil::operator()(bool x)'],['../struct_cos.html#ae222f8710f6b8254c471ebd475aa5bda',1,'Cos::operator()(T x)'],['../struct_cos.html#a5f26feb1dcc4bec5f59a9ff511c5b163',1,'Cos::operator()(complex64_t x)'],['../struct_cosh.html#a5847ebeebb236fdc926798ddc16475ba',1,'Cosh::operator()(T x)'],['../struct_cosh.html#aefdd91298dac16d528d29ee47e2f7252',1,'Cosh::operator()(complex64_t x)'],['../struct_conjugate.html#acb0a2694285f1f57c7654b371ce8cbd8',1,'Conjugate::operator()()'],['../struct_erf.html#a80719402ad7f7d418859a6677d7b604d',1,'Erf::operator()()'],['../struct_erf_inv.html#afbf3668d1a512e889f093a0bc7673309',1,'ErfInv::operator()()'],['../struct_exp.html#a5ef395868e055348c0802fd5fe45669c',1,'Exp::operator()(T x)'],['../struct_exp.html#a2b341ac400c4d145397950eb60734336',1,'Exp::operator()(complex64_t x)'],['../struct_expm1.html#a4b834d42cf0b84daf03fec62c222091a',1,'Expm1::operator()()'],['../struct_floor.html#ace3551f28429081e9f3a3dab0c84212b',1,'Floor::operator()(T x)'],['../struct_floor.html#a10d7fd05b4c224c9f135451246d13014',1,'Floor::operator()(int8_t x)'],['../struct_floor.html#a2865a04a492e3590302f4bd3215a10d7',1,'Floor::operator()(int16_t x)'],['../struct_floor.html#a41012343ff0463ec44b4d06196f41182',1,'Floor::operator()(int32_t x)'],['../struct_floor.html#aae3181d15856796aa0628cf30c92aa2e',1,'Floor::operator()(int64_t x)'],['../struct_floor.html#ac6cf38d82c8e270911afdca4c69ad51b',1,'Floor::operator()(uint8_t x)'],['../struct_floor.html#a78969b9e2b53ae248e72a67259eea5d8',1,'Floor::operator()(uint16_t x)'],['../struct_floor.html#a959009320ed622ed45b39becab1d5b98',1,'Floor::operator()(uint32_t x)'],['../struct_floor.html#a7d04b83c3345cd867315cae2d7ff68ab',1,'Floor::operator()(uint64_t x)'],['../struct_floor.html#abea845fe5e8e6b93bd4bca8717337e0b',1,'Floor::operator()(bool x)'],['../struct_imag.html#a3b29e9f8a46c194d683f6a9938314400',1,'Imag::operator()()'],['../struct_log.html#a32a383cb6be06e616a75f23bf49089c3',1,'Log::operator()()'],['../struct_log2.html#ac1e067ecdcbdbffb6106e789c2b98b64',1,'Log2::operator()()'],['../struct_log10.html#ac596a74c1642a00f3eced07ee3334122',1,'Log10::operator()()'],['../struct_log1p.html#a4464c6e7bdbe55ffd7d961c695cd13ce',1,'Log1p::operator()()'],['../struct_logical_not.html#a8a620bac957ab8c09ac85adfddd96708',1,'LogicalNot::operator()()'],['../struct_negative.html#af6879b374314a559faa321e8cce3d710',1,'Negative::operator()()'],['../struct_real.html#a85b9c5b9e65297994fa26ff68e19e809',1,'Real::operator()()'],['../struct_round.html#aa06a0195867e2ceb679c403b6909a1c4',1,'Round::operator()(T x)'],['../struct_round.html#ad3a08f2276ff1033900bc0a7da812655',1,'Round::operator()(complex64_t x)'],['../struct_sigmoid.html#a75a24cd75cb4d4c9a072811b2d70ad55',1,'Sigmoid::operator()()'],['../struct_sign.html#aa3304c6b43bcad53061614b741d8403c',1,'Sign::operator()(T x)'],['../struct_sign.html#ac48992b675b8b28be1e27e1f2ec5d2f7',1,'Sign::operator()(uint32_t x)'],['../struct_sign.html#ae07a4249e1b61419a3b9ca6c337b7bb5',1,'Sign::operator()(complex64_t x)'],['../struct_sin.html#a7caf98c777521fa5d5c6ddaaa3b779fd',1,'Sin::operator()(T x)'],['../struct_sin.html#aa510cf4595b6d49065ab6b602d8fcb14',1,'Sin::operator()(complex64_t x)'],['../struct_sinh.html#a02cf32bcf560657b9ee34fb1affed8e2',1,'Sinh::operator()(T x)'],['../struct_sinh.html#a1f8ba1858d352ee68861cd6ea861af43',1,'Sinh::operator()(complex64_t x)'],['../struct_square.html#afde739fc544e45dd30964c02dca94310',1,'Square::operator()()'],['../struct_sqrt.html#ab9b16d2b9b03a1c54190f4479a56a4ad',1,'Sqrt::operator()()'],['../struct_rsqrt.html#ae16699fd829e40416436247a39233fda',1,'Rsqrt::operator()()'],['../struct_tan.html#a1e6fb8c691621c69cb9bd393de4f6e78',1,'Tan::operator()(T x)'],['../struct_tan.html#a2ef120c9f92b0d2e9cec8389eda05724',1,'Tan::operator()(complex64_t x)'],['../struct_tanh.html#adce11a7ad33226c6ecff34f46f5c45d7',1,'Tanh::operator()(T x)'],['../struct_tanh.html#aa8423b43c725bb4b88965a11e8cf20f6',1,'Tanh::operator()(complex64_t x)'],['../structmlx_1_1core_1_1_function_exporter.html#ada4e13daeb3ba0f5ebe20ec0663727b3',1,'mlx::core::FunctionExporter::operator()(const std::initializer_list< array > &args)'],['../structmlx_1_1core_1_1_function_exporter.html#a82aeb5fa32ef5638f42dc2372278427e',1,'mlx::core::FunctionExporter::operator()(const Args &args)'],['../structmlx_1_1core_1_1_function_exporter.html#ac8b8fa0a23d58a94e2e9b923dc7324e8',1,'mlx::core::FunctionExporter::operator()(const Kwargs &kwargs)'],['../structmlx_1_1core_1_1_function_exporter.html#a35a3c1d94249ce0fe0e82b0ea047d441',1,'mlx::core::FunctionExporter::operator()(const Args &args, const Kwargs &kwargs)'],['../structmlx_1_1core_1_1_imported_function.html#a3555db23026d30eaeee265fed99947b2',1,'mlx::core::ImportedFunction::operator()(const std::initializer_list< array > &args) const'],['../structmlx_1_1core_1_1_imported_function.html#a5953b3f47c094cc47bcbb0845379ca8d',1,'mlx::core::ImportedFunction::operator()(const Args &args) const'],['../structmlx_1_1core_1_1_imported_function.html#a10fec4eab5851ed825a9b46a31cedcc9',1,'mlx::core::ImportedFunction::operator()(const Kwargs &kwargs) const'],['../structmlx_1_1core_1_1_imported_function.html#a7d1accece61230eec256e0f70610776d',1,'mlx::core::ImportedFunction::operator()(const Args &args, const Kwargs &kwargs) const']]], + ['operator_2a_27',['operator*',['../structpocketfft_1_1detail_1_1cmplx.html#a26bf3d709a58f06228e502af6db8e5ac',1,'pocketfft::detail::cmplx::operator*(const T2 &other) const -> cmplx< decltype(r *other)>'],['../structpocketfft_1_1detail_1_1cmplx.html#ad9c591ef8ae976293f207937d273e9a1',1,'pocketfft::detail::cmplx::operator*(const cmplx< T2 > &other) const -> cmplx< decltype(r+other.r)>'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a153756072fda6d3e53bcca11b46a1238',1,'mlx::core::array::ArrayIterator::operator*()'],['../namespacemlx_1_1core_1_1simd.html#a08c1e7a00b1b4bc60e30d1554f4f46f2',1,'mlx::core::simd::operator*(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#ae4ec5f1f081d20b46b13eb83eb1b6431',1,'mlx::core::simd::operator*(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a4555cd6a3b50af00700f97fdf00f63a7',1,'mlx::core::simd::operator*(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ab6a73491bcb185cd91ae4db6b0f21e49',1,'mlx::core::simd::operator*(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value *b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a4030444ea38ce1529a8cbb8c183a28bd',1,'mlx::core::simd::operator*(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a *b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#acd5ac48dc7895f06daf55f0a7e0667fb',1,'mlx::core::simd::operator*(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value *b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a6f6d26e3fe39ee1ba0a7380d0ecf7b45',1,'mlx::core::simd::operator*(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a5373c1af09825b5f701ebd106508fa6b',1,'mlx::core::simd::operator*(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#ac50da923a4b7ac682554bd1d74c306d9',1,'mlx::core::simd::operator*(T a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#a681d4fb076973f58f7dac894ec62a385',1,'operator*(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8f06316063fc91747533105f256b55b5',1,'operator*(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7b3bce3f6f17089d87e13e91f580a581',1,'operator*(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a54ae7216b82c5cea362f6b83e1df3a9b',1,'operator*(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a852689073c17596de4fb545bc046b380',1,'operator*(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a168300bbd04d8e97c5e4218cb14ae378',1,'operator*(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6278bd2e0e2805090b33ef666bf7f6bb',1,'operator*(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aecf703522d9ce32dfeefe1e6e903db06',1,'operator*(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7cd44d27fa9a4f13df39894c34fdb348',1,'operator*(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee64dc1890abb6d1035361cb8c751f96',1,'operator*(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1a559ab88dbbb4fd2c7509d2c94e55b',1,'operator*(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a495ae2d9be5d97c4c6448fc4e50a03e1',1,'operator*(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a87ab4b7a502430da664ccb8abd383058',1,'operator*(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5f997839cf49c24ab594a0dff486a7bc',1,'operator*(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa0c2d29950926ae579adf6337fbea64b',1,'mlx::steel::operator*()'],['../group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0',1,'mlx::core::operator*(const array &a, const array &b)'],['../group__ops.html#gac22a67f7de797b1ae59029843cbdcab6',1,'mlx::core::operator*(T a, const array &b)'],['../group__ops.html#ga6f2369ed5fae8ff9b1528670a004dde2',1,'mlx::core::operator*(const array &a, T b)'],['../namespacemlx_1_1core.html#a0cc824d6318f97f7058918ab64ddfc25',1,'mlx::core::operator*(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a81e1c727c3fc48910b030cb65a9e7afa',1,'mlx::core::operator*(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a861d948220d8f48d46c68d2ddb16a096',1,'mlx::core::operator*(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a13d16561812679b36e68185dc4b2d04d',1,'mlx::core::operator*(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a5287610200ff573730c9c92413f48881',1,'mlx::core::operator*(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a377ccc6b4ef36767abca102dca56dc10',1,'mlx::core::operator*(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a5d696b63635ce6967526d6a410f7f6b1',1,'mlx::core::operator*(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abe90e9527bfa3e1c813d41df4a2372e7',1,'mlx::core::operator*(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a5f14963c77f96bcb5a3bef5661a86ba4',1,'mlx::core::operator*(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#acfb06fe9f5fee01dbb5a2b23bccfd0d3',1,'mlx::core::operator*(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#afc9a87f1fccbac05242b91bfbb35c24d',1,'mlx::core::operator*(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0b9678af9b487900cacf6639a4693de0',1,'mlx::core::operator*(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ad5950619081389e6ed7512f38358d33d',1,'mlx::core::operator*(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a65d25d082374761c05b056e1046d1d4e',1,'mlx::core::operator*(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a759191fb984e7737f0ef529c2053ad73',1,'mlx::core::operator*(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3a52675c3d4552b319dd9707844abdec',1,'mlx::core::operator*(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a45d67f5d80fba4d42e34c682a8d22beb',1,'mlx::core::operator*(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ad25880c67bbcbfafbe54dc16418bf736',1,'mlx::core::operator*(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a63c836e1141e07ae72cee770bad01200',1,'mlx::core::operator*(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a265a37b8ee4a97390213e9ec49693e66',1,'mlx::core::operator*(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab5a457da04dcb157a0b5172c4b2244b6',1,'mlx::core::operator*(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#aa56a8bda08be9ef3711496e216a75c95',1,'mlx::core::operator*(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af89612098dd355b1eefb841c753b36ab',1,'mlx::core::operator*(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a4552687a0637f710b5d55bb6378fcabe',1,'mlx::core::operator*(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af69db7def588d7da430434a69456e29c',1,'mlx::core::operator*(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a00af6e5095888f00791ee0ab6d993ad6',1,'mlx::core::operator*(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab48feddc1aa304383e5493923506ad7a',1,'mlx::core::operator*(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a0367b582e85162b4180e086f725e49e9',1,'mlx::core::operator*(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a45f0479526fbccdb00bc73ea7f3b7625',1,'mlx::core::operator*(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a394797646010ba9ef2a1f9b9a4b8ddd9',1,'mlx::core::operator*(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acaaa86b59c7ceb2e092ac07f2a75225c',1,'mlx::core::operator*(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a067d47823a322b88043cce7ce4a3ec78',1,'mlx::core::operator*(bfloat16_t lhs, float16_t rhs)']]], + ['operator_2a_3d_28',['operator*=',['../structpocketfft_1_1detail_1_1cmplx.html#a683fd490182c9189fa2c05b1823edd93',1,'pocketfft::detail::cmplx::operator*=(T2 other)'],['../structpocketfft_1_1detail_1_1cmplx.html#a06f2c26c6fc4722e61b44da4c242ed87',1,'pocketfft::detail::cmplx::operator*=(const cmplx< T2 > &other)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7232b0a0e193b3c6172d6fc2578bf419',1,'operator*=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ade65ebca11e38d56408c512df89b99f4',1,'operator*=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af4348ce3425dd99d069e8fdf06e25a3c',1,'operator*=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2c3c5f793b3d957d7295d7f1faabebee',1,'operator*=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac66657077d55e94197b52b63acb50b7d',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a383165ea838cc3feeee4d9cf54aa77cc',1,'operator*=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab706af260b61f735b28464877d02137c',1,'operator*=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a979374b1dd4e0eaf602326fa901336d1',1,'operator*=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac815eec2c1b15a47b1c6ea6790e77d24',1,'operator*=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8110fae7bcc34a0de5927546b24aa935',1,'operator*=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae4acef3e7ae7dfe359422503f894e885',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc268cdbc30500f3009f5de2b2f0f67a',1,'operator*=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a81f65b04a87a25c7eb1a751d1be9fa55',1,'operator*=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08c1f916302eb9d48c93f8b7260538fe',1,'operator*=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc8e82b8f593b12c6d405e2250ab0f62',1,'operator*=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4611728172afea51860a77fdb06cafa0',1,'operator*=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0b8736e2ae24758b6e24ea72668df5b4',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad920df9579603f0b0ee2689eba330617',1,'operator*=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae97ab6c3ddcc2754b24f86319a5398be',1,'operator*=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3ff4ff59f411010ac8502cfabda4bd6f',1,'operator*=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abd3d82e2dec1847e97eb8fc3bab2985a',1,'operator*=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a738078eb7d5ff94ff48156a555d763a5',1,'operator*=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a435f2f4256aadb1b57fd62bb7f733cf7',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0e4377b120d6305335d296e031ee5b30',1,'operator*=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a917354f77eac26189da8a2f610a00074',1,'operator*=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af725f935bfa0405e5ff17ede3ac47283',1,'operator*=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7c56980c234a04260b8b19298085e526',1,'operator*=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab840ff9de0cdd0e9afffb8baa2a850a3',1,'operator*=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a73416a7415f3fe31525e33419e5e8aab',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a16978f4b16d954ef4d4cf0f32f6c0b94',1,'operator*=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a99aa4cc110d1c7aa3b4c8c5cbf9235b7',1,'operator*=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2179abbc91ce8763e96e39e1917bfa6e',1,'operator*=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab070ea4676d10a10ff3e9379a4068a57',1,'operator*=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0197e039d4c65bf49649a6f250c2d436',1,'operator*=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad3565cc6fd1e088d052b1108aa065851',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a711693988c437c2fb4d7da505982fe21',1,'operator*=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeff4c28986f98c23de1df17043edb0f5',1,'operator*=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7dbf0c75df4817cb4ef8b60c417a89d0',1,'operator*=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a323a80492cd17a49e2c3dd18f8c8b5cc',1,'operator*=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adb465776d3868bda0525d632ffc4d129',1,'operator*=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a12a98d71d670b409b8065e0d61672d55',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5d00eb2ec2b0e15b2753d100694c45ae',1,'operator*=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1a2a683ff40490226eb1371fb905023d',1,'operator*=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4126fb7ed5bbb27a2332c543cf56a337',1,'operator*=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab092d9790ef20fc0386707530aee89db',1,'operator*=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abff1fd2439e31e6e64a3d2fdee3c7821',1,'operator*=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a625dcb133f1f953f263e6200399866c6',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08b6071245513e1726ec68e3b63edc53',1,'operator*=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a13aa79165ec87710e977f33fe0361e91',1,'operator*=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3796dcf819adb1ef8152f57ba63ff6b1',1,'operator*=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aaab79d0b4c9e9bdc059ace6ec58c5b00',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a0dd3893abc8986901872c8365ab1509d',1,'mlx::core::operator*=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a3cc5c154e4ad9a83ad43da8513146fdc',1,'mlx::core::operator*=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a600e77dbc72e78207b5f5dbf4b298781',1,'mlx::core::operator*=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a54833be1d44bc3adfc9ea218fc3685bd',1,'mlx::core::operator*=(float &lhs, _MLX_Float16 rhs)']]], + ['operator_2b_29',['operator+',['../structpocketfft_1_1detail_1_1cmplx.html#a76447ef141c8732d57421749fc81b236',1,'pocketfft::detail::cmplx::operator+()'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ae2adde594b5a4853f6bc78263a957d85',1,'mlx::core::array::ArrayIterator::operator+()'],['../namespacemlx_1_1core_1_1simd.html#aac6acd134f1498b4fb45fdbc882335bf',1,'mlx::core::simd::operator+(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a8b622c47d07b171b2303ea744bf72284',1,'mlx::core::simd::operator+(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#aed655ffa017ade5e0f954f906d9f7ae6',1,'mlx::core::simd::operator+(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a27dfc3843dbefbbebed5b7137bacbb59',1,'mlx::core::simd::operator+(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value+b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#aa78806bf6a3be64b44e9a1f04bad3862',1,'mlx::core::simd::operator+(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a+b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a98b77f1ca24bff373f48ef62f0013a02',1,'mlx::core::simd::operator+(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value+b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#ae690b57b386cbad40565487d6d2393bb',1,'mlx::core::simd::operator+(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a417109cdd61f35954ba2cc37af9b4460',1,'mlx::core::simd::operator+(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#ac33643b5f3cdbd3be0fa7d5784e35007',1,'mlx::core::simd::operator+(T a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#ad6af5c6c5ed4898b49758618e5aee189',1,'operator+(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a09c1a797eb7f43742578680899932f50',1,'operator+(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a551b970f73bb4a3b287653021d000b60',1,'operator+(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a43a225e7e548bb041f3a5d844faaf0da',1,'operator+(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8b6c3fd9d068a2159084359df8b9b449',1,'operator+(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0a5bfe15d95ba540795f4c25ebfa4f07',1,'operator+(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa415ce182fe7582d885fe633fc3527ce',1,'operator+(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a62f891b7dbba0000749cf338f594bedb',1,'operator+(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab43932322f81bf322aa1b0deeee9a987',1,'operator+(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acd15d46ea5827a2a39898ccbb8352eb8',1,'operator+(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a006763fae6e0577fc168ec9446f0f747',1,'operator+(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a12a47e8ac0be788edff57ae0a96d7830',1,'operator+(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af87dfa2122e9c76042dc41fb7f338a87',1,'operator+(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af2737d09c887ee8cd43fdeabceddbe82',1,'operator+(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a12ff4f38aa8474bf76770c7b8e3e18cb',1,'mlx::steel::operator+()'],['../group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c',1,'mlx::core::operator+(const array &a, const array &b)'],['../group__ops.html#ga7d0ec8d01e7cefa6a6b25f11876761b5',1,'mlx::core::operator+(T a, const array &b)'],['../group__ops.html#ga7cc080a4f9d4a667f2099aa0dbfefadd',1,'mlx::core::operator+(const array &a, T b)'],['../namespacemlx_1_1core.html#ac14b984970cafd8fbe24d080949515cc',1,'mlx::core::operator+(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab076069c6f0047c548a8dc29d35dd36a',1,'mlx::core::operator+(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aab9d96b0a168f4d05146000a6212b5d8',1,'mlx::core::operator+(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac4e6f03d7e4ae701b4eefa784f36185b',1,'mlx::core::operator+(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a4cabd600a5271b0d416c91e8d31dd9c1',1,'mlx::core::operator+(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af26df9dc279d71b7cc10892c72162b58',1,'mlx::core::operator+(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#ac3b97eecec9bd8efb313f8f201560343',1,'mlx::core::operator+(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2e3bb121cbde30c2e6d806df0d41ff59',1,'mlx::core::operator+(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ac87ecce4b44b0826e666a169ddc6f878',1,'mlx::core::operator+(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aed3d9cd32698ef0fe65b1280f103b3f5',1,'mlx::core::operator+(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6fa13b9359cf3f575fbda5260e6e035d',1,'mlx::core::operator+(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af240a6471ff827819192808bffeb857a',1,'mlx::core::operator+(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ac25a05679f312b724c406d8b282803c9',1,'mlx::core::operator+(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a54863a54f258acf2b5c734950618e4e1',1,'mlx::core::operator+(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a9f81f5ea8909db9660197217612ee446',1,'mlx::core::operator+(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a13e26c38da0a4e332e0ae4eb0aed9cb8',1,'mlx::core::operator+(const std::complex< float > &x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a59bb13a0bb7f748c8de34415b248bc57',1,'mlx::core::operator+(const complex64_t &x, const std::complex< float > &y)'],['../namespacemlx_1_1core.html#a38a44c412c8be4c8b952d3082cc7db74',1,'mlx::core::operator+(const complex64_t &x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a011dbdbd2413e59e744cf82b05431340',1,'mlx::core::operator+(bool x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a230e3b7c479add1b171fa0aaa3a8b13c',1,'mlx::core::operator+(const complex64_t &x, bool y)'],['../namespacemlx_1_1core.html#a3a6f43c2485f0d42293184f1aecbeaee',1,'mlx::core::operator+(uint32_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a766157c5d5d00fdf3da95eb7cb2981b9',1,'mlx::core::operator+(const complex64_t &x, uint32_t y)'],['../namespacemlx_1_1core.html#a64dceec2bb03eee963a2a1bc1ac69284',1,'mlx::core::operator+(uint64_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#ae36badb78a17cd7d13663a69645fc328',1,'mlx::core::operator+(const complex64_t &x, uint64_t y)'],['../namespacemlx_1_1core.html#ac1afa5d4c856e4b58109eff086e70ffd',1,'mlx::core::operator+(int32_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a8978def3c2cfe2a96314d564613b80db',1,'mlx::core::operator+(const complex64_t &x, int32_t y)'],['../namespacemlx_1_1core.html#a5b8af5ca4c0e37aba0b7530542bd64c2',1,'mlx::core::operator+(int64_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a3eaa72850205c18450c3af9a01cda219',1,'mlx::core::operator+(const complex64_t &x, int64_t y)'],['../namespacemlx_1_1core.html#ad38b38a3faf050735d45eed4438ee27a',1,'mlx::core::operator+(float16_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a358e66ff205bda3e8542427b6d2edadc',1,'mlx::core::operator+(const complex64_t &x, float16_t y)'],['../namespacemlx_1_1core.html#af56d4b85e329e39a825c01a50e3a2522',1,'mlx::core::operator+(bfloat16_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a806a495a129ebaab69cc57ca7db831d6',1,'mlx::core::operator+(const complex64_t &x, bfloat16_t y)'],['../namespacemlx_1_1core.html#a09fc6ebda917969383783a112a8547e7',1,'mlx::core::operator+(float x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a7ed0e2cdb65612f54e67166762cb6408',1,'mlx::core::operator+(const complex64_t &x, float y)'],['../namespacemlx_1_1core.html#af7577c91b8c43682f0ebc9eb9758aae4',1,'mlx::core::operator+(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#abe36af9951afd8dd3ffe90ceedeb7f2b',1,'mlx::core::operator+(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#afb9f780dd056a4f975518f71a3b021ee',1,'mlx::core::operator+(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a6a8e093b24c4c789b7cd160f7e7f7de9',1,'mlx::core::operator+(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#af3a603690fd3de9e4f7f2035a4d25621',1,'mlx::core::operator+(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afa2a4bccfeea9688ac922cb638341511',1,'mlx::core::operator+(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a6111e94d51de12391e5d68b765f28fc3',1,'mlx::core::operator+(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7c7dd6d346e0cdf398a896f2c6958258',1,'mlx::core::operator+(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a00872a443f462b0ae0a30c84fb001bc0',1,'mlx::core::operator+(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4f5d80d03bae6d8d90455d3c47a8c116',1,'mlx::core::operator+(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a78f1f388f9d81ed93f60311f4645d8d0',1,'mlx::core::operator+(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aa43e1d6958c5d5a6fa9a625a1660e741',1,'mlx::core::operator+(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ae877e1d5e3cf57734da8b49535fe3fb3',1,'mlx::core::operator+(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a9a5ae769f67f886d59c8e292a8218550',1,'mlx::core::operator+(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a058878237ce50baa4c909d8d15448d7e',1,'mlx::core::operator+(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a95fd207028f125eefbafe9e0522407fe',1,'mlx::core::operator+(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#abc6425a3fbb386f5ea5964b42507e989',1,'mlx::core::operator+(bfloat16_t lhs, float16_t rhs)']]], + ['operator_2b_2b_30',['operator++',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a3efe69356a84d0d4438f033992fcbd9d',1,'mlx::core::array::ArrayIterator']]], + ['operator_2b_3d_31',['operator+=',['../structpocketfft_1_1detail_1_1cmplx.html#ad4e69dcd89bdb7764c9c5807168f911e',1,'pocketfft::detail::cmplx::operator+=(const cmplx &other)'],['../structpocketfft_1_1detail_1_1cmplx.html#affa618d8850a7c232793b7c61db6d184',1,'pocketfft::detail::cmplx::operator+=(const cmplx< T2 > &other)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab04f480aea9fbba0895068c7558dd400',1,'operator+=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a251780ac4592cc2b1a543e417ff57770',1,'operator+=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24381d991c2d570aa953694f396a69b5',1,'operator+=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7595740d4cc12924905d6bd1b99ee4da',1,'operator+=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac1498acb8c3623b5f412f70ab6a6528b',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abce5ab327110c164f054b43ed47f79a0',1,'operator+=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae0c70198e236ffe1a98f79987c686419',1,'operator+=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a13b3338935440ae51ecc4a356093efc5',1,'operator+=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a0cb8544b4ebd2906ba8e7f2868e8de',1,'operator+=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7b134429ea0c8493800ff8b465410f9c',1,'operator+=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4154f90ab7857ca856f9e15fe1bf5acf',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab9ae6a51e2027b02cac9966e05f3ba68',1,'operator+=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab93ce536eb7998bee00de4af868e31a9',1,'operator+=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad0ae9e2b4874f991a2c853e1c1fe735d',1,'operator+=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a194a6670cc25ade35a24b566f31af785',1,'operator+=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3d0d689516c99003659c5d026847bd2e',1,'operator+=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a007f58508b98bb79e5c323ed0dec89b6',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa7198e580e2a83c1fd01a4b6fdf86a80',1,'operator+=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a15573fefd880adefbba079b1c1bd8082',1,'operator+=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a104cf94cb9e359d1b6ef92ced2ce0c27',1,'operator+=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa04cfcb52191fd23205a1a3572b46ae0',1,'operator+=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad684bc2ae1a2a627cd3e4a4c641e2d77',1,'operator+=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1e28448e35f4934075b397c34ba3d66',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8ad16afd7f1711de83c0cec5af868f76',1,'operator+=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac45e9ca0c7155caebe3d0f7261518077',1,'operator+=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3c62ac679d6aa515144d40ebafe4a188',1,'operator+=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9ff5ab3aef1057fa083b53a65c8aba03',1,'operator+=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae74bb0a3c12cd1a23f3d29ce307d6fb1',1,'operator+=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac188bd19f236b098d603b0d8acd08921',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef9fa600d107b509f2e3df7d6b080e01',1,'operator+=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af5713afb3a62967a02c3c20661951ee4',1,'operator+=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7f1b84352a3ed6171444a43da1fc7e92',1,'operator+=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af1983edd26245e6e51c6e47354095e32',1,'operator+=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8cd55d1a579540eb450e12a8a8a950be',1,'operator+=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a588ef0f7e03f306758524d378278976f',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a74751abec7086f85f4f26ced44f1ca1f',1,'operator+=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4dd3cf0e5aa116ff330352a50c18cde7',1,'operator+=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afb9a0e18c0e40c77e6143fb7d84ebfba',1,'operator+=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adf0cfd9a608a6fb3d57933e32e7d81d2',1,'operator+=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4bd92db6c8b9b5dc96332c7ae3eff8c7',1,'operator+=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5d628a5bc4fa755610392f47a523a1f1',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7c790442f77f2437b482c4a55e224fc3',1,'operator+=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a77bab4481b41be50297b257e95058706',1,'operator+=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7816a97d16b1d2f8a90227bb1da2f6ac',1,'operator+=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac244d140c6149726ea44174d3e836ca3',1,'operator+=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af802541c4c65ee4442acd495de4d27fe',1,'operator+=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac06eb2fea47a09a8a8abdaa1aa9b4603',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5263b2463fecdc97f9521d00bffea059',1,'operator+=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24ca436ab299a710263d65302532dd3b',1,'operator+=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee1bdf0ab2e445293708b476e8cfde3b',1,'operator+=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a190e27077f0fba642a86f5c8f488bcc2',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a9f2c9d2f21fbf9fbbacd940c6967c9d1',1,'mlx::core::operator+=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a0b1b3c48afc0a785282e43435bba8418',1,'mlx::core::operator+=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7b763db8194e6fcb1b87eab143dfa47a',1,'mlx::core::operator+=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a827167f6a1ae55428fd218ddd51ec3b6',1,'mlx::core::operator+=(float &lhs, _MLX_Float16 rhs)']]], + ['operator_2d_32',['operator-',['../structpocketfft_1_1detail_1_1cmplx.html#a460da5db36d1c72fb1ed3496fd3abde4',1,'pocketfft::detail::cmplx::operator-()'],['../namespacemlx_1_1core_1_1simd.html#af5be79b8dada8f8e91ae7c03c16606ec',1,'mlx::core::simd::operator-(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ad5761065b4a655cd086d88846ae08d97',1,'mlx::core::simd::operator-(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#acc490f7f5195acfa7b7c5df7afb39438',1,'mlx::core::simd::operator-(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a678cddce777549a39474449d56fd1de6',1,'mlx::core::simd::operator-(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a70563bcd6c28802d11199812ffef38c8',1,'mlx::core::simd::operator-(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#ab1f7f553d3a9176a70404a29cad06619',1,'mlx::core::simd::operator-(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value - b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#aa73282cb05b65b931b97ce35c46bae20',1,'mlx::core::simd::operator-(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a - b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#ab35a129d6e31b86c06b61252c7b26d4e',1,'mlx::core::simd::operator-(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value - b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a727a13b3d26f9e7cae7f091105867904',1,'mlx::core::simd::operator-(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a6e39cc693b30ad8e530392baf4bb5b0e',1,'mlx::core::simd::operator-(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aad9cc064528e4189a5b7dd816a134ae6',1,'mlx::core::simd::operator-(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a7434ba1ab2ad798fe8557a9b45035e81',1,'mlx::core::simd::operator-(T a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#a226cfd54d49f02e35c5aab3139c7596b',1,'operator-(complex64_t x): complex.h'],['../backend_2metal_2kernels_2complex_8h.html#af5608264cf920688607059b4e8cd3117',1,'operator-(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6aedc8d6d0980134ac69b96f22d9a855',1,'operator-(_MLX_BFloat16 x): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a333f67614dbf8027439a7e124052cb85',1,'operator-(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a891aa4bf46c20a26a55061736aba25f1',1,'operator-(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7ad7ff44a3200853711869f7a577d931',1,'operator-(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af69ef8f1d8ecae0e6f755bf1c46cf075',1,'operator-(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5bd875a54b79b2dcedf674807c3e53c5',1,'operator-(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab02f8646b47806e1d2038f248df03f06',1,'operator-(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab27b26182c7c6e08af37e6d511fd9253',1,'operator-(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5868c85c988ec3432cf86d7df40e464d',1,'operator-(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad03ef47e6cc7521bbfb45740dee20f88',1,'operator-(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab789f8a400512ff27e36b3373170f0c5',1,'operator-(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7f601b22ecc480132d82ad782e5363bf',1,'operator-(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a152366ab4e2ccc867e919af6c74ced91',1,'operator-(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a42bead8ef0beb9f3452128d64cd4df9d',1,'operator-(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aca8ef21c16984ccb329b3bd0c1e4be48',1,'mlx::steel::operator-()'],['../group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816',1,'mlx::core::operator-(const array &a)'],['../group__ops.html#ga0c7f3cb36d4ca516c7a33142f88b9181',1,'mlx::core::operator-(const array &a, const array &b)'],['../group__ops.html#gae68d3d0691ba951501218e98439f3465',1,'mlx::core::operator-(T a, const array &b)'],['../group__ops.html#gaf5e5d882c51ad0a0ea315c274d5439b2',1,'mlx::core::operator-(const array &a, T b)'],['../namespacemlx_1_1core.html#a622ce842fe44e4b6a95e03242341b459',1,'mlx::core::operator-(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af32a99d930d49e9b178472d7a65531ab',1,'mlx::core::operator-(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a3555a2b31fc0925850d3240e85e03ec5',1,'mlx::core::operator-(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a46080889fd9e5c3f9916508e97dff5ad',1,'mlx::core::operator-(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a9ca27fd1e512c8ed126342e565da12ae',1,'mlx::core::operator-(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3803f8d36558d32bb7dd6e580ea683b4',1,'mlx::core::operator-(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#af5d865528989ca66b3d357e5ce4e0300',1,'mlx::core::operator-(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#afb784b960f55aeb4edd7f567fa74d443',1,'mlx::core::operator-(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a29cbacf4b399c24728fb0808fad498f9',1,'mlx::core::operator-(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aececc0e451237aa6c0d1a2c3d828c86e',1,'mlx::core::operator-(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a383a26cc2689c98fd6c4435ade8dc669',1,'mlx::core::operator-(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad6311ef8df59bdfb212b5cf8169246b2',1,'mlx::core::operator-(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a23b7329bc1c93c8ac0a1f576565fefb0',1,'mlx::core::operator-(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad8d650bf63998abd716ee0ca28e1cbb9',1,'mlx::core::operator-(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a7339b33201254e9119d99d3a728ded72',1,'mlx::core::operator-(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a064318b7a16e5cb6d0a6407501b5c7dc',1,'mlx::core::operator-(_MLX_BFloat16 lhs)'],['../namespacemlx_1_1core.html#a7bae3ff296d9a60ff3c7e448f7fbc6bd',1,'mlx::core::operator-(const complex64_t &v)'],['../namespacemlx_1_1core.html#afb5069ecebdfd9d388c26f83df12c93c',1,'mlx::core::operator-(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a8d126e3f3fa9f8c1c1ae1b09f94df487',1,'mlx::core::operator-(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ad04f1ccd2cd7c487a2f2aaa055939f64',1,'mlx::core::operator-(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a15eb2ea76508ff823fa0591e811d0b7d',1,'mlx::core::operator-(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a96d9577db38d6809d022893e32feeda1',1,'mlx::core::operator-(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a5d9c02765c1672930757416411567bf2',1,'mlx::core::operator-(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a6105d3b5266666b7c6bb9469285a9ec3',1,'mlx::core::operator-(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a777aa772dfb205b25d26f3180d98a2f6',1,'mlx::core::operator-(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a085eb092f4ada47f8169de62886cff90',1,'mlx::core::operator-(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab25e5d211e2c8785b45c3a81a6282e2b',1,'mlx::core::operator-(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#abf5d09561a81b0f0b32d59d77e32e16f',1,'mlx::core::operator-(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4ce6867dbb4d1631d1870dac14022dbb',1,'mlx::core::operator-(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a8a049e646e0442064cfe9e202d7047c5',1,'mlx::core::operator-(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a78e2a1cfc65453185bcca13bd4f523cf',1,'mlx::core::operator-(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#af143cf68673e06390d4bb2ec2892bd22',1,'mlx::core::operator-(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a46d502dfe0b027955950d4e716c2eb26',1,'mlx::core::operator-(_MLX_Float16 lhs)'],['../namespacemlx_1_1core.html#a2631e78c6f0a602f6754ac577ec75f83',1,'mlx::core::operator-(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a73d79cbd75d543d0837b8a51bf103f9e',1,'mlx::core::operator-(bfloat16_t lhs, float16_t rhs)']]], + ['operator_2d_3d_33',['operator-=',['../structpocketfft_1_1detail_1_1cmplx.html#a12441ff423274bd1b54245933d69ad7e',1,'pocketfft::detail::cmplx::operator-=()'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab225043bd02bb423930bc98aae9c2bca',1,'operator-=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac2f1e1f2365cfa531b1519aa9ff67695',1,'operator-=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a513501355a5912a1263fd8b10864142b',1,'operator-=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab4f4ecd62c3d8b3363d02019573dc9f1',1,'operator-=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a92d1348f201d78fcd474f75d5b23ef68',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3eefe9a7f5fb226335ea687012f32d5c',1,'operator-=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef62c7e3e494b6a511a7833c0d942a60',1,'operator-=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad30726cc8b69fd300d33c2a46e123c28',1,'operator-=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8859b5b8dc241e4f58243c85d2630cc8',1,'operator-=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7003e1e5881e3d106257f22b6a3e59fe',1,'operator-=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3165e37d393be50c2cfa9ddcba153684',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a76f5bd895b7214cbc3cea3440992718a',1,'operator-=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7167343d90eb70e5a0d5fa9ec5398e94',1,'operator-=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9b31c363ebc93d592b6fa0e27b00335a',1,'operator-=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a967a1d7b5664f616e5b6f2d257367f0c',1,'operator-=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aff19193e1b2cee29a8737318e95cc74a',1,'operator-=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aede0cc4179507b739849948f1a2fed4b',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7e1a6056f9c96f3c89fe204dbf103be5',1,'operator-=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9d06cceea5c179bcc608452188bd7d6a',1,'operator-=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0aa9ffe056f49fda181bbacbd60556ea',1,'operator-=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ada5685d99c2d6708d1c4ef826d68e879',1,'operator-=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a726cecf778b8584b6f7c37db1b064576',1,'operator-=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3816a35f8468156d59c239256c12dcf3',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa332fae098e7c6dc23b98bc0026f1070',1,'operator-=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afb3cd302e0b78902c62111dce4494fe8',1,'operator-=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abb884888f14086cc674657677cb4b8bc',1,'operator-=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a38bb89f925eca4f9c042f6ee7a2c0193',1,'operator-=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac30c580713f354916088a7dc049ae4cd',1,'operator-=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a715c824ee8c87e0256114a85624d9949',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7bc91aaaf476a37063264d1d53d862cc',1,'operator-=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab155f418f15cabd86ff942c6f9472ddb',1,'operator-=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aaa66dc6d7b2c5efbfaa97ca9c7872bd8',1,'operator-=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a696978d9401e09200045b2d8aad045c2',1,'operator-=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae998d8f423a9fb73405cfbd4b836bc72',1,'operator-=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a279d09ab8542f1c1a8dc8173b65946b6',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a491dadfae957cd7cc0c36188d910f6f6',1,'operator-=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9a837c3b9c4e42f53d7cd1ed0d266e2f',1,'operator-=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acf7af2284269544064b68e807064bba4',1,'operator-=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a28d297705e29009197418546ef435393',1,'operator-=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a948579a4d9ba276523190b03b09578fb',1,'operator-=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a4b98a0a11db5b77cf9168df37c8bc7',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31a3d8f2ff8038f7e0d717845c039808',1,'operator-=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1dac193d9f1c8c0eb4473441895f8c58',1,'operator-=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad2817d53fdd4b112babfb6f0b38c8f39',1,'operator-=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa705d87cf4b78e9d7c6b07dd0c66cac6',1,'operator-=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a542affc376726840647a6e93acf2c1a7',1,'operator-=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#add18cfe4c0d38e95c6dff6bab3e7a932',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab1de7e7e7304ff3598925d2e69134764',1,'operator-=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0d3fb52437c677c5d0f1a3642384b15c',1,'operator-=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adda64cae388baac1f138b06dc8595237',1,'operator-=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af20874a61c6c3f4c3fd045a96e806644',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a8b8a55690df46d97fcfc2a60120783af',1,'mlx::core::operator-=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#ab03949b1f60fa035ce454a894cd73ae9',1,'mlx::core::operator-=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#adaf70bbfb3667df0d08fd3c99896e20a',1,'mlx::core::operator-=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a321c98e5a78621d3c9a3895f707f2f1c',1,'mlx::core::operator-=(float &lhs, _MLX_Float16 rhs)']]], + ['operator_2f_34',['operator/',['../namespacemlx_1_1core_1_1simd.html#ac86a54a5e2ccc79bc92739f143bc0bef',1,'mlx::core::simd::operator/(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#ac5d10f465c21ab259041042ff0159187',1,'mlx::core::simd::operator/(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a18a2689f4ae197c5b204fe9b3370da4c',1,'mlx::core::simd::operator/(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a1d45c3b97cecfff86a2e43ae1f7fa185',1,'mlx::core::simd::operator/(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value/b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a89be64949908f19dd42aa7e38b320b0c',1,'mlx::core::simd::operator/(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a/b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a1c61bd3ac3ec5d8d2da65b45d59f543e',1,'mlx::core::simd::operator/(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value/b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#aab8837750c84794369e630d8ea0b408c',1,'mlx::core::simd::operator/(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a0585ea196b665710115e48b7ebef0fc1',1,'mlx::core::simd::operator/(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a075f637ff3f983ada0fd6288ab8d91d7',1,'mlx::core::simd::operator/(T a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#ae6a708f67d6fd9b0962aa8877cec6d35',1,'operator/(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9f16a44e1c9836ca57edc1d7b93b5d7c',1,'operator/(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aacaedf12f862c76457133336dd6fc446',1,'operator/(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a584a513596de20663dad951a5b81695e',1,'operator/(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad8f7b11669736fbd6ed2e28211d877d4',1,'operator/(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a59515695ebc48844345fa5120511aed1',1,'operator/(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8c8ac6736440fdca366ebdefe2a12b9f',1,'operator/(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad6859b04680d0d26d75fd6c4dd74ee24',1,'operator/(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4720cc79ab2b8e39952ea9ef20e51250',1,'operator/(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a72d10ec0e62949247da129eb3a83fb9b',1,'operator/(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad6399ba2b8708899739b4cdbb44add8d',1,'operator/(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a998b1ba877a606aedf722ab46b290403',1,'operator/(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa3277ae33976c70f7bd937ddff027b72',1,'operator/(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa708a970a200822c99c0489f389469fa',1,'operator/(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a6bde717aca2051499f73a3eee199bfdd',1,'mlx::steel::operator/()'],['../group__ops.html#gaeedf77f722b394429f1a7f6c367883bf',1,'mlx::core::operator/(const array &a, const array &b)'],['../group__ops.html#ga7366ec7f453be2a4dc449f0faa1bf554',1,'mlx::core::operator/(double a, const array &b)'],['../group__ops.html#gadfb324ae9b4feb2c7ea0ac6ade639f38',1,'mlx::core::operator/(const array &a, double b)'],['../namespacemlx_1_1core.html#a7573ac3b93ddecd69e9c88a26fc84ba9',1,'mlx::core::operator/(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a40e868dad70401d9aa9ee9c32235c315',1,'mlx::core::operator/(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a7587c28fbd2023b134e5fc12bb0dde23',1,'mlx::core::operator/(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a92cdd377c408becf4cf83c1ee9b7085d',1,'mlx::core::operator/(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aef89566301cb133d98c8e7bdd2b7bec6',1,'mlx::core::operator/(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a94e7b51185590492b46916685641276f',1,'mlx::core::operator/(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a04584788c08180835219d0ea1e2b97b1',1,'mlx::core::operator/(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad5af96e2ff09d207eb1e1980fe3e7c2d',1,'mlx::core::operator/(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ac2217bf760038cd011781158923149ed',1,'mlx::core::operator/(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aea414c04bddc4b9b609262e97398f1b4',1,'mlx::core::operator/(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a27fe23230cd082c0363b9451b731ce6b',1,'mlx::core::operator/(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abdd9bb8fb4411e5924f3eb7ef1bb52f8',1,'mlx::core::operator/(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a50bae338a7353f8b0ed3441071bb0cf6',1,'mlx::core::operator/(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aab26a3284dd3ac7d47c8b5b3a3290ce3',1,'mlx::core::operator/(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a749f48db01de38f259a0c6750a97fa77',1,'mlx::core::operator/(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a32a6a08a2a4652975b0a1bd1fcf3eafd',1,'mlx::core::operator/(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4b66fb38ddc5cc0c2489583d5c499602',1,'mlx::core::operator/(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a45726f1905b709cf8253e6efa046027b',1,'mlx::core::operator/(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afd4170c1e364384f30e6bae341146fa6',1,'mlx::core::operator/(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aef85739d150b9d5609973da8a3f1086a',1,'mlx::core::operator/(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af52a941f8ed9b25eec91402c7b9e281f',1,'mlx::core::operator/(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a477cade78296bc85894170f62db68870',1,'mlx::core::operator/(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a22f5a2257e11423fc2fe18e2dce91590',1,'mlx::core::operator/(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a640d3574dfe6ad934c720ae8bdd78bfa',1,'mlx::core::operator/(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a6f65d8fd0cdddc96fc01f6af95804873',1,'mlx::core::operator/(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a517019d42d4e426b7b98e1c719bb47ce',1,'mlx::core::operator/(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a0beb7a223c542015a4eff4aed814a9dd',1,'mlx::core::operator/(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#abc9b1bd5018d46514bc19d23db2e5063',1,'mlx::core::operator/(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af22937df654ddbd6e398ef12764d18c0',1,'mlx::core::operator/(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a775aed5f49b530c57e71cbac81404d45',1,'mlx::core::operator/(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a97efcd96d6be666e5608034ae77289ef',1,'mlx::core::operator/(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a899851f85dbddd96f9d36319b82542a0',1,'mlx::core::operator/(bfloat16_t lhs, float16_t rhs)']]], + ['operator_2f_3d_35',['operator/=',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5aa3b8c68a2b58d41ea33eaabbf83095',1,'operator/=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a90a1c5130db515db48624d8587edbb91',1,'operator/=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a65f30a2dc199134e35bc7c5d431b2263',1,'operator/=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7172d84db640e6c49dff0d08dd64b53e',1,'operator/=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acf7cb9927bf09022088401923f2e1916',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a86b2a001cbec0d3a8d762a3c7ff47b0b',1,'operator/=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a744f72ba83522fe3cc2a49a007b42543',1,'operator/=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a77c678665b34df7652dcde053ca73185',1,'operator/=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae0614b6b199d8a65ae95d4621b118b82',1,'operator/=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa846fde89c7d2d18b18ef180a8a9c8a3',1,'operator/=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08e778be18e4a291c108fcc528b981d3',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6b9e49ad9ea256d2d0220c0d81552602',1,'operator/=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab933bc3cdf9adfea10ab9dba5292c812',1,'operator/=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a25e7c5d2ecf3375756d59074f333858f',1,'operator/=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ae4a80fde67eea9a0a37b2803946544',1,'operator/=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a912393b7208fa45bd1e87f30b218b68b',1,'operator/=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a18963246f2b640874bef6dca7049f64d',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0e2c2c2cb50b3a55ff213f18978aca35',1,'operator/=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a64f1136b17006f168ef837e17240814f',1,'operator/=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae46d75b8046d557452d74513f1106710',1,'operator/=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08d2460e259b9106d90d889481ad60d5',1,'operator/=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f7fd418408806ef498745c6fdb2c062',1,'operator/=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac703495cb370b52526a5a2d36ae26038',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ca11d43174baf0a729f93b35eabcbea',1,'operator/=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9f835a0a80c411580c97b65fdc5bdfd3',1,'operator/=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a17f47ec9cff60f8e1b3477a2793b7ac0',1,'operator/=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5be23e296bbed3a885586a6424b1666e',1,'operator/=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afba39221eb54e272aae79910b3cd7ef5',1,'operator/=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac057d95a2bf087575584aa6f9a2c6bf5',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab986ae2cec780a1f494b7b4468b7ba11',1,'operator/=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a44522c2304c6396bbe6b9d32000f4b6f',1,'operator/=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef8e7e499ea9d432aa743d83c076f945',1,'operator/=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3a0a3edbf1ba2314551454059c3f422b',1,'operator/=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acb9f0aef9fbdfde8a4f46e33b0d6c52f',1,'operator/=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a303dfcc81ffd355f866f863d7d9f0fa5',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a359edd4bcb8776861ceb26a3005624c0',1,'operator/=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc9f32cc6f40768df4285fba2e4783c7',1,'operator/=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae71f66d814a03f6377c9d86cf0a2b5d7',1,'operator/=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad0125b6baba3065a87a174ec27aa9a61',1,'operator/=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5cc74ad3e522d7104e6e2117751151ad',1,'operator/=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab3b594321fb42b0c2da99954d1e0976c',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4a0023e2fd08875156cd6ef747fbb5cd',1,'operator/=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4358ee606e66ba2081fcf94f9c3b5915',1,'operator/=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1e7ef6f065695d4b1d017547b60ef62',1,'operator/=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a284dfc702f0f67b9c233b87162eeabdd',1,'operator/=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab8f211ea896fc5190004f3ad6ad8932f',1,'operator/=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7e1bcf3bc06cbcbc304c0cdf729802bc',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abbe42648a46092137b303ccd08f7df86',1,'operator/=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af1a12a1efb618a57da6dd41ae18cb53c',1,'operator/=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a94686039356dfa9aa45608a8b0562fdc',1,'operator/=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa251d6483d3b099d1b5311fbe6f0bce2',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a045ff27257cb6d8ab7a94771ba5a17e6',1,'mlx::core::operator/=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a58112951a56a0f9f8c90b60fe74f9508',1,'mlx::core::operator/=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae736defc89a04fbaf7627ad2695bb838',1,'mlx::core::operator/=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#ab1f260710251256ef737dd59be9e143c',1,'mlx::core::operator/=(float &lhs, _MLX_Float16 rhs)']]], + ['operator_3c_36',['operator<',['../namespacemlx_1_1core_1_1simd.html#a6cd6e41660608d17ca8d38658d5e385c',1,'mlx::core::simd::operator<(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#ad9bebf95b37fa0c6517be82af5ccd4eb',1,'mlx::core::simd::operator<(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ac962a14c88c87082fc70a9c0370f35b0',1,'mlx::core::simd::operator<(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a23b59272b0760326844fffe20db9b3e2',1,'mlx::core::simd::operator<(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a01259c9188e6ecd48979cdc2fd766372',1,'mlx::core::simd::operator<(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#acf35d81032bb9043804fd1de43540f60',1,'mlx::core::simd::operator<(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#a3f63139b42029ba8d7b3b8ef10f5ac96',1,'mlx::core::simd::operator<(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#aaf29bfdcfdbb9a0acb9f4a6ed622868f',1,'mlx::core::simd::operator<(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a9e0c9b3e986809be5e87aacc4612bb8e',1,'mlx::core::simd::operator<(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#a67674e32596a9dae2258bb8e0e6a2058',1,'operator<(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9ef6a57b7185e9ca49e255fec1a44e25',1,'operator<(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aab02c65bc38ea66335b2192ead4095a8',1,'operator<(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae91686513e284bcc9635833744bbdda1',1,'operator<(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2486f3b5de85b0d57f458d8f21f82b42',1,'operator<(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a435a2aec4c777b4b184ff5d24992e8a1',1,'operator<(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abdd04257e6a73883b5f56f1186d0e906',1,'operator<(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a69984aaa05ae1d4fccccf7f57e8ecb4a',1,'operator<(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a501cc01d5bf15d9f03aa28545f9624ea',1,'operator<(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1b029e4ca72125a5f9471f582c819705',1,'operator<(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0736a76f56578d26ba1422dc8b744a18',1,'operator<(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24b1fa8998c892f90f8dde7c34fb10a5',1,'operator<(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af80ff2020ec2c4b406c5fdae3fe55e63',1,'operator<(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac03f6eefb836373d37dc280b0d813d78',1,'operator<(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#adb5f24b57d98214fc215a06475f21412',1,'mlx::steel::operator<()'],['../group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3',1,'mlx::core::operator<(const array &a, const array &b)'],['../group__ops.html#ga1ef8ea11cf15ce628c54201fa42748ef',1,'mlx::core::operator<(T a, const array &b)'],['../group__ops.html#ga95e72226dc7a79c40b3d16f990922050',1,'mlx::core::operator<(const array &a, T b)'],['../namespacemlx_1_1core.html#a987d631e1508e8df55d98ddd57e4d086',1,'mlx::core::operator<(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad3fb46370cd8f0992866fad9e2c64a3c',1,'mlx::core::operator<(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a3026691bf7ee5095243a8611bf3411aa',1,'mlx::core::operator<(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0d42d6c1d5f77a96e2f296b8ebd79ee6',1,'mlx::core::operator<(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ab5ce08a7de0a0ca00d61f7a7f8ea3ab4',1,'mlx::core::operator<(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abce8b7f24b61e5ec0f9a3afe20845caf',1,'mlx::core::operator<(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#aff97612627ae1ed260c43c0a7af0d306',1,'mlx::core::operator<(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a9119e518234df7923cae2b3802d59bf2',1,'mlx::core::operator<(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#aefb9b05ce8864ada99a920ab32017b89',1,'mlx::core::operator<(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abc55f3676c2d112a6e9ab276bd6b1796',1,'mlx::core::operator<(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#afe6581a2c45f24d7fab1e4006c1e3c70',1,'mlx::core::operator<(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aca1d50cdd9506481dcc4cd1ad4a4f734',1,'mlx::core::operator<(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a310720f513b6a2490e9df80c65f1bfb3',1,'mlx::core::operator<(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a29e457a170b6cefb6ba1e394c96c6f7b',1,'mlx::core::operator<(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#afd4519985b6b207ec41ad8530d1036df',1,'mlx::core::operator<(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae1e41ca94022e43a00cdfc5845102daa',1,'mlx::core::operator<(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ac80f4022bffd95b57526685ce8e1cbc1',1,'mlx::core::operator<(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a3a8f6f0af477788c4f0aa98abfc5f1ab',1,'mlx::core::operator<(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a3728ed9b6cbd152bf675251a0501b466',1,'mlx::core::operator<(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a5b9ad811a5e1358100c5423dd70ea387',1,'mlx::core::operator<(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a5c77e1db83995d3e06a8a26265bce5d6',1,'mlx::core::operator<(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab8a0a3f70664049b35ce1887bd8ff5c2',1,'mlx::core::operator<(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6652d93bfb2d426e261a1712a181a4d2',1,'mlx::core::operator<(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a03758b8d13da2de07cc4f4fc45d2854b',1,'mlx::core::operator<(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a325161b81a9ff179fd37d949780a17ba',1,'mlx::core::operator<(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a92eca79fce8233e4299343eee3996511',1,'mlx::core::operator<(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#adb016662b8f7eb680abfe1a421eabe72',1,'mlx::core::operator<(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3c_3c_37',['operator<<',['../namespacemlx_1_1core_1_1simd.html#ae21cbfd232edd7fe0f6f6c9fa11a354e',1,'mlx::core::simd::operator<<(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a56fccba38270fe3ae9fa7b2ecdeb5e87',1,'mlx::core::simd::operator<<(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a4ecd782ffa497ac7dc2482a232b0dd00',1,'mlx::core::simd::operator<<(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a33232e2342d5a3e542c9428924a25830',1,'mlx::core::simd::operator<<(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value<< b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a50044315dc365f026830416f6b615c77',1,'mlx::core::simd::operator<<(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a<< b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a68e7b952915e629d246d1ffac98b54ce',1,'mlx::core::simd::operator<<(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value<< b), 1 >'],['../group__ops.html#gad656c30f9fd7d9467e405657b325aa7e',1,'mlx::core::operator<<(const array &a, const array &b)'],['../namespacemlx_1_1core.html#a1e5c30e316afa30c14bc48b92afdb794',1,'mlx::core::operator<<(std::ostream &os, const Device &d)'],['../namespacemlx_1_1core.html#a4ddd07021b36c848d6fb1dd9ac276822',1,'mlx::core::operator<<(std::ostream &os, const Stream &s)'],['../namespacemlx_1_1core.html#a0023c267cf81345fad65e7a797954cd3',1,'mlx::core::operator<<(std::ostream &os, const Dtype &d)'],['../namespacemlx_1_1core.html#a1fd58658474fb842d648dcf8f7d9f078',1,'mlx::core::operator<<(std::ostream &os, const Dtype::Kind &k)'],['../namespacemlx_1_1core.html#a123331f01188bd76e37623b63b6b4340',1,'mlx::core::operator<<(std::ostream &os, array a)'],['../namespacemlx_1_1core.html#a4e733bba89760abed32393e085812b22',1,'mlx::core::operator<<(std::ostream &os, const std::vector< int > &v)'],['../namespacemlx_1_1core.html#a5e5bd5c57b1cf19776bdb41e732861d9',1,'mlx::core::operator<<(std::ostream &os, const std::vector< int64_t > &v)'],['../namespacemlx_1_1core.html#a42a19c8442b173606e714364227e7d45',1,'mlx::core::operator<<(std::ostream &os, const complex64_t &v)'],['../namespacemlx_1_1core.html#a57eb97a5eba99a846ac429795e407574',1,'mlx::core::operator<<(std::ostream &os, const float16_t &v)'],['../namespacemlx_1_1core.html#a7db909d54cf07375e89424c32c07a29c',1,'mlx::core::operator<<(std::ostream &os, const bfloat16_t &v)']]], + ['operator_3c_3d_38',['operator<=',['../namespacemlx_1_1core_1_1simd.html#a4d5e4c31af23d2871e09b88c1f6e418c',1,'mlx::core::simd::operator<=(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#ae0fcb84973e4762a543ad3843db4f153',1,'mlx::core::simd::operator<=(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aadd49786edc08f867e592d234327a031',1,'mlx::core::simd::operator<=(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aec6783f79ca181d6782a810ffb267482',1,'mlx::core::simd::operator<=(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a05240b8fd6f54632b676d4b66449f799',1,'mlx::core::simd::operator<=(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a914e821c358e05dfe8d0208888646793',1,'mlx::core::simd::operator<=(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#ad1570f6937d194a09e61d0e3a70ef578',1,'mlx::core::simd::operator<=(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a46c6ea18a9edd2a9cdba2ab62ca4782c',1,'mlx::core::simd::operator<=(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#accd17f741cab18590fdbe388d4783967',1,'mlx::core::simd::operator<=(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#aee04c9a63c6716a99a027418354debb0',1,'operator<=(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af469c58cffeab488c681f4b33f02cd05',1,'operator<=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a81eae168dfafd299c2b94e3e8558cf',1,'operator<=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f486bf02c6ad5b9b6a96d3450f03e47',1,'operator<=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acba9efe192d22b7781b4622103c7a944',1,'operator<=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aff100489cc40ad276c2d5d67a9df67db',1,'operator<=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7eac96f64ca42991caf819c8e8c8d2bc',1,'operator<=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a88c11cd37600de5480570da3d2ae5732',1,'operator<=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08c7d12a0d16565fbf052dba2db8b22d',1,'operator<=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2b9de9624c0a507b4ead85f898ad9daf',1,'operator<=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a28f8d21c5eef047c701cf690ce9c2ef0',1,'operator<=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a14b56c687053ee2432398a25663c068f',1,'operator<=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f360806708b95a3be400af0b8871b57',1,'operator<=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a80d288f22cadfdf5e904410349e616a1',1,'operator<=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a6cc3bab5e7f6e7c719c82afa90ad2827',1,'mlx::steel::operator<=()'],['../group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6',1,'mlx::core::operator<=(const array &a, const array &b)'],['../group__ops.html#ga150a9be467c9f91482a6d6fc13504bc4',1,'mlx::core::operator<=(T a, const array &b)'],['../group__ops.html#ga624eeccef0cc4b130e1325abfea057cb',1,'mlx::core::operator<=(const array &a, T b)'],['../namespacemlx_1_1core.html#a0066a47cb21223ddebc77992ee874fb9',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2593dbace3ce50e7146d9514726a543f',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a88654bcf6c9728517a2933ca2e29a7c1',1,'mlx::core::operator<=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a5d4f449e9c1699b99fcf894dd15e8af3',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a6b678bea8fdcda1f11c6691b56a15211',1,'mlx::core::operator<=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae8aacc606ea16f018a90eae758830a35',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a25668dea4ffb51c7c00eeecb9530d1d8',1,'mlx::core::operator<=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a084558b6a5487549799c49c37c9e9652',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ade2e2a0daa79d5c52f278f85f03dde2e',1,'mlx::core::operator<=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a750a2d2b4976ad94b08994d081f83445',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ade5a175ff45347689ac4c798d04c8ffc',1,'mlx::core::operator<=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae25e0c01b46612f039313a4825ba6428',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a5c90f16d8f6edf4b75c96b945b9fa591',1,'mlx::core::operator<=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a8cd6583fa0fc9957f993e00b2ec01d91',1,'mlx::core::operator<=(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a012130a0458cbc30b88365e0e0eab232',1,'mlx::core::operator<=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae8c890bdcffadee8c5dab85c907f57eb',1,'mlx::core::operator<=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a43cb070553c1f2fffb32ef6670e30980',1,'mlx::core::operator<=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ac759b7798d668a99535e59e26d6ba192',1,'mlx::core::operator<=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a70e528a789b5660d98e783b045aaa379',1,'mlx::core::operator<=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a40bd8abb8a4d989ddabbb298518bd7f5',1,'mlx::core::operator<=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a4155d4b0c76f37ab5e0b54f9cd683f35',1,'mlx::core::operator<=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad8bb648d0603a206e0392990c911ca0b',1,'mlx::core::operator<=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ace72a5853f2afd6510dcb97d54fa650d',1,'mlx::core::operator<=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab38f7a0d3c0809071ff5d3af859018d6',1,'mlx::core::operator<=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a7904b886d7b535a6af0a885d00597323',1,'mlx::core::operator<=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a57952168bd0b54c2677204d4ab1cb6e5',1,'mlx::core::operator<=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a6235dc5f4db517618bb3449b08c96e8b',1,'mlx::core::operator<=(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3d_39',['operator=',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a027b84cddc8d476f736ac1f1a9991fe4',1,'mlx::core::allocator::Allocator::operator=(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a2e971b47339b1d0849a334a902a9df3c',1,'mlx::core::allocator::Allocator::operator=(Allocator &&other)=delete'],['../classmlx_1_1core_1_1array.html#a8acf2b4c75f9b7f79da6675dbc36cf36',1,'mlx::core::array::operator=(const array &other) &&=delete'],['../classmlx_1_1core_1_1array.html#a5c89c2406a610b32943955f9a5060fbd',1,'mlx::core::array::operator=(array &&other) &&=delete'],['../classmlx_1_1core_1_1array.html#ad3277ff68f1336aa217f9cbe40181479',1,'mlx::core::array::operator=(array &&other) &=default'],['../classmlx_1_1core_1_1array.html#a5da41aabecf4c8055b7515341bf57147',1,'mlx::core::array::operator=(const array &other) &'],['../structmlx_1_1core_1_1array_1_1_data.html#a68e9417954fe811b5e41e6317a526748',1,'mlx::core::array::Data::operator=()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a3f42a1362b4a513fa89e7b3dcc570a8e',1,'mlx::core::metal::CommandEncoder::operator=()'],['../classmlx_1_1core_1_1metal_1_1_device.html#ad1d6382fd18a46b1906e1b43e0bd2e73',1,'mlx::core::metal::Device::operator=()'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aef97dbbc755940789f99a26164591c45',1,'mlx::core::metal::ResidencySet::operator=()'],['../structmlx_1_1core_1_1_function_exporter.html#a7ec0f53eb2783d5b1953be612e36d5c7',1,'mlx::core::FunctionExporter::operator=()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a957211656a13b4c0d126989a9aba3e25',1,'mlx::core::io::FileWriter::operator=()'],['../classmlx_1_1core_1_1_primitive.html#a6b1be7ea92f3a7bb19875c70259dad6b',1,'mlx::core::Primitive::operator=(const Primitive &other)=delete'],['../classmlx_1_1core_1_1_primitive.html#a50bbddd43e1ba0cf5f127cd7aa756a9e',1,'mlx::core::Primitive::operator=(Primitive &&other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a0a859309a4f192f2679e07f2e4ff4d22',1,'mlx::core::UnaryPrimitive::operator=(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#ab90b2ea80f1d914be03cf44def5db5a5',1,'mlx::core::UnaryPrimitive::operator=(UnaryPrimitive &&other)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ab170dbd2ce34c51e2eeebf5d08e7e2db',1,'mlx::core::scheduler::Scheduler::operator=(const Scheduler &)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a035ea35f4dd8ee985973080f14029379',1,'mlx::core::scheduler::Scheduler::operator=(Scheduler &&)=delete'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#a0f65b0523b8ddd989f338da6cb2860e3',1,'mlx::core::_MLX_BFloat16::operator=(std::vector< bool >::reference x)'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#abb8cd44ee22b17c55333ff2eb4e13a14',1,'mlx::core::_MLX_BFloat16::operator=(const float &x)'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a608a099bf7116ee608dcfd31ea3ade2c',1,'mlx::core::_MLX_Float16::operator=(std::vector< bool >::reference x)'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a35543c3653d477c46350697fb808373d',1,'mlx::core::_MLX_Float16::operator=(const float &x)'],['../structmlx_1_1core_1_1_command_encoder.html#a3f42a1362b4a513fa89e7b3dcc570a8e',1,'mlx::core::CommandEncoder::operator=()']]], + ['operator_3d_3d_40',['operator==',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a1afd6d2a19a2b0d712063f221ab4eba7',1,'mlx::core::array::ArrayIterator::operator==()'],['../namespacemlx_1_1core_1_1simd.html#a273fcc5387c1c9878e658ba6bc32f00c',1,'mlx::core::simd::operator==(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#a46ede415296683771bb22246a813482a',1,'mlx::core::simd::operator==(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a63768090c16e5dcffccadf550d169abc',1,'mlx::core::simd::operator==(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a7928482ed5d25932be80413c7239125c',1,'mlx::core::simd::operator==(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a51de2acf3dcd55c7c52e3ce7ed6ed9d7',1,'mlx::core::simd::operator==(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a4877ae5406d081680b785a86ad656e03',1,'mlx::core::simd::operator==(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#acafae9e62680565cd1f1c50c64d7ce4f',1,'mlx::core::simd::operator==(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#aa837052ddcb02f4d9bc39b07399b4d91',1,'mlx::core::simd::operator==(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aaacbf6671080409e822fbb218e3fdf00',1,'mlx::core::simd::operator==(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#abfc19f03616441245dfc7726b278f190',1,'operator==(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a49a13b06a325ed3cca4004b6a0cde065',1,'operator==(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0aa3bfcfab53700488e5f386e6de60d5',1,'operator==(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3936148781ab1c4f33f58d12c116f370',1,'operator==(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae753526b669fba27771089dc809abd66',1,'operator==(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a05a4f197a71d0f16879032f44492bb79',1,'operator==(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae86f5917847b1ec9f313996250f2e0be',1,'operator==(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aab74ec4d33a64b92b908717d500f1ecf',1,'operator==(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac30a2c1fa6f172af903fdeb6a8632606',1,'operator==(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab4e9ad547aa23daa351075e0ecc58fa2',1,'operator==(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa5fa1a8f2b39c3508fe38205469756d1',1,'operator==(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeadc1f36c6bdc219294ce9341d80afa5',1,'operator==(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3ae2091ada1e39e857fbc53c97bdb79f',1,'operator==(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac7b4d295f3c7b1e09964f24f306422da',1,'operator==(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#abcc797f27e87e857b41c1a8d33ee2c78',1,'mlx::steel::operator==()'],['../namespacemlx_1_1core.html#a937503d72b66c661bf3f5fdcd98ef97c',1,'mlx::core::operator==(const Device &lhs, const Device &rhs)'],['../group__ops.html#gaa30cf69f3d22f65615f5e1696dd5703f',1,'mlx::core::operator==(const array &a, const array &b)'],['../group__ops.html#gaf115782d009ac2a547fcca395c9ec797',1,'mlx::core::operator==(T a, const array &b)'],['../group__ops.html#ga3ad3ed7aece2650943a35082dbe3a0a5',1,'mlx::core::operator==(const array &a, T b)'],['../namespacemlx_1_1core.html#ac470f937a379d6356c8f567c97cd7481',1,'mlx::core::operator==(const Stream &lhs, const Stream &rhs)'],['../namespacemlx_1_1core.html#aec63a0472cb943fe39f31e7678555572',1,'mlx::core::operator==(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad05311ca8e2f19ffe5849e963837cec7',1,'mlx::core::operator==(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aaaf591cb2188381e6cbd857132d04eb7',1,'mlx::core::operator==(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7ef33c33509ccccf1ab217500e8b3c1a',1,'mlx::core::operator==(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#abec4200a718b7c5ed80b7abcc4447260',1,'mlx::core::operator==(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad853981b1c5ba69b07d54c7b77055d22',1,'mlx::core::operator==(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a752d6cb4172a9cb91e5da19582329c6d',1,'mlx::core::operator==(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0175beb3de139faa08479a88215b35ea',1,'mlx::core::operator==(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a61da2851cb3beeef28049228346c28b5',1,'mlx::core::operator==(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aa24713cb9e39bacb516c992eb03d2b2b',1,'mlx::core::operator==(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a6d565dd93c46259f9486d9fdf0969589',1,'mlx::core::operator==(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a24e79a82557861de64dad66d36e6ff30',1,'mlx::core::operator==(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#af27d515ac390d62bd852b73ea759a947',1,'mlx::core::operator==(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae3e1e8b7a5410e0edf35f31f74295e2f',1,'mlx::core::operator==(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aaa22230a66b15c3e774d8ce45783a746',1,'mlx::core::operator==(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ae2a0bcdc171d7e9745d33e1d9aac4f8a',1,'mlx::core::operator==(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a331ec62442a8d3eb8ccba7b4de5168d1',1,'mlx::core::operator==(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#acfcaefe0990eb3533e2b11a6f2657492',1,'mlx::core::operator==(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a8d48dbd49cccff07777affb2a412058c',1,'mlx::core::operator==(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a88eae27edd22fa4418776672023cb276',1,'mlx::core::operator==(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a188b363f633ea360407b3f9cf4e1f1a6',1,'mlx::core::operator==(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ae065fe5c42c1a333d7858d19f6434fa9',1,'mlx::core::operator==(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a2f98db199deb6d7a82551fa4afec655a',1,'mlx::core::operator==(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a85f83add412cb320b5cd1c3da6aadbd5',1,'mlx::core::operator==(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7e2cee66c3ca1b56f4f3d7fd1d6e0be1',1,'mlx::core::operator==(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#ad436557da5c7fea71fc58182a876cfe5',1,'mlx::core::operator==(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3e_41',['operator>',['../namespacemlx_1_1core_1_1simd.html#abd37e62eff936a64677b5aba787b4d18',1,'mlx::core::simd::operator>(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#a71a6902e729e3facdc609e93cd12d485',1,'mlx::core::simd::operator>(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ab7b291b3559792e18208e17432d25342',1,'mlx::core::simd::operator>(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ad8b67f9ced9c7f3cb472b9c3df817f08',1,'mlx::core::simd::operator>(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a4113a94fb8dcd0d88f14ec9d82089508',1,'mlx::core::simd::operator>(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#ac971bfa5c7ec8abc432eab5f3c5646aa',1,'mlx::core::simd::operator>(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#a35d875fa7bce02a6171f37240a346e1d',1,'mlx::core::simd::operator>(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#acf2391cc4d945887d7820501ba14ba89',1,'mlx::core::simd::operator>(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aa17e031474fa87f6ea7855257dcc9ece',1,'mlx::core::simd::operator>(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#a032a8d3eec2384c9f03066f7fd945995',1,'operator>(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae394c0a10e47d1d047854a888402eb57',1,'operator>(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab9cd098786d2f4c855c42e4a6f30ab3e',1,'operator>(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a55600f3b9859e2891e0e0b5690867b72',1,'operator>(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afd7cdb8ed2a9820efe9cf322c06f188c',1,'operator>(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31bbdbe0b62b90a4d6ea4bb0a7db586b',1,'operator>(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a68125e66f74eaffe5ea9267638ce870d',1,'operator>(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac89eb6b29edad8cca63727ab97171c29',1,'operator>(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a74e477567c9477c2cf0684f81ef4498f',1,'operator>(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2d37130b6fd79b425f5ba92b65e36bed',1,'operator>(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a41d55d167e9dc63bf29d15e0ff004869',1,'operator>(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa95f9ebfdab3c5f524775651362ce914',1,'operator>(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2826bd301bb5393473ccd363f2052c0d',1,'operator>(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a62a512d0edd894759c69f724b970fbdb',1,'operator>(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a7512eadda6160e4c9d9e6aa4049fac20',1,'mlx::steel::operator>()'],['../group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb',1,'mlx::core::operator>(const array &a, const array &b)'],['../group__ops.html#ga32e106e794e2c32e4e7decee2df2477f',1,'mlx::core::operator>(T a, const array &b)'],['../group__ops.html#ga96552b90e89923c5d2064cc427775ec5',1,'mlx::core::operator>(const array &a, T b)'],['../namespacemlx_1_1core.html#aedc4e9df4bf71c0ac34fcfae60cdf550',1,'mlx::core::operator>(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a14c188303d09b97867bcfd34519aa4a6',1,'mlx::core::operator>(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ac97736fadafa7efa201624d0e1128ee8',1,'mlx::core::operator>(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3c41a304126bc225bdc68062d1eb6e7e',1,'mlx::core::operator>(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ab594f3ae1ee13227fae940fef0d00cb9',1,'mlx::core::operator>(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a01dabc077a872c115a9a9ccd95f1acec',1,'mlx::core::operator>(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#adabbd8768d216873617768249473a5c7',1,'mlx::core::operator>(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#adae1b14669d27ce1fe0c214771c07b77',1,'mlx::core::operator>(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ab03a22961d99fa12d3e74b3116e94e8f',1,'mlx::core::operator>(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a42011a27a3d23a60be5be44ee7cac87c',1,'mlx::core::operator>(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a50f6a94bb36d89cf28817aff88ab89c8',1,'mlx::core::operator>(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac173de50ee57b1b066d49363ba978c53',1,'mlx::core::operator>(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#ab09f1b4879aa3190c2f66c9bd1224021',1,'mlx::core::operator>(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a91eb6ca854217424129a55ae95a123b5',1,'mlx::core::operator>(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a58d5795d8312599d101ae16f194e4a2a',1,'mlx::core::operator>(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aafa3bbeda78610c4285f3e57042268f3',1,'mlx::core::operator>(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a8a928d76a6fbf3d336296401e14617a4',1,'mlx::core::operator>(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ade2f9222fd433cd4d673c6182f256235',1,'mlx::core::operator>(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ae24c337810c841ff23e327efde7045e1',1,'mlx::core::operator>(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acf401ede354fcc998b13ea6442994d7e',1,'mlx::core::operator>(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a2bb28a9a0894a73ae1b27e7f4da0841a',1,'mlx::core::operator>(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a09d631e8a85fd7ae72e1a868b8f9b9cb',1,'mlx::core::operator>(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a49421ea65b5a98df080d75b1636b2157',1,'mlx::core::operator>(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a692ce931b660415e17f92d18a8e0d446',1,'mlx::core::operator>(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a579bb87b3ede5663d7cd68c7c0f6fb9e',1,'mlx::core::operator>(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af810587a17e692f4eec256d3c3cd27de',1,'mlx::core::operator>(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a50f4177d3ca03a95fc2614e100c7391d',1,'mlx::core::operator>(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3e_3d_42',['operator>=',['../namespacemlx_1_1core_1_1simd.html#a87e11ab36aae3328fe3d5230bdf31692',1,'mlx::core::simd::operator>=(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#a4e65febbfa8b4df2970c1d78801b3c66',1,'mlx::core::simd::operator>=(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a673b4d8d228f35f06cf5b882335f04d5',1,'mlx::core::simd::operator>=(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a530ac8728e4d7e7be2482d5b2467906c',1,'mlx::core::simd::operator>=(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#ac7f3848b48c8e23c71c85fcc9909b933',1,'mlx::core::simd::operator>=(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a034d7b57cb3c6ca711c573515327d1a8',1,'mlx::core::simd::operator>=(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#a8d7dcf1914ce8fe8518d84b0f2a5fe91',1,'mlx::core::simd::operator>=(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#aecdc08fcc70b158749a93a7a0f688aa3',1,'mlx::core::simd::operator>=(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ab9097573af69cc66d1427d0f52507e7a',1,'mlx::core::simd::operator>=(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#aafbd686c180398c98b33d7643f893a46',1,'operator>=(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a430dd11fbf4c6f39bc1506ab43b2341f',1,'operator>=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a64f6787a96386246f83a8981d274150e',1,'operator>=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1a788f82212afad30e4c2ee40f1c313c',1,'operator>=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae88617c4a012c5dc12781a349a28c886',1,'operator>=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a467a88531150a4d9d30fce07c49c126e',1,'operator>=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9e21c5ea9dd724dc2ca8c54ad908f09c',1,'operator>=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2f6286d222e2176bcbdc824c5d598100',1,'operator>=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abec53064aa96265385ecc57de5fbc74c',1,'operator>=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac766839f8f9e4863e8e18418c342c875',1,'operator>=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2807fa6862b0f9689c81199b1e695ed8',1,'operator>=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee3ae0d0d1f941463b06eca0bf041b2b',1,'operator>=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a523eda93c809733368e2b45382d2add6',1,'operator>=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1f4e90909ac1c7280f4c7d1977c55fb7',1,'operator>=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa3c95c60cf69603705bb4636de547bcb',1,'mlx::steel::operator>=()'],['../group__ops.html#ga3a41895f25ed083a36994d95fa102546',1,'mlx::core::operator>=(const array &a, const array &b)'],['../group__ops.html#gaf509f2cb3b18963232f20d6c3bd229b2',1,'mlx::core::operator>=(T a, const array &b)'],['../group__ops.html#gafa0eb25d5978674bfc9e59d4145ec590',1,'mlx::core::operator>=(const array &a, T b)'],['../namespacemlx_1_1core.html#a8494764f5c686743ede66dc76d85d955',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a019df48807b506d9995856684bf7797a',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a96ab6405430efb887cdb5c828cb67d6e',1,'mlx::core::operator>=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac18be72269b1bcfb0249cc00a0600681',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aeb879815228efbd2c8f80986e1c8d41f',1,'mlx::core::operator>=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0051156f6a568f58cd54850f746fb507',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ae93556906e115625ed1b62d36cf21b70',1,'mlx::core::operator>=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab81ad16e3be591dfc9e42ac3c19b055f',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6cfe9b03e7c5f1eb9374208a552c3cc9',1,'mlx::core::operator>=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2f5add83812fb137dd9226c6c01e45d5',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ad1014a836e7ce9301de8588eef1e89ee',1,'mlx::core::operator>=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a17791561434dc995de9f268d145c0ed1',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a3755925b24a903045937464be117de2f',1,'mlx::core::operator>=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a6262aeb513d27fc8313293b261e72abb',1,'mlx::core::operator>=(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a6feb4b3ea511b0eda4d1ec9725f3fb4c',1,'mlx::core::operator>=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a03b3f7fcb755ec075985ab26336926f0',1,'mlx::core::operator>=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aecfbf5ef4872ae447eb4a374e4db28e4',1,'mlx::core::operator>=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae4690f349b2483f5d1a4b75aba67399f',1,'mlx::core::operator>=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a667e95146dd5199e67bcb121b984b1f0',1,'mlx::core::operator>=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a3375f1562f148bdc07451f2b6e54e6df',1,'mlx::core::operator>=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ae83df12368cb07ccb1c10c1117ff3922',1,'mlx::core::operator>=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad41251938cf852b5560c1180944ebb49',1,'mlx::core::operator>=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a4ddb5ef0b88929086f9b09729fda0dde',1,'mlx::core::operator>=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a0908a61ab261aff726922b33fa6ed159',1,'mlx::core::operator>=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a0fdadf87edd8a0a57c63953fb0ebe053',1,'mlx::core::operator>=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a47c82778e43032c0bbf5d59407e81dc9',1,'mlx::core::operator>=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a14e6c43b924eacca1b2dac1d5d00ca2b',1,'mlx::core::operator>=(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3e_3e_43',['operator>>',['../namespacemlx_1_1core_1_1simd.html#a6e45c9c2f0591d9d5dd37a07ebcc3c2a',1,'mlx::core::simd::operator>>(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#aa35a2aab733e4bfc80a9f4e3f508daee',1,'mlx::core::simd::operator>>(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#aebf93b8179621e83bb3f3c4a8816eca8',1,'mlx::core::simd::operator>>(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a1108d186d57c2010c743d3f9297befc7',1,'mlx::core::simd::operator>>(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value > > b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a9ac36abfb7dffc7ad24b4d0c295452e5',1,'mlx::core::simd::operator>>(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a > > b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a4bf8c887eb6943563ceb1e603d1325b1',1,'mlx::core::simd::operator>>(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value > > b), 1 >'],['../group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a',1,'mlx::core::operator>>()']]], + ['operator_5b_5d_44',['operator[]',['../classpocketfft_1_1detail_1_1arr.html#aea0bd899b19e03f54dfd6c188727061a',1,'pocketfft::detail::arr::operator[](size_t idx)'],['../classpocketfft_1_1detail_1_1arr.html#a99c54f96bc79c7cdd8925c1663462842',1,'pocketfft::detail::arr::operator[](size_t idx) const'],['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a71b02f67c47b24adb296eafd2c7a3598',1,'pocketfft::detail::sincos_2pibyn::operator[]()'],['../classpocketfft_1_1detail_1_1cndarr.html#ae4852d1fe936a5d61832b507816c7054',1,'pocketfft::detail::cndarr::operator[]()'],['../classpocketfft_1_1detail_1_1ndarr.html#a2b2c4e205e8b5c32c9fe55dfd7b8c8d8',1,'pocketfft::detail::ndarr::operator[]()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a235268dc56eb1bb5b86cd3aade67b77c',1,'mlx::core::simd::Simd::operator[](int idx) const'],['../structmlx_1_1core_1_1simd_1_1_simd.html#ae877ce4884241399de8e28090441f557',1,'mlx::core::simd::Simd::operator[](int idx)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a4b24316469cd9ecc88f8c073ab1a862e',1,'mlx::core::simd::Simd< float16_t, N >::operator[](int idx) const'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a97043111a44318b5eb68977ecacbb638',1,'mlx::core::simd::Simd< float16_t, N >::operator[](int idx)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a235268dc56eb1bb5b86cd3aade67b77c',1,'mlx::core::simd::Simd< T, 1 >::operator[](int idx) const'],['../structmlx_1_1core_1_1simd_1_1_simd.html#ae877ce4884241399de8e28090441f557',1,'mlx::core::simd::Simd< T, 1 >::operator[](int idx)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a235268dc56eb1bb5b86cd3aade67b77c',1,'mlx::core::simd::Simd< float16_t, N >::operator[](int idx) const'],['../structmlx_1_1core_1_1simd_1_1_simd.html#ae877ce4884241399de8e28090441f557',1,'mlx::core::simd::Simd< float16_t, N >::operator[](int idx)']]], + ['operator_5e_45',['operator^',['../namespacemlx_1_1core_1_1simd.html#a25b3de1947dbab7c4864b41ec226453b',1,'mlx::core::simd::operator^(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a93e69a8170b8fe14f0a3188b4e8ccd49',1,'mlx::core::simd::operator^(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a16c4a2c8fc59a2e2fcc05db243289706',1,'mlx::core::simd::operator^(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a369178519e0e91fa936c0fd4aa9ee109',1,'mlx::core::simd::operator^(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value ^ b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a5b877b5eb7044d9b2a42a9af4af21f01',1,'mlx::core::simd::operator^(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a ^ b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a16fa3c809e46b5cae3e8abfaf98199a4',1,'mlx::core::simd::operator^(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value ^ b), 1 >'],['../group__ops.html#gac3a6fe18694e84b3d63458e9553ac181',1,'mlx::core::operator^(const array &a, const array &b)'],['../namespacemlx_1_1core.html#ae36ea40b8477bfa12d41aae8245225c9',1,'mlx::core::operator^(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a03fc96696f5c6d9411841889d05f4670',1,'mlx::core::operator^(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a55130edf926366db0d6207989e609b7c',1,'mlx::core::operator^(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0b75198f364d742a1c25dd13e398f2c2',1,'mlx::core::operator^(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7f205f1b10b23180a23bf2be4bb726b1',1,'mlx::core::operator^(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a9edfe65f3c6da583c7b109290ec94b22',1,'mlx::core::operator^(uint16_t lhs, _MLX_Float16 rhs)']]], + ['operator_5e_3d_46',['operator^=',['../namespacemlx_1_1core.html#a97cb7d3eac404a442e84656cefe7cfb4',1,'mlx::core::operator^=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abcfd2d9615c96561fd44dfb9c341cf8e',1,'mlx::core::operator^=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ae78083d766b9cf6f87cded341bbcd63e',1,'mlx::core::operator^=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acf36c10779fbf1efbe1e6a7fd41176cd',1,'mlx::core::operator^=(_MLX_Float16 &lhs, uint16_t rhs)']]], + ['operator_7c_47',['operator|',['../namespacemlx_1_1core_1_1simd.html#ab2b540d7329491000e7722f9b3ef797d',1,'mlx::core::simd::operator|(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a0cd57bba23daed624df5e2b06b676dca',1,'mlx::core::simd::operator|(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#acd57dc91aa205d9d3f8804df4261a7fb',1,'mlx::core::simd::operator|(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a60805b5f57ddbbf74f700b54cd3fc4f8',1,'mlx::core::simd::operator|(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value|b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a495d15a18ee4a6dda22e37e8dc02e45b',1,'mlx::core::simd::operator|(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a|b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a6449faa1666afe1186d55b61bb3e5b5a',1,'mlx::core::simd::operator|(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value|b), 1 >'],['../group__ops.html#ga52392a2a98f09a80da8d338c4908bd02',1,'mlx::core::operator|(const array &a, const array &b)'],['../namespacemlx_1_1core.html#af84ed854132c1514dca5a524fdb7ed05',1,'mlx::core::operator|(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7423aac70f9f2e3fb6a5c9a3fc96f703',1,'mlx::core::operator|(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a19805f505cb7ac72bfab66c339ea7900',1,'mlx::core::operator|(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2d933573edf4ed305fddd8a0caef1ee8',1,'mlx::core::operator|(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afab3d4eb1b36a276922879ce6e44b7f5',1,'mlx::core::operator|(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ab132729fa6912d22a8e402057eb4ba12',1,'mlx::core::operator|(uint16_t lhs, _MLX_Float16 rhs)']]], + ['operator_7c_3d_48',['operator|=',['../namespacemlx_1_1core.html#a8e1d21375ae4b89b3cbea3a46d262abd',1,'mlx::core::operator|=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a28d6c2f89e73b7b874dd1f67f853a96f',1,'mlx::core::operator|=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a2d8470b69cbbeefece08d3ffd46c0082',1,'mlx::core::operator|=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a359c6257097a304c00d41d64296ef4c9',1,'mlx::core::operator|=(_MLX_Float16 &lhs, uint16_t rhs)']]], + ['operator_7c_7c_49',['operator||',['../namespacemlx_1_1core_1_1simd.html#ab380b8f73672727a38ea0931e731fe4a',1,'mlx::core::simd::operator||(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#ac34f6b278627949d2ee68cdbf3d2f50f',1,'mlx::core::simd::operator||(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ab2bc61c02b9096163e9db91a3f88788f',1,'mlx::core::simd::operator||(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a7a1c3be1c37d41e450469f2e98cd9dde',1,'mlx::core::simd::operator||(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value||b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a0c8bd67982681ecd53cd8d739be3a5a9',1,'mlx::core::simd::operator||(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a||b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#aad2d440fbb9e5478b5ed24400a859942',1,'mlx::core::simd::operator||(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value||b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a38e83534a648d0743dc4c7deb9a7fd49',1,'mlx::core::simd::operator||(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#acdcdaea84869a0b05c08139c10f13a06',1,'mlx::core::simd::operator||(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#aa9ac1951153211b2ff95dd34a3427797',1,'mlx::core::simd::operator||(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1steel.html#a1bb3ac5061a04e407fc4cdcc9f6ea03f',1,'mlx::steel::operator||()'],['../group__ops.html#ga27af56a98270d4d76d139f0f9171b83a',1,'mlx::core::operator||()']]], + ['ops_2eh_50',['ops.h',['../backend_2metal_2kernels_2reduction_2ops_8h.html',1,'(Global Namespace)'],['../distributed_2ops_8h.html',1,'(Global Namespace)'],['../ops_8h.html',1,'(Global Namespace)']]], + ['or_51',['Or',['../struct_or.html',1,'Or< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a7a959bb7b33f410a03b3c887173fd7ed',1,'mlx::core::distributed::AllReduce::Or'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23da51065a44e7f9a76a6dab6de637c6db22',1,'mlx::core::BitwiseBinary::Or'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a2e53e38f8b906ed4def9a5653aeb51fe',1,'mlx::core::Reduce::Or']]], + ['ortho_52',['ortho',['../structpocketfft_1_1detail_1_1_exec_dcst.html#aea17551a49acaca5e7808dc181d38b7f',1,'pocketfft::detail::ExecDcst']]], + ['os_53',['oS',['../struct_m_l_x_conv_params.html#a19ccb9fecfccdc18b6a7f0cc43adbc6e',1,'MLXConvParams']]], + ['out_54',['out',['../struct_read_writer.html#abea3b913c952c505d0ca4e529c7316ef',1,'ReadWriter']]], + ['out_5fof_5fbounds_55',['out_of_bounds',['../struct_read_writer.html#a08e10626fbc789b6dff9172fd6c36f7c',1,'ReadWriter::out_of_bounds() const'],['../struct_read_writer.html#a6f946aea5452109dca7fc70ed39c6efe',1,'ReadWriter::out_of_bounds() const'],['../struct_read_writer.html#a8f40d7f343d32134fe27a694abfde6bf',1,'ReadWriter::out_of_bounds() const']]], + ['out_5fstrides_56',['out_strides',['../struct_m_l_x_conv_params.html#adfca77f9a3c2b4c74752f90636ff5667',1,'MLXConvParams']]], + ['outer_57',['outer',['../group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d',1,'mlx::core']]], + ['output_5fshape_58',['output_shape',['../classmlx_1_1core_1_1_broadcast_axes.html#aaa495110c16fbbc642fbb224ef8dfae6',1,'mlx::core::BroadcastAxes::output_shape()'],['../classmlx_1_1core_1_1_broadcast.html#a00c39c113fe3e698771e2e6b595c32cd',1,'mlx::core::Broadcast::output_shape()'],['../classmlx_1_1core_1_1_expand_dims.html#a3814ad4697eccb75fdb9275017a3fd67',1,'mlx::core::ExpandDims::output_shape()'],['../classmlx_1_1core_1_1_flatten.html#a2f8e1defb9c33af2dec29ff8697132aa',1,'mlx::core::Flatten::output_shape()'],['../classmlx_1_1core_1_1_reshape.html#aa15020d7d844d714d42bc60b44aeefc1',1,'mlx::core::Reshape::output_shape()'],['../classmlx_1_1core_1_1_squeeze.html#aadf1d3b85839390a2ec560603aeed04a',1,'mlx::core::Squeeze::output_shape()'],['../classmlx_1_1core_1_1_unflatten.html#a4c760c8fe981fd2ac17a31ff9faff10a',1,'mlx::core::Unflatten::output_shape()']]], + ['output_5fshapes_59',['output_shapes',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a50934862ccdb16a3dcce6626c5727080',1,'mlx::core::fast::AffineQuantize::output_shapes()'],['../classmlx_1_1core_1_1_primitive.html#aa5b443d71db1c7ed31a5ae6e31b7fe29',1,'mlx::core::Primitive::output_shapes()'],['../classmlx_1_1core_1_1_abs.html#ac9d55481e5490423e4aaf02b95cafc75',1,'mlx::core::Abs::output_shapes()'],['../classmlx_1_1core_1_1_add.html#a50877893083fd78b31aa25152f750418',1,'mlx::core::Add::output_shapes()'],['../classmlx_1_1core_1_1_arange.html#a447083a1403d3d42a7ad9c307a666946',1,'mlx::core::Arange::output_shapes()'],['../classmlx_1_1core_1_1_arc_cos.html#a67a5025f8d7e5bac22888ad4bf813679',1,'mlx::core::ArcCos::output_shapes()'],['../classmlx_1_1core_1_1_arc_cosh.html#a3ab82e9f0452faea735338abccb5f0ac',1,'mlx::core::ArcCosh::output_shapes()'],['../classmlx_1_1core_1_1_arc_sin.html#a0217b9a4e18196ed65ba96b4ad096ecd',1,'mlx::core::ArcSin::output_shapes()'],['../classmlx_1_1core_1_1_arc_sinh.html#a2f668f230d93c7b90e62200a0b7cb6f6',1,'mlx::core::ArcSinh::output_shapes()'],['../classmlx_1_1core_1_1_arc_tan.html#a2ebabfd1c2963199df0d7610b7ddf422',1,'mlx::core::ArcTan::output_shapes()'],['../classmlx_1_1core_1_1_arc_tan2.html#acb8e5cf85c4bc58f909ce2e8b83c3619',1,'mlx::core::ArcTan2::output_shapes()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6806f04142d850f107a18a71900759c6',1,'mlx::core::ArcTanh::output_shapes()'],['../classmlx_1_1core_1_1_arg_partition.html#a587ce69b0639683ba646652f887d0239',1,'mlx::core::ArgPartition::output_shapes()'],['../classmlx_1_1core_1_1_arg_reduce.html#a81a70885480c1d436329025091b2fa4c',1,'mlx::core::ArgReduce::output_shapes()'],['../classmlx_1_1core_1_1_arg_sort.html#a219ce04a811397a900c3235d8e6aef5c',1,'mlx::core::ArgSort::output_shapes()'],['../classmlx_1_1core_1_1_as_type.html#a3975b31cfd86d6eb33dc73554b357b88',1,'mlx::core::AsType::output_shapes()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a49c9d2688d3cca8abf5698a250d57d56',1,'mlx::core::BitwiseBinary::output_shapes()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a42c4385e65851d58e4411a4afe73f58e',1,'mlx::core::BroadcastAxes::output_shapes()'],['../classmlx_1_1core_1_1_broadcast.html#adef65b1ec75efbe43e5574ec81b7c0ac',1,'mlx::core::Broadcast::output_shapes()'],['../classmlx_1_1core_1_1_ceil.html#a3bf7db5178ed26e23d9ba360ba34ab85',1,'mlx::core::Ceil::output_shapes()'],['../classmlx_1_1core_1_1_compiled.html#a15cb081590ee024ba11476494581a4d4',1,'mlx::core::Compiled::output_shapes()'],['../classmlx_1_1core_1_1_concatenate.html#af8415a2fe28804a1437d0876ba15615f',1,'mlx::core::Concatenate::output_shapes()'],['../classmlx_1_1core_1_1_conjugate.html#afd68332463d12e69c47388f6b81ae96c',1,'mlx::core::Conjugate::output_shapes()'],['../classmlx_1_1core_1_1_contiguous.html#afff58fbf61f0c26b3606208dd2fa2072',1,'mlx::core::Contiguous::output_shapes()'],['../classmlx_1_1core_1_1_copy.html#a6bbe5fd9ce3cb5a39853b316106d2674',1,'mlx::core::Copy::output_shapes()'],['../classmlx_1_1core_1_1_cos.html#a923312e71c5a003a38b37ab67ec82580',1,'mlx::core::Cos::output_shapes()'],['../classmlx_1_1core_1_1_cosh.html#adf58c7e24b5059e66007132bc16dfe49',1,'mlx::core::Cosh::output_shapes()'],['../classmlx_1_1core_1_1_divide.html#a9563d9ee243204cfdaac6aca34853cd7',1,'mlx::core::Divide::output_shapes()'],['../classmlx_1_1core_1_1_div_mod.html#a1b7f104346cb5423ac15371b45c7ef86',1,'mlx::core::DivMod::output_shapes()'],['../classmlx_1_1core_1_1_select.html#a10e837a391542b364186288a87e11513',1,'mlx::core::Select::output_shapes()'],['../classmlx_1_1core_1_1_remainder.html#ab4de49818d1fdea8cdfef502f519b255',1,'mlx::core::Remainder::output_shapes()'],['../classmlx_1_1core_1_1_equal.html#ae714c2b0641fc9c339a2f8483bb4e257',1,'mlx::core::Equal::output_shapes()'],['../classmlx_1_1core_1_1_erf.html#ace70b96c48419e29243982ed697f6411',1,'mlx::core::Erf::output_shapes()'],['../classmlx_1_1core_1_1_erf_inv.html#a067cac7a7244b4dae6629c7e4466589f',1,'mlx::core::ErfInv::output_shapes()'],['../classmlx_1_1core_1_1_exp.html#aef2b3c24dba3ca3a63a210d3bd8e39b6',1,'mlx::core::Exp::output_shapes()'],['../classmlx_1_1core_1_1_expm1.html#ae78f03a204687f16164ed702cfc0d5cc',1,'mlx::core::Expm1::output_shapes()'],['../classmlx_1_1core_1_1_expand_dims.html#af64bd4bc2cc5f5c58869f34cd974bb3c',1,'mlx::core::ExpandDims::output_shapes()'],['../classmlx_1_1core_1_1_flatten.html#a5069a73ba1e7b52b7b051f692db6d0d2',1,'mlx::core::Flatten::output_shapes()'],['../classmlx_1_1core_1_1_floor.html#a0a62dee6df6a82fcd955bf7670be2cd5',1,'mlx::core::Floor::output_shapes()'],['../classmlx_1_1core_1_1_gather.html#a53d89a6c4ebb634bc208bd85aa2fcda1',1,'mlx::core::Gather::output_shapes()'],['../classmlx_1_1core_1_1_gather_axis.html#abc483c7da7747263b2f1498f98b4d96d',1,'mlx::core::GatherAxis::output_shapes()'],['../classmlx_1_1core_1_1_greater.html#af798a7cd704a2a9a8b3ecb6ef49583b0',1,'mlx::core::Greater::output_shapes()'],['../classmlx_1_1core_1_1_greater_equal.html#a1a77c18d89ee227171ff38efef6cacf6',1,'mlx::core::GreaterEqual::output_shapes()'],['../classmlx_1_1core_1_1_hadamard.html#aa709166de3c493308689769579d665e8',1,'mlx::core::Hadamard::output_shapes()'],['../classmlx_1_1core_1_1_imag.html#ad4f847483ba07d20aba5b927c2689be8',1,'mlx::core::Imag::output_shapes()'],['../classmlx_1_1core_1_1_less.html#ad7604a75b79260d263ac0c7d959cadd5',1,'mlx::core::Less::output_shapes()'],['../classmlx_1_1core_1_1_less_equal.html#a5598c700e881673098928e47b4da9ff8',1,'mlx::core::LessEqual::output_shapes()'],['../classmlx_1_1core_1_1_log.html#ab2cae6889352ca0674f6463f8f52d77d',1,'mlx::core::Log::output_shapes()'],['../classmlx_1_1core_1_1_log1p.html#a73a02ddf0f125fff83462d97146a0a08',1,'mlx::core::Log1p::output_shapes()'],['../classmlx_1_1core_1_1_logical_not.html#ad3889969521c6a040aa2f26caee219b7',1,'mlx::core::LogicalNot::output_shapes()'],['../classmlx_1_1core_1_1_logical_and.html#a266f1eaced19b8b11e273de9219cf9ed',1,'mlx::core::LogicalAnd::output_shapes()'],['../classmlx_1_1core_1_1_logical_or.html#a931b98fca3e19085af9fa97a43db8ced',1,'mlx::core::LogicalOr::output_shapes()'],['../classmlx_1_1core_1_1_log_add_exp.html#a234f8c8ea5f5bf2fb7e371588fea98b9',1,'mlx::core::LogAddExp::output_shapes()'],['../classmlx_1_1core_1_1_matmul.html#abfabe69f428f7f125bf5665713a0eb5c',1,'mlx::core::Matmul::output_shapes()'],['../classmlx_1_1core_1_1_maximum.html#a888a69fb68726c3c18973f3ea38cfd2b',1,'mlx::core::Maximum::output_shapes()'],['../classmlx_1_1core_1_1_minimum.html#af921b5202ebf9716972bcf0e3056742a',1,'mlx::core::Minimum::output_shapes()'],['../classmlx_1_1core_1_1_multiply.html#adfd4c7f89660b42ab58e088b1ae19435',1,'mlx::core::Multiply::output_shapes()'],['../classmlx_1_1core_1_1_negative.html#a606fb13a48d10c88707f1a2c41bee9e8',1,'mlx::core::Negative::output_shapes()'],['../classmlx_1_1core_1_1_not_equal.html#ad1e8a577dc103d96f1ab65bf3b389d35',1,'mlx::core::NotEqual::output_shapes()'],['../classmlx_1_1core_1_1_number_of_elements.html#a6cdf307348ba22b3dc8f90f1fb1e0757',1,'mlx::core::NumberOfElements::output_shapes()'],['../classmlx_1_1core_1_1_partition.html#a5e62aa0109e53fb4acb861ef39787b4a',1,'mlx::core::Partition::output_shapes()'],['../classmlx_1_1core_1_1_power.html#af23ed795bdcdc4c3f91f0d4c1bb1d928',1,'mlx::core::Power::output_shapes()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a7d57a31d41c58e1bd88ffe9c6b0dbf52',1,'mlx::core::QuantizedMatmul::output_shapes()'],['../classmlx_1_1core_1_1_real.html#a75999bd0b97d97a5675b9cdbab27dcff',1,'mlx::core::Real::output_shapes()'],['../classmlx_1_1core_1_1_reshape.html#aed3a83606d6917b2c344607101a2c43d',1,'mlx::core::Reshape::output_shapes()'],['../classmlx_1_1core_1_1_reduce.html#aaf3da1c98cdf530803118b382c5f58bc',1,'mlx::core::Reduce::output_shapes()'],['../classmlx_1_1core_1_1_round.html#a61821399e177e142723fc986e437d459',1,'mlx::core::Round::output_shapes()'],['../classmlx_1_1core_1_1_scatter_axis.html#af9688c010e1abee9b7b3788f11d91cc5',1,'mlx::core::ScatterAxis::output_shapes()'],['../classmlx_1_1core_1_1_sigmoid.html#aff024a3309584724c9842f172a4e440b',1,'mlx::core::Sigmoid::output_shapes()'],['../classmlx_1_1core_1_1_sign.html#a2260f2e8e081010192eb8a6f90acde6e',1,'mlx::core::Sign::output_shapes()'],['../classmlx_1_1core_1_1_sin.html#abdd433ecbb54898161b43aa9e14ec7f1',1,'mlx::core::Sin::output_shapes()'],['../classmlx_1_1core_1_1_sinh.html#ae04d8f6175c691a8f0d2a9fdd15af0ad',1,'mlx::core::Sinh::output_shapes()'],['../classmlx_1_1core_1_1_slice_update.html#abb6376f13c4269bd9e739e131893da53',1,'mlx::core::SliceUpdate::output_shapes()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a920dc4d1ee4976065e6d91fe3ecfbbf3',1,'mlx::core::DynamicSlice::output_shapes()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a804c03c745fc563e209a7bfb3d425a91',1,'mlx::core::DynamicSliceUpdate::output_shapes()'],['../classmlx_1_1core_1_1_softmax.html#a1a798a4dcd62486362d4b58582357490',1,'mlx::core::Softmax::output_shapes()'],['../classmlx_1_1core_1_1_sort.html#acc0a3f078b3f4c83e6e1137cb81ee62c',1,'mlx::core::Sort::output_shapes()'],['../classmlx_1_1core_1_1_square.html#a0513541766bb997ed166643fe95a6d38',1,'mlx::core::Square::output_shapes()'],['../classmlx_1_1core_1_1_sqrt.html#ae45215d61e2e99749d9a0bae291edd45',1,'mlx::core::Sqrt::output_shapes()'],['../classmlx_1_1core_1_1_stop_gradient.html#a8af7641d478505d1dc39c75ba7d5a3cf',1,'mlx::core::StopGradient::output_shapes()'],['../classmlx_1_1core_1_1_subtract.html#aaaff4872bde70ad40cf90e6131ea0489',1,'mlx::core::Subtract::output_shapes()'],['../classmlx_1_1core_1_1_squeeze.html#a839d9d72ac0a19e1146b5b470292a174',1,'mlx::core::Squeeze::output_shapes()'],['../classmlx_1_1core_1_1_tan.html#a9e4bba311bb24617dbb5ca591bc2868e',1,'mlx::core::Tan::output_shapes()'],['../classmlx_1_1core_1_1_tanh.html#a8873286b69b805486fa83c4806843f3d',1,'mlx::core::Tanh::output_shapes()'],['../classmlx_1_1core_1_1_unflatten.html#a068cf053b5b0612fafd4a2d53d42f9fa',1,'mlx::core::Unflatten::output_shapes()'],['../classmlx_1_1core_1_1_transpose.html#ac9328f43900bedec555909d09202ccd7',1,'mlx::core::Transpose::output_shapes()'],['../classmlx_1_1core_1_1_eigh.html#a9892f5b72dec19a5a2f7af5efcf2a952',1,'mlx::core::Eigh::output_shapes()']]], + ['outputs_60',['outputs',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a55a7a92c6abad369c99a5ede7a2521b9',1,'mlx::core::metal::DeviceStream::outputs'],['../classmlx_1_1core_1_1array.html#a2c186fd527f984f0589d4183b4976289',1,'mlx::core::array::outputs()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aefa48740fdee884f02e2d379bca4e78f',1,'mlx::core::metal::CommandEncoder::outputs()'],['../structmlx_1_1core_1_1_command_encoder.html#aefa48740fdee884f02e2d379bca4e78f',1,'mlx::core::CommandEncoder::outputs()']]], + ['overwrite_5fdescriptor_61',['overwrite_descriptor',['../classmlx_1_1core_1_1array.html#a95e6b156c8e05439f076b85c05079387',1,'mlx::core::array']]] ]; diff --git a/docs/build/html/search/classes_1.js b/docs/build/html/search/classes_1.js index cf4c90cb0..1a29858f0 100644 --- a/docs/build/html/search/classes_1.js +++ b/docs/build/html/search/classes_1.js @@ -4,7 +4,7 @@ var searchData= ['accumhelper_1',['AccumHelper',['../structmlx_1_1steel_1_1_accum_helper.html',1,'mlx::steel']]], ['add_2',['Add',['../struct_add.html',1,'Add'],['../classmlx_1_1core_1_1_add.html',1,'mlx::core::Add'],['../structmlx_1_1core_1_1detail_1_1_add.html',1,'mlx::core::detail::Add']]], ['add_5fvec_3',['add_vec',['../structpocketfft_1_1detail_1_1add__vec.html',1,'pocketfft::detail']]], - ['add_5fvec_3c_20cmplx_3c_20t_20_3e_20_3e_4',['add_vec< cmplx< T > >',['../structpocketfft_1_1detail_1_1add__vec.html',1,'pocketfft::detail']]], + ['add_5fvec_3c_20cmplx_3c_20t_20_3e_20_3e_4',['add_vec< cmplx< T > >',['../structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html',1,'pocketfft::detail']]], ['addmm_5',['AddMM',['../classmlx_1_1core_1_1_add_m_m.html',1,'mlx::core']]], ['affinequantize_6',['AffineQuantize',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html',1,'mlx::core::fast']]], ['aligned_5fallocator_7',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html',1,'pocketfft::detail::threading']]], diff --git a/docs/build/html/search/classes_10.js b/docs/build/html/search/classes_10.js index e7553bdff..1501af001 100644 --- a/docs/build/html/search/classes_10.js +++ b/docs/build/html/search/classes_10.js @@ -6,10 +6,10 @@ var searchData= ['pocketfft_5fc_3',['pocketfft_c',['../classpocketfft_1_1detail_1_1pocketfft__c.html',1,'pocketfft::detail']]], ['pocketfft_5fr_4',['pocketfft_r',['../classpocketfft_1_1detail_1_1pocketfft__r.html',1,'pocketfft::detail']]], ['pointer_5felement_5',['pointer_element',['../structmetal_1_1pointer__element.html',1,'metal']]], - ['pointer_5felement_3c_20constant_20t_20_2a_20_3e_6',['pointer_element< constant T * >',['../structmetal_1_1pointer__element.html',1,'metal']]], - ['pointer_5felement_3c_20device_20t_20_2a_20_3e_7',['pointer_element< device T * >',['../structmetal_1_1pointer__element.html',1,'metal']]], - ['pointer_5felement_3c_20thread_20t_20_2a_20_3e_8',['pointer_element< thread T * >',['../structmetal_1_1pointer__element.html',1,'metal']]], - ['pointer_5felement_3c_20threadgroup_20t_20_2a_20_3e_9',['pointer_element< threadgroup T * >',['../structmetal_1_1pointer__element.html',1,'metal']]], + ['pointer_5felement_3c_20constant_20t_20_2a_20_3e_6',['pointer_element< constant T * >',['../structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.html',1,'metal']]], + ['pointer_5felement_3c_20device_20t_20_2a_20_3e_7',['pointer_element< device T * >',['../structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.html',1,'metal']]], + ['pointer_5felement_3c_20thread_20t_20_2a_20_3e_8',['pointer_element< thread T * >',['../structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.html',1,'metal']]], + ['pointer_5felement_3c_20threadgroup_20t_20_2a_20_3e_9',['pointer_element< threadgroup T * >',['../structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.html',1,'metal']]], ['power_10',['Power',['../structmlx_1_1core_1_1detail_1_1_power.html',1,'mlx::core::detail::Power'],['../classmlx_1_1core_1_1_power.html',1,'mlx::core::Power'],['../struct_power.html',1,'Power']]], ['primitive_11',['Primitive',['../classmlx_1_1core_1_1_primitive.html',1,'mlx::core']]], ['printformatter_12',['PrintFormatter',['../structmlx_1_1core_1_1_print_formatter.html',1,'mlx::core']]], diff --git a/docs/build/html/search/classes_13.js b/docs/build/html/search/classes_13.js index 9f4545215..a3bd326e2 100644 --- a/docs/build/html/search/classes_13.js +++ b/docs/build/html/search/classes_13.js @@ -1,34 +1,44 @@ var searchData= [ - ['scaleddotproductattention_0',['ScaledDotProductAttention',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html',1,'mlx::core::fast']]], - ['scaleop_1',['ScaleOp',['../struct_scale_op.html',1,'']]], - ['scan_2',['Scan',['../classmlx_1_1core_1_1_scan.html',1,'mlx::core']]], - ['scatter_3',['Scatter',['../classmlx_1_1core_1_1_scatter.html',1,'mlx::core']]], - ['scheduler_4',['Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html',1,'mlx::core::scheduler']]], - ['select_5',['Select',['../structmlx_1_1core_1_1detail_1_1_select.html',1,'mlx::core::detail::Select'],['../classmlx_1_1core_1_1_select.html',1,'mlx::core::Select'],['../struct_select.html',1,'Select']]], - ['send_6',['Send',['../classmlx_1_1core_1_1distributed_1_1_send.html',1,'mlx::core::distributed']]], - ['shape2d_7',['Shape2D',['../structmlx_1_1steel_1_1_shape2_d.html',1,'mlx::steel']]], - ['sigmoid_8',['Sigmoid',['../structmlx_1_1core_1_1detail_1_1_sigmoid.html',1,'mlx::core::detail::Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html',1,'mlx::core::Sigmoid'],['../struct_sigmoid.html',1,'Sigmoid']]], - ['sign_9',['Sign',['../structmlx_1_1core_1_1detail_1_1_sign.html',1,'mlx::core::detail::Sign'],['../classmlx_1_1core_1_1_sign.html',1,'mlx::core::Sign'],['../struct_sign.html',1,'Sign']]], - ['simple_5fiter_10',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html',1,'pocketfft::detail']]], - ['sin_11',['Sin',['../structmlx_1_1core_1_1detail_1_1_sin.html',1,'mlx::core::detail::Sin'],['../classmlx_1_1core_1_1_sin.html',1,'mlx::core::Sin'],['../struct_sin.html',1,'Sin']]], - ['sincos_5f2pibyn_12',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html',1,'pocketfft::detail']]], - ['sinh_13',['Sinh',['../structmlx_1_1core_1_1detail_1_1_sinh.html',1,'mlx::core::detail::Sinh'],['../classmlx_1_1core_1_1_sinh.html',1,'mlx::core::Sinh'],['../struct_sinh.html',1,'Sinh']]], - ['slice_14',['Slice',['../classmlx_1_1core_1_1_slice.html',1,'mlx::core']]], - ['sliceupdate_15',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html',1,'mlx::core']]], - ['softmax_16',['Softmax',['../classmlx_1_1core_1_1_softmax.html',1,'mlx::core']]], - ['sort_17',['Sort',['../classmlx_1_1core_1_1_sort.html',1,'mlx::core']]], - ['split_18',['Split',['../classmlx_1_1core_1_1_split.html',1,'mlx::core']]], - ['sqrt_19',['Sqrt',['../structmlx_1_1core_1_1detail_1_1_sqrt.html',1,'mlx::core::detail::Sqrt'],['../classmlx_1_1core_1_1_sqrt.html',1,'mlx::core::Sqrt'],['../struct_sqrt.html',1,'Sqrt']]], - ['square_20',['Square',['../structmlx_1_1core_1_1detail_1_1_square.html',1,'mlx::core::detail::Square'],['../classmlx_1_1core_1_1_square.html',1,'mlx::core::Square'],['../struct_square.html',1,'Square']]], - ['squeeze_21',['Squeeze',['../classmlx_1_1core_1_1_squeeze.html',1,'mlx::core']]], - ['stopgradient_22',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html',1,'mlx::core']]], - ['stream_23',['Stream',['../structmlx_1_1core_1_1_stream.html',1,'mlx::core']]], - ['streamcontext_24',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html',1,'mlx::core']]], - ['streamthread_25',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html',1,'mlx::core::scheduler']]], - ['subop_26',['SubOp',['../struct_sub_op.html',1,'']]], - ['subtract_27',['Subtract',['../structmlx_1_1core_1_1detail_1_1_subtract.html',1,'mlx::core::detail::Subtract'],['../classmlx_1_1core_1_1_subtract.html',1,'mlx::core::Subtract'],['../struct_subtract.html',1,'Subtract']]], - ['sum_28',['Sum',['../struct_sum.html',1,'']]], - ['sumop_29',['SumOp',['../struct_sum_op.html',1,'']]], - ['svd_30',['SVD',['../classmlx_1_1core_1_1_s_v_d.html',1,'mlx::core']]] + ['scalart_0',['ScalarT',['../structmlx_1_1core_1_1simd_1_1_scalar_t.html',1,'mlx::core::simd']]], + ['scalart_3c_20bool_2c_20n_20_3e_1',['ScalarT< bool, N >',['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['scalart_3c_20int64_5ft_2c_20n_20_3e_2',['ScalarT< int64_t, N >',['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['scalart_3c_20int8_5ft_2c_20n_20_3e_3',['ScalarT< int8_t, N >',['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['scalart_3c_20uint64_5ft_2c_20n_20_3e_4',['ScalarT< uint64_t, N >',['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['scalarvector_5',['ScalarVector',['../structmlx_1_1core_1_1_scalar_vector.html',1,'mlx::core']]], + ['scaleddotproductattention_6',['ScaledDotProductAttention',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html',1,'mlx::core::fast']]], + ['scaleop_7',['ScaleOp',['../struct_scale_op.html',1,'']]], + ['scan_8',['Scan',['../classmlx_1_1core_1_1_scan.html',1,'mlx::core']]], + ['scatter_9',['Scatter',['../classmlx_1_1core_1_1_scatter.html',1,'mlx::core']]], + ['scatteraxis_10',['ScatterAxis',['../classmlx_1_1core_1_1_scatter_axis.html',1,'mlx::core']]], + ['scheduler_11',['Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html',1,'mlx::core::scheduler']]], + ['select_12',['Select',['../structmlx_1_1core_1_1detail_1_1_select.html',1,'mlx::core::detail::Select'],['../classmlx_1_1core_1_1_select.html',1,'mlx::core::Select'],['../struct_select.html',1,'Select']]], + ['send_13',['Send',['../classmlx_1_1core_1_1distributed_1_1_send.html',1,'mlx::core::distributed']]], + ['shape2d_14',['Shape2D',['../structmlx_1_1steel_1_1_shape2_d.html',1,'mlx::steel']]], + ['sigmoid_15',['Sigmoid',['../structmlx_1_1core_1_1detail_1_1_sigmoid.html',1,'mlx::core::detail::Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html',1,'mlx::core::Sigmoid'],['../struct_sigmoid.html',1,'Sigmoid']]], + ['sign_16',['Sign',['../structmlx_1_1core_1_1detail_1_1_sign.html',1,'mlx::core::detail::Sign'],['../classmlx_1_1core_1_1_sign.html',1,'mlx::core::Sign'],['../struct_sign.html',1,'Sign']]], + ['simd_17',['Simd',['../structmlx_1_1core_1_1simd_1_1_simd.html',1,'mlx::core::simd']]], + ['simd_3c_20float16_5ft_2c_20n_20_3e_18',['Simd< float16_t, N >',['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html',1,'mlx::core::simd']]], + ['simd_3c_20t_2c_201_20_3e_19',['Simd< T, 1 >',['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html',1,'mlx::core::simd']]], + ['simple_5fiter_20',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html',1,'pocketfft::detail']]], + ['sin_21',['Sin',['../structmlx_1_1core_1_1detail_1_1_sin.html',1,'mlx::core::detail::Sin'],['../classmlx_1_1core_1_1_sin.html',1,'mlx::core::Sin'],['../struct_sin.html',1,'Sin']]], + ['sincos_5f2pibyn_22',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html',1,'pocketfft::detail']]], + ['sinh_23',['Sinh',['../structmlx_1_1core_1_1detail_1_1_sinh.html',1,'mlx::core::detail::Sinh'],['../classmlx_1_1core_1_1_sinh.html',1,'mlx::core::Sinh'],['../struct_sinh.html',1,'Sinh']]], + ['slice_24',['Slice',['../classmlx_1_1core_1_1_slice.html',1,'mlx::core']]], + ['sliceupdate_25',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html',1,'mlx::core']]], + ['softmax_26',['Softmax',['../classmlx_1_1core_1_1_softmax.html',1,'mlx::core']]], + ['sort_27',['Sort',['../classmlx_1_1core_1_1_sort.html',1,'mlx::core']]], + ['split_28',['Split',['../classmlx_1_1core_1_1_split.html',1,'mlx::core']]], + ['sqrt_29',['Sqrt',['../structmlx_1_1core_1_1detail_1_1_sqrt.html',1,'mlx::core::detail::Sqrt'],['../classmlx_1_1core_1_1_sqrt.html',1,'mlx::core::Sqrt'],['../struct_sqrt.html',1,'Sqrt']]], + ['square_30',['Square',['../structmlx_1_1core_1_1detail_1_1_square.html',1,'mlx::core::detail::Square'],['../classmlx_1_1core_1_1_square.html',1,'mlx::core::Square'],['../struct_square.html',1,'Square']]], + ['squeeze_31',['Squeeze',['../classmlx_1_1core_1_1_squeeze.html',1,'mlx::core']]], + ['stopgradient_32',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html',1,'mlx::core']]], + ['stream_33',['Stream',['../structmlx_1_1core_1_1_stream.html',1,'mlx::core']]], + ['streamcontext_34',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html',1,'mlx::core']]], + ['streamthread_35',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html',1,'mlx::core::scheduler']]], + ['subop_36',['SubOp',['../struct_sub_op.html',1,'']]], + ['subtract_37',['Subtract',['../structmlx_1_1core_1_1detail_1_1_subtract.html',1,'mlx::core::detail::Subtract'],['../classmlx_1_1core_1_1_subtract.html',1,'mlx::core::Subtract'],['../struct_subtract.html',1,'Subtract']]], + ['sum_38',['Sum',['../struct_sum.html',1,'']]], + ['sumop_39',['SumOp',['../struct_sum_op.html',1,'']]], + ['svd_40',['SVD',['../classmlx_1_1core_1_1_s_v_d.html',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/classes_16.js b/docs/build/html/search/classes_16.js index e6aaa0804..c0c6525fd 100644 --- a/docs/build/html/search/classes_16.js +++ b/docs/build/html/search/classes_16.js @@ -1,6 +1,8 @@ var searchData= [ - ['view_0',['View',['../classmlx_1_1core_1_1_view.html',1,'mlx::core']]], - ['vlen_1',['VLEN',['../structpocketfft_1_1detail_1_1_v_l_e_n.html',1,'pocketfft::detail']]], - ['vtype_2',['VTYPE',['../structpocketfft_1_1detail_1_1_v_t_y_p_e.html',1,'pocketfft::detail']]] + ['vectorscalar_0',['VectorScalar',['../structmlx_1_1core_1_1_vector_scalar.html',1,'mlx::core']]], + ['vectorvector_1',['VectorVector',['../structmlx_1_1core_1_1_vector_vector.html',1,'mlx::core']]], + ['view_2',['View',['../classmlx_1_1core_1_1_view.html',1,'mlx::core']]], + ['vlen_3',['VLEN',['../structpocketfft_1_1detail_1_1_v_l_e_n.html',1,'pocketfft::detail']]], + ['vtype_4',['VTYPE',['../structpocketfft_1_1detail_1_1_v_t_y_p_e.html',1,'pocketfft::detail']]] ]; diff --git a/docs/build/html/search/classes_2.js b/docs/build/html/search/classes_2.js index 8561b318e..c4696c579 100644 --- a/docs/build/html/search/classes_2.js +++ b/docs/build/html/search/classes_2.js @@ -2,24 +2,23 @@ var searchData= [ ['basemmafrag_0',['BaseMMAFrag',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]], ['basemmafrag_3c_20accumtype_2c_20kfragsize_2c_20kfragsize_20_3e_1',['BaseMMAFrag< AccumType, kFragSize, kFragSize >',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]], - ['basemmafrag_3c_20t_2c_208_2c_208_20_3e_2',['BaseMMAFrag< T, 8, 8 >',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html',1,'mlx::steel']]], - ['bfloat_5for_5fbits_3',['bfloat_or_bits',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core::numeric_limits< bfloat16_t >']]], - ['bits_5fto_5fbfloat_5fstruct_4',['bits_to_bfloat_struct',['../struct___m_l_x___b_float16_1_1bits__to__bfloat__struct.html',1,'_MLX_BFloat16']]], - ['bitwiseand_5',['BitwiseAnd',['../struct_bitwise_and.html',1,'BitwiseAnd'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html',1,'mlx::core::detail::BitwiseAnd']]], - ['bitwisebinary_6',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html',1,'mlx::core']]], - ['bitwiseor_7',['BitwiseOr',['../struct_bitwise_or.html',1,'BitwiseOr'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html',1,'mlx::core::detail::BitwiseOr']]], - ['bitwisexor_8',['BitwiseXor',['../struct_bitwise_xor.html',1,'BitwiseXor'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html',1,'mlx::core::detail::BitwiseXor']]], - ['blockloader_9',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], - ['blockloader_3c_20t_2c_20transpose_5fa_20_3f_20bk_20_3abm_2c_20transpose_5fa_20_3f_20bm_20_3abk_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20_21transpose_5fa_2c_20tgp_5fsize_20_3e_10',['BlockLoader< T, transpose_a ? BK :BM, transpose_a ? BM :BK, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, !transpose_a, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], - ['blockloader_3c_20t_2c_20transpose_5fb_20_3f_20bn_20_3abk_2c_20transpose_5fb_20_3f_20bk_20_3abn_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20transpose_5fb_2c_20tgp_5fsize_20_3e_11',['BlockLoader< T, transpose_b ? BN :BK, transpose_b ? BK :BN, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, transpose_b, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], - ['blockloadert_12',['BlockLoaderT',['../structmlx_1_1steel_1_1_block_loader_t.html',1,'mlx::steel']]], - ['blockmaskedmm_13',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html',1,'mlx::core']]], - ['blockmergesort_14',['BlockMergeSort',['../struct_block_merge_sort.html',1,'']]], - ['blockmma_15',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel']]], - ['blockmma_3c_20t_2c_20u_2c_20bm_2c_20bn_2c_20bk_2c_20wm_2c_20wn_2c_20transpose_5fa_2c_20transpose_5fb_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20accumtype_2c_20epilogue_20_3e_16',['BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, AccumType, Epilogue >',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel']]], - ['blockswizzle_17',['BlockSwizzle',['../structmlx_1_1steel_1_1_block_swizzle.html',1,'mlx::steel']]], - ['bool4_5for_5fuint_18',['bool4_or_uint',['../unionbool4__or__uint.html',1,'']]], - ['broadcast_19',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html',1,'mlx::core']]], - ['broadcastaxes_20',['BroadcastAxes',['../classmlx_1_1core_1_1_broadcast_axes.html',1,'mlx::core']]], - ['buffer_21',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html',1,'mlx::core::allocator::Buffer'],['../classmlx_1_1core_1_1metal_1_1_buffer.html',1,'mlx::core::metal::Buffer']]] + ['basemmafrag_3c_20t_2c_208_2c_208_20_3e_2',['BaseMMAFrag< T, 8, 8 >',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html',1,'mlx::steel']]], + ['bits_5fto_5fbfloat_5fstruct_3',['bits_to_bfloat_struct',['../struct___m_l_x___b_float16_1_1bits__to__bfloat__struct.html',1,'_MLX_BFloat16']]], + ['bitwiseand_4',['BitwiseAnd',['../struct_bitwise_and.html',1,'BitwiseAnd'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html',1,'mlx::core::detail::BitwiseAnd']]], + ['bitwisebinary_5',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html',1,'mlx::core']]], + ['bitwiseor_6',['BitwiseOr',['../struct_bitwise_or.html',1,'BitwiseOr'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html',1,'mlx::core::detail::BitwiseOr']]], + ['bitwisexor_7',['BitwiseXor',['../struct_bitwise_xor.html',1,'BitwiseXor'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html',1,'mlx::core::detail::BitwiseXor']]], + ['blockloader_8',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], + ['blockloader_3c_20t_2c_20transpose_5fa_20_3f_20bk_20_3abm_2c_20transpose_5fa_20_3f_20bm_20_3abk_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20_21transpose_5fa_2c_20tgp_5fsize_20_3e_9',['BlockLoader< T, transpose_a ? BK :BM, transpose_a ? BM :BK, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, !transpose_a, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], + ['blockloader_3c_20t_2c_20transpose_5fb_20_3f_20bn_20_3abk_2c_20transpose_5fb_20_3f_20bk_20_3abn_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20transpose_5fb_2c_20tgp_5fsize_20_3e_10',['BlockLoader< T, transpose_b ? BN :BK, transpose_b ? BK :BN, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, transpose_b, tgp_size >',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel']]], + ['blockloadert_11',['BlockLoaderT',['../structmlx_1_1steel_1_1_block_loader_t.html',1,'mlx::steel']]], + ['blockmaskedmm_12',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html',1,'mlx::core']]], + ['blockmergesort_13',['BlockMergeSort',['../struct_block_merge_sort.html',1,'']]], + ['blockmma_14',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel']]], + ['blockmma_3c_20t_2c_20u_2c_20bm_2c_20bn_2c_20bk_2c_20wm_2c_20wn_2c_20transpose_5fa_2c_20transpose_5fb_2c_20transpose_5fa_20_3f_20bm_2btgp_5fpadding_5fa_20_3abk_2btgp_5fpadding_5fa_2c_20transpose_5fb_20_3f_20bk_2btgp_5fpadding_5fb_20_3abn_2btgp_5fpadding_5fb_2c_20accumtype_2c_20epilogue_20_3e_15',['BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, transpose_a ? BM+tgp_padding_a :BK+tgp_padding_a, transpose_b ? BK+tgp_padding_b :BN+tgp_padding_b, AccumType, Epilogue >',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel']]], + ['blockswizzle_16',['BlockSwizzle',['../structmlx_1_1steel_1_1_block_swizzle.html',1,'mlx::steel']]], + ['bool4_5for_5fuint_17',['bool4_or_uint',['../unionbool4__or__uint.html',1,'']]], + ['broadcast_18',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html',1,'mlx::core']]], + ['broadcastaxes_19',['BroadcastAxes',['../classmlx_1_1core_1_1_broadcast_axes.html',1,'mlx::core']]], + ['buffer_20',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html',1,'mlx::core::allocator::Buffer'],['../classmlx_1_1core_1_1metal_1_1_buffer.html',1,'mlx::core::metal::Buffer']]] ]; diff --git a/docs/build/html/search/classes_3.js b/docs/build/html/search/classes_3.js index c1cf1a75f..5e6fefacf 100644 --- a/docs/build/html/search/classes_3.js +++ b/docs/build/html/search/classes_3.js @@ -3,10 +3,10 @@ var searchData= ['ceil_0',['Ceil',['../struct_ceil.html',1,'Ceil'],['../classmlx_1_1core_1_1_ceil.html',1,'mlx::core::Ceil'],['../structmlx_1_1core_1_1detail_1_1_ceil.html',1,'mlx::core::detail::Ceil']]], ['cfftp_1',['cfftp',['../classpocketfft_1_1detail_1_1cfftp.html',1,'pocketfft::detail']]], ['channelhelper_2',['ChannelHelper',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], - ['channelhelper_3c_201_20_3e_3',['ChannelHelper< 1 >',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], - ['channelhelper_3c_202_20_3e_4',['ChannelHelper< 2 >',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], - ['channelhelper_3c_203_20_3e_5',['ChannelHelper< 3 >',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], - ['channelhelper_3c_204_20_3e_6',['ChannelHelper< 4 >',['../structmlx_1_1steel_1_1_channel_helper.html',1,'mlx::steel']]], + ['channelhelper_3c_201_20_3e_3',['ChannelHelper< 1 >',['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html',1,'mlx::steel']]], + ['channelhelper_3c_202_20_3e_4',['ChannelHelper< 2 >',['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html',1,'mlx::steel']]], + ['channelhelper_3c_203_20_3e_5',['ChannelHelper< 3 >',['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html',1,'mlx::steel']]], + ['channelhelper_3c_204_20_3e_6',['ChannelHelper< 4 >',['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html',1,'mlx::steel']]], ['cholesky_7',['Cholesky',['../classmlx_1_1core_1_1_cholesky.html',1,'mlx::core']]], ['cmplx_8',['cmplx',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], ['cmplx_3c_20t0_20_3e_9',['cmplx< T0 >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], @@ -23,7 +23,7 @@ var searchData= ['concurrent_5fqueue_3c_20std_3a_3afunction_3c_20void_28_29_3e_20_3e_20',['concurrent_queue< std::function< void()> >',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html',1,'pocketfft::detail::threading']]], ['concurrentcontext_21',['ConcurrentContext',['../structmlx_1_1core_1_1_command_encoder_1_1_concurrent_context.html',1,'mlx::core::CommandEncoder::ConcurrentContext'],['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html',1,'mlx::core::metal::CommandEncoder::ConcurrentContext']]], ['conditionaltype_22',['ConditionalType',['../struct_conditional_type.html',1,'']]], - ['conditionaltype_3c_20true_2c_20t_2c_20u_20_3e_23',['ConditionalType< true, T, U >',['../struct_conditional_type.html',1,'']]], + ['conditionaltype_3c_20true_2c_20t_2c_20u_20_3e_23',['ConditionalType< true, T, U >',['../struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.html',1,'']]], ['conjugate_24',['Conjugate',['../struct_conjugate.html',1,'Conjugate'],['../classmlx_1_1core_1_1_conjugate.html',1,'mlx::core::Conjugate'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html',1,'mlx::core::detail::Conjugate']]], ['contiguous_25',['Contiguous',['../classmlx_1_1core_1_1_contiguous.html',1,'mlx::core']]], ['contiguousiterator_26',['ContiguousIterator',['../structmlx_1_1core_1_1_contiguous_iterator.html',1,'mlx::core']]], @@ -44,7 +44,7 @@ var searchData= ['cummax_41',['CumMax',['../struct_cum_max.html',1,'']]], ['cummin_42',['CumMin',['../struct_cum_min.html',1,'']]], ['cumprod_43',['CumProd',['../struct_cum_prod.html',1,'']]], - ['cumprod_3c_20bool_20_3e_44',['CumProd< bool >',['../struct_cum_prod.html',1,'']]], + ['cumprod_3c_20bool_20_3e_44',['CumProd< bool >',['../struct_cum_prod_3_01bool_01_4.html',1,'']]], ['cumsum_45',['CumSum',['../struct_cum_sum.html',1,'']]], ['custom_46',['Custom',['../classmlx_1_1core_1_1fast_1_1_custom.html',1,'mlx::core::fast']]], ['customkernel_47',['CustomKernel',['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html',1,'mlx::core::fast']]], diff --git a/docs/build/html/search/classes_4.js b/docs/build/html/search/classes_4.js index 60eb61d49..3800cc4cb 100644 --- a/docs/build/html/search/classes_4.js +++ b/docs/build/html/search/classes_4.js @@ -1,16 +1,14 @@ var searchData= [ ['data_0',['Data',['../structmlx_1_1core_1_1array_1_1_data.html',1,'mlx::core::array']]], - ['defaultcontiguousreduce_1',['DefaultContiguousReduce',['../structmlx_1_1core_1_1_default_contiguous_reduce.html',1,'mlx::core']]], - ['defaultstridedreduce_2',['DefaultStridedReduce',['../structmlx_1_1core_1_1_default_strided_reduce.html',1,'mlx::core']]], - ['depends_3',['Depends',['../classmlx_1_1core_1_1_depends.html',1,'mlx::core']]], - ['device_4',['Device',['../structmlx_1_1core_1_1_device.html',1,'mlx::core::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html',1,'mlx::core::metal::Device']]], - ['devicestream_5',['DeviceStream',['../structmlx_1_1core_1_1metal_1_1_device_stream.html',1,'mlx::core::metal']]], - ['distprimitive_6',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html',1,'mlx::core::distributed']]], - ['divide_7',['Divide',['../struct_divide.html',1,'Divide'],['../structmlx_1_1core_1_1detail_1_1_divide.html',1,'mlx::core::detail::Divide'],['../classmlx_1_1core_1_1_divide.html',1,'mlx::core::Divide']]], - ['divmod_8',['DivMod',['../struct_div_mod.html',1,'DivMod'],['../classmlx_1_1core_1_1_div_mod.html',1,'mlx::core::DivMod']]], - ['divop_9',['DivOp',['../struct_div_op.html',1,'']]], - ['dtype_10',['Dtype',['../structmlx_1_1core_1_1_dtype.html',1,'mlx::core']]], - ['dynamicslice_11',['DynamicSlice',['../classmlx_1_1core_1_1_dynamic_slice.html',1,'mlx::core']]], - ['dynamicsliceupdate_12',['DynamicSliceUpdate',['../classmlx_1_1core_1_1_dynamic_slice_update.html',1,'mlx::core']]] + ['depends_1',['Depends',['../classmlx_1_1core_1_1_depends.html',1,'mlx::core']]], + ['device_2',['Device',['../structmlx_1_1core_1_1_device.html',1,'mlx::core::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html',1,'mlx::core::metal::Device']]], + ['devicestream_3',['DeviceStream',['../structmlx_1_1core_1_1metal_1_1_device_stream.html',1,'mlx::core::metal']]], + ['distprimitive_4',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html',1,'mlx::core::distributed']]], + ['divide_5',['Divide',['../struct_divide.html',1,'Divide'],['../structmlx_1_1core_1_1detail_1_1_divide.html',1,'mlx::core::detail::Divide'],['../classmlx_1_1core_1_1_divide.html',1,'mlx::core::Divide']]], + ['divmod_6',['DivMod',['../struct_div_mod.html',1,'DivMod'],['../classmlx_1_1core_1_1_div_mod.html',1,'mlx::core::DivMod']]], + ['divop_7',['DivOp',['../struct_div_op.html',1,'']]], + ['dtype_8',['Dtype',['../structmlx_1_1core_1_1_dtype.html',1,'mlx::core']]], + ['dynamicslice_9',['DynamicSlice',['../classmlx_1_1core_1_1_dynamic_slice.html',1,'mlx::core']]], + ['dynamicsliceupdate_10',['DynamicSliceUpdate',['../classmlx_1_1core_1_1_dynamic_slice_update.html',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/classes_6.js b/docs/build/html/search/classes_6.js index 1559cdbcf..1d2aef597 100644 --- a/docs/build/html/search/classes_6.js +++ b/docs/build/html/search/classes_6.js @@ -1,6 +1,6 @@ var searchData= [ - ['fence_0',['Fence',['../structmlx_1_1core_1_1metal_1_1_fence.html',1,'mlx::core::metal']]], + ['fence_0',['Fence',['../classmlx_1_1core_1_1_fence.html',1,'mlx::core::Fence'],['../structmlx_1_1core_1_1metal_1_1_fence.html',1,'mlx::core::metal::Fence']]], ['fft_1',['FFT',['../classmlx_1_1core_1_1_f_f_t.html',1,'mlx::core']]], ['fftblue_2',['fftblue',['../classpocketfft_1_1detail_1_1fftblue.html',1,'pocketfft::detail']]], ['filewriter_3',['FileWriter',['../classmlx_1_1core_1_1io_1_1_file_writer.html',1,'mlx::core::io']]], diff --git a/docs/build/html/search/classes_7.js b/docs/build/html/search/classes_7.js index 812f37a40..c224972c3 100644 --- a/docs/build/html/search/classes_7.js +++ b/docs/build/html/search/classes_7.js @@ -1,16 +1,17 @@ var searchData= [ ['gather_0',['Gather',['../classmlx_1_1core_1_1_gather.html',1,'mlx::core']]], - ['gathermm_1',['GatherMM',['../classmlx_1_1core_1_1_gather_m_m.html',1,'mlx::core']]], - ['gatherqmm_2',['GatherQMM',['../classmlx_1_1core_1_1_gather_q_m_m.html',1,'mlx::core']]], - ['gemmaddmmparams_3',['GEMMAddMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html',1,'mlx::steel']]], - ['gemmkernel_4',['GEMMKernel',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html',1,'mlx::steel']]], - ['gemmparams_5',['GEMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_params.html',1,'mlx::steel']]], - ['gemmspiltkparams_6',['GEMMSpiltKParams',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html',1,'mlx::steel']]], - ['gemvkernel_7',['GEMVKernel',['../struct_g_e_m_v_kernel.html',1,'']]], - ['gemvtkernel_8',['GEMVTKernel',['../struct_g_e_m_v_t_kernel.html',1,'']]], - ['greater_9',['Greater',['../struct_greater.html',1,'Greater'],['../structmlx_1_1core_1_1detail_1_1_greater.html',1,'mlx::core::detail::Greater'],['../classmlx_1_1core_1_1_greater.html',1,'mlx::core::Greater']]], - ['greaterequal_10',['GreaterEqual',['../struct_greater_equal.html',1,'GreaterEqual'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html',1,'mlx::core::detail::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html',1,'mlx::core::GreaterEqual']]], - ['group_11',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html',1,'mlx::core::distributed']]], - ['groupimpl_12',['GroupImpl',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html',1,'mlx::core::distributed::detail']]] + ['gatheraxis_1',['GatherAxis',['../classmlx_1_1core_1_1_gather_axis.html',1,'mlx::core']]], + ['gathermm_2',['GatherMM',['../classmlx_1_1core_1_1_gather_m_m.html',1,'mlx::core']]], + ['gatherqmm_3',['GatherQMM',['../classmlx_1_1core_1_1_gather_q_m_m.html',1,'mlx::core']]], + ['gemmaddmmparams_4',['GEMMAddMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html',1,'mlx::steel']]], + ['gemmkernel_5',['GEMMKernel',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html',1,'mlx::steel']]], + ['gemmparams_6',['GEMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_params.html',1,'mlx::steel']]], + ['gemmspiltkparams_7',['GEMMSpiltKParams',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html',1,'mlx::steel']]], + ['gemvkernel_8',['GEMVKernel',['../struct_g_e_m_v_kernel.html',1,'']]], + ['gemvtkernel_9',['GEMVTKernel',['../struct_g_e_m_v_t_kernel.html',1,'']]], + ['greater_10',['Greater',['../struct_greater.html',1,'Greater'],['../structmlx_1_1core_1_1detail_1_1_greater.html',1,'mlx::core::detail::Greater'],['../classmlx_1_1core_1_1_greater.html',1,'mlx::core::Greater']]], + ['greaterequal_11',['GreaterEqual',['../struct_greater_equal.html',1,'GreaterEqual'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html',1,'mlx::core::detail::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html',1,'mlx::core::GreaterEqual']]], + ['group_12',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html',1,'mlx::core::distributed']]], + ['groupimpl_13',['GroupImpl',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html',1,'mlx::core::distributed::detail']]] ]; diff --git a/docs/build/html/search/classes_8.js b/docs/build/html/search/classes_8.js index 0515a44a9..802cbce8e 100644 --- a/docs/build/html/search/classes_8.js +++ b/docs/build/html/search/classes_8.js @@ -1,5 +1,4 @@ var searchData= [ - ['hadamard_0',['Hadamard',['../classmlx_1_1core_1_1_hadamard.html',1,'mlx::core']]], - ['half_5for_5fbits_1',['half_or_bits',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core::numeric_limits< float16_t >']]] + ['hadamard_0',['Hadamard',['../classmlx_1_1core_1_1_hadamard.html',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/classes_9.js b/docs/build/html/search/classes_9.js index 8672ed52a..df05cfbab 100644 --- a/docs/build/html/search/classes_9.js +++ b/docs/build/html/search/classes_9.js @@ -9,11 +9,10 @@ var searchData= ['integral_5fconstant_3c_20bool_2c_20b_20_3e_3c_20false_20_3e_6',['integral_constant< bool, B >< false >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], ['integral_5fconstant_3c_20bool_2c_20b_20_3e_3c_20true_20_3e_7',['integral_constant< bool, B >< true >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], ['integral_5fconstant_3c_20int_2c_20val_20_3e_8',['integral_constant< int, val >',['../structmlx_1_1steel_1_1integral__constant.html',1,'mlx::steel']]], - ['intorfloat_9',['IntOrFloat',['../unionmlx_1_1core_1_1detail_1_1_int_or_float.html',1,'mlx::core::detail']]], - ['intracing_10',['InTracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html',1,'mlx::core::detail']]], - ['inverse_11',['Inverse',['../classmlx_1_1core_1_1_inverse.html',1,'mlx::core']]], - ['is_5fempty_12',['is_empty',['../structmetal_1_1is__empty.html',1,'metal']]], - ['is_5fintegral_13',['is_integral',['../structmlx_1_1steel_1_1is__integral.html',1,'mlx::steel']]], - ['is_5fintegral_3c_20integral_5fconstant_3c_20t_2c_20v_20_3e_20_3e_14',['is_integral< integral_constant< T, v > >',['../structmlx_1_1steel_1_1is__integral.html',1,'mlx::steel']]], - ['is_5fstatic_15',['is_static',['../structmetal_1_1is__static.html',1,'metal']]] + ['intracing_9',['InTracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html',1,'mlx::core::detail']]], + ['inverse_10',['Inverse',['../classmlx_1_1core_1_1_inverse.html',1,'mlx::core']]], + ['is_5fempty_11',['is_empty',['../structmetal_1_1is__empty.html',1,'metal']]], + ['is_5fintegral_12',['is_integral',['../structmlx_1_1steel_1_1is__integral.html',1,'mlx::steel']]], + ['is_5fintegral_3c_20integral_5fconstant_3c_20t_2c_20v_20_3e_20_3e_13',['is_integral< integral_constant< T, v > >',['../structmlx_1_1steel_1_1is__integral_3_01integral__constant_3_01_t_00_01v_01_4_01_4.html',1,'mlx::steel']]], + ['is_5fstatic_14',['is_static',['../structmetal_1_1is__static.html',1,'metal']]] ]; diff --git a/docs/build/html/search/classes_c.js b/docs/build/html/search/classes_c.js index 3fe54eda7..0a104128e 100644 --- a/docs/build/html/search/classes_c.js +++ b/docs/build/html/search/classes_c.js @@ -9,19 +9,19 @@ var searchData= ['lessequal_6',['LessEqual',['../struct_less_equal.html',1,'LessEqual'],['../structmlx_1_1core_1_1detail_1_1_less_equal.html',1,'mlx::core::detail::LessEqual'],['../classmlx_1_1core_1_1_less_equal.html',1,'mlx::core::LessEqual']]], ['lessthan_7',['LessThan',['../struct_less_than.html',1,'']]], ['limits_8',['Limits',['../struct_limits.html',1,'']]], - ['limits_3c_20bfloat16_5ft_20_3e_9',['Limits< bfloat16_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20bool_20_3e_10',['Limits< bool >',['../struct_limits.html',1,'']]], - ['limits_3c_20complex64_5ft_20_3e_11',['Limits< complex64_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20float_20_3e_12',['Limits< float >',['../struct_limits.html',1,'']]], - ['limits_3c_20half_20_3e_13',['Limits< half >',['../struct_limits.html',1,'']]], - ['limits_3c_20int16_5ft_20_3e_14',['Limits< int16_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20int32_5ft_20_3e_15',['Limits< int32_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20int64_5ft_20_3e_16',['Limits< int64_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20int8_5ft_20_3e_17',['Limits< int8_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20uint16_5ft_20_3e_18',['Limits< uint16_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20uint32_5ft_20_3e_19',['Limits< uint32_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20uint64_5ft_20_3e_20',['Limits< uint64_t >',['../struct_limits.html',1,'']]], - ['limits_3c_20uint8_5ft_20_3e_21',['Limits< uint8_t >',['../struct_limits.html',1,'']]], + ['limits_3c_20bfloat16_5ft_20_3e_9',['Limits< bfloat16_t >',['../struct_limits_3_01bfloat16__t_01_4.html',1,'']]], + ['limits_3c_20bool_20_3e_10',['Limits< bool >',['../struct_limits_3_01bool_01_4.html',1,'']]], + ['limits_3c_20complex64_5ft_20_3e_11',['Limits< complex64_t >',['../struct_limits_3_01complex64__t_01_4.html',1,'']]], + ['limits_3c_20float_20_3e_12',['Limits< float >',['../struct_limits_3_01float_01_4.html',1,'']]], + ['limits_3c_20half_20_3e_13',['Limits< half >',['../struct_limits_3_01half_01_4.html',1,'']]], + ['limits_3c_20int16_5ft_20_3e_14',['Limits< int16_t >',['../struct_limits_3_01int16__t_01_4.html',1,'']]], + ['limits_3c_20int32_5ft_20_3e_15',['Limits< int32_t >',['../struct_limits_3_01int32__t_01_4.html',1,'']]], + ['limits_3c_20int64_5ft_20_3e_16',['Limits< int64_t >',['../struct_limits_3_01int64__t_01_4.html',1,'']]], + ['limits_3c_20int8_5ft_20_3e_17',['Limits< int8_t >',['../struct_limits_3_01int8__t_01_4.html',1,'']]], + ['limits_3c_20uint16_5ft_20_3e_18',['Limits< uint16_t >',['../struct_limits_3_01uint16__t_01_4.html',1,'']]], + ['limits_3c_20uint32_5ft_20_3e_19',['Limits< uint32_t >',['../struct_limits_3_01uint32__t_01_4.html',1,'']]], + ['limits_3c_20uint64_5ft_20_3e_20',['Limits< uint64_t >',['../struct_limits_3_01uint64__t_01_4.html',1,'']]], + ['limits_3c_20uint8_5ft_20_3e_21',['Limits< uint8_t >',['../struct_limits_3_01uint8__t_01_4.html',1,'']]], ['load_22',['Load',['../classmlx_1_1core_1_1_load.html',1,'mlx::core']]], ['log_23',['Log',['../struct_log.html',1,'Log'],['../structmlx_1_1core_1_1detail_1_1_log.html',1,'mlx::core::detail::Log'],['../classmlx_1_1core_1_1_log.html',1,'mlx::core::Log']]], ['log10_24',['Log10',['../struct_log10.html',1,'Log10'],['../structmlx_1_1core_1_1detail_1_1_log10.html',1,'mlx::core::detail::Log10']]], @@ -33,6 +33,6 @@ var searchData= ['logicalor_30',['LogicalOr',['../struct_logical_or.html',1,'LogicalOr'],['../structmlx_1_1core_1_1detail_1_1_logical_or.html',1,'mlx::core::detail::LogicalOr'],['../classmlx_1_1core_1_1_logical_or.html',1,'mlx::core::LogicalOr']]], ['loopalignment_31',['LoopAlignment',['../structmlx_1_1steel_1_1_loop_alignment.html',1,'mlx::steel']]], ['loopedelemtoloc_32',['LoopedElemToLoc',['../struct_looped_elem_to_loc.html',1,'']]], - ['loopedelemtoloc_3c_201_2c_20offsett_2c_20false_20_3e_33',['LoopedElemToLoc< 1, OffsetT, false >',['../struct_looped_elem_to_loc.html',1,'']]], - ['loopedelemtoloc_3c_201_2c_20offsett_2c_20true_20_3e_34',['LoopedElemToLoc< 1, OffsetT, true >',['../struct_looped_elem_to_loc.html',1,'']]] + ['loopedelemtoloc_3c_201_2c_20offsett_2c_20false_20_3e_33',['LoopedElemToLoc< 1, OffsetT, false >',['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html',1,'']]], + ['loopedelemtoloc_3c_201_2c_20offsett_2c_20true_20_3e_34',['LoopedElemToLoc< 1, OffsetT, true >',['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html',1,'']]] ]; diff --git a/docs/build/html/search/classes_d.js b/docs/build/html/search/classes_d.js index 37b59f3a5..f43d285e5 100644 --- a/docs/build/html/search/classes_d.js +++ b/docs/build/html/search/classes_d.js @@ -9,7 +9,7 @@ var searchData= ['min_6',['Min',['../struct_min.html',1,'']]], ['minimum_7',['Minimum',['../struct_minimum.html',1,'Minimum'],['../structmlx_1_1core_1_1detail_1_1_minimum.html',1,'mlx::core::detail::Minimum'],['../classmlx_1_1core_1_1_minimum.html',1,'mlx::core::Minimum']]], ['mlx_5fatomic_8',['mlx_atomic',['../structmlx__atomic.html',1,'']]], - ['mlx_5fatomic_3c_20t_2c_20enable_5fif_5ft_3c_20is_5fmetal_5fatomic_3c_20t_20_3e_20_3e_20_3e_9',['mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >',['../structmlx__atomic.html',1,'']]], + ['mlx_5fatomic_3c_20t_2c_20enable_5fif_5ft_3c_20is_5fmetal_5fatomic_3c_20t_20_3e_20_3e_20_3e_9',['mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >',['../structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html',1,'']]], ['mlxconvparams_10',['MLXConvParams',['../struct_m_l_x_conv_params.html',1,'']]], ['mlxconvparams_3c_202_20_3e_11',['MLXConvParams< 2 >',['../struct_m_l_x_conv_params.html',1,'']]], ['mmatile_12',['MMATile',['../structmlx_1_1steel_1_1_m_m_a_tile.html',1,'mlx::steel']]], diff --git a/docs/build/html/search/classes_e.js b/docs/build/html/search/classes_e.js index 03adf6a06..23f44d71c 100644 --- a/docs/build/html/search/classes_e.js +++ b/docs/build/html/search/classes_e.js @@ -8,7 +8,7 @@ var searchData= ['notequal_5',['NotEqual',['../structmlx_1_1core_1_1detail_1_1_not_equal.html',1,'mlx::core::detail::NotEqual'],['../classmlx_1_1core_1_1_not_equal.html',1,'mlx::core::NotEqual'],['../struct_not_equal.html',1,'NotEqual']]], ['numberofelements_6',['NumberOfElements',['../classmlx_1_1core_1_1_number_of_elements.html',1,'mlx::core']]], ['numeric_5flimits_7',['numeric_limits',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core']]], - ['numeric_5flimits_3c_20bfloat16_5ft_20_3e_8',['numeric_limits< bfloat16_t >',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core']]], - ['numeric_5flimits_3c_20float_20_3e_9',['numeric_limits< float >',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core']]], - ['numeric_5flimits_3c_20float16_5ft_20_3e_10',['numeric_limits< float16_t >',['../structmlx_1_1core_1_1numeric__limits.html',1,'mlx::core']]] + ['numeric_5flimits_3c_20bfloat16_5ft_20_3e_8',['numeric_limits< bfloat16_t >',['../structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html',1,'mlx::core']]], + ['numeric_5flimits_3c_20float_20_3e_9',['numeric_limits< float >',['../structmlx_1_1core_1_1numeric__limits_3_01float_01_4.html',1,'mlx::core']]], + ['numeric_5flimits_3c_20float16_5ft_20_3e_10',['numeric_limits< float16_t >',['../structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/defines_2.js b/docs/build/html/search/defines_2.js index 03e79577b..38d742007 100644 --- a/docs/build/html/search/defines_2.js +++ b/docs/build/html/search/defines_2.js @@ -8,5 +8,6 @@ var searchData= ['bfloat_5finplace_5fbitop_5',['bfloat_inplace_bitop',['../types_2bf16_8h.html#af13b46bc58e6e6f675ae47aabec37711',1,'bf16.h']]], ['bfloat_5finplace_5fop_6',['bfloat_inplace_op',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ac82467fbc674e990090f482b9c1e5c',1,'bfloat_inplace_op: bf16.h'],['../types_2bf16_8h.html#aee905053f51f76e0c1af94199714d514',1,'bfloat_inplace_op: bf16.h']]], ['bfloat_5finplace_5fop_5faddr_5fspace_5fhelper_7',['bfloat_inplace_op_addr_space_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af30a2cbd2c3415516203b83bd21872f8',1,'bfloat_inplace_op_addr_space_helper: bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1457da931c28fa4e2500daa4e6441e8b',1,'bfloat_inplace_op_addr_space_helper: bf16.h']]], - ['bfloat_5finplace_5fop_5fhelper_8',['bfloat_inplace_op_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2846fd11b5e19b435e9f7ef0998c9b1d',1,'bfloat_inplace_op_helper: bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afe5988aa8147be2bafda6a5b7792fe15',1,'bfloat_inplace_op_helper: bf16.h']]] + ['bfloat_5finplace_5fop_5fhelper_8',['bfloat_inplace_op_helper',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2846fd11b5e19b435e9f7ef0998c9b1d',1,'bfloat_inplace_op_helper: bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afe5988aa8147be2bafda6a5b7792fe15',1,'bfloat_inplace_op_helper: bf16.h']]], + ['binary_5fsingle_9',['BINARY_SINGLE',['../cpu_2binary__ops_8h.html#ad606fe7b9586413e760cfe461c9be119',1,'binary_ops.h']]] ]; diff --git a/docs/build/html/search/defines_4.js b/docs/build/html/search/defines_4.js index ffff99cd1..1951ec29e 100644 --- a/docs/build/html/search/defines_4.js +++ b/docs/build/html/search/defines_4.js @@ -1,11 +1,21 @@ var searchData= [ - ['define_5fdefault_5fis_5fequivalent_0',['DEFINE_DEFAULT_IS_EQUIVALENT',['../primitives_8h.html#a0fb9d19207dc4869aca35abfbdf4d70a',1,'primitives.h']]], - ['define_5fgrads_1',['DEFINE_GRADS',['../primitives_8h.html#a77abdcb55bc2eb0f9a45edc5ee639bf6',1,'primitives.h']]], - ['define_5finput_5foutput_5fshape_2',['DEFINE_INPUT_OUTPUT_SHAPE',['../primitives_8h.html#a649a06267b75e007224ea4ddefedb999',1,'primitives.h']]], - ['define_5fprint_3',['DEFINE_PRINT',['../primitives_8h.html#a1d3a37af519e16f6a703b1e9ebd0f592',1,'primitives.h']]], - ['define_5fsimd_5fexclusive_5fscan_4',['DEFINE_SIMD_EXCLUSIVE_SCAN',['../scan_8h.html#a185f66aac8c5317587e6abd43f3013fc',1,'scan.h']]], - ['define_5fsimd_5freduce_5',['DEFINE_SIMD_REDUCE',['../backend_2metal_2kernels_2reduction_2ops_8h.html#acacf99e0ba629ed062ccc3c2eba89b05',1,'ops.h']]], - ['define_5fsimd_5fscan_6',['DEFINE_SIMD_SCAN',['../scan_8h.html#a0d8d6a9b0f3a1263629380bda8eca7bc',1,'scan.h']]], - ['define_5fvmap_7',['DEFINE_VMAP',['../primitives_8h.html#adc0fbd79fe0d1114dc85da4ed99798bd',1,'primitives.h']]] + ['default_5fbinary_0',['DEFAULT_BINARY',['../base__simd_8h.html#a1d01fad7230543d56f4facf17dcc4047',1,'base_simd.h']]], + ['default_5fbinary_5fop_1',['DEFAULT_BINARY_OP',['../cpu_2binary__ops_8h.html#a3f97e0a58306201b54eb73fac0affc8a',1,'binary_ops.h']]], + ['default_5fbool_5fop_2',['DEFAULT_BOOL_OP',['../cpu_2binary__ops_8h.html#a98fb0ee78e9fd26ad6d0a2b365f7caa7',1,'binary_ops.h']]], + ['default_5fcomparisons_3',['DEFAULT_COMPARISONS',['../base__simd_8h.html#af5153f51cdd18d56d1de24b5da9f4366',1,'base_simd.h']]], + ['default_5fop_4',['DEFAULT_OP',['../cpu_2unary__ops_8h.html#a83702f31e6dbd79c339a6aad67319f64',1,'unary_ops.h']]], + ['default_5freduction_5',['DEFAULT_REDUCTION',['../base__simd_8h.html#a0adf6d25084019eff671abc59031573e',1,'base_simd.h']]], + ['default_5funary_6',['DEFAULT_UNARY',['../base__simd_8h.html#a122d6a2fa4dcfe78b221e52155419124',1,'base_simd.h']]], + ['define_5fdefault_5fis_5fequivalent_7',['DEFINE_DEFAULT_IS_EQUIVALENT',['../primitives_8h.html#a0fb9d19207dc4869aca35abfbdf4d70a',1,'primitives.h']]], + ['define_5fgrads_8',['DEFINE_GRADS',['../primitives_8h.html#a77abdcb55bc2eb0f9a45edc5ee639bf6',1,'primitives.h']]], + ['define_5finput_5foutput_5fshape_9',['DEFINE_INPUT_OUTPUT_SHAPE',['../primitives_8h.html#a649a06267b75e007224ea4ddefedb999',1,'primitives.h']]], + ['define_5fneon_5fbinary_5fop_10',['DEFINE_NEON_BINARY_OP',['../neon__fp16__simd_8h.html#a63ba6e115ef14e134081310f7d4579a5',1,'neon_fp16_simd.h']]], + ['define_5fneon_5fcomparison_11',['DEFINE_NEON_COMPARISON',['../neon__fp16__simd_8h.html#a9fcd94c7369a6b4437f9c310a805c79d',1,'neon_fp16_simd.h']]], + ['define_5fneon_5funary_5fop_12',['DEFINE_NEON_UNARY_OP',['../neon__fp16__simd_8h.html#af8138a463be93b9e0c9b685e94a1fd00',1,'neon_fp16_simd.h']]], + ['define_5fprint_13',['DEFINE_PRINT',['../primitives_8h.html#a1d3a37af519e16f6a703b1e9ebd0f592',1,'primitives.h']]], + ['define_5fsimd_5fexclusive_5fscan_14',['DEFINE_SIMD_EXCLUSIVE_SCAN',['../scan_8h.html#a185f66aac8c5317587e6abd43f3013fc',1,'scan.h']]], + ['define_5fsimd_5freduce_15',['DEFINE_SIMD_REDUCE',['../backend_2metal_2kernels_2reduction_2ops_8h.html#acacf99e0ba629ed062ccc3c2eba89b05',1,'ops.h']]], + ['define_5fsimd_5fscan_16',['DEFINE_SIMD_SCAN',['../scan_8h.html#a0d8d6a9b0f3a1263629380bda8eca7bc',1,'scan.h']]], + ['define_5fvmap_17',['DEFINE_VMAP',['../primitives_8h.html#adc0fbd79fe0d1114dc85da4ed99798bd',1,'primitives.h']]] ]; diff --git a/docs/build/html/search/defines_9.js b/docs/build/html/search/defines_9.js index 09218ba2b..2d664ab68 100644 --- a/docs/build/html/search/defines_9.js +++ b/docs/build/html/search/defines_9.js @@ -3,8 +3,9 @@ var searchData= ['max_5foutput_5fsize_0',['MAX_OUTPUT_SIZE',['../backend_2metal_2kernels_2fft_8h.html#a28d683cf067736d76f867f30c066317e',1,'fft.h']]], ['max_5fradix_1',['MAX_RADIX',['../backend_2metal_2kernels_2fft_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: fft.h'],['../readwrite_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: readwrite.h']]], ['mlx_5flapack_5ffunc_2',['MLX_LAPACK_FUNC',['../lapack_8h.html#ae22db9704827bf013a0a61f21a47464b',1,'lapack.h']]], - ['mlx_5fmtl_5fconst_3',['MLX_MTL_CONST',['../kernels_2gemv__masked_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: gemv_masked.h'],['../quantized_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: quantized.h'],['../sort_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: sort.h']]], + ['mlx_5fmtl_5fconst_3',['MLX_MTL_CONST',['../gemv__masked_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: gemv_masked.h'],['../quantized_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: quantized.h'],['../sort_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: sort.h']]], ['mlx_5fmtl_5floop_5funroll_4',['MLX_MTL_LOOP_UNROLL',['../sort_8h.html#ad34b622323cebef136669fedd7229515',1,'sort.h']]], - ['mlx_5fmtl_5fpragma_5funroll_5',['MLX_MTL_PRAGMA_UNROLL',['../kernels_2gemv__masked_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: gemv_masked.h'],['../backend_2metal_2kernels_2utils_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: utils.h']]], - ['mtl_5fconst_6',['MTL_CONST',['../defines_8h.html#a767ed9f2604de22b259cee02c4ce1d22',1,'defines.h']]] + ['mlx_5fmtl_5fpragma_5funroll_5',['MLX_MTL_PRAGMA_UNROLL',['../gemv__masked_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: gemv_masked.h'],['../backend_2metal_2kernels_2utils_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: utils.h']]], + ['mlx_5fsimd_5flibrary_5fversion_6',['MLX_SIMD_LIBRARY_VERSION',['../accelerate__simd_8h.html#a628420ebe2f37d352c8c4b0dd0db59d0',1,'accelerate_simd.h']]], + ['mtl_5fconst_7',['MTL_CONST',['../defines_8h.html#a767ed9f2604de22b259cee02c4ce1d22',1,'defines.h']]] ]; diff --git a/docs/build/html/search/defines_d.js b/docs/build/html/search/defines_d.js index 3a705c846..cc48f7946 100644 --- a/docs/build/html/search/defines_d.js +++ b/docs/build/html/search/defines_d.js @@ -1,5 +1,11 @@ var searchData= [ - ['steel_5fconst_0',['STEEL_CONST',['../steel_2defines_8h.html#a90b91c866313ffa46eff6d9cc944ad2b',1,'defines.h']]], - ['steel_5fpragma_5funroll_1',['STEEL_PRAGMA_UNROLL',['../steel_2defines_8h.html#a5a5c3095b132a7589bc19cd5cb80e2c6',1,'defines.h']]] + ['simd_5fdefault_5fbinary_0',['SIMD_DEFAULT_BINARY',['../accelerate__simd_8h.html#a9aa795d90ddc485f24cd4a5268fe0e82',1,'accelerate_simd.h']]], + ['simd_5fdefault_5fcomparisons_1',['SIMD_DEFAULT_COMPARISONS',['../accelerate__simd_8h.html#aca22aa431d399cea13c969926689b459',1,'accelerate_simd.h']]], + ['simd_5fdefault_5funary_2',['SIMD_DEFAULT_UNARY',['../accelerate__simd_8h.html#a3b449b9ce2c623ab4dce3f6fe349bb6d',1,'accelerate_simd.h']]], + ['simd_5ffp16_5fdefault_5fbinary_3',['SIMD_FP16_DEFAULT_BINARY',['../accelerate__fp16__simd_8h.html#a6437645d67e6a8b7c1349b0040aaa6bb',1,'accelerate_fp16_simd.h']]], + ['simd_5ffp16_5fdefault_5funary_4',['SIMD_FP16_DEFAULT_UNARY',['../accelerate__fp16__simd_8h.html#a60af883c0392ecab270ecc1146f5e8a2',1,'accelerate_fp16_simd.h']]], + ['single_5',['SINGLE',['../cpu_2unary__ops_8h.html#a602aea95990389a45c255195f849d5de',1,'unary_ops.h']]], + ['steel_5fconst_6',['STEEL_CONST',['../steel_2defines_8h.html#a90b91c866313ffa46eff6d9cc944ad2b',1,'defines.h']]], + ['steel_5fpragma_5funroll_7',['STEEL_PRAGMA_UNROLL',['../steel_2defines_8h.html#a5a5c3095b132a7589bc19cd5cb80e2c6',1,'defines.h']]] ]; diff --git a/docs/build/html/search/enums_0.js b/docs/build/html/search/enums_0.js index ce385f300..889fa39a5 100644 --- a/docs/build/html/search/enums_0.js +++ b/docs/build/html/search/enums_0.js @@ -1,4 +1,5 @@ var searchData= [ - ['base_0',['Base',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421',1,'mlx::core::Log']]] + ['base_0',['Base',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421',1,'mlx::core::Log']]], + ['binaryoptype_1',['BinaryOpType',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/enums_5.js b/docs/build/html/search/enums_5.js index c1bd77374..7eba5a656 100644 --- a/docs/build/html/search/enums_5.js +++ b/docs/build/html/search/enums_5.js @@ -1,5 +1,5 @@ var searchData= [ - ['reducetype_0',['ReduceType',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924',1,'mlx::core::distributed::AllReduce::ReduceType'],['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3',1,'mlx::core::ArgReduce::ReduceType'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9',1,'mlx::core::Reduce::ReduceType'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1',1,'mlx::core::Scan::ReduceType'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613c',1,'mlx::core::Scatter::ReduceType']]], + ['reducetype_0',['ReduceType',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924',1,'mlx::core::distributed::AllReduce::ReduceType'],['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3',1,'mlx::core::ArgReduce::ReduceType'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9',1,'mlx::core::Reduce::ReduceType'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1',1,'mlx::core::Scan::ReduceType'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613c',1,'mlx::core::Scatter::ReduceType'],['../classmlx_1_1core_1_1_scatter_axis.html#aa292e6cb2a4b32c42ad4f7a258b334f2',1,'mlx::core::ScatterAxis::ReduceType']]], ['reductionoptype_1',['ReductionOpType',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/enums_7.js b/docs/build/html/search/enums_7.js index 6a3fc021b..df1d54a58 100644 --- a/docs/build/html/search/enums_7.js +++ b/docs/build/html/search/enums_7.js @@ -1,4 +1,4 @@ var searchData= [ - ['val_0',['Val',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1da',1,'mlx::core::Dtype']]] + ['ternaryoptype_0',['TernaryOpType',['../namespacemlx_1_1core.html#ac2b8997537c7f25dd2b244d4c0a865a1',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/enums_8.js b/docs/build/html/search/enums_8.js new file mode 100644 index 000000000..6a3fc021b --- /dev/null +++ b/docs/build/html/search/enums_8.js @@ -0,0 +1,4 @@ +var searchData= +[ + ['val_0',['Val',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1da',1,'mlx::core::Dtype']]] +]; diff --git a/docs/build/html/search/enumvalues_11.js b/docs/build/html/search/enumvalues_11.js index c5c21816a..43d5939d1 100644 --- a/docs/build/html/search/enumvalues_11.js +++ b/docs/build/html/search/enumvalues_11.js @@ -1,5 +1,8 @@ var searchData= [ ['v_0',['V',['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a5206560a306a2e085a437fd258eb57ce',1,'mlx::core::Dtype']]], - ['vector_1',['Vector',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a57dea6f5039281b7fee517fc43bf3110',1,'mlx::core']]] + ['vector_1',['Vector',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a57dea6f5039281b7fee517fc43bf3110',1,'mlx::core']]], + ['vectorscalar_2',['VectorScalar',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6a8a94416459b638cebf3bfbce26a6ce78',1,'mlx::core']]], + ['vectorvector_3',['VectorVector',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6a4d8269410dcd9cadc9722e9a118bddfb',1,'mlx::core']]], + ['vectorvectorvector_4',['VectorVectorVector',['../namespacemlx_1_1core.html#ac2b8997537c7f25dd2b244d4c0a865a1acbcaeeb0e232871afe48bcf063a14b42',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/enumvalues_6.js b/docs/build/html/search/enumvalues_6.js index 5deb1fa6e..cabbf83d3 100644 --- a/docs/build/html/search/enumvalues_6.js +++ b/docs/build/html/search/enumvalues_6.js @@ -1,6 +1,6 @@ var searchData= [ - ['general_0',['General',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a0db377921f4ce762c62526131097968f',1,'mlx::core']]], + ['general_0',['General',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6a0db377921f4ce762c62526131097968f',1,'mlx::core::General'],['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a0db377921f4ce762c62526131097968f',1,'mlx::core::General'],['../namespacemlx_1_1core.html#ac2b8997537c7f25dd2b244d4c0a865a1a0db377921f4ce762c62526131097968f',1,'mlx::core::General']]], ['generalcontiguousreduce_1',['GeneralContiguousReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a540cf31fe6858115a02e789938297cdb',1,'mlx::core']]], ['generalgeneral_2',['GeneralGeneral',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a6fe62e8ce1fae1e70cb9eeaa67d29dab',1,'mlx::core']]], ['generalreduce_3',['GeneralReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a3622f95ed0ec99657f9ad8ef39ec2184',1,'mlx::core']]], diff --git a/docs/build/html/search/enumvalues_a.js b/docs/build/html/search/enumvalues_a.js index bcafee1d6..7e7c9547d 100644 --- a/docs/build/html/search/enumvalues_a.js +++ b/docs/build/html/search/enumvalues_a.js @@ -2,6 +2,6 @@ var searchData= [ ['no_5ffuse_0',['no_fuse',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4ada8df7fd43da6073fec4fe5666b03dbb',1,'mlx::core']]], ['no_5fsimplify_1',['no_simplify',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a8e5611dfddbae6e68624c59aa3e4e3e2',1,'mlx::core']]], - ['none_2',['None',['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca87a6a1927de175b71d7d0b5c11b8665c',1,'mlx::core::Scatter']]], + ['none_2',['None',['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca87a6a1927de175b71d7d0b5c11b8665c',1,'mlx::core::Scatter::None'],['../classmlx_1_1core_1_1_scatter_axis.html#aa292e6cb2a4b32c42ad4f7a258b334f2a93146c4280504f1f67459e6ae0d25c38',1,'mlx::core::ScatterAxis::None']]], ['number_3',['number',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2dab1bc248a7ff2b2e95569f56de68615df',1,'mlx::core::Dtype']]] ]; diff --git a/docs/build/html/search/enumvalues_e.js b/docs/build/html/search/enumvalues_e.js index 6a250e52f..5885b8201 100644 --- a/docs/build/html/search/enumvalues_e.js +++ b/docs/build/html/search/enumvalues_e.js @@ -1,7 +1,10 @@ var searchData= [ ['scalar_0',['Scalar',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337af60357a8d17e45793298323f1b372a74',1,'mlx::core']]], - ['scheduled_1',['scheduled',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078af8a6f8eed2395ab89a758dec434393ae',1,'mlx::core::array']]], - ['signedinteger_2',['signedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2daed58b4631ff157bec9e35ed1182d2c10',1,'mlx::core::Dtype']]], - ['sum_3',['Sum',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a1fc7c1f09c80650ab0497e2d6781d65f',1,'mlx::core::distributed::AllReduce::Sum'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a8582875544f1d3d396a1a376473ef1dd',1,'mlx::core::Reduce::Sum'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ade23893033e4849f5596e7ce76a5fc36',1,'mlx::core::Scan::Sum'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca14abe2d8818efa71726be4e156813d6f',1,'mlx::core::Scatter::Sum']]] + ['scalarscalar_1',['ScalarScalar',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6a7b15cb76e0535ea81a5b6af9c96dcde4',1,'mlx::core']]], + ['scalarscalarscalar_2',['ScalarScalarScalar',['../namespacemlx_1_1core.html#ac2b8997537c7f25dd2b244d4c0a865a1a09c2e68746fa22c9903625cea17464db',1,'mlx::core']]], + ['scalarvector_3',['ScalarVector',['../namespacemlx_1_1core.html#a546e3d3c8957fbf2758f9504f4a2d0b6aabac63719294588466e3c2f00cccb0a6',1,'mlx::core']]], + ['scheduled_4',['scheduled',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078af8a6f8eed2395ab89a758dec434393ae',1,'mlx::core::array']]], + ['signedinteger_5',['signedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2daed58b4631ff157bec9e35ed1182d2c10',1,'mlx::core::Dtype']]], + ['sum_6',['Sum',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a1fc7c1f09c80650ab0497e2d6781d65f',1,'mlx::core::distributed::AllReduce::Sum'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a8582875544f1d3d396a1a376473ef1dd',1,'mlx::core::Reduce::Sum'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ade23893033e4849f5596e7ce76a5fc36',1,'mlx::core::Scan::Sum'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca14abe2d8818efa71726be4e156813d6f',1,'mlx::core::Scatter::Sum'],['../classmlx_1_1core_1_1_scatter_axis.html#aa292e6cb2a4b32c42ad4f7a258b334f2a702b8cfdaf7fe3e063873595ff0508f2',1,'mlx::core::ScatterAxis::Sum']]] ]; diff --git a/docs/build/html/search/files_0.js b/docs/build/html/search/files_0.js index 8570df320..0a58a104b 100644 --- a/docs/build/html/search/files_0.js +++ b/docs/build/html/search/files_0.js @@ -1,8 +1,10 @@ var searchData= [ - ['allocator_2eh_0',['allocator.h',['../allocator_8h.html',1,'(Global Namespace)'],['../backend_2metal_2allocator_8h.html',1,'(Global Namespace)']]], - ['arange_2eh_1',['arange.h',['../common_2arange_8h.html',1,'(Global Namespace)'],['../metal_2jit_2arange_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2arange_8h.html',1,'(Global Namespace)']]], - ['array_2eh_2',['array.h',['../array_8h.html',1,'']]], - ['atomic_2eh_3',['atomic.h',['../atomic_8h.html',1,'']]], - ['attn_2eh_4',['attn.h',['../attn_8h.html',1,'']]] + ['accelerate_5ffp16_5fsimd_2eh_0',['accelerate_fp16_simd.h',['../accelerate__fp16__simd_8h.html',1,'']]], + ['accelerate_5fsimd_2eh_1',['accelerate_simd.h',['../accelerate__simd_8h.html',1,'']]], + ['allocator_2eh_2',['allocator.h',['../allocator_8h.html',1,'(Global Namespace)'],['../backend_2metal_2allocator_8h.html',1,'(Global Namespace)']]], + ['arange_2eh_3',['arange.h',['../cpu_2arange_8h.html',1,'(Global Namespace)'],['../metal_2jit_2arange_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2arange_8h.html',1,'(Global Namespace)']]], + ['array_2eh_4',['array.h',['../array_8h.html',1,'']]], + ['atomic_2eh_5',['atomic.h',['../atomic_8h.html',1,'']]], + ['attn_2eh_6',['attn.h',['../attn_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/files_1.js b/docs/build/html/search/files_1.js index af8aecb92..01eadf3d4 100644 --- a/docs/build/html/search/files_1.js +++ b/docs/build/html/search/files_1.js @@ -1,8 +1,9 @@ var searchData= [ - ['bf16_2eh_0',['bf16.h',['../backend_2metal_2kernels_2jit_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html',1,'(Global Namespace)'],['../types_2bf16_8h.html',1,'(Global Namespace)']]], - ['bf16_5fmath_2eh_1',['bf16_math.h',['../bf16__math_8h.html',1,'']]], - ['binary_2eh_2',['binary.h',['../common_2binary_8h.html',1,'(Global Namespace)'],['../metal_2binary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary_8h.html',1,'(Global Namespace)']]], - ['binary_5fops_2eh_3',['binary_ops.h',['../binary__ops_8h.html',1,'']]], - ['binary_5ftwo_2eh_4',['binary_two.h',['../common_2binary__two_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__two_8h.html',1,'(Global Namespace)']]] + ['base_5fsimd_2eh_0',['base_simd.h',['../base__simd_8h.html',1,'']]], + ['bf16_2eh_1',['bf16.h',['../backend_2metal_2kernels_2jit_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html',1,'(Global Namespace)'],['../types_2bf16_8h.html',1,'(Global Namespace)']]], + ['bf16_5fmath_2eh_2',['bf16_math.h',['../bf16__math_8h.html',1,'']]], + ['binary_2eh_3',['binary.h',['../common_2binary_8h.html',1,'(Global Namespace)'],['../cpu_2binary_8h.html',1,'(Global Namespace)'],['../metal_2binary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary_8h.html',1,'(Global Namespace)']]], + ['binary_5fops_2eh_4',['binary_ops.h',['../cpu_2binary__ops_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__ops_8h.html',1,'(Global Namespace)']]], + ['binary_5ftwo_2eh_5',['binary_two.h',['../cpu_2binary__two_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__two_8h.html',1,'(Global Namespace)']]] ]; diff --git a/docs/build/html/search/files_10.js b/docs/build/html/search/files_10.js index 84e339292..9dd3bc541 100644 --- a/docs/build/html/search/files_10.js +++ b/docs/build/html/search/files_10.js @@ -1,13 +1,4 @@ var searchData= [ - ['radix_2eh_0',['radix.h',['../radix_8h.html',1,'']]], - ['random_2eh_1',['random.h',['../random_8h.html',1,'']]], - ['readwrite_2eh_2',['readwrite.h',['../readwrite_8h.html',1,'']]], - ['reduce_2eh_3',['reduce.h',['../common_2reduce_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2reduce_8h.html',1,'(Global Namespace)'],['../metal_2reduce_8h.html',1,'(Global Namespace)']]], - ['reduce_5fall_2eh_4',['reduce_all.h',['../reduce__all_8h.html',1,'']]], - ['reduce_5fcol_2eh_5',['reduce_col.h',['../reduce__col_8h.html',1,'']]], - ['reduce_5finit_2eh_6',['reduce_init.h',['../reduce__init_8h.html',1,'']]], - ['reduce_5frow_2eh_7',['reduce_row.h',['../reduce__row_8h.html',1,'']]], - ['reduce_5futils_2eh_8',['reduce_utils.h',['../reduce__utils_8h.html',1,'']]], - ['resident_2eh_9',['resident.h',['../resident_8h.html',1,'']]] + ['quantized_2eh_0',['quantized.h',['../quantized_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/files_11.js b/docs/build/html/search/files_11.js index 01964bcf5..92f5de768 100644 --- a/docs/build/html/search/files_11.js +++ b/docs/build/html/search/files_11.js @@ -1,18 +1,14 @@ var searchData= [ - ['scan_2eh_0',['scan.h',['../scan_8h.html',1,'']]], - ['scatter_2eh_1',['scatter.h',['../scatter_8h.html',1,'']]], - ['scheduler_2eh_2',['scheduler.h',['../scheduler_8h.html',1,'']]], - ['sdpa_5fvector_2eh_3',['sdpa_vector.h',['../sdpa__vector_8h.html',1,'']]], - ['slicing_2eh_4',['slicing.h',['../common_2slicing_8h.html',1,'(Global Namespace)'],['../metal_2slicing_8h.html',1,'(Global Namespace)']]], - ['softmax_2eh_5',['softmax.h',['../jit_2softmax_8h.html',1,'(Global Namespace)'],['../kernels_2softmax_8h.html',1,'(Global Namespace)']]], - ['sort_2eh_6',['sort.h',['../sort_8h.html',1,'']]], - ['steel_5fattention_2eh_7',['steel_attention.h',['../steel__attention_8h.html',1,'']]], - ['steel_5fconv_2eh_8',['steel_conv.h',['../jit_2steel__conv_8h.html',1,'(Global Namespace)'],['../kernels_2steel_2conv_2kernels_2steel__conv_8h.html',1,'(Global Namespace)']]], - ['steel_5fconv_5fgeneral_2eh_9',['steel_conv_general.h',['../steel__conv__general_8h.html',1,'']]], - ['steel_5fgemm_2eh_10',['steel_gemm.h',['../steel__gemm_8h.html',1,'']]], - ['steel_5fgemm_5ffused_2eh_11',['steel_gemm_fused.h',['../steel__gemm__fused_8h.html',1,'']]], - ['steel_5fgemm_5fmasked_2eh_12',['steel_gemm_masked.h',['../steel__gemm__masked_8h.html',1,'']]], - ['steel_5fgemm_5fsplitk_2eh_13',['steel_gemm_splitk.h',['../steel__gemm__splitk_8h.html',1,'']]], - ['stream_2eh_14',['stream.h',['../stream_8h.html',1,'']]] + ['radix_2eh_0',['radix.h',['../radix_8h.html',1,'']]], + ['random_2eh_1',['random.h',['../random_8h.html',1,'']]], + ['readwrite_2eh_2',['readwrite.h',['../readwrite_8h.html',1,'']]], + ['reduce_2eh_3',['reduce.h',['../common_2reduce_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2reduce_8h.html',1,'(Global Namespace)'],['../metal_2reduce_8h.html',1,'(Global Namespace)']]], + ['reduce_5fall_2eh_4',['reduce_all.h',['../reduce__all_8h.html',1,'']]], + ['reduce_5fcol_2eh_5',['reduce_col.h',['../reduce__col_8h.html',1,'']]], + ['reduce_5finit_2eh_6',['reduce_init.h',['../reduce__init_8h.html',1,'']]], + ['reduce_5frow_2eh_7',['reduce_row.h',['../reduce__row_8h.html',1,'']]], + ['reduce_5futils_2eh_8',['reduce_utils.h',['../reduce__utils_8h.html',1,'']]], + ['resident_2eh_9',['resident.h',['../resident_8h.html',1,'']]], + ['ring_2eh_10',['ring.h',['../ring_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/files_12.js b/docs/build/html/search/files_12.js index b7521b551..596e2f052 100644 --- a/docs/build/html/search/files_12.js +++ b/docs/build/html/search/files_12.js @@ -1,10 +1,19 @@ var searchData= [ - ['ternary_2eh_0',['ternary.h',['../common_2ternary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2ternary_8h.html',1,'(Global Namespace)'],['../metal_2ternary_8h.html',1,'(Global Namespace)']]], - ['ternary_5fops_2eh_1',['ternary_ops.h',['../ternary__ops_8h.html',1,'']]], - ['threadpool_2eh_2',['threadpool.h',['../threadpool_8h.html',1,'']]], - ['threefry_2eh_3',['threefry.h',['../threefry_8h.html',1,'']]], - ['transforms_2eh_4',['transforms.h',['../backend_2metal_2kernels_2steel_2attn_2transforms_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2gemm_2transforms_8h.html',1,'(Global Namespace)'],['../transforms_8h.html',1,'(Global Namespace)']]], - ['transforms_5fimpl_2eh_5',['transforms_impl.h',['../transforms__impl_8h.html',1,'']]], - ['type_5ftraits_2eh_6',['type_traits.h',['../type__traits_8h.html',1,'']]] + ['scan_2eh_0',['scan.h',['../scan_8h.html',1,'']]], + ['scatter_2eh_1',['scatter.h',['../scatter_8h.html',1,'']]], + ['scatter_5faxis_2eh_2',['scatter_axis.h',['../scatter__axis_8h.html',1,'']]], + ['scheduler_2eh_3',['scheduler.h',['../scheduler_8h.html',1,'']]], + ['sdpa_5fvector_2eh_4',['sdpa_vector.h',['../sdpa__vector_8h.html',1,'']]], + ['simd_2eh_5',['simd.h',['../simd_8h.html',1,'']]], + ['slicing_2eh_6',['slicing.h',['../common_2slicing_8h.html',1,'(Global Namespace)'],['../cpu_2slicing_8h.html',1,'(Global Namespace)'],['../metal_2slicing_8h.html',1,'(Global Namespace)']]], + ['softmax_2eh_7',['softmax.h',['../jit_2softmax_8h.html',1,'(Global Namespace)'],['../kernels_2softmax_8h.html',1,'(Global Namespace)']]], + ['sort_2eh_8',['sort.h',['../sort_8h.html',1,'']]], + ['steel_5fattention_2eh_9',['steel_attention.h',['../steel__attention_8h.html',1,'']]], + ['steel_5fconv_2eh_10',['steel_conv.h',['../steel__conv_8h.html',1,'']]], + ['steel_5fconv_5fgeneral_2eh_11',['steel_conv_general.h',['../steel__conv__general_8h.html',1,'']]], + ['steel_5fgemm_5ffused_2eh_12',['steel_gemm_fused.h',['../steel__gemm__fused_8h.html',1,'']]], + ['steel_5fgemm_5fmasked_2eh_13',['steel_gemm_masked.h',['../steel__gemm__masked_8h.html',1,'']]], + ['steel_5fgemm_5fsplitk_2eh_14',['steel_gemm_splitk.h',['../steel__gemm__splitk_8h.html',1,'']]], + ['stream_2eh_15',['stream.h',['../stream_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/files_13.js b/docs/build/html/search/files_13.js index 3cd8eae9d..6a1303cd2 100644 --- a/docs/build/html/search/files_13.js +++ b/docs/build/html/search/files_13.js @@ -1,6 +1,11 @@ var searchData= [ - ['unary_2eh_0',['unary.h',['../common_2unary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2unary_8h.html',1,'(Global Namespace)'],['../metal_2unary_8h.html',1,'(Global Namespace)']]], - ['unary_5fops_2eh_1',['unary_ops.h',['../unary__ops_8h.html',1,'']]], - ['utils_2eh_2',['utils.h',['../backend_2accelerate_2utils_8h.html',1,'(Global Namespace)'],['../backend_2common_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2utils_8h.html',1,'(Global Namespace)'],['../utils_8h.html',1,'(Global Namespace)']]] + ['ternary_2eh_0',['ternary.h',['../common_2ternary_8h.html',1,'(Global Namespace)'],['../cpu_2ternary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2ternary_8h.html',1,'(Global Namespace)'],['../metal_2ternary_8h.html',1,'(Global Namespace)']]], + ['ternary_5fops_2eh_1',['ternary_ops.h',['../ternary__ops_8h.html',1,'']]], + ['threadpool_2eh_2',['threadpool.h',['../threadpool_8h.html',1,'']]], + ['threefry_2eh_3',['threefry.h',['../threefry_8h.html',1,'']]], + ['transforms_2eh_4',['transforms.h',['../backend_2metal_2kernels_2steel_2attn_2transforms_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2gemm_2transforms_8h.html',1,'(Global Namespace)'],['../transforms_8h.html',1,'(Global Namespace)']]], + ['transforms_5fimpl_2eh_5',['transforms_impl.h',['../transforms__impl_8h.html',1,'']]], + ['type_2eh_6',['type.h',['../type_8h.html',1,'']]], + ['type_5ftraits_2eh_7',['type_traits.h',['../type__traits_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/files_14.js b/docs/build/html/search/files_14.js new file mode 100644 index 000000000..6cd9cefc3 --- /dev/null +++ b/docs/build/html/search/files_14.js @@ -0,0 +1,6 @@ +var searchData= +[ + ['unary_2eh_0',['unary.h',['../cpu_2unary_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2unary_8h.html',1,'(Global Namespace)'],['../metal_2unary_8h.html',1,'(Global Namespace)']]], + ['unary_5fops_2eh_1',['unary_ops.h',['../cpu_2unary__ops_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2unary__ops_8h.html',1,'(Global Namespace)']]], + ['utils_2eh_2',['utils.h',['../backend_2common_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2utils_8h.html',1,'(Global Namespace)'],['../utils_8h.html',1,'(Global Namespace)']]] +]; diff --git a/docs/build/html/search/files_2.js b/docs/build/html/search/files_2.js index 6b69718f7..c82020c67 100644 --- a/docs/build/html/search/files_2.js +++ b/docs/build/html/search/files_2.js @@ -6,5 +6,5 @@ var searchData= ['compiled_5fpreamble_2eh_3',['compiled_preamble.h',['../compiled__preamble_8h.html',1,'']]], ['complex_2eh_4',['complex.h',['../backend_2metal_2kernels_2complex_8h.html',1,'(Global Namespace)'],['../types_2complex_8h.html',1,'(Global Namespace)']]], ['conv_2eh_5',['conv.h',['../conv_8h.html',1,'']]], - ['copy_2eh_6',['copy.h',['../common_2copy_8h.html',1,'(Global Namespace)'],['../metal_2copy_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2copy_8h.html',1,'(Global Namespace)']]] + ['copy_2eh_6',['copy.h',['../common_2copy_8h.html',1,'(Global Namespace)'],['../cpu_2copy_8h.html',1,'(Global Namespace)'],['../metal_2copy_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2copy_8h.html',1,'(Global Namespace)']]] ]; diff --git a/docs/build/html/search/files_4.js b/docs/build/html/search/files_4.js index 2c567202c..67039b83b 100644 --- a/docs/build/html/search/files_4.js +++ b/docs/build/html/search/files_4.js @@ -2,7 +2,7 @@ var searchData= [ ['einsum_2eh_0',['einsum.h',['../einsum_8h.html',1,'']]], ['erf_2eh_1',['erf.h',['../erf_8h.html',1,'']]], - ['event_2eh_2',['event.h',['../event_8h.html',1,'']]], + ['event_2eh_2',['event.h',['../backend_2metal_2event_8h.html',1,'(Global Namespace)'],['../event_8h.html',1,'(Global Namespace)']]], ['expm1f_2eh_3',['expm1f.h',['../expm1f_8h.html',1,'']]], ['export_2eh_4',['export.h',['../export_8h.html',1,'']]], ['export_5fimpl_2eh_5',['export_impl.h',['../export__impl_8h.html',1,'']]] diff --git a/docs/build/html/search/files_5.js b/docs/build/html/search/files_5.js index d25e2e8c3..288ad793a 100644 --- a/docs/build/html/search/files_5.js +++ b/docs/build/html/search/files_5.js @@ -2,6 +2,7 @@ var searchData= [ ['fast_2eh_0',['fast.h',['../fast_8h.html',1,'']]], ['fast_5fprimitives_2eh_1',['fast_primitives.h',['../fast__primitives_8h.html',1,'']]], - ['fft_2eh_2',['fft.h',['../backend_2metal_2kernels_2fft_8h.html',1,'(Global Namespace)'],['../fft_8h.html',1,'(Global Namespace)']]], - ['fp16_2eh_3',['fp16.h',['../fp16_8h.html',1,'']]] + ['fence_2eh_2',['fence.h',['../fence_8h.html',1,'']]], + ['fft_2eh_3',['fft.h',['../backend_2metal_2kernels_2fft_8h.html',1,'(Global Namespace)'],['../fft_8h.html',1,'(Global Namespace)']]], + ['fp16_2eh_4',['fp16.h',['../fp16_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/files_6.js b/docs/build/html/search/files_6.js index d246684e7..22209f447 100644 --- a/docs/build/html/search/files_6.js +++ b/docs/build/html/search/files_6.js @@ -1,8 +1,9 @@ var searchData= [ ['gather_2eh_0',['gather.h',['../gather_8h.html',1,'']]], - ['gemm_2eh_1',['gemm.h',['../gemm_8h.html',1,'']]], - ['gemv_5fmasked_2eh_2',['gemv_masked.h',['../jit_2gemv__masked_8h.html',1,'(Global Namespace)'],['../kernels_2gemv__masked_8h.html',1,'(Global Namespace)']]], - ['gguf_2eh_3',['gguf.h',['../gguf_8h.html',1,'']]], - ['graph_5futils_2eh_4',['graph_utils.h',['../graph__utils_8h.html',1,'']]] + ['gather_5faxis_2eh_1',['gather_axis.h',['../gather__axis_8h.html',1,'']]], + ['gemm_2eh_2',['gemm.h',['../cpu_2gemm_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2steel_2gemm_2gemm_8h.html',1,'(Global Namespace)']]], + ['gemv_5fmasked_2eh_3',['gemv_masked.h',['../gemv__masked_8h.html',1,'']]], + ['gguf_2eh_4',['gguf.h',['../gguf_8h.html',1,'']]], + ['graph_5futils_2eh_5',['graph_utils.h',['../graph__utils_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/files_c.js b/docs/build/html/search/files_c.js index 15bfbc3a6..df1295c6a 100644 --- a/docs/build/html/search/files_c.js +++ b/docs/build/html/search/files_c.js @@ -1,9 +1,10 @@ var searchData= [ - ['matmul_2eh_0',['matmul.h',['../matmul_8h.html',1,'']]], - ['metal_2eh_1',['metal.h',['../metal_8h.html',1,'']]], - ['metal_5fimpl_2eh_2',['metal_impl.h',['../metal__impl_8h.html',1,'']]], - ['mlx_2eh_3',['mlx.h',['../mlx_8h.html',1,'']]], - ['mma_2eh_4',['mma.h',['../attn_2mma_8h.html',1,'(Global Namespace)'],['../gemm_2mma_8h.html',1,'(Global Namespace)']]], - ['mpi_2eh_5',['mpi.h',['../mpi_8h.html',1,'']]] + ['math_2eh_0',['math.h',['../math_8h.html',1,'']]], + ['matmul_2eh_1',['matmul.h',['../matmul_8h.html',1,'']]], + ['metal_2eh_2',['metal.h',['../metal_8h.html',1,'']]], + ['metal_5fimpl_2eh_3',['metal_impl.h',['../metal__impl_8h.html',1,'']]], + ['mlx_2eh_4',['mlx.h',['../mlx_8h.html',1,'']]], + ['mma_2eh_5',['mma.h',['../attn_2mma_8h.html',1,'(Global Namespace)'],['../gemm_2mma_8h.html',1,'(Global Namespace)']]], + ['mpi_2eh_6',['mpi.h',['../mpi_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/files_d.js b/docs/build/html/search/files_d.js index 1679dc0b8..20cd218c9 100644 --- a/docs/build/html/search/files_d.js +++ b/docs/build/html/search/files_d.js @@ -1,4 +1,4 @@ var searchData= [ - ['ops_2eh_0',['ops.h',['../backend_2common_2ops_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2reduction_2ops_8h.html',1,'(Global Namespace)'],['../distributed_2ops_8h.html',1,'(Global Namespace)'],['../ops_8h.html',1,'(Global Namespace)']]] + ['neon_5ffp16_5fsimd_2eh_0',['neon_fp16_simd.h',['../neon__fp16__simd_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/files_e.js b/docs/build/html/search/files_e.js index 0ea2bf8fb..9ba5f4fa7 100644 --- a/docs/build/html/search/files_e.js +++ b/docs/build/html/search/files_e.js @@ -1,6 +1,4 @@ var searchData= [ - ['params_2eh_0',['params.h',['../attn_2params_8h.html',1,'(Global Namespace)'],['../conv_2params_8h.html',1,'(Global Namespace)'],['../gemm_2params_8h.html',1,'(Global Namespace)']]], - ['pocketfft_2eh_1',['pocketfft.h',['../pocketfft_8h.html',1,'']]], - ['primitives_2eh_2',['primitives.h',['../distributed_2primitives_8h.html',1,'(Global Namespace)'],['../primitives_8h.html',1,'(Global Namespace)']]] + ['ops_2eh_0',['ops.h',['../backend_2metal_2kernels_2reduction_2ops_8h.html',1,'(Global Namespace)'],['../distributed_2ops_8h.html',1,'(Global Namespace)'],['../ops_8h.html',1,'(Global Namespace)']]] ]; diff --git a/docs/build/html/search/files_f.js b/docs/build/html/search/files_f.js index 9dd3bc541..0ea2bf8fb 100644 --- a/docs/build/html/search/files_f.js +++ b/docs/build/html/search/files_f.js @@ -1,4 +1,6 @@ var searchData= [ - ['quantized_2eh_0',['quantized.h',['../quantized_8h.html',1,'']]] + ['params_2eh_0',['params.h',['../attn_2params_8h.html',1,'(Global Namespace)'],['../conv_2params_8h.html',1,'(Global Namespace)'],['../gemm_2params_8h.html',1,'(Global Namespace)']]], + ['pocketfft_2eh_1',['pocketfft.h',['../pocketfft_8h.html',1,'']]], + ['primitives_2eh_2',['primitives.h',['../distributed_2primitives_8h.html',1,'(Global Namespace)'],['../primitives_8h.html',1,'(Global Namespace)']]] ]; diff --git a/docs/build/html/search/functions_1.js b/docs/build/html/search/functions_1.js index 05e535e06..14c519496 100644 --- a/docs/build/html/search/functions_1.js +++ b/docs/build/html/search/functions_1.js @@ -2,9 +2,9 @@ var searchData= [ ['abort_5fwith_5fexception_0',['abort_with_exception',['../namespacemlx_1_1core.html#ae1b6e5cfd27b1526285648686898e011',1,'mlx::core']]], ['abs_1',['Abs',['../classmlx_1_1core_1_1_abs.html#a1247e72feb640fb562d036b2dd1ae4ad',1,'mlx::core::Abs']]], - ['abs_2',['abs',['../namespacemetal.html#a87c5122c60f9a12afceb9925a5b78ffb',1,'metal::abs()'],['../namespacemetal_1_1fast.html#a90d2973f71f83180e7f02e38d11c7a8f',1,'metal::fast::abs()'],['../namespacemetal_1_1precise.html#a99f2b2746e813b9ca7b4249afbaf2a14',1,'metal::precise::abs()'],['../group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920',1,'mlx::core::abs()']]], - ['acos_3',['acos',['../namespacemetal.html#ad4537748b3c832b6569ff7ccb209fcb2',1,'metal::acos()'],['../namespacemetal_1_1fast.html#a805ce5c3a94b618b7349d70bbb82f0b2',1,'metal::fast::acos()'],['../namespacemetal_1_1precise.html#a8a2bcc89fc0b7e74f0453f82f89a8604',1,'metal::precise::acos()']]], - ['acosh_4',['acosh',['../namespacemetal.html#a2d0efb92b7f61eff342d776bd6c5f3a0',1,'metal::acosh()'],['../namespacemetal_1_1fast.html#afb656fc3406649a238b6f1e0509de751',1,'metal::fast::acosh()'],['../namespacemetal_1_1precise.html#a1f489fabffab969b8677b56bb1136067',1,'metal::precise::acosh()']]], + ['abs_2',['abs',['../namespacemlx_1_1core_1_1simd.html#a4f3cc8b2493586e83fd65640df3b60ad',1,'mlx::core::simd::abs(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a1958f026f26f313d17155ac87ea6eca3',1,'mlx::core::simd::abs(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a7696a0628a1c6ccb293ebd6f2328ea48',1,'mlx::core::simd::abs(Simd< float16_t, N > a)'],['../namespacemetal.html#a87c5122c60f9a12afceb9925a5b78ffb',1,'metal::abs()'],['../namespacemetal_1_1fast.html#a90d2973f71f83180e7f02e38d11c7a8f',1,'metal::fast::abs()'],['../namespacemetal_1_1precise.html#a99f2b2746e813b9ca7b4249afbaf2a14',1,'metal::precise::abs()'],['../group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920',1,'mlx::core::abs()']]], + ['acos_3',['acos',['../namespacemlx_1_1core_1_1simd.html#ae623449dfa7aab3031aa2f14c1b10a2d',1,'mlx::core::simd::acos(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a85999467c83b07e4fa5f093f7ddf19e1',1,'mlx::core::simd::acos(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a3d4f9d08d1902e3d62c6f63d39329dbd',1,'mlx::core::simd::acos(Simd< T, 1 > in)'],['../namespacemetal.html#ad4537748b3c832b6569ff7ccb209fcb2',1,'metal::acos()'],['../namespacemetal_1_1fast.html#a805ce5c3a94b618b7349d70bbb82f0b2',1,'metal::fast::acos()'],['../namespacemetal_1_1precise.html#a8a2bcc89fc0b7e74f0453f82f89a8604',1,'metal::precise::acos()']]], + ['acosh_4',['acosh',['../namespacemlx_1_1core_1_1simd.html#a4f8a64e7742fcd8f759f723a36a7c826',1,'mlx::core::simd::acosh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a90092f3826ad3be4b2b1785f7ff4a86b',1,'mlx::core::simd::acosh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a51071c8104494b5bd8097990da3bf943',1,'mlx::core::simd::acosh(Simd< T, 1 > in)'],['../namespacemetal.html#a2d0efb92b7f61eff342d776bd6c5f3a0',1,'metal::acosh()'],['../namespacemetal_1_1fast.html#afb656fc3406649a238b6f1e0509de751',1,'metal::fast::acosh()'],['../namespacemetal_1_1precise.html#a1f489fabffab969b8677b56bb1136067',1,'metal::precise::acosh()']]], ['add_5',['Add',['../classmlx_1_1core_1_1_add.html#ae3fd5483f3454eac3df256e3f5f3cdae',1,'mlx::core::Add']]], ['add_6',['add',['../group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7',1,'mlx::core']]], ['add_5ftemporaries_7',['add_temporaries',['../classmlx_1_1core_1_1metal_1_1_device.html#a72ad17c96fc6ce825bc77f0bed657901',1,'mlx::core::metal::Device']]], @@ -19,7 +19,7 @@ var searchData= ['aligned_5falloc_16',['aligned_alloc',['../namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76',1,'pocketfft::detail']]], ['aligned_5fallocator_17',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator(const aligned_allocator< U > &)'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator()=default']]], ['aligned_5fdealloc_18',['aligned_dealloc',['../namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8',1,'pocketfft::detail']]], - ['all_19',['all',['../group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68',1,'mlx::core::all(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4',1,'mlx::core::all(const array &a, StreamOrDevice s={})'],['../group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59',1,'mlx::core::all(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc',1,'mlx::core::all(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['all_19',['all',['../namespacemlx_1_1core_1_1simd.html#a5109118acb6766855878b9e8a56b156a',1,'mlx::core::simd::all(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#a4ba3690489c2bf861e22e1175255438c',1,'mlx::core::simd::all(Simd< T, 1 > x)'],['../group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68',1,'mlx::core::all(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4',1,'mlx::core::all(const array &a, StreamOrDevice s={})'],['../group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59',1,'mlx::core::all(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc',1,'mlx::core::all(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], ['all_5fgather_20',['all_gather',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a04bb1df23abe5b1f3fa0126375c6cea4',1,'mlx::core::distributed::detail::GroupImpl::all_gather()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04',1,'mlx::core::distributed::detail::all_gather()'],['../namespacemlx_1_1core_1_1distributed.html#a82ef5e8cc7ac62cd228e51b1c1b77cb7',1,'mlx::core::distributed::all_gather()']]], ['all_5freduce_21',['all_reduce',['../reduce__all_8h.html#a9086a585eda5a887160ee24baae0a7b8',1,'reduce_all.h']]], ['all_5freduce_5fdispatch_22',['all_reduce_dispatch',['../namespacemlx_1_1core.html#a3ab0fd997d9a35782106ff083a72e098',1,'mlx::core']]], @@ -31,7 +31,7 @@ var searchData= ['allocator_28',['Allocator',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a5803678a418fef687fc65fa9d5c37b65',1,'mlx::core::allocator::Allocator::Allocator()=default'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#aa05c081ce80dc036f9d3dd8c195259d2',1,'mlx::core::allocator::Allocator::Allocator(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a8e8ce346a16cf0c62847bed9289f9959',1,'mlx::core::allocator::Allocator::Allocator(Allocator &&other)=delete']]], ['allocator_29',['allocator',['../namespacemlx_1_1core_1_1allocator.html#aa23e2f20a336d0b159c097087194634e',1,'mlx::core::allocator::allocator()'],['../namespacemlx_1_1core_1_1metal.html#a74b3558bd518aecde6b14b0ba5e1a0d5',1,'mlx::core::metal::allocator()']]], ['allreduce_30',['AllReduce',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a2d1ea56cbf72a316680ea90aa6da1c2d',1,'mlx::core::distributed::AllReduce']]], - ['any_31',['any',['../group__ops.html#ga8598dd718fb05cb28535e250372d4e6f',1,'mlx::core::any(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gad37df97f253a963bece124198dbaf9ba',1,'mlx::core::any(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef',1,'mlx::core::any(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce',1,'mlx::core::any(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['any_31',['any',['../namespacemlx_1_1core_1_1simd.html#a9d968537ad5ef18630f5afce8453b30e',1,'mlx::core::simd::any(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#ae3b138b4bbcee0ca70b58a3e2ebd818c',1,'mlx::core::simd::any(Simd< T, 1 > x)'],['../group__ops.html#ga8598dd718fb05cb28535e250372d4e6f',1,'mlx::core::any(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gad37df97f253a963bece124198dbaf9ba',1,'mlx::core::any(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef',1,'mlx::core::any(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce',1,'mlx::core::any(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], ['apply_32',['apply',['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply()'],['../struct_transform_scale.html#a9dd329422e5b8da43486cdce17132e16',1,'TransformScale::apply()'],['../struct_max_op.html#ab3d3c3040017a13c170e7bdd1ffac46e',1,'MaxOp::apply()'],['../struct_sum_op.html#aa9563a98cbbe1b1921ade0c63ab38b4d',1,'SumOp::apply()'],['../struct_mul_op.html#a1b93d804653d92fc7e46747de9e9c756',1,'MulOp::apply()'],['../struct_sub_op.html#ad211f879a212ed0e98136217ca8e4143',1,'SubOp::apply()'],['../struct_exp_sub_op.html#a00e457a01cb38f959dfd789455e7f334',1,'ExpSubOp::apply()'],['../struct_div_op.html#a1b8df47142dc6ea15315ce3a310f9221',1,'DivOp::apply()'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const'],['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply()'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const']]], ['apply_5fepilogue_33',['apply_epilogue',['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)']]], ['apply_5fepilogue_5fsafe_34',['apply_epilogue_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA::apply_epilogue_safe(const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const BinaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA::apply_epilogue_safe(const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const BinaryEpilogue &epilogue_op)']]], @@ -66,15 +66,15 @@ var searchData= ['array_5fequal_63',['array_equal',['../group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645',1,'mlx::core::array_equal(const array &a, const array &b, bool equal_nan, StreamOrDevice s={})'],['../group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14',1,'mlx::core::array_equal(const array &a, const array &b, StreamOrDevice s={})']]], ['arrayiterator_64',['ArrayIterator',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ad3afcb24c6db7642bbc06835f7f3e27a',1,'mlx::core::array::ArrayIterator']]], ['as_5fstrided_65',['as_strided',['../group__ops.html#ga6085b03f2662ef2a61de523fd609f3bf',1,'mlx::core']]], - ['asin_66',['asin',['../namespacemetal.html#a16e843194df3fd136404bf80ba5ac95c',1,'metal::asin()'],['../namespacemetal_1_1fast.html#a769455a283da99654b6e42c3acf13eb1',1,'metal::fast::asin()'],['../namespacemetal_1_1precise.html#adc7b8b6e12e320cb32030f728dcbf438',1,'metal::precise::asin()']]], - ['asinh_67',['asinh',['../namespacemetal.html#abcc3251866930cfe880f89e7473d0e63',1,'metal::asinh()'],['../namespacemetal_1_1fast.html#a4367034b7b3e14310803bb2be975a556',1,'metal::fast::asinh()'],['../namespacemetal_1_1precise.html#aaad1cdde6687c8011fbc5fda1bb13424',1,'metal::precise::asinh()']]], + ['asin_66',['asin',['../namespacemlx_1_1core_1_1simd.html#a9c7723fc49137394fa817136a7ffb50f',1,'mlx::core::simd::asin(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a069963ffb15f06d1c48258054750dadf',1,'mlx::core::simd::asin(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a75349994f899aecb68553c2247580163',1,'mlx::core::simd::asin(Simd< T, 1 > in)'],['../namespacemetal.html#a16e843194df3fd136404bf80ba5ac95c',1,'metal::asin()'],['../namespacemetal_1_1fast.html#a769455a283da99654b6e42c3acf13eb1',1,'metal::fast::asin()'],['../namespacemetal_1_1precise.html#adc7b8b6e12e320cb32030f728dcbf438',1,'metal::precise::asin()']]], + ['asinh_67',['asinh',['../namespacemlx_1_1core_1_1simd.html#aaa76bdf1db09261d84da51d394837f5d',1,'mlx::core::simd::asinh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a7b47a5f370e8e59e1debfa5405e13266',1,'mlx::core::simd::asinh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a9323e370f6740651ebfd51367985d0e2',1,'mlx::core::simd::asinh(Simd< T, 1 > in)'],['../namespacemetal.html#abcc3251866930cfe880f89e7473d0e63',1,'metal::asinh()'],['../namespacemetal_1_1fast.html#a4367034b7b3e14310803bb2be975a556',1,'metal::fast::asinh()'],['../namespacemetal_1_1precise.html#aaad1cdde6687c8011fbc5fda1bb13424',1,'metal::precise::asinh()']]], ['asstrided_68',['AsStrided',['../classmlx_1_1core_1_1_as_strided.html#aee21aadc21343fd15aacb8f2f8ac3761',1,'mlx::core::AsStrided']]], ['astype_69',['AsType',['../classmlx_1_1core_1_1_as_type.html#a8c3241d402a8977bb4db037e225f5b47',1,'mlx::core::AsType']]], ['astype_70',['astype',['../group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62',1,'mlx::core']]], ['async_5feval_71',['async_eval',['../namespacemlx_1_1core.html#a15dda19aa7fa1fc5fca35df5cf963297',1,'mlx::core::async_eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#abf8dae3b87a03a409711691cd9c097b6',1,'mlx::core::async_eval(Arrays &&... outputs)']]], - ['atan_72',['atan',['../namespacemetal.html#a80a771553d9a0012b93620d19c48b00f',1,'metal::atan()'],['../namespacemetal_1_1fast.html#a769503b4b7f89071d0983258c5a3ac5a',1,'metal::fast::atan()'],['../namespacemetal_1_1precise.html#aaaf4b5f4786a912089bbf0ae7619a6be',1,'metal::precise::atan()']]], - ['atan2_73',['atan2',['../namespacemetal.html#a1d430793eaa38ccf0d07145e3fcd1e61',1,'metal::atan2()'],['../namespacemetal_1_1fast.html#a00e687ea46f5affe26e6aef8fd62b89a',1,'metal::fast::atan2()'],['../namespacemetal_1_1precise.html#a6f161b049cc6884f87b09b33c2d1cd7f',1,'metal::precise::atan2()']]], - ['atanh_74',['atanh',['../namespacemetal.html#a57116427997ba71dd3863bfb15de33bf',1,'metal::atanh()'],['../namespacemetal_1_1fast.html#af24608fc605db9a14427d37c36dc1c53',1,'metal::fast::atanh()'],['../namespacemetal_1_1precise.html#a902994837653b90c47f4285673e712c4',1,'metal::precise::atanh()']]], + ['atan_72',['atan',['../namespacemlx_1_1core_1_1simd.html#aa7550a1210e50c996d0db84034b8a22e',1,'mlx::core::simd::atan(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a99e84cece5722fb764844a2badc5426b',1,'mlx::core::simd::atan(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a421845a6f68f88c58f520d2c1fa15914',1,'mlx::core::simd::atan(Simd< T, 1 > in)'],['../namespacemetal.html#a80a771553d9a0012b93620d19c48b00f',1,'metal::atan()'],['../namespacemetal_1_1fast.html#a769503b4b7f89071d0983258c5a3ac5a',1,'metal::fast::atan()'],['../namespacemetal_1_1precise.html#aaaf4b5f4786a912089bbf0ae7619a6be',1,'metal::precise::atan()']]], + ['atan2_73',['atan2',['../namespacemlx_1_1core_1_1simd.html#a23dba4ee3f0811b41c381733a6e6ff16',1,'mlx::core::simd::atan2(Simd< float16_t, N > x, Simd< float16_t, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ac6104b5667e0eb379528bf7e2de23bee',1,'mlx::core::simd::atan2(Simd< T, N > a, Simd< T, N > b)'],['../namespacemlx_1_1core_1_1simd.html#af5e8e8230c7d7af8201a3aaa7f491a2d',1,'mlx::core::simd::atan2(Simd< T, 1 > a, Simd< T, 1 > b)'],['../namespacemetal.html#a1d430793eaa38ccf0d07145e3fcd1e61',1,'metal::atan2()'],['../namespacemetal_1_1fast.html#a00e687ea46f5affe26e6aef8fd62b89a',1,'metal::fast::atan2()'],['../namespacemetal_1_1precise.html#a6f161b049cc6884f87b09b33c2d1cd7f',1,'metal::precise::atan2()']]], + ['atanh_74',['atanh',['../namespacemlx_1_1core_1_1simd.html#a18d330fd2c7360b2890a722232ba35b7',1,'mlx::core::simd::atanh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ae55fd26c3e18a6a27679d2b47566f8bc',1,'mlx::core::simd::atanh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a5ebae2e6cce1889513f15be3adb265ea',1,'mlx::core::simd::atanh(Simd< T, 1 > in)'],['../namespacemetal.html#a57116427997ba71dd3863bfb15de33bf',1,'metal::atanh()'],['../namespacemetal_1_1fast.html#af24608fc605db9a14427d37c36dc1c53',1,'metal::fast::atanh()'],['../namespacemetal_1_1precise.html#a902994837653b90c47f4285673e712c4',1,'metal::precise::atanh()']]], ['atleast_5f1d_75',['atleast_1d',['../group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396',1,'mlx::core::atleast_1d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5',1,'mlx::core::atleast_1d(const std::vector< array > &a, StreamOrDevice s={})']]], ['atleast_5f2d_76',['atleast_2d',['../group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087',1,'mlx::core::atleast_2d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga9950299a80c2562f13448758f856d1f5',1,'mlx::core::atleast_2d(const std::vector< array > &a, StreamOrDevice s={})']]], ['atleast_5f3d_77',['atleast_3d',['../group__ops.html#ga4afd919601e67782ff964465919956a0',1,'mlx::core::atleast_3d(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaffdf742ad79440a60dda40062a8074fe',1,'mlx::core::atleast_3d(const std::vector< array > &a, StreamOrDevice s={})']]], diff --git a/docs/build/html/search/functions_10.js b/docs/build/html/search/functions_10.js index 1bb59d4f4..cb34db364 100644 --- a/docs/build/html/search/functions_10.js +++ b/docs/build/html/search/functions_10.js @@ -1,7 +1,7 @@ var searchData= [ ['pad_0',['Pad',['../classmlx_1_1core_1_1_pad.html#aa55090a94f574c29678d841d091cdf44',1,'mlx::core::Pad']]], - ['pad_1',['pad',['../group__ops.html#ga58e03f39aa5d63aba1646f530706a920',1,'mlx::core::pad(const array &a, const std::vector< int > &axes, const Shape &low_pad_size, const Shape &high_pad_size, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#ga8495c9902b0a67f288918cb528d5abeb',1,'mlx::core::pad(const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#gaeaae2f9637b0b3e2b4c6c0517e285935',1,'mlx::core::pad(const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#ga5b8684361f57cbfd5e6eedf82a62ded0',1,'mlx::core::pad(const array &a, int pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})']]], + ['pad_1',['pad',['../group__ops.html#gab95ebd20bd7c6d1c840007cc020cbc0c',1,'mlx::core::pad(const array &a, const std::vector< int > &axes, const Shape &low_pad_size, const Shape &high_pad_size, const array &pad_value=array(0), const std::string &mode="constant", StreamOrDevice s={})'],['../group__ops.html#gad89f464e92c356faab1f1c2e763b1fb7',1,'mlx::core::pad(const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), const std::string &mode="constant", StreamOrDevice s={})'],['../group__ops.html#gab2d5a17d3d5225fed34905e786c31c5f',1,'mlx::core::pad(const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), const std::string &mode="constant", StreamOrDevice s={})'],['../group__ops.html#gaa73ac5674467b1d5c74de6fef7204c44',1,'mlx::core::pad(const array &a, int pad_width, const array &pad_value=array(0), const std::string &mode="constant", StreamOrDevice s={})']]], ['pad_5fgpu_2',['pad_gpu',['../namespacemlx_1_1core.html#a830324cd1b6231218b3e561e247e69b9',1,'mlx::core']]], ['parallelfilereader_3',['ParallelFileReader',['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a6cdb4547408f8cbca9e2ddd82514e697',1,'mlx::core::io::ParallelFileReader']]], ['partition_4',['Partition',['../classmlx_1_1core_1_1_partition.html#a7b82ca3895b6654308fac566b277ac0d',1,'mlx::core::Partition']]], @@ -15,7 +15,7 @@ var searchData= ['pocketfft_5fc_12',['pocketfft_c',['../classpocketfft_1_1detail_1_1pocketfft__c.html#a44f46cf06f572b42ad8b53c7c6f38e4d',1,'pocketfft::detail::pocketfft_c']]], ['pocketfft_5fr_13',['pocketfft_r',['../classpocketfft_1_1detail_1_1pocketfft__r.html#a60e9b3d1f9b20ec5b86d46b0398f6f7d',1,'pocketfft::detail::pocketfft_r']]], ['post_5fin_14',['post_in',['../struct_read_writer.html#ab555cec93b66eead607e6a03d9324e1c',1,'ReadWriter::post_in(float2 elem) const'],['../struct_read_writer.html#a6c47a25b2135393045fa5f95ada59d9d',1,'ReadWriter::post_in(float elem) const']]], - ['pow_15',['pow',['../namespacemetal.html#acd288d4552215bd10455584a214c57b8',1,'metal::pow()'],['../namespacemetal_1_1fast.html#ade2367eaec894bd2e14a1351c363e003',1,'metal::fast::pow()'],['../namespacemetal_1_1precise.html#a4cce64f1f20c1c6dfd29115bdb7c8d42',1,'metal::precise::pow()']]], + ['pow_15',['pow',['../namespacemlx_1_1core_1_1simd.html#a19d535de1fc179cc39ec9643c9863cbc',1,'mlx::core::simd::pow(Simd< float16_t, N > x, Simd< float16_t, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ab18b3a88a2439fd026b6551b38d1f14a',1,'mlx::core::simd::pow(Simd< T, N > base, Simd< T, N > exp)'],['../namespacemlx_1_1core_1_1simd.html#afe3d50bc4a11061898aa57377fa9536d',1,'mlx::core::simd::pow(Simd< T, 1 > a, Simd< T, 1 > b)'],['../namespacemetal.html#acd288d4552215bd10455584a214c57b8',1,'metal::pow()'],['../namespacemetal_1_1fast.html#ade2367eaec894bd2e14a1351c363e003',1,'metal::fast::pow()'],['../namespacemetal_1_1precise.html#a4cce64f1f20c1c6dfd29115bdb7c8d42',1,'metal::precise::pow()']]], ['power_16',['Power',['../classmlx_1_1core_1_1_power.html#a7bc6c64179b7a2aef56fe1dafb6459b2',1,'mlx::core::Power']]], ['power_17',['power',['../group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d',1,'mlx::core']]], ['powr_18',['powr',['../namespacemetal.html#ae529e431f178bafedc18a889323c0bc2',1,'metal::powr()'],['../namespacemetal_1_1fast.html#a4293cbc94175b4dcc724fe4747eb5d5a',1,'metal::fast::powr()'],['../namespacemetal_1_1precise.html#ac9dbab0bd99b2b94e364aba5353bdcd7',1,'metal::precise::powr()']]], @@ -26,13 +26,13 @@ var searchData= ['primitive_23',['primitive',['../classmlx_1_1core_1_1array.html#a790548666511d8c6d9f92ee79d2ce14c',1,'mlx::core::array']]], ['primitive_5fid_24',['primitive_id',['../classmlx_1_1core_1_1array.html#af5ad83605d4eea81561246873bee1d7c',1,'mlx::core::array']]], ['primitive_5fptr_25',['primitive_ptr',['../classmlx_1_1core_1_1array.html#a5119cd616ec3c05d65878944b8889469',1,'mlx::core::array']]], - ['print_26',['print',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a6814f9008a683c6911d5b8991ef770ab',1,'mlx::core::distributed::AllReduce::print()'],['../classmlx_1_1core_1_1_primitive.html#ae1aff91354ce036596088a3e19474ecb',1,'mlx::core::Primitive::print()'],['../classmlx_1_1core_1_1_abs.html#a643d6db5116eed978e3208804a992107',1,'mlx::core::Abs::print()'],['../classmlx_1_1core_1_1_add.html#a8a96345aa63724f22b68bca7b861211d',1,'mlx::core::Add::print()'],['../classmlx_1_1core_1_1_add_m_m.html#a1262ac2c4c6e9ff6b6047bf7605e5cc9',1,'mlx::core::AddMM::print()'],['../classmlx_1_1core_1_1_arange.html#abd73d2b793da796dc7cf04c9f7d5c19e',1,'mlx::core::Arange::print()'],['../classmlx_1_1core_1_1_arc_cos.html#aa48d8bec4efbac569d809cf11648b739',1,'mlx::core::ArcCos::print()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6a9a2ab0cc360d7e2f9676db17f8e630',1,'mlx::core::ArcCosh::print()'],['../classmlx_1_1core_1_1_arc_sin.html#a895a35c9dd22fdb06e7b971bfd6fde87',1,'mlx::core::ArcSin::print()'],['../classmlx_1_1core_1_1_arc_sinh.html#aa8b2934a8a0b2eedec8257bbb5726430',1,'mlx::core::ArcSinh::print()'],['../classmlx_1_1core_1_1_arc_tan.html#ab0309e4feca36f221b3d672dc92cac05',1,'mlx::core::ArcTan::print()'],['../classmlx_1_1core_1_1_arc_tan2.html#abdfef9f572d06df1251c28222756a361',1,'mlx::core::ArcTan2::print()'],['../classmlx_1_1core_1_1_arc_tanh.html#aa9549311240d7ba225b84e1df9ad8523',1,'mlx::core::ArcTanh::print()'],['../classmlx_1_1core_1_1_arg_partition.html#aa8678d94fa1571ea71a7bf790cdb8d63',1,'mlx::core::ArgPartition::print()'],['../classmlx_1_1core_1_1_arg_reduce.html#a153a6d8dba7301c4fcd0e429154ead8f',1,'mlx::core::ArgReduce::print()'],['../classmlx_1_1core_1_1_arg_sort.html#a0b59ce43e0982d634a01631728b419bd',1,'mlx::core::ArgSort::print()'],['../classmlx_1_1core_1_1_as_type.html#aa617e29147c14bd5d1fa8ad0bf65af0c',1,'mlx::core::AsType::print()'],['../classmlx_1_1core_1_1_as_strided.html#af2e21b77ea9e6c70bca45224967745bf',1,'mlx::core::AsStrided::print()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a69b28e239da7fdb89f0a9f9467dd797d',1,'mlx::core::BitwiseBinary::print()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a37ecf6fa296d28efb7651a3c510fe159',1,'mlx::core::BlockMaskedMM::print()'],['../classmlx_1_1core_1_1_gather_m_m.html#ae7a6f4eecb15e95b21e6c87068ebd758',1,'mlx::core::GatherMM::print()'],['../classmlx_1_1core_1_1_broadcast_axes.html#aa15f81d08cabe43ac92de7534fb940df',1,'mlx::core::BroadcastAxes::print()'],['../classmlx_1_1core_1_1_broadcast.html#a6a610412861c6e472f930b6721b99a11',1,'mlx::core::Broadcast::print()'],['../classmlx_1_1core_1_1_ceil.html#a14a0048dd6496341cacaddada68276ee',1,'mlx::core::Ceil::print()'],['../classmlx_1_1core_1_1_compiled.html#a271521f92eef49c39799f38e26b64a9b',1,'mlx::core::Compiled::print()'],['../classmlx_1_1core_1_1_concatenate.html#a56f29b585a6d1d958954a68dcc893f33',1,'mlx::core::Concatenate::print()'],['../classmlx_1_1core_1_1_conjugate.html#a40281539bbd543ac8fd8e28650de17e4',1,'mlx::core::Conjugate::print()'],['../classmlx_1_1core_1_1_contiguous.html#aca8a4ba9a58cc10f063e6b082fa2fc23',1,'mlx::core::Contiguous::print()'],['../classmlx_1_1core_1_1_convolution.html#a844eab7c4cc99e775cfb561265ed14fd',1,'mlx::core::Convolution::print()'],['../classmlx_1_1core_1_1_copy.html#acfa1a02ab9cdab593e928faa515a8008',1,'mlx::core::Copy::print()'],['../classmlx_1_1core_1_1_cos.html#a81858457e4bea931a4bc6f6e38b0f696',1,'mlx::core::Cos::print()'],['../classmlx_1_1core_1_1_cosh.html#ac247faad68c1050cda9f72d7d6d040e2',1,'mlx::core::Cosh::print()'],['../classmlx_1_1core_1_1_custom_transforms.html#a2ddbacbc468271b11caee0ad97005298',1,'mlx::core::CustomTransforms::print()'],['../classmlx_1_1core_1_1_depends.html#aed575b0d927f4341f60442c70adeeb82',1,'mlx::core::Depends::print()'],['../classmlx_1_1core_1_1_divide.html#af3c15337ac15522cc34ed98b97895bb6',1,'mlx::core::Divide::print()'],['../classmlx_1_1core_1_1_div_mod.html#a7edbed50d07869d921e529157931b7a1',1,'mlx::core::DivMod::print()'],['../classmlx_1_1core_1_1_select.html#a678285f2c0b9dae85692399c3aa692a7',1,'mlx::core::Select::print()'],['../classmlx_1_1core_1_1_remainder.html#aeaecac5ea8e606d7ecd393d8019029e4',1,'mlx::core::Remainder::print()'],['../classmlx_1_1core_1_1_equal.html#a0787bf32f0b405a8b2ac809d2d990774',1,'mlx::core::Equal::print()'],['../classmlx_1_1core_1_1_erf.html#a186af7b783cf832c3b25eec3a09f5a0c',1,'mlx::core::Erf::print()'],['../classmlx_1_1core_1_1_erf_inv.html#a0acb31bd5780abf61877bd1a3e0fd4f9',1,'mlx::core::ErfInv::print()'],['../classmlx_1_1core_1_1_exp.html#ad87cc1b2ae595a613b03b0fdca63ae6a',1,'mlx::core::Exp::print()'],['../classmlx_1_1core_1_1_expm1.html#af1a99266fc50aa5948cdd298e2916ef1',1,'mlx::core::Expm1::print()'],['../classmlx_1_1core_1_1_expand_dims.html#ac8f1d849562b2222158fbe476fc2dc2e',1,'mlx::core::ExpandDims::print()'],['../classmlx_1_1core_1_1_f_f_t.html#a15a2a5f7647f5fb78611a251d3270edf',1,'mlx::core::FFT::print()'],['../classmlx_1_1core_1_1_flatten.html#ad0495ee66601c7527d836d2db77a6aec',1,'mlx::core::Flatten::print()'],['../classmlx_1_1core_1_1_floor.html#ac289e87c5fac15e2f491e2513be610f6',1,'mlx::core::Floor::print()'],['../classmlx_1_1core_1_1_full.html#a68e08303f4960ab373b84a3312edc013',1,'mlx::core::Full::print()'],['../classmlx_1_1core_1_1_gather.html#a9d57637a8a65008683c3847251bdcf91',1,'mlx::core::Gather::print()'],['../classmlx_1_1core_1_1_greater.html#aa2980e45cd2c79ebfb394012d3108a04',1,'mlx::core::Greater::print()'],['../classmlx_1_1core_1_1_greater_equal.html#ab98045c861d2d2ffb0398c2c1d671cef',1,'mlx::core::GreaterEqual::print()'],['../classmlx_1_1core_1_1_hadamard.html#a3df6e7e3b3b71bf50be5f1a05d0870b6',1,'mlx::core::Hadamard::print()'],['../classmlx_1_1core_1_1_imag.html#a0c8d48e2a1474d80a314ea9b96dbaa8d',1,'mlx::core::Imag::print()'],['../classmlx_1_1core_1_1_less.html#ad67e6f66d7b75546fd98dbee6b631d78',1,'mlx::core::Less::print()'],['../classmlx_1_1core_1_1_less_equal.html#a409842d3862113c53cbbdf7467a06950',1,'mlx::core::LessEqual::print()'],['../classmlx_1_1core_1_1_load.html#a54e08a0ca41b7c9f1a76b00c889f0bfa',1,'mlx::core::Load::print()'],['../classmlx_1_1core_1_1_log.html#a7b946d98d4a228c6be9f606a3bd8a30d',1,'mlx::core::Log::print()'],['../classmlx_1_1core_1_1_log1p.html#a8a1569dde30440ce11ea466ccc69d2d4',1,'mlx::core::Log1p::print()'],['../classmlx_1_1core_1_1_logical_not.html#a001ff3eca46440f0d8a287e0b98a8a2c',1,'mlx::core::LogicalNot::print()'],['../classmlx_1_1core_1_1_logical_and.html#a9a5220eb56e1fd94fd879394ee5ad397',1,'mlx::core::LogicalAnd::print()'],['../classmlx_1_1core_1_1_logical_or.html#a6becc5fbfadde850de9857099dcd5003',1,'mlx::core::LogicalOr::print()'],['../classmlx_1_1core_1_1_log_add_exp.html#a702a2eff0bd1ae7b6fb829dd0b0b11b9',1,'mlx::core::LogAddExp::print()'],['../classmlx_1_1core_1_1_matmul.html#abb4a16a265a05d56a2f5d2e89d6f9dfd',1,'mlx::core::Matmul::print()'],['../classmlx_1_1core_1_1_maximum.html#a3b708a1d6b526719c62850294776f8ca',1,'mlx::core::Maximum::print()'],['../classmlx_1_1core_1_1_minimum.html#a137677bf32c626a768b732a7b8575512',1,'mlx::core::Minimum::print()'],['../classmlx_1_1core_1_1_multiply.html#aa4f1f7af68346ce80c2636df415c9909',1,'mlx::core::Multiply::print()'],['../classmlx_1_1core_1_1_negative.html#a0d5c30e267ff6468d64f1987f9f83f91',1,'mlx::core::Negative::print()'],['../classmlx_1_1core_1_1_not_equal.html#a12aa2f764880d29e627540610b63af09',1,'mlx::core::NotEqual::print()'],['../classmlx_1_1core_1_1_number_of_elements.html#aecde30826970938f3aa688979a668f52',1,'mlx::core::NumberOfElements::print()'],['../classmlx_1_1core_1_1_pad.html#af87754daaf51f6a6cf8bd4949ca1e70a',1,'mlx::core::Pad::print()'],['../classmlx_1_1core_1_1_partition.html#ab5c7aa4fed325475b33d4004649f0dc0',1,'mlx::core::Partition::print()'],['../classmlx_1_1core_1_1_power.html#a33e2d7ff078426fe66ea2370ceb5af60',1,'mlx::core::Power::print()'],['../classmlx_1_1core_1_1_quantized_matmul.html#aaef8c96d4d40b4fa08ced540d341a4db',1,'mlx::core::QuantizedMatmul::print()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a53c3fa7beb51ce2e1c2da28633406fe0',1,'mlx::core::GatherQMM::print()'],['../classmlx_1_1core_1_1_random_bits.html#a8a5593c34fd868d94b36a8ced1390271',1,'mlx::core::RandomBits::print()'],['../classmlx_1_1core_1_1_real.html#a740a0dfb54c2a4467a0a59f11fe69e1b',1,'mlx::core::Real::print()'],['../classmlx_1_1core_1_1_reshape.html#a0f2323d5d67ece0eb25ecff565b21862',1,'mlx::core::Reshape::print()'],['../classmlx_1_1core_1_1_reduce.html#a399be3a89553787a0a687706881f03cd',1,'mlx::core::Reduce::print()'],['../classmlx_1_1core_1_1_round.html#af0dfe8943109c936b35ab0082f566f72',1,'mlx::core::Round::print()'],['../classmlx_1_1core_1_1_scan.html#ad5b6308c79e9b985a49df35eadd15b22',1,'mlx::core::Scan::print()'],['../classmlx_1_1core_1_1_scatter.html#aa9d45cbfb27b814517f6016092b30efa',1,'mlx::core::Scatter::print()'],['../classmlx_1_1core_1_1_sigmoid.html#ad4cd19938e5159754aa7516f405580c2',1,'mlx::core::Sigmoid::print()'],['../classmlx_1_1core_1_1_sign.html#a2aa0720fe0a6d2408eb43c25d3d45b0a',1,'mlx::core::Sign::print()'],['../classmlx_1_1core_1_1_sin.html#a73b31005551015897f15c00e8b0222e4',1,'mlx::core::Sin::print()'],['../classmlx_1_1core_1_1_sinh.html#a5b4753d52e80799d4fea0b9172d25a77',1,'mlx::core::Sinh::print()'],['../classmlx_1_1core_1_1_slice.html#a50851148948d924b71817cfbd4401504',1,'mlx::core::Slice::print()'],['../classmlx_1_1core_1_1_slice_update.html#a751eefb9922c56479b4b0de2ad45439b',1,'mlx::core::SliceUpdate::print()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a421283744fe5554ac9a8288cf47edeab',1,'mlx::core::DynamicSlice::print()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a3669f4d939ba36256c43143b603eb12b',1,'mlx::core::DynamicSliceUpdate::print()'],['../classmlx_1_1core_1_1_softmax.html#aa783610ef6b82b92681e78fc99412d83',1,'mlx::core::Softmax::print()'],['../classmlx_1_1core_1_1_sort.html#ada81b9343f80958174eba708452927a2',1,'mlx::core::Sort::print()'],['../classmlx_1_1core_1_1_split.html#ad0c31fe5972643cc75fde10445fc47f2',1,'mlx::core::Split::print()'],['../classmlx_1_1core_1_1_square.html#a75feb558cd1d615e96309dd7d1e81384',1,'mlx::core::Square::print()'],['../classmlx_1_1core_1_1_sqrt.html#a8681c8de2f50049848d320c47f713c0f',1,'mlx::core::Sqrt::print()'],['../classmlx_1_1core_1_1_stop_gradient.html#acc7a7d51cbf014dae8ba3d20bedcad50',1,'mlx::core::StopGradient::print()'],['../classmlx_1_1core_1_1_subtract.html#a3834fd305435fb5a8e512566832e372b',1,'mlx::core::Subtract::print()'],['../classmlx_1_1core_1_1_squeeze.html#a74c9c825b5b968badb9bca8159eabcdf',1,'mlx::core::Squeeze::print()'],['../classmlx_1_1core_1_1_tan.html#aeea7c284d595a2a928d5f28a55e9be7f',1,'mlx::core::Tan::print()'],['../classmlx_1_1core_1_1_tanh.html#a73f4976d641daf697cc1a231d773d78e',1,'mlx::core::Tanh::print()'],['../classmlx_1_1core_1_1_unflatten.html#a77820cf21bd1277c173305b72599bdef',1,'mlx::core::Unflatten::print()'],['../classmlx_1_1core_1_1_view.html#a513b034919a8a494add3155f910a360c',1,'mlx::core::View::print()'],['../classmlx_1_1core_1_1_transpose.html#ac6c87b850f4e5560aa13a5e1e9f9fe04',1,'mlx::core::Transpose::print()'],['../classmlx_1_1core_1_1_q_r_f.html#aba3526722b3a52b41fa8103b909f7f3b',1,'mlx::core::QRF::print()'],['../classmlx_1_1core_1_1_s_v_d.html#ab87a4e7ef857936bea66ba9e24662f53',1,'mlx::core::SVD::print()'],['../classmlx_1_1core_1_1_inverse.html#a543f18f1ce5c06c897141091e95a66e9',1,'mlx::core::Inverse::print()'],['../classmlx_1_1core_1_1_cholesky.html#a0a8b51ff7f5369d22bdc58910d4aaf84',1,'mlx::core::Cholesky::print()'],['../classmlx_1_1core_1_1_eigh.html#a2b8e47ecd60cd7330716761c5fb1fe84',1,'mlx::core::Eigh::print()'],['../structmlx_1_1core_1_1_print_formatter.html#a79fad4cf5844db8c92b066539146281b',1,'mlx::core::PrintFormatter::print(std::ostream &os, bool val)'],['../structmlx_1_1core_1_1_print_formatter.html#a8da448a8adae671b26359341ea514316',1,'mlx::core::PrintFormatter::print(std::ostream &os, int16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9d750c134a6fbfa8251c5b1d01d73287',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#adbbb9cbff767f9db73c659a0c07ba633',1,'mlx::core::PrintFormatter::print(std::ostream &os, int32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a520adb07fafd911b22bc24b295e4f6cf',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ab0c702f1ae201e17cd328c9855cf522e',1,'mlx::core::PrintFormatter::print(std::ostream &os, int64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac59a5137ddd8b32aae057bb9826ee80d',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac4b7895d1168cfc1a3d1186d8a414d2f',1,'mlx::core::PrintFormatter::print(std::ostream &os, float16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ae21005f92bc641f2d657096f5d176a6d',1,'mlx::core::PrintFormatter::print(std::ostream &os, bfloat16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a57af5c32561b95d6ac2a3a1dc4f5d43e',1,'mlx::core::PrintFormatter::print(std::ostream &os, float val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9e1dc67c9afb0a09966336504790823d',1,'mlx::core::PrintFormatter::print(std::ostream &os, complex64_t val)']]], + ['print_26',['print',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a6814f9008a683c6911d5b8991ef770ab',1,'mlx::core::distributed::AllReduce::print()'],['../classmlx_1_1core_1_1_primitive.html#ae1aff91354ce036596088a3e19474ecb',1,'mlx::core::Primitive::print()'],['../classmlx_1_1core_1_1_abs.html#a643d6db5116eed978e3208804a992107',1,'mlx::core::Abs::print()'],['../classmlx_1_1core_1_1_add.html#a8a96345aa63724f22b68bca7b861211d',1,'mlx::core::Add::print()'],['../classmlx_1_1core_1_1_add_m_m.html#a1262ac2c4c6e9ff6b6047bf7605e5cc9',1,'mlx::core::AddMM::print()'],['../classmlx_1_1core_1_1_arange.html#abd73d2b793da796dc7cf04c9f7d5c19e',1,'mlx::core::Arange::print()'],['../classmlx_1_1core_1_1_arc_cos.html#aa48d8bec4efbac569d809cf11648b739',1,'mlx::core::ArcCos::print()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6a9a2ab0cc360d7e2f9676db17f8e630',1,'mlx::core::ArcCosh::print()'],['../classmlx_1_1core_1_1_arc_sin.html#a895a35c9dd22fdb06e7b971bfd6fde87',1,'mlx::core::ArcSin::print()'],['../classmlx_1_1core_1_1_arc_sinh.html#aa8b2934a8a0b2eedec8257bbb5726430',1,'mlx::core::ArcSinh::print()'],['../classmlx_1_1core_1_1_arc_tan.html#ab0309e4feca36f221b3d672dc92cac05',1,'mlx::core::ArcTan::print()'],['../classmlx_1_1core_1_1_arc_tan2.html#abdfef9f572d06df1251c28222756a361',1,'mlx::core::ArcTan2::print()'],['../classmlx_1_1core_1_1_arc_tanh.html#aa9549311240d7ba225b84e1df9ad8523',1,'mlx::core::ArcTanh::print()'],['../classmlx_1_1core_1_1_arg_partition.html#aa8678d94fa1571ea71a7bf790cdb8d63',1,'mlx::core::ArgPartition::print()'],['../classmlx_1_1core_1_1_arg_reduce.html#a153a6d8dba7301c4fcd0e429154ead8f',1,'mlx::core::ArgReduce::print()'],['../classmlx_1_1core_1_1_arg_sort.html#a0b59ce43e0982d634a01631728b419bd',1,'mlx::core::ArgSort::print()'],['../classmlx_1_1core_1_1_as_type.html#aa617e29147c14bd5d1fa8ad0bf65af0c',1,'mlx::core::AsType::print()'],['../classmlx_1_1core_1_1_as_strided.html#af2e21b77ea9e6c70bca45224967745bf',1,'mlx::core::AsStrided::print()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a69b28e239da7fdb89f0a9f9467dd797d',1,'mlx::core::BitwiseBinary::print()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a37ecf6fa296d28efb7651a3c510fe159',1,'mlx::core::BlockMaskedMM::print()'],['../classmlx_1_1core_1_1_gather_m_m.html#ae7a6f4eecb15e95b21e6c87068ebd758',1,'mlx::core::GatherMM::print()'],['../classmlx_1_1core_1_1_broadcast_axes.html#aa15f81d08cabe43ac92de7534fb940df',1,'mlx::core::BroadcastAxes::print()'],['../classmlx_1_1core_1_1_broadcast.html#a6a610412861c6e472f930b6721b99a11',1,'mlx::core::Broadcast::print()'],['../classmlx_1_1core_1_1_ceil.html#a14a0048dd6496341cacaddada68276ee',1,'mlx::core::Ceil::print()'],['../classmlx_1_1core_1_1_compiled.html#a271521f92eef49c39799f38e26b64a9b',1,'mlx::core::Compiled::print()'],['../classmlx_1_1core_1_1_concatenate.html#a56f29b585a6d1d958954a68dcc893f33',1,'mlx::core::Concatenate::print()'],['../classmlx_1_1core_1_1_conjugate.html#a40281539bbd543ac8fd8e28650de17e4',1,'mlx::core::Conjugate::print()'],['../classmlx_1_1core_1_1_contiguous.html#aca8a4ba9a58cc10f063e6b082fa2fc23',1,'mlx::core::Contiguous::print()'],['../classmlx_1_1core_1_1_convolution.html#a844eab7c4cc99e775cfb561265ed14fd',1,'mlx::core::Convolution::print()'],['../classmlx_1_1core_1_1_copy.html#acfa1a02ab9cdab593e928faa515a8008',1,'mlx::core::Copy::print()'],['../classmlx_1_1core_1_1_cos.html#a81858457e4bea931a4bc6f6e38b0f696',1,'mlx::core::Cos::print()'],['../classmlx_1_1core_1_1_cosh.html#ac247faad68c1050cda9f72d7d6d040e2',1,'mlx::core::Cosh::print()'],['../classmlx_1_1core_1_1_custom_transforms.html#a2ddbacbc468271b11caee0ad97005298',1,'mlx::core::CustomTransforms::print()'],['../classmlx_1_1core_1_1_depends.html#aed575b0d927f4341f60442c70adeeb82',1,'mlx::core::Depends::print()'],['../classmlx_1_1core_1_1_divide.html#af3c15337ac15522cc34ed98b97895bb6',1,'mlx::core::Divide::print()'],['../classmlx_1_1core_1_1_div_mod.html#a7edbed50d07869d921e529157931b7a1',1,'mlx::core::DivMod::print()'],['../classmlx_1_1core_1_1_select.html#a678285f2c0b9dae85692399c3aa692a7',1,'mlx::core::Select::print()'],['../classmlx_1_1core_1_1_remainder.html#aeaecac5ea8e606d7ecd393d8019029e4',1,'mlx::core::Remainder::print()'],['../classmlx_1_1core_1_1_equal.html#a0787bf32f0b405a8b2ac809d2d990774',1,'mlx::core::Equal::print()'],['../classmlx_1_1core_1_1_erf.html#a186af7b783cf832c3b25eec3a09f5a0c',1,'mlx::core::Erf::print()'],['../classmlx_1_1core_1_1_erf_inv.html#a0acb31bd5780abf61877bd1a3e0fd4f9',1,'mlx::core::ErfInv::print()'],['../classmlx_1_1core_1_1_exp.html#ad87cc1b2ae595a613b03b0fdca63ae6a',1,'mlx::core::Exp::print()'],['../classmlx_1_1core_1_1_expm1.html#af1a99266fc50aa5948cdd298e2916ef1',1,'mlx::core::Expm1::print()'],['../classmlx_1_1core_1_1_expand_dims.html#ac8f1d849562b2222158fbe476fc2dc2e',1,'mlx::core::ExpandDims::print()'],['../classmlx_1_1core_1_1_f_f_t.html#a15a2a5f7647f5fb78611a251d3270edf',1,'mlx::core::FFT::print()'],['../classmlx_1_1core_1_1_flatten.html#ad0495ee66601c7527d836d2db77a6aec',1,'mlx::core::Flatten::print()'],['../classmlx_1_1core_1_1_floor.html#ac289e87c5fac15e2f491e2513be610f6',1,'mlx::core::Floor::print()'],['../classmlx_1_1core_1_1_full.html#a68e08303f4960ab373b84a3312edc013',1,'mlx::core::Full::print()'],['../classmlx_1_1core_1_1_gather.html#a9d57637a8a65008683c3847251bdcf91',1,'mlx::core::Gather::print()'],['../classmlx_1_1core_1_1_gather_axis.html#a9108bd9dfc153e6260e6340ff923ba38',1,'mlx::core::GatherAxis::print()'],['../classmlx_1_1core_1_1_greater.html#aa2980e45cd2c79ebfb394012d3108a04',1,'mlx::core::Greater::print()'],['../classmlx_1_1core_1_1_greater_equal.html#ab98045c861d2d2ffb0398c2c1d671cef',1,'mlx::core::GreaterEqual::print()'],['../classmlx_1_1core_1_1_hadamard.html#a3df6e7e3b3b71bf50be5f1a05d0870b6',1,'mlx::core::Hadamard::print()'],['../classmlx_1_1core_1_1_imag.html#a0c8d48e2a1474d80a314ea9b96dbaa8d',1,'mlx::core::Imag::print()'],['../classmlx_1_1core_1_1_less.html#ad67e6f66d7b75546fd98dbee6b631d78',1,'mlx::core::Less::print()'],['../classmlx_1_1core_1_1_less_equal.html#a409842d3862113c53cbbdf7467a06950',1,'mlx::core::LessEqual::print()'],['../classmlx_1_1core_1_1_load.html#a54e08a0ca41b7c9f1a76b00c889f0bfa',1,'mlx::core::Load::print()'],['../classmlx_1_1core_1_1_log.html#a7b946d98d4a228c6be9f606a3bd8a30d',1,'mlx::core::Log::print()'],['../classmlx_1_1core_1_1_log1p.html#a8a1569dde30440ce11ea466ccc69d2d4',1,'mlx::core::Log1p::print()'],['../classmlx_1_1core_1_1_logical_not.html#a001ff3eca46440f0d8a287e0b98a8a2c',1,'mlx::core::LogicalNot::print()'],['../classmlx_1_1core_1_1_logical_and.html#a9a5220eb56e1fd94fd879394ee5ad397',1,'mlx::core::LogicalAnd::print()'],['../classmlx_1_1core_1_1_logical_or.html#a6becc5fbfadde850de9857099dcd5003',1,'mlx::core::LogicalOr::print()'],['../classmlx_1_1core_1_1_log_add_exp.html#a702a2eff0bd1ae7b6fb829dd0b0b11b9',1,'mlx::core::LogAddExp::print()'],['../classmlx_1_1core_1_1_matmul.html#abb4a16a265a05d56a2f5d2e89d6f9dfd',1,'mlx::core::Matmul::print()'],['../classmlx_1_1core_1_1_maximum.html#a3b708a1d6b526719c62850294776f8ca',1,'mlx::core::Maximum::print()'],['../classmlx_1_1core_1_1_minimum.html#a137677bf32c626a768b732a7b8575512',1,'mlx::core::Minimum::print()'],['../classmlx_1_1core_1_1_multiply.html#aa4f1f7af68346ce80c2636df415c9909',1,'mlx::core::Multiply::print()'],['../classmlx_1_1core_1_1_negative.html#a0d5c30e267ff6468d64f1987f9f83f91',1,'mlx::core::Negative::print()'],['../classmlx_1_1core_1_1_not_equal.html#a12aa2f764880d29e627540610b63af09',1,'mlx::core::NotEqual::print()'],['../classmlx_1_1core_1_1_number_of_elements.html#aecde30826970938f3aa688979a668f52',1,'mlx::core::NumberOfElements::print()'],['../classmlx_1_1core_1_1_pad.html#af87754daaf51f6a6cf8bd4949ca1e70a',1,'mlx::core::Pad::print()'],['../classmlx_1_1core_1_1_partition.html#ab5c7aa4fed325475b33d4004649f0dc0',1,'mlx::core::Partition::print()'],['../classmlx_1_1core_1_1_power.html#a33e2d7ff078426fe66ea2370ceb5af60',1,'mlx::core::Power::print()'],['../classmlx_1_1core_1_1_quantized_matmul.html#aaef8c96d4d40b4fa08ced540d341a4db',1,'mlx::core::QuantizedMatmul::print()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a53c3fa7beb51ce2e1c2da28633406fe0',1,'mlx::core::GatherQMM::print()'],['../classmlx_1_1core_1_1_random_bits.html#a8a5593c34fd868d94b36a8ced1390271',1,'mlx::core::RandomBits::print()'],['../classmlx_1_1core_1_1_real.html#a740a0dfb54c2a4467a0a59f11fe69e1b',1,'mlx::core::Real::print()'],['../classmlx_1_1core_1_1_reshape.html#a0f2323d5d67ece0eb25ecff565b21862',1,'mlx::core::Reshape::print()'],['../classmlx_1_1core_1_1_reduce.html#a399be3a89553787a0a687706881f03cd',1,'mlx::core::Reduce::print()'],['../classmlx_1_1core_1_1_round.html#af0dfe8943109c936b35ab0082f566f72',1,'mlx::core::Round::print()'],['../classmlx_1_1core_1_1_scan.html#ad5b6308c79e9b985a49df35eadd15b22',1,'mlx::core::Scan::print()'],['../classmlx_1_1core_1_1_scatter.html#aa9d45cbfb27b814517f6016092b30efa',1,'mlx::core::Scatter::print()'],['../classmlx_1_1core_1_1_scatter_axis.html#a657843d4d9846ecd56e35e066986eb96',1,'mlx::core::ScatterAxis::print()'],['../classmlx_1_1core_1_1_sigmoid.html#ad4cd19938e5159754aa7516f405580c2',1,'mlx::core::Sigmoid::print()'],['../classmlx_1_1core_1_1_sign.html#a2aa0720fe0a6d2408eb43c25d3d45b0a',1,'mlx::core::Sign::print()'],['../classmlx_1_1core_1_1_sin.html#a73b31005551015897f15c00e8b0222e4',1,'mlx::core::Sin::print()'],['../classmlx_1_1core_1_1_sinh.html#a5b4753d52e80799d4fea0b9172d25a77',1,'mlx::core::Sinh::print()'],['../classmlx_1_1core_1_1_slice.html#a50851148948d924b71817cfbd4401504',1,'mlx::core::Slice::print()'],['../classmlx_1_1core_1_1_slice_update.html#a751eefb9922c56479b4b0de2ad45439b',1,'mlx::core::SliceUpdate::print()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a421283744fe5554ac9a8288cf47edeab',1,'mlx::core::DynamicSlice::print()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a3669f4d939ba36256c43143b603eb12b',1,'mlx::core::DynamicSliceUpdate::print()'],['../classmlx_1_1core_1_1_softmax.html#aa783610ef6b82b92681e78fc99412d83',1,'mlx::core::Softmax::print()'],['../classmlx_1_1core_1_1_sort.html#ada81b9343f80958174eba708452927a2',1,'mlx::core::Sort::print()'],['../classmlx_1_1core_1_1_split.html#ad0c31fe5972643cc75fde10445fc47f2',1,'mlx::core::Split::print()'],['../classmlx_1_1core_1_1_square.html#a75feb558cd1d615e96309dd7d1e81384',1,'mlx::core::Square::print()'],['../classmlx_1_1core_1_1_sqrt.html#a8681c8de2f50049848d320c47f713c0f',1,'mlx::core::Sqrt::print()'],['../classmlx_1_1core_1_1_stop_gradient.html#acc7a7d51cbf014dae8ba3d20bedcad50',1,'mlx::core::StopGradient::print()'],['../classmlx_1_1core_1_1_subtract.html#a3834fd305435fb5a8e512566832e372b',1,'mlx::core::Subtract::print()'],['../classmlx_1_1core_1_1_squeeze.html#a74c9c825b5b968badb9bca8159eabcdf',1,'mlx::core::Squeeze::print()'],['../classmlx_1_1core_1_1_tan.html#aeea7c284d595a2a928d5f28a55e9be7f',1,'mlx::core::Tan::print()'],['../classmlx_1_1core_1_1_tanh.html#a73f4976d641daf697cc1a231d773d78e',1,'mlx::core::Tanh::print()'],['../classmlx_1_1core_1_1_unflatten.html#a77820cf21bd1277c173305b72599bdef',1,'mlx::core::Unflatten::print()'],['../classmlx_1_1core_1_1_view.html#a513b034919a8a494add3155f910a360c',1,'mlx::core::View::print()'],['../classmlx_1_1core_1_1_transpose.html#ac6c87b850f4e5560aa13a5e1e9f9fe04',1,'mlx::core::Transpose::print()'],['../classmlx_1_1core_1_1_q_r_f.html#aba3526722b3a52b41fa8103b909f7f3b',1,'mlx::core::QRF::print()'],['../classmlx_1_1core_1_1_s_v_d.html#ab87a4e7ef857936bea66ba9e24662f53',1,'mlx::core::SVD::print()'],['../classmlx_1_1core_1_1_inverse.html#a543f18f1ce5c06c897141091e95a66e9',1,'mlx::core::Inverse::print()'],['../classmlx_1_1core_1_1_cholesky.html#a0a8b51ff7f5369d22bdc58910d4aaf84',1,'mlx::core::Cholesky::print()'],['../classmlx_1_1core_1_1_eigh.html#a2b8e47ecd60cd7330716761c5fb1fe84',1,'mlx::core::Eigh::print()'],['../structmlx_1_1core_1_1_print_formatter.html#a79fad4cf5844db8c92b066539146281b',1,'mlx::core::PrintFormatter::print(std::ostream &os, bool val)'],['../structmlx_1_1core_1_1_print_formatter.html#a8da448a8adae671b26359341ea514316',1,'mlx::core::PrintFormatter::print(std::ostream &os, int16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9d750c134a6fbfa8251c5b1d01d73287',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#adbbb9cbff767f9db73c659a0c07ba633',1,'mlx::core::PrintFormatter::print(std::ostream &os, int32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a520adb07fafd911b22bc24b295e4f6cf',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ab0c702f1ae201e17cd328c9855cf522e',1,'mlx::core::PrintFormatter::print(std::ostream &os, int64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac59a5137ddd8b32aae057bb9826ee80d',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac4b7895d1168cfc1a3d1186d8a414d2f',1,'mlx::core::PrintFormatter::print(std::ostream &os, float16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ae21005f92bc641f2d657096f5d176a6d',1,'mlx::core::PrintFormatter::print(std::ostream &os, bfloat16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a57af5c32561b95d6ac2a3a1dc4f5d43e',1,'mlx::core::PrintFormatter::print(std::ostream &os, float val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9e1dc67c9afb0a09966336504790823d',1,'mlx::core::PrintFormatter::print(std::ostream &os, complex64_t val)']]], ['print_5fcomplex_5fconstant_27',['print_complex_constant',['../namespacemlx_1_1core.html#a2b78f270942c6eb185e8045f1c5b4286',1,'mlx::core']]], ['print_5fconstant_28',['print_constant',['../namespacemlx_1_1core.html#a7d11b000895d44d183260634f4192d92',1,'mlx::core']]], ['print_5ffloat_5fconstant_29',['print_float_constant',['../namespacemlx_1_1core.html#a93a8ac59c644b801ec8881a58368caf2',1,'mlx::core']]], ['print_5fgraph_30',['print_graph',['../namespacemlx_1_1core.html#a9b33065059fd68fed26da94603cfcce3',1,'mlx::core::print_graph(std::ostream &os, NodeNamer namer, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a8ac23fb7f4d4c52e592d6296e63b80d4',1,'mlx::core::print_graph(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a17505ed8064dcaddc011cb3d52da2523',1,'mlx::core::print_graph(std::ostream &os, Arrays &&... outputs)'],['../namespacemlx_1_1core.html#a59e84542600e1a23464c100da3cfb7c4',1,'mlx::core::print_graph(std::ostream &os, NodeNamer namer, Arrays &&... outputs)']]], ['print_5fint_5fconstant_31',['print_int_constant',['../namespacemlx_1_1core.html#a948ce3dfc4520d3aa98b33e42f617c64',1,'mlx::core']]], - ['prod_32',['prod',['../structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89',1,'pocketfft::detail::util::prod()'],['../group__ops.html#ga4a09b7241d564d92548bc2773eb1d544',1,'mlx::core::prod(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga61832191f3c42ea549cf04953edc3602',1,'mlx::core::prod(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2b3935108f641e20a70dbf63f540d970',1,'mlx::core::prod(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga8a10a10b81c69996d0aca8ba401f8ff0',1,'mlx::core::prod(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['prod_32',['prod',['../structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89',1,'pocketfft::detail::util::prod()'],['../namespacemlx_1_1core_1_1simd.html#a271cedfc48efc69db43813e8c424bf7c',1,'mlx::core::simd::prod(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#ac27cdc630e86b25ad607ca409de2b274',1,'mlx::core::simd::prod(Simd< T, 1 > x)'],['../namespacemlx_1_1core_1_1simd.html#a0a26dff48b078fb3e9fef688232183ed',1,'mlx::core::simd::prod(Simd< float16_t, N > x)'],['../group__ops.html#ga4a09b7241d564d92548bc2773eb1d544',1,'mlx::core::prod(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga61832191f3c42ea549cf04953edc3602',1,'mlx::core::prod(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2b3935108f641e20a70dbf63f540d970',1,'mlx::core::prod(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga8a10a10b81c69996d0aca8ba401f8ff0',1,'mlx::core::prod(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], ['promote_5ftypes_33',['promote_types',['../namespacemlx_1_1core.html#a505922e54acd43114308e3bdbda0e497',1,'mlx::core']]], ['ptr_34',['ptr',['../classmlx_1_1core_1_1allocator_1_1_buffer.html#a990643feac06961c5599aac098c17b94',1,'mlx::core::allocator::Buffer::ptr() const'],['../classmlx_1_1core_1_1allocator_1_1_buffer.html#acb15b2f057568828ea09635ed968b62a',1,'mlx::core::allocator::Buffer::ptr()'],['../classmlx_1_1core_1_1metal_1_1_buffer.html#a990643feac06961c5599aac098c17b94',1,'mlx::core::metal::Buffer::ptr() const'],['../classmlx_1_1core_1_1metal_1_1_buffer.html#acb15b2f057568828ea09635ed968b62a',1,'mlx::core::metal::Buffer::ptr()']]], ['push_35',['push',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#ae8ebd6bc1b4d18e75bd046005e7dde7b',1,'pocketfft::detail::threading::concurrent_queue']]], diff --git a/docs/build/html/search/functions_12.js b/docs/build/html/search/functions_12.js index aa78e9c35..6b6526d18 100644 --- a/docs/build/html/search/functions_12.js +++ b/docs/build/html/search/functions_12.js @@ -29,56 +29,57 @@ var searchData= ['read_26',['read',['../classmlx_1_1core_1_1io_1_1_reader.html#ad8d74e2c62b579511089faa4cc6f50a1',1,'mlx::core::io::Reader::read(char *data, size_t n)=0'],['../classmlx_1_1core_1_1io_1_1_reader.html#a3e82cc31bd2a8594f19dc9858dca3efc',1,'mlx::core::io::Reader::read(char *data, size_t n, size_t offset)=0'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a6691826fc8d28f83792bfa2f92660a3b',1,'mlx::core::io::ParallelFileReader::read(char *data, size_t n) override'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a2b83b4576f1942db869171cccbf607df',1,'mlx::core::io::ParallelFileReader::read(char *data, size_t n, size_t offset) override']]], ['readwriter_27',['ReadWriter',['../struct_read_writer.html#a1aa07e41d7ac286ad79bd26a072dfa0c',1,'ReadWriter']]], ['real_28',['Real',['../classmlx_1_1core_1_1_real.html#acd4480e3f0834d70ff6b5f1ecef17892',1,'mlx::core::Real']]], - ['real_29',['real',['../group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764',1,'mlx::core']]], - ['reciprocal_30',['reciprocal',['../group__ops.html#ga4d29556bb93e2f66916116cf1f062b36',1,'mlx::core']]], - ['recv_31',['Recv',['../classmlx_1_1core_1_1distributed_1_1_recv.html#a511dd4e0259da18a181a25579d9b55db',1,'mlx::core::distributed::Recv']]], - ['recv_32',['recv',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac4af5fc16a82ba8c72df04d7694f8352',1,'mlx::core::distributed::detail::GroupImpl::recv()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#a003de04deb00ecbb19179b3f557df548',1,'mlx::core::distributed::detail::recv()'],['../namespacemlx_1_1core_1_1distributed.html#af93c1680b656e98158d5f6eed8e092e8',1,'mlx::core::distributed::recv(Shape shape, Dtype dtype, int src, std::optional< Group > group=std::nullopt, StreamOrDevice s={})']]], - ['recv_5flike_33',['recv_like',['../namespacemlx_1_1core_1_1distributed.html#a2822b78bce2c679e6ff940b2fca944f0',1,'mlx::core::distributed']]], - ['reduce_34',['Reduce',['../classmlx_1_1core_1_1_reduce.html#a055368c1d036fb953a23ef230e33dcbf',1,'mlx::core::Reduce']]], - ['reduce_35',['reduce',['../namespacemlx_1_1core_1_1metal.html#abb997ccbed4c9a9ccd975b1574755fca',1,'mlx::core::metal']]], - ['reduce_5futils_36',['reduce_utils',['../namespacemlx_1_1core_1_1metal.html#a2ec39572806310cf528aea06530e8af8',1,'mlx::core::metal']]], - ['reduction_5fop_37',['reduction_op',['../namespacemlx_1_1core.html#aa08ffc1e8f2c58afb2d463496f827ef0',1,'mlx::core::reduction_op(const array &x, array &out, const std::vector< int > &axes, U init, OpS ops, OpC opc, Op op)'],['../namespacemlx_1_1core.html#a064d61b6ddc9e5d1e261a7e33de71083',1,'mlx::core::reduction_op(const array &x, array &out, const std::vector< int > &axes, U init, Op op)']]], + ['real_29',['real',['../namespacemlx_1_1core_1_1simd.html#acdf822b7626bbab6a495552aea3457b5',1,'mlx::core::simd::real()'],['../group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764',1,'mlx::core::real()']]], + ['recip_30',['recip',['../namespacemlx_1_1core_1_1simd.html#ae344abefc91c7d9c0a9506c868a84d61',1,'mlx::core::simd::recip(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#afc915aed256295475ac88fde3a736f1f',1,'mlx::core::simd::recip(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a6235990c43aaf0e0c126c82d10f01b45',1,'mlx::core::simd::recip(Simd< float16_t, N > a)']]], + ['reciprocal_31',['reciprocal',['../group__ops.html#ga4d29556bb93e2f66916116cf1f062b36',1,'mlx::core']]], + ['recv_32',['Recv',['../classmlx_1_1core_1_1distributed_1_1_recv.html#a511dd4e0259da18a181a25579d9b55db',1,'mlx::core::distributed::Recv']]], + ['recv_33',['recv',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac4af5fc16a82ba8c72df04d7694f8352',1,'mlx::core::distributed::detail::GroupImpl::recv()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#a003de04deb00ecbb19179b3f557df548',1,'mlx::core::distributed::detail::recv()'],['../namespacemlx_1_1core_1_1distributed.html#af93c1680b656e98158d5f6eed8e092e8',1,'mlx::core::distributed::recv(Shape shape, Dtype dtype, int src, std::optional< Group > group=std::nullopt, StreamOrDevice s={})']]], + ['recv_5flike_34',['recv_like',['../namespacemlx_1_1core_1_1distributed.html#a2822b78bce2c679e6ff940b2fca944f0',1,'mlx::core::distributed']]], + ['reduce_35',['Reduce',['../classmlx_1_1core_1_1_reduce.html#a055368c1d036fb953a23ef230e33dcbf',1,'mlx::core::Reduce']]], + ['reduce_36',['reduce',['../namespacemlx_1_1core_1_1metal.html#abb997ccbed4c9a9ccd975b1574755fca',1,'mlx::core::metal']]], + ['reduce_5futils_37',['reduce_utils',['../namespacemlx_1_1core_1_1metal.html#a2ec39572806310cf528aea06530e8af8',1,'mlx::core::metal']]], ['reductionplan_38',['ReductionPlan',['../structmlx_1_1core_1_1_reduction_plan.html#a07d9eb40a259918ce23360416b3e9db8',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_, Shape shape_, Strides strides_)'],['../structmlx_1_1core_1_1_reduction_plan.html#aec7496f3740a0b0d51aaa606f6fd68f4',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_)']]], ['register_5flibrary_39',['register_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a45945f2efcd242d915ffa2171e92bf9d',1,'mlx::core::metal::Device::register_library(const std::string &lib_name, const std::string &lib_path)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a99ff72689b7beb65ad4541391b0eeabf',1,'mlx::core::metal::Device::register_library(const std::string &lib_name)']]], - ['remainder_40',['Remainder',['../classmlx_1_1core_1_1_remainder.html#a4f3eada4a21898af4a77d1d27ce14641',1,'mlx::core::Remainder']]], - ['remainder_41',['remainder',['../group__ops.html#ga99f5c904f724156a814d7817188351d2',1,'mlx::core']]], - ['remaining_42',['remaining',['../classpocketfft_1_1detail_1_1multi__iter.html#a034d12f842df90e6471dffd3fa6ba4bd',1,'pocketfft::detail::multi_iter::remaining()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a9267d37f51a9a5aecc69293c7ed1b1f6',1,'pocketfft::detail::simple_iter::remaining()'],['../classpocketfft_1_1detail_1_1rev__iter.html#a143637135c441a4b9a2959c2370d8c63',1,'pocketfft::detail::rev_iter::remaining()']]], - ['repeat_43',['repeat',['../group__ops.html#gab49e3a687e826554ed1574186e8ae974',1,'mlx::core::repeat(const array &arr, int repeats, int axis, StreamOrDevice s={})'],['../group__ops.html#ga4f75f5d5db999f02f43ecbc6dccf3ba6',1,'mlx::core::repeat(const array &arr, int repeats, StreamOrDevice s={})']]], - ['reset_44',['reset',['../structmlx_1_1core_1_1_contiguous_iterator.html#afa2e2bde9bfa57ac759bc7f5b881262a',1,'mlx::core::ContiguousIterator']]], - ['reset_5fpeak_5fmemory_45',['reset_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a26b9c8ac7ed56c3bb7ddc194009ec5a6',1,'mlx::core::metal::MetalAllocator::reset_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#adec8bb375da6c9dd5ff625a3a8434122',1,'mlx::core::metal::reset_peak_memory()']]], - ['reshape_46',['Reshape',['../classmlx_1_1core_1_1_reshape.html#aa5a5d520b6ec6c8d9ba9d79808e36312',1,'mlx::core::Reshape']]], - ['reshape_47',['reshape',['../group__ops.html#ga084f03ce2b22258afb7c8b45e17af828',1,'mlx::core']]], - ['residencyset_48',['ResidencySet',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#abb69d020da017a7e52e9e3903b877eec',1,'mlx::core::metal::ResidencySet::ResidencySet(MTL::Device *d)'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aabbf8c16f269f38e4c38097b947d18b7',1,'mlx::core::metal::ResidencySet::ResidencySet(const ResidencySet &)=delete']]], - ['resize_49',['resize',['../classpocketfft_1_1detail_1_1arr.html#a8d73baaefa02dff8714e4398c83917e0',1,'pocketfft::detail::arr::resize()'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#a0364647bca4324ac41ea3900925a69b5',1,'mlx::core::metal::ResidencySet::resize()']]], - ['restart_50',['restart',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288',1,'pocketfft::detail::threading::thread_pool']]], - ['result_5ftype_51',['result_type',['../namespacemlx_1_1core.html#a8b984eef832f757e28cd262d64a49ae7',1,'mlx::core::result_type(const array &a, const array &b)'],['../namespacemlx_1_1core.html#ac457c232f956ba802acb69c5a621633d',1,'mlx::core::result_type(const array &a, const array &b, const array &c)'],['../namespacemlx_1_1core.html#aafaf24a28297428caf6d0c36c623489e',1,'mlx::core::result_type(const std::vector< array > &arrays)']]], - ['retain_5fgraph_52',['retain_graph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a12ead93cb70ebab865c5e9ce7718f814',1,'mlx::core::detail::RetainGraph::retain_graph()'],['../namespacemlx_1_1core_1_1detail.html#a38af45eb92e437207c722a088f381cd3',1,'mlx::core::detail::retain_graph()']]], - ['retaingraph_53',['RetainGraph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a7fac0244c14cc9e8f580bc1298ff68da',1,'mlx::core::detail::RetainGraph']]], - ['rev_5fiter_54',['rev_iter',['../classpocketfft_1_1detail_1_1rev__iter.html#af7b8c2f1534d3038ba2a3c6b9919e134',1,'pocketfft::detail::rev_iter']]], - ['rev_5fofs_55',['rev_ofs',['../classpocketfft_1_1detail_1_1rev__iter.html#a7f112afa76cb7a4c29cff217a6f5f5a9',1,'pocketfft::detail::rev_iter']]], - ['rfft_56',['rfft',['../namespacemlx_1_1core_1_1fft.html#a9cb0edfb831b1ed607a8124d38540c13',1,'mlx::core::fft::rfft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a464016cbc948bb3af17d43ce39cf54bd',1,'mlx::core::fft::rfft(const array &a, int axis=-1, StreamOrDevice s={})']]], - ['rfft2_57',['rfft2',['../namespacemlx_1_1core_1_1fft.html#a99397f5d9de6551f967120546ec96728',1,'mlx::core::fft::rfft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a59ca0c3c455e4ff1fed3dbd2327c55f0',1,'mlx::core::fft::rfft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], - ['rfftn_58',['rfftn',['../namespacemlx_1_1core_1_1fft.html#ab60d121ff5509c5a144b2fab7ae0f93b',1,'mlx::core::fft::rfftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#ab502e092ba4bb571ecc421a25e4cb968',1,'mlx::core::fft::rfftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a53d44fd9b6c7645f9303c24099755bf2',1,'mlx::core::fft::rfftn(const array &a, StreamOrDevice s={})']]], - ['rfftp_59',['rfftp',['../classpocketfft_1_1detail_1_1rfftp.html#a0c590f917b8e8afa3ff53ccff52e68c5',1,'pocketfft::detail::rfftp']]], - ['right_5fshift_60',['right_shift',['../group__ops.html#gafa376ad57d38ba87378f0272dc379b23',1,'mlx::core']]], - ['rint_61',['rint',['../namespacemetal.html#a29ab6060527120eee745aec0daa06e01',1,'metal::rint()'],['../namespacemetal_1_1fast.html#aa613bc252f8d8069e175ec9e9d05a7ec',1,'metal::fast::rint()'],['../namespacemetal_1_1precise.html#ab17bd408098270ad92f37bcd1039c254',1,'metal::precise::rint()']]], - ['rms_5fnorm_62',['rms_norm',['../namespacemlx_1_1core_1_1fast.html#ac7b620275c6386f822b7aacc6b312e62',1,'mlx::core::fast']]], - ['rmsnorm_63',['RMSNorm',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a22adaff0749711263388ec151fcfebe2',1,'mlx::core::fast::RMSNorm']]], - ['rmsnormvjp_64',['RMSNormVJP',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#aac060129b2e1af79bf388bfe705381ca',1,'mlx::core::fast::RMSNormVJP']]], - ['roll_65',['roll',['../group__ops.html#gac40e48c69f9c715a767912c30836e75c',1,'mlx::core::roll(const array &a, int shift, StreamOrDevice s={})'],['../group__ops.html#ga5011d1a5735c64e5b91afa56c7e2cc02',1,'mlx::core::roll(const array &a, const Shape &shift, StreamOrDevice s={})'],['../group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436',1,'mlx::core::roll(const array &a, int shift, int axis, StreamOrDevice s={})'],['../group__ops.html#ga665f502ecc96f1f4467556b784abf9ae',1,'mlx::core::roll(const array &a, int shift, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga79137f90bc44ac9e35f408c012701df9',1,'mlx::core::roll(const array &a, const Shape &shift, int axis, StreamOrDevice s={})'],['../group__ops.html#ga9d76930fb567a7d459ff96fb851abe36',1,'mlx::core::roll(const array &a, const Shape &shift, const std::vector< int > &axes, StreamOrDevice s={})']]], - ['rope_66',['RoPE',['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a60b399d7f38c0f5f50342a6b97f0eb1a',1,'mlx::core::fast::RoPE']]], - ['rope_67',['rope',['../namespacemlx_1_1core_1_1fast.html#a534ef357eae24892684a6ecd866d3fab',1,'mlx::core::fast::rope(const array &x, int dims, bool traditional, std::optional< float > base, float scale, int offset, const std::optional< array > &freqs=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fast.html#a1632b78950f0c8c31b24be7d80faeb39',1,'mlx::core::fast::rope(const array &x, int dims, bool traditional, std::optional< float > base, float scale, const array &offset, const std::optional< array > &freqs=std::nullopt, StreamOrDevice s={})']]], - ['rot90_68',['ROT90',['../namespacepocketfft_1_1detail.html#a928bad5278df636ee47402c0a75f64ef',1,'pocketfft::detail']]], - ['rotx90_69',['ROTX90',['../namespacepocketfft_1_1detail.html#ab6a43dc0cec4291e163e68a0875ac501',1,'pocketfft::detail']]], - ['round_70',['Round',['../classmlx_1_1core_1_1_round.html#a1327a359b2aed91f576145a0e70d1dde',1,'mlx::core::Round']]], - ['round_71',['round',['../namespacemetal.html#a46c667e169ff9d51a9204a045305442f',1,'metal::round()'],['../namespacemetal_1_1fast.html#a4cb687257a004726d49e496417eaa40f',1,'metal::fast::round()'],['../namespacemetal_1_1precise.html#a5295ab08055d12534cc3775da855ac12',1,'metal::precise::round()'],['../group__ops.html#ga2d74d43f007a069384e89d8416525331',1,'mlx::core::round(const array &a, int decimals, StreamOrDevice s={})'],['../group__ops.html#gaf18fb7e98bf8cf3b7fbc5e64c988a95b',1,'mlx::core::round(const array &a, StreamOrDevice s={})']]], - ['round_5ferror_72',['round_error',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#afa223448fa4f04c1113a85345dd720c3',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['row_5fbin_5fop_73',['row_bin_op',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a318c4279bdc7b39b7919f108b1cd8010',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::row_bin_op()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a3d0d5b9c7962658cc6d5afbbbb2f19e2',1,'mlx::steel::MMATile::row_bin_op()']]], - ['row_5freduce_74',['row_reduce',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a51d662e4cff88b5ad17d7c44bb6b6970',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::row_reduce()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa0ad5cb750ace934bf230385d8bd9f88',1,'mlx::steel::MMATile::row_reduce()']]], - ['row_5freduce_5fgeneral_5fdispatch_75',['row_reduce_general_dispatch',['../namespacemlx_1_1core.html#ab1eeca8ec6fa31819ee108fa6ed2c41b',1,'mlx::core']]], - ['row_5freduce_5flooped_76',['row_reduce_looped',['../reduce__row_8h.html#a72611b8006ae5642b69f4d250d69865b',1,'reduce_row.h']]], - ['row_5freduce_5fsimple_77',['row_reduce_simple',['../reduce__row_8h.html#a7dbd7cc81b1ce5a4271d56b99ce595d4',1,'reduce_row.h']]], - ['row_5freduce_5fsmall_78',['row_reduce_small',['../reduce__row_8h.html#a1e1b59fa73d2b0be978494a759f2c6ee',1,'reduce_row.h']]], - ['rsqrt_79',['rsqrt',['../namespacemetal.html#a1cf4b605c0aa7ff5bfe5e979a16f5157',1,'metal::rsqrt()'],['../namespacemetal_1_1fast.html#aa62097c750f1e4b69d09277f19976ab1',1,'metal::fast::rsqrt()'],['../namespacemetal_1_1precise.html#afb397b477745f12a44423934fa2b05ac',1,'metal::precise::rsqrt()'],['../group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1',1,'mlx::core::rsqrt()']]], - ['run_80',['run',['../struct_g_e_m_v_kernel.html#ac4a7b5011a0ea938ab1949bb1767fc1a',1,'GEMVKernel::run()'],['../struct_g_e_m_v_t_kernel.html#a5d68656832de892f33db939005713927',1,'GEMVTKernel::run()'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run(const device T *A, const device T *B, device U *D, const constant GEMMParams *params, threadgroup T *As, threadgroup T *Bs, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run(const device T *A, const device T *B, device U *D, const constant GEMMParams *params, threadgroup T *As, threadgroup T *Bs, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)']]] + ['register_5foutput_5farray_40',['register_output_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ada20558738968ca2ecdcd95f228e028a',1,'mlx::core::metal::CommandEncoder::register_output_array()'],['../structmlx_1_1core_1_1_command_encoder.html#ada20558738968ca2ecdcd95f228e028a',1,'mlx::core::CommandEncoder::register_output_array()']]], + ['remainder_41',['Remainder',['../classmlx_1_1core_1_1_remainder.html#a4f3eada4a21898af4a77d1d27ce14641',1,'mlx::core::Remainder']]], + ['remainder_42',['remainder',['../namespacemlx_1_1core_1_1simd.html#ac66bdf1a8e86a4d350c85037bc764da5',1,'mlx::core::simd::remainder(Simd< float16_t, N > x, Simd< float16_t, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ab020d2c434fad0cdf79fd37b0f6c1676',1,'mlx::core::simd::remainder(Simd< T, N > a, Simd< T, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a54c7f2f2b995eb767462b1228982967f',1,'mlx::core::simd::remainder(Simd< T, 1 > a_, Simd< T, 1 > b_)'],['../group__ops.html#ga99f5c904f724156a814d7817188351d2',1,'mlx::core::remainder()']]], + ['remaining_43',['remaining',['../classpocketfft_1_1detail_1_1multi__iter.html#a034d12f842df90e6471dffd3fa6ba4bd',1,'pocketfft::detail::multi_iter::remaining()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a9267d37f51a9a5aecc69293c7ed1b1f6',1,'pocketfft::detail::simple_iter::remaining()'],['../classpocketfft_1_1detail_1_1rev__iter.html#a143637135c441a4b9a2959c2370d8c63',1,'pocketfft::detail::rev_iter::remaining()']]], + ['repeat_44',['repeat',['../group__ops.html#gab49e3a687e826554ed1574186e8ae974',1,'mlx::core::repeat(const array &arr, int repeats, int axis, StreamOrDevice s={})'],['../group__ops.html#ga4f75f5d5db999f02f43ecbc6dccf3ba6',1,'mlx::core::repeat(const array &arr, int repeats, StreamOrDevice s={})']]], + ['reset_45',['reset',['../structmlx_1_1core_1_1_contiguous_iterator.html#afa2e2bde9bfa57ac759bc7f5b881262a',1,'mlx::core::ContiguousIterator']]], + ['reset_5fpeak_5fmemory_46',['reset_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a26b9c8ac7ed56c3bb7ddc194009ec5a6',1,'mlx::core::metal::MetalAllocator::reset_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#adec8bb375da6c9dd5ff625a3a8434122',1,'mlx::core::metal::reset_peak_memory()']]], + ['reshape_47',['Reshape',['../classmlx_1_1core_1_1_reshape.html#aa5a5d520b6ec6c8d9ba9d79808e36312',1,'mlx::core::Reshape']]], + ['reshape_48',['reshape',['../group__ops.html#ga084f03ce2b22258afb7c8b45e17af828',1,'mlx::core']]], + ['residencyset_49',['ResidencySet',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#abb69d020da017a7e52e9e3903b877eec',1,'mlx::core::metal::ResidencySet::ResidencySet(MTL::Device *d)'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aabbf8c16f269f38e4c38097b947d18b7',1,'mlx::core::metal::ResidencySet::ResidencySet(const ResidencySet &)=delete']]], + ['resize_50',['resize',['../classpocketfft_1_1detail_1_1arr.html#a8d73baaefa02dff8714e4398c83917e0',1,'pocketfft::detail::arr::resize()'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#a0364647bca4324ac41ea3900925a69b5',1,'mlx::core::metal::ResidencySet::resize()'],['../class_thread_pool.html#a33d9a848213206e95997eb050702ecbf',1,'ThreadPool::resize()']]], + ['restart_51',['restart',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288',1,'pocketfft::detail::threading::thread_pool']]], + ['result_5ftype_52',['result_type',['../namespacemlx_1_1core.html#a8b984eef832f757e28cd262d64a49ae7',1,'mlx::core::result_type(const array &a, const array &b)'],['../namespacemlx_1_1core.html#ac457c232f956ba802acb69c5a621633d',1,'mlx::core::result_type(const array &a, const array &b, const array &c)'],['../namespacemlx_1_1core.html#aafaf24a28297428caf6d0c36c623489e',1,'mlx::core::result_type(const std::vector< array > &arrays)']]], + ['retain_5fgraph_53',['retain_graph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a12ead93cb70ebab865c5e9ce7718f814',1,'mlx::core::detail::RetainGraph::retain_graph()'],['../namespacemlx_1_1core_1_1detail.html#a38af45eb92e437207c722a088f381cd3',1,'mlx::core::detail::retain_graph()']]], + ['retaingraph_54',['RetainGraph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a7fac0244c14cc9e8f580bc1298ff68da',1,'mlx::core::detail::RetainGraph']]], + ['rev_5fiter_55',['rev_iter',['../classpocketfft_1_1detail_1_1rev__iter.html#af7b8c2f1534d3038ba2a3c6b9919e134',1,'pocketfft::detail::rev_iter']]], + ['rev_5fofs_56',['rev_ofs',['../classpocketfft_1_1detail_1_1rev__iter.html#a7f112afa76cb7a4c29cff217a6f5f5a9',1,'pocketfft::detail::rev_iter']]], + ['rfft_57',['rfft',['../namespacemlx_1_1core_1_1fft.html#a9cb0edfb831b1ed607a8124d38540c13',1,'mlx::core::fft::rfft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a464016cbc948bb3af17d43ce39cf54bd',1,'mlx::core::fft::rfft(const array &a, int axis=-1, StreamOrDevice s={})']]], + ['rfft2_58',['rfft2',['../namespacemlx_1_1core_1_1fft.html#a99397f5d9de6551f967120546ec96728',1,'mlx::core::fft::rfft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a59ca0c3c455e4ff1fed3dbd2327c55f0',1,'mlx::core::fft::rfft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], + ['rfftn_59',['rfftn',['../namespacemlx_1_1core_1_1fft.html#ab60d121ff5509c5a144b2fab7ae0f93b',1,'mlx::core::fft::rfftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#ab502e092ba4bb571ecc421a25e4cb968',1,'mlx::core::fft::rfftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a53d44fd9b6c7645f9303c24099755bf2',1,'mlx::core::fft::rfftn(const array &a, StreamOrDevice s={})']]], + ['rfftp_60',['rfftp',['../classpocketfft_1_1detail_1_1rfftp.html#a0c590f917b8e8afa3ff53ccff52e68c5',1,'pocketfft::detail::rfftp']]], + ['right_5fshift_61',['right_shift',['../group__ops.html#gafa376ad57d38ba87378f0272dc379b23',1,'mlx::core']]], + ['rint_62',['rint',['../namespacemlx_1_1core_1_1simd.html#a400d89d040f43d471b306a8e8bdb3974',1,'mlx::core::simd::rint(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a797196eccc3690aac5c45e5f9c804ceb',1,'mlx::core::simd::rint(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a8c200919c0eeefb2e2e5d9d19741a805',1,'mlx::core::simd::rint(Simd< float16_t, N > a)'],['../namespacemetal.html#a29ab6060527120eee745aec0daa06e01',1,'metal::rint()'],['../namespacemetal_1_1fast.html#aa613bc252f8d8069e175ec9e9d05a7ec',1,'metal::fast::rint()'],['../namespacemetal_1_1precise.html#ab17bd408098270ad92f37bcd1039c254',1,'metal::precise::rint()']]], + ['rms_5fnorm_63',['rms_norm',['../namespacemlx_1_1core_1_1fast.html#ac7b620275c6386f822b7aacc6b312e62',1,'mlx::core::fast']]], + ['rmsnorm_64',['RMSNorm',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a22adaff0749711263388ec151fcfebe2',1,'mlx::core::fast::RMSNorm']]], + ['rmsnormvjp_65',['RMSNormVJP',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#aac060129b2e1af79bf388bfe705381ca',1,'mlx::core::fast::RMSNormVJP']]], + ['roll_66',['roll',['../group__ops.html#gac40e48c69f9c715a767912c30836e75c',1,'mlx::core::roll(const array &a, int shift, StreamOrDevice s={})'],['../group__ops.html#ga5011d1a5735c64e5b91afa56c7e2cc02',1,'mlx::core::roll(const array &a, const Shape &shift, StreamOrDevice s={})'],['../group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436',1,'mlx::core::roll(const array &a, int shift, int axis, StreamOrDevice s={})'],['../group__ops.html#ga665f502ecc96f1f4467556b784abf9ae',1,'mlx::core::roll(const array &a, int shift, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga79137f90bc44ac9e35f408c012701df9',1,'mlx::core::roll(const array &a, const Shape &shift, int axis, StreamOrDevice s={})'],['../group__ops.html#ga9d76930fb567a7d459ff96fb851abe36',1,'mlx::core::roll(const array &a, const Shape &shift, const std::vector< int > &axes, StreamOrDevice s={})']]], + ['rope_67',['RoPE',['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a60b399d7f38c0f5f50342a6b97f0eb1a',1,'mlx::core::fast::RoPE']]], + ['rope_68',['rope',['../namespacemlx_1_1core_1_1fast.html#a534ef357eae24892684a6ecd866d3fab',1,'mlx::core::fast::rope(const array &x, int dims, bool traditional, std::optional< float > base, float scale, int offset, const std::optional< array > &freqs=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fast.html#a1632b78950f0c8c31b24be7d80faeb39',1,'mlx::core::fast::rope(const array &x, int dims, bool traditional, std::optional< float > base, float scale, const array &offset, const std::optional< array > &freqs=std::nullopt, StreamOrDevice s={})']]], + ['rot90_69',['ROT90',['../namespacepocketfft_1_1detail.html#a928bad5278df636ee47402c0a75f64ef',1,'pocketfft::detail']]], + ['rotx90_70',['ROTX90',['../namespacepocketfft_1_1detail.html#ab6a43dc0cec4291e163e68a0875ac501',1,'pocketfft::detail']]], + ['round_71',['Round',['../classmlx_1_1core_1_1_round.html#a1327a359b2aed91f576145a0e70d1dde',1,'mlx::core::Round']]], + ['round_72',['round',['../namespacemetal.html#a46c667e169ff9d51a9204a045305442f',1,'metal::round()'],['../namespacemetal_1_1fast.html#a4cb687257a004726d49e496417eaa40f',1,'metal::fast::round()'],['../namespacemetal_1_1precise.html#a5295ab08055d12534cc3775da855ac12',1,'metal::precise::round()'],['../group__ops.html#ga2d74d43f007a069384e89d8416525331',1,'mlx::core::round(const array &a, int decimals, StreamOrDevice s={})'],['../group__ops.html#gaf18fb7e98bf8cf3b7fbc5e64c988a95b',1,'mlx::core::round(const array &a, StreamOrDevice s={})']]], + ['round_5ferror_73',['round_error',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#afa223448fa4f04c1113a85345dd720c3',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['row_5fbin_5fop_74',['row_bin_op',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a318c4279bdc7b39b7919f108b1cd8010',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::row_bin_op()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a3d0d5b9c7962658cc6d5afbbbb2f19e2',1,'mlx::steel::MMATile::row_bin_op()']]], + ['row_5freduce_75',['row_reduce',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a51d662e4cff88b5ad17d7c44bb6b6970',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::row_reduce()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa0ad5cb750ace934bf230385d8bd9f88',1,'mlx::steel::MMATile::row_reduce()']]], + ['row_5freduce_5fgeneral_5fdispatch_76',['row_reduce_general_dispatch',['../namespacemlx_1_1core.html#ab1eeca8ec6fa31819ee108fa6ed2c41b',1,'mlx::core']]], + ['row_5freduce_5flooped_77',['row_reduce_looped',['../reduce__row_8h.html#a72611b8006ae5642b69f4d250d69865b',1,'reduce_row.h']]], + ['row_5freduce_5fsimple_78',['row_reduce_simple',['../reduce__row_8h.html#a7dbd7cc81b1ce5a4271d56b99ce595d4',1,'reduce_row.h']]], + ['row_5freduce_5fsmall_79',['row_reduce_small',['../reduce__row_8h.html#a1e1b59fa73d2b0be978494a759f2c6ee',1,'reduce_row.h']]], + ['rsqrt_80',['rsqrt',['../namespacemlx_1_1core_1_1simd.html#aea75ddf8c696efc2e5e924667ed48e70',1,'mlx::core::simd::rsqrt(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a74ac0fd799967b0f303bfd26fc6a17cf',1,'mlx::core::simd::rsqrt(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a3345cb53830d1afd625acc7bdc3a0435',1,'mlx::core::simd::rsqrt(Simd< float16_t, N > a)'],['../namespacemetal.html#a1cf4b605c0aa7ff5bfe5e979a16f5157',1,'metal::rsqrt()'],['../namespacemetal_1_1fast.html#aa62097c750f1e4b69d09277f19976ab1',1,'metal::fast::rsqrt()'],['../namespacemetal_1_1precise.html#afb397b477745f12a44423934fa2b05ac',1,'metal::precise::rsqrt()'],['../group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1',1,'mlx::core::rsqrt()']]], + ['run_81',['run',['../struct_g_e_m_v_kernel.html#ac4a7b5011a0ea938ab1949bb1767fc1a',1,'GEMVKernel::run()'],['../struct_g_e_m_v_t_kernel.html#a5d68656832de892f33db939005713927',1,'GEMVTKernel::run()'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run(const device T *A, const device T *B, device U *D, const constant GEMMParams *params, threadgroup T *As, threadgroup T *Bs, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run(const device T *A, const device T *B, device U *D, const constant GEMMParams *params, threadgroup T *As, threadgroup T *Bs, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)']]] ]; diff --git a/docs/build/html/search/functions_13.js b/docs/build/html/search/functions_13.js index 5ce820f30..f23f2da40 100644 --- a/docs/build/html/search/functions_13.js +++ b/docs/build/html/search/functions_13.js @@ -4,147 +4,158 @@ var searchData= ['save_1',['save',['../namespacemlx_1_1core.html#ad4c2cebe9e54582295f98c5a448a1f32',1,'mlx::core::save(std::shared_ptr< io::Writer > out_stream, array a)'],['../namespacemlx_1_1core.html#a22a37f3e33e0658680f6227bdd2d0b91',1,'mlx::core::save(std::string file, array a)']]], ['save_5fgguf_2',['save_gguf',['../namespacemlx_1_1core.html#a8bcc29ca8846ec99dce333df4a34dc5f',1,'mlx::core']]], ['save_5fsafetensors_3',['save_safetensors',['../namespacemlx_1_1core.html#a9f158db20c2405557f3ebc397e876de8',1,'mlx::core::save_safetensors(std::shared_ptr< io::Writer > in_stream, std::unordered_map< std::string, array >, std::unordered_map< std::string, std::string > metadata={})'],['../namespacemlx_1_1core.html#a21e256d852d587bcdc0827831b2c5c16',1,'mlx::core::save_safetensors(std::string file, std::unordered_map< std::string, array >, std::unordered_map< std::string, std::string > metadata={})']]], - ['scaled_5fdot_5fproduct_5fattention_4',['scaled_dot_product_attention',['../namespacemlx_1_1core_1_1fast.html#a3663b50265b0a9c0cca2b5376852e059',1,'mlx::core::fast']]], - ['scaleddotproductattention_5',['ScaledDotProductAttention',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ab3f78d30e5bb3e76cfe701f2358e4748',1,'mlx::core::fast::ScaledDotProductAttention']]], - ['scan_6',['Scan',['../classmlx_1_1core_1_1_scan.html#ac93e8f9c6771de825d2186ef34fa7087',1,'mlx::core::Scan']]], - ['scan_7',['scan',['../namespacemlx_1_1core_1_1metal.html#a81c2cf124b0803098a54a78f8f6873a6',1,'mlx::core::metal']]], - ['scatter_8',['Scatter',['../classmlx_1_1core_1_1_scatter.html#ac9b3eff67389ef9aa820753379ffeaa3',1,'mlx::core::Scatter']]], - ['scatter_9',['scatter',['../namespacemlx_1_1core_1_1metal.html#a32e902c6cd6d35fcc3119ed6685a170f',1,'mlx::core::metal::scatter()'],['../group__ops.html#gad438be8f90bae9d37c6853b8f4225d61',1,'mlx::core::scatter(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gac2c2b379a3ce959dbe1c4a68f112edfe',1,'mlx::core::scatter(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scatter_5fadd_10',['scatter_add',['../group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174',1,'mlx::core::scatter_add(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gac13318518e5703f1273c5366eb523a5a',1,'mlx::core::scatter_add(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scatter_5fimpl_11',['scatter_impl',['../scatter_8h.html#ab72d4fe1dbd4ae4dc529ee2ec8164fa4',1,'scatter.h']]], - ['scatter_5fmax_12',['scatter_max',['../group__ops.html#ga05881a4157cd113c9392d168a79e6673',1,'mlx::core::scatter_max(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga9adda5f9202bb3486e4d9e1114e3a56f',1,'mlx::core::scatter_max(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scatter_5fmin_13',['scatter_min',['../group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5',1,'mlx::core::scatter_min(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga51fa762a997c243ca7a19e1ed3e83199',1,'mlx::core::scatter_min(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scatter_5fprod_14',['scatter_prod',['../group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0',1,'mlx::core::scatter_prod(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gaf83c53c453faa9083ba27e4b97539339',1,'mlx::core::scatter_prod(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], - ['scheduler_15',['Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a3ae42aed78a2200e9d02776fcd2316ba',1,'mlx::core::scheduler::Scheduler::Scheduler()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a61a74e3628899e66dde600e24a750648',1,'mlx::core::scheduler::Scheduler::Scheduler(const Scheduler &)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ac3f77b7c93220dadd0b3bb2e903b7059',1,'mlx::core::scheduler::Scheduler::Scheduler(Scheduler &&)=delete']]], - ['scheduler_16',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html#ae856e468c2f7c8f8ec672522cc13730b',1,'mlx::core::scheduler']]], - ['sdpa_5fvector_17',['sdpa_vector',['../sdpa__vector_8h.html#aee8f6015c5b4fc81cde3c5edf30af8af',1,'sdpa_vector.h']]], - ['sdpa_5fvector_5f2pass_5f1_18',['sdpa_vector_2pass_1',['../sdpa__vector_8h.html#adef3c1f2f7ec098b0ef43f5f5bfeab0e',1,'sdpa_vector.h']]], - ['sdpa_5fvector_5f2pass_5f2_19',['sdpa_vector_2pass_2',['../sdpa__vector_8h.html#a1368cf3618a4e03dbf743b3463205efe',1,'sdpa_vector.h']]], - ['seed_20',['seed',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a9f19c5da2031cba50d0ff996924347d8',1,'mlx::core::random::KeySequence::seed()'],['../namespacemlx_1_1core_1_1random.html#ac4ad325b613257306df74595d3d0e23b',1,'mlx::core::random::seed()']]], - ['seek_21',['seek',['../structmlx_1_1core_1_1_contiguous_iterator.html#af08f009e0a72414d274db2ff1b2c7dd5',1,'mlx::core::ContiguousIterator::seek()'],['../classmlx_1_1core_1_1io_1_1_reader.html#acea55078bd39ccaa27a9a36f17a39cd1',1,'mlx::core::io::Reader::seek()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a9c1716dda53aa36faea9c8fb1a3e34d4',1,'mlx::core::io::Writer::seek()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a4434ee18ff8bbf1b4fce670a337b535f',1,'mlx::core::io::ParallelFileReader::seek()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9646f4ea048ae58719daeb588e2de433',1,'mlx::core::io::FileWriter::seek()']]], - ['select_22',['Select',['../classmlx_1_1core_1_1_select.html#a6f833fe55dd68ad3726bbf9a8f75eec9',1,'mlx::core::Select']]], - ['send_23',['Send',['../classmlx_1_1core_1_1distributed_1_1_send.html#a2481dd876b14d4a13ac466cbca9c4eac',1,'mlx::core::distributed::Send']]], - ['send_24',['send',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac8472eb2f96d1b14c7e4ccef56268ba0',1,'mlx::core::distributed::detail::GroupImpl::send()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#abf33511660ac71df5fc92f2aad6c6e08',1,'mlx::core::distributed::detail::send()'],['../namespacemlx_1_1core_1_1distributed.html#a5a8360edaa3a528a3927fce4d2cf1777',1,'mlx::core::distributed::send()']]], - ['set_25',['Set',['../structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807',1,'pocketfft::detail::cmplx::Set(T r_, T i_)'],['../structpocketfft_1_1detail_1_1cmplx.html#a447d26b2e07f6e45f29d865e906c0a98',1,'pocketfft::detail::cmplx::Set(T r_)']]], - ['set_5fbytes_26',['set_bytes',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a9c343f791812a45c6c03a5c9f27f74d5',1,'mlx::core::metal::CommandEncoder::set_bytes(const T *v, int n, int idx)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#abc52d18ea87d213c47fd26062c829849',1,'mlx::core::metal::CommandEncoder::set_bytes(const T &v, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a9c343f791812a45c6c03a5c9f27f74d5',1,'mlx::core::CommandEncoder::set_bytes(const T *v, int n, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#abc52d18ea87d213c47fd26062c829849',1,'mlx::core::CommandEncoder::set_bytes(const T &v, int idx)']]], - ['set_5fcache_5flimit_27',['set_cache_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#af392bced29d9e4e3f1a7cc4725d83764',1,'mlx::core::metal::MetalAllocator::set_cache_limit()'],['../namespacemlx_1_1core_1_1metal.html#ab09c9b60f1e886ab859e6a066c9a5b9d',1,'mlx::core::metal::set_cache_limit()']]], - ['set_5fcompile_5fmode_28',['set_compile_mode',['../namespacemlx_1_1core.html#a49445a55f976c4397f25ea18e1e92bef',1,'mlx::core']]], - ['set_5fcompute_5fpipeline_5fstate_29',['set_compute_pipeline_state',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a6d4c03a6585deedb5ccd1a1057d0c6ef',1,'mlx::core::metal::CommandEncoder::set_compute_pipeline_state()'],['../structmlx_1_1core_1_1_command_encoder.html#a6d4c03a6585deedb5ccd1a1057d0c6ef',1,'mlx::core::CommandEncoder::set_compute_pipeline_state()']]], - ['set_5fdata_30',['set_data',['../classmlx_1_1core_1_1array.html#af9e3a02b4c0023c36248dc75c887214f',1,'mlx::core::array::set_data(allocator::Buffer buffer, Deleter d=allocator::free)'],['../classmlx_1_1core_1_1array.html#a5f338202a39d37fa3f4241e851a15838',1,'mlx::core::array::set_data(allocator::Buffer buffer, size_t data_size, Strides strides, Flags flags, Deleter d=allocator::free)']]], - ['set_5fdefault_5fdevice_31',['set_default_device',['../namespacemlx_1_1core.html#a312a2de41367fe52caeaf8c0f596a120',1,'mlx::core']]], - ['set_5fdefault_5fstream_32',['set_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a6d15314ac9cf25efc9bd1278de9a66bb',1,'mlx::core::scheduler::Scheduler::set_default_stream()'],['../namespacemlx_1_1core.html#af35a2b06517d8bb7dbb469692b4f841c',1,'mlx::core::set_default_stream()']]], - ['set_5finput_5farray_33',['set_input_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ab69ff0d7f14b9b59db4df0608193dce4',1,'mlx::core::metal::CommandEncoder::set_input_array()'],['../structmlx_1_1core_1_1_command_encoder.html#ab69ff0d7f14b9b59db4df0608193dce4',1,'mlx::core::CommandEncoder::set_input_array()']]], - ['set_5fmemory_5flimit_34',['set_memory_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a179e3127ef9377ce54295f771c34ba1b',1,'mlx::core::metal::MetalAllocator::set_memory_limit()'],['../namespacemlx_1_1core_1_1metal.html#a3fb2c4a237fa4bfdff798156146c4937',1,'mlx::core::metal::set_memory_limit()']]], - ['set_5fname_35',['set_name',['../structmlx_1_1core_1_1_node_namer.html#a57a574e48f8a9cd122616d80b138c768',1,'mlx::core::NodeNamer']]], - ['set_5foutput_5farray_36',['set_output_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a6a2e28e542eaa2886041bddd51ff6522',1,'mlx::core::metal::CommandEncoder::set_output_array()'],['../structmlx_1_1core_1_1_command_encoder.html#a6a2e28e542eaa2886041bddd51ff6522',1,'mlx::core::CommandEncoder::set_output_array()']]], - ['set_5fresidency_5fset_37',['set_residency_set',['../classmlx_1_1core_1_1metal_1_1_device.html#a03a2f0c712660a1bd437cb16e4aba79f',1,'mlx::core::metal::Device']]], - ['set_5fsiblings_38',['set_siblings',['../classmlx_1_1core_1_1array.html#a8fccbe7a4edfd8cca168161124e263b1',1,'mlx::core::array']]], - ['set_5fstatus_39',['set_status',['../classmlx_1_1core_1_1array.html#a63598018999b49f1340b183cb303f05c',1,'mlx::core::array']]], - ['set_5ftracer_40',['set_tracer',['../classmlx_1_1core_1_1array.html#af26e6be1a9e6239471a4c24310c0c7c8',1,'mlx::core::array']]], - ['set_5fvalue_41',['set_value',['../classmlx_1_1core_1_1_event.html#a0d077b11f4b28f882b42440b7ac6d40d',1,'mlx::core::Event']]], - ['set_5fvector_5fbytes_42',['set_vector_bytes',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a68c3c6a036e11ec40211c09811bbed1b',1,'mlx::core::metal::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, size_t nelems, int idx)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a7375adf9ee5355bcf4b7f5f210efd115',1,'mlx::core::metal::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a68c3c6a036e11ec40211c09811bbed1b',1,'mlx::core::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, size_t nelems, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a7375adf9ee5355bcf4b7f5f210efd115',1,'mlx::core::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, int idx)']]], - ['set_5fwired_5flimit_43',['set_wired_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a84fa0347da18055bc13ba0a5c4b57253',1,'mlx::core::metal::MetalAllocator::set_wired_limit()'],['../namespacemlx_1_1core_1_1metal.html#a31eab4828d31d292bc84e07b0d961e1e',1,'mlx::core::metal::set_wired_limit()']]], - ['shape_44',['shape',['../classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0',1,'pocketfft::detail::arr_info::shape() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8',1,'pocketfft::detail::arr_info::shape(size_t i) const'],['../classmlx_1_1core_1_1array.html#a1d06c76b0f3010a5c329d0e9e29e0597',1,'mlx::core::array::shape() const'],['../classmlx_1_1core_1_1array.html#ac50382b652f6e8fbd50d42b7ff595810',1,'mlx::core::array::shape(int dim) const']]], - ['shape2d_45',['Shape2D',['../structmlx_1_1steel_1_1_shape2_d.html#a070ce70eb6d84361c7f313159c438a5c',1,'mlx::steel::Shape2D']]], - ['shapes_5fwithout_5freduction_5faxes_46',['shapes_without_reduction_axes',['../namespacemlx_1_1core.html#a0bea91a360a984e72d2815353f97ee25',1,'mlx::core']]], - ['shared_5fbuffer_5freshape_47',['shared_buffer_reshape',['../namespacemlx_1_1core.html#a88d88987bd8bf3ca46bf3b5e8aacce9d',1,'mlx::core']]], - ['shared_5fbuffer_5fslice_48',['shared_buffer_slice',['../namespacemlx_1_1core.html#a349a9fc2bfd950f679a3fe39b8bdedad',1,'mlx::core']]], - ['shutdown_49',['shutdown',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411',1,'pocketfft::detail::threading::thread_pool']]], - ['siblings_50',['siblings',['../classmlx_1_1core_1_1array.html#acf80fde8f743f65ad5b4be69fcb7a74d',1,'mlx::core::array::siblings() const'],['../classmlx_1_1core_1_1array.html#a7263f23e70a580a9bc2129fbcde36e6c',1,'mlx::core::array::siblings()']]], - ['sigmoid_51',['Sigmoid',['../classmlx_1_1core_1_1_sigmoid.html#a47eca99113ec19f0eb60b6a0472c592b',1,'mlx::core::Sigmoid']]], - ['sigmoid_52',['sigmoid',['../group__ops.html#ga708abf8f79609cd6831db7c38cafac0e',1,'mlx::core']]], - ['sign_53',['Sign',['../classmlx_1_1core_1_1_sign.html#afe951e50907bc23a601ec5fa9eae5763',1,'mlx::core::Sign']]], - ['sign_54',['sign',['../group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa',1,'mlx::core']]], - ['signal_55',['signal',['../classmlx_1_1core_1_1_event.html#a65a858445506a61be5889ae0e3651b89',1,'mlx::core::Event']]], - ['signaling_5fnan_56',['signaling_NaN',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ad1f76a43c7d51a3765174aa6e0dd9f80',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['simd_5fbroadcast_57',['simd_broadcast',['../namespacemetal.html#a498f1e85107eb5f01ba4435977f8efe0',1,'metal']]], - ['simd_5fexclusive_5fscan_58',['simd_exclusive_scan',['../struct_cum_prod.html#a1a86e9398bae24182b7be0a6577bf223',1,'CumProd< bool >::simd_exclusive_scan()'],['../struct_cum_max.html#ae11b67aa6c998e9a01615b2a79af4403',1,'CumMax::simd_exclusive_scan()'],['../struct_cum_min.html#a83e65017ff33018b585c043fb803773b',1,'CumMin::simd_exclusive_scan()']]], - ['simd_5fmax_59',['simd_max',['../namespacemetal.html#a048cad0aca52cb737ebf103e76bd1c49',1,'metal']]], - ['simd_5fmin_60',['simd_min',['../namespacemetal.html#ae9e2a23e00724ba2d7868bc4112b386b',1,'metal']]], - ['simd_5fprefix_5fexclusive_5fproduct_61',['simd_prefix_exclusive_product',['../namespacemetal.html#a5ca40242390b632f737e29636829b2e4',1,'metal']]], - ['simd_5fprefix_5fexclusive_5fsum_62',['simd_prefix_exclusive_sum',['../namespacemetal.html#abfbb70c7471f28bf7ff36a612ad014b2',1,'metal']]], - ['simd_5fprefix_5finclusive_5fproduct_63',['simd_prefix_inclusive_product',['../namespacemetal.html#a6ca6a7e1996228fa536e969e9e45c446',1,'metal']]], - ['simd_5fprefix_5finclusive_5fsum_64',['simd_prefix_inclusive_sum',['../namespacemetal.html#a567acb18199ac0107712eb8cb8aeb8e9',1,'metal']]], - ['simd_5fproduct_65',['simd_product',['../namespacemetal.html#ac6e883a04e2265a9790d7db76059e1b4',1,'metal']]], - ['simd_5fscan_66',['simd_scan',['../struct_cum_prod.html#abeb5ec4237b330e7219f4e881cf10d7a',1,'CumProd< bool >::simd_scan()'],['../struct_cum_max.html#adc9ec8bb09b4433d4c2f03022c43d781',1,'CumMax::simd_scan()'],['../struct_cum_min.html#a0a1005d91b1c90e90e2c6dbd6c296649',1,'CumMin::simd_scan()']]], - ['simd_5fshuffle_67',['simd_shuffle',['../namespacemetal.html#a259ed115bc3c58f88eb35830916b26d4',1,'metal::simd_shuffle()'],['../backend_2metal_2kernels_2utils_8h.html#a71986ecdd7d18f975dd22c3df7421ce2',1,'simd_shuffle(uint64_t data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a3bdbdfeb7a1dde40cd3ce1df8d9213b5',1,'simd_shuffle(int64_t data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ab4cbcdb054f9165130da91a3334da0cf',1,'simd_shuffle(bool data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ab8175b66bcc080fb89f738143568c30b',1,'simd_shuffle(complex64_t data, uint16_t lane): utils.h']]], - ['simd_5fshuffle_5fand_5ffill_5fdown_68',['simd_shuffle_and_fill_down',['../namespacemetal.html#ae29a06f0eac636ad7af21dea5b04938b',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a0ee6239fa29a5f9ee0201e0dc5ddc8e0',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], - ['simd_5fshuffle_5fand_5ffill_5fup_69',['simd_shuffle_and_fill_up',['../namespacemetal.html#a1ca14116bf50639b214d8414b5bbaaa6',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a5138d5cdc18139e135707916a243cd8e',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta)'],['../backend_2metal_2kernels_2utils_8h.html#a5862d5ea154c9b76cf56a630cf6385b4',1,'simd_shuffle_and_fill_up(uint64_t data, uint64_t filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a7bb56415c5412a6a26f70a990915f064',1,'simd_shuffle_and_fill_up(int64_t data, int64_t filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ad55bd473647f2c6c68e65e5312c132d1',1,'simd_shuffle_and_fill_up(bool data, bool filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a94e02a6ae8c39cbf4cb23aa44df9dbd5',1,'simd_shuffle_and_fill_up(complex64_t data, complex64_t filling, uint16_t delta): utils.h']]], - ['simd_5fshuffle_5fdown_70',['simd_shuffle_down',['../namespacemetal.html#af6e2dd7ae087aba6abac4f0350b7611c',1,'metal::simd_shuffle_down()'],['../backend_2metal_2kernels_2utils_8h.html#aba6279624b1d30c525efee856a222b5c',1,'simd_shuffle_down(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a0c1e4d782fcc56e1ab5565cef12430dd',1,'simd_shuffle_down(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a48ae83a8caf5c74810df60b6c6cdb062',1,'simd_shuffle_down(bool data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ad9a671a5f9aaa729ae7a77026f16bcb0',1,'simd_shuffle_down(complex64_t data, uint16_t delta): utils.h']]], - ['simd_5fshuffle_5frotate_5fdown_71',['simd_shuffle_rotate_down',['../namespacemetal.html#a4bb203647a421032db47e73cd649841b',1,'metal']]], - ['simd_5fshuffle_5frotate_5fup_72',['simd_shuffle_rotate_up',['../namespacemetal.html#a729b22077d6c944491a6027c18ea80c9',1,'metal']]], - ['simd_5fshuffle_5fup_73',['simd_shuffle_up',['../namespacemetal.html#afe81c5fbde3f4890458b081909242c55',1,'metal::simd_shuffle_up()'],['../backend_2metal_2kernels_2utils_8h.html#a39e436e0a942912266aae7e0bd82d7c0',1,'simd_shuffle_up(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a617f3857caf33c569afa6148135f8b7a',1,'simd_shuffle_up(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ae0f5c42020275a588234e69f1eb7a485',1,'simd_shuffle_up(bool data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a92b455bac6a23af51c35ea83de2383eb',1,'simd_shuffle_up(complex64_t data, uint16_t delta): utils.h']]], - ['simd_5fshuffle_5fxor_74',['simd_shuffle_xor',['../namespacemetal.html#a5017efc9605e069cfb507137cd1a1852',1,'metal']]], - ['simd_5fsum_75',['simd_sum',['../namespacemetal.html#a85181e37a00cb4a4217f1bb25389bce5',1,'metal']]], - ['simd_5fxor_76',['simd_xor',['../namespacemetal.html#a1308decbf2d5c33d34d6be523ea1c30f',1,'metal']]], - ['simple_5fiter_77',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398',1,'pocketfft::detail::simple_iter']]], - ['sin_78',['Sin',['../classmlx_1_1core_1_1_sin.html#a10d1ecc0ca96e79cdf55b57073d126ea',1,'mlx::core::Sin']]], - ['sin_79',['sin',['../namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258',1,'pocketfft::detail::sin()'],['../namespacemetal.html#a619a159ca5f2ddfe3647d3a6bb6e804c',1,'metal::sin()'],['../namespacemetal_1_1fast.html#a3af771cfe7a135104f9d063147dba270',1,'metal::fast::sin()'],['../namespacemetal_1_1precise.html#a71acf77ffd29c56f56afae0195c98a1c',1,'metal::precise::sin()'],['../group__ops.html#gaebf0a73ad3732fba39df37826c235692',1,'mlx::core::sin()']]], - ['sincos_5f2pibyn_80',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40',1,'pocketfft::detail::sincos_2pibyn']]], - ['sinh_81',['Sinh',['../classmlx_1_1core_1_1_sinh.html#a4a4f6814d403c2ce5d6c574b0dca3c96',1,'mlx::core::Sinh']]], - ['sinh_82',['sinh',['../namespacemetal.html#a83ba4235ae350ab8880a9df09158620b',1,'metal::sinh()'],['../namespacemetal_1_1fast.html#a990d90b3440e38d1fb4ff5065c6c189b',1,'metal::fast::sinh()'],['../namespacemetal_1_1precise.html#abc8f4f59dd6e7204ab5d84f0af96331c',1,'metal::precise::sinh()'],['../group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7',1,'mlx::core::sinh()']]], - ['sinpi_83',['sinpi',['../namespacemetal.html#ae9655f7fa2ba6c0625ca25fbb278e269',1,'metal::sinpi()'],['../namespacemetal_1_1fast.html#ab07a32fe544aa304577d29e0251e87b2',1,'metal::fast::sinpi()'],['../namespacemetal_1_1precise.html#a78b17dab93519d9c82c2575dafec49c9',1,'metal::precise::sinpi()']]], - ['size_84',['size',['../classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952',1,'pocketfft::detail::arr::size()'],['../classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5',1,'pocketfft::detail::arr_info::size()'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a2adf9a9c968f113dde830cc0dc27dcc6',1,'mlx::core::allocator::Allocator::size()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#aafa92e8310db089b1ac72b840777e26b',1,'mlx::core::allocator::CommonAllocator::size()'],['../classmlx_1_1core_1_1array.html#a598f87161926d9e0b516860f0ea2c8f6',1,'mlx::core::array::size()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a51f6587e8065be16f0418ca42a796e05',1,'mlx::core::metal::MetalAllocator::size()'],['../structmlx_1_1core_1_1distributed_1_1_group.html#ac0a5a1e463a9330355e8bfe09c0feaf2',1,'mlx::core::distributed::Group::size()'],['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ab1c8044b05f185c4bcc53002d4587599',1,'mlx::core::distributed::detail::GroupImpl::size()'],['../structmlx_1_1core_1_1_dtype.html#ab54051563d85212c7f0f049166bc9971',1,'mlx::core::Dtype::size()']]], - ['size_5fof_85',['size_of',['../namespacemlx_1_1core.html#add4794cc0ffe5d717fc146084a235d95',1,'mlx::core']]], - ['slice_86',['Slice',['../classmlx_1_1core_1_1_slice.html#a3aa025acbf4a9ca9e030a1e6bda102f7',1,'mlx::core::Slice']]], - ['slice_87',['slice',['../group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d',1,'mlx::core::slice(const array &a, Shape start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#gaedcbdf4040f5e6a02a74b1ed1c6c2ebc',1,'mlx::core::slice(const array &a, std::initializer_list< int > start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#gaec56dcb94d5e7f7b885fb60b4bf4aa9d',1,'mlx::core::slice(const array &a, Shape start, Shape stop, StreamOrDevice s={})'],['../group__ops.html#ga797996e53ea34317a55dc2f314736b89',1,'mlx::core::slice(const array &a, const array &start, std::vector< int > axes, Shape slice_size, StreamOrDevice s={})']]], - ['slice_5fgpu_88',['slice_gpu',['../namespacemlx_1_1core.html#a327578951a44116e5da2db651661265f',1,'mlx::core']]], - ['slice_5fupdate_89',['slice_update',['../group__ops.html#ga7b7e786985d27789aaed20bb2f9509be',1,'mlx::core::slice_update(const array &src, const array &update, Shape start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#ga3b7c6136ae5a38dd6457b65c1833aa67',1,'mlx::core::slice_update(const array &src, const array &update, Shape start, Shape stop, StreamOrDevice s={})'],['../group__ops.html#gacd906ffb96149a4998c321cbf2231d7e',1,'mlx::core::slice_update(const array &src, const array &update, const array &start, std::vector< int > axes, StreamOrDevice s={})']]], - ['sliceupdate_90',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html#a63a49264b18340f2bc442c081a7b4c7a',1,'mlx::core::SliceUpdate']]], - ['softmax_91',['Softmax',['../classmlx_1_1core_1_1_softmax.html#a4ec686aac4e06f0dfe2cbd6801af40eb',1,'mlx::core::Softmax']]], - ['softmax_92',['softmax',['../namespacemlx_1_1core_1_1metal.html#a4fe937c2c584fd646926057f31d54ca6',1,'mlx::core::metal::softmax()'],['../group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6',1,'mlx::core::softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga1ae3614d07d873892a530d14c3857d0b',1,'mlx::core::softmax(const array &a, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga06f570d73716a24303e6de3aaba4457b',1,'mlx::core::softmax(const array &a, int axis, bool precise=false, StreamOrDevice s={})']]], - ['softmax_5fexp_93',['softmax_exp',['../kernels_2softmax_8h.html#a440d4031ee5e86159a4dd715e44a438b',1,'softmax.h']]], - ['softmax_5flooped_94',['softmax_looped',['../kernels_2softmax_8h.html#a8c47b0924ebfeebcca25f3dd17373276',1,'softmax.h']]], - ['softmax_5fsingle_5frow_95',['softmax_single_row',['../kernels_2softmax_8h.html#a815fe70f879f318e5d6e99acf043f52b',1,'softmax.h']]], - ['sort_96',['Sort',['../classmlx_1_1core_1_1_sort.html#a62943032dbd72e85ceb9b4b7211f4a44',1,'mlx::core::Sort']]], - ['sort_97',['sort',['../struct_thread_sort.html#ad9ab3e6b47f7e9b91c0f3b773596986d',1,'ThreadSort::sort()'],['../struct_block_merge_sort.html#acc970f5eb963f7f2010f5ae5ea8b8bc0',1,'BlockMergeSort::sort()'],['../namespacemlx_1_1core_1_1metal.html#ab77c9a9ecaeeab8c66b712862777c24b',1,'mlx::core::metal::sort()'],['../group__ops.html#ga7fb616054665b3c2d61fa234f501f079',1,'mlx::core::sort(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaae1bc47aa737f705d0e5884270063fea',1,'mlx::core::sort(const array &a, int axis, StreamOrDevice s={})']]], - ['special_5fmul_98',['special_mul',['../structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874',1,'pocketfft::detail::cmplx::special_mul()'],['../namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76',1,'pocketfft::detail::special_mul()']]], - ['split_99',['Split',['../classmlx_1_1core_1_1_split.html#ad3f4ed34b85c73683bad5d530309342f',1,'mlx::core::Split']]], - ['split_100',['split',['../structmlx_1_1core_1_1distributed_1_1_group.html#ad3682f4dc85bfe7e5464b87f6f0fd931',1,'mlx::core::distributed::Group::split()'],['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a87800a23c8160933a2d77a55a959194d',1,'mlx::core::distributed::detail::GroupImpl::split()'],['../group__ops.html#ga7534290bceab5fb3831a05d67bebce7d',1,'mlx::core::split(const array &a, int num_splits, int axis, StreamOrDevice s={})'],['../group__ops.html#ga56882d24e5fde59c266774624c892d41',1,'mlx::core::split(const array &a, int num_splits, StreamOrDevice s={})'],['../group__ops.html#ga19005414e7d8020cd6e94e06bf399b09',1,'mlx::core::split(const array &a, const Shape &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#ga9ea089f42b9940510619052b7166d9ac',1,'mlx::core::split(const array &a, const Shape &indices, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a42847b435d037a977592e355eed072af',1,'mlx::core::random::split(const array &key, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7ec057064c7326c41b536f08178861e5',1,'mlx::core::random::split(const array &key, int num, StreamOrDevice s={})']]], - ['sqrt_101',['Sqrt',['../classmlx_1_1core_1_1_sqrt.html#a6682a7c31ca427c9d2c5ddb6a479bf29',1,'mlx::core::Sqrt']]], - ['sqrt_102',['sqrt',['../namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3',1,'pocketfft::detail::sqrt()'],['../namespacemetal.html#ab3f4d4852ca0e591104fbd8e5b50d31b',1,'metal::sqrt()'],['../namespacemetal_1_1fast.html#a4218a85c7d8a74cb8055b4755205627e',1,'metal::fast::sqrt()'],['../namespacemetal_1_1precise.html#acb213467361cd2cab93a8d5ea1aa5bfd',1,'metal::precise::sqrt()'],['../group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1',1,'mlx::core::sqrt()']]], - ['square_103',['Square',['../classmlx_1_1core_1_1_square.html#ab94e28d5c92e6febc1c74e525f730dc4',1,'mlx::core::Square']]], - ['square_104',['square',['../group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e',1,'mlx::core']]], - ['squeeze_105',['Squeeze',['../classmlx_1_1core_1_1_squeeze.html#a032bd53dcc3d71a11d810bc3ca3ef4b0',1,'mlx::core::Squeeze']]], - ['squeeze_106',['squeeze',['../group__ops.html#ga710daa7ec721bd4d3f326082cb195576',1,'mlx::core::squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3',1,'mlx::core::squeeze(const array &a, int axis, StreamOrDevice s={})'],['../group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad',1,'mlx::core::squeeze(const array &a, StreamOrDevice s={})']]], - ['stack_107',['stack',['../group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032',1,'mlx::core::stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga82216209dce901296fc737fe8efa5c94',1,'mlx::core::stack(const std::vector< array > &arrays, StreamOrDevice s={})']]], - ['start_5fcapture_108',['start_capture',['../namespacemlx_1_1core_1_1metal.html#aa47cb5651bf3b65c46ab216b7e504d77',1,'mlx::core::metal']]], - ['start_5fconcurrent_109',['start_concurrent',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::metal::CommandEncoder::start_concurrent()'],['../structmlx_1_1core_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::CommandEncoder::start_concurrent()']]], - ['state_110',['state',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a379b27ac336ef351aa81142c5626ad76',1,'mlx::core::fast::RMSNormVJP::state()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a963e672c607b5f86080e6cc32a3cd6e5',1,'mlx::core::fast::LayerNormVJP::state()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#aa5a03284c6f5639d684dd34d86050cf9',1,'mlx::core::fast::AffineQuantize::state()'],['../classmlx_1_1core_1_1_add_m_m.html#a24ab73de46d0589780dac5ade43f93b8',1,'mlx::core::AddMM::state()'],['../classmlx_1_1core_1_1_arange.html#ac4a9f48a11c2af03ed57fdf2422cbfad',1,'mlx::core::Arange::state()'],['../classmlx_1_1core_1_1_arg_partition.html#a5033c46f5aae9b14859cc8b0ca4c8e19',1,'mlx::core::ArgPartition::state()'],['../classmlx_1_1core_1_1_arg_reduce.html#acac3b26364260aac7511b4cb7add3604',1,'mlx::core::ArgReduce::state()'],['../classmlx_1_1core_1_1_arg_sort.html#a90548429765f9e7e2332f01b72692fa2',1,'mlx::core::ArgSort::state()'],['../classmlx_1_1core_1_1_as_type.html#a98ea769fc9cd6d76b07817444e7a78ab',1,'mlx::core::AsType::state()'],['../classmlx_1_1core_1_1_as_strided.html#ae730aeff375498ba774d4207c7af8c36',1,'mlx::core::AsStrided::state()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8a67d6f431b4055ab66656201622af4d',1,'mlx::core::BitwiseBinary::state()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a6bbcc34b256840e4df2953563f2b4a07',1,'mlx::core::BlockMaskedMM::state()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a5b2594b7a70dd4873a07e742140a245f',1,'mlx::core::BroadcastAxes::state()'],['../classmlx_1_1core_1_1_broadcast.html#a49fdb421047860733af7dfbbb478da8d',1,'mlx::core::Broadcast::state()'],['../classmlx_1_1core_1_1_concatenate.html#a60cd572a42b346399ee539af2dfbf29e',1,'mlx::core::Concatenate::state()'],['../classmlx_1_1core_1_1_convolution.html#a7f44f0caea20cc2858717afba1e915d8',1,'mlx::core::Convolution::state()'],['../classmlx_1_1core_1_1_equal.html#aa27ff7525f109edc56b731a6df78f6bc',1,'mlx::core::Equal::state()'],['../classmlx_1_1core_1_1_expand_dims.html#a7cacc704c533c00ba072f0a7872631cf',1,'mlx::core::ExpandDims::state()'],['../classmlx_1_1core_1_1_f_f_t.html#a710c6f6e8412da0af0fdbe58fbae320e',1,'mlx::core::FFT::state()'],['../classmlx_1_1core_1_1_flatten.html#af95dd89c47cd2342233dc0b6d36822a3',1,'mlx::core::Flatten::state()'],['../classmlx_1_1core_1_1_gather.html#aee59ff90127ef4c2d7fcbe2955b95b27',1,'mlx::core::Gather::state()'],['../classmlx_1_1core_1_1_hadamard.html#afd67d09fde38ab3b6ba873b797f03dae',1,'mlx::core::Hadamard::state()'],['../classmlx_1_1core_1_1_log.html#a86fca2ec3766f5d4a2e6d8ba2983c3aa',1,'mlx::core::Log::state()'],['../classmlx_1_1core_1_1_number_of_elements.html#afbfee716b4896e98bdf502ceab87ac09',1,'mlx::core::NumberOfElements::state()'],['../classmlx_1_1core_1_1_pad.html#a00a7cff2ae640f45b43f62cc25d6346c',1,'mlx::core::Pad::state()'],['../classmlx_1_1core_1_1_partition.html#adde13e40924c016473864119465cad4b',1,'mlx::core::Partition::state()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ad83bfd32fda988c29e5ca277a84c0655',1,'mlx::core::QuantizedMatmul::state()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a60c908bc836f930bb33f60b3e9db43af',1,'mlx::core::GatherQMM::state()'],['../classmlx_1_1core_1_1_random_bits.html#a75a34d7541a1c124710dc4d0ec2dfa60',1,'mlx::core::RandomBits::state()'],['../classmlx_1_1core_1_1_reshape.html#aa8ad5958aac8723dd6ce49820eaba029',1,'mlx::core::Reshape::state()'],['../classmlx_1_1core_1_1_reduce.html#afca1398c042a3b1ca9a9a2e40fe62367',1,'mlx::core::Reduce::state()'],['../classmlx_1_1core_1_1_scan.html#a7249ca4c3316b1b1248df32c71fee0ea',1,'mlx::core::Scan::state()'],['../classmlx_1_1core_1_1_scatter.html#a50a65033dc2a1cc84bf529ba718c9c60',1,'mlx::core::Scatter::state()'],['../classmlx_1_1core_1_1_slice.html#a069dafc62bf71e3ebc0bd99d96ec23be',1,'mlx::core::Slice::state()'],['../classmlx_1_1core_1_1_slice_update.html#aac5156a19209274b7de1dff231ef25fd',1,'mlx::core::SliceUpdate::state()'],['../classmlx_1_1core_1_1_dynamic_slice.html#aec9084e603d7562f3a75c5fc32918581',1,'mlx::core::DynamicSlice::state()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a0b0b2a0e4d97305fd6f3c635fcdccd76',1,'mlx::core::DynamicSliceUpdate::state()'],['../classmlx_1_1core_1_1_softmax.html#adf665f7c292e84f56c630016b75427f3',1,'mlx::core::Softmax::state()'],['../classmlx_1_1core_1_1_sort.html#a5ab15d1a89efd8661887c805c35fc617',1,'mlx::core::Sort::state()'],['../classmlx_1_1core_1_1_split.html#a915156cde0448ec26776e329004b1a92',1,'mlx::core::Split::state()'],['../classmlx_1_1core_1_1_sqrt.html#a053853757ad99195e3f2b1cca571e31b',1,'mlx::core::Sqrt::state()'],['../classmlx_1_1core_1_1_squeeze.html#a04f9d2595cb7d4ec988479cd33fe9362',1,'mlx::core::Squeeze::state()'],['../classmlx_1_1core_1_1_unflatten.html#aeba13680064238191811230171365598',1,'mlx::core::Unflatten::state()'],['../classmlx_1_1core_1_1_view.html#a37620f6548630bd2d0dd44e9ab084b93',1,'mlx::core::View::state()'],['../classmlx_1_1core_1_1_transpose.html#a23167291e2bf12e2bb2e51d1db340909',1,'mlx::core::Transpose::state()'],['../classmlx_1_1core_1_1_inverse.html#aa1fce744f4a2d660c65901a7542056f2',1,'mlx::core::Inverse::state()'],['../classmlx_1_1core_1_1_cholesky.html#a64f03d32ed249a3b2a59b6af66d23727',1,'mlx::core::Cholesky::state()'],['../classmlx_1_1core_1_1_eigh.html#aa3b6c33b5679c5528863f3de2ab2f914',1,'mlx::core::Eigh::state()']]], - ['status_111',['status',['../classmlx_1_1core_1_1array.html#a7102659be87e9ef62966696ab9b07dad',1,'mlx::core::array']]], - ['std_112',['std',['../group__ops.html#ga2a466024f8061febc0a64be557644cb0',1,'mlx::core::std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#gafdcb04d77c64405a3990078a77dd984c',1,'mlx::core::std(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8',1,'mlx::core::std(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga62721a206df671ef5797449eea97af9f',1,'mlx::core::std(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], - ['steel_5fconv_113',['steel_conv',['../namespacemlx_1_1core_1_1metal.html#a92f1e559b1121d545746f81ff86eaca1',1,'mlx::core::metal']]], - ['steel_5fconv_5fgeneral_114',['steel_conv_general',['../namespacemlx_1_1core_1_1metal.html#a02edb6a90bdf30f4c9f0d6c25b0267b5',1,'mlx::core::metal']]], - ['steel_5fgemm_5ffused_115',['steel_gemm_fused',['../namespacemlx_1_1core_1_1metal.html#a17764366deed71c160fb26091400a803',1,'mlx::core::metal']]], - ['steel_5fgemm_5fmasked_116',['steel_gemm_masked',['../namespacemlx_1_1core_1_1metal.html#a962272ca73d26c08f76f706a128fd71f',1,'mlx::core::metal']]], - ['steel_5fgemm_5fsplitk_117',['steel_gemm_splitk',['../namespacemlx_1_1core_1_1metal.html#ad0dfd40ba7c09755711ceb731e57a5ac',1,'mlx::core::metal']]], - ['steel_5fmatmul_118',['steel_matmul',['../namespacemlx_1_1core.html#aaff208bbac7021c4265580885874499a',1,'mlx::core']]], - ['steel_5fmatmul_5fregular_119',['steel_matmul_regular',['../namespacemlx_1_1core.html#a2a8a09851097571fb51ac5b608550e44',1,'mlx::core']]], - ['step_120',['step',['../structmlx_1_1core_1_1_contiguous_iterator.html#aad921dd422adb0a0f555e19a2f42239c',1,'mlx::core::ContiguousIterator']]], - ['stop_5fcapture_121',['stop_capture',['../namespacemlx_1_1core_1_1metal.html#ac90714424e36fb01e04550de69b8314f',1,'mlx::core::metal']]], - ['stop_5fgradient_122',['stop_gradient',['../group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f',1,'mlx::core']]], - ['stopgradient_123',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html#ac70d1ab819d04e00f76bc25aeebaf84f',1,'mlx::core::StopGradient']]], - ['store_124',['store',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#aa8f50ea8961ec5b35c1b81366d64f2cb',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a2aadaa3239cb3af0c2ee8af9b88c8a98',1,'mlx::steel::MMATile::store(threadgroup U *dst) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a752f708e4fe5ef37fdd902dae153179f',1,'mlx::steel::MMATile::store(device U *dst, const int ld) const'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#aa8f50ea8961ec5b35c1b81366d64f2cb',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a2aadaa3239cb3af0c2ee8af9b88c8a98',1,'mlx::steel::MMATile::store(threadgroup U *dst) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a752f708e4fe5ef37fdd902dae153179f',1,'mlx::steel::MMATile::store(device U *dst, const int ld) const']]], - ['store_5fresult_125',['store_result',['../structmlx_1_1steel_1_1_block_m_m_a.html#a0461451ffb5041b6a916ea17ed34288b',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a0461451ffb5041b6a916ea17ed34288b',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const']]], - ['store_5fresult_5fsafe_126',['store_result_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a081ba538d30d1d02498a7f341e6bd611',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a081ba538d30d1d02498a7f341e6bd611',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const']]], - ['store_5fsafe_127',['store_safe',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a1f0b00daad8eba2f855bb306e70d2328',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a57703f522c7409dbe2c0a68bb7acc2ba',1,'mlx::steel::MMATile::store_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a1f0b00daad8eba2f855bb306e70d2328',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a57703f522c7409dbe2c0a68bb7acc2ba',1,'mlx::steel::MMATile::store_safe()']]], - ['stream_128',['Stream',['../structmlx_1_1core_1_1_stream.html#a7f0815ff4886da74cbbff5f93d82dd3e',1,'mlx::core::Stream']]], - ['stream_129',['stream',['../classmlx_1_1core_1_1_event.html#a193143bad31b68c699fa27f135b45614',1,'mlx::core::Event::stream()'],['../classmlx_1_1core_1_1_primitive.html#a46e6257397a662528f9f831842ac456a',1,'mlx::core::Primitive::stream()']]], - ['streamcontext_130',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html#a89d803151e9d7dce29382aa83d5c6ef1',1,'mlx::core::StreamContext']]], - ['streamthread_131',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#ac528109a11abcb82e6e221c5efa4493c',1,'mlx::core::scheduler::StreamThread']]], - ['stride_132',['stride',['../classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093',1,'pocketfft::detail::arr_info::stride() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b',1,'pocketfft::detail::arr_info::stride(size_t i) const']]], - ['stride_5fin_133',['stride_in',['../classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c',1,'pocketfft::detail::multi_iter']]], - ['stride_5fout_134',['stride_out',['../classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7',1,'pocketfft::detail::multi_iter']]], - ['strided_5freduce_5fgeneral_5fdispatch_135',['strided_reduce_general_dispatch',['../namespacemlx_1_1core.html#aa0332c64ee9965f05026c30a0b778000',1,'mlx::core']]], - ['strided_5fscan_136',['strided_scan',['../scan_8h.html#a7abb6ffb6c3b96b88c2a63cd4cc2f7ae',1,'scan.h']]], - ['strides_137',['strides',['../classmlx_1_1core_1_1array.html#a28cf1928f5ec2f972a94ff1c0e71187d',1,'mlx::core::array::strides() const'],['../classmlx_1_1core_1_1array.html#ac9bfc251a9937eaefbe7f8c5ffd304d1',1,'mlx::core::array::strides(int dim) const']]], - ['submit_138',['submit',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4',1,'pocketfft::detail::threading::thread_pool']]], - ['subtract_139',['Subtract',['../classmlx_1_1core_1_1_subtract.html#a834854757394f8de7082af65bf86ed9c',1,'mlx::core::Subtract']]], - ['subtract_140',['subtract',['../group__ops.html#ga196c240d3d0fcbb4713802c485e15133',1,'mlx::core']]], - ['sum_141',['sum',['../namespacemlx_1_1steel.html#ab4a6ddea4beb7c447cf5b69b9d46cc3b',1,'mlx::steel::sum(T x)'],['../namespacemlx_1_1steel.html#acd6e194d37b617d7a5818bc384a97fe4',1,'mlx::steel::sum(T x, Us... us)'],['../group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6',1,'mlx::core::sum(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3627754d7868487bdab1bd83f05d9c81',1,'mlx::core::sum(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4',1,'mlx::core::sum(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d',1,'mlx::core::sum(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['svd_142',['SVD',['../classmlx_1_1core_1_1_s_v_d.html#ae89ff583e34fa894cccb8e7a475ee6d1',1,'mlx::core::SVD']]], - ['svd_143',['svd',['../namespacemlx_1_1core_1_1linalg.html#a64364b880e99914cf47bf756fa8dbaf0',1,'mlx::core::linalg']]], - ['swapaxes_144',['swapaxes',['../group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb',1,'mlx::core']]], - ['swizzle_145',['swizzle',['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle::swizzle(uint3 tid, const int swizzle_log)'],['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle::swizzle(uint3 tid, const int swizzle_log)']]], - ['synchronize_146',['synchronize',['../namespacemlx_1_1core.html#a14287949d82ffefad0306cef5eb5f9e4',1,'mlx::core::synchronize()'],['../namespacemlx_1_1core.html#a6648a71937b055e5ff513d98056c2fb5',1,'mlx::core::synchronize(Stream)']]] + ['scalarvector_4',['ScalarVector',['../structmlx_1_1core_1_1_scalar_vector.html#a69d6a3ddd7586e8e19a42c5e6f5a287b',1,'mlx::core::ScalarVector']]], + ['scaled_5fdot_5fproduct_5fattention_5',['scaled_dot_product_attention',['../namespacemlx_1_1core_1_1fast.html#a3663b50265b0a9c0cca2b5376852e059',1,'mlx::core::fast']]], + ['scaleddotproductattention_6',['ScaledDotProductAttention',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ab3f78d30e5bb3e76cfe701f2358e4748',1,'mlx::core::fast::ScaledDotProductAttention']]], + ['scan_7',['Scan',['../classmlx_1_1core_1_1_scan.html#ac93e8f9c6771de825d2186ef34fa7087',1,'mlx::core::Scan']]], + ['scan_8',['scan',['../namespacemlx_1_1core_1_1metal.html#a81c2cf124b0803098a54a78f8f6873a6',1,'mlx::core::metal']]], + ['scatter_9',['Scatter',['../classmlx_1_1core_1_1_scatter.html#ac9b3eff67389ef9aa820753379ffeaa3',1,'mlx::core::Scatter']]], + ['scatter_10',['scatter',['../namespacemlx_1_1core_1_1metal.html#a32e902c6cd6d35fcc3119ed6685a170f',1,'mlx::core::metal::scatter()'],['../group__ops.html#gad438be8f90bae9d37c6853b8f4225d61',1,'mlx::core::scatter(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gac2c2b379a3ce959dbe1c4a68f112edfe',1,'mlx::core::scatter(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatter_5fadd_11',['scatter_add',['../group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174',1,'mlx::core::scatter_add(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gac13318518e5703f1273c5366eb523a5a',1,'mlx::core::scatter_add(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatter_5fadd_5faxis_12',['scatter_add_axis',['../group__ops.html#gab3fd98c0d06b84b836f93bddbd7a2a0d',1,'mlx::core']]], + ['scatter_5faxis_13',['scatter_axis',['../namespacemlx_1_1core_1_1metal.html#a88c1d42d525fcdfb2f9e8aa2c3f82ea6',1,'mlx::core::metal::scatter_axis()'],['../scatter__axis_8h.html#af78a7935b05dabd42c2cdff4cf375130',1,'scatter_axis(const device T *upd, const device IdxT *indices, device mlx_atomic< T > *out, const constant int *shape, const constant int64_t *upd_strides, const constant int64_t *idx_strides, const constant size_t &ndim, const constant int &axis, const constant int &out_axis_size, const constant size_t &upd_ax_stride, const constant size_t &idx_ax_stride, uint3 index, uint3 grid_dim): scatter_axis.h']]], + ['scatter_5fimpl_14',['scatter_impl',['../scatter_8h.html#ab72d4fe1dbd4ae4dc529ee2ec8164fa4',1,'scatter.h']]], + ['scatter_5fmax_15',['scatter_max',['../group__ops.html#ga05881a4157cd113c9392d168a79e6673',1,'mlx::core::scatter_max(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga9adda5f9202bb3486e4d9e1114e3a56f',1,'mlx::core::scatter_max(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatter_5fmin_16',['scatter_min',['../group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5',1,'mlx::core::scatter_min(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga51fa762a997c243ca7a19e1ed3e83199',1,'mlx::core::scatter_min(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatter_5fprod_17',['scatter_prod',['../group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0',1,'mlx::core::scatter_prod(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#gaf83c53c453faa9083ba27e4b97539339',1,'mlx::core::scatter_prod(const array &a, const array &indices, const array &updates, int axis, StreamOrDevice s={})']]], + ['scatteraxis_18',['ScatterAxis',['../classmlx_1_1core_1_1_scatter_axis.html#a7365a2c5fddb1c39509998598de411db',1,'mlx::core::ScatterAxis']]], + ['scheduler_19',['Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a3ae42aed78a2200e9d02776fcd2316ba',1,'mlx::core::scheduler::Scheduler::Scheduler()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a61a74e3628899e66dde600e24a750648',1,'mlx::core::scheduler::Scheduler::Scheduler(const Scheduler &)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ac3f77b7c93220dadd0b3bb2e903b7059',1,'mlx::core::scheduler::Scheduler::Scheduler(Scheduler &&)=delete']]], + ['scheduler_20',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html#ae856e468c2f7c8f8ec672522cc13730b',1,'mlx::core::scheduler']]], + ['sdpa_5fvector_21',['sdpa_vector',['../sdpa__vector_8h.html#a826f7a3c7ab843abc0842241db3e57b3',1,'sdpa_vector.h']]], + ['sdpa_5fvector_5f2pass_5f1_22',['sdpa_vector_2pass_1',['../sdpa__vector_8h.html#aae1a2f23b03e24734805b08ebc5c1a59',1,'sdpa_vector.h']]], + ['sdpa_5fvector_5f2pass_5f2_23',['sdpa_vector_2pass_2',['../sdpa__vector_8h.html#a1368cf3618a4e03dbf743b3463205efe',1,'sdpa_vector.h']]], + ['seed_24',['seed',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a9f19c5da2031cba50d0ff996924347d8',1,'mlx::core::random::KeySequence::seed()'],['../namespacemlx_1_1core_1_1random.html#ac4ad325b613257306df74595d3d0e23b',1,'mlx::core::random::seed()']]], + ['seek_25',['seek',['../structmlx_1_1core_1_1_contiguous_iterator.html#af08f009e0a72414d274db2ff1b2c7dd5',1,'mlx::core::ContiguousIterator::seek()'],['../classmlx_1_1core_1_1io_1_1_reader.html#acea55078bd39ccaa27a9a36f17a39cd1',1,'mlx::core::io::Reader::seek()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a9c1716dda53aa36faea9c8fb1a3e34d4',1,'mlx::core::io::Writer::seek()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a4434ee18ff8bbf1b4fce670a337b535f',1,'mlx::core::io::ParallelFileReader::seek()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9646f4ea048ae58719daeb588e2de433',1,'mlx::core::io::FileWriter::seek()']]], + ['select_26',['Select',['../classmlx_1_1core_1_1_select.html#a6f833fe55dd68ad3726bbf9a8f75eec9',1,'mlx::core::Select']]], + ['select_27',['select',['../namespacemlx_1_1core_1_1simd.html#afb3bcbd8d8b34128cd0c8eb677a170ef',1,'mlx::core::simd::select(Simd< MaskT, N > mask, Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a9e3e7b35d564c70de8fa0b6150570ed8',1,'mlx::core::simd::select(Simd< MaskT, 1 > mask, Simd< T, 1 > x, Simd< T, 1 > y)'],['../namespacemlx_1_1core_1_1simd.html#a3b5ebb46e7beae839c97b2e7ed9c7426',1,'mlx::core::simd::select(Simd< MaskT, N > mask, Simd< float16_t, N > x, Simd< float16_t, N > y)']]], + ['send_28',['Send',['../classmlx_1_1core_1_1distributed_1_1_send.html#a2481dd876b14d4a13ac466cbca9c4eac',1,'mlx::core::distributed::Send']]], + ['send_29',['send',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac8472eb2f96d1b14c7e4ccef56268ba0',1,'mlx::core::distributed::detail::GroupImpl::send()'],['../namespacemlx_1_1core_1_1distributed_1_1detail.html#abf33511660ac71df5fc92f2aad6c6e08',1,'mlx::core::distributed::detail::send()'],['../namespacemlx_1_1core_1_1distributed.html#a5a8360edaa3a528a3927fce4d2cf1777',1,'mlx::core::distributed::send()']]], + ['set_30',['Set',['../structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807',1,'pocketfft::detail::cmplx::Set(T r_, T i_)'],['../structpocketfft_1_1detail_1_1cmplx.html#a447d26b2e07f6e45f29d865e906c0a98',1,'pocketfft::detail::cmplx::Set(T r_)']]], + ['set_5fbinary_5fop_5foutput_5fdata_31',['set_binary_op_output_data',['../namespacemlx_1_1core.html#a6a52856325c2eb031d3983eba2108d59',1,'mlx::core']]], + ['set_5fbuffer_32',['set_buffer',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ae890f5cefa4ae24ae0f5d8e46a313a92',1,'mlx::core::metal::CommandEncoder::set_buffer()'],['../structmlx_1_1core_1_1_command_encoder.html#ae890f5cefa4ae24ae0f5d8e46a313a92',1,'mlx::core::CommandEncoder::set_buffer()']]], + ['set_5fbytes_33',['set_bytes',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a9c343f791812a45c6c03a5c9f27f74d5',1,'mlx::core::metal::CommandEncoder::set_bytes(const T *v, int n, int idx)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#abc52d18ea87d213c47fd26062c829849',1,'mlx::core::metal::CommandEncoder::set_bytes(const T &v, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a9c343f791812a45c6c03a5c9f27f74d5',1,'mlx::core::CommandEncoder::set_bytes(const T *v, int n, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#abc52d18ea87d213c47fd26062c829849',1,'mlx::core::CommandEncoder::set_bytes(const T &v, int idx)']]], + ['set_5fcache_5flimit_34',['set_cache_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#af392bced29d9e4e3f1a7cc4725d83764',1,'mlx::core::metal::MetalAllocator::set_cache_limit()'],['../namespacemlx_1_1core_1_1metal.html#ab09c9b60f1e886ab859e6a066c9a5b9d',1,'mlx::core::metal::set_cache_limit()']]], + ['set_5fcompile_5fmode_35',['set_compile_mode',['../namespacemlx_1_1core.html#a49445a55f976c4397f25ea18e1e92bef',1,'mlx::core']]], + ['set_5fcompute_5fpipeline_5fstate_36',['set_compute_pipeline_state',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a6d4c03a6585deedb5ccd1a1057d0c6ef',1,'mlx::core::metal::CommandEncoder::set_compute_pipeline_state()'],['../structmlx_1_1core_1_1_command_encoder.html#a6d4c03a6585deedb5ccd1a1057d0c6ef',1,'mlx::core::CommandEncoder::set_compute_pipeline_state()']]], + ['set_5fdata_37',['set_data',['../classmlx_1_1core_1_1array.html#af9e3a02b4c0023c36248dc75c887214f',1,'mlx::core::array::set_data(allocator::Buffer buffer, Deleter d=allocator::free)'],['../classmlx_1_1core_1_1array.html#a5f338202a39d37fa3f4241e851a15838',1,'mlx::core::array::set_data(allocator::Buffer buffer, size_t data_size, Strides strides, Flags flags, Deleter d=allocator::free)']]], + ['set_5fdefault_5fdevice_38',['set_default_device',['../namespacemlx_1_1core.html#a312a2de41367fe52caeaf8c0f596a120',1,'mlx::core']]], + ['set_5fdefault_5fstream_39',['set_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a6d15314ac9cf25efc9bd1278de9a66bb',1,'mlx::core::scheduler::Scheduler::set_default_stream()'],['../namespacemlx_1_1core.html#af35a2b06517d8bb7dbb469692b4f841c',1,'mlx::core::set_default_stream()']]], + ['set_5finput_5farray_40',['set_input_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ab69ff0d7f14b9b59db4df0608193dce4',1,'mlx::core::metal::CommandEncoder::set_input_array()'],['../structmlx_1_1core_1_1_command_encoder.html#ab69ff0d7f14b9b59db4df0608193dce4',1,'mlx::core::CommandEncoder::set_input_array()']]], + ['set_5fmemory_5flimit_41',['set_memory_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a179e3127ef9377ce54295f771c34ba1b',1,'mlx::core::metal::MetalAllocator::set_memory_limit()'],['../namespacemlx_1_1core_1_1metal.html#a3fb2c4a237fa4bfdff798156146c4937',1,'mlx::core::metal::set_memory_limit()']]], + ['set_5fname_42',['set_name',['../structmlx_1_1core_1_1_node_namer.html#a57a574e48f8a9cd122616d80b138c768',1,'mlx::core::NodeNamer']]], + ['set_5foutput_5farray_43',['set_output_array',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a6a2e28e542eaa2886041bddd51ff6522',1,'mlx::core::metal::CommandEncoder::set_output_array()'],['../structmlx_1_1core_1_1_command_encoder.html#a6a2e28e542eaa2886041bddd51ff6522',1,'mlx::core::CommandEncoder::set_output_array()']]], + ['set_5fresidency_5fset_44',['set_residency_set',['../classmlx_1_1core_1_1metal_1_1_device.html#a03a2f0c712660a1bd437cb16e4aba79f',1,'mlx::core::metal::Device']]], + ['set_5fsiblings_45',['set_siblings',['../classmlx_1_1core_1_1array.html#a8fccbe7a4edfd8cca168161124e263b1',1,'mlx::core::array']]], + ['set_5fstatus_46',['set_status',['../classmlx_1_1core_1_1array.html#a63598018999b49f1340b183cb303f05c',1,'mlx::core::array']]], + ['set_5fternary_5fop_5foutput_5fdata_47',['set_ternary_op_output_data',['../namespacemlx_1_1core.html#a6f4528d0d338ea5e1f19d345875c26a2',1,'mlx::core']]], + ['set_5ftracer_48',['set_tracer',['../classmlx_1_1core_1_1array.html#af26e6be1a9e6239471a4c24310c0c7c8',1,'mlx::core::array']]], + ['set_5funary_5foutput_5fdata_49',['set_unary_output_data',['../namespacemlx_1_1core.html#a4c6a4241bfcdd7bbf30d0e521b79e5a3',1,'mlx::core']]], + ['set_5fvalue_50',['set_value',['../classmlx_1_1core_1_1_event.html#a0d077b11f4b28f882b42440b7ac6d40d',1,'mlx::core::Event']]], + ['set_5fvector_5fbytes_51',['set_vector_bytes',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a68c3c6a036e11ec40211c09811bbed1b',1,'mlx::core::metal::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, size_t nelems, int idx)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a7375adf9ee5355bcf4b7f5f210efd115',1,'mlx::core::metal::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a68c3c6a036e11ec40211c09811bbed1b',1,'mlx::core::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, size_t nelems, int idx)'],['../structmlx_1_1core_1_1_command_encoder.html#a7375adf9ee5355bcf4b7f5f210efd115',1,'mlx::core::CommandEncoder::set_vector_bytes(const std::vector< T > &vec, int idx)']]], + ['set_5fwired_5flimit_52',['set_wired_limit',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a84fa0347da18055bc13ba0a5c4b57253',1,'mlx::core::metal::MetalAllocator::set_wired_limit()'],['../namespacemlx_1_1core_1_1metal.html#a31eab4828d31d292bc84e07b0d961e1e',1,'mlx::core::metal::set_wired_limit()']]], + ['shape_53',['shape',['../classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0',1,'pocketfft::detail::arr_info::shape() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8',1,'pocketfft::detail::arr_info::shape(size_t i) const'],['../classmlx_1_1core_1_1array.html#a1d06c76b0f3010a5c329d0e9e29e0597',1,'mlx::core::array::shape() const'],['../classmlx_1_1core_1_1array.html#ac50382b652f6e8fbd50d42b7ff595810',1,'mlx::core::array::shape(int dim) const']]], + ['shape2d_54',['Shape2D',['../structmlx_1_1steel_1_1_shape2_d.html#a070ce70eb6d84361c7f313159c438a5c',1,'mlx::steel::Shape2D']]], + ['shapes_5fwithout_5freduction_5faxes_55',['shapes_without_reduction_axes',['../namespacemlx_1_1core.html#a0bea91a360a984e72d2815353f97ee25',1,'mlx::core']]], + ['shared_5fbuffer_5freshape_56',['shared_buffer_reshape',['../namespacemlx_1_1core.html#a88d88987bd8bf3ca46bf3b5e8aacce9d',1,'mlx::core']]], + ['shared_5fbuffer_5fslice_57',['shared_buffer_slice',['../namespacemlx_1_1core.html#a349a9fc2bfd950f679a3fe39b8bdedad',1,'mlx::core']]], + ['shutdown_58',['shutdown',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411',1,'pocketfft::detail::threading::thread_pool']]], + ['siblings_59',['siblings',['../classmlx_1_1core_1_1array.html#acf80fde8f743f65ad5b4be69fcb7a74d',1,'mlx::core::array::siblings() const'],['../classmlx_1_1core_1_1array.html#a7263f23e70a580a9bc2129fbcde36e6c',1,'mlx::core::array::siblings()']]], + ['sigmoid_60',['Sigmoid',['../classmlx_1_1core_1_1_sigmoid.html#a47eca99113ec19f0eb60b6a0472c592b',1,'mlx::core::Sigmoid']]], + ['sigmoid_61',['sigmoid',['../group__ops.html#ga708abf8f79609cd6831db7c38cafac0e',1,'mlx::core']]], + ['sign_62',['Sign',['../classmlx_1_1core_1_1_sign.html#afe951e50907bc23a601ec5fa9eae5763',1,'mlx::core::Sign']]], + ['sign_63',['sign',['../group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa',1,'mlx::core']]], + ['signal_64',['signal',['../classmlx_1_1core_1_1_event.html#a65a858445506a61be5889ae0e3651b89',1,'mlx::core::Event']]], + ['signaling_5fnan_65',['signaling_NaN',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ad1f76a43c7d51a3765174aa6e0dd9f80',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['simd_66',['Simd',['../structmlx_1_1core_1_1simd_1_1_simd.html#a5c24246e05e833fd81d900226a29e6ab',1,'mlx::core::simd::Simd::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a1693c8e542dddf2ab60d309d64de71b6',1,'mlx::core::simd::Simd::Simd(Simd< U, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a74091cd8b7c72014f73ec215b52ea2ce',1,'mlx::core::simd::Simd::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aabe757c8fd93dadd6f8859cda99f4927',1,'mlx::core::simd::Simd::Simd(Simd< T, N/2 > x, Simd< T, N/2 > y)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#a3f6e4a83ecf897465f44160b6fad5a7a',1,'mlx::core::simd::Simd< T, 1 >::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#a585bc4768c4f7e1313d7e8756fbb00cc',1,'mlx::core::simd::Simd< T, 1 >::Simd(Simd< U, 1 > v)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#acf948f7c5e8829432c0ac17fc9f911e2',1,'mlx::core::simd::Simd< T, 1 >::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a04a3a73f98fa5c9090b6cf6154e99e8d',1,'mlx::core::simd::Simd< float16_t, N >::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#ad8b628f8834e983853d557cc1e4124bb',1,'mlx::core::simd::Simd< float16_t, N >::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a5e76655d70c0e9ae49eea536c0e3b8cf',1,'mlx::core::simd::Simd< float16_t, N >::Simd(float16x8_t v)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#ae1dfcaca51f9a6fcdb757cb8413ac223',1,'mlx::core::simd::Simd< float16_t, N >::Simd(Simd< float, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a1f30c088a6828d0673e927ed6c0a4b2b',1,'mlx::core::simd::Simd< float16_t, N >::Simd(Simd< uint16_t, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a5c24246e05e833fd81d900226a29e6ab',1,'mlx::core::simd::Simd< T, 1 >::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a1693c8e542dddf2ab60d309d64de71b6',1,'mlx::core::simd::Simd< T, 1 >::Simd(Simd< U, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a74091cd8b7c72014f73ec215b52ea2ce',1,'mlx::core::simd::Simd< T, 1 >::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aabe757c8fd93dadd6f8859cda99f4927',1,'mlx::core::simd::Simd< T, 1 >::Simd(Simd< T, N/2 > x, Simd< T, N/2 > y)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a5c24246e05e833fd81d900226a29e6ab',1,'mlx::core::simd::Simd< float16_t, N >::Simd()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a1693c8e542dddf2ab60d309d64de71b6',1,'mlx::core::simd::Simd< float16_t, N >::Simd(Simd< U, N > other)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a74091cd8b7c72014f73ec215b52ea2ce',1,'mlx::core::simd::Simd< float16_t, N >::Simd(U v)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aabe757c8fd93dadd6f8859cda99f4927',1,'mlx::core::simd::Simd< float16_t, N >::Simd(Simd< float16_t, N/2 > x, Simd< float16_t, N/2 > y)']]], + ['simd_5fbroadcast_67',['simd_broadcast',['../namespacemetal.html#a498f1e85107eb5f01ba4435977f8efe0',1,'metal']]], + ['simd_5fexclusive_5fscan_68',['simd_exclusive_scan',['../struct_cum_prod_3_01bool_01_4.html#a1a86e9398bae24182b7be0a6577bf223',1,'CumProd< bool >::simd_exclusive_scan()'],['../struct_cum_max.html#ae11b67aa6c998e9a01615b2a79af4403',1,'CumMax::simd_exclusive_scan()'],['../struct_cum_min.html#a83e65017ff33018b585c043fb803773b',1,'CumMin::simd_exclusive_scan()']]], + ['simd_5fmax_69',['simd_max',['../namespacemetal.html#a048cad0aca52cb737ebf103e76bd1c49',1,'metal']]], + ['simd_5fmin_70',['simd_min',['../namespacemetal.html#ae9e2a23e00724ba2d7868bc4112b386b',1,'metal']]], + ['simd_5fprefix_5fexclusive_5fproduct_71',['simd_prefix_exclusive_product',['../namespacemetal.html#a5ca40242390b632f737e29636829b2e4',1,'metal']]], + ['simd_5fprefix_5fexclusive_5fsum_72',['simd_prefix_exclusive_sum',['../namespacemetal.html#abfbb70c7471f28bf7ff36a612ad014b2',1,'metal']]], + ['simd_5fprefix_5finclusive_5fproduct_73',['simd_prefix_inclusive_product',['../namespacemetal.html#a6ca6a7e1996228fa536e969e9e45c446',1,'metal']]], + ['simd_5fprefix_5finclusive_5fsum_74',['simd_prefix_inclusive_sum',['../namespacemetal.html#a567acb18199ac0107712eb8cb8aeb8e9',1,'metal']]], + ['simd_5fproduct_75',['simd_product',['../namespacemetal.html#ac6e883a04e2265a9790d7db76059e1b4',1,'metal']]], + ['simd_5fscan_76',['simd_scan',['../struct_cum_prod_3_01bool_01_4.html#abeb5ec4237b330e7219f4e881cf10d7a',1,'CumProd< bool >::simd_scan()'],['../struct_cum_max.html#adc9ec8bb09b4433d4c2f03022c43d781',1,'CumMax::simd_scan()'],['../struct_cum_min.html#a0a1005d91b1c90e90e2c6dbd6c296649',1,'CumMin::simd_scan()']]], + ['simd_5fshuffle_77',['simd_shuffle',['../namespacemetal.html#a259ed115bc3c58f88eb35830916b26d4',1,'metal::simd_shuffle()'],['../backend_2metal_2kernels_2utils_8h.html#a71986ecdd7d18f975dd22c3df7421ce2',1,'simd_shuffle(uint64_t data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a3bdbdfeb7a1dde40cd3ce1df8d9213b5',1,'simd_shuffle(int64_t data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ab4cbcdb054f9165130da91a3334da0cf',1,'simd_shuffle(bool data, uint16_t lane): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ab8175b66bcc080fb89f738143568c30b',1,'simd_shuffle(complex64_t data, uint16_t lane): utils.h']]], + ['simd_5fshuffle_5fand_5ffill_5fdown_78',['simd_shuffle_and_fill_down',['../namespacemetal.html#ae29a06f0eac636ad7af21dea5b04938b',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a0ee6239fa29a5f9ee0201e0dc5ddc8e0',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], + ['simd_5fshuffle_5fand_5ffill_5fup_79',['simd_shuffle_and_fill_up',['../namespacemetal.html#a1ca14116bf50639b214d8414b5bbaaa6',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a5138d5cdc18139e135707916a243cd8e',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta)'],['../backend_2metal_2kernels_2utils_8h.html#a5862d5ea154c9b76cf56a630cf6385b4',1,'simd_shuffle_and_fill_up(uint64_t data, uint64_t filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a7bb56415c5412a6a26f70a990915f064',1,'simd_shuffle_and_fill_up(int64_t data, int64_t filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ad55bd473647f2c6c68e65e5312c132d1',1,'simd_shuffle_and_fill_up(bool data, bool filling, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a94e02a6ae8c39cbf4cb23aa44df9dbd5',1,'simd_shuffle_and_fill_up(complex64_t data, complex64_t filling, uint16_t delta): utils.h']]], + ['simd_5fshuffle_5fdown_80',['simd_shuffle_down',['../namespacemetal.html#af6e2dd7ae087aba6abac4f0350b7611c',1,'metal::simd_shuffle_down()'],['../backend_2metal_2kernels_2utils_8h.html#aba6279624b1d30c525efee856a222b5c',1,'simd_shuffle_down(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a0c1e4d782fcc56e1ab5565cef12430dd',1,'simd_shuffle_down(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a48ae83a8caf5c74810df60b6c6cdb062',1,'simd_shuffle_down(bool data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ad9a671a5f9aaa729ae7a77026f16bcb0',1,'simd_shuffle_down(complex64_t data, uint16_t delta): utils.h']]], + ['simd_5fshuffle_5frotate_5fdown_81',['simd_shuffle_rotate_down',['../namespacemetal.html#a4bb203647a421032db47e73cd649841b',1,'metal']]], + ['simd_5fshuffle_5frotate_5fup_82',['simd_shuffle_rotate_up',['../namespacemetal.html#a729b22077d6c944491a6027c18ea80c9',1,'metal']]], + ['simd_5fshuffle_5fup_83',['simd_shuffle_up',['../namespacemetal.html#afe81c5fbde3f4890458b081909242c55',1,'metal::simd_shuffle_up()'],['../backend_2metal_2kernels_2utils_8h.html#a39e436e0a942912266aae7e0bd82d7c0',1,'simd_shuffle_up(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a617f3857caf33c569afa6148135f8b7a',1,'simd_shuffle_up(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ae0f5c42020275a588234e69f1eb7a485',1,'simd_shuffle_up(bool data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a92b455bac6a23af51c35ea83de2383eb',1,'simd_shuffle_up(complex64_t data, uint16_t delta): utils.h']]], + ['simd_5fshuffle_5fxor_84',['simd_shuffle_xor',['../namespacemetal.html#a5017efc9605e069cfb507137cd1a1852',1,'metal']]], + ['simd_5fsum_85',['simd_sum',['../namespacemetal.html#a85181e37a00cb4a4217f1bb25389bce5',1,'metal']]], + ['simd_5fxor_86',['simd_xor',['../namespacemetal.html#a1308decbf2d5c33d34d6be523ea1c30f',1,'metal']]], + ['simple_5fiter_87',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398',1,'pocketfft::detail::simple_iter']]], + ['sin_88',['Sin',['../classmlx_1_1core_1_1_sin.html#a10d1ecc0ca96e79cdf55b57073d126ea',1,'mlx::core::Sin']]], + ['sin_89',['sin',['../namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258',1,'pocketfft::detail::sin()'],['../namespacemlx_1_1core_1_1simd.html#ab4d582d72c0a7ee313e19c906e43cef1',1,'mlx::core::simd::sin()'],['../namespacemetal.html#a619a159ca5f2ddfe3647d3a6bb6e804c',1,'metal::sin()'],['../namespacemetal_1_1fast.html#a3af771cfe7a135104f9d063147dba270',1,'metal::fast::sin()'],['../namespacemetal_1_1precise.html#a71acf77ffd29c56f56afae0195c98a1c',1,'metal::precise::sin()'],['../group__ops.html#gaebf0a73ad3732fba39df37826c235692',1,'mlx::core::sin()']]], + ['sincos_90',['sincos',['../namespacemlx_1_1core_1_1simd.html#ad78056685c9732c3465c0d8b8ec1bef7',1,'mlx::core::simd']]], + ['sincos_5f2pibyn_91',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40',1,'pocketfft::detail::sincos_2pibyn']]], + ['sinh_92',['Sinh',['../classmlx_1_1core_1_1_sinh.html#a4a4f6814d403c2ce5d6c574b0dca3c96',1,'mlx::core::Sinh']]], + ['sinh_93',['sinh',['../namespacemlx_1_1core_1_1simd.html#a09a2f3f2bc999c16babf3d8d90994d6e',1,'mlx::core::simd::sinh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a479ccddac341bd0760857b77e449e5e1',1,'mlx::core::simd::sinh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a40879bf874309c0a5abef783aea2057d',1,'mlx::core::simd::sinh(Simd< T, 1 > in)'],['../namespacemetal.html#a83ba4235ae350ab8880a9df09158620b',1,'metal::sinh()'],['../namespacemetal_1_1fast.html#a990d90b3440e38d1fb4ff5065c6c189b',1,'metal::fast::sinh()'],['../namespacemetal_1_1precise.html#abc8f4f59dd6e7204ab5d84f0af96331c',1,'metal::precise::sinh()'],['../group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7',1,'mlx::core::sinh()']]], + ['sinpi_94',['sinpi',['../namespacemetal.html#ae9655f7fa2ba6c0625ca25fbb278e269',1,'metal::sinpi()'],['../namespacemetal_1_1fast.html#ab07a32fe544aa304577d29e0251e87b2',1,'metal::fast::sinpi()'],['../namespacemetal_1_1precise.html#a78b17dab93519d9c82c2575dafec49c9',1,'metal::precise::sinpi()']]], + ['size_95',['size',['../classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952',1,'pocketfft::detail::arr::size()'],['../classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5',1,'pocketfft::detail::arr_info::size()'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a2adf9a9c968f113dde830cc0dc27dcc6',1,'mlx::core::allocator::Allocator::size()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#aafa92e8310db089b1ac72b840777e26b',1,'mlx::core::allocator::CommonAllocator::size()'],['../classmlx_1_1core_1_1array.html#a598f87161926d9e0b516860f0ea2c8f6',1,'mlx::core::array::size()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a51f6587e8065be16f0418ca42a796e05',1,'mlx::core::metal::MetalAllocator::size()'],['../structmlx_1_1core_1_1distributed_1_1_group.html#ac0a5a1e463a9330355e8bfe09c0feaf2',1,'mlx::core::distributed::Group::size()'],['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ab1c8044b05f185c4bcc53002d4587599',1,'mlx::core::distributed::detail::GroupImpl::size()'],['../structmlx_1_1core_1_1_dtype.html#ab54051563d85212c7f0f049166bc9971',1,'mlx::core::Dtype::size()']]], + ['size_5fof_96',['size_of',['../namespacemlx_1_1core.html#add4794cc0ffe5d717fc146084a235d95',1,'mlx::core']]], + ['slice_97',['Slice',['../classmlx_1_1core_1_1_slice.html#a3aa025acbf4a9ca9e030a1e6bda102f7',1,'mlx::core::Slice']]], + ['slice_98',['slice',['../namespacemlx_1_1core.html#a5adff87687b78bfc18dafbc654925cdb',1,'mlx::core::slice(const array &in, array &out, const Shape &start_indices, const Shape &strides)'],['../group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d',1,'mlx::core::slice(const array &a, Shape start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#gaedcbdf4040f5e6a02a74b1ed1c6c2ebc',1,'mlx::core::slice(const array &a, std::initializer_list< int > start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#gaec56dcb94d5e7f7b885fb60b4bf4aa9d',1,'mlx::core::slice(const array &a, Shape start, Shape stop, StreamOrDevice s={})'],['../group__ops.html#ga797996e53ea34317a55dc2f314736b89',1,'mlx::core::slice(const array &a, const array &start, std::vector< int > axes, Shape slice_size, StreamOrDevice s={})']]], + ['slice_5fgpu_99',['slice_gpu',['../namespacemlx_1_1core.html#a327578951a44116e5da2db651661265f',1,'mlx::core']]], + ['slice_5fupdate_100',['slice_update',['../group__ops.html#ga7b7e786985d27789aaed20bb2f9509be',1,'mlx::core::slice_update(const array &src, const array &update, Shape start, Shape stop, Shape strides, StreamOrDevice s={})'],['../group__ops.html#ga3b7c6136ae5a38dd6457b65c1833aa67',1,'mlx::core::slice_update(const array &src, const array &update, Shape start, Shape stop, StreamOrDevice s={})'],['../group__ops.html#gacd906ffb96149a4998c321cbf2231d7e',1,'mlx::core::slice_update(const array &src, const array &update, const array &start, std::vector< int > axes, StreamOrDevice s={})']]], + ['sliceupdate_101',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html#a63a49264b18340f2bc442c081a7b4c7a',1,'mlx::core::SliceUpdate']]], + ['softmax_102',['Softmax',['../classmlx_1_1core_1_1_softmax.html#a4ec686aac4e06f0dfe2cbd6801af40eb',1,'mlx::core::Softmax']]], + ['softmax_103',['softmax',['../namespacemlx_1_1core_1_1metal.html#a4fe937c2c584fd646926057f31d54ca6',1,'mlx::core::metal::softmax()'],['../group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6',1,'mlx::core::softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga1ae3614d07d873892a530d14c3857d0b',1,'mlx::core::softmax(const array &a, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga06f570d73716a24303e6de3aaba4457b',1,'mlx::core::softmax(const array &a, int axis, bool precise=false, StreamOrDevice s={})']]], + ['softmax_5fexp_104',['softmax_exp',['../kernels_2softmax_8h.html#a440d4031ee5e86159a4dd715e44a438b',1,'softmax.h']]], + ['softmax_5flooped_105',['softmax_looped',['../kernels_2softmax_8h.html#a8c47b0924ebfeebcca25f3dd17373276',1,'softmax.h']]], + ['softmax_5fsingle_5frow_106',['softmax_single_row',['../kernels_2softmax_8h.html#a815fe70f879f318e5d6e99acf043f52b',1,'softmax.h']]], + ['sort_107',['Sort',['../classmlx_1_1core_1_1_sort.html#a62943032dbd72e85ceb9b4b7211f4a44',1,'mlx::core::Sort']]], + ['sort_108',['sort',['../struct_thread_sort.html#a7c7cf65ffe4362596009fc1c05633e7a',1,'ThreadSort::sort()'],['../struct_block_merge_sort.html#ad25513c2d02802ac35edb086aceb4ce8',1,'BlockMergeSort::sort()'],['../namespacemlx_1_1core_1_1metal.html#ab77c9a9ecaeeab8c66b712862777c24b',1,'mlx::core::metal::sort()'],['../group__ops.html#ga7fb616054665b3c2d61fa234f501f079',1,'mlx::core::sort(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaae1bc47aa737f705d0e5884270063fea',1,'mlx::core::sort(const array &a, int axis, StreamOrDevice s={})']]], + ['special_5fmul_109',['special_mul',['../structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874',1,'pocketfft::detail::cmplx::special_mul()'],['../namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76',1,'pocketfft::detail::special_mul()']]], + ['split_110',['Split',['../classmlx_1_1core_1_1_split.html#ad3f4ed34b85c73683bad5d530309342f',1,'mlx::core::Split']]], + ['split_111',['split',['../structmlx_1_1core_1_1distributed_1_1_group.html#ad3682f4dc85bfe7e5464b87f6f0fd931',1,'mlx::core::distributed::Group::split()'],['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a87800a23c8160933a2d77a55a959194d',1,'mlx::core::distributed::detail::GroupImpl::split()'],['../group__ops.html#ga7534290bceab5fb3831a05d67bebce7d',1,'mlx::core::split(const array &a, int num_splits, int axis, StreamOrDevice s={})'],['../group__ops.html#ga56882d24e5fde59c266774624c892d41',1,'mlx::core::split(const array &a, int num_splits, StreamOrDevice s={})'],['../group__ops.html#ga19005414e7d8020cd6e94e06bf399b09',1,'mlx::core::split(const array &a, const Shape &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#ga9ea089f42b9940510619052b7166d9ac',1,'mlx::core::split(const array &a, const Shape &indices, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a42847b435d037a977592e355eed072af',1,'mlx::core::random::split(const array &key, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7ec057064c7326c41b536f08178861e5',1,'mlx::core::random::split(const array &key, int num, StreamOrDevice s={})']]], + ['sqrt_112',['Sqrt',['../classmlx_1_1core_1_1_sqrt.html#a6682a7c31ca427c9d2c5ddb6a479bf29',1,'mlx::core::Sqrt']]], + ['sqrt_113',['sqrt',['../namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3',1,'pocketfft::detail::sqrt()'],['../namespacemlx_1_1core_1_1simd.html#ae39b8e1d1fff94947406eeb8ec6e0414',1,'mlx::core::simd::sqrt(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a2ba6c75c0821db3e9ac525a89b3ac859',1,'mlx::core::simd::sqrt(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a7e80d3e33f2edd02310641d3e3dd5658',1,'mlx::core::simd::sqrt(Simd< float16_t, N > a)'],['../namespacemetal.html#ab3f4d4852ca0e591104fbd8e5b50d31b',1,'metal::sqrt()'],['../namespacemetal_1_1fast.html#a4218a85c7d8a74cb8055b4755205627e',1,'metal::fast::sqrt()'],['../namespacemetal_1_1precise.html#acb213467361cd2cab93a8d5ea1aa5bfd',1,'metal::precise::sqrt()'],['../group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1',1,'mlx::core::sqrt()']]], + ['square_114',['Square',['../classmlx_1_1core_1_1_square.html#ab94e28d5c92e6febc1c74e525f730dc4',1,'mlx::core::Square']]], + ['square_115',['square',['../group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e',1,'mlx::core']]], + ['squeeze_116',['Squeeze',['../classmlx_1_1core_1_1_squeeze.html#a032bd53dcc3d71a11d810bc3ca3ef4b0',1,'mlx::core::Squeeze']]], + ['squeeze_117',['squeeze',['../group__ops.html#ga710daa7ec721bd4d3f326082cb195576',1,'mlx::core::squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3',1,'mlx::core::squeeze(const array &a, int axis, StreamOrDevice s={})'],['../group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad',1,'mlx::core::squeeze(const array &a, StreamOrDevice s={})']]], + ['stack_118',['stack',['../group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032',1,'mlx::core::stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga82216209dce901296fc737fe8efa5c94',1,'mlx::core::stack(const std::vector< array > &arrays, StreamOrDevice s={})']]], + ['start_5fcapture_119',['start_capture',['../namespacemlx_1_1core_1_1metal.html#aa47cb5651bf3b65c46ab216b7e504d77',1,'mlx::core::metal']]], + ['start_5fconcurrent_120',['start_concurrent',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::metal::CommandEncoder::start_concurrent()'],['../structmlx_1_1core_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::CommandEncoder::start_concurrent()']]], + ['state_121',['state',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a379b27ac336ef351aa81142c5626ad76',1,'mlx::core::fast::RMSNormVJP::state()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a963e672c607b5f86080e6cc32a3cd6e5',1,'mlx::core::fast::LayerNormVJP::state()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#aa5a03284c6f5639d684dd34d86050cf9',1,'mlx::core::fast::AffineQuantize::state()'],['../classmlx_1_1core_1_1_add_m_m.html#a24ab73de46d0589780dac5ade43f93b8',1,'mlx::core::AddMM::state()'],['../classmlx_1_1core_1_1_arange.html#ac4a9f48a11c2af03ed57fdf2422cbfad',1,'mlx::core::Arange::state()'],['../classmlx_1_1core_1_1_arg_partition.html#a5033c46f5aae9b14859cc8b0ca4c8e19',1,'mlx::core::ArgPartition::state()'],['../classmlx_1_1core_1_1_arg_reduce.html#acac3b26364260aac7511b4cb7add3604',1,'mlx::core::ArgReduce::state()'],['../classmlx_1_1core_1_1_arg_sort.html#a90548429765f9e7e2332f01b72692fa2',1,'mlx::core::ArgSort::state()'],['../classmlx_1_1core_1_1_as_type.html#a98ea769fc9cd6d76b07817444e7a78ab',1,'mlx::core::AsType::state()'],['../classmlx_1_1core_1_1_as_strided.html#ae730aeff375498ba774d4207c7af8c36',1,'mlx::core::AsStrided::state()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8a67d6f431b4055ab66656201622af4d',1,'mlx::core::BitwiseBinary::state()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a6bbcc34b256840e4df2953563f2b4a07',1,'mlx::core::BlockMaskedMM::state()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a5b2594b7a70dd4873a07e742140a245f',1,'mlx::core::BroadcastAxes::state()'],['../classmlx_1_1core_1_1_broadcast.html#a49fdb421047860733af7dfbbb478da8d',1,'mlx::core::Broadcast::state()'],['../classmlx_1_1core_1_1_concatenate.html#a60cd572a42b346399ee539af2dfbf29e',1,'mlx::core::Concatenate::state()'],['../classmlx_1_1core_1_1_convolution.html#a7f44f0caea20cc2858717afba1e915d8',1,'mlx::core::Convolution::state()'],['../classmlx_1_1core_1_1_equal.html#aa27ff7525f109edc56b731a6df78f6bc',1,'mlx::core::Equal::state()'],['../classmlx_1_1core_1_1_expand_dims.html#a7cacc704c533c00ba072f0a7872631cf',1,'mlx::core::ExpandDims::state()'],['../classmlx_1_1core_1_1_f_f_t.html#a710c6f6e8412da0af0fdbe58fbae320e',1,'mlx::core::FFT::state()'],['../classmlx_1_1core_1_1_flatten.html#af95dd89c47cd2342233dc0b6d36822a3',1,'mlx::core::Flatten::state()'],['../classmlx_1_1core_1_1_gather.html#aee59ff90127ef4c2d7fcbe2955b95b27',1,'mlx::core::Gather::state()'],['../classmlx_1_1core_1_1_gather_axis.html#adff37b05799654b1a589e334d1cd6b46',1,'mlx::core::GatherAxis::state()'],['../classmlx_1_1core_1_1_hadamard.html#afd67d09fde38ab3b6ba873b797f03dae',1,'mlx::core::Hadamard::state()'],['../classmlx_1_1core_1_1_log.html#a86fca2ec3766f5d4a2e6d8ba2983c3aa',1,'mlx::core::Log::state()'],['../classmlx_1_1core_1_1_number_of_elements.html#afbfee716b4896e98bdf502ceab87ac09',1,'mlx::core::NumberOfElements::state()'],['../classmlx_1_1core_1_1_pad.html#a00a7cff2ae640f45b43f62cc25d6346c',1,'mlx::core::Pad::state()'],['../classmlx_1_1core_1_1_partition.html#adde13e40924c016473864119465cad4b',1,'mlx::core::Partition::state()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ad83bfd32fda988c29e5ca277a84c0655',1,'mlx::core::QuantizedMatmul::state()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a60c908bc836f930bb33f60b3e9db43af',1,'mlx::core::GatherQMM::state()'],['../classmlx_1_1core_1_1_random_bits.html#a75a34d7541a1c124710dc4d0ec2dfa60',1,'mlx::core::RandomBits::state()'],['../classmlx_1_1core_1_1_reshape.html#aa8ad5958aac8723dd6ce49820eaba029',1,'mlx::core::Reshape::state()'],['../classmlx_1_1core_1_1_reduce.html#afca1398c042a3b1ca9a9a2e40fe62367',1,'mlx::core::Reduce::state()'],['../classmlx_1_1core_1_1_scan.html#a7249ca4c3316b1b1248df32c71fee0ea',1,'mlx::core::Scan::state()'],['../classmlx_1_1core_1_1_scatter.html#a50a65033dc2a1cc84bf529ba718c9c60',1,'mlx::core::Scatter::state()'],['../classmlx_1_1core_1_1_scatter_axis.html#a1a0125be908a1d80875236c817f34495',1,'mlx::core::ScatterAxis::state()'],['../classmlx_1_1core_1_1_slice.html#a069dafc62bf71e3ebc0bd99d96ec23be',1,'mlx::core::Slice::state()'],['../classmlx_1_1core_1_1_slice_update.html#aac5156a19209274b7de1dff231ef25fd',1,'mlx::core::SliceUpdate::state()'],['../classmlx_1_1core_1_1_dynamic_slice.html#aec9084e603d7562f3a75c5fc32918581',1,'mlx::core::DynamicSlice::state()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a0b0b2a0e4d97305fd6f3c635fcdccd76',1,'mlx::core::DynamicSliceUpdate::state()'],['../classmlx_1_1core_1_1_softmax.html#adf665f7c292e84f56c630016b75427f3',1,'mlx::core::Softmax::state()'],['../classmlx_1_1core_1_1_sort.html#a5ab15d1a89efd8661887c805c35fc617',1,'mlx::core::Sort::state()'],['../classmlx_1_1core_1_1_split.html#a915156cde0448ec26776e329004b1a92',1,'mlx::core::Split::state()'],['../classmlx_1_1core_1_1_sqrt.html#a053853757ad99195e3f2b1cca571e31b',1,'mlx::core::Sqrt::state()'],['../classmlx_1_1core_1_1_squeeze.html#a04f9d2595cb7d4ec988479cd33fe9362',1,'mlx::core::Squeeze::state()'],['../classmlx_1_1core_1_1_unflatten.html#aeba13680064238191811230171365598',1,'mlx::core::Unflatten::state()'],['../classmlx_1_1core_1_1_view.html#a37620f6548630bd2d0dd44e9ab084b93',1,'mlx::core::View::state()'],['../classmlx_1_1core_1_1_transpose.html#a23167291e2bf12e2bb2e51d1db340909',1,'mlx::core::Transpose::state()'],['../classmlx_1_1core_1_1_inverse.html#aa1fce744f4a2d660c65901a7542056f2',1,'mlx::core::Inverse::state()'],['../classmlx_1_1core_1_1_cholesky.html#a64f03d32ed249a3b2a59b6af66d23727',1,'mlx::core::Cholesky::state()'],['../classmlx_1_1core_1_1_eigh.html#aa3b6c33b5679c5528863f3de2ab2f914',1,'mlx::core::Eigh::state()']]], + ['status_122',['status',['../classmlx_1_1core_1_1array.html#a7102659be87e9ef62966696ab9b07dad',1,'mlx::core::array']]], + ['std_123',['std',['../group__ops.html#ga2a466024f8061febc0a64be557644cb0',1,'mlx::core::std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#gafdcb04d77c64405a3990078a77dd984c',1,'mlx::core::std(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8',1,'mlx::core::std(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga62721a206df671ef5797449eea97af9f',1,'mlx::core::std(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], + ['steel_5fconv_124',['steel_conv',['../namespacemlx_1_1core_1_1metal.html#a92f1e559b1121d545746f81ff86eaca1',1,'mlx::core::metal']]], + ['steel_5fconv_5fgeneral_125',['steel_conv_general',['../namespacemlx_1_1core_1_1metal.html#a02edb6a90bdf30f4c9f0d6c25b0267b5',1,'mlx::core::metal']]], + ['steel_5fgemm_5ffused_126',['steel_gemm_fused',['../namespacemlx_1_1core_1_1metal.html#a17764366deed71c160fb26091400a803',1,'mlx::core::metal']]], + ['steel_5fgemm_5fmasked_127',['steel_gemm_masked',['../namespacemlx_1_1core_1_1metal.html#a962272ca73d26c08f76f706a128fd71f',1,'mlx::core::metal']]], + ['steel_5fgemm_5fsplitk_128',['steel_gemm_splitk',['../namespacemlx_1_1core_1_1metal.html#ad0dfd40ba7c09755711ceb731e57a5ac',1,'mlx::core::metal']]], + ['steel_5fmatmul_129',['steel_matmul',['../namespacemlx_1_1core.html#aaff208bbac7021c4265580885874499a',1,'mlx::core']]], + ['steel_5fmatmul_5fregular_130',['steel_matmul_regular',['../namespacemlx_1_1core.html#a2a8a09851097571fb51ac5b608550e44',1,'mlx::core']]], + ['step_131',['step',['../structmlx_1_1core_1_1_contiguous_iterator.html#aad921dd422adb0a0f555e19a2f42239c',1,'mlx::core::ContiguousIterator']]], + ['stop_5fcapture_132',['stop_capture',['../namespacemlx_1_1core_1_1metal.html#ac90714424e36fb01e04550de69b8314f',1,'mlx::core::metal']]], + ['stop_5fgradient_133',['stop_gradient',['../group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f',1,'mlx::core']]], + ['stopgradient_134',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html#ac70d1ab819d04e00f76bc25aeebaf84f',1,'mlx::core::StopGradient']]], + ['store_135',['store',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#aa8f50ea8961ec5b35c1b81366d64f2cb',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a2aadaa3239cb3af0c2ee8af9b88c8a98',1,'mlx::steel::MMATile::store(threadgroup U *dst) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a752f708e4fe5ef37fdd902dae153179f',1,'mlx::steel::MMATile::store(device U *dst, const int ld) const'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#aa8f50ea8961ec5b35c1b81366d64f2cb',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a2aadaa3239cb3af0c2ee8af9b88c8a98',1,'mlx::steel::MMATile::store(threadgroup U *dst) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a752f708e4fe5ef37fdd902dae153179f',1,'mlx::steel::MMATile::store(device U *dst, const int ld) const'],['../namespacemlx_1_1core_1_1simd.html#afa2236afddfdec312eb7e27b89a5316a',1,'mlx::core::simd::store()']]], + ['store_5fresult_136',['store_result',['../structmlx_1_1steel_1_1_block_m_m_a.html#a0461451ffb5041b6a916ea17ed34288b',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a0461451ffb5041b6a916ea17ed34288b',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const']]], + ['store_5fresult_5fsafe_137',['store_result_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a081ba538d30d1d02498a7f341e6bd611',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a081ba538d30d1d02498a7f341e6bd611',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const']]], + ['store_5fsafe_138',['store_safe',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1f0b00daad8eba2f855bb306e70d2328',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a57703f522c7409dbe2c0a68bb7acc2ba',1,'mlx::steel::MMATile::store_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1f0b00daad8eba2f855bb306e70d2328',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a57703f522c7409dbe2c0a68bb7acc2ba',1,'mlx::steel::MMATile::store_safe()']]], + ['stream_139',['Stream',['../structmlx_1_1core_1_1_stream.html#a7f0815ff4886da74cbbff5f93d82dd3e',1,'mlx::core::Stream']]], + ['stream_140',['stream',['../classmlx_1_1core_1_1_event.html#a193143bad31b68c699fa27f135b45614',1,'mlx::core::Event::stream()'],['../classmlx_1_1core_1_1_primitive.html#a46e6257397a662528f9f831842ac456a',1,'mlx::core::Primitive::stream()']]], + ['streamcontext_141',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html#a89d803151e9d7dce29382aa83d5c6ef1',1,'mlx::core::StreamContext']]], + ['streamthread_142',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#ac528109a11abcb82e6e221c5efa4493c',1,'mlx::core::scheduler::StreamThread']]], + ['stride_143',['stride',['../classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093',1,'pocketfft::detail::arr_info::stride() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b',1,'pocketfft::detail::arr_info::stride(size_t i) const']]], + ['stride_5fin_144',['stride_in',['../classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c',1,'pocketfft::detail::multi_iter']]], + ['stride_5fout_145',['stride_out',['../classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7',1,'pocketfft::detail::multi_iter']]], + ['strided_5freduce_5fgeneral_5fdispatch_146',['strided_reduce_general_dispatch',['../namespacemlx_1_1core.html#aa0332c64ee9965f05026c30a0b778000',1,'mlx::core']]], + ['strided_5fscan_147',['strided_scan',['../scan_8h.html#a7abb6ffb6c3b96b88c2a63cd4cc2f7ae',1,'scan.h']]], + ['strides_148',['strides',['../classmlx_1_1core_1_1array.html#a28cf1928f5ec2f972a94ff1c0e71187d',1,'mlx::core::array::strides() const'],['../classmlx_1_1core_1_1array.html#ac9bfc251a9937eaefbe7f8c5ffd304d1',1,'mlx::core::array::strides(int dim) const']]], + ['submit_149',['submit',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4',1,'pocketfft::detail::threading::thread_pool']]], + ['subtract_150',['Subtract',['../classmlx_1_1core_1_1_subtract.html#a834854757394f8de7082af65bf86ed9c',1,'mlx::core::Subtract']]], + ['subtract_151',['subtract',['../group__ops.html#ga196c240d3d0fcbb4713802c485e15133',1,'mlx::core']]], + ['sum_152',['sum',['../namespacemlx_1_1core_1_1simd.html#a53b547b886918dc13d4da88eeb8811d2',1,'mlx::core::simd::sum(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#a96ce7d90b3b8b6dddab36ef5b49fffc2',1,'mlx::core::simd::sum(Simd< T, 1 > x)'],['../namespacemlx_1_1core_1_1simd.html#ab0e7c082fc6bed52d522765ef91d205d',1,'mlx::core::simd::sum(Simd< float16_t, N > x)'],['../namespacemlx_1_1steel.html#ab4a6ddea4beb7c447cf5b69b9d46cc3b',1,'mlx::steel::sum(T x)'],['../namespacemlx_1_1steel.html#acd6e194d37b617d7a5818bc384a97fe4',1,'mlx::steel::sum(T x, Us... us)'],['../group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6',1,'mlx::core::sum(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3627754d7868487bdab1bd83f05d9c81',1,'mlx::core::sum(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4',1,'mlx::core::sum(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d',1,'mlx::core::sum(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['svd_153',['SVD',['../classmlx_1_1core_1_1_s_v_d.html#ae89ff583e34fa894cccb8e7a475ee6d1',1,'mlx::core::SVD']]], + ['svd_154',['svd',['../namespacemlx_1_1core_1_1linalg.html#a64364b880e99914cf47bf756fa8dbaf0',1,'mlx::core::linalg']]], + ['swapaxes_155',['swapaxes',['../group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb',1,'mlx::core']]], + ['swizzle_156',['swizzle',['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle::swizzle(uint3 tid, const int swizzle_log)'],['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle::swizzle(uint3 tid, const int swizzle_log)']]], + ['synchronize_157',['synchronize',['../namespacemlx_1_1core.html#a14287949d82ffefad0306cef5eb5f9e4',1,'mlx::core::synchronize()'],['../namespacemlx_1_1core.html#a6648a71937b055e5ff513d98056c2fb5',1,'mlx::core::synchronize(Stream)']]] ]; diff --git a/docs/build/html/search/functions_14.js b/docs/build/html/search/functions_14.js index 2368453c9..3730d8149 100644 --- a/docs/build/html/search/functions_14.js +++ b/docs/build/html/search/functions_14.js @@ -7,9 +7,9 @@ var searchData= ['take_4',['take',['../group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7',1,'mlx::core::take(const array &a, const array &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#ga731af77b7de547a73336f91c829c40b4',1,'mlx::core::take(const array &a, int index, int axis, StreamOrDevice s={})'],['../group__ops.html#ga45d0f423a5e030440ef753f36c5aabf1',1,'mlx::core::take(const array &a, const array &indices, StreamOrDevice s={})'],['../group__ops.html#ga7b3c9b4c8ee02dc23cfd7dcd855dbf20',1,'mlx::core::take(const array &a, int index, StreamOrDevice s={})']]], ['take_5falong_5faxis_5',['take_along_axis',['../group__ops.html#gae0a81d4f983e296a87302e36d65bfc76',1,'mlx::core']]], ['tan_6',['Tan',['../classmlx_1_1core_1_1_tan.html#a8dcc9ff660210ccf05134dd95f47de08',1,'mlx::core::Tan']]], - ['tan_7',['tan',['../namespacemetal.html#a862215a8ddacb086296ba02567c9b158',1,'metal::tan()'],['../namespacemetal_1_1fast.html#ae34754afa152a6170ac2ae3294174506',1,'metal::fast::tan()'],['../namespacemetal_1_1precise.html#a8261ed22f03122ef15b89512358acb1f',1,'metal::precise::tan()'],['../group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5',1,'mlx::core::tan()']]], + ['tan_7',['tan',['../namespacemlx_1_1core_1_1simd.html#a445ddc4ed928656df64d889942588cfd',1,'mlx::core::simd::tan(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a829842f854aecfae93b7d42f83aec9a7',1,'mlx::core::simd::tan(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a5abc381a85fe8b0e9cb472f874704652',1,'mlx::core::simd::tan(Simd< T, 1 > in)'],['../namespacemetal.html#a862215a8ddacb086296ba02567c9b158',1,'metal::tan()'],['../namespacemetal_1_1fast.html#ae34754afa152a6170ac2ae3294174506',1,'metal::fast::tan()'],['../namespacemetal_1_1precise.html#a8261ed22f03122ef15b89512358acb1f',1,'metal::precise::tan()'],['../group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5',1,'mlx::core::tan()']]], ['tanh_8',['Tanh',['../classmlx_1_1core_1_1_tanh.html#ae551297bf573e1802fb831440276dee4',1,'mlx::core::Tanh']]], - ['tanh_9',['tanh',['../namespacemetal.html#aa97fc50bd6addfc6de0aae8570fe963d',1,'metal::tanh()'],['../namespacemetal_1_1fast.html#a13e6e6ae087b7c558e9a94ddbc864d43',1,'metal::fast::tanh()'],['../namespacemetal_1_1precise.html#a741c27a10cc968dd1e63473d9fcd8f99',1,'metal::precise::tanh()'],['../group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d',1,'mlx::core::tanh()']]], + ['tanh_9',['tanh',['../namespacemlx_1_1core_1_1simd.html#aa244fbe7456b653aa50a473108fd6a2b',1,'mlx::core::simd::tanh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ad78f543dc5da87a14ca113a1dd9852fd',1,'mlx::core::simd::tanh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ab80a7db8d99e3f4032e761c60216027d',1,'mlx::core::simd::tanh(Simd< T, 1 > in)'],['../namespacemetal.html#aa97fc50bd6addfc6de0aae8570fe963d',1,'metal::tanh()'],['../namespacemetal_1_1fast.html#a13e6e6ae087b7c558e9a94ddbc864d43',1,'metal::fast::tanh()'],['../namespacemetal_1_1precise.html#a741c27a10cc968dd1e63473d9fcd8f99',1,'metal::precise::tanh()'],['../group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d',1,'mlx::core::tanh()']]], ['tanpi_10',['tanpi',['../namespacemetal.html#ae2046d163a525fc1822a9ec8a0aeaeb3',1,'metal::tanpi()'],['../namespacemetal_1_1fast.html#a39b2952d4adf1400016c63243798aaf8',1,'metal::fast::tanpi()'],['../namespacemetal_1_1precise.html#a8fae8c20deff43a8e855bba6f3ba20a5',1,'metal::precise::tanpi()']]], ['tell_11',['tell',['../classmlx_1_1core_1_1io_1_1_reader.html#a27697ccc1ce45da0233db3bd4f298aed',1,'mlx::core::io::Reader::tell()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a11ad80749894993232fbb5c70fd7b282',1,'mlx::core::io::Writer::tell()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a2e92131428f0ffa98fff781b8c35d9e5',1,'mlx::core::io::ParallelFileReader::tell()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#aa883a722789c962164fd0ddcc5f6ffc5',1,'mlx::core::io::FileWriter::tell()']]], ['tensordot_12',['tensordot',['../group__ops.html#gaf5c9735f4690327e1500e04e728fae70',1,'mlx::core::tensordot(const array &a, const array &b, const int axis=2, StreamOrDevice s={})'],['../group__ops.html#gad7fe00b566f89d607639c1a497cabbc6',1,'mlx::core::tensordot(const array &a, const array &b, const std::vector< int > &axes_a, const std::vector< int > &axes_b, StreamOrDevice s={})']]], @@ -18,38 +18,40 @@ var searchData= ['ternary_5fg_5fnd1_15',['ternary_g_nd1',['../metal_2kernels_2ternary_8h.html#af8400389f3bb11498fb1c4057e638a27',1,'ternary.h']]], ['ternary_5fg_5fnd2_16',['ternary_g_nd2',['../metal_2kernels_2ternary_8h.html#a0971bb39ec881e97af7ab9584e1ce22a',1,'ternary.h']]], ['ternary_5fg_5fnd3_17',['ternary_g_nd3',['../metal_2kernels_2ternary_8h.html#ad7968ba7b638b85ff67a65eef768f59c',1,'ternary.h']]], - ['ternary_5fop_5fgpu_18',['ternary_op_gpu',['../namespacemlx_1_1core.html#aa63e62b6d3906e4cac871d498515a1cd',1,'mlx::core']]], - ['ternary_5fop_5fgpu_5finplace_19',['ternary_op_gpu_inplace',['../namespacemlx_1_1core.html#a37645c0adccb3eb46844115def1a68d7',1,'mlx::core']]], - ['ternary_5fops_20',['ternary_ops',['../namespacemlx_1_1core_1_1metal.html#a11b593b07e9a33e5f78fe4695fb99ec9',1,'mlx::core::metal']]], - ['ternary_5fv_21',['ternary_v',['../metal_2kernels_2ternary_8h.html#a83f93644d21ee774e06e8190d0725ccb',1,'ternary.h']]], - ['ternary_5fv2_22',['ternary_v2',['../metal_2kernels_2ternary_8h.html#a3e610f3b01966bdbf23fdfebe5d2c508',1,'ternary.h']]], - ['thread_5fcount_23',['thread_count',['../structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87',1,'pocketfft::detail::util']]], - ['thread_5ffn_24',['thread_fn',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a06a62c21c1174e4eb4d242e50aad7adf',1,'mlx::core::scheduler::StreamThread']]], - ['thread_5fid_25',['thread_id',['../namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b',1,'pocketfft::detail::threading']]], - ['thread_5fmap_26',['thread_map',['../namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e',1,'pocketfft::detail::threading']]], - ['thread_5fpool_27',['thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560',1,'pocketfft::detail::threading::thread_pool::thread_pool(size_t nthreads)'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0',1,'pocketfft::detail::threading::thread_pool::thread_pool()'],['../namespacemlx_1_1core_1_1io.html#a05f27b765443a178a972abae772e863d',1,'mlx::core::io::thread_pool()']]], - ['thread_5freduce_28',['thread_reduce',['../reduce__row_8h.html#afd80a25fa84e6cc884dcc8698859ade1',1,'reduce_row.h']]], - ['thread_5fswap_29',['thread_swap',['../sort_8h.html#a6e8c2da4975a8001fd5ddf211a3058b7',1,'sort.h']]], - ['threadgroup_5freduce_30',['threadgroup_reduce',['../reduce__row_8h.html#aa146bb611069fd2892f03714fd1cc3cf',1,'reduce_row.h']]], - ['threadpool_31',['ThreadPool',['../class_thread_pool.html#ac291710e33dbbed96ee20711080d506d',1,'ThreadPool']]], - ['threefry2x32_5fhash_32',['threefry2x32_hash',['../namespacemlx_1_1core_1_1random.html#ac7e92c89a2bac1b0bed922a3d4c3c66b',1,'mlx::core::random']]], - ['tile_33',['tile',['../group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361',1,'mlx::core']]], - ['tile_5fmatmad_34',['tile_matmad',['../namespacemlx_1_1steel.html#ad583e6038efc119542410f43b603d4ad',1,'mlx::steel']]], - ['to_5fbnns_5fdtype_35',['to_bnns_dtype',['../namespacemlx_1_1core.html#a7a4193f37b1de9c33c31d1da09c77edb',1,'mlx::core']]], - ['to_5fstream_36',['to_stream',['../namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9',1,'mlx::core']]], - ['topk_37',['topk',['../group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a',1,'mlx::core::topk(const array &a, int k, StreamOrDevice s={})'],['../group__ops.html#ga35b8436c79ff953f6c809598b646f498',1,'mlx::core::topk(const array &a, int k, int axis, StreamOrDevice s={})']]], - ['trace_38',['trace',['../group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, StreamOrDevice s={})'],['../group__ops.html#gaf25c00108feaafaa6350a4434cb0062e',1,'mlx::core::trace(const array &a, StreamOrDevice s={})']]], - ['transformadd_39',['TransformAdd',['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd(const float, const float)'],['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd(const float, const float)']]], - ['transformaxpby_40',['TransformAxpby',['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby(const float alpha_, const float beta_)'],['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby(const float alpha_, const float beta_)']]], - ['transformscale_41',['TransformScale',['../struct_transform_scale.html#ae109cf7c963ba13df96977e7563f7b70',1,'TransformScale']]], - ['transpose_42',['Transpose',['../classmlx_1_1core_1_1_transpose.html#a1a9ba023584c61c7ac93d6dce536760a',1,'mlx::core::Transpose']]], - ['transpose_43',['transpose',['../group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b',1,'mlx::core::transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5',1,'mlx::core::transpose(const array &a, std::initializer_list< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a',1,'mlx::core::transpose(const array &a, StreamOrDevice s={})']]], - ['tri_44',['tri',['../group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78',1,'mlx::core::tri(int n, int m, int k, Dtype type, StreamOrDevice s={})'],['../group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5',1,'mlx::core::tri(int n, Dtype type, StreamOrDevice s={})']]], - ['tri_5finv_45',['tri_inv',['../namespacemlx_1_1core_1_1linalg.html#aba1994571326326717b5b5e38c2e0661',1,'mlx::core::linalg']]], - ['tril_46',['tril',['../group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2',1,'mlx::core']]], - ['triu_47',['triu',['../group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412',1,'mlx::core']]], - ['trunc_48',['trunc',['../namespacemetal.html#a93cb75a11a362bfc8310ea19c554c887',1,'metal::trunc()'],['../namespacemetal_1_1fast.html#aa62e1075e86c626d97038f16e9433415',1,'metal::fast::trunc()'],['../namespacemetal_1_1precise.html#a334183e7a2dd49b983d072d1e8ee2b27',1,'metal::precise::trunc()']]], - ['truncated_5fnormal_49',['truncated_normal',['../namespacemlx_1_1core_1_1random.html#aece7dc5a29e0488d8b9648f340dbff72',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a39663eda0fd7b274d01499a7b1c9035f',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['try_5fpop_50',['try_pop',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43',1,'pocketfft::detail::threading::concurrent_queue']]], - ['type_5fto_5fname_51',['type_to_name',['../namespacemlx_1_1core.html#aef60e3a8d9c987c9c338b193673d2164',1,'mlx::core::type_to_name(const Dtype &t)'],['../namespacemlx_1_1core.html#af1fdfdaa5644394362e6baba30701bae',1,'mlx::core::type_to_name(const array &a)']]] + ['ternary_5fop_18',['ternary_op',['../namespacemlx_1_1core.html#a9dcc3018702ee31c21c8652bdc2182b1',1,'mlx::core']]], + ['ternary_5fop_5fdims_19',['ternary_op_dims',['../namespacemlx_1_1core.html#a8096c7a688ac3f09cca69a3a85f7f157',1,'mlx::core']]], + ['ternary_5fop_5fdispatch_5fdims_20',['ternary_op_dispatch_dims',['../namespacemlx_1_1core.html#ac1c085e305954247d042f5d8803cd85b',1,'mlx::core']]], + ['ternary_5fop_5fgpu_21',['ternary_op_gpu',['../namespacemlx_1_1core.html#aa63e62b6d3906e4cac871d498515a1cd',1,'mlx::core']]], + ['ternary_5fop_5fgpu_5finplace_22',['ternary_op_gpu_inplace',['../namespacemlx_1_1core.html#a37645c0adccb3eb46844115def1a68d7',1,'mlx::core']]], + ['ternary_5fops_23',['ternary_ops',['../namespacemlx_1_1core_1_1metal.html#a11b593b07e9a33e5f78fe4695fb99ec9',1,'mlx::core::metal']]], + ['ternary_5fv_24',['ternary_v',['../metal_2kernels_2ternary_8h.html#a83f93644d21ee774e06e8190d0725ccb',1,'ternary.h']]], + ['ternary_5fv2_25',['ternary_v2',['../metal_2kernels_2ternary_8h.html#a3e610f3b01966bdbf23fdfebe5d2c508',1,'ternary.h']]], + ['thread_5fcount_26',['thread_count',['../structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87',1,'pocketfft::detail::util']]], + ['thread_5ffn_27',['thread_fn',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a06a62c21c1174e4eb4d242e50aad7adf',1,'mlx::core::scheduler::StreamThread']]], + ['thread_5fid_28',['thread_id',['../namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b',1,'pocketfft::detail::threading']]], + ['thread_5fmap_29',['thread_map',['../namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e',1,'pocketfft::detail::threading']]], + ['thread_5fpool_30',['thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560',1,'pocketfft::detail::threading::thread_pool::thread_pool(size_t nthreads)'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0',1,'pocketfft::detail::threading::thread_pool::thread_pool()'],['../namespacemlx_1_1core_1_1io.html#a05f27b765443a178a972abae772e863d',1,'mlx::core::io::thread_pool()']]], + ['thread_5freduce_31',['thread_reduce',['../reduce__row_8h.html#afd80a25fa84e6cc884dcc8698859ade1',1,'reduce_row.h']]], + ['thread_5fswap_32',['thread_swap',['../sort_8h.html#a6e8c2da4975a8001fd5ddf211a3058b7',1,'sort.h']]], + ['threadgroup_5freduce_33',['threadgroup_reduce',['../reduce__row_8h.html#aa146bb611069fd2892f03714fd1cc3cf',1,'reduce_row.h']]], + ['threadpool_34',['ThreadPool',['../class_thread_pool.html#ac291710e33dbbed96ee20711080d506d',1,'ThreadPool']]], + ['threefry2x32_5fhash_35',['threefry2x32_hash',['../namespacemlx_1_1core_1_1random.html#ac7e92c89a2bac1b0bed922a3d4c3c66b',1,'mlx::core::random']]], + ['tile_36',['tile',['../group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361',1,'mlx::core']]], + ['tile_5fmatmad_37',['tile_matmad',['../namespacemlx_1_1steel.html#ad583e6038efc119542410f43b603d4ad',1,'mlx::steel']]], + ['to_5fstream_38',['to_stream',['../namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9',1,'mlx::core']]], + ['topk_39',['topk',['../group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a',1,'mlx::core::topk(const array &a, int k, StreamOrDevice s={})'],['../group__ops.html#ga35b8436c79ff953f6c809598b646f498',1,'mlx::core::topk(const array &a, int k, int axis, StreamOrDevice s={})']]], + ['trace_40',['trace',['../group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, StreamOrDevice s={})'],['../group__ops.html#gaf25c00108feaafaa6350a4434cb0062e',1,'mlx::core::trace(const array &a, StreamOrDevice s={})']]], + ['transformadd_41',['TransformAdd',['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd(const float, const float)'],['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd(const float, const float)']]], + ['transformaxpby_42',['TransformAxpby',['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby(const float alpha_, const float beta_)'],['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby(const float alpha_, const float beta_)']]], + ['transformscale_43',['TransformScale',['../struct_transform_scale.html#ae109cf7c963ba13df96977e7563f7b70',1,'TransformScale']]], + ['transpose_44',['Transpose',['../classmlx_1_1core_1_1_transpose.html#a1a9ba023584c61c7ac93d6dce536760a',1,'mlx::core::Transpose']]], + ['transpose_45',['transpose',['../group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b',1,'mlx::core::transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5',1,'mlx::core::transpose(const array &a, std::initializer_list< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a',1,'mlx::core::transpose(const array &a, StreamOrDevice s={})']]], + ['tri_46',['tri',['../group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78',1,'mlx::core::tri(int n, int m, int k, Dtype type, StreamOrDevice s={})'],['../group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5',1,'mlx::core::tri(int n, Dtype type, StreamOrDevice s={})']]], + ['tri_5finv_47',['tri_inv',['../namespacemlx_1_1core_1_1linalg.html#aba1994571326326717b5b5e38c2e0661',1,'mlx::core::linalg']]], + ['tril_48',['tril',['../group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2',1,'mlx::core']]], + ['triu_49',['triu',['../group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412',1,'mlx::core']]], + ['trunc_50',['trunc',['../namespacemetal.html#a93cb75a11a362bfc8310ea19c554c887',1,'metal::trunc()'],['../namespacemetal_1_1fast.html#aa62e1075e86c626d97038f16e9433415',1,'metal::fast::trunc()'],['../namespacemetal_1_1precise.html#a334183e7a2dd49b983d072d1e8ee2b27',1,'metal::precise::trunc()']]], + ['truncated_5fnormal_51',['truncated_normal',['../namespacemlx_1_1core_1_1random.html#aece7dc5a29e0488d8b9648f340dbff72',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a39663eda0fd7b274d01499a7b1c9035f',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['try_5fpop_52',['try_pop',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43',1,'pocketfft::detail::threading::concurrent_queue']]], + ['type_5fto_5fname_53',['type_to_name',['../namespacemlx_1_1core.html#aef60e3a8d9c987c9c338b193673d2164',1,'mlx::core::type_to_name(const Dtype &t)'],['../namespacemlx_1_1core.html#af1fdfdaa5644394362e6baba30701bae',1,'mlx::core::type_to_name(const array &a)']]] ]; diff --git a/docs/build/html/search/functions_15.js b/docs/build/html/search/functions_15.js index 72cc66742..ea9d13da6 100644 --- a/docs/build/html/search/functions_15.js +++ b/docs/build/html/search/functions_15.js @@ -1,17 +1,21 @@ var searchData= [ ['uint16_5fto_5fbfloat16_0',['uint16_to_bfloat16',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8d066e48cf3e2a0583c71816fa40f7f4',1,'uint16_to_bfloat16(const uint16_t x): bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a8d066e48cf3e2a0583c71816fa40f7f4',1,'uint16_to_bfloat16(const uint16_t x): bf16.h']]], - ['unary_1',['unary',['../namespacemlx_1_1core_1_1metal.html#afac64fd56ac492d6baf6de7e8a00b039',1,'mlx::core::metal']]], - ['unary_5fg_2',['unary_g',['../metal_2kernels_2unary_8h.html#af13d20efb568db3ab7cd7ec0311c87be',1,'unary.h']]], - ['unary_5fop_5fgpu_3',['unary_op_gpu',['../namespacemlx_1_1core.html#aba2b4accc059f30d4dca88db9f7a6e13',1,'mlx::core']]], - ['unary_5fop_5fgpu_5finplace_4',['unary_op_gpu_inplace',['../namespacemlx_1_1core.html#a668fde2bd280a88f63a68b68a343d375',1,'mlx::core']]], - ['unary_5fops_5',['unary_ops',['../namespacemlx_1_1core_1_1metal.html#a17b471fa52ea5f24ee63e081f46528f5',1,'mlx::core::metal']]], - ['unary_5fv_6',['unary_v',['../metal_2kernels_2unary_8h.html#a64e4f6737edddb72122e262977ee3014',1,'unary.h']]], - ['unary_5fv2_7',['unary_v2',['../metal_2kernels_2unary_8h.html#a7c7690f0df9d2acc60b63be58d9c7777',1,'unary.h']]], - ['unaryprimitive_8',['UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html#a189f6d4ed369f82a4b724a29eb056d4e',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(Stream stream)'],['../classmlx_1_1core_1_1_unary_primitive.html#a9935cffc4f246d3d883bc3d26c5163f2',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a780281fb04e2daf1be630c124bd605e3',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(UnaryPrimitive &&other)=delete']]], - ['unflatten_9',['Unflatten',['../classmlx_1_1core_1_1_unflatten.html#a2d1c32eb1fe2bc7641ade600453c7966',1,'mlx::core::Unflatten']]], - ['unflatten_10',['unflatten',['../group__ops.html#ga666bcc2187a144247e8c0c224b016625',1,'mlx::core']]], - ['uniform_11',['uniform',['../namespacemlx_1_1core_1_1random.html#ac461a0be91e448c9887b38b832c61cc2',1,'mlx::core::random::uniform(const array &low, const array &high, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#abe65438fbb52624386f50f77863a2c5e',1,'mlx::core::random::uniform(T low, U high, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a52913f952387ee3943b3c1f572583ac0',1,'mlx::core::random::uniform(const Shape &shape, Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a0ffb2f91da490f372f898ca2f82104a8',1,'mlx::core::random::uniform(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['update_5ffence_12',['update_fence',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aeef08f5f3c015578d40de756a6465aa2',1,'mlx::core::metal::CommandEncoder::update_fence()'],['../structmlx_1_1core_1_1_command_encoder.html#aeef08f5f3c015578d40de756a6465aa2',1,'mlx::core::CommandEncoder::update_fence()']]], - ['utils_13',['utils',['../namespacemlx_1_1core_1_1metal.html#a529dc6c2d4a37ba544b66b2c3cd792cc',1,'mlx::core::metal']]] + ['unary_1',['unary',['../namespacemlx_1_1core.html#a6c8fdd03ef891d7f47804bf02e9a8507',1,'mlx::core::unary()'],['../namespacemlx_1_1core_1_1metal.html#afac64fd56ac492d6baf6de7e8a00b039',1,'mlx::core::metal::unary()']]], + ['unary_5ffp_2',['unary_fp',['../namespacemlx_1_1core.html#a76a2cb4634f5fd6970a8c3b3753d7a4a',1,'mlx::core']]], + ['unary_5fg_3',['unary_g',['../metal_2kernels_2unary_8h.html#af13d20efb568db3ab7cd7ec0311c87be',1,'unary.h']]], + ['unary_5fop_4',['unary_op',['../namespacemlx_1_1core.html#a27f00519f9756896734fd4d47fec0625',1,'mlx::core::unary_op(const T *a, U *out, Op op, size_t shape, size_t stride)'],['../namespacemlx_1_1core.html#ae20f207ad1ed3badc17cecf08f118b5e',1,'mlx::core::unary_op(const array &a, array &out, Op op)']]], + ['unary_5fop_5fgpu_5',['unary_op_gpu',['../namespacemlx_1_1core.html#aba2b4accc059f30d4dca88db9f7a6e13',1,'mlx::core']]], + ['unary_5fop_5fgpu_5finplace_6',['unary_op_gpu_inplace',['../namespacemlx_1_1core.html#a668fde2bd280a88f63a68b68a343d375',1,'mlx::core']]], + ['unary_5fops_7',['unary_ops',['../namespacemlx_1_1core_1_1metal.html#a17b471fa52ea5f24ee63e081f46528f5',1,'mlx::core::metal']]], + ['unary_5fv_8',['unary_v',['../metal_2kernels_2unary_8h.html#a64e4f6737edddb72122e262977ee3014',1,'unary.h']]], + ['unary_5fv2_9',['unary_v2',['../metal_2kernels_2unary_8h.html#a7c7690f0df9d2acc60b63be58d9c7777',1,'unary.h']]], + ['unaryprimitive_10',['UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html#a189f6d4ed369f82a4b724a29eb056d4e',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(Stream stream)'],['../classmlx_1_1core_1_1_unary_primitive.html#a9935cffc4f246d3d883bc3d26c5163f2',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a780281fb04e2daf1be630c124bd605e3',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(UnaryPrimitive &&other)=delete']]], + ['unflatten_11',['Unflatten',['../classmlx_1_1core_1_1_unflatten.html#a2d1c32eb1fe2bc7641ade600453c7966',1,'mlx::core::Unflatten']]], + ['unflatten_12',['unflatten',['../group__ops.html#ga666bcc2187a144247e8c0c224b016625',1,'mlx::core']]], + ['uniform_13',['uniform',['../namespacemlx_1_1core_1_1random.html#ac461a0be91e448c9887b38b832c61cc2',1,'mlx::core::random::uniform(const array &low, const array &high, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#abe65438fbb52624386f50f77863a2c5e',1,'mlx::core::random::uniform(T low, U high, const Shape &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a52913f952387ee3943b3c1f572583ac0',1,'mlx::core::random::uniform(const Shape &shape, Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a0ffb2f91da490f372f898ca2f82104a8',1,'mlx::core::random::uniform(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['update_14',['update',['../classmlx_1_1core_1_1_fence.html#a653279d4023d69751a930a91d3bf010a',1,'mlx::core::Fence']]], + ['update_5ffence_15',['update_fence',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aeef08f5f3c015578d40de756a6465aa2',1,'mlx::core::metal::CommandEncoder::update_fence()'],['../structmlx_1_1core_1_1_command_encoder.html#aeef08f5f3c015578d40de756a6465aa2',1,'mlx::core::CommandEncoder::update_fence()']]], + ['update_5fgpu_16',['update_gpu',['../classmlx_1_1core_1_1_fence.html#a6c5652aad6e93b06c72258bb8d9c19fc',1,'mlx::core::Fence']]], + ['utils_17',['utils',['../namespacemlx_1_1core_1_1metal.html#a529dc6c2d4a37ba544b66b2c3cd792cc',1,'mlx::core::metal']]] ]; diff --git a/docs/build/html/search/functions_16.js b/docs/build/html/search/functions_16.js index f8460f5e4..34767aacc 100644 --- a/docs/build/html/search/functions_16.js +++ b/docs/build/html/search/functions_16.js @@ -5,10 +5,12 @@ var searchData= ['value_2',['value',['../classmlx_1_1core_1_1_event.html#ab71c7baee3d1d02ad6a2001bbf90b970',1,'mlx::core::Event']]], ['value_5fand_5fgrad_3',['value_and_grad',['../namespacemlx_1_1core.html#abf49b337a00997231c0f7fd389efa8f3',1,'mlx::core::value_and_grad(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#a7b987f404b8699de00f9e0099ab6b1b0',1,'mlx::core::value_and_grad(const std::function< std::vector< array >(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a5a64dc878b29403d27e50bd7a288cc04',1,'mlx::core::value_and_grad(const std::function< array(const array &)> &fun)'],['../namespacemlx_1_1core.html#a7620f1ae298127cb6181db9162f012a7',1,'mlx::core::value_and_grad(const std::function< array(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#a2f69ffc30d66b1fca8f24b65be161a51',1,'mlx::core::value_and_grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)']]], ['var_4',['var',['../group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9',1,'mlx::core::var(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga7d7b38d118fa2613214078ef0f7d5a42',1,'mlx::core::var(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga78ddeb966cbe7a5b0aa17e1de43025f2',1,'mlx::core::var(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga4fbf3e3f98f2e4956faf87af320aa9d0',1,'mlx::core::var(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], - ['view_5',['View',['../classmlx_1_1core_1_1_view.html#ad7eed156c308e9a29a8b41f965ec941e',1,'mlx::core::View']]], - ['view_6',['view',['../group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02',1,'mlx::core']]], - ['vjp_7',['vjp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abbf6d1d63dcda207ad7d9eeb4fc36225',1,'mlx::core::distributed::AllReduce::vjp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#aa5eff6fc128b71220899aab8ab9116fb',1,'mlx::core::distributed::AllGather::vjp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a74be4bcd0382f7f6400bf73fd5569c91',1,'mlx::core::fast::Custom::vjp()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#aacfbbbc15fcee0a5ce4f519ca3cca5eb',1,'mlx::core::fast::RMSNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#ae5e1b5df0705a6b1d141691a4396b0b6',1,'mlx::core::fast::LayerNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#ad999105414badd66c8fd9e069454a533',1,'mlx::core::fast::RoPE::vjp()'],['../classmlx_1_1core_1_1_primitive.html#a1dcb6807326eeab62474c6a0e3836d42',1,'mlx::core::Primitive::vjp()'],['../classmlx_1_1core_1_1_abs.html#aa2dd8ec0989e716b77394ac349b34592',1,'mlx::core::Abs::vjp()'],['../classmlx_1_1core_1_1_add.html#ac28e581862880e24ed2b99bb6a916607',1,'mlx::core::Add::vjp()'],['../classmlx_1_1core_1_1_add_m_m.html#ac1562a37cec6928e01281926ebeb47c6',1,'mlx::core::AddMM::vjp()'],['../classmlx_1_1core_1_1_arc_cos.html#a78e73e5e639d1249c7fe9614bf157c92',1,'mlx::core::ArcCos::vjp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a856c677f16e2b3f2edd2491e35db2d26',1,'mlx::core::ArcCosh::vjp()'],['../classmlx_1_1core_1_1_arc_sin.html#ab4057cd5ef1a8359f97493018e10d3a1',1,'mlx::core::ArcSin::vjp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a7988ee5b9e1e7e498dcab73d61ba147e',1,'mlx::core::ArcSinh::vjp()'],['../classmlx_1_1core_1_1_arc_tan.html#a5fefc3634b96a67ff8ae011a8ee180c2',1,'mlx::core::ArcTan::vjp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a99840c282e37b2b2a9c312e6e8ade1d2',1,'mlx::core::ArcTan2::vjp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a07da5797f7aaf3dfe43bf24e8562ac72',1,'mlx::core::ArcTanh::vjp()'],['../classmlx_1_1core_1_1_arg_partition.html#ade23d014717a0b0235d00073503aeac0',1,'mlx::core::ArgPartition::vjp()'],['../classmlx_1_1core_1_1_arg_reduce.html#a60d272685a373e6fe879416481a1ce1a',1,'mlx::core::ArgReduce::vjp()'],['../classmlx_1_1core_1_1_as_type.html#ac38a4f889311a3b5e5be9a67dcb93e18',1,'mlx::core::AsType::vjp()'],['../classmlx_1_1core_1_1_as_strided.html#a34783284c9b2f5b4a62c3c3ee5dd4062',1,'mlx::core::AsStrided::vjp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6131ed1c317ff8700a3e9b13fdaa9d61',1,'mlx::core::BitwiseBinary::vjp()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a1adf20087ee2f685bf39c2724b8e7120',1,'mlx::core::BlockMaskedMM::vjp()'],['../classmlx_1_1core_1_1_gather_m_m.html#a76c9f27c57354f6230b43944882e1bda',1,'mlx::core::GatherMM::vjp()'],['../classmlx_1_1core_1_1_broadcast_axes.html#aea8ef2b2616568a2bb56695381a035be',1,'mlx::core::BroadcastAxes::vjp()'],['../classmlx_1_1core_1_1_broadcast.html#a0318847c9be40f00b23907ad56037d18',1,'mlx::core::Broadcast::vjp()'],['../classmlx_1_1core_1_1_ceil.html#ac2f5a2bd84b8f013e5ce688419a88acb',1,'mlx::core::Ceil::vjp()'],['../classmlx_1_1core_1_1_compiled.html#a32462e65c52f84b708188130cc508133',1,'mlx::core::Compiled::vjp()'],['../classmlx_1_1core_1_1_concatenate.html#a8155db9100ec3b8bd0bc94baeaeee3b0',1,'mlx::core::Concatenate::vjp()'],['../classmlx_1_1core_1_1_contiguous.html#abf488f02057fd5852f38b2e8a600ad2a',1,'mlx::core::Contiguous::vjp()'],['../classmlx_1_1core_1_1_convolution.html#af8eb9c0c055ad20aa74b547016917690',1,'mlx::core::Convolution::vjp()'],['../classmlx_1_1core_1_1_copy.html#a6c4dee582001e9983e9517485ee37efd',1,'mlx::core::Copy::vjp()'],['../classmlx_1_1core_1_1_cos.html#a51d84113728e651ef9d4a1fe671c4d00',1,'mlx::core::Cos::vjp()'],['../classmlx_1_1core_1_1_cosh.html#a0791abd4305a333fb3b181a5357ce0f4',1,'mlx::core::Cosh::vjp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa1da36cef632df767cd9809d6cf06209',1,'mlx::core::CustomTransforms::vjp()'],['../classmlx_1_1core_1_1_depends.html#a02996fa45f01f7cb9f37074d5f8ccab0',1,'mlx::core::Depends::vjp()'],['../classmlx_1_1core_1_1_divide.html#ad3af7c70cad22c1a1a75b4a78ef793b6',1,'mlx::core::Divide::vjp()'],['../classmlx_1_1core_1_1_div_mod.html#a8c914a07f666a1d9377a27ed5d55e7c1',1,'mlx::core::DivMod::vjp()'],['../classmlx_1_1core_1_1_select.html#a9b522487b78fceeca7f827cd1c29a9a3',1,'mlx::core::Select::vjp()'],['../classmlx_1_1core_1_1_remainder.html#ab18f7bca1027ae71847a50da0933cec6',1,'mlx::core::Remainder::vjp()'],['../classmlx_1_1core_1_1_equal.html#af3c1bfcd1bf50922fc00e302bb193736',1,'mlx::core::Equal::vjp()'],['../classmlx_1_1core_1_1_erf.html#a1f529e95a42a2d69a8b18979d3ee2909',1,'mlx::core::Erf::vjp()'],['../classmlx_1_1core_1_1_erf_inv.html#a48afff12a58ddefae7ae0245c3580189',1,'mlx::core::ErfInv::vjp()'],['../classmlx_1_1core_1_1_exp.html#a94b9b7d137c3640d290b96c5e8b7e1a8',1,'mlx::core::Exp::vjp()'],['../classmlx_1_1core_1_1_expm1.html#af6ce416169190479c9792bb9cdbe2f43',1,'mlx::core::Expm1::vjp()'],['../classmlx_1_1core_1_1_expand_dims.html#a2fb3c65ba7a3b2d1f33a3c681fda8896',1,'mlx::core::ExpandDims::vjp()'],['../classmlx_1_1core_1_1_f_f_t.html#aafc895614a6e368c0e6d64af20d01090',1,'mlx::core::FFT::vjp()'],['../classmlx_1_1core_1_1_flatten.html#ab549a8c38b63055e2d5cd672f7676aab',1,'mlx::core::Flatten::vjp()'],['../classmlx_1_1core_1_1_floor.html#a589e2cf99b6fd1a5ba85534a2a31338e',1,'mlx::core::Floor::vjp()'],['../classmlx_1_1core_1_1_full.html#a49e76e7a8641f990701abc1b3bd49969',1,'mlx::core::Full::vjp()'],['../classmlx_1_1core_1_1_gather.html#aacf612a8f5f1cdbbfd19707d8d33c426',1,'mlx::core::Gather::vjp()'],['../classmlx_1_1core_1_1_greater.html#a341766a8a7e41d2a1160d35d4e781679',1,'mlx::core::Greater::vjp()'],['../classmlx_1_1core_1_1_greater_equal.html#a62f07a4ac54c708307c82aac0e5693ee',1,'mlx::core::GreaterEqual::vjp()'],['../classmlx_1_1core_1_1_hadamard.html#af4134775427b8998d66f489468b98656',1,'mlx::core::Hadamard::vjp()'],['../classmlx_1_1core_1_1_imag.html#a80da5fdd0fa549eebd7804c0e261848b',1,'mlx::core::Imag::vjp()'],['../classmlx_1_1core_1_1_less.html#aaf205d389b5e602e0814b68f66de8f50',1,'mlx::core::Less::vjp()'],['../classmlx_1_1core_1_1_less_equal.html#aab2aab7590c299885e815c18eedd1028',1,'mlx::core::LessEqual::vjp()'],['../classmlx_1_1core_1_1_log.html#a40885dccfbf928c4d035881be1d49280',1,'mlx::core::Log::vjp()'],['../classmlx_1_1core_1_1_log1p.html#a3113c1d2b4c5e73d0b470f42dc48a880',1,'mlx::core::Log1p::vjp()'],['../classmlx_1_1core_1_1_logical_not.html#af2c3c241cf3910fbaba013c69d052a50',1,'mlx::core::LogicalNot::vjp()'],['../classmlx_1_1core_1_1_logical_and.html#ae42f8fc454577b0fd6410cae9d5f3b54',1,'mlx::core::LogicalAnd::vjp()'],['../classmlx_1_1core_1_1_logical_or.html#a51aed488f52d5031998689af9cb17847',1,'mlx::core::LogicalOr::vjp()'],['../classmlx_1_1core_1_1_log_add_exp.html#ae231af0ed24a93eb647ee58c2d2b20b4',1,'mlx::core::LogAddExp::vjp()'],['../classmlx_1_1core_1_1_matmul.html#a524136cca481598ea20894d85ca66bb0',1,'mlx::core::Matmul::vjp()'],['../classmlx_1_1core_1_1_maximum.html#a7de15d7b28784e24bbfc7e85ddcbcff3',1,'mlx::core::Maximum::vjp()'],['../classmlx_1_1core_1_1_minimum.html#a48a0cbe3a6c4f7473c00e343f63b5204',1,'mlx::core::Minimum::vjp()'],['../classmlx_1_1core_1_1_multiply.html#a74b7556ec03e2c3d3f971666d06f5db1',1,'mlx::core::Multiply::vjp()'],['../classmlx_1_1core_1_1_negative.html#a889585f056d33bda30c30311257af52a',1,'mlx::core::Negative::vjp()'],['../classmlx_1_1core_1_1_not_equal.html#a0361f29f4ae1235bdf3f3304527e2d4b',1,'mlx::core::NotEqual::vjp()'],['../classmlx_1_1core_1_1_pad.html#ad8a7e547644f2717a24322968e971038',1,'mlx::core::Pad::vjp()'],['../classmlx_1_1core_1_1_partition.html#a7110772b6cd2d430a2b825cf5c952ca9',1,'mlx::core::Partition::vjp()'],['../classmlx_1_1core_1_1_power.html#a1453bb8307d6ff33134f1e00263bf082',1,'mlx::core::Power::vjp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#acb975e272b4a88ab232ef7f7c3a2bf26',1,'mlx::core::QuantizedMatmul::vjp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#ae08a4b7d28902d46f39e66beeb0e23ab',1,'mlx::core::GatherQMM::vjp()'],['../classmlx_1_1core_1_1_real.html#a29f6109339c5141a862ceae72c8b80fe',1,'mlx::core::Real::vjp()'],['../classmlx_1_1core_1_1_reshape.html#ab17294ecc6b5d4e89626fb48c7516365',1,'mlx::core::Reshape::vjp()'],['../classmlx_1_1core_1_1_reduce.html#a684883d2a96315f548ca769510e28e4e',1,'mlx::core::Reduce::vjp()'],['../classmlx_1_1core_1_1_round.html#af8f085e08b7fa8840c52a20b12ca35ce',1,'mlx::core::Round::vjp()'],['../classmlx_1_1core_1_1_scan.html#aaf13f72620b4b5d6a20e1228930e848e',1,'mlx::core::Scan::vjp()'],['../classmlx_1_1core_1_1_scatter.html#a0b51287fba789bb139ed61d40a0c636a',1,'mlx::core::Scatter::vjp()'],['../classmlx_1_1core_1_1_sigmoid.html#aac2f56a4c8362e36a28e232758ca52cf',1,'mlx::core::Sigmoid::vjp()'],['../classmlx_1_1core_1_1_sign.html#aa60ac52edd739fbdf388a997acd01bce',1,'mlx::core::Sign::vjp()'],['../classmlx_1_1core_1_1_sin.html#aedefe550ab4b0687858981bc0bcfbfa0',1,'mlx::core::Sin::vjp()'],['../classmlx_1_1core_1_1_sinh.html#a6b39fdd429bbb4de389e7c904fd561f0',1,'mlx::core::Sinh::vjp()'],['../classmlx_1_1core_1_1_slice.html#a291746a527ff991b66249fb2b54b685f',1,'mlx::core::Slice::vjp()'],['../classmlx_1_1core_1_1_slice_update.html#aedcdc60a0477997a96306c02b66d3f77',1,'mlx::core::SliceUpdate::vjp()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a29caf03256945f7732a52d551191f8fa',1,'mlx::core::DynamicSlice::vjp()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ab2817cb9d1bfcd3de6454d841909da1f',1,'mlx::core::DynamicSliceUpdate::vjp()'],['../classmlx_1_1core_1_1_softmax.html#abb68c311c45ee422a7c966accde9041b',1,'mlx::core::Softmax::vjp()'],['../classmlx_1_1core_1_1_sort.html#a3a8900dce53ee4eb7a1b83806e629358',1,'mlx::core::Sort::vjp()'],['../classmlx_1_1core_1_1_split.html#a7e8730f9cffa9872fff6f8d577031674',1,'mlx::core::Split::vjp()'],['../classmlx_1_1core_1_1_square.html#abcd9516da7f02dc906368c23b0bca263',1,'mlx::core::Square::vjp()'],['../classmlx_1_1core_1_1_sqrt.html#a08a21bd2c3a016f042d95aca294e68f3',1,'mlx::core::Sqrt::vjp()'],['../classmlx_1_1core_1_1_subtract.html#a3a3322be7c3bcaa0397cf099091df16b',1,'mlx::core::Subtract::vjp()'],['../classmlx_1_1core_1_1_squeeze.html#a8d95a13d7cc5586d48a38e9199180d06',1,'mlx::core::Squeeze::vjp()'],['../classmlx_1_1core_1_1_tan.html#a4639836cff03d73c769387d6943e92d7',1,'mlx::core::Tan::vjp()'],['../classmlx_1_1core_1_1_tanh.html#afe7b05e2b36b99c3a1b66f0cd3544e95',1,'mlx::core::Tanh::vjp()'],['../classmlx_1_1core_1_1_unflatten.html#a34f1218fa1d0e28f3ee10b65e6b0e319',1,'mlx::core::Unflatten::vjp()'],['../classmlx_1_1core_1_1_transpose.html#ac7805aa29b34afdf8852554f1e759f80',1,'mlx::core::Transpose::vjp()'],['../namespacemlx_1_1core.html#a1b33e2c2e3471420490cf0be2de6de18',1,'mlx::core::vjp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &cotangents)'],['../namespacemlx_1_1core.html#a2065a11249c3f4356ffd69b7a8c487ff',1,'mlx::core::vjp(const std::function< array(const array &)> &fun, const array &primal, const array &cotangent)']]], - ['vmap_8',['vmap',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a3f2dc71859847ca675ec4bfbe125035a',1,'mlx::core::distributed::AllReduce::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ad532d1d51f089dec3c84799b724ea031',1,'mlx::core::distributed::AllGather::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a5cfb66191b9e8b86649da77af55b0f93',1,'mlx::core::distributed::Send::vmap()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a7f4c3a4c48c6807faa36fb31e39dad8d',1,'mlx::core::fast::Custom::vmap()'],['../classmlx_1_1core_1_1_primitive.html#ac632b9619dd7a6a0f177bd36202e8103',1,'mlx::core::Primitive::vmap()'],['../classmlx_1_1core_1_1_abs.html#a4c9c98f1d71432fd3752ad9a6a8e7f2f',1,'mlx::core::Abs::vmap()'],['../classmlx_1_1core_1_1_add.html#a0e557d4d896153f84a25532562e4c646',1,'mlx::core::Add::vmap()'],['../classmlx_1_1core_1_1_add_m_m.html#a73ce80b3a37ec2523943028d50ebce81',1,'mlx::core::AddMM::vmap()'],['../classmlx_1_1core_1_1_arc_cos.html#a7548e23ace6827674aa6d284d44ccf83',1,'mlx::core::ArcCos::vmap()'],['../classmlx_1_1core_1_1_arc_cosh.html#af8ff78e910a9e485a203e1d3347bd461',1,'mlx::core::ArcCosh::vmap()'],['../classmlx_1_1core_1_1_arc_sin.html#a7cabb1e5a2bda44944378822c671ec82',1,'mlx::core::ArcSin::vmap()'],['../classmlx_1_1core_1_1_arc_sinh.html#a9e72b9751939387c333b5d4e19a37f6d',1,'mlx::core::ArcSinh::vmap()'],['../classmlx_1_1core_1_1_arc_tan.html#a1fb921554544a56498bc54f82e4a0556',1,'mlx::core::ArcTan::vmap()'],['../classmlx_1_1core_1_1_arc_tan2.html#ae02cb9fbf25e93dc1d7fbc9e3fb28634',1,'mlx::core::ArcTan2::vmap()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6ddcae68873559211cb91e7740dfc040',1,'mlx::core::ArcTanh::vmap()'],['../classmlx_1_1core_1_1_arg_partition.html#a441093795bcc31495ab5fbc9957b740a',1,'mlx::core::ArgPartition::vmap()'],['../classmlx_1_1core_1_1_arg_reduce.html#abfec42fa06ea15edaf393593751fb1ba',1,'mlx::core::ArgReduce::vmap()'],['../classmlx_1_1core_1_1_arg_sort.html#a3522bbbe4626a467394c1a8a9d7ac34e',1,'mlx::core::ArgSort::vmap()'],['../classmlx_1_1core_1_1_as_type.html#a7ebaf86fd6cad4a1ecfd7cde1ee0b0cc',1,'mlx::core::AsType::vmap()'],['../classmlx_1_1core_1_1_bitwise_binary.html#aa10be55f05bc1868bf4b375dc475f965',1,'mlx::core::BitwiseBinary::vmap()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a4e04f564d440e2d312c335db52c308e1',1,'mlx::core::BroadcastAxes::vmap()'],['../classmlx_1_1core_1_1_broadcast.html#aee4c71c2588ad01eb57e10f346cd666f',1,'mlx::core::Broadcast::vmap()'],['../classmlx_1_1core_1_1_ceil.html#ae86819990b43bdb0c2b3a25719b3a7a4',1,'mlx::core::Ceil::vmap()'],['../classmlx_1_1core_1_1_compiled.html#a732e7548f53977b4513bb7f30a04c30d',1,'mlx::core::Compiled::vmap()'],['../classmlx_1_1core_1_1_concatenate.html#a58c54dcf8e4b045d25edd3afc2caffc1',1,'mlx::core::Concatenate::vmap()'],['../classmlx_1_1core_1_1_conjugate.html#a2c7632c8ae0ca07777e23a0a79344e60',1,'mlx::core::Conjugate::vmap()'],['../classmlx_1_1core_1_1_contiguous.html#a563221e90b15aa90bfae23d29c10e4ec',1,'mlx::core::Contiguous::vmap()'],['../classmlx_1_1core_1_1_copy.html#a669b10253c15b769d90058d1ad7d0e61',1,'mlx::core::Copy::vmap()'],['../classmlx_1_1core_1_1_cos.html#aec9460daf0131156734013d03b230cd6',1,'mlx::core::Cos::vmap()'],['../classmlx_1_1core_1_1_cosh.html#a1ab2386e7d96219b6e4a525f7dac0406',1,'mlx::core::Cosh::vmap()'],['../classmlx_1_1core_1_1_custom_transforms.html#a906a2ff30d9c5281fbf1fa927e4c021b',1,'mlx::core::CustomTransforms::vmap()'],['../classmlx_1_1core_1_1_divide.html#a83e7da52831165b3a026e97b63770242',1,'mlx::core::Divide::vmap()'],['../classmlx_1_1core_1_1_div_mod.html#ae709e0fdd83994bd1d156e0d0e6a7942',1,'mlx::core::DivMod::vmap()'],['../classmlx_1_1core_1_1_select.html#a84e80361c8cf02536b4b98098793550f',1,'mlx::core::Select::vmap()'],['../classmlx_1_1core_1_1_remainder.html#a79867e1099a2e3c2d3e87407b2ab6e3d',1,'mlx::core::Remainder::vmap()'],['../classmlx_1_1core_1_1_equal.html#aea9cc3c88924ac824d72c39c2e83b0ca',1,'mlx::core::Equal::vmap()'],['../classmlx_1_1core_1_1_erf.html#abe554f553356654a3e800ba368108aaa',1,'mlx::core::Erf::vmap()'],['../classmlx_1_1core_1_1_erf_inv.html#ad5d7634e8568af8cc4a54a558a48d0e9',1,'mlx::core::ErfInv::vmap()'],['../classmlx_1_1core_1_1_exp.html#a0fcd579fe148b4c3dbc72e514b81bb37',1,'mlx::core::Exp::vmap()'],['../classmlx_1_1core_1_1_expm1.html#aa4caa848b2ea97e71ee3dd33de039296',1,'mlx::core::Expm1::vmap()'],['../classmlx_1_1core_1_1_expand_dims.html#a380c9ddc25a1f973c3d71b42f8a19486',1,'mlx::core::ExpandDims::vmap()'],['../classmlx_1_1core_1_1_f_f_t.html#ac32d6cc9b67289124f855ea68a61ede1',1,'mlx::core::FFT::vmap()'],['../classmlx_1_1core_1_1_flatten.html#a244a03915313286d36ed4d36b01a99f2',1,'mlx::core::Flatten::vmap()'],['../classmlx_1_1core_1_1_floor.html#aea4dc79a65774990e775ad49519a5d10',1,'mlx::core::Floor::vmap()'],['../classmlx_1_1core_1_1_full.html#afc57ab6bd9ebdbbf042af54a59785d95',1,'mlx::core::Full::vmap()'],['../classmlx_1_1core_1_1_gather.html#abab0c4c204e66489825ce80d2194a275',1,'mlx::core::Gather::vmap()'],['../classmlx_1_1core_1_1_greater.html#a6d8267411fc4951de781f9e8e6c53aa0',1,'mlx::core::Greater::vmap()'],['../classmlx_1_1core_1_1_greater_equal.html#ab0e1be93eb01b0ce7fa83e953f5e3e1d',1,'mlx::core::GreaterEqual::vmap()'],['../classmlx_1_1core_1_1_hadamard.html#a9f1a172e6246859e813002abe9b8f99c',1,'mlx::core::Hadamard::vmap()'],['../classmlx_1_1core_1_1_imag.html#ace9906672bd88df0573653883d58ecb3',1,'mlx::core::Imag::vmap()'],['../classmlx_1_1core_1_1_less.html#a5fee5956cf087d8405359121aa62ba7e',1,'mlx::core::Less::vmap()'],['../classmlx_1_1core_1_1_less_equal.html#a3d5df21db184f2b7620cda9da1684480',1,'mlx::core::LessEqual::vmap()'],['../classmlx_1_1core_1_1_log.html#a007ddbcf911093231f607a8b9ed5cd49',1,'mlx::core::Log::vmap()'],['../classmlx_1_1core_1_1_log1p.html#a7122576f95ce479926bbbbc690891f71',1,'mlx::core::Log1p::vmap()'],['../classmlx_1_1core_1_1_logical_not.html#a5308a271619ee74df561b0aaf525915d',1,'mlx::core::LogicalNot::vmap()'],['../classmlx_1_1core_1_1_logical_and.html#aacc5f6f53ffc327b7771485e3da2a4e5',1,'mlx::core::LogicalAnd::vmap()'],['../classmlx_1_1core_1_1_logical_or.html#a6e2e77e6aaf47872b2e96b151c32daf3',1,'mlx::core::LogicalOr::vmap()'],['../classmlx_1_1core_1_1_log_add_exp.html#a82190aa1421a9734b6e9480debffac78',1,'mlx::core::LogAddExp::vmap()'],['../classmlx_1_1core_1_1_matmul.html#a3a1c6e70bac300240760fe41a58340c2',1,'mlx::core::Matmul::vmap()'],['../classmlx_1_1core_1_1_maximum.html#ab664918e0d71cfec1318a9879e78c5d3',1,'mlx::core::Maximum::vmap()'],['../classmlx_1_1core_1_1_minimum.html#adab0f31acf68075a0be908d8eb882980',1,'mlx::core::Minimum::vmap()'],['../classmlx_1_1core_1_1_multiply.html#ae7e82c8fc8cbaf4e00c27eb54fac7dbf',1,'mlx::core::Multiply::vmap()'],['../classmlx_1_1core_1_1_negative.html#a1f8a6079e272f1a0599f88a1a8419cf0',1,'mlx::core::Negative::vmap()'],['../classmlx_1_1core_1_1_not_equal.html#ab8b57932f03c8eee664bf89adeaa43b5',1,'mlx::core::NotEqual::vmap()'],['../classmlx_1_1core_1_1_number_of_elements.html#a977d83eae845b8bd8c0b98b48cb1c6c2',1,'mlx::core::NumberOfElements::vmap()'],['../classmlx_1_1core_1_1_pad.html#a85658812a0f3275ba3eb74b7c75686cf',1,'mlx::core::Pad::vmap()'],['../classmlx_1_1core_1_1_partition.html#aa0cc55e4d4d2cb5d129d32832321df2c',1,'mlx::core::Partition::vmap()'],['../classmlx_1_1core_1_1_power.html#a5e22749592413a9adbdc877b03b87c8f',1,'mlx::core::Power::vmap()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a3434394140177b285f971c9ffe7e8763',1,'mlx::core::QuantizedMatmul::vmap()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a13ce5e138ebddb8780a034452f68892f',1,'mlx::core::GatherQMM::vmap()'],['../classmlx_1_1core_1_1_random_bits.html#a0dc12f053c6492f934bc18031412c415',1,'mlx::core::RandomBits::vmap()'],['../classmlx_1_1core_1_1_real.html#a07fbbefb6a1bc1ebd3985b24c36693b6',1,'mlx::core::Real::vmap()'],['../classmlx_1_1core_1_1_reshape.html#ae239dd3c6cab147e4af572dc58204f9d',1,'mlx::core::Reshape::vmap()'],['../classmlx_1_1core_1_1_reduce.html#abab1b5aa01ccad44f213f510c3596b38',1,'mlx::core::Reduce::vmap()'],['../classmlx_1_1core_1_1_round.html#a6fad8799a7982e1ccbe05be7cc38a7fd',1,'mlx::core::Round::vmap()'],['../classmlx_1_1core_1_1_scan.html#a297c7cc89c9bf9d186ebdebb634c7804',1,'mlx::core::Scan::vmap()'],['../classmlx_1_1core_1_1_scatter.html#a696c38b373a7a7c71bc112bd1117e322',1,'mlx::core::Scatter::vmap()'],['../classmlx_1_1core_1_1_sigmoid.html#a12712c23037e38192cbccd2d4b14cc85',1,'mlx::core::Sigmoid::vmap()'],['../classmlx_1_1core_1_1_sign.html#aa7296045907015b4e0ae8a93e5e6e295',1,'mlx::core::Sign::vmap()'],['../classmlx_1_1core_1_1_sin.html#a45533996f3d72d9dd97d4c61cd684fba',1,'mlx::core::Sin::vmap()'],['../classmlx_1_1core_1_1_sinh.html#ae171df22bc34c32e31b8135dc4caa788',1,'mlx::core::Sinh::vmap()'],['../classmlx_1_1core_1_1_slice.html#ae33583b0db22fcfeae34dfe1c0e3eaa2',1,'mlx::core::Slice::vmap()'],['../classmlx_1_1core_1_1_slice_update.html#adbf1c76de6ab2f986758530d351d6fa3',1,'mlx::core::SliceUpdate::vmap()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a825a6d4d1499b287525462854b841ef2',1,'mlx::core::DynamicSlice::vmap()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a750fb3548d8f3a5c6f4e54958649936f',1,'mlx::core::DynamicSliceUpdate::vmap()'],['../classmlx_1_1core_1_1_softmax.html#ad22d3dcc71054d3dba529cf2dc981e19',1,'mlx::core::Softmax::vmap()'],['../classmlx_1_1core_1_1_sort.html#abfabb9e625cc0cb9335c7454ed27505c',1,'mlx::core::Sort::vmap()'],['../classmlx_1_1core_1_1_split.html#ab7c40e02a842e83bdb4698608472c7a6',1,'mlx::core::Split::vmap()'],['../classmlx_1_1core_1_1_square.html#a55bf43f878d4741c57a08d5fef472ea5',1,'mlx::core::Square::vmap()'],['../classmlx_1_1core_1_1_sqrt.html#a9d30e306ce08980c27d98c898577017e',1,'mlx::core::Sqrt::vmap()'],['../classmlx_1_1core_1_1_stop_gradient.html#aca680c8befef81da414c4375b11b16b0',1,'mlx::core::StopGradient::vmap()'],['../classmlx_1_1core_1_1_subtract.html#aa98f960e621a767c8a03624fd292f098',1,'mlx::core::Subtract::vmap()'],['../classmlx_1_1core_1_1_squeeze.html#aa098a5850741bfb621800c7badce3532',1,'mlx::core::Squeeze::vmap()'],['../classmlx_1_1core_1_1_tan.html#ae2f67ca2adc83b10009cf28498bf58b7',1,'mlx::core::Tan::vmap()'],['../classmlx_1_1core_1_1_tanh.html#a32df3564c1ecb858c1ba9f855376762f',1,'mlx::core::Tanh::vmap()'],['../classmlx_1_1core_1_1_unflatten.html#a0f6ee31b99aca962d887c856414813fe',1,'mlx::core::Unflatten::vmap()'],['../classmlx_1_1core_1_1_view.html#a2230d3e5f434fb2b888de50b529ac121',1,'mlx::core::View::vmap()'],['../classmlx_1_1core_1_1_transpose.html#a5ef848b69def9a246665b67e6e3ffdfe',1,'mlx::core::Transpose::vmap()'],['../classmlx_1_1core_1_1_s_v_d.html#a0366c958f6cdac8d1d9e1a4eda53fae8',1,'mlx::core::SVD::vmap()'],['../classmlx_1_1core_1_1_inverse.html#a98419b9f0b8a6c9185fe012d523552c2',1,'mlx::core::Inverse::vmap()'],['../classmlx_1_1core_1_1_cholesky.html#ab5c3f6199ec3b399c91243a05d116aa5',1,'mlx::core::Cholesky::vmap()'],['../classmlx_1_1core_1_1_eigh.html#ab2f2ea5326e2f6045f9b7250692c240f',1,'mlx::core::Eigh::vmap()'],['../namespacemlx_1_1core.html#ac3caec2fa65375ed4c3bf1206177b84c',1,'mlx::core::vmap(const std::function< array(const array &)> &fun, int in_axis=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a8481a3bb4c12c2b7dc6ba576c2be3d0d',1,'mlx::core::vmap(const std::function< array(const array &, const array &)> &fun, int in_axis_a=0, int in_axis_b=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a95a7757e8d18fced38acfc6a3e8d686a',1,'mlx::core::vmap(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &in_axes={}, const std::vector< int > &out_axes={})']]], - ['vmap_5freplace_9',['vmap_replace',['../namespacemlx_1_1core_1_1detail.html#a31a5582530faea230eb8acafc0f7e154',1,'mlx::core::detail']]], - ['vmap_5ftrace_10',['vmap_trace',['../namespacemlx_1_1core_1_1detail.html#a5ba794afe1a557e0505887cfb481c515',1,'mlx::core::detail']]] + ['vectorscalar_5',['VectorScalar',['../structmlx_1_1core_1_1_vector_scalar.html#a97088143e6d301d753dcdd1ccdd82287',1,'mlx::core::VectorScalar']]], + ['vectorvector_6',['VectorVector',['../structmlx_1_1core_1_1_vector_vector.html#a4867666c95c597a113afb64f173cc022',1,'mlx::core::VectorVector']]], + ['view_7',['View',['../classmlx_1_1core_1_1_view.html#ad7eed156c308e9a29a8b41f965ec941e',1,'mlx::core::View']]], + ['view_8',['view',['../group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02',1,'mlx::core']]], + ['vjp_9',['vjp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abbf6d1d63dcda207ad7d9eeb4fc36225',1,'mlx::core::distributed::AllReduce::vjp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#aa5eff6fc128b71220899aab8ab9116fb',1,'mlx::core::distributed::AllGather::vjp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a74be4bcd0382f7f6400bf73fd5569c91',1,'mlx::core::fast::Custom::vjp()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#aacfbbbc15fcee0a5ce4f519ca3cca5eb',1,'mlx::core::fast::RMSNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#ae5e1b5df0705a6b1d141691a4396b0b6',1,'mlx::core::fast::LayerNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#ad999105414badd66c8fd9e069454a533',1,'mlx::core::fast::RoPE::vjp()'],['../classmlx_1_1core_1_1_primitive.html#a1dcb6807326eeab62474c6a0e3836d42',1,'mlx::core::Primitive::vjp()'],['../classmlx_1_1core_1_1_abs.html#aa2dd8ec0989e716b77394ac349b34592',1,'mlx::core::Abs::vjp()'],['../classmlx_1_1core_1_1_add.html#ac28e581862880e24ed2b99bb6a916607',1,'mlx::core::Add::vjp()'],['../classmlx_1_1core_1_1_add_m_m.html#ac1562a37cec6928e01281926ebeb47c6',1,'mlx::core::AddMM::vjp()'],['../classmlx_1_1core_1_1_arc_cos.html#a78e73e5e639d1249c7fe9614bf157c92',1,'mlx::core::ArcCos::vjp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a856c677f16e2b3f2edd2491e35db2d26',1,'mlx::core::ArcCosh::vjp()'],['../classmlx_1_1core_1_1_arc_sin.html#ab4057cd5ef1a8359f97493018e10d3a1',1,'mlx::core::ArcSin::vjp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a7988ee5b9e1e7e498dcab73d61ba147e',1,'mlx::core::ArcSinh::vjp()'],['../classmlx_1_1core_1_1_arc_tan.html#a5fefc3634b96a67ff8ae011a8ee180c2',1,'mlx::core::ArcTan::vjp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a99840c282e37b2b2a9c312e6e8ade1d2',1,'mlx::core::ArcTan2::vjp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a07da5797f7aaf3dfe43bf24e8562ac72',1,'mlx::core::ArcTanh::vjp()'],['../classmlx_1_1core_1_1_arg_partition.html#ade23d014717a0b0235d00073503aeac0',1,'mlx::core::ArgPartition::vjp()'],['../classmlx_1_1core_1_1_arg_reduce.html#a60d272685a373e6fe879416481a1ce1a',1,'mlx::core::ArgReduce::vjp()'],['../classmlx_1_1core_1_1_as_type.html#ac38a4f889311a3b5e5be9a67dcb93e18',1,'mlx::core::AsType::vjp()'],['../classmlx_1_1core_1_1_as_strided.html#a34783284c9b2f5b4a62c3c3ee5dd4062',1,'mlx::core::AsStrided::vjp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6131ed1c317ff8700a3e9b13fdaa9d61',1,'mlx::core::BitwiseBinary::vjp()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a1adf20087ee2f685bf39c2724b8e7120',1,'mlx::core::BlockMaskedMM::vjp()'],['../classmlx_1_1core_1_1_gather_m_m.html#a76c9f27c57354f6230b43944882e1bda',1,'mlx::core::GatherMM::vjp()'],['../classmlx_1_1core_1_1_broadcast_axes.html#aea8ef2b2616568a2bb56695381a035be',1,'mlx::core::BroadcastAxes::vjp()'],['../classmlx_1_1core_1_1_broadcast.html#a0318847c9be40f00b23907ad56037d18',1,'mlx::core::Broadcast::vjp()'],['../classmlx_1_1core_1_1_ceil.html#ac2f5a2bd84b8f013e5ce688419a88acb',1,'mlx::core::Ceil::vjp()'],['../classmlx_1_1core_1_1_compiled.html#a32462e65c52f84b708188130cc508133',1,'mlx::core::Compiled::vjp()'],['../classmlx_1_1core_1_1_concatenate.html#a8155db9100ec3b8bd0bc94baeaeee3b0',1,'mlx::core::Concatenate::vjp()'],['../classmlx_1_1core_1_1_contiguous.html#abf488f02057fd5852f38b2e8a600ad2a',1,'mlx::core::Contiguous::vjp()'],['../classmlx_1_1core_1_1_convolution.html#af8eb9c0c055ad20aa74b547016917690',1,'mlx::core::Convolution::vjp()'],['../classmlx_1_1core_1_1_copy.html#a6c4dee582001e9983e9517485ee37efd',1,'mlx::core::Copy::vjp()'],['../classmlx_1_1core_1_1_cos.html#a51d84113728e651ef9d4a1fe671c4d00',1,'mlx::core::Cos::vjp()'],['../classmlx_1_1core_1_1_cosh.html#a0791abd4305a333fb3b181a5357ce0f4',1,'mlx::core::Cosh::vjp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa1da36cef632df767cd9809d6cf06209',1,'mlx::core::CustomTransforms::vjp()'],['../classmlx_1_1core_1_1_depends.html#a02996fa45f01f7cb9f37074d5f8ccab0',1,'mlx::core::Depends::vjp()'],['../classmlx_1_1core_1_1_divide.html#ad3af7c70cad22c1a1a75b4a78ef793b6',1,'mlx::core::Divide::vjp()'],['../classmlx_1_1core_1_1_div_mod.html#a8c914a07f666a1d9377a27ed5d55e7c1',1,'mlx::core::DivMod::vjp()'],['../classmlx_1_1core_1_1_select.html#a9b522487b78fceeca7f827cd1c29a9a3',1,'mlx::core::Select::vjp()'],['../classmlx_1_1core_1_1_remainder.html#ab18f7bca1027ae71847a50da0933cec6',1,'mlx::core::Remainder::vjp()'],['../classmlx_1_1core_1_1_equal.html#af3c1bfcd1bf50922fc00e302bb193736',1,'mlx::core::Equal::vjp()'],['../classmlx_1_1core_1_1_erf.html#a1f529e95a42a2d69a8b18979d3ee2909',1,'mlx::core::Erf::vjp()'],['../classmlx_1_1core_1_1_erf_inv.html#a48afff12a58ddefae7ae0245c3580189',1,'mlx::core::ErfInv::vjp()'],['../classmlx_1_1core_1_1_exp.html#a94b9b7d137c3640d290b96c5e8b7e1a8',1,'mlx::core::Exp::vjp()'],['../classmlx_1_1core_1_1_expm1.html#af6ce416169190479c9792bb9cdbe2f43',1,'mlx::core::Expm1::vjp()'],['../classmlx_1_1core_1_1_expand_dims.html#a2fb3c65ba7a3b2d1f33a3c681fda8896',1,'mlx::core::ExpandDims::vjp()'],['../classmlx_1_1core_1_1_f_f_t.html#aafc895614a6e368c0e6d64af20d01090',1,'mlx::core::FFT::vjp()'],['../classmlx_1_1core_1_1_flatten.html#ab549a8c38b63055e2d5cd672f7676aab',1,'mlx::core::Flatten::vjp()'],['../classmlx_1_1core_1_1_floor.html#a589e2cf99b6fd1a5ba85534a2a31338e',1,'mlx::core::Floor::vjp()'],['../classmlx_1_1core_1_1_full.html#a49e76e7a8641f990701abc1b3bd49969',1,'mlx::core::Full::vjp()'],['../classmlx_1_1core_1_1_gather.html#aacf612a8f5f1cdbbfd19707d8d33c426',1,'mlx::core::Gather::vjp()'],['../classmlx_1_1core_1_1_gather_axis.html#a9c73b4ebed01bbdbaa316eddb6b5606d',1,'mlx::core::GatherAxis::vjp()'],['../classmlx_1_1core_1_1_greater.html#a341766a8a7e41d2a1160d35d4e781679',1,'mlx::core::Greater::vjp()'],['../classmlx_1_1core_1_1_greater_equal.html#a62f07a4ac54c708307c82aac0e5693ee',1,'mlx::core::GreaterEqual::vjp()'],['../classmlx_1_1core_1_1_hadamard.html#af4134775427b8998d66f489468b98656',1,'mlx::core::Hadamard::vjp()'],['../classmlx_1_1core_1_1_imag.html#a80da5fdd0fa549eebd7804c0e261848b',1,'mlx::core::Imag::vjp()'],['../classmlx_1_1core_1_1_less.html#aaf205d389b5e602e0814b68f66de8f50',1,'mlx::core::Less::vjp()'],['../classmlx_1_1core_1_1_less_equal.html#aab2aab7590c299885e815c18eedd1028',1,'mlx::core::LessEqual::vjp()'],['../classmlx_1_1core_1_1_log.html#a40885dccfbf928c4d035881be1d49280',1,'mlx::core::Log::vjp()'],['../classmlx_1_1core_1_1_log1p.html#a3113c1d2b4c5e73d0b470f42dc48a880',1,'mlx::core::Log1p::vjp()'],['../classmlx_1_1core_1_1_logical_not.html#af2c3c241cf3910fbaba013c69d052a50',1,'mlx::core::LogicalNot::vjp()'],['../classmlx_1_1core_1_1_logical_and.html#ae42f8fc454577b0fd6410cae9d5f3b54',1,'mlx::core::LogicalAnd::vjp()'],['../classmlx_1_1core_1_1_logical_or.html#a51aed488f52d5031998689af9cb17847',1,'mlx::core::LogicalOr::vjp()'],['../classmlx_1_1core_1_1_log_add_exp.html#ae231af0ed24a93eb647ee58c2d2b20b4',1,'mlx::core::LogAddExp::vjp()'],['../classmlx_1_1core_1_1_matmul.html#a524136cca481598ea20894d85ca66bb0',1,'mlx::core::Matmul::vjp()'],['../classmlx_1_1core_1_1_maximum.html#a7de15d7b28784e24bbfc7e85ddcbcff3',1,'mlx::core::Maximum::vjp()'],['../classmlx_1_1core_1_1_minimum.html#a48a0cbe3a6c4f7473c00e343f63b5204',1,'mlx::core::Minimum::vjp()'],['../classmlx_1_1core_1_1_multiply.html#a74b7556ec03e2c3d3f971666d06f5db1',1,'mlx::core::Multiply::vjp()'],['../classmlx_1_1core_1_1_negative.html#a889585f056d33bda30c30311257af52a',1,'mlx::core::Negative::vjp()'],['../classmlx_1_1core_1_1_not_equal.html#a0361f29f4ae1235bdf3f3304527e2d4b',1,'mlx::core::NotEqual::vjp()'],['../classmlx_1_1core_1_1_pad.html#ad8a7e547644f2717a24322968e971038',1,'mlx::core::Pad::vjp()'],['../classmlx_1_1core_1_1_partition.html#a7110772b6cd2d430a2b825cf5c952ca9',1,'mlx::core::Partition::vjp()'],['../classmlx_1_1core_1_1_power.html#a1453bb8307d6ff33134f1e00263bf082',1,'mlx::core::Power::vjp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#acb975e272b4a88ab232ef7f7c3a2bf26',1,'mlx::core::QuantizedMatmul::vjp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#ae08a4b7d28902d46f39e66beeb0e23ab',1,'mlx::core::GatherQMM::vjp()'],['../classmlx_1_1core_1_1_real.html#a29f6109339c5141a862ceae72c8b80fe',1,'mlx::core::Real::vjp()'],['../classmlx_1_1core_1_1_reshape.html#ab17294ecc6b5d4e89626fb48c7516365',1,'mlx::core::Reshape::vjp()'],['../classmlx_1_1core_1_1_reduce.html#a684883d2a96315f548ca769510e28e4e',1,'mlx::core::Reduce::vjp()'],['../classmlx_1_1core_1_1_round.html#af8f085e08b7fa8840c52a20b12ca35ce',1,'mlx::core::Round::vjp()'],['../classmlx_1_1core_1_1_scan.html#aaf13f72620b4b5d6a20e1228930e848e',1,'mlx::core::Scan::vjp()'],['../classmlx_1_1core_1_1_scatter.html#a0b51287fba789bb139ed61d40a0c636a',1,'mlx::core::Scatter::vjp()'],['../classmlx_1_1core_1_1_scatter_axis.html#a450f97b0be61a2bdfbfef4b2eb7cd198',1,'mlx::core::ScatterAxis::vjp()'],['../classmlx_1_1core_1_1_sigmoid.html#aac2f56a4c8362e36a28e232758ca52cf',1,'mlx::core::Sigmoid::vjp()'],['../classmlx_1_1core_1_1_sign.html#aa60ac52edd739fbdf388a997acd01bce',1,'mlx::core::Sign::vjp()'],['../classmlx_1_1core_1_1_sin.html#aedefe550ab4b0687858981bc0bcfbfa0',1,'mlx::core::Sin::vjp()'],['../classmlx_1_1core_1_1_sinh.html#a6b39fdd429bbb4de389e7c904fd561f0',1,'mlx::core::Sinh::vjp()'],['../classmlx_1_1core_1_1_slice.html#a291746a527ff991b66249fb2b54b685f',1,'mlx::core::Slice::vjp()'],['../classmlx_1_1core_1_1_slice_update.html#aedcdc60a0477997a96306c02b66d3f77',1,'mlx::core::SliceUpdate::vjp()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a29caf03256945f7732a52d551191f8fa',1,'mlx::core::DynamicSlice::vjp()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ab2817cb9d1bfcd3de6454d841909da1f',1,'mlx::core::DynamicSliceUpdate::vjp()'],['../classmlx_1_1core_1_1_softmax.html#abb68c311c45ee422a7c966accde9041b',1,'mlx::core::Softmax::vjp()'],['../classmlx_1_1core_1_1_sort.html#a3a8900dce53ee4eb7a1b83806e629358',1,'mlx::core::Sort::vjp()'],['../classmlx_1_1core_1_1_split.html#a7e8730f9cffa9872fff6f8d577031674',1,'mlx::core::Split::vjp()'],['../classmlx_1_1core_1_1_square.html#abcd9516da7f02dc906368c23b0bca263',1,'mlx::core::Square::vjp()'],['../classmlx_1_1core_1_1_sqrt.html#a08a21bd2c3a016f042d95aca294e68f3',1,'mlx::core::Sqrt::vjp()'],['../classmlx_1_1core_1_1_subtract.html#a3a3322be7c3bcaa0397cf099091df16b',1,'mlx::core::Subtract::vjp()'],['../classmlx_1_1core_1_1_squeeze.html#a8d95a13d7cc5586d48a38e9199180d06',1,'mlx::core::Squeeze::vjp()'],['../classmlx_1_1core_1_1_tan.html#a4639836cff03d73c769387d6943e92d7',1,'mlx::core::Tan::vjp()'],['../classmlx_1_1core_1_1_tanh.html#afe7b05e2b36b99c3a1b66f0cd3544e95',1,'mlx::core::Tanh::vjp()'],['../classmlx_1_1core_1_1_unflatten.html#a34f1218fa1d0e28f3ee10b65e6b0e319',1,'mlx::core::Unflatten::vjp()'],['../classmlx_1_1core_1_1_transpose.html#ac7805aa29b34afdf8852554f1e759f80',1,'mlx::core::Transpose::vjp()'],['../namespacemlx_1_1core.html#a1b33e2c2e3471420490cf0be2de6de18',1,'mlx::core::vjp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &cotangents)'],['../namespacemlx_1_1core.html#a2065a11249c3f4356ffd69b7a8c487ff',1,'mlx::core::vjp(const std::function< array(const array &)> &fun, const array &primal, const array &cotangent)']]], + ['vmap_10',['vmap',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a3f2dc71859847ca675ec4bfbe125035a',1,'mlx::core::distributed::AllReduce::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ad532d1d51f089dec3c84799b724ea031',1,'mlx::core::distributed::AllGather::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a5cfb66191b9e8b86649da77af55b0f93',1,'mlx::core::distributed::Send::vmap()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a7f4c3a4c48c6807faa36fb31e39dad8d',1,'mlx::core::fast::Custom::vmap()'],['../classmlx_1_1core_1_1_primitive.html#ac632b9619dd7a6a0f177bd36202e8103',1,'mlx::core::Primitive::vmap()'],['../classmlx_1_1core_1_1_abs.html#a4c9c98f1d71432fd3752ad9a6a8e7f2f',1,'mlx::core::Abs::vmap()'],['../classmlx_1_1core_1_1_add.html#a0e557d4d896153f84a25532562e4c646',1,'mlx::core::Add::vmap()'],['../classmlx_1_1core_1_1_add_m_m.html#a73ce80b3a37ec2523943028d50ebce81',1,'mlx::core::AddMM::vmap()'],['../classmlx_1_1core_1_1_arc_cos.html#a7548e23ace6827674aa6d284d44ccf83',1,'mlx::core::ArcCos::vmap()'],['../classmlx_1_1core_1_1_arc_cosh.html#af8ff78e910a9e485a203e1d3347bd461',1,'mlx::core::ArcCosh::vmap()'],['../classmlx_1_1core_1_1_arc_sin.html#a7cabb1e5a2bda44944378822c671ec82',1,'mlx::core::ArcSin::vmap()'],['../classmlx_1_1core_1_1_arc_sinh.html#a9e72b9751939387c333b5d4e19a37f6d',1,'mlx::core::ArcSinh::vmap()'],['../classmlx_1_1core_1_1_arc_tan.html#a1fb921554544a56498bc54f82e4a0556',1,'mlx::core::ArcTan::vmap()'],['../classmlx_1_1core_1_1_arc_tan2.html#ae02cb9fbf25e93dc1d7fbc9e3fb28634',1,'mlx::core::ArcTan2::vmap()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6ddcae68873559211cb91e7740dfc040',1,'mlx::core::ArcTanh::vmap()'],['../classmlx_1_1core_1_1_arg_partition.html#a441093795bcc31495ab5fbc9957b740a',1,'mlx::core::ArgPartition::vmap()'],['../classmlx_1_1core_1_1_arg_reduce.html#abfec42fa06ea15edaf393593751fb1ba',1,'mlx::core::ArgReduce::vmap()'],['../classmlx_1_1core_1_1_arg_sort.html#a3522bbbe4626a467394c1a8a9d7ac34e',1,'mlx::core::ArgSort::vmap()'],['../classmlx_1_1core_1_1_as_type.html#a7ebaf86fd6cad4a1ecfd7cde1ee0b0cc',1,'mlx::core::AsType::vmap()'],['../classmlx_1_1core_1_1_bitwise_binary.html#aa10be55f05bc1868bf4b375dc475f965',1,'mlx::core::BitwiseBinary::vmap()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a4e04f564d440e2d312c335db52c308e1',1,'mlx::core::BroadcastAxes::vmap()'],['../classmlx_1_1core_1_1_broadcast.html#aee4c71c2588ad01eb57e10f346cd666f',1,'mlx::core::Broadcast::vmap()'],['../classmlx_1_1core_1_1_ceil.html#ae86819990b43bdb0c2b3a25719b3a7a4',1,'mlx::core::Ceil::vmap()'],['../classmlx_1_1core_1_1_compiled.html#a732e7548f53977b4513bb7f30a04c30d',1,'mlx::core::Compiled::vmap()'],['../classmlx_1_1core_1_1_concatenate.html#a58c54dcf8e4b045d25edd3afc2caffc1',1,'mlx::core::Concatenate::vmap()'],['../classmlx_1_1core_1_1_conjugate.html#a2c7632c8ae0ca07777e23a0a79344e60',1,'mlx::core::Conjugate::vmap()'],['../classmlx_1_1core_1_1_contiguous.html#a563221e90b15aa90bfae23d29c10e4ec',1,'mlx::core::Contiguous::vmap()'],['../classmlx_1_1core_1_1_copy.html#a669b10253c15b769d90058d1ad7d0e61',1,'mlx::core::Copy::vmap()'],['../classmlx_1_1core_1_1_cos.html#aec9460daf0131156734013d03b230cd6',1,'mlx::core::Cos::vmap()'],['../classmlx_1_1core_1_1_cosh.html#a1ab2386e7d96219b6e4a525f7dac0406',1,'mlx::core::Cosh::vmap()'],['../classmlx_1_1core_1_1_custom_transforms.html#a906a2ff30d9c5281fbf1fa927e4c021b',1,'mlx::core::CustomTransforms::vmap()'],['../classmlx_1_1core_1_1_divide.html#a83e7da52831165b3a026e97b63770242',1,'mlx::core::Divide::vmap()'],['../classmlx_1_1core_1_1_div_mod.html#ae709e0fdd83994bd1d156e0d0e6a7942',1,'mlx::core::DivMod::vmap()'],['../classmlx_1_1core_1_1_select.html#a84e80361c8cf02536b4b98098793550f',1,'mlx::core::Select::vmap()'],['../classmlx_1_1core_1_1_remainder.html#a79867e1099a2e3c2d3e87407b2ab6e3d',1,'mlx::core::Remainder::vmap()'],['../classmlx_1_1core_1_1_equal.html#aea9cc3c88924ac824d72c39c2e83b0ca',1,'mlx::core::Equal::vmap()'],['../classmlx_1_1core_1_1_erf.html#abe554f553356654a3e800ba368108aaa',1,'mlx::core::Erf::vmap()'],['../classmlx_1_1core_1_1_erf_inv.html#ad5d7634e8568af8cc4a54a558a48d0e9',1,'mlx::core::ErfInv::vmap()'],['../classmlx_1_1core_1_1_exp.html#a0fcd579fe148b4c3dbc72e514b81bb37',1,'mlx::core::Exp::vmap()'],['../classmlx_1_1core_1_1_expm1.html#aa4caa848b2ea97e71ee3dd33de039296',1,'mlx::core::Expm1::vmap()'],['../classmlx_1_1core_1_1_expand_dims.html#a380c9ddc25a1f973c3d71b42f8a19486',1,'mlx::core::ExpandDims::vmap()'],['../classmlx_1_1core_1_1_f_f_t.html#ac32d6cc9b67289124f855ea68a61ede1',1,'mlx::core::FFT::vmap()'],['../classmlx_1_1core_1_1_flatten.html#a244a03915313286d36ed4d36b01a99f2',1,'mlx::core::Flatten::vmap()'],['../classmlx_1_1core_1_1_floor.html#aea4dc79a65774990e775ad49519a5d10',1,'mlx::core::Floor::vmap()'],['../classmlx_1_1core_1_1_full.html#afc57ab6bd9ebdbbf042af54a59785d95',1,'mlx::core::Full::vmap()'],['../classmlx_1_1core_1_1_gather.html#abab0c4c204e66489825ce80d2194a275',1,'mlx::core::Gather::vmap()'],['../classmlx_1_1core_1_1_gather_axis.html#a48d50bad33b69e29f75bedc794f7b785',1,'mlx::core::GatherAxis::vmap()'],['../classmlx_1_1core_1_1_greater.html#a6d8267411fc4951de781f9e8e6c53aa0',1,'mlx::core::Greater::vmap()'],['../classmlx_1_1core_1_1_greater_equal.html#ab0e1be93eb01b0ce7fa83e953f5e3e1d',1,'mlx::core::GreaterEqual::vmap()'],['../classmlx_1_1core_1_1_hadamard.html#a9f1a172e6246859e813002abe9b8f99c',1,'mlx::core::Hadamard::vmap()'],['../classmlx_1_1core_1_1_imag.html#ace9906672bd88df0573653883d58ecb3',1,'mlx::core::Imag::vmap()'],['../classmlx_1_1core_1_1_less.html#a5fee5956cf087d8405359121aa62ba7e',1,'mlx::core::Less::vmap()'],['../classmlx_1_1core_1_1_less_equal.html#a3d5df21db184f2b7620cda9da1684480',1,'mlx::core::LessEqual::vmap()'],['../classmlx_1_1core_1_1_log.html#a007ddbcf911093231f607a8b9ed5cd49',1,'mlx::core::Log::vmap()'],['../classmlx_1_1core_1_1_log1p.html#a7122576f95ce479926bbbbc690891f71',1,'mlx::core::Log1p::vmap()'],['../classmlx_1_1core_1_1_logical_not.html#a5308a271619ee74df561b0aaf525915d',1,'mlx::core::LogicalNot::vmap()'],['../classmlx_1_1core_1_1_logical_and.html#aacc5f6f53ffc327b7771485e3da2a4e5',1,'mlx::core::LogicalAnd::vmap()'],['../classmlx_1_1core_1_1_logical_or.html#a6e2e77e6aaf47872b2e96b151c32daf3',1,'mlx::core::LogicalOr::vmap()'],['../classmlx_1_1core_1_1_log_add_exp.html#a82190aa1421a9734b6e9480debffac78',1,'mlx::core::LogAddExp::vmap()'],['../classmlx_1_1core_1_1_matmul.html#a3a1c6e70bac300240760fe41a58340c2',1,'mlx::core::Matmul::vmap()'],['../classmlx_1_1core_1_1_maximum.html#ab664918e0d71cfec1318a9879e78c5d3',1,'mlx::core::Maximum::vmap()'],['../classmlx_1_1core_1_1_minimum.html#adab0f31acf68075a0be908d8eb882980',1,'mlx::core::Minimum::vmap()'],['../classmlx_1_1core_1_1_multiply.html#ae7e82c8fc8cbaf4e00c27eb54fac7dbf',1,'mlx::core::Multiply::vmap()'],['../classmlx_1_1core_1_1_negative.html#a1f8a6079e272f1a0599f88a1a8419cf0',1,'mlx::core::Negative::vmap()'],['../classmlx_1_1core_1_1_not_equal.html#ab8b57932f03c8eee664bf89adeaa43b5',1,'mlx::core::NotEqual::vmap()'],['../classmlx_1_1core_1_1_number_of_elements.html#a977d83eae845b8bd8c0b98b48cb1c6c2',1,'mlx::core::NumberOfElements::vmap()'],['../classmlx_1_1core_1_1_pad.html#a85658812a0f3275ba3eb74b7c75686cf',1,'mlx::core::Pad::vmap()'],['../classmlx_1_1core_1_1_partition.html#aa0cc55e4d4d2cb5d129d32832321df2c',1,'mlx::core::Partition::vmap()'],['../classmlx_1_1core_1_1_power.html#a5e22749592413a9adbdc877b03b87c8f',1,'mlx::core::Power::vmap()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a3434394140177b285f971c9ffe7e8763',1,'mlx::core::QuantizedMatmul::vmap()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a13ce5e138ebddb8780a034452f68892f',1,'mlx::core::GatherQMM::vmap()'],['../classmlx_1_1core_1_1_random_bits.html#a0dc12f053c6492f934bc18031412c415',1,'mlx::core::RandomBits::vmap()'],['../classmlx_1_1core_1_1_real.html#a07fbbefb6a1bc1ebd3985b24c36693b6',1,'mlx::core::Real::vmap()'],['../classmlx_1_1core_1_1_reshape.html#ae239dd3c6cab147e4af572dc58204f9d',1,'mlx::core::Reshape::vmap()'],['../classmlx_1_1core_1_1_reduce.html#abab1b5aa01ccad44f213f510c3596b38',1,'mlx::core::Reduce::vmap()'],['../classmlx_1_1core_1_1_round.html#a6fad8799a7982e1ccbe05be7cc38a7fd',1,'mlx::core::Round::vmap()'],['../classmlx_1_1core_1_1_scan.html#a297c7cc89c9bf9d186ebdebb634c7804',1,'mlx::core::Scan::vmap()'],['../classmlx_1_1core_1_1_scatter.html#a696c38b373a7a7c71bc112bd1117e322',1,'mlx::core::Scatter::vmap()'],['../classmlx_1_1core_1_1_scatter_axis.html#ae78709d1be122618f210ff595d888df8',1,'mlx::core::ScatterAxis::vmap()'],['../classmlx_1_1core_1_1_sigmoid.html#a12712c23037e38192cbccd2d4b14cc85',1,'mlx::core::Sigmoid::vmap()'],['../classmlx_1_1core_1_1_sign.html#aa7296045907015b4e0ae8a93e5e6e295',1,'mlx::core::Sign::vmap()'],['../classmlx_1_1core_1_1_sin.html#a45533996f3d72d9dd97d4c61cd684fba',1,'mlx::core::Sin::vmap()'],['../classmlx_1_1core_1_1_sinh.html#ae171df22bc34c32e31b8135dc4caa788',1,'mlx::core::Sinh::vmap()'],['../classmlx_1_1core_1_1_slice.html#ae33583b0db22fcfeae34dfe1c0e3eaa2',1,'mlx::core::Slice::vmap()'],['../classmlx_1_1core_1_1_slice_update.html#adbf1c76de6ab2f986758530d351d6fa3',1,'mlx::core::SliceUpdate::vmap()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a825a6d4d1499b287525462854b841ef2',1,'mlx::core::DynamicSlice::vmap()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a750fb3548d8f3a5c6f4e54958649936f',1,'mlx::core::DynamicSliceUpdate::vmap()'],['../classmlx_1_1core_1_1_softmax.html#ad22d3dcc71054d3dba529cf2dc981e19',1,'mlx::core::Softmax::vmap()'],['../classmlx_1_1core_1_1_sort.html#abfabb9e625cc0cb9335c7454ed27505c',1,'mlx::core::Sort::vmap()'],['../classmlx_1_1core_1_1_split.html#ab7c40e02a842e83bdb4698608472c7a6',1,'mlx::core::Split::vmap()'],['../classmlx_1_1core_1_1_square.html#a55bf43f878d4741c57a08d5fef472ea5',1,'mlx::core::Square::vmap()'],['../classmlx_1_1core_1_1_sqrt.html#a9d30e306ce08980c27d98c898577017e',1,'mlx::core::Sqrt::vmap()'],['../classmlx_1_1core_1_1_stop_gradient.html#aca680c8befef81da414c4375b11b16b0',1,'mlx::core::StopGradient::vmap()'],['../classmlx_1_1core_1_1_subtract.html#aa98f960e621a767c8a03624fd292f098',1,'mlx::core::Subtract::vmap()'],['../classmlx_1_1core_1_1_squeeze.html#aa098a5850741bfb621800c7badce3532',1,'mlx::core::Squeeze::vmap()'],['../classmlx_1_1core_1_1_tan.html#ae2f67ca2adc83b10009cf28498bf58b7',1,'mlx::core::Tan::vmap()'],['../classmlx_1_1core_1_1_tanh.html#a32df3564c1ecb858c1ba9f855376762f',1,'mlx::core::Tanh::vmap()'],['../classmlx_1_1core_1_1_unflatten.html#a0f6ee31b99aca962d887c856414813fe',1,'mlx::core::Unflatten::vmap()'],['../classmlx_1_1core_1_1_view.html#a2230d3e5f434fb2b888de50b529ac121',1,'mlx::core::View::vmap()'],['../classmlx_1_1core_1_1_transpose.html#a5ef848b69def9a246665b67e6e3ffdfe',1,'mlx::core::Transpose::vmap()'],['../classmlx_1_1core_1_1_s_v_d.html#a0366c958f6cdac8d1d9e1a4eda53fae8',1,'mlx::core::SVD::vmap()'],['../classmlx_1_1core_1_1_inverse.html#a98419b9f0b8a6c9185fe012d523552c2',1,'mlx::core::Inverse::vmap()'],['../classmlx_1_1core_1_1_cholesky.html#ab5c3f6199ec3b399c91243a05d116aa5',1,'mlx::core::Cholesky::vmap()'],['../classmlx_1_1core_1_1_eigh.html#ab2f2ea5326e2f6045f9b7250692c240f',1,'mlx::core::Eigh::vmap()'],['../namespacemlx_1_1core.html#ac3caec2fa65375ed4c3bf1206177b84c',1,'mlx::core::vmap(const std::function< array(const array &)> &fun, int in_axis=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a8481a3bb4c12c2b7dc6ba576c2be3d0d',1,'mlx::core::vmap(const std::function< array(const array &, const array &)> &fun, int in_axis_a=0, int in_axis_b=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a95a7757e8d18fced38acfc6a3e8d686a',1,'mlx::core::vmap(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &in_axes={}, const std::vector< int > &out_axes={})']]], + ['vmap_5freplace_11',['vmap_replace',['../namespacemlx_1_1core_1_1detail.html#a31a5582530faea230eb8acafc0f7e154',1,'mlx::core::detail']]], + ['vmap_5ftrace_12',['vmap_trace',['../namespacemlx_1_1core_1_1detail.html#a5ba794afe1a557e0505887cfb481c515',1,'mlx::core::detail']]] ]; diff --git a/docs/build/html/search/functions_17.js b/docs/build/html/search/functions_17.js index 8dd234bbd..e812d2f47 100644 --- a/docs/build/html/search/functions_17.js +++ b/docs/build/html/search/functions_17.js @@ -1,12 +1,13 @@ var searchData= [ - ['wait_0',['wait',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#af503189cc9247047fbdfc3ebf1daacc1',1,'pocketfft::detail::threading::latch::wait()'],['../classmlx_1_1core_1_1array.html#a648592006f1c92287734ba2428eaa45e',1,'mlx::core::array::wait()'],['../classmlx_1_1core_1_1_event.html#a634afd918e6ed847f354531ba9f48252',1,'mlx::core::Event::wait()']]], + ['wait_0',['wait',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#af503189cc9247047fbdfc3ebf1daacc1',1,'pocketfft::detail::threading::latch::wait()'],['../classmlx_1_1core_1_1array.html#a648592006f1c92287734ba2428eaa45e',1,'mlx::core::array::wait()'],['../classmlx_1_1core_1_1_fence.html#a1ccbe354d043e6c4d76286c6635a38e2',1,'mlx::core::Fence::wait()'],['../classmlx_1_1core_1_1_event.html#a634afd918e6ed847f354531ba9f48252',1,'mlx::core::Event::wait()']]], ['wait_5ffor_5ffence_1',['wait_for_fence',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aefdadbff4e003dc6f77506840babc088',1,'mlx::core::metal::CommandEncoder::wait_for_fence()'],['../structmlx_1_1core_1_1_command_encoder.html#aefdadbff4e003dc6f77506840babc088',1,'mlx::core::CommandEncoder::wait_for_fence()']]], ['wait_5ffor_5fone_2',['wait_for_one',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a01c574bb388f10d67aaaaa541894d807',1,'mlx::core::scheduler::Scheduler::wait_for_one()'],['../namespacemlx_1_1core_1_1scheduler.html#a8cc4d5fd1f5ce722b377ead1863a2291',1,'mlx::core::scheduler::wait_for_one()']]], - ['where_3',['where',['../group__ops.html#ga8a2056f8c9bb30914c40bcf509386491',1,'mlx::core']]], - ['write_4',['write',['../struct_read_writer.html#ac2ea71e41740ddc863890e3e8e6f09d0',1,'ReadWriter::write()'],['../classmlx_1_1core_1_1io_1_1_writer.html#ad9515b7f007338674de1e124cf77e125',1,'mlx::core::io::Writer::write()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#abca32838c9886f734d93430c34c07d7f',1,'mlx::core::io::FileWriter::write()'],['../struct_read_writer.html#a7a3d1396b0f83aa7506207bd6e7336bf',1,'ReadWriter::write() const'],['../struct_read_writer.html#ae1f0d3555b74998cc2d2288bce72a1f4',1,'ReadWriter::write() const']]], - ['write_5fpadded_5',['write_padded',['../struct_read_writer.html#a95367307acace2aa88226cf8956d2d88',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#abaf2a6ad4c88bd9f65fe1db1f73a8d87',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#a420453a56e77d6b3891ed4b5f178af9c',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const']]], - ['write_5fsafe_6',['write_safe',['../scan_8h.html#ae86aef08e5ebc8790031eb51eefa754c',1,'scan.h']]], - ['write_5fstrided_7',['write_strided',['../struct_read_writer.html#a77a4d7eac217305e22a3c25b3756ef67',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a12e7f43cd9de2d9990054184c0a32839',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a959ccaa08f2999c50cea063b01e492e4',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a5592b24dad5ad030a1e4769b0a278f35',1,'ReadWriter::write_strided(int stride, int overall_n)']]], - ['write_5funsafe_8',['write_unsafe',['../scan_8h.html#a8010e7bdf7a72cbd35ce7cd7ecb08e32',1,'scan.h']]] + ['wait_5fgpu_3',['wait_gpu',['../classmlx_1_1core_1_1_fence.html#ab6d783dee02656ebb8ffcbbfa6de5b53',1,'mlx::core::Fence']]], + ['where_4',['where',['../group__ops.html#ga8a2056f8c9bb30914c40bcf509386491',1,'mlx::core']]], + ['write_5',['write',['../struct_read_writer.html#ac2ea71e41740ddc863890e3e8e6f09d0',1,'ReadWriter::write()'],['../classmlx_1_1core_1_1io_1_1_writer.html#ad9515b7f007338674de1e124cf77e125',1,'mlx::core::io::Writer::write()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#abca32838c9886f734d93430c34c07d7f',1,'mlx::core::io::FileWriter::write()'],['../struct_read_writer.html#a7a3d1396b0f83aa7506207bd6e7336bf',1,'ReadWriter::write() const'],['../struct_read_writer.html#ae1f0d3555b74998cc2d2288bce72a1f4',1,'ReadWriter::write() const']]], + ['write_5fpadded_6',['write_padded',['../struct_read_writer.html#a95367307acace2aa88226cf8956d2d88',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#abaf2a6ad4c88bd9f65fe1db1f73a8d87',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#a420453a56e77d6b3891ed4b5f178af9c',1,'ReadWriter::write_padded(int length, const device float2 *w_k) const']]], + ['write_5fsafe_7',['write_safe',['../scan_8h.html#ae86aef08e5ebc8790031eb51eefa754c',1,'scan.h']]], + ['write_5fstrided_8',['write_strided',['../struct_read_writer.html#a77a4d7eac217305e22a3c25b3756ef67',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a12e7f43cd9de2d9990054184c0a32839',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a959ccaa08f2999c50cea063b01e492e4',1,'ReadWriter::write_strided(int stride, int overall_n)'],['../struct_read_writer.html#a5592b24dad5ad030a1e4769b0a278f35',1,'ReadWriter::write_strided(int stride, int overall_n)']]], + ['write_5funsafe_9',['write_unsafe',['../scan_8h.html#a8010e7bdf7a72cbd35ce7cd7ecb08e32',1,'scan.h']]] ]; diff --git a/docs/build/html/search/functions_19.js b/docs/build/html/search/functions_19.js index fc00a1651..bab955a35 100644 --- a/docs/build/html/search/functions_19.js +++ b/docs/build/html/search/functions_19.js @@ -10,17 +10,18 @@ var searchData= ['_7edevicestream_7',['~DeviceStream',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a1c4397732f64f5811381dd01e30e020e',1,'mlx::core::metal::DeviceStream']]], ['_7efence_8',['~Fence',['../structmlx_1_1core_1_1metal_1_1_fence.html#a4940c1aece13814af7727de9abb511f2',1,'mlx::core::metal::Fence']]], ['_7efilewriter_9',['~FileWriter',['../classmlx_1_1core_1_1io_1_1_file_writer.html#ac325f51cd22050b6359056290e8ef42c',1,'mlx::core::io::FileWriter']]], - ['_7eintracing_10',['~InTracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#a83d57d7fa63bcb0ff72080191d0f177a',1,'mlx::core::detail::InTracing']]], - ['_7eparallelfilereader_11',['~ParallelFileReader',['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#ae9e3fc1cc3e827dae4d3d107f6780817',1,'mlx::core::io::ParallelFileReader']]], - ['_7eprimitive_12',['~Primitive',['../classmlx_1_1core_1_1_primitive.html#a29f70eb2d3b7e6c5fe52779c03f03777',1,'mlx::core::Primitive']]], - ['_7ereader_13',['~Reader',['../classmlx_1_1core_1_1io_1_1_reader.html#a81cd4747d81311c87dc6753f2d0d8b16',1,'mlx::core::io::Reader']]], - ['_7eresidencyset_14',['~ResidencySet',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#a998f07776f489bea9a7dd3c290ea7a79',1,'mlx::core::metal::ResidencySet']]], - ['_7eretaingraph_15',['~RetainGraph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a6bd6dc2e1caf2f764f39856a72ff6cbc',1,'mlx::core::detail::RetainGraph']]], - ['_7escheduler_16',['~Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a6626c4a743a2b3004fc14042bc8b0edf',1,'mlx::core::scheduler::Scheduler']]], - ['_7estreamcontext_17',['~StreamContext',['../structmlx_1_1core_1_1_stream_context.html#ac5be1c576d22b3d0b0a6fcc7e6abe659',1,'mlx::core::StreamContext']]], - ['_7estreamthread_18',['~StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a71de50591388b6e2cc6c57827e1a1ad4',1,'mlx::core::scheduler::StreamThread']]], - ['_7ethread_5fpool_19',['~thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a68f71e79fa93379fcc5819e533b45472',1,'pocketfft::detail::threading::thread_pool']]], - ['_7ethreadpool_20',['~ThreadPool',['../class_thread_pool.html#a44d3d2ab618970605e684efc216655eb',1,'ThreadPool']]], - ['_7eunaryprimitive_21',['~UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html#ac0677ab99a5ca660ed6ab7902ea364de',1,'mlx::core::UnaryPrimitive']]], - ['_7ewriter_22',['~Writer',['../classmlx_1_1core_1_1io_1_1_writer.html#a0e42f93a64118e9f5ede54ffe1bda045',1,'mlx::core::io::Writer']]] + ['_7egroupimpl_10',['~GroupImpl',['../classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a83f119b87210f53438c5ba675a78065e',1,'mlx::core::distributed::detail::GroupImpl']]], + ['_7eintracing_11',['~InTracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#a83d57d7fa63bcb0ff72080191d0f177a',1,'mlx::core::detail::InTracing']]], + ['_7eparallelfilereader_12',['~ParallelFileReader',['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#ae9e3fc1cc3e827dae4d3d107f6780817',1,'mlx::core::io::ParallelFileReader']]], + ['_7eprimitive_13',['~Primitive',['../classmlx_1_1core_1_1_primitive.html#a29f70eb2d3b7e6c5fe52779c03f03777',1,'mlx::core::Primitive']]], + ['_7ereader_14',['~Reader',['../classmlx_1_1core_1_1io_1_1_reader.html#a81cd4747d81311c87dc6753f2d0d8b16',1,'mlx::core::io::Reader']]], + ['_7eresidencyset_15',['~ResidencySet',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#a998f07776f489bea9a7dd3c290ea7a79',1,'mlx::core::metal::ResidencySet']]], + ['_7eretaingraph_16',['~RetainGraph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a6bd6dc2e1caf2f764f39856a72ff6cbc',1,'mlx::core::detail::RetainGraph']]], + ['_7escheduler_17',['~Scheduler',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a6626c4a743a2b3004fc14042bc8b0edf',1,'mlx::core::scheduler::Scheduler']]], + ['_7estreamcontext_18',['~StreamContext',['../structmlx_1_1core_1_1_stream_context.html#ac5be1c576d22b3d0b0a6fcc7e6abe659',1,'mlx::core::StreamContext']]], + ['_7estreamthread_19',['~StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a71de50591388b6e2cc6c57827e1a1ad4',1,'mlx::core::scheduler::StreamThread']]], + ['_7ethread_5fpool_20',['~thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a68f71e79fa93379fcc5819e533b45472',1,'pocketfft::detail::threading::thread_pool']]], + ['_7ethreadpool_21',['~ThreadPool',['../class_thread_pool.html#a44d3d2ab618970605e684efc216655eb',1,'ThreadPool']]], + ['_7eunaryprimitive_22',['~UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html#ac0677ab99a5ca660ed6ab7902ea364de',1,'mlx::core::UnaryPrimitive']]], + ['_7ewriter_23',['~Writer',['../classmlx_1_1core_1_1io_1_1_writer.html#a0e42f93a64118e9f5ede54ffe1bda045',1,'mlx::core::io::Writer']]] ]; diff --git a/docs/build/html/search/functions_2.js b/docs/build/html/search/functions_2.js index b22de5526..a6816c71f 100644 --- a/docs/build/html/search/functions_2.js +++ b/docs/build/html/search/functions_2.js @@ -1,54 +1,58 @@ var searchData= [ - ['begin_0',['begin',['../classmlx_1_1core_1_1array.html#a76b258b169d7d73419ebbf85340fb914',1,'mlx::core::array']]], - ['bernoulli_1',['bernoulli',['../namespacemlx_1_1core_1_1random.html#a1c601b637f60071dfc85cad19a841744',1,'mlx::core::random::bernoulli(const array &p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aaa49f6c2af5496822fa09435e54275cb',1,'mlx::core::random::bernoulli(const array &p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa9e360f9cb7bd23221352ed9e31d83c2',1,'mlx::core::random::bernoulli(T p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a2778876cd2318fec69cd1f3fc0955d68',1,'mlx::core::random::bernoulli(T p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#ad7eb4467e2f9d5f74a5607b29a935b6e',1,'mlx::core::random::bernoulli(const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['bfloat16_5fto_5fuint16_2',['bfloat16_to_uint16',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h']]], - ['bfloat_5fbits_5fto_5ffloat_3',['bfloat_bits_to_float',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3b33ae338dc4f223d0f3c748de07bad1',1,'bf16.h']]], - ['bfs_5fmax_5fwidth_4',['bfs_max_width',['../namespacemlx_1_1core_1_1env.html#ac3266e1259a64c8b56bdc6c7029179f2',1,'mlx::core::env']]], - ['binary_5',['binary',['../namespacemlx_1_1core_1_1metal.html#a269d591ec02e2f7c0f7a718fbfa37f73',1,'mlx::core::metal']]], - ['binary_5fg_6',['binary_g',['../metal_2kernels_2binary_8h.html#ab6b062acfb0497230e9476482d9dac20',1,'binary_g(device const T *a, device const T *b, device U *c, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a84dbd99589c7a18ff4d91056dc6fe17a',1,'binary_g(device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary_two.h']]], - ['binary_5fg_5fnd1_7',['binary_g_nd1',['../metal_2kernels_2binary_8h.html#aba4dd8bf59ed391789110e08ecfec6c2',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a1f2e8e7dc1998bab3864184878a75776',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary_two.h']]], - ['binary_5fg_5fnd2_8',['binary_g_nd2',['../metal_2kernels_2binary_8h.html#a73869f3771f16108f37120e485ce6e0b',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a0d31d8d6c8a1845d3329ef2e23e3fff6',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary_two.h']]], - ['binary_5fg_5fnd3_9',['binary_g_nd3',['../metal_2kernels_2binary_8h.html#ab6eb3c1f7349a52d5befff14e796320a',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#aae95e8f5fa772f780fc88054a2e35878',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary_two.h']]], - ['binary_5fop_5fgpu_10',['binary_op_gpu',['../namespacemlx_1_1core.html#ad884f4a36308b5b4f8a5d990d2e086df',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a094876ea5a2a2445ab64efc8222da202',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], - ['binary_5fop_5fgpu_5finplace_11',['binary_op_gpu_inplace',['../namespacemlx_1_1core.html#a8616c0b7b0fc118a75400bc86404c367',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a7e6af6624e322e7ad60a3873a66e18a3',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], - ['binary_5fops_12',['binary_ops',['../namespacemlx_1_1core_1_1metal.html#a8db7f9cc781d4bfb08423a401665f322',1,'mlx::core::metal']]], - ['binary_5fss_13',['binary_ss',['../metal_2kernels_2binary_8h.html#a242b8b29a852c255467e50628c6dccf5',1,'binary_ss(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#af8a791ac7ca88d32cd8f4e9ac0f9ab4f',1,'binary_ss(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fsv_14',['binary_sv',['../metal_2kernels_2binary_8h.html#a4116c35f2e4632366d1611d5a95ba141',1,'binary_sv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab18c6ecf5065275c93701efd095c916c',1,'binary_sv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fsv2_15',['binary_sv2',['../metal_2kernels_2binary_8h.html#aa8c48b1b21d8f5a181f5443de2346589',1,'binary_sv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a08822ff98ea6f61a98b49a9e9a38b891',1,'binary_sv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], - ['binary_5ftwo_16',['binary_two',['../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb',1,'mlx::core::metal']]], - ['binary_5fvs_17',['binary_vs',['../metal_2kernels_2binary_8h.html#a649851d133358dd5832a73b1061b3313',1,'binary_vs(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca',1,'binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fvs2_18',['binary_vs2',['../metal_2kernels_2binary_8h.html#a48bd82eb10f9c623ce7d28daec4fa512',1,'binary_vs2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a273d2f31691f2c64623c2a97eab344be',1,'binary_vs2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], - ['binary_5fvv_19',['binary_vv',['../metal_2kernels_2binary_8h.html#add6a9aeee3cb0ba909574f27fa9ecd5b',1,'binary_vv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9',1,'binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fvv2_20',['binary_vv2',['../metal_2kernels_2binary_8h.html#a19dbbf8fea68b64bdd25dc8d36865171',1,'binary_vv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12e80730e43dfaa4c79ce8d5f99edc50',1,'binary_vv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], - ['bits_21',['bits',['../namespacemlx_1_1core_1_1random.html#ad7d1c0b530906538dd8fb31b17382f2b',1,'mlx::core::random::bits(const Shape &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a035d36774135faabad33d8f64a879df7',1,'mlx::core::random::bits(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['bits_5fto_5fbfloat_22',['bits_to_bfloat',['../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca',1,'_MLX_BFloat16']]], - ['bitwise_5fand_23',['bitwise_and',['../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada',1,'mlx::core']]], - ['bitwise_5for_24',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]], - ['bitwise_5fxor_25',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]], - ['bitwisebinary_26',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary']]], - ['block_5fmasked_5fgemm_27',['block_masked_gemm',['../steel__gemm__masked_8h.html#a3642df61cba8eff47a08a20834137eac',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h'],['../steel__gemm__masked_8h.html#ac59086cee8cf396751da691fe80cda7b',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device bool *out_mask, const device bool *lhs_mask, const device bool *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h']]], - ['block_5fmasked_5fmm_28',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]], - ['block_5fsort_29',['block_sort',['../struct_kernel_merge_sort.html#a56b644ec66f7fb5c01b280f124304be9',1,'KernelMergeSort::block_sort()'],['../struct_kernel_multi_block_merge_sort.html#a322ed2eac315a561e0fd90af2fd577eb',1,'KernelMultiBlockMergeSort::block_sort()'],['../sort_8h.html#a93f14092416169c4449141043ac45ffd',1,'block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid): sort.h']]], - ['block_5fsort_5fnc_30',['block_sort_nc',['../sort_8h.html#ad272ef4422c75565a79b7f5e751bb247',1,'sort.h']]], - ['blockloader_31',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)']]], - ['blockloadert_32',['BlockLoaderT',['../structmlx_1_1steel_1_1_block_loader_t.html#a076616a7c67ad1b847e0e6b046077ee2',1,'mlx::steel::BlockLoaderT']]], - ['blockmaskedmm_33',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM']]], - ['blockmma_34',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)']]], - ['bluestein_5ffft_35',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]], - ['broadcast_36',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html#a004cce3029c0427569830016f99648cb',1,'mlx::core::Broadcast']]], - ['broadcast_5farrays_37',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]], - ['broadcast_5fshapes_38',['broadcast_shapes',['../namespacemlx_1_1core.html#aefa6a2ec6439e9619cafd227a1dc14ab',1,'mlx::core']]], - ['broadcast_5fto_39',['broadcast_to',['../group__ops.html#ga2fd5891f11593b7f09550e884f969013',1,'mlx::core']]], - ['broadcastaxes_40',['BroadcastAxes',['../classmlx_1_1core_1_1_broadcast_axes.html#a5136f33489670cdc0802e46725288195',1,'mlx::core::BroadcastAxes']]], - ['bs_5fqmm_5fn_41',['bs_qmm_n',['../quantized_8h.html#a323d2d70799c8d9dffe4b64a4285a799',1,'quantized.h']]], - ['bs_5fqmm_5ft_42',['bs_qmm_t',['../quantized_8h.html#a693057a0c311a73ac0759e005b9806c1',1,'quantized.h']]], - ['bs_5fqmv_43',['bs_qmv',['../quantized_8h.html#aaf4fb9c4318c5cd27d118004dbdeba61',1,'quantized.h']]], - ['bs_5fqmv_5ffast_44',['bs_qmv_fast',['../quantized_8h.html#a359282a9f71e487e5d86d246896ab33d',1,'quantized.h']]], - ['bs_5fqvm_45',['bs_qvm',['../quantized_8h.html#a2c53419ba5019d4722c0f4c2026b1142',1,'quantized.h']]], - ['buffer_46',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::allocator::Buffer::Buffer()'],['../classmlx_1_1core_1_1metal_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::metal::Buffer::Buffer()']]], - ['buffer_47',['buffer',['../classmlx_1_1core_1_1array.html#ab3daf04c27c4593d9d73c397b8484a08',1,'mlx::core::array::buffer()'],['../classmlx_1_1core_1_1array.html#a634466ce661485394f2fdc3bd6796bcd',1,'mlx::core::array::buffer() const']]], - ['buffer_5fsize_48',['buffer_size',['../classmlx_1_1core_1_1array.html#a914577c63755b2e862d2da68bbf8e3dd',1,'mlx::core::array']]], - ['build_5fcommand_49',['build_command',['../classmlx_1_1core_1_1_jit_compiler.html#a10a5cde91ab929ccbdbdf4c4d940f156',1,'mlx::core::JitCompiler']]], - ['build_5flib_5fname_50',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]] + ['barrier_0',['barrier',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a7f028c6ca48e75bf2c1806b5b8cfc90e',1,'mlx::core::metal::CommandEncoder::barrier()'],['../structmlx_1_1core_1_1_command_encoder.html#a7f028c6ca48e75bf2c1806b5b8cfc90e',1,'mlx::core::CommandEncoder::barrier()']]], + ['begin_1',['begin',['../classmlx_1_1core_1_1array.html#a76b258b169d7d73419ebbf85340fb914',1,'mlx::core::array']]], + ['bernoulli_2',['bernoulli',['../namespacemlx_1_1core_1_1random.html#a1c601b637f60071dfc85cad19a841744',1,'mlx::core::random::bernoulli(const array &p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aaa49f6c2af5496822fa09435e54275cb',1,'mlx::core::random::bernoulli(const array &p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa9e360f9cb7bd23221352ed9e31d83c2',1,'mlx::core::random::bernoulli(T p, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a2778876cd2318fec69cd1f3fc0955d68',1,'mlx::core::random::bernoulli(T p, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#ad7eb4467e2f9d5f74a5607b29a935b6e',1,'mlx::core::random::bernoulli(const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['bfloat16_5fto_5fuint16_3',['bfloat16_to_uint16',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a1420e191fa60d707dce327d0938e3088',1,'bfloat16_to_uint16(const bfloat16_t x): bf16.h']]], + ['bfloat_5fbits_5fto_5ffloat_4',['bfloat_bits_to_float',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3b33ae338dc4f223d0f3c748de07bad1',1,'bf16.h']]], + ['bfs_5fmax_5fwidth_5',['bfs_max_width',['../namespacemlx_1_1core_1_1env.html#ac3266e1259a64c8b56bdc6c7029179f2',1,'mlx::core::env']]], + ['binary_6',['binary',['../namespacemlx_1_1core.html#ae374861abd45cf019c3e6be2026f3798',1,'mlx::core::binary()'],['../namespacemlx_1_1core_1_1metal.html#a269d591ec02e2f7c0f7a718fbfa37f73',1,'mlx::core::metal::binary()']]], + ['binary_5fg_7',['binary_g',['../metal_2kernels_2binary_8h.html#ab6b062acfb0497230e9476482d9dac20',1,'binary_g(device const T *a, device const T *b, device U *c, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a84dbd99589c7a18ff4d91056dc6fe17a',1,'binary_g(device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary_two.h']]], + ['binary_5fg_5fnd1_8',['binary_g_nd1',['../metal_2kernels_2binary_8h.html#aba4dd8bf59ed391789110e08ecfec6c2',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a1f2e8e7dc1998bab3864184878a75776',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index): binary_two.h']]], + ['binary_5fg_5fnd2_9',['binary_g_nd2',['../metal_2kernels_2binary_8h.html#a73869f3771f16108f37120e485ce6e0b',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a0d31d8d6c8a1845d3329ef2e23e3fff6',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5fg_5fnd3_10',['binary_g_nd3',['../metal_2kernels_2binary_8h.html#ab6eb3c1f7349a52d5befff14e796320a',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#aae95e8f5fa772f780fc88054a2e35878',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim): binary_two.h']]], + ['binary_5fop_11',['binary_op',['../namespacemlx_1_1core.html#a9c1c1fdf9a0840a16a4d10a8f74f761d',1,'mlx::core::binary_op(const array &a, const array &b, array &out, Op op)'],['../namespacemlx_1_1core.html#a2aca3458c56605a74d07ec39876549bc',1,'mlx::core::binary_op(const array &a, const array &b, array &out, Op op)']]], + ['binary_5fop_5fdims_12',['binary_op_dims',['../namespacemlx_1_1core.html#a7ca09ebf776fe32db580f9038587ec31',1,'mlx::core']]], + ['binary_5fop_5fdispatch_5fdims_13',['binary_op_dispatch_dims',['../namespacemlx_1_1core.html#a66c9ee5018168b9101de52e0122d9755',1,'mlx::core']]], + ['binary_5fop_5fgpu_14',['binary_op_gpu',['../namespacemlx_1_1core.html#ad884f4a36308b5b4f8a5d990d2e086df',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a094876ea5a2a2445ab64efc8222da202',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], + ['binary_5fop_5fgpu_5finplace_15',['binary_op_gpu_inplace',['../namespacemlx_1_1core.html#a8616c0b7b0fc118a75400bc86404c367',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a7e6af6624e322e7ad60a3873a66e18a3',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], + ['binary_5fops_16',['binary_ops',['../namespacemlx_1_1core_1_1metal.html#a8db7f9cc781d4bfb08423a401665f322',1,'mlx::core::metal']]], + ['binary_5fss_17',['binary_ss',['../metal_2kernels_2binary_8h.html#a242b8b29a852c255467e50628c6dccf5',1,'binary_ss(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#af8a791ac7ca88d32cd8f4e9ac0f9ab4f',1,'binary_ss(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fsv_18',['binary_sv',['../metal_2kernels_2binary_8h.html#a4116c35f2e4632366d1611d5a95ba141',1,'binary_sv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab18c6ecf5065275c93701efd095c916c',1,'binary_sv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fsv2_19',['binary_sv2',['../metal_2kernels_2binary_8h.html#aa8c48b1b21d8f5a181f5443de2346589',1,'binary_sv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a08822ff98ea6f61a98b49a9e9a38b891',1,'binary_sv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5ftwo_20',['binary_two',['../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb',1,'mlx::core::metal']]], + ['binary_5fvs_21',['binary_vs',['../metal_2kernels_2binary_8h.html#a649851d133358dd5832a73b1061b3313',1,'binary_vs(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca',1,'binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fvs2_22',['binary_vs2',['../metal_2kernels_2binary_8h.html#a48bd82eb10f9c623ce7d28daec4fa512',1,'binary_vs2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a273d2f31691f2c64623c2a97eab344be',1,'binary_vs2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5fvv_23',['binary_vv',['../metal_2kernels_2binary_8h.html#add6a9aeee3cb0ba909574f27fa9ecd5b',1,'binary_vv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9',1,'binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fvv2_24',['binary_vv2',['../metal_2kernels_2binary_8h.html#a19dbbf8fea68b64bdd25dc8d36865171',1,'binary_vv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12e80730e43dfaa4c79ce8d5f99edc50',1,'binary_vv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['bits_25',['bits',['../namespacemlx_1_1core_1_1random.html#ad7d1c0b530906538dd8fb31b17382f2b',1,'mlx::core::random::bits(const Shape &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a035d36774135faabad33d8f64a879df7',1,'mlx::core::random::bits(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['bits_5fto_5fbfloat_26',['bits_to_bfloat',['../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca',1,'_MLX_BFloat16']]], + ['bitwise_5fand_27',['bitwise_and',['../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada',1,'mlx::core']]], + ['bitwise_5for_28',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]], + ['bitwise_5fxor_29',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]], + ['bitwisebinary_30',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary']]], + ['block_5fmasked_5fgemm_31',['block_masked_gemm',['../steel__gemm__masked_8h.html#a3642df61cba8eff47a08a20834137eac',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h'],['../steel__gemm__masked_8h.html#ac59086cee8cf396751da691fe80cda7b',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant int64_t *batch_strides, const device bool *out_mask, const device bool *lhs_mask, const device bool *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h']]], + ['block_5fmasked_5fmm_32',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]], + ['block_5fsort_33',['block_sort',['../struct_kernel_merge_sort.html#a90102e73c3f6e9652e9870d57b129e71',1,'KernelMergeSort::block_sort()'],['../struct_kernel_multi_block_merge_sort.html#a56b3d8bf726e5619325dd07752666f8a',1,'KernelMultiBlockMergeSort::block_sort()'],['../sort_8h.html#a93f14092416169c4449141043ac45ffd',1,'block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid): sort.h']]], + ['block_5fsort_5fnc_34',['block_sort_nc',['../sort_8h.html#ad272ef4422c75565a79b7f5e751bb247',1,'sort.h']]], + ['blockloader_35',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader(const device T *src_, const int src_ld_, threadgroup T *dst_, ushort simd_group_id, ushort simd_lane_id)']]], + ['blockloadert_36',['BlockLoaderT',['../structmlx_1_1steel_1_1_block_loader_t.html#a076616a7c67ad1b847e0e6b046077ee2',1,'mlx::steel::BlockLoaderT']]], + ['blockmaskedmm_37',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM']]], + ['blockmma_38',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA(ushort simd_group_id, ushort simd_lane_id)']]], + ['bluestein_5ffft_39',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]], + ['broadcast_40',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html#a004cce3029c0427569830016f99648cb',1,'mlx::core::Broadcast']]], + ['broadcast_5farrays_41',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]], + ['broadcast_5fshapes_42',['broadcast_shapes',['../namespacemlx_1_1core.html#aefa6a2ec6439e9619cafd227a1dc14ab',1,'mlx::core']]], + ['broadcast_5fto_43',['broadcast_to',['../group__ops.html#ga2fd5891f11593b7f09550e884f969013',1,'mlx::core']]], + ['broadcastaxes_44',['BroadcastAxes',['../classmlx_1_1core_1_1_broadcast_axes.html#a5136f33489670cdc0802e46725288195',1,'mlx::core::BroadcastAxes']]], + ['bs_5fqmm_5fn_45',['bs_qmm_n',['../quantized_8h.html#a323d2d70799c8d9dffe4b64a4285a799',1,'quantized.h']]], + ['bs_5fqmm_5ft_46',['bs_qmm_t',['../quantized_8h.html#a693057a0c311a73ac0759e005b9806c1',1,'quantized.h']]], + ['bs_5fqmv_47',['bs_qmv',['../quantized_8h.html#aaf4fb9c4318c5cd27d118004dbdeba61',1,'quantized.h']]], + ['bs_5fqmv_5ffast_48',['bs_qmv_fast',['../quantized_8h.html#a359282a9f71e487e5d86d246896ab33d',1,'quantized.h']]], + ['bs_5fqvm_49',['bs_qvm',['../quantized_8h.html#a2c53419ba5019d4722c0f4c2026b1142',1,'quantized.h']]], + ['buffer_50',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::allocator::Buffer::Buffer()'],['../classmlx_1_1core_1_1metal_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::metal::Buffer::Buffer()']]], + ['buffer_51',['buffer',['../classmlx_1_1core_1_1array.html#ab3daf04c27c4593d9d73c397b8484a08',1,'mlx::core::array::buffer()'],['../classmlx_1_1core_1_1array.html#a634466ce661485394f2fdc3bd6796bcd',1,'mlx::core::array::buffer() const']]], + ['buffer_5fsize_52',['buffer_size',['../classmlx_1_1core_1_1array.html#a914577c63755b2e862d2da68bbf8e3dd',1,'mlx::core::array']]], + ['build_5fcommand_53',['build_command',['../classmlx_1_1core_1_1_jit_compiler.html#a10a5cde91ab929ccbdbdf4c4d940f156',1,'mlx::core::JitCompiler']]], + ['build_5flib_5fname_54',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/functions_3.js b/docs/build/html/search/functions_3.js index dc6b3a319..7e951d71a 100644 --- a/docs/build/html/search/functions_3.js +++ b/docs/build/html/search/functions_3.js @@ -4,7 +4,7 @@ var searchData= ['c2r_1',['c2r',['../namespacepocketfft_1_1detail.html#ab26cbfed16f487b987f50bf63bfc1ab9',1,'pocketfft::detail::c2r(const shape_t &shape_out, const stride_t &stride_in, const stride_t &stride_out, size_t axis, bool forward, const std::complex< T > *data_in, T *data_out, T fct, size_t nthreads=1)'],['../namespacepocketfft_1_1detail.html#a788506fff59f8e13056247076cac51c1',1,'pocketfft::detail::c2r(const shape_t &shape_out, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, bool forward, const std::complex< T > *data_in, T *data_out, T fct, size_t nthreads=1)'],['../namespacepocketfft.html#ab26cbfed16f487b987f50bf63bfc1ab9',1,'pocketfft::c2r()']]], ['categorical_2',['categorical',['../namespacemlx_1_1core_1_1random.html#a1ede4c7f5c86e7b7e834953853ffae9a',1,'mlx::core::random::categorical(const array &logits, int axis, const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa7104c436b3972a2480cfeb54554855f',1,'mlx::core::random::categorical(const array &logits_, int axis, int num_samples, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa336e774783543705dffe2ad5b2c49c1',1,'mlx::core::random::categorical(const array &logits, int axis=-1, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], ['ceil_3',['Ceil',['../classmlx_1_1core_1_1_ceil.html#aede38610ca25429f229301546bc9b682',1,'mlx::core::Ceil']]], - ['ceil_4',['ceil',['../namespacemetal.html#ad63204d38bc01df6ffc64583f7886b3c',1,'metal::ceil()'],['../namespacemetal_1_1fast.html#a97b0bbd79f1f45d9d3104d712914e6b8',1,'metal::fast::ceil()'],['../namespacemetal_1_1precise.html#a8577549a1afeea206dd9a2004af2868d',1,'metal::precise::ceil()'],['../group__ops.html#ga1404ecceff83fd9b9139b7520f55e096',1,'mlx::core::ceil()']]], + ['ceil_4',['ceil',['../namespacemlx_1_1core_1_1simd.html#ae1d5460c58c507a0104d8dfa90343f12',1,'mlx::core::simd::ceil(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a4e54bd4ceb51ec41b0f95ebabe558713',1,'mlx::core::simd::ceil(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#ae5714693df24c8e26384fe5b5888376d',1,'mlx::core::simd::ceil(Simd< float16_t, N > a)'],['../namespacemetal.html#ad63204d38bc01df6ffc64583f7886b3c',1,'metal::ceil()'],['../namespacemetal_1_1fast.html#a97b0bbd79f1f45d9d3104d712914e6b8',1,'metal::fast::ceil()'],['../namespacemetal_1_1precise.html#a8577549a1afeea206dd9a2004af2868d',1,'metal::precise::ceil()'],['../group__ops.html#ga1404ecceff83fd9b9139b7520f55e096',1,'mlx::core::ceil()']]], ['ceildiv_5',['ceildiv',['../backend_2metal_2kernels_2utils_8h.html#a8e5a4b0fb5d018d7b078d147efe4f1e3',1,'utils.h']]], ['cfftp_6',['cfftp',['../classpocketfft_1_1detail_1_1cfftp.html#a121dcd8d4db180061d09fc1c5eb7da27',1,'pocketfft::detail::cfftp']]], ['check_5fcontiguity_7',['check_contiguity',['../namespacemlx_1_1core.html#ab2b50a44a9d3a06282be4611f5fc7447',1,'mlx::core']]], @@ -12,104 +12,105 @@ var searchData= ['cholesky_9',['Cholesky',['../classmlx_1_1core_1_1_cholesky.html#a6ae2e30b85f99f4f0d7f14c7949818ab',1,'mlx::core::Cholesky']]], ['cholesky_10',['cholesky',['../namespacemlx_1_1core_1_1linalg.html#a46c8a4f806f0a97a4323e91189aa512b',1,'mlx::core::linalg']]], ['cholesky_5finv_11',['cholesky_inv',['../namespacemlx_1_1core_1_1linalg.html#aef0fe4894c5cf98792d59859c6d20511',1,'mlx::core::linalg']]], - ['clear_12',['clear',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa97a98e423827a889c13a92217626ec7',1,'mlx::steel::MMATile::clear()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa97a98e423827a889c13a92217626ec7',1,'mlx::steel::MMATile::clear()']]], - ['clear_5fcache_13',['clear_cache',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a447c1eb38c00d2e8e521675297f4a9b1',1,'mlx::core::metal::MetalAllocator::clear_cache()'],['../namespacemlx_1_1core_1_1metal.html#a22b3384ebd17f2fca198f81b9f1b6dc3',1,'mlx::core::metal::clear_cache()']]], - ['clip_14',['clip',['../group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8',1,'mlx::core']]], - ['close_15',['close',['../structmlx_1_1core_1_1_function_exporter.html#a43454277d21709d2f9b23a7ead0e674f',1,'mlx::core::FunctionExporter']]], - ['cmplx_16',['cmplx',['../structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c',1,'pocketfft::detail::cmplx::cmplx()'],['../structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a',1,'pocketfft::detail::cmplx::cmplx(T r_, T i_)']]], - ['cndarr_17',['cndarr',['../classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129',1,'pocketfft::detail::cndarr']]], - ['col_5freduce_5f2pass_18',['col_reduce_2pass',['../reduce__col_8h.html#a5a3d4847e4ae940d3a532d790c0b9b27',1,'reduce_col.h']]], - ['col_5freduce_5flongcolumn_19',['col_reduce_longcolumn',['../reduce__col_8h.html#a00bb1ed154aef8c1d2cb15d0c4231a51',1,'reduce_col.h']]], - ['col_5freduce_5flooped_20',['col_reduce_looped',['../reduce__col_8h.html#a59be8d8c4bca2217f62ea868550393c0',1,'reduce_col.h']]], - ['col_5freduce_5fsmall_21',['col_reduce_small',['../reduce__col_8h.html#a674f4b6075bab1b89778e10ab24c557e',1,'reduce_col.h']]], - ['collapse_5fcontiguous_5fdims_22',['collapse_contiguous_dims',['../namespacemlx_1_1core.html#a4d594bb84abeff4619d1abb77b20123e',1,'mlx::core::collapse_contiguous_dims(const Shape &shape, const std::vector< Strides > &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#a977c7c84de79ad67055ae2a89b7f6869',1,'mlx::core::collapse_contiguous_dims(const std::vector< array > &xs, size_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#ac813412cce77fc1340dcfefc6e099276',1,'mlx::core::collapse_contiguous_dims(Arrays &&... xs)'],['../namespacemlx_1_1core.html#a79acfa8bc30c1f213bf893b5983eb666',1,'mlx::core::collapse_contiguous_dims(const Shape &shape, const Strides &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#ab607cd6974ca6606826e785807156d6a',1,'mlx::core::collapse_contiguous_dims(const array &a, int64_t size_cap=std::numeric_limits< int32_t >::max())']]], - ['commandencoder_23',['CommandEncoder',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::metal::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::metal::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete'],['../structmlx_1_1core_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete']]], - ['commit_5fcommand_5fbuffer_24',['commit_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a95248f1387824067fd4fed23ace5ac0c',1,'mlx::core::metal::Device']]], - ['communication_5fstream_25',['communication_stream',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6',1,'mlx::core::distributed::detail']]], - ['compile_26',['compile',['../namespacemlx_1_1core.html#a55933c6665de9f81059120d6b0de1c87',1,'mlx::core::compile(std::function< std::vector< array >(const std::vector< array > &)> fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#a56de948c2dffbf35c88035bc1561db53',1,'mlx::core::compile(std::vector< array >(fun)(const std::vector< array > &), bool shapeless=false)'],['../namespacemlx_1_1core.html#ace67713d269595f5f2265e46728a6f9c',1,'mlx::core::compile(F &&f, bool shapeless=false)'],['../namespacemlx_1_1core_1_1detail.html#af556c7576658b2e2498ead70339d95e5',1,'mlx::core::detail::compile(std::function< std::vector< array >(const std::vector< array > &)> fun, std::uintptr_t fun_id, bool shapeless=false, std::vector< uint64_t > constants={})']]], - ['compile_5favailable_5ffor_5fdevice_27',['compile_available_for_device',['../namespacemlx_1_1core_1_1detail.html#aeeff2ba6ec3d9d4ed090de6d2681dbc2',1,'mlx::core::detail']]], - ['compile_5fclear_5fcache_28',['compile_clear_cache',['../namespacemlx_1_1core_1_1detail.html#a3fb927c209b946aefebb195993fbe4cf',1,'mlx::core::detail']]], - ['compile_5fdfs_29',['compile_dfs',['../namespacemlx_1_1core_1_1detail.html#a545fccdb5dc365b154cf4f0a2ca4753b',1,'mlx::core::detail']]], - ['compile_5ferase_30',['compile_erase',['../namespacemlx_1_1core_1_1detail.html#a69eb76a14f845ca000f1ccb2edda0175',1,'mlx::core::detail']]], - ['compile_5freplace_31',['compile_replace',['../namespacemlx_1_1core_1_1detail.html#a56fc01df6ba4c508d1da8b366b1328ac',1,'mlx::core::detail']]], - ['compile_5fsimplify_32',['compile_simplify',['../namespacemlx_1_1core_1_1detail.html#a33c878c900ca06f35d479f99c57b9e39',1,'mlx::core::detail']]], - ['compile_5ftrace_33',['compile_trace',['../namespacemlx_1_1core_1_1detail.html#ac2163a401119bb6edecfeb43373ef0dd',1,'mlx::core::detail']]], - ['compile_5fvalidate_5fshapeless_34',['compile_validate_shapeless',['../namespacemlx_1_1core_1_1detail.html#a10d612cb45a17fa17b704a357a902a68',1,'mlx::core::detail']]], - ['compiled_35',['Compiled',['../classmlx_1_1core_1_1_compiled.html#a2d8cefff835c419a48a077d306b8e051',1,'mlx::core::Compiled']]], - ['compiled_5fallocate_5foutputs_36',['compiled_allocate_outputs',['../namespacemlx_1_1core.html#ab8c3c4fc05745f586de922c8266f4fce',1,'mlx::core']]], - ['compiled_5fcheck_5fcontiguity_37',['compiled_check_contiguity',['../namespacemlx_1_1core.html#a562040f4a03f2c0a5d50eb9c8f14a8be',1,'mlx::core']]], - ['complex128_5ft_38',['complex128_t',['../structmlx_1_1core_1_1complex128__t.html#aa15d0b805f8790f7c7b76fc7b9d677e0',1,'mlx::core::complex128_t::complex128_t(double v, double u)'],['../structmlx_1_1core_1_1complex128__t.html#abf2842253b874f9f13f39ea68a89e5b6',1,'mlx::core::complex128_t::complex128_t(std::complex< double > v)'],['../structmlx_1_1core_1_1complex128__t.html#a526fba96d7e815360cb4226af085a1bf',1,'mlx::core::complex128_t::complex128_t(T x)']]], - ['complex64_5ft_39',['complex64_t',['../structcomplex64__t.html#adbd392a5e92d31997380ad0a38be4be8',1,'complex64_t::complex64_t(float real, float imag)'],['../structcomplex64__t.html#a29782289bb90d6294099667b86509cd3',1,'complex64_t::complex64_t()'],['../structcomplex64__t.html#a905b048d70eb8d748a62454268242291',1,'complex64_t::complex64_t() threadgroup'],['../structcomplex64__t.html#a33a2452eb33b5ed53655773539c357a5',1,'complex64_t::complex64_t(T x) thread'],['../structcomplex64__t.html#a89b65ace8588b7bf215355f705eb23d9',1,'complex64_t::complex64_t(T x) threadgroup'],['../structcomplex64__t.html#ac81b486f642fb3b26c5d659917bdbcd0',1,'complex64_t::complex64_t(T x) device'],['../structcomplex64__t.html#a0a27a41206400f1e62b60ceb56960c93',1,'complex64_t::complex64_t(T x) const ant'],['../structmlx_1_1core_1_1complex64__t.html#a697cc973ae27d63c8e00d830e780bd8c',1,'mlx::core::complex64_t::complex64_t(float v, float u)'],['../structmlx_1_1core_1_1complex64__t.html#ae065e39938f9c4374b4116f4c67d4d09',1,'mlx::core::complex64_t::complex64_t(std::complex< float > v)'],['../structmlx_1_1core_1_1complex64__t.html#a2232cbbe591a9d2bc228cb23fac38b50',1,'mlx::core::complex64_t::complex64_t(T x)']]], - ['complex_5fmul_40',['complex_mul',['../radix_8h.html#a5bfc53b531214c9ce277bebc18aa67d6',1,'radix.h']]], - ['complex_5fmul_5fconj_41',['complex_mul_conj',['../radix_8h.html#a0e2dfd3d1dda09f47ccc64eec35629f3',1,'radix.h']]], - ['compute_5fstrided_5findices_42',['compute_strided_indices',['../struct_read_writer.html#a7c903fbb8b85a856ba5564d7df537cdf',1,'ReadWriter']]], - ['concatenate_43',['Concatenate',['../classmlx_1_1core_1_1_concatenate.html#acff07853de2d31faeec7c4ca40ce0888',1,'mlx::core::Concatenate']]], - ['concatenate_44',['concatenate',['../namespacemlx_1_1core.html#a76a2e310857f60f5ea6f1388d45b964d',1,'mlx::core::concatenate(std::string &acc, T first)'],['../namespacemlx_1_1core.html#aaf51544472fa87fa974686eacdd2a4a6',1,'mlx::core::concatenate(std::string &acc, T first, Args... args)'],['../group__ops.html#ga52838af566948b1b96e7aa00832071b3',1,'mlx::core::concatenate(std::vector< array > arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga666ac69778984fafdc2f51d296270468',1,'mlx::core::concatenate(std::vector< array > arrays, StreamOrDevice s={})']]], - ['concatenate_5fgpu_45',['concatenate_gpu',['../namespacemlx_1_1core.html#a050299d0d366ca5c9d09d1004dcc3e7d',1,'mlx::core']]], - ['concurrentcontext_46',['ConcurrentContext',['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::metal::CommandEncoder::ConcurrentContext::ConcurrentContext()'],['../structmlx_1_1core_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::CommandEncoder::ConcurrentContext::ConcurrentContext()']]], - ['conj_47',['conj',['../namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3',1,'pocketfft::detail']]], - ['conjugate_48',['Conjugate',['../classmlx_1_1core_1_1_conjugate.html#a627f9e6a8729fb3ffb3ca3228d007c87',1,'mlx::core::Conjugate']]], - ['conjugate_49',['conjugate',['../group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23',1,'mlx::core']]], - ['contiguous_50',['Contiguous',['../classmlx_1_1core_1_1_contiguous.html#a3e83f414c02ae0b92a50b6f8e402e1c0',1,'mlx::core::Contiguous']]], - ['contiguous_51',['contiguous',['../group__ops.html#ga8ab10aa6c41416d739791164a52b25d5',1,'mlx::core']]], - ['contiguous_5fscan_52',['contiguous_scan',['../scan_8h.html#a60d279b9add7d56639bb209408f09d79',1,'scan.h']]], - ['contiguousiterator_53',['ContiguousIterator',['../structmlx_1_1core_1_1_contiguous_iterator.html#a727442ddff5fd3c3ebe09b000a01c9d3',1,'mlx::core::ContiguousIterator::ContiguousIterator()'],['../structmlx_1_1core_1_1_contiguous_iterator.html#aa82bec516eb54656c74fdaa74de1d735',1,'mlx::core::ContiguousIterator::ContiguousIterator(const array &a)'],['../structmlx_1_1core_1_1_contiguous_iterator.html#a8760380bff7462a886e7a4edd2955375',1,'mlx::core::ContiguousIterator::ContiguousIterator(const Shape &shape, const Strides &strides, int dims)']]], - ['conv_54',['conv',['../namespacemlx_1_1core_1_1metal.html#ab1704e853394c725668c06752ebb5c24',1,'mlx::core::metal']]], - ['conv1d_55',['conv1d',['../group__ops.html#ga30d47e08093c03a3676f235f9f559411',1,'mlx::core']]], - ['conv2d_56',['conv2d',['../group__ops.html#ga73b02833229678786e7f302d458d5a83',1,'mlx::core']]], - ['conv2dinputblockloadergeneral_57',['Conv2DInputBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1d83af561a483432bf8dcb42e734b23b',1,'mlx::steel::Conv2DInputBlockLoaderGeneral']]], - ['conv2dinputblockloaderlargefilter_58',['Conv2DInputBlockLoaderLargeFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8755116a535539744e4947bc69f9c50f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter']]], - ['conv2dinputblockloadersmallchannels_59',['Conv2DInputBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ab9fd3fdeab94470dde3326f1dd5c455a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels']]], - ['conv2dinputblockloadersmallfilter_60',['Conv2DInputBlockLoaderSmallFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0a2cbf57c51cd928722e3f06aafcf933',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], - ['conv2dweightblockloader_61',['Conv2DWeightBlockLoader',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a9a7dca3512b64cffb6eac305d795831c',1,'mlx::steel::Conv2DWeightBlockLoader']]], - ['conv2dweightblockloadergeneral_62',['Conv2DWeightBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ad0550fabbdc9297559381a5b488e9af1',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], - ['conv2dweightblockloadersmallchannels_63',['Conv2DWeightBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae1806ea1c19713819dee83a38ab35fa6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels']]], - ['conv3d_64',['conv3d',['../group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca',1,'mlx::core']]], - ['conv_5fgeneral_65',['conv_general',['../group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e',1,'mlx::core::conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})'],['../group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d',1,'mlx::core::conv_general(const array &input, const array &weight, std::vector< int > stride={}, std::vector< int > padding={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})']]], - ['conv_5ftranspose1d_66',['conv_transpose1d',['../group__ops.html#gaa30bf1adcd78d1c2595d07b215731714',1,'mlx::core']]], - ['conv_5ftranspose2d_67',['conv_transpose2d',['../group__ops.html#gaebb59971cb9bc45005dc1d398e4f0a3d',1,'mlx::core']]], - ['conv_5ftranspose3d_68',['conv_transpose3d',['../group__ops.html#ga8db814da631d9cd32a8d6563bf4ac530',1,'mlx::core']]], - ['convolution_69',['Convolution',['../classmlx_1_1core_1_1_convolution.html#a6f1de77b719bb13217b0d8c64cabb8ef',1,'mlx::core::Convolution']]], - ['copy_70',['Copy',['../classmlx_1_1core_1_1_copy.html#a6243e044af119105ffaaed7d405cd584',1,'mlx::core::Copy']]], - ['copy_71',['copy',['../namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2',1,'mlx::core::copy()'],['../namespacemlx_1_1core_1_1metal.html#aa215e631e2680f04a591b88d91571719',1,'mlx::core::metal::copy()'],['../group__ops.html#gae306e93af12f774bd80bad6c231b09d6',1,'mlx::core::copy()']]], - ['copy_5fg_72',['copy_g',['../metal_2kernels_2copy_8h.html#a71e4103db4689d90ef6f9d5ba93604cf',1,'copy.h']]], - ['copy_5fg_5fnd1_73',['copy_g_nd1',['../metal_2kernels_2copy_8h.html#a232c5c6b8386cf8ecbf4cdadb6e4176e',1,'copy.h']]], - ['copy_5fg_5fnd2_74',['copy_g_nd2',['../metal_2kernels_2copy_8h.html#a39ec5b7b8351e4332b842982a2ee6260',1,'copy.h']]], - ['copy_5fg_5fnd3_75',['copy_g_nd3',['../metal_2kernels_2copy_8h.html#aab82689380897ff4716b5eafd6ef3ecc',1,'copy.h']]], - ['copy_5fgg_76',['copy_gg',['../metal_2kernels_2copy_8h.html#ade9a9eea9b8262a854a11721fe2bb9fa',1,'copy.h']]], - ['copy_5fgg_5fdynamic_77',['copy_gg_dynamic',['../metal_2kernels_2copy_8h.html#ad0f05a73165d4ee38c9f02c705ea6ca8',1,'copy.h']]], - ['copy_5fgg_5fdynamic_5fnd1_78',['copy_gg_dynamic_nd1',['../metal_2kernels_2copy_8h.html#a8548ea41cac179084ddd33d26921576f',1,'copy.h']]], - ['copy_5fgg_5fdynamic_5fnd2_79',['copy_gg_dynamic_nd2',['../metal_2kernels_2copy_8h.html#a9b9266ee25a4dbcbe4fde883b40170f1',1,'copy.h']]], - ['copy_5fgg_5fdynamic_5fnd3_80',['copy_gg_dynamic_nd3',['../metal_2kernels_2copy_8h.html#af33ccc02f10bcb5c19ea7b1dd0af4956',1,'copy.h']]], - ['copy_5fgg_5fnd1_81',['copy_gg_nd1',['../metal_2kernels_2copy_8h.html#a370d7bbba1a4b0d64da873bafd29a78b',1,'copy.h']]], - ['copy_5fgg_5fnd2_82',['copy_gg_nd2',['../metal_2kernels_2copy_8h.html#af0b06ac3a96852a64fa4274a94b58301',1,'copy.h']]], - ['copy_5fgg_5fnd3_83',['copy_gg_nd3',['../metal_2kernels_2copy_8h.html#a3f3836ad0b6545ec9b9e1864224f7a13',1,'copy.h']]], - ['copy_5fgpu_84',['copy_gpu',['../namespacemlx_1_1core.html#addaa46a13ac2deb1d9ce621338320e0e',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a6a6f4e46c8fc44fdc74c50ace02bcf38',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype)']]], - ['copy_5fgpu_5finplace_85',['copy_gpu_inplace',['../namespacemlx_1_1core.html#a473fb602368f6c73d9105c9a151c4c82',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s, const std::optional< array > &dynamic_i_offset=std::nullopt, const std::optional< array > &dynamic_o_offset=std::nullopt)'],['../namespacemlx_1_1core.html#a58ef0842dd1b8f79159d5fb6777d30a1',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a49fc043a981925b9be79e37fc415d966',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const Strides &i_strides, int64_t i_offset, CopyType ctype, const Stream &s)']]], - ['copy_5fhartley_86',['copy_hartley',['../namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#ae7b44d2773d9d06a9787aff01d66b3ed',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], - ['copy_5finplace_87',['copy_inplace',['../namespacemlx_1_1core.html#a98495894a796b2cc6d022e7a03432c64',1,'mlx::core::copy_inplace(const array &src, array &dst, CopyType ctype)'],['../namespacemlx_1_1core.html#ae85bafda5ab0b4b2289591260cf07685',1,'mlx::core::copy_inplace(const array &src, array &dst, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)']]], - ['copy_5finput_88',['copy_input',['../namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< cmplx< T > > &src, cmplx< vtype_t< T > > *dst)'],['../namespacepocketfft_1_1detail.html#a30fc708f9d8f9cfa74194925c7863c0a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, vtype_t< T > *dst)'],['../namespacepocketfft_1_1detail.html#a3387bd35f237870e42b8461769e6aec4',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, T *dst)']]], - ['copy_5foutput_89',['copy_output',['../namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const cmplx< vtype_t< T > > *src, ndarr< cmplx< T > > &dst)'],['../namespacepocketfft_1_1detail.html#a21980853aca4d92ed06e3dcffe7ef660',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#a310481c334e46674710ba794ad7403c0',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], - ['copy_5fs_90',['copy_s',['../metal_2kernels_2copy_8h.html#aef09f9b9475345b1bba121d037d222ea',1,'copy.h']]], - ['copy_5fs2_91',['copy_s2',['../metal_2kernels_2copy_8h.html#a8023e9335cc5334847a8d315042be3a3',1,'copy.h']]], - ['copy_5fshared_5fbuffer_92',['copy_shared_buffer',['../classmlx_1_1core_1_1array.html#ad2814dbffa5ad174d9c97a10bf4cf26b',1,'mlx::core::array::copy_shared_buffer(const array &other, const Strides &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a92974c656c35a972ad241f80584bbd29',1,'mlx::core::array::copy_shared_buffer(const array &other)']]], - ['copy_5fv_93',['copy_v',['../metal_2kernels_2copy_8h.html#ae26a13e0c8e6c15f7b10078e65970659',1,'copy.h']]], - ['copy_5fv2_94',['copy_v2',['../metal_2kernels_2copy_8h.html#aee14a5326f53d9b30b0b38e27d180ef3',1,'copy.h']]], - ['cos_95',['Cos',['../classmlx_1_1core_1_1_cos.html#a2acb9fcf0901462189c476756fd99995',1,'mlx::core::Cos']]], - ['cos_96',['cos',['../namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d',1,'pocketfft::detail::cos()'],['../namespacemetal.html#a2fa4778a6fe2fa43253ea724e5a608a3',1,'metal::cos()'],['../namespacemetal_1_1fast.html#a75b6bb32fa3870eda46a7bfc9f481f88',1,'metal::fast::cos()'],['../namespacemetal_1_1precise.html#ac4941f62e7d8ab9d7cabbd967aa9f220',1,'metal::precise::cos()'],['../group__ops.html#ga39dfdf72b556012aa35ff27a94116e74',1,'mlx::core::cos()']]], - ['cosh_97',['Cosh',['../classmlx_1_1core_1_1_cosh.html#a44e8ac2e09a55ec32e9dc6641eedc8f1',1,'mlx::core::Cosh']]], - ['cosh_98',['cosh',['../namespacemetal.html#a8a68a88cc110830d057dbd71431b93c0',1,'metal::cosh()'],['../namespacemetal_1_1fast.html#a31544ad9de28012a4ddda86e3966a77e',1,'metal::fast::cosh()'],['../namespacemetal_1_1precise.html#a72d86d508300a9b58f4ccbbe70da4fbc',1,'metal::precise::cosh()'],['../group__ops.html#ga2181b71cda88007a3092be4795ff0715',1,'mlx::core::cosh()']]], - ['cospi_99',['cospi',['../namespacemetal.html#a5c2f37939ad705ddea4409d3bedb8ce1',1,'metal::cospi()'],['../namespacemetal_1_1fast.html#a9906b41f75319b384ffb570cc94d67ce',1,'metal::fast::cospi()'],['../namespacemetal_1_1precise.html#a2392b78bd196efdbbac65901c4ab20e7',1,'metal::precise::cospi()']]], - ['cost_5fguess_100',['cost_guess',['../structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0',1,'pocketfft::detail::util']]], - ['count_5fdown_101',['count_down',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342',1,'pocketfft::detail::threading::latch']]], - ['cross_102',['cross',['../namespacemlx_1_1core_1_1linalg.html#abcda3fbda45183c21e7f27aa0dde64e6',1,'mlx::core::linalg']]], - ['cummax_103',['cummax',['../group__ops.html#gaee37cac8476e8f8d666bcded5bc59143',1,'mlx::core']]], - ['cummin_104',['cummin',['../group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8',1,'mlx::core']]], - ['cumprod_105',['cumprod',['../group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f',1,'mlx::core']]], - ['cumsum_106',['cumsum',['../group__ops.html#gaddc825a5c173e195ab0fda83ad630420',1,'mlx::core']]], - ['custom_107',['Custom',['../classmlx_1_1core_1_1fast_1_1_custom.html#a4186fea23f7156c38960426821fca313',1,'mlx::core::fast::Custom']]], - ['custom_5ffunction_108',['custom_function',['../namespacemlx_1_1core.html#a8d3ca5fbaecdb995660c24cde5aeebaf',1,'mlx::core']]], - ['custom_5fvjp_109',['custom_vjp',['../namespacemlx_1_1core.html#a9290596250fa308df4c69b44483bb8aa',1,'mlx::core']]], - ['customkernel_110',['CustomKernel',['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a954893e07f0d36715b4e1e414b6f2153',1,'mlx::core::fast::CustomKernel']]], - ['customtransforms_111',['CustomTransforms',['../classmlx_1_1core_1_1_custom_transforms.html#ab52abadb9c6f6db83d087c7b751be488',1,'mlx::core::CustomTransforms']]] + ['clamp_12',['clamp',['../namespacemlx_1_1core_1_1simd.html#ae4be4d88cd8eba7a8c1784fd53b86edb',1,'mlx::core::simd::clamp(Simd< T, N > v, Simd< T, N > min, Simd< T, N > max)'],['../namespacemlx_1_1core_1_1simd.html#a567c06bf988af03988478679055a6c45',1,'mlx::core::simd::clamp(Simd< T, 1 > v, Simd< T, 1 > min, Simd< T, 1 > max)'],['../namespacemlx_1_1core_1_1simd.html#a146d2a834c936a381c1f86caffa822d7',1,'mlx::core::simd::clamp(Simd< float16_t, N > v, Simd< float16_t, N > min, Simd< float16_t, N > max)']]], + ['clear_13',['clear',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa97a98e423827a889c13a92217626ec7',1,'mlx::steel::MMATile::clear()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa97a98e423827a889c13a92217626ec7',1,'mlx::steel::MMATile::clear()']]], + ['clear_5fcache_14',['clear_cache',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a447c1eb38c00d2e8e521675297f4a9b1',1,'mlx::core::metal::MetalAllocator::clear_cache()'],['../namespacemlx_1_1core_1_1metal.html#a22b3384ebd17f2fca198f81b9f1b6dc3',1,'mlx::core::metal::clear_cache()']]], + ['clip_15',['clip',['../group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8',1,'mlx::core']]], + ['close_16',['close',['../structmlx_1_1core_1_1_function_exporter.html#a43454277d21709d2f9b23a7ead0e674f',1,'mlx::core::FunctionExporter']]], + ['cmplx_17',['cmplx',['../structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c',1,'pocketfft::detail::cmplx::cmplx()'],['../structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a',1,'pocketfft::detail::cmplx::cmplx(T r_, T i_)']]], + ['cndarr_18',['cndarr',['../classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129',1,'pocketfft::detail::cndarr']]], + ['col_5freduce_5f2pass_19',['col_reduce_2pass',['../reduce__col_8h.html#a5a3d4847e4ae940d3a532d790c0b9b27',1,'reduce_col.h']]], + ['col_5freduce_5flongcolumn_20',['col_reduce_longcolumn',['../reduce__col_8h.html#a00bb1ed154aef8c1d2cb15d0c4231a51',1,'reduce_col.h']]], + ['col_5freduce_5flooped_21',['col_reduce_looped',['../reduce__col_8h.html#a59be8d8c4bca2217f62ea868550393c0',1,'reduce_col.h']]], + ['col_5freduce_5fsmall_22',['col_reduce_small',['../reduce__col_8h.html#a674f4b6075bab1b89778e10ab24c557e',1,'reduce_col.h']]], + ['collapse_5fcontiguous_5fdims_23',['collapse_contiguous_dims',['../namespacemlx_1_1core.html#a4d594bb84abeff4619d1abb77b20123e',1,'mlx::core::collapse_contiguous_dims(const Shape &shape, const std::vector< Strides > &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#a977c7c84de79ad67055ae2a89b7f6869',1,'mlx::core::collapse_contiguous_dims(const std::vector< array > &xs, size_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#ac813412cce77fc1340dcfefc6e099276',1,'mlx::core::collapse_contiguous_dims(Arrays &&... xs)'],['../namespacemlx_1_1core.html#a79acfa8bc30c1f213bf893b5983eb666',1,'mlx::core::collapse_contiguous_dims(const Shape &shape, const Strides &strides, int64_t size_cap=std::numeric_limits< int32_t >::max())'],['../namespacemlx_1_1core.html#ab607cd6974ca6606826e785807156d6a',1,'mlx::core::collapse_contiguous_dims(const array &a, int64_t size_cap=std::numeric_limits< int32_t >::max())']]], + ['commandencoder_24',['CommandEncoder',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::metal::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::metal::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete'],['../structmlx_1_1core_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete']]], + ['commit_5fcommand_5fbuffer_25',['commit_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a95248f1387824067fd4fed23ace5ac0c',1,'mlx::core::metal::Device']]], + ['communication_5fstream_26',['communication_stream',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6',1,'mlx::core::distributed::detail']]], + ['compile_27',['compile',['../namespacemlx_1_1core.html#a55933c6665de9f81059120d6b0de1c87',1,'mlx::core::compile(std::function< std::vector< array >(const std::vector< array > &)> fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#a56de948c2dffbf35c88035bc1561db53',1,'mlx::core::compile(std::vector< array >(fun)(const std::vector< array > &), bool shapeless=false)'],['../namespacemlx_1_1core.html#ace67713d269595f5f2265e46728a6f9c',1,'mlx::core::compile(F &&f, bool shapeless=false)'],['../namespacemlx_1_1core_1_1detail.html#af556c7576658b2e2498ead70339d95e5',1,'mlx::core::detail::compile(std::function< std::vector< array >(const std::vector< array > &)> fun, std::uintptr_t fun_id, bool shapeless=false, std::vector< uint64_t > constants={})']]], + ['compile_5favailable_5ffor_5fdevice_28',['compile_available_for_device',['../namespacemlx_1_1core_1_1detail.html#aeeff2ba6ec3d9d4ed090de6d2681dbc2',1,'mlx::core::detail']]], + ['compile_5fclear_5fcache_29',['compile_clear_cache',['../namespacemlx_1_1core_1_1detail.html#a3fb927c209b946aefebb195993fbe4cf',1,'mlx::core::detail']]], + ['compile_5fdfs_30',['compile_dfs',['../namespacemlx_1_1core_1_1detail.html#a545fccdb5dc365b154cf4f0a2ca4753b',1,'mlx::core::detail']]], + ['compile_5ferase_31',['compile_erase',['../namespacemlx_1_1core_1_1detail.html#a69eb76a14f845ca000f1ccb2edda0175',1,'mlx::core::detail']]], + ['compile_5freplace_32',['compile_replace',['../namespacemlx_1_1core_1_1detail.html#a56fc01df6ba4c508d1da8b366b1328ac',1,'mlx::core::detail']]], + ['compile_5fsimplify_33',['compile_simplify',['../namespacemlx_1_1core_1_1detail.html#a33c878c900ca06f35d479f99c57b9e39',1,'mlx::core::detail']]], + ['compile_5ftrace_34',['compile_trace',['../namespacemlx_1_1core_1_1detail.html#ac2163a401119bb6edecfeb43373ef0dd',1,'mlx::core::detail']]], + ['compile_5fvalidate_5fshapeless_35',['compile_validate_shapeless',['../namespacemlx_1_1core_1_1detail.html#a10d612cb45a17fa17b704a357a902a68',1,'mlx::core::detail']]], + ['compiled_36',['Compiled',['../classmlx_1_1core_1_1_compiled.html#a2d8cefff835c419a48a077d306b8e051',1,'mlx::core::Compiled']]], + ['compiled_5fallocate_5foutputs_37',['compiled_allocate_outputs',['../namespacemlx_1_1core.html#ab8c3c4fc05745f586de922c8266f4fce',1,'mlx::core']]], + ['compiled_5fcheck_5fcontiguity_38',['compiled_check_contiguity',['../namespacemlx_1_1core.html#a562040f4a03f2c0a5d50eb9c8f14a8be',1,'mlx::core']]], + ['complex128_5ft_39',['complex128_t',['../structmlx_1_1core_1_1complex128__t.html#a4330d04587f3282bcd650e36532da178',1,'mlx::core::complex128_t::complex128_t()'],['../structmlx_1_1core_1_1complex128__t.html#aa15d0b805f8790f7c7b76fc7b9d677e0',1,'mlx::core::complex128_t::complex128_t(double v, double u)'],['../structmlx_1_1core_1_1complex128__t.html#abf2842253b874f9f13f39ea68a89e5b6',1,'mlx::core::complex128_t::complex128_t(std::complex< double > v)'],['../structmlx_1_1core_1_1complex128__t.html#a526fba96d7e815360cb4226af085a1bf',1,'mlx::core::complex128_t::complex128_t(T x)']]], + ['complex64_5ft_40',['complex64_t',['../structcomplex64__t.html#adbd392a5e92d31997380ad0a38be4be8',1,'complex64_t::complex64_t(float real, float imag)'],['../structcomplex64__t.html#a29782289bb90d6294099667b86509cd3',1,'complex64_t::complex64_t()'],['../structcomplex64__t.html#a905b048d70eb8d748a62454268242291',1,'complex64_t::complex64_t() threadgroup'],['../structcomplex64__t.html#a33a2452eb33b5ed53655773539c357a5',1,'complex64_t::complex64_t(T x) thread'],['../structcomplex64__t.html#a89b65ace8588b7bf215355f705eb23d9',1,'complex64_t::complex64_t(T x) threadgroup'],['../structcomplex64__t.html#ac81b486f642fb3b26c5d659917bdbcd0',1,'complex64_t::complex64_t(T x) device'],['../structcomplex64__t.html#a0a27a41206400f1e62b60ceb56960c93',1,'complex64_t::complex64_t(T x) const ant'],['../structmlx_1_1core_1_1complex64__t.html#ad27bed7d6b7966bfcf563af06bedddf3',1,'mlx::core::complex64_t::complex64_t()'],['../structmlx_1_1core_1_1complex64__t.html#a697cc973ae27d63c8e00d830e780bd8c',1,'mlx::core::complex64_t::complex64_t(float v, float u)'],['../structmlx_1_1core_1_1complex64__t.html#ae065e39938f9c4374b4116f4c67d4d09',1,'mlx::core::complex64_t::complex64_t(std::complex< float > v)'],['../structmlx_1_1core_1_1complex64__t.html#a2232cbbe591a9d2bc228cb23fac38b50',1,'mlx::core::complex64_t::complex64_t(T x)']]], + ['complex_5fmul_41',['complex_mul',['../radix_8h.html#a5bfc53b531214c9ce277bebc18aa67d6',1,'radix.h']]], + ['complex_5fmul_5fconj_42',['complex_mul_conj',['../radix_8h.html#a0e2dfd3d1dda09f47ccc64eec35629f3',1,'radix.h']]], + ['compute_5fstrided_5findices_43',['compute_strided_indices',['../struct_read_writer.html#a7c903fbb8b85a856ba5564d7df537cdf',1,'ReadWriter']]], + ['concatenate_44',['Concatenate',['../classmlx_1_1core_1_1_concatenate.html#acff07853de2d31faeec7c4ca40ce0888',1,'mlx::core::Concatenate']]], + ['concatenate_45',['concatenate',['../namespacemlx_1_1core.html#a76a2e310857f60f5ea6f1388d45b964d',1,'mlx::core::concatenate(std::string &acc, T first)'],['../namespacemlx_1_1core.html#aaf51544472fa87fa974686eacdd2a4a6',1,'mlx::core::concatenate(std::string &acc, T first, Args... args)'],['../group__ops.html#ga52838af566948b1b96e7aa00832071b3',1,'mlx::core::concatenate(std::vector< array > arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga666ac69778984fafdc2f51d296270468',1,'mlx::core::concatenate(std::vector< array > arrays, StreamOrDevice s={})']]], + ['concatenate_5fgpu_46',['concatenate_gpu',['../namespacemlx_1_1core.html#a050299d0d366ca5c9d09d1004dcc3e7d',1,'mlx::core']]], + ['concurrentcontext_47',['ConcurrentContext',['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::metal::CommandEncoder::ConcurrentContext::ConcurrentContext()'],['../structmlx_1_1core_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::CommandEncoder::ConcurrentContext::ConcurrentContext()']]], + ['conj_48',['conj',['../namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3',1,'pocketfft::detail::conj()'],['../namespacemlx_1_1core_1_1simd.html#a660b79a51fb439f4aba91e2aea276300',1,'mlx::core::simd::conj()']]], + ['conjugate_49',['Conjugate',['../classmlx_1_1core_1_1_conjugate.html#a627f9e6a8729fb3ffb3ca3228d007c87',1,'mlx::core::Conjugate']]], + ['conjugate_50',['conjugate',['../group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23',1,'mlx::core']]], + ['contiguous_51',['Contiguous',['../classmlx_1_1core_1_1_contiguous.html#a3e83f414c02ae0b92a50b6f8e402e1c0',1,'mlx::core::Contiguous']]], + ['contiguous_52',['contiguous',['../group__ops.html#ga8ab10aa6c41416d739791164a52b25d5',1,'mlx::core']]], + ['contiguous_5fscan_53',['contiguous_scan',['../scan_8h.html#a60d279b9add7d56639bb209408f09d79',1,'scan.h']]], + ['contiguousiterator_54',['ContiguousIterator',['../structmlx_1_1core_1_1_contiguous_iterator.html#a727442ddff5fd3c3ebe09b000a01c9d3',1,'mlx::core::ContiguousIterator::ContiguousIterator()'],['../structmlx_1_1core_1_1_contiguous_iterator.html#aa82bec516eb54656c74fdaa74de1d735',1,'mlx::core::ContiguousIterator::ContiguousIterator(const array &a)'],['../structmlx_1_1core_1_1_contiguous_iterator.html#a8760380bff7462a886e7a4edd2955375',1,'mlx::core::ContiguousIterator::ContiguousIterator(const Shape &shape, const Strides &strides, int dims)']]], + ['conv_55',['conv',['../namespacemlx_1_1core_1_1metal.html#ab1704e853394c725668c06752ebb5c24',1,'mlx::core::metal']]], + ['conv1d_56',['conv1d',['../group__ops.html#ga30d47e08093c03a3676f235f9f559411',1,'mlx::core']]], + ['conv2d_57',['conv2d',['../group__ops.html#ga73b02833229678786e7f302d458d5a83',1,'mlx::core']]], + ['conv2dinputblockloadergeneral_58',['Conv2DInputBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1d83af561a483432bf8dcb42e734b23b',1,'mlx::steel::Conv2DInputBlockLoaderGeneral']]], + ['conv2dinputblockloaderlargefilter_59',['Conv2DInputBlockLoaderLargeFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8755116a535539744e4947bc69f9c50f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter']]], + ['conv2dinputblockloadersmallchannels_60',['Conv2DInputBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ab9fd3fdeab94470dde3326f1dd5c455a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels']]], + ['conv2dinputblockloadersmallfilter_61',['Conv2DInputBlockLoaderSmallFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0a2cbf57c51cd928722e3f06aafcf933',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], + ['conv2dweightblockloader_62',['Conv2DWeightBlockLoader',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a9a7dca3512b64cffb6eac305d795831c',1,'mlx::steel::Conv2DWeightBlockLoader']]], + ['conv2dweightblockloadergeneral_63',['Conv2DWeightBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ad0550fabbdc9297559381a5b488e9af1',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], + ['conv2dweightblockloadersmallchannels_64',['Conv2DWeightBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae1806ea1c19713819dee83a38ab35fa6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels']]], + ['conv3d_65',['conv3d',['../group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca',1,'mlx::core']]], + ['conv_5fgeneral_66',['conv_general',['../group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e',1,'mlx::core::conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})'],['../group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d',1,'mlx::core::conv_general(const array &input, const array &weight, std::vector< int > stride={}, std::vector< int > padding={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})']]], + ['conv_5ftranspose1d_67',['conv_transpose1d',['../group__ops.html#gaa30bf1adcd78d1c2595d07b215731714',1,'mlx::core']]], + ['conv_5ftranspose2d_68',['conv_transpose2d',['../group__ops.html#gaebb59971cb9bc45005dc1d398e4f0a3d',1,'mlx::core']]], + ['conv_5ftranspose3d_69',['conv_transpose3d',['../group__ops.html#ga8db814da631d9cd32a8d6563bf4ac530',1,'mlx::core']]], + ['convolution_70',['Convolution',['../classmlx_1_1core_1_1_convolution.html#a6f1de77b719bb13217b0d8c64cabb8ef',1,'mlx::core::Convolution']]], + ['copy_71',['Copy',['../classmlx_1_1core_1_1_copy.html#a6243e044af119105ffaaed7d405cd584',1,'mlx::core::Copy']]], + ['copy_72',['copy',['../namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2',1,'mlx::core::copy()'],['../namespacemlx_1_1core_1_1metal.html#aa215e631e2680f04a591b88d91571719',1,'mlx::core::metal::copy()'],['../group__ops.html#gae306e93af12f774bd80bad6c231b09d6',1,'mlx::core::copy()']]], + ['copy_5fg_73',['copy_g',['../metal_2kernels_2copy_8h.html#a71e4103db4689d90ef6f9d5ba93604cf',1,'copy.h']]], + ['copy_5fg_5fnd1_74',['copy_g_nd1',['../metal_2kernels_2copy_8h.html#a232c5c6b8386cf8ecbf4cdadb6e4176e',1,'copy.h']]], + ['copy_5fg_5fnd2_75',['copy_g_nd2',['../metal_2kernels_2copy_8h.html#a39ec5b7b8351e4332b842982a2ee6260',1,'copy.h']]], + ['copy_5fg_5fnd3_76',['copy_g_nd3',['../metal_2kernels_2copy_8h.html#aab82689380897ff4716b5eafd6ef3ecc',1,'copy.h']]], + ['copy_5fgg_77',['copy_gg',['../metal_2kernels_2copy_8h.html#ade9a9eea9b8262a854a11721fe2bb9fa',1,'copy.h']]], + ['copy_5fgg_5fdynamic_78',['copy_gg_dynamic',['../metal_2kernels_2copy_8h.html#ad0f05a73165d4ee38c9f02c705ea6ca8',1,'copy.h']]], + ['copy_5fgg_5fdynamic_5fnd1_79',['copy_gg_dynamic_nd1',['../metal_2kernels_2copy_8h.html#a8548ea41cac179084ddd33d26921576f',1,'copy.h']]], + ['copy_5fgg_5fdynamic_5fnd2_80',['copy_gg_dynamic_nd2',['../metal_2kernels_2copy_8h.html#a9b9266ee25a4dbcbe4fde883b40170f1',1,'copy.h']]], + ['copy_5fgg_5fdynamic_5fnd3_81',['copy_gg_dynamic_nd3',['../metal_2kernels_2copy_8h.html#af33ccc02f10bcb5c19ea7b1dd0af4956',1,'copy.h']]], + ['copy_5fgg_5fnd1_82',['copy_gg_nd1',['../metal_2kernels_2copy_8h.html#a370d7bbba1a4b0d64da873bafd29a78b',1,'copy.h']]], + ['copy_5fgg_5fnd2_83',['copy_gg_nd2',['../metal_2kernels_2copy_8h.html#af0b06ac3a96852a64fa4274a94b58301',1,'copy.h']]], + ['copy_5fgg_5fnd3_84',['copy_gg_nd3',['../metal_2kernels_2copy_8h.html#a3f3836ad0b6545ec9b9e1864224f7a13',1,'copy.h']]], + ['copy_5fgpu_85',['copy_gpu',['../namespacemlx_1_1core.html#addaa46a13ac2deb1d9ce621338320e0e',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a6a6f4e46c8fc44fdc74c50ace02bcf38',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype)']]], + ['copy_5fgpu_5finplace_86',['copy_gpu_inplace',['../namespacemlx_1_1core.html#a473fb602368f6c73d9105c9a151c4c82',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s, const std::optional< array > &dynamic_i_offset=std::nullopt, const std::optional< array > &dynamic_o_offset=std::nullopt)'],['../namespacemlx_1_1core.html#a58ef0842dd1b8f79159d5fb6777d30a1',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a49fc043a981925b9be79e37fc415d966',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const Strides &i_strides, int64_t i_offset, CopyType ctype, const Stream &s)']]], + ['copy_5fhartley_87',['copy_hartley',['../namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#ae7b44d2773d9d06a9787aff01d66b3ed',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], + ['copy_5finplace_88',['copy_inplace',['../namespacemlx_1_1core.html#a98495894a796b2cc6d022e7a03432c64',1,'mlx::core::copy_inplace(const array &src, array &dst, CopyType ctype)'],['../namespacemlx_1_1core.html#ae85bafda5ab0b4b2289591260cf07685',1,'mlx::core::copy_inplace(const array &src, array &dst, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)']]], + ['copy_5finput_89',['copy_input',['../namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< cmplx< T > > &src, cmplx< vtype_t< T > > *dst)'],['../namespacepocketfft_1_1detail.html#a30fc708f9d8f9cfa74194925c7863c0a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, vtype_t< T > *dst)'],['../namespacepocketfft_1_1detail.html#a3387bd35f237870e42b8461769e6aec4',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, T *dst)']]], + ['copy_5foutput_90',['copy_output',['../namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const cmplx< vtype_t< T > > *src, ndarr< cmplx< T > > &dst)'],['../namespacepocketfft_1_1detail.html#a21980853aca4d92ed06e3dcffe7ef660',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#a310481c334e46674710ba794ad7403c0',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], + ['copy_5fs_91',['copy_s',['../metal_2kernels_2copy_8h.html#aef09f9b9475345b1bba121d037d222ea',1,'copy.h']]], + ['copy_5fs2_92',['copy_s2',['../metal_2kernels_2copy_8h.html#a8023e9335cc5334847a8d315042be3a3',1,'copy.h']]], + ['copy_5fshared_5fbuffer_93',['copy_shared_buffer',['../classmlx_1_1core_1_1array.html#ad2814dbffa5ad174d9c97a10bf4cf26b',1,'mlx::core::array::copy_shared_buffer(const array &other, const Strides &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a92974c656c35a972ad241f80584bbd29',1,'mlx::core::array::copy_shared_buffer(const array &other)']]], + ['copy_5fv_94',['copy_v',['../metal_2kernels_2copy_8h.html#ae26a13e0c8e6c15f7b10078e65970659',1,'copy.h']]], + ['copy_5fv2_95',['copy_v2',['../metal_2kernels_2copy_8h.html#aee14a5326f53d9b30b0b38e27d180ef3',1,'copy.h']]], + ['cos_96',['Cos',['../classmlx_1_1core_1_1_cos.html#a2acb9fcf0901462189c476756fd99995',1,'mlx::core::Cos']]], + ['cos_97',['cos',['../namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d',1,'pocketfft::detail::cos()'],['../namespacemlx_1_1core_1_1simd.html#ab179f429e34cd6d5c37050ea7e7c54ad',1,'mlx::core::simd::cos()'],['../namespacemetal.html#a2fa4778a6fe2fa43253ea724e5a608a3',1,'metal::cos()'],['../namespacemetal_1_1fast.html#a75b6bb32fa3870eda46a7bfc9f481f88',1,'metal::fast::cos()'],['../namespacemetal_1_1precise.html#ac4941f62e7d8ab9d7cabbd967aa9f220',1,'metal::precise::cos()'],['../group__ops.html#ga39dfdf72b556012aa35ff27a94116e74',1,'mlx::core::cos()']]], + ['cosh_98',['Cosh',['../classmlx_1_1core_1_1_cosh.html#a44e8ac2e09a55ec32e9dc6641eedc8f1',1,'mlx::core::Cosh']]], + ['cosh_99',['cosh',['../namespacemlx_1_1core_1_1simd.html#aedc18b6fdb820cce9125c977c02833aa',1,'mlx::core::simd::cosh(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#aa5b4f7d3b776e8d16907e15a11800f01',1,'mlx::core::simd::cosh(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ae1265896d855818d20f2de2a9ebb684a',1,'mlx::core::simd::cosh(Simd< T, 1 > in)'],['../namespacemetal.html#a8a68a88cc110830d057dbd71431b93c0',1,'metal::cosh()'],['../namespacemetal_1_1fast.html#a31544ad9de28012a4ddda86e3966a77e',1,'metal::fast::cosh()'],['../namespacemetal_1_1precise.html#a72d86d508300a9b58f4ccbbe70da4fbc',1,'metal::precise::cosh()'],['../group__ops.html#ga2181b71cda88007a3092be4795ff0715',1,'mlx::core::cosh()']]], + ['cospi_100',['cospi',['../namespacemetal.html#a5c2f37939ad705ddea4409d3bedb8ce1',1,'metal::cospi()'],['../namespacemetal_1_1fast.html#a9906b41f75319b384ffb570cc94d67ce',1,'metal::fast::cospi()'],['../namespacemetal_1_1precise.html#a2392b78bd196efdbbac65901c4ab20e7',1,'metal::precise::cospi()']]], + ['cost_5fguess_101',['cost_guess',['../structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0',1,'pocketfft::detail::util']]], + ['count_5fdown_102',['count_down',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342',1,'pocketfft::detail::threading::latch']]], + ['cross_103',['cross',['../namespacemlx_1_1core_1_1linalg.html#abcda3fbda45183c21e7f27aa0dde64e6',1,'mlx::core::linalg']]], + ['cummax_104',['cummax',['../group__ops.html#gaee37cac8476e8f8d666bcded5bc59143',1,'mlx::core']]], + ['cummin_105',['cummin',['../group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8',1,'mlx::core']]], + ['cumprod_106',['cumprod',['../group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f',1,'mlx::core']]], + ['cumsum_107',['cumsum',['../group__ops.html#gaddc825a5c173e195ab0fda83ad630420',1,'mlx::core']]], + ['custom_108',['Custom',['../classmlx_1_1core_1_1fast_1_1_custom.html#a4186fea23f7156c38960426821fca313',1,'mlx::core::fast::Custom']]], + ['custom_5ffunction_109',['custom_function',['../namespacemlx_1_1core.html#a8d3ca5fbaecdb995660c24cde5aeebaf',1,'mlx::core']]], + ['custom_5fvjp_110',['custom_vjp',['../namespacemlx_1_1core.html#a9290596250fa308df4c69b44483bb8aa',1,'mlx::core']]], + ['customkernel_111',['CustomKernel',['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a954893e07f0d36715b4e1e414b6f2153',1,'mlx::core::fast::CustomKernel']]], + ['customtransforms_112',['CustomTransforms',['../classmlx_1_1core_1_1_custom_transforms.html#ab52abadb9c6f6db83d087c7b751be488',1,'mlx::core::CustomTransforms']]] ]; diff --git a/docs/build/html/search/functions_4.js b/docs/build/html/search/functions_4.js index 10dafde1e..1b2cdedd0 100644 --- a/docs/build/html/search/functions_4.js +++ b/docs/build/html/search/functions_4.js @@ -12,33 +12,31 @@ var searchData= ['default_5f_9',['default_',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#ab5993daeed822c6b970caddab7e3fd90',1,'mlx::core::random::KeySequence']]], ['default_5fdevice_10',['default_device',['../namespacemlx_1_1core.html#a0196171cfe6ee2953113abce597dc815',1,'mlx::core']]], ['default_5fstream_11',['default_stream',['../namespacemlx_1_1core.html#ac198b7e282957c724c84a435e8f1215e',1,'mlx::core']]], - ['defaultcontiguousreduce_12',['DefaultContiguousReduce',['../structmlx_1_1core_1_1_default_contiguous_reduce.html#aeb4fb7fa1a4c8e7d1da1f450ce95c57f',1,'mlx::core::DefaultContiguousReduce']]], - ['defaultstridedreduce_13',['DefaultStridedReduce',['../structmlx_1_1core_1_1_default_strided_reduce.html#a477e5dd0dd33071e48769d11d19a13fb',1,'mlx::core::DefaultStridedReduce']]], - ['define_5finput_5foutput_5fshape_14',['DEFINE_INPUT_OUTPUT_SHAPE',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a2965dbda1bed67128e97c3c5d864c82f',1,'mlx::core::fast::RMSNorm::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#afd0818925ffea79f4e3dda0dd8cf0366',1,'mlx::core::fast::LayerNorm::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a361cc8e0e56ff45ec98dbf81ed8eff2c',1,'mlx::core::fast::RoPE::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a64d2ce4b46b529a6a9ef068947bc623e',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_INPUT_OUTPUT_SHAPE()']]], - ['define_5fprint_15',['DEFINE_PRINT',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a8af1e90d4aa56f31ec40ad152ebd2421',1,'mlx::core::distributed::AllGather::DEFINE_PRINT()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a31bf76e24cf3836cf1fd26da30712e31',1,'mlx::core::distributed::Send::DEFINE_PRINT()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a7a0cad13da7cf8e565934318a2bc34f1',1,'mlx::core::distributed::Recv::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae6eea81b5e3789c2f6f376cc07f0a47c',1,'mlx::core::fast::RMSNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a9895733eab845e11484d86cf6ecedced',1,'mlx::core::fast::RMSNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a467fcf02b3ddf1d8b6d476b244ae3568',1,'mlx::core::fast::LayerNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a5ab3eb5402c7e8060916056eb2b7887f',1,'mlx::core::fast::LayerNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a2b06fe64fa8feca65140632087065e16',1,'mlx::core::fast::RoPE::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a6cc2092fa5b8e7585921b8e0f3ec3db7',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a4b8f1b1f633002c8ca6fa8f0ef4dd587',1,'mlx::core::fast::AffineQuantize::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a116ecf31c8672c94e5ea06c1d43e9534',1,'mlx::core::fast::CustomKernel::DEFINE_PRINT()']]], - ['degrees_16',['degrees',['../group__ops.html#ga3a70569b50e1083c5ded199d73fb960c',1,'mlx::core']]], - ['denorm_5fmin_17',['denorm_min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a6a9dbcba4dd79cad50876dda506b9eed',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['depends_18',['Depends',['../classmlx_1_1core_1_1_depends.html#a4ccb792c99f5d8d133d3fac29f7d3f62',1,'mlx::core::Depends']]], - ['depends_19',['depends',['../group__ops.html#gac4a51a68fbe1725436b026d2fbb95759',1,'mlx::core']]], - ['dequantize_20',['dequantize',['../quantized_8h.html#aecff265b63566d0d5689cfc4e5b037d2',1,'dequantize(): quantized.h'],['../group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077',1,'mlx::core::dequantize()']]], - ['detach_21',['detach',['../classmlx_1_1core_1_1array.html#a84948c29df8c957904919c8602692bd2',1,'mlx::core::array']]], - ['device_22',['Device',['../classmlx_1_1core_1_1metal_1_1_device.html#ae0db74570eb4b19d8cf19774db91bfd6',1,'mlx::core::metal::Device::Device()'],['../classmlx_1_1core_1_1metal_1_1_device.html#abf59a4addb5473f9e814e3651ba85f06',1,'mlx::core::metal::Device::Device(const Device &)=delete'],['../structmlx_1_1core_1_1_device.html#a481ccfb94d689994396bd353e966b489',1,'mlx::core::Device::Device()']]], - ['device_23',['device',['../classmlx_1_1core_1_1_primitive.html#a8ae61e3289c4134232a69295268f8261',1,'mlx::core::Primitive::device()'],['../namespacemlx_1_1core_1_1metal.html#a910797b74824e6ee576fbb533dee8b57',1,'mlx::core::metal::device(mlx::core::Device)']]], - ['device_5finfo_24',['device_info',['../namespacemlx_1_1core_1_1metal.html#a6ad19c44efabb7423f973407926ead61',1,'mlx::core::metal']]], - ['devicestream_25',['DeviceStream',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a573326bc8b48e39076850c7bf52ad0d7',1,'mlx::core::metal::DeviceStream']]], - ['diag_26',['diag',['../group__ops.html#ga11af511875640e1fa88e0ca87e199344',1,'mlx::core']]], - ['diagonal_27',['diagonal',['../group__ops.html#ga9236b085a88ead3128ed8079d009cac6',1,'mlx::core']]], - ['disable_5fcompile_28',['disable_compile',['../namespacemlx_1_1core.html#a5f5fea955057bb3842b271b037909e66',1,'mlx::core']]], - ['dispatch_5fthreadgroups_29',['dispatch_threadgroups',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a85796b2bf41dbf347ae0978d4660600d',1,'mlx::core::metal::CommandEncoder::dispatch_threadgroups()'],['../structmlx_1_1core_1_1_command_encoder.html#a85796b2bf41dbf347ae0978d4660600d',1,'mlx::core::CommandEncoder::dispatch_threadgroups()']]], - ['dispatch_5fthreads_30',['dispatch_threads',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a0a8501b940e5a347475fa4bc38fb4c05',1,'mlx::core::metal::CommandEncoder::dispatch_threads()'],['../structmlx_1_1core_1_1_command_encoder.html#a0a8501b940e5a347475fa4bc38fb4c05',1,'mlx::core::CommandEncoder::dispatch_threads()']]], - ['distprimitive_31',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8c54166951522c2a52ef39fce8c87f8f',1,'mlx::core::distributed::DistPrimitive']]], - ['divide_32',['Divide',['../classmlx_1_1core_1_1_divide.html#a62fc71e8998be65ff18285dbbd21eedb',1,'mlx::core::Divide']]], - ['divide_33',['divide',['../namespacemetal.html#a2aea493fc1a874970b77ed0031e965df',1,'metal::divide()'],['../namespacemetal_1_1fast.html#ae70bc2185e4649369cf7b15f5e1d48be',1,'metal::fast::divide()'],['../namespacemetal_1_1precise.html#aec0982cdb96a08b61f51129150d82e9d',1,'metal::precise::divide()'],['../group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f',1,'mlx::core::divide()']]], - ['divmod_34',['DivMod',['../classmlx_1_1core_1_1_div_mod.html#a859e3b6149cdceab1c7ccfd2246fb826',1,'mlx::core::DivMod']]], - ['divmod_35',['divmod',['../group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894',1,'mlx::core']]], - ['dst_36',['dst',['../namespacepocketfft_1_1detail.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::detail::dst()'],['../namespacepocketfft.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::dst()']]], - ['dtype_37',['Dtype',['../structmlx_1_1core_1_1_dtype.html#aec17f0a4a51729e5ac40b62f0aa765d1',1,'mlx::core::Dtype']]], - ['dtype_38',['dtype',['../classmlx_1_1core_1_1array.html#ae29e7d6fbfbea1e5e321a8d1ea3cfacd',1,'mlx::core::array']]], - ['dynamicslice_39',['DynamicSlice',['../classmlx_1_1core_1_1_dynamic_slice.html#a97f23f7d45b69219dee1a208d9a3063b',1,'mlx::core::DynamicSlice']]], - ['dynamicsliceupdate_40',['DynamicSliceUpdate',['../classmlx_1_1core_1_1_dynamic_slice_update.html#a16bbd8d756598cf620e3b3c95dd23213',1,'mlx::core::DynamicSliceUpdate']]] + ['define_5finput_5foutput_5fshape_12',['DEFINE_INPUT_OUTPUT_SHAPE',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a2965dbda1bed67128e97c3c5d864c82f',1,'mlx::core::fast::RMSNorm::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#afd0818925ffea79f4e3dda0dd8cf0366',1,'mlx::core::fast::LayerNorm::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a361cc8e0e56ff45ec98dbf81ed8eff2c',1,'mlx::core::fast::RoPE::DEFINE_INPUT_OUTPUT_SHAPE()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a64d2ce4b46b529a6a9ef068947bc623e',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_INPUT_OUTPUT_SHAPE()']]], + ['define_5fprint_13',['DEFINE_PRINT',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a8af1e90d4aa56f31ec40ad152ebd2421',1,'mlx::core::distributed::AllGather::DEFINE_PRINT()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a31bf76e24cf3836cf1fd26da30712e31',1,'mlx::core::distributed::Send::DEFINE_PRINT()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a7a0cad13da7cf8e565934318a2bc34f1',1,'mlx::core::distributed::Recv::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae6eea81b5e3789c2f6f376cc07f0a47c',1,'mlx::core::fast::RMSNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a9895733eab845e11484d86cf6ecedced',1,'mlx::core::fast::RMSNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a467fcf02b3ddf1d8b6d476b244ae3568',1,'mlx::core::fast::LayerNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a5ab3eb5402c7e8060916056eb2b7887f',1,'mlx::core::fast::LayerNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a2b06fe64fa8feca65140632087065e16',1,'mlx::core::fast::RoPE::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a6cc2092fa5b8e7585921b8e0f3ec3db7',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a4b8f1b1f633002c8ca6fa8f0ef4dd587',1,'mlx::core::fast::AffineQuantize::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a116ecf31c8672c94e5ea06c1d43e9534',1,'mlx::core::fast::CustomKernel::DEFINE_PRINT()']]], + ['degrees_14',['degrees',['../group__ops.html#ga3a70569b50e1083c5ded199d73fb960c',1,'mlx::core']]], + ['denorm_5fmin_15',['denorm_min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a6a9dbcba4dd79cad50876dda506b9eed',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['depends_16',['Depends',['../classmlx_1_1core_1_1_depends.html#a4ccb792c99f5d8d133d3fac29f7d3f62',1,'mlx::core::Depends']]], + ['depends_17',['depends',['../group__ops.html#gac4a51a68fbe1725436b026d2fbb95759',1,'mlx::core']]], + ['dequantize_18',['dequantize',['../quantized_8h.html#aecff265b63566d0d5689cfc4e5b037d2',1,'dequantize(): quantized.h'],['../group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077',1,'mlx::core::dequantize()']]], + ['detach_19',['detach',['../classmlx_1_1core_1_1array.html#a84948c29df8c957904919c8602692bd2',1,'mlx::core::array']]], + ['device_20',['Device',['../classmlx_1_1core_1_1metal_1_1_device.html#ae0db74570eb4b19d8cf19774db91bfd6',1,'mlx::core::metal::Device::Device()'],['../classmlx_1_1core_1_1metal_1_1_device.html#abf59a4addb5473f9e814e3651ba85f06',1,'mlx::core::metal::Device::Device(const Device &)=delete'],['../structmlx_1_1core_1_1_device.html#a481ccfb94d689994396bd353e966b489',1,'mlx::core::Device::Device()']]], + ['device_21',['device',['../classmlx_1_1core_1_1_primitive.html#a8ae61e3289c4134232a69295268f8261',1,'mlx::core::Primitive::device()'],['../namespacemlx_1_1core_1_1metal.html#a910797b74824e6ee576fbb533dee8b57',1,'mlx::core::metal::device(mlx::core::Device)']]], + ['device_5finfo_22',['device_info',['../namespacemlx_1_1core_1_1metal.html#a6ad19c44efabb7423f973407926ead61',1,'mlx::core::metal']]], + ['devicestream_23',['DeviceStream',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a573326bc8b48e39076850c7bf52ad0d7',1,'mlx::core::metal::DeviceStream']]], + ['diag_24',['diag',['../group__ops.html#ga11af511875640e1fa88e0ca87e199344',1,'mlx::core']]], + ['diagonal_25',['diagonal',['../group__ops.html#ga9236b085a88ead3128ed8079d009cac6',1,'mlx::core']]], + ['disable_5fcompile_26',['disable_compile',['../namespacemlx_1_1core.html#a5f5fea955057bb3842b271b037909e66',1,'mlx::core']]], + ['dispatch_5fthreadgroups_27',['dispatch_threadgroups',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a85796b2bf41dbf347ae0978d4660600d',1,'mlx::core::metal::CommandEncoder::dispatch_threadgroups()'],['../structmlx_1_1core_1_1_command_encoder.html#a85796b2bf41dbf347ae0978d4660600d',1,'mlx::core::CommandEncoder::dispatch_threadgroups()']]], + ['dispatch_5fthreads_28',['dispatch_threads',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a0a8501b940e5a347475fa4bc38fb4c05',1,'mlx::core::metal::CommandEncoder::dispatch_threads()'],['../structmlx_1_1core_1_1_command_encoder.html#a0a8501b940e5a347475fa4bc38fb4c05',1,'mlx::core::CommandEncoder::dispatch_threads()']]], + ['distprimitive_29',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8c54166951522c2a52ef39fce8c87f8f',1,'mlx::core::distributed::DistPrimitive']]], + ['divide_30',['Divide',['../classmlx_1_1core_1_1_divide.html#a62fc71e8998be65ff18285dbbd21eedb',1,'mlx::core::Divide']]], + ['divide_31',['divide',['../namespacemetal.html#a2aea493fc1a874970b77ed0031e965df',1,'metal::divide()'],['../namespacemetal_1_1fast.html#ae70bc2185e4649369cf7b15f5e1d48be',1,'metal::fast::divide()'],['../namespacemetal_1_1precise.html#aec0982cdb96a08b61f51129150d82e9d',1,'metal::precise::divide()'],['../group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f',1,'mlx::core::divide()']]], + ['divmod_32',['DivMod',['../classmlx_1_1core_1_1_div_mod.html#a859e3b6149cdceab1c7ccfd2246fb826',1,'mlx::core::DivMod']]], + ['divmod_33',['divmod',['../group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894',1,'mlx::core']]], + ['dst_34',['dst',['../namespacepocketfft_1_1detail.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::detail::dst()'],['../namespacepocketfft.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::dst()']]], + ['dtype_35',['Dtype',['../structmlx_1_1core_1_1_dtype.html#aec17f0a4a51729e5ac40b62f0aa765d1',1,'mlx::core::Dtype']]], + ['dtype_36',['dtype',['../classmlx_1_1core_1_1array.html#ae29e7d6fbfbea1e5e321a8d1ea3cfacd',1,'mlx::core::array']]], + ['dynamicslice_37',['DynamicSlice',['../classmlx_1_1core_1_1_dynamic_slice.html#a97f23f7d45b69219dee1a208d9a3063b',1,'mlx::core::DynamicSlice']]], + ['dynamicsliceupdate_38',['DynamicSliceUpdate',['../classmlx_1_1core_1_1_dynamic_slice_update.html#a16bbd8d756598cf620e3b3c95dd23213',1,'mlx::core::DynamicSliceUpdate']]] ]; diff --git a/docs/build/html/search/functions_5.js b/docs/build/html/search/functions_5.js index 724042447..4173fa699 100644 --- a/docs/build/html/search/functions_5.js +++ b/docs/build/html/search/functions_5.js @@ -15,36 +15,38 @@ var searchData= ['elems_12',['elems',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a865ece5ad0b9a56937b6d77a18b5a1dc',1,'mlx::steel::MMATile::elems()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ae21bb7cce701290de84c6015e064d8a1',1,'mlx::steel::MMATile::elems() const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a865ece5ad0b9a56937b6d77a18b5a1dc',1,'mlx::steel::MMATile::elems()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ae21bb7cce701290de84c6015e064d8a1',1,'mlx::steel::MMATile::elems() const']]], ['empty_13',['empty',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#a1269e5da40c3f5145c895cee3641879a',1,'pocketfft::detail::threading::concurrent_queue']]], ['enable_5fcompile_14',['enable_compile',['../namespacemlx_1_1core.html#a1983a2466bff3bae4d23cf34bd0946c9',1,'mlx::core']]], - ['end_15',['end',['../classmlx_1_1core_1_1array.html#a5daf64552fb450825c9b382f3a5fa2d4',1,'mlx::core::array']]], - ['end_5fencoding_16',['end_encoding',['../classmlx_1_1core_1_1metal_1_1_device.html#a60689f97347811b27e8c5ca23e0372bf',1,'mlx::core::metal::Device']]], - ['enqueue_17',['enqueue',['../class_thread_pool.html#a375fa2d63197282277be640b54e8a196',1,'ThreadPool::enqueue()'],['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4918720319cf224a1b4208568964c286',1,'mlx::core::scheduler::StreamThread::enqueue()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a03809c783bd1866362dc7cb9118abbcc',1,'mlx::core::scheduler::Scheduler::enqueue()'],['../namespacemlx_1_1core_1_1scheduler.html#aa2d4eacf5d5cbc778a51aafd4fd8e4d7',1,'mlx::core::scheduler::enqueue()']]], - ['epsilon_18',['epsilon',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a96c4197e3076f0aa9065370b8ece49ca',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['equal_19',['Equal',['../classmlx_1_1core_1_1_equal.html#a4af81cf2dd071db5bbf8ce1df95fdf36',1,'mlx::core::Equal']]], - ['equal_20',['equal',['../group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde',1,'mlx::core']]], - ['erase_21',['erase',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#ae136ad270522210c85c13cacf5165238',1,'mlx::core::metal::ResidencySet']]], - ['erf_22',['Erf',['../classmlx_1_1core_1_1_erf.html#a702f76f848928d8d7d3d0881ac6e4c82',1,'mlx::core::Erf']]], - ['erf_23',['erf',['../erf_8h.html#a6ce199ee56105c67adbf8c48c019a8b2',1,'erf(): erf.h'],['../group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0',1,'mlx::core::erf()']]], - ['erfinv_24',['ErfInv',['../classmlx_1_1core_1_1_erf_inv.html#a5d0279247b67da4592311559f04e1478',1,'mlx::core::ErfInv']]], - ['erfinv_25',['erfinv',['../erf_8h.html#a1846e0d683c7aff826bb32addcc3b885',1,'erfinv(): erf.h'],['../group__ops.html#ga76fb9062c64264e34d2e07013390557c',1,'mlx::core::erfinv()']]], - ['eval_26',['eval',['../classmlx_1_1core_1_1array.html#a2820c45188071a22175e9fa42e10a49a',1,'mlx::core::array::eval()'],['../namespacemlx_1_1core.html#a7d6e097d8effed52f4713672e471f299',1,'mlx::core::eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#adb14f689c9f75f7901edb196c2bfb971',1,'mlx::core::eval(Arrays &&... outputs)']]], - ['eval_5fcpu_27',['eval_cpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#acdc1965ad64ee9ee6328fe150a97902e',1,'mlx::core::distributed::AllReduce::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ab721fe0072fffbddbc3c4334dd033ba5',1,'mlx::core::distributed::AllGather::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#af2620837bfc1b97217d006ed6e374051',1,'mlx::core::distributed::Send::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a3be84b08122a939edd6062d26261358a',1,'mlx::core::distributed::Recv::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a7da6e0cfd630958d9633b2e2bd97a54f',1,'mlx::core::fast::RMSNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#adfc1d52bc266466ab29ee45fd8fab439',1,'mlx::core::fast::RMSNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a5d7a4c1c9ee84e327d1c371733108c05',1,'mlx::core::fast::LayerNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a0d8c4c6e7462befc38f7e08244fa1c2b',1,'mlx::core::fast::LayerNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a05a7d595c6b9dadf7ddfd6e3fd402f0e',1,'mlx::core::fast::RoPE::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ae20851e002f7fcb6d4f97817596f6328',1,'mlx::core::fast::ScaledDotProductAttention::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a3b5d628628d245b38911118d4a0ff9fd',1,'mlx::core::fast::AffineQuantize::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a4ad1b7a9919753c759093f3e21a15bad',1,'mlx::core::fast::CustomKernel::eval_cpu()'],['../classmlx_1_1core_1_1_primitive.html#a1596dc50b910538eae14878e98f07575',1,'mlx::core::Primitive::eval_cpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a7e8f6f5d6ae0a33f6abc0f5a46e0b132',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#aa0ed6e32c36200a3ff9bc592c9b300db',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0d3e697496ef8e842d21195cb3c14e60',1,'mlx::core::Abs::eval_cpu()'],['../classmlx_1_1core_1_1_add.html#a5bacfc51dfa2a5a931bad2dd7bdc7a5f',1,'mlx::core::Add::eval_cpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a15694e3bf2ed5c193237b2b9ca00867c',1,'mlx::core::AddMM::eval_cpu()'],['../classmlx_1_1core_1_1_arange.html#aba44432491cbd599bf72712f5f4267a1',1,'mlx::core::Arange::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a58dcba9e706cb12bab062bb7fa5fa006',1,'mlx::core::ArcCos::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#a0f6d989bcbbc38f15ef17a136879a9c9',1,'mlx::core::ArcCosh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sin.html#ab3542492c14021329788de8f2a9be1e4',1,'mlx::core::ArcSin::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a52574b24d8d16839c58673f51f8ac066',1,'mlx::core::ArcSinh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a1211bc31241227528f04435239ddb9a3',1,'mlx::core::ArcTan::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a13094e6b702769928ca0da468f5ce45c',1,'mlx::core::ArcTan2::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a5af9224e1f1ffec412b0baa0af7e1ecd',1,'mlx::core::ArcTanh::eval_cpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a896f75c5325798ac3f9093f6a4581828',1,'mlx::core::ArgPartition::eval_cpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#ad8d48725623ede1ff654fa13eccf2287',1,'mlx::core::ArgReduce::eval_cpu()'],['../classmlx_1_1core_1_1_arg_sort.html#a022079683774bfeb531b3a002cff16fa',1,'mlx::core::ArgSort::eval_cpu()'],['../classmlx_1_1core_1_1_as_type.html#aa89dbf4d73b00c6a44cffd04d5bb228d',1,'mlx::core::AsType::eval_cpu()'],['../classmlx_1_1core_1_1_as_strided.html#acdd4705e4503ff0b124215c4676b4193',1,'mlx::core::AsStrided::eval_cpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a2194bf585213bda1b2966aa02d2fe283',1,'mlx::core::BitwiseBinary::eval_cpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aa85da478cdc6d4a97be06e5d4abee1f2',1,'mlx::core::BlockMaskedMM::eval_cpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#a62352074a480df0e1f879b0bae425730',1,'mlx::core::GatherMM::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a6423095cd28b2f90893c03166257a568',1,'mlx::core::BroadcastAxes::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast.html#a53d48d9778e2d4c24a124cd767900780',1,'mlx::core::Broadcast::eval_cpu()'],['../classmlx_1_1core_1_1_ceil.html#a9791801fff3f8b79944e15ac2a45a035',1,'mlx::core::Ceil::eval_cpu()'],['../classmlx_1_1core_1_1_compiled.html#ac45b1d0fedd85feefbff7ce7e168b151',1,'mlx::core::Compiled::eval_cpu()'],['../classmlx_1_1core_1_1_concatenate.html#a609e76bede7fc5581ec84ddcb727a258',1,'mlx::core::Concatenate::eval_cpu()'],['../classmlx_1_1core_1_1_conjugate.html#ae39643e2178f442ffba05139f8609d61',1,'mlx::core::Conjugate::eval_cpu()'],['../classmlx_1_1core_1_1_contiguous.html#a742de24e6c0310cd85a606dec0cd8336',1,'mlx::core::Contiguous::eval_cpu()'],['../classmlx_1_1core_1_1_convolution.html#ac74256068da01730629109fa4fa8432b',1,'mlx::core::Convolution::eval_cpu()'],['../classmlx_1_1core_1_1_copy.html#af4a0ebec423e84ffe8083a5e9ed0d70c',1,'mlx::core::Copy::eval_cpu()'],['../classmlx_1_1core_1_1_cos.html#a061fc446268fe56237ae6b20ccf78152',1,'mlx::core::Cos::eval_cpu()'],['../classmlx_1_1core_1_1_cosh.html#ae8702df7e8f0e20cbeccb2a548961d3d',1,'mlx::core::Cosh::eval_cpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#adba1c40c77a2138df6b5f75483f62184',1,'mlx::core::CustomTransforms::eval_cpu()'],['../classmlx_1_1core_1_1_depends.html#a0c7ea6db97337591fa53c6e6bde41e5e',1,'mlx::core::Depends::eval_cpu()'],['../classmlx_1_1core_1_1_divide.html#a823443c2a8e8b81bbcaeee6ddbcdbf49',1,'mlx::core::Divide::eval_cpu()'],['../classmlx_1_1core_1_1_div_mod.html#ae350b7b93ad128e3133ee14f247193b3',1,'mlx::core::DivMod::eval_cpu()'],['../classmlx_1_1core_1_1_select.html#aa51aa36e0adbd69e0d23d7c7adf88de2',1,'mlx::core::Select::eval_cpu()'],['../classmlx_1_1core_1_1_remainder.html#ac6c6c86a0bf02e6e529eb87f6e617ccc',1,'mlx::core::Remainder::eval_cpu()'],['../classmlx_1_1core_1_1_equal.html#aabb8aa61fa581defddcdca1274b1b454',1,'mlx::core::Equal::eval_cpu()'],['../classmlx_1_1core_1_1_erf.html#a84ea16e43d5b7f83bbc2d5ece78a3fb6',1,'mlx::core::Erf::eval_cpu()'],['../classmlx_1_1core_1_1_erf_inv.html#af579627402af3249565134884701d39e',1,'mlx::core::ErfInv::eval_cpu()'],['../classmlx_1_1core_1_1_exp.html#a47934c5a5023bc7ae7ae89bff45ebb2c',1,'mlx::core::Exp::eval_cpu()'],['../classmlx_1_1core_1_1_expm1.html#ab9c8b7aa50fe4592d55f8957baac647a',1,'mlx::core::Expm1::eval_cpu()'],['../classmlx_1_1core_1_1_expand_dims.html#a34058a87582a6ab2e5d82a75bc713030',1,'mlx::core::ExpandDims::eval_cpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a6bc262a0c2b5d4fe655e3e2e0ff28635',1,'mlx::core::FFT::eval_cpu()'],['../classmlx_1_1core_1_1_flatten.html#a72ade7d22386b349712f6c7c1f619842',1,'mlx::core::Flatten::eval_cpu()'],['../classmlx_1_1core_1_1_floor.html#a1a7dc5f571b7b73e7ef3cbdc1dd1fcf7',1,'mlx::core::Floor::eval_cpu()'],['../classmlx_1_1core_1_1_full.html#a3dccd3756599d7fd018b2af0093b082c',1,'mlx::core::Full::eval_cpu()'],['../classmlx_1_1core_1_1_gather.html#a9ed5587f0d04b59a2b9186c0aac21290',1,'mlx::core::Gather::eval_cpu()'],['../classmlx_1_1core_1_1_greater.html#abe1c03f311d0e0b610f3392a6566f2ae',1,'mlx::core::Greater::eval_cpu()'],['../classmlx_1_1core_1_1_greater_equal.html#a15469125b9bea89b64bfeac01590c075',1,'mlx::core::GreaterEqual::eval_cpu()'],['../classmlx_1_1core_1_1_hadamard.html#ab27d6a9df42b3aab41ace3073a4c880d',1,'mlx::core::Hadamard::eval_cpu()'],['../classmlx_1_1core_1_1_imag.html#a17d1f1f9f8528668fcdf39b636720829',1,'mlx::core::Imag::eval_cpu()'],['../classmlx_1_1core_1_1_less.html#a32624124ffece066f496b3299056bcef',1,'mlx::core::Less::eval_cpu()'],['../classmlx_1_1core_1_1_less_equal.html#a55d1352b0e97841a92503bc57c19ed16',1,'mlx::core::LessEqual::eval_cpu()'],['../classmlx_1_1core_1_1_load.html#ada026ac30566f3109d8182e35d307c0a',1,'mlx::core::Load::eval_cpu()'],['../classmlx_1_1core_1_1_log.html#aadc7bb4cb24f3ecbbb9ed54a699ab74f',1,'mlx::core::Log::eval_cpu()'],['../classmlx_1_1core_1_1_log1p.html#a8192e5438de99c4cda056987935cba23',1,'mlx::core::Log1p::eval_cpu()'],['../classmlx_1_1core_1_1_logical_not.html#acf3f7b3b20ca69533536e0e0a05725b3',1,'mlx::core::LogicalNot::eval_cpu()'],['../classmlx_1_1core_1_1_logical_and.html#adbe1c1785af1a8b827289d22b0d170b3',1,'mlx::core::LogicalAnd::eval_cpu()'],['../classmlx_1_1core_1_1_logical_or.html#a13cd4cbf26589287e85aeaaca42d7f62',1,'mlx::core::LogicalOr::eval_cpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#abef17fb590b1a8d356f2a580e45d41f0',1,'mlx::core::LogAddExp::eval_cpu()'],['../classmlx_1_1core_1_1_matmul.html#a357a7f57a2a220a91977f810a69413fc',1,'mlx::core::Matmul::eval_cpu()'],['../classmlx_1_1core_1_1_maximum.html#a62b38fbe5f96db58c2b60165ac4eadcf',1,'mlx::core::Maximum::eval_cpu()'],['../classmlx_1_1core_1_1_minimum.html#a6b93f493ee87089943a8085fe59dfc6e',1,'mlx::core::Minimum::eval_cpu()'],['../classmlx_1_1core_1_1_multiply.html#a624fce06c047cdc4dfdbdcaaddb25f34',1,'mlx::core::Multiply::eval_cpu()'],['../classmlx_1_1core_1_1_negative.html#af43553dc418c8ebe75fa9cdcba103c3b',1,'mlx::core::Negative::eval_cpu()'],['../classmlx_1_1core_1_1_not_equal.html#a8f95f8b5873850b875b1641df8196047',1,'mlx::core::NotEqual::eval_cpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#acc328321cf5300874ee884367cbede3f',1,'mlx::core::NumberOfElements::eval_cpu()'],['../classmlx_1_1core_1_1_pad.html#aaf82dd163cd536fbf97304f8b29080cb',1,'mlx::core::Pad::eval_cpu()'],['../classmlx_1_1core_1_1_partition.html#a784596ab567f9f3cb4fe1a69466523d8',1,'mlx::core::Partition::eval_cpu()'],['../classmlx_1_1core_1_1_power.html#a6783da16fb6ff393aaa57737f1973206',1,'mlx::core::Power::eval_cpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ab3dfa73b74d8f4f2e9ab4f0eb016b0e3',1,'mlx::core::QuantizedMatmul::eval_cpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a89aae98bfbdd6563df44ef7d70f0bf8c',1,'mlx::core::GatherQMM::eval_cpu()'],['../classmlx_1_1core_1_1_random_bits.html#a5752d051cd16cf5f8d4754c0a656f0d2',1,'mlx::core::RandomBits::eval_cpu()'],['../classmlx_1_1core_1_1_real.html#a365d046caac91b521f0f5a5518037934',1,'mlx::core::Real::eval_cpu()'],['../classmlx_1_1core_1_1_reshape.html#a658de2c5f710991b48e14b2bd19b229f',1,'mlx::core::Reshape::eval_cpu()'],['../classmlx_1_1core_1_1_reduce.html#aeb8a58b560c0a09ae3a695df7829acfa',1,'mlx::core::Reduce::eval_cpu()'],['../classmlx_1_1core_1_1_round.html#ad066b0944b437f64ab546025efa00007',1,'mlx::core::Round::eval_cpu()'],['../classmlx_1_1core_1_1_scan.html#a15676d9fd066e935782a923fba3e940b',1,'mlx::core::Scan::eval_cpu()'],['../classmlx_1_1core_1_1_scatter.html#a7623f590f8b77167b5ebb4f14bc9dc97',1,'mlx::core::Scatter::eval_cpu()'],['../classmlx_1_1core_1_1_sigmoid.html#aa930ce05734cca529ebcb8d0ca8e1255',1,'mlx::core::Sigmoid::eval_cpu()'],['../classmlx_1_1core_1_1_sign.html#a7498ec993b66879be30c5d9762c45a97',1,'mlx::core::Sign::eval_cpu()'],['../classmlx_1_1core_1_1_sin.html#ab34f9cebc2aed55a0b6ab4c991f02eb5',1,'mlx::core::Sin::eval_cpu()'],['../classmlx_1_1core_1_1_sinh.html#ab6d5f6f40d177f6435f6a51c71b939dd',1,'mlx::core::Sinh::eval_cpu()'],['../classmlx_1_1core_1_1_slice.html#a4b13503f5b2f5c6a90d394b020f9b3f2',1,'mlx::core::Slice::eval_cpu()'],['../classmlx_1_1core_1_1_slice_update.html#ad82ca0e3ab88a0e086431050deea831b',1,'mlx::core::SliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a4e8c22c24a587ea0648ce89f461ed1ee',1,'mlx::core::DynamicSlice::eval_cpu()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a379185914db0326a5d4839839fe4fc83',1,'mlx::core::DynamicSliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_softmax.html#ac9ebc2eab1683b682e689ed8f4622b79',1,'mlx::core::Softmax::eval_cpu()'],['../classmlx_1_1core_1_1_sort.html#a459769a0241b2620e55bedaba19827cd',1,'mlx::core::Sort::eval_cpu()'],['../classmlx_1_1core_1_1_split.html#aff2889cb9074f0fda53edf8fa40b1fd4',1,'mlx::core::Split::eval_cpu()'],['../classmlx_1_1core_1_1_square.html#a1f4d327a705950616da63b83c2829e59',1,'mlx::core::Square::eval_cpu()'],['../classmlx_1_1core_1_1_sqrt.html#a5a64ecc4eef1e30a2963435dca7cefd5',1,'mlx::core::Sqrt::eval_cpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a56207714d374b08f60e4d9cdbc7340b2',1,'mlx::core::StopGradient::eval_cpu()'],['../classmlx_1_1core_1_1_subtract.html#a47574258b6c95f8ad260c114d6d36a12',1,'mlx::core::Subtract::eval_cpu()'],['../classmlx_1_1core_1_1_squeeze.html#a9bcb7476041020f59ef816196ddb81cb',1,'mlx::core::Squeeze::eval_cpu()'],['../classmlx_1_1core_1_1_tan.html#a9c9a731158fa60eef30067fe0da9f3e9',1,'mlx::core::Tan::eval_cpu()'],['../classmlx_1_1core_1_1_tanh.html#af7ed4345f622da069e5b0284067923f5',1,'mlx::core::Tanh::eval_cpu()'],['../classmlx_1_1core_1_1_unflatten.html#a507c22306b7afcdd5970cfaa32188f0a',1,'mlx::core::Unflatten::eval_cpu()'],['../classmlx_1_1core_1_1_view.html#a0ad6deb11914a242f10e8039fcb02497',1,'mlx::core::View::eval_cpu()'],['../classmlx_1_1core_1_1_transpose.html#a1fbcfcca43f9ec06c63a3c14708c30f8',1,'mlx::core::Transpose::eval_cpu()'],['../classmlx_1_1core_1_1_q_r_f.html#a48493887395d65a27f04de1804d277d2',1,'mlx::core::QRF::eval_cpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a637f5c39fa8b10722c04a066f6c1ada6',1,'mlx::core::SVD::eval_cpu()'],['../classmlx_1_1core_1_1_inverse.html#aeb1d8dc9bc4052a616023f65b3c7bb81',1,'mlx::core::Inverse::eval_cpu()'],['../classmlx_1_1core_1_1_cholesky.html#a4bdec36c1cc99aadf9a4a39d4c57bea5',1,'mlx::core::Cholesky::eval_cpu()'],['../classmlx_1_1core_1_1_eigh.html#a894b32e17229394f6a43b4a0655fd8be',1,'mlx::core::Eigh::eval_cpu()']]], - ['eval_5fgpu_28',['eval_gpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a52df7155f56b8450581b2fd2747cad20',1,'mlx::core::distributed::AllReduce::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a4251ce0f2db2045226b66210b828af7a',1,'mlx::core::distributed::AllGather::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a0c8dbd2a912be91be04ec701e29fba3d',1,'mlx::core::distributed::Send::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a932e39624bc3d234a7489c3decc4749e',1,'mlx::core::distributed::Recv::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae7955e8d43c097eecae264e804b4d8ca',1,'mlx::core::fast::RMSNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a48efb8fa84c4ba6cc9fb560ebbe01560',1,'mlx::core::fast::RMSNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a77abda7f47bffa2c037a5d60cccc1528',1,'mlx::core::fast::LayerNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a954a003a4a27c8c4c60a5a14142a9cc3',1,'mlx::core::fast::LayerNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a913b6b00fc518b25ac3947e4e15790f2',1,'mlx::core::fast::RoPE::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a505f38ba93a3499895f5312e0112e73d',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ad51666e69f670e286293aff96eb435a9',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, array &out)'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a63812b2abaf26ad7e7fa4c9e82db1628',1,'mlx::core::fast::AffineQuantize::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a2ed2a16b23053f8195068386a99fd6db',1,'mlx::core::fast::CustomKernel::eval_gpu()'],['../classmlx_1_1core_1_1_primitive.html#ad217376dcf5eff691d731566faec2ba2',1,'mlx::core::Primitive::eval_gpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a6b7f80abaf038d53ec6ffbb0dfac6adb',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#a971fe9ad47f6569118879ce1d0f41447',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0a976e636dd8505b473fbdddf949f514',1,'mlx::core::Abs::eval_gpu()'],['../classmlx_1_1core_1_1_add.html#aa0aacbc1e26b95a2f040f62aa4f69c3d',1,'mlx::core::Add::eval_gpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a5f933be14baebc32a0be0f9a69148aa9',1,'mlx::core::AddMM::eval_gpu()'],['../classmlx_1_1core_1_1_arange.html#a7a2e9787c6c3a78b4a6df91206974031',1,'mlx::core::Arange::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a46f72d4af89b0a0f5f203783fb44589c',1,'mlx::core::ArcCos::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#aa6a2587485a0e015ac2d5211d7d045fc',1,'mlx::core::ArcCosh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sin.html#a7fa4ae7a85bc8bed97ea258ae30762f3',1,'mlx::core::ArcSin::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79f648a86de4c10386a1ce3b5e38e8ac',1,'mlx::core::ArcSinh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a77866feb27028865d844070447c9a254',1,'mlx::core::ArcTan::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a76d3f0c29e0ff4642b8d39dac90d3f50',1,'mlx::core::ArcTan2::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a10566b9d3b2c7d090895b46d9040bc1d',1,'mlx::core::ArcTanh::eval_gpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a9a60995eaf85f63c877e86b23cbc15fc',1,'mlx::core::ArgPartition::eval_gpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#aafa982ce2abc0cd9e81e43aa2c823d29',1,'mlx::core::ArgReduce::eval_gpu()'],['../classmlx_1_1core_1_1_arg_sort.html#abc2d730850ec4ee8d7968b7417911709',1,'mlx::core::ArgSort::eval_gpu()'],['../classmlx_1_1core_1_1_as_type.html#a5b111b9d74c60d27b4a7ebaa49f96e0b',1,'mlx::core::AsType::eval_gpu()'],['../classmlx_1_1core_1_1_as_strided.html#ab6771a208323994927ca162ba7bb10ed',1,'mlx::core::AsStrided::eval_gpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#ac831a29fc46701b00bbe63ee33832afd',1,'mlx::core::BitwiseBinary::eval_gpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ab372b6df4de00a33795a052a23bb1df9',1,'mlx::core::BlockMaskedMM::eval_gpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#ad754c35f460a055cc383ad93a5f72da1',1,'mlx::core::GatherMM::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a56d16e75a0df867d2f1ba4e5198f15cb',1,'mlx::core::BroadcastAxes::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast.html#ab9bd9dbcedcefc9b29c84911b5ce69fe',1,'mlx::core::Broadcast::eval_gpu()'],['../classmlx_1_1core_1_1_ceil.html#abe178e0058e44b6618be414215e96887',1,'mlx::core::Ceil::eval_gpu()'],['../classmlx_1_1core_1_1_compiled.html#aa3d5ff0f2b3554ad48fbbf2a0f3336d5',1,'mlx::core::Compiled::eval_gpu()'],['../classmlx_1_1core_1_1_concatenate.html#a309a1c50e97f9925866433ee2841c474',1,'mlx::core::Concatenate::eval_gpu()'],['../classmlx_1_1core_1_1_conjugate.html#aff0a802166e3724db88ab5d3feb2d3de',1,'mlx::core::Conjugate::eval_gpu()'],['../classmlx_1_1core_1_1_contiguous.html#a519cd16fd0c55b371ea7625fbb37c70f',1,'mlx::core::Contiguous::eval_gpu()'],['../classmlx_1_1core_1_1_convolution.html#a30b64109eeb1778f002b99447dff9dd2',1,'mlx::core::Convolution::eval_gpu()'],['../classmlx_1_1core_1_1_copy.html#a1eda7b2ea771a168f67421f0d384b3a1',1,'mlx::core::Copy::eval_gpu()'],['../classmlx_1_1core_1_1_cos.html#a5ef41aafad595f6cdd8c535e36e12060',1,'mlx::core::Cos::eval_gpu()'],['../classmlx_1_1core_1_1_cosh.html#a23f71b43792934c3ec0ebe9b74f32559',1,'mlx::core::Cosh::eval_gpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#a7b3538681acbb20af3ed37b0877f6667',1,'mlx::core::CustomTransforms::eval_gpu()'],['../classmlx_1_1core_1_1_depends.html#ae5057f65e69490ad0add8eeda2b75e28',1,'mlx::core::Depends::eval_gpu()'],['../classmlx_1_1core_1_1_divide.html#abffda0ce37221ddc28dc9eea794f6bc7',1,'mlx::core::Divide::eval_gpu()'],['../classmlx_1_1core_1_1_div_mod.html#a003117c9ecf3c06a27248f72a76348dc',1,'mlx::core::DivMod::eval_gpu()'],['../classmlx_1_1core_1_1_select.html#a2a82b6cba4c386b2b87f225a4b08ea9b',1,'mlx::core::Select::eval_gpu()'],['../classmlx_1_1core_1_1_remainder.html#a7919ea9b84e42522d51bf0d5a396e161',1,'mlx::core::Remainder::eval_gpu()'],['../classmlx_1_1core_1_1_equal.html#ac3757001fec42ceb5ece2954df42161c',1,'mlx::core::Equal::eval_gpu()'],['../classmlx_1_1core_1_1_erf.html#ad8551be664d767dccc3c0d8cc1eca008',1,'mlx::core::Erf::eval_gpu()'],['../classmlx_1_1core_1_1_erf_inv.html#a4a2413d0634db1f3dae1806ddfa632db',1,'mlx::core::ErfInv::eval_gpu()'],['../classmlx_1_1core_1_1_exp.html#a7d63695a97a14760fd33b5d4e6590822',1,'mlx::core::Exp::eval_gpu()'],['../classmlx_1_1core_1_1_expm1.html#a82930071f4b77d883b300f77966aff5f',1,'mlx::core::Expm1::eval_gpu()'],['../classmlx_1_1core_1_1_expand_dims.html#ad350ede3abecc55371ddeb89fbba2b90',1,'mlx::core::ExpandDims::eval_gpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a1c21b26d1e9ad7c4da78ae845721b2dd',1,'mlx::core::FFT::eval_gpu()'],['../classmlx_1_1core_1_1_flatten.html#acb2219cc122d218b273af2cb9a882e7f',1,'mlx::core::Flatten::eval_gpu()'],['../classmlx_1_1core_1_1_floor.html#aaa29c83538099eb8f951c95a41f2eb65',1,'mlx::core::Floor::eval_gpu()'],['../classmlx_1_1core_1_1_full.html#aa54f99bb4cba12a551392dea56003872',1,'mlx::core::Full::eval_gpu()'],['../classmlx_1_1core_1_1_gather.html#aec48ee529cb2449915a7b27a3c4361e8',1,'mlx::core::Gather::eval_gpu()'],['../classmlx_1_1core_1_1_greater.html#ae8957cccf4c924d941f57a1bb751c878',1,'mlx::core::Greater::eval_gpu()'],['../classmlx_1_1core_1_1_greater_equal.html#ac246263b4548126c3d4ab7e392575d24',1,'mlx::core::GreaterEqual::eval_gpu()'],['../classmlx_1_1core_1_1_hadamard.html#a2470feb690f5463138490763c38b5733',1,'mlx::core::Hadamard::eval_gpu()'],['../classmlx_1_1core_1_1_imag.html#a247a4d059b0a99678c6be8c15e42c1e6',1,'mlx::core::Imag::eval_gpu()'],['../classmlx_1_1core_1_1_less.html#a353335ce06ddbe8498d86d129c835917',1,'mlx::core::Less::eval_gpu()'],['../classmlx_1_1core_1_1_less_equal.html#acf035a82b11e6f63742143ea540fedac',1,'mlx::core::LessEqual::eval_gpu()'],['../classmlx_1_1core_1_1_load.html#a06933e887ea94a4d01d81195c5e07a3d',1,'mlx::core::Load::eval_gpu()'],['../classmlx_1_1core_1_1_log.html#aaaa49e9455f3a197bc319646b5ca6390',1,'mlx::core::Log::eval_gpu()'],['../classmlx_1_1core_1_1_log1p.html#a1b97decae7338d46874e736c95fa7431',1,'mlx::core::Log1p::eval_gpu()'],['../classmlx_1_1core_1_1_logical_not.html#a1d0d2bc93f935eca6c85ef7bf67f2d6a',1,'mlx::core::LogicalNot::eval_gpu()'],['../classmlx_1_1core_1_1_logical_and.html#a132b2eedaa3978de5a5350da3c2ca40f',1,'mlx::core::LogicalAnd::eval_gpu()'],['../classmlx_1_1core_1_1_logical_or.html#a3be1da328f0f8620de2e4fc1d22a077a',1,'mlx::core::LogicalOr::eval_gpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#acace355b62ec00df649f9f99e8f2eb7a',1,'mlx::core::LogAddExp::eval_gpu()'],['../classmlx_1_1core_1_1_matmul.html#a8707a4e9b75c769e8f1dbca15c6a1ae7',1,'mlx::core::Matmul::eval_gpu()'],['../classmlx_1_1core_1_1_maximum.html#ade0f721b10a6b3a12bdadd34c48f72a7',1,'mlx::core::Maximum::eval_gpu()'],['../classmlx_1_1core_1_1_minimum.html#aadc68afa0afbe2103f19d161f5e0a2ba',1,'mlx::core::Minimum::eval_gpu()'],['../classmlx_1_1core_1_1_multiply.html#a634fcb4e981d8d3f4d94252caf25bee0',1,'mlx::core::Multiply::eval_gpu()'],['../classmlx_1_1core_1_1_negative.html#a97f1b316eace0c6d9e576d766940c75b',1,'mlx::core::Negative::eval_gpu()'],['../classmlx_1_1core_1_1_not_equal.html#a61179747e34e203150e9c660dfddb5f2',1,'mlx::core::NotEqual::eval_gpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#a2c98c42915fb2bfe12f5c99ea553eff5',1,'mlx::core::NumberOfElements::eval_gpu()'],['../classmlx_1_1core_1_1_pad.html#aefd4d3a5bd8b6b35b266c9e558ada153',1,'mlx::core::Pad::eval_gpu()'],['../classmlx_1_1core_1_1_partition.html#a8eca1be21ae9ccfda46e6f3e85f506ef',1,'mlx::core::Partition::eval_gpu()'],['../classmlx_1_1core_1_1_power.html#a80577d4c0853c24027777c90a1ec7e11',1,'mlx::core::Power::eval_gpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a2812ad007d695ed1aaf9cf706fb9c4b3',1,'mlx::core::QuantizedMatmul::eval_gpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a86eb048afc95646b2e96ec5493e3d887',1,'mlx::core::GatherQMM::eval_gpu()'],['../classmlx_1_1core_1_1_random_bits.html#a578756866665358577418e4cdd94aa3a',1,'mlx::core::RandomBits::eval_gpu()'],['../classmlx_1_1core_1_1_real.html#a1e209e88a43bdd1eea43ad0b03f9a7f2',1,'mlx::core::Real::eval_gpu()'],['../classmlx_1_1core_1_1_reshape.html#aa1e85f28471875750c47351520b56059',1,'mlx::core::Reshape::eval_gpu()'],['../classmlx_1_1core_1_1_reduce.html#ae9caaf42edadfe73ea208d98f526890f',1,'mlx::core::Reduce::eval_gpu()'],['../classmlx_1_1core_1_1_round.html#af7fe5ff8f3db166c203b4be4b07f13ec',1,'mlx::core::Round::eval_gpu()'],['../classmlx_1_1core_1_1_scan.html#aef22c6fc2b2cb2a907cd8965c7413dde',1,'mlx::core::Scan::eval_gpu()'],['../classmlx_1_1core_1_1_scatter.html#ab304345db3d8cfeea15e27461ae2e678',1,'mlx::core::Scatter::eval_gpu()'],['../classmlx_1_1core_1_1_sigmoid.html#a7a6bd0222d51d7f25f2719a91ccdfeca',1,'mlx::core::Sigmoid::eval_gpu()'],['../classmlx_1_1core_1_1_sign.html#afa2b48b99a194106006b44af69ffda8b',1,'mlx::core::Sign::eval_gpu()'],['../classmlx_1_1core_1_1_sin.html#a6b59f1156cf8bdad8d45acd1d825cb5e',1,'mlx::core::Sin::eval_gpu()'],['../classmlx_1_1core_1_1_sinh.html#a5a1af2399f166d5b228b5e83a1837c75',1,'mlx::core::Sinh::eval_gpu()'],['../classmlx_1_1core_1_1_slice.html#aa53c21ff06a7c659e889af6b97d10a4a',1,'mlx::core::Slice::eval_gpu()'],['../classmlx_1_1core_1_1_slice_update.html#aac1a1d122e5697be057d63552141032b',1,'mlx::core::SliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_dynamic_slice.html#ab0a2e31c03f02a4f25700e240cf18e3e',1,'mlx::core::DynamicSlice::eval_gpu()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a249dab28690c45203c3995698de0cab7',1,'mlx::core::DynamicSliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_softmax.html#a35dac69ddcc7e2ec0e1a76fe93db85af',1,'mlx::core::Softmax::eval_gpu()'],['../classmlx_1_1core_1_1_sort.html#a4141c48f0e8670c728663f3722675382',1,'mlx::core::Sort::eval_gpu()'],['../classmlx_1_1core_1_1_split.html#a78ddda89c4daee73c74cfbc1e44656df',1,'mlx::core::Split::eval_gpu()'],['../classmlx_1_1core_1_1_square.html#a0ea2a78a5bb52daa4103263bf2f98045',1,'mlx::core::Square::eval_gpu()'],['../classmlx_1_1core_1_1_sqrt.html#a6d205e679a593d1ba20206c5c47ba501',1,'mlx::core::Sqrt::eval_gpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a907b96f0a1ce608e211d87ccf2b9ca89',1,'mlx::core::StopGradient::eval_gpu()'],['../classmlx_1_1core_1_1_subtract.html#a69021b23daf061764d97fabbc0f4f06c',1,'mlx::core::Subtract::eval_gpu()'],['../classmlx_1_1core_1_1_squeeze.html#a18d382c8bc59d60b38e9fd1cb70660fd',1,'mlx::core::Squeeze::eval_gpu()'],['../classmlx_1_1core_1_1_tan.html#aca7dbb4836507005a2032ac957a04d3f',1,'mlx::core::Tan::eval_gpu()'],['../classmlx_1_1core_1_1_tanh.html#a48df896599ae93dbce84a5c0f50cf761',1,'mlx::core::Tanh::eval_gpu()'],['../classmlx_1_1core_1_1_unflatten.html#adfbb8208355f9c3cb2e4cb1fd4fe788f',1,'mlx::core::Unflatten::eval_gpu()'],['../classmlx_1_1core_1_1_view.html#add6e12ff1e476fe1db7718b14f21b075',1,'mlx::core::View::eval_gpu()'],['../classmlx_1_1core_1_1_transpose.html#a38d25739c08aa594a6775015a1d7d92e',1,'mlx::core::Transpose::eval_gpu()'],['../classmlx_1_1core_1_1_q_r_f.html#ae5fa3482192f4713605cd07e7fc1c6c9',1,'mlx::core::QRF::eval_gpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a7067b2207f826a25549d571856b94e83',1,'mlx::core::SVD::eval_gpu()'],['../classmlx_1_1core_1_1_inverse.html#a086fbbc947ad232e01686ad063a78ed2',1,'mlx::core::Inverse::eval_gpu()'],['../classmlx_1_1core_1_1_cholesky.html#a8c918594bf129888044ef37fcae56795',1,'mlx::core::Cholesky::eval_gpu()'],['../classmlx_1_1core_1_1_eigh.html#a67775b41c0a15e356f08d51d9736baa2',1,'mlx::core::Eigh::eval_gpu()']]], - ['event_29',['Event',['../classmlx_1_1core_1_1_event.html#a833506419b2110ad1abd89b2dd238b4d',1,'mlx::core::Event::Event()=default'],['../classmlx_1_1core_1_1_event.html#a13e4835f2ffb2cc22e29148a448ea184',1,'mlx::core::Event::Event(const Stream &steam)']]], - ['event_30',['event',['../classmlx_1_1core_1_1array.html#a0a8e4d6e67e739a712876bb36f88f9bf',1,'mlx::core::array']]], - ['exec_31',['exec',['../classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2',1,'pocketfft::detail::cfftp::exec()'],['../classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac',1,'pocketfft::detail::rfftp::exec()'],['../classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c',1,'pocketfft::detail::fftblue::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d',1,'pocketfft::detail::pocketfft_c::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33',1,'pocketfft::detail::pocketfft_r::exec()'],['../classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058',1,'pocketfft::detail::T_dct1::exec()'],['../classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26',1,'pocketfft::detail::T_dst1::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac',1,'pocketfft::detail::T_dcst23::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708',1,'pocketfft::detail::T_dcst4::exec()'],['../classmlx_1_1core_1_1_jit_compiler.html#adcf98f940e1919388eaab907ea17a540',1,'mlx::core::JitCompiler::exec()']]], - ['exec_5fr_32',['exec_r',['../classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00',1,'pocketfft::detail::fftblue']]], - ['exp_33',['Exp',['../classmlx_1_1core_1_1_exp.html#a1d0a618cbb91ab29ef53b57ff6ed6e06',1,'mlx::core::Exp']]], - ['exp_34',['exp',['../namespacemetal.html#ac2a0b3618d922ac014baac8189d44650',1,'metal::exp()'],['../namespacemetal_1_1fast.html#ad3dbd387b63373c29e3449609f763ede',1,'metal::fast::exp()'],['../namespacemetal_1_1precise.html#a8d8d2d5700ce432b33cf47cf22528e8f',1,'metal::precise::exp()'],['../group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125',1,'mlx::core::exp()']]], - ['exp10_35',['exp10',['../namespacemetal.html#a4c63707d13c89364496a48906631c204',1,'metal::exp10()'],['../namespacemetal_1_1fast.html#a453122f982485cbb4e471b3ac282ee5e',1,'metal::fast::exp10()'],['../namespacemetal_1_1precise.html#af9addb343c967da3a83e9e123a8521fd',1,'metal::precise::exp10()']]], - ['exp2_36',['exp2',['../namespacemetal.html#a228201c20777848804a4d0589c1d33e7',1,'metal::exp2()'],['../namespacemetal_1_1fast.html#ac092b65a46720adaf22f6266671d2d71',1,'metal::fast::exp2()'],['../namespacemetal_1_1precise.html#a92a880bd2197efc0da0f8f0f7ec1e4c9',1,'metal::precise::exp2()']]], - ['expand_5fdims_37',['expand_dims',['../group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133',1,'mlx::core::expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a',1,'mlx::core::expand_dims(const array &a, int axis, StreamOrDevice s={})']]], - ['expanddims_38',['ExpandDims',['../classmlx_1_1core_1_1_expand_dims.html#aea2479ea4dd93941eb83a22e087983a8',1,'mlx::core::ExpandDims']]], - ['expm1_39',['Expm1',['../classmlx_1_1core_1_1_expm1.html#a47c2a1b2a4ef6bb07ba77c55ddddaec2',1,'mlx::core::Expm1']]], - ['expm1_40',['expm1',['../group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0',1,'mlx::core']]], - ['expm1f_41',['expm1f',['../expm1f_8h.html#a87f66d30e185950f42ce3641783cdc40',1,'expm1f.h']]], - ['expm1f_5fscaled_5funchecked_42',['expm1f_scaled_unchecked',['../expm1f_8h.html#adf20e03405fba634ca8d01acac24592e',1,'expm1f.h']]], - ['export_5ffunction_43',['export_function',['../namespacemlx_1_1core.html#a2afa4ea816ac9317200fd5c964fc89dc',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, const Args &args, bool shapeless=false)'],['../namespacemlx_1_1core.html#ace51644e2aa72f8d56b86eaa0e1a68b7',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, const Kwargs &kwargs, bool shapeless=false)'],['../namespacemlx_1_1core.html#a9692d7bb6de3456abc535d0f4bac7a94',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, const Args &args, const Kwargs &kwargs, bool shapeless=false)']]], - ['export_5fto_5fdot_44',['export_to_dot',['../namespacemlx_1_1core.html#a30338cb7d259334e46dc7a4819716fa6',1,'mlx::core::export_to_dot(std::ostream &os, NodeNamer namer, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a57395bdf43d9c5c134e610c169222cca',1,'mlx::core::export_to_dot(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a839f94dbad44f0d37333006fc876b42e',1,'mlx::core::export_to_dot(std::ostream &os, Arrays &&... outputs)'],['../namespacemlx_1_1core.html#a12faa564e0d85a1daa77c08babb75261',1,'mlx::core::export_to_dot(std::ostream &os, NodeNamer namer, Arrays &&... outputs)']]], - ['exporter_45',['exporter',['../namespacemlx_1_1core.html#a0303e26b737c9fd197ed9caa90fd21a7',1,'mlx::core::exporter(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#af2735df8513ecce88456585f5aea50f5',1,'mlx::core::exporter(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#af38d5718f517e50a590fdb3d63a90df1',1,'mlx::core::exporter(const std::string &path, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, bool shapeless=false)']]], - ['eye_46',['eye',['../group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3',1,'mlx::core::eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd',1,'mlx::core::eye(int n, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga61657db78ef35d41112d362c869c25d2',1,'mlx::core::eye(int n, int m, StreamOrDevice s={})'],['../group__ops.html#ga908a15b42834be498a46856c99dfc779',1,'mlx::core::eye(int n, int m, int k, StreamOrDevice s={})'],['../group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f',1,'mlx::core::eye(int n, StreamOrDevice s={})']]] + ['encode_5fsignal_15',['encode_signal',['../namespacemlx_1_1core.html#a6d452306f0f046a7d021bd94f8713a89',1,'mlx::core']]], + ['encode_5fwait_16',['encode_wait',['../namespacemlx_1_1core.html#a2874ba55b73057b76c23a7429fdd2d6e',1,'mlx::core']]], + ['end_17',['end',['../classmlx_1_1core_1_1array.html#a5daf64552fb450825c9b382f3a5fa2d4',1,'mlx::core::array']]], + ['end_5fencoding_18',['end_encoding',['../classmlx_1_1core_1_1metal_1_1_device.html#a60689f97347811b27e8c5ca23e0372bf',1,'mlx::core::metal::Device']]], + ['enqueue_19',['enqueue',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4918720319cf224a1b4208568964c286',1,'mlx::core::scheduler::StreamThread::enqueue()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a03809c783bd1866362dc7cb9118abbcc',1,'mlx::core::scheduler::Scheduler::enqueue()'],['../class_thread_pool.html#a375fa2d63197282277be640b54e8a196',1,'ThreadPool::enqueue()'],['../namespacemlx_1_1core_1_1scheduler.html#aa2d4eacf5d5cbc778a51aafd4fd8e4d7',1,'mlx::core::scheduler::enqueue()']]], + ['epsilon_20',['epsilon',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a96c4197e3076f0aa9065370b8ece49ca',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['equal_21',['Equal',['../classmlx_1_1core_1_1_equal.html#a4af81cf2dd071db5bbf8ce1df95fdf36',1,'mlx::core::Equal']]], + ['equal_22',['equal',['../group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde',1,'mlx::core']]], + ['erase_23',['erase',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#ae136ad270522210c85c13cacf5165238',1,'mlx::core::metal::ResidencySet']]], + ['erf_24',['Erf',['../classmlx_1_1core_1_1_erf.html#a702f76f848928d8d7d3d0881ac6e4c82',1,'mlx::core::Erf']]], + ['erf_25',['erf',['../namespacemlx_1_1core_1_1simd.html#a60e33ebb16d9bab375a64aec8015a5c2',1,'mlx::core::simd::erf()'],['../erf_8h.html#a6ce199ee56105c67adbf8c48c019a8b2',1,'erf(): erf.h'],['../group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0',1,'mlx::core::erf()']]], + ['erfinv_26',['ErfInv',['../classmlx_1_1core_1_1_erf_inv.html#a5d0279247b67da4592311559f04e1478',1,'mlx::core::ErfInv']]], + ['erfinv_27',['erfinv',['../namespacemlx_1_1core_1_1simd.html#a7687f3d14077b51fb421f0efb5b626db',1,'mlx::core::simd::erfinv()'],['../erf_8h.html#a1846e0d683c7aff826bb32addcc3b885',1,'erfinv(): erf.h'],['../group__ops.html#ga76fb9062c64264e34d2e07013390557c',1,'mlx::core::erfinv()']]], + ['eval_28',['eval',['../classmlx_1_1core_1_1array.html#a2820c45188071a22175e9fa42e10a49a',1,'mlx::core::array::eval()'],['../namespacemlx_1_1core.html#a7d6e097d8effed52f4713672e471f299',1,'mlx::core::eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#adb14f689c9f75f7901edb196c2bfb971',1,'mlx::core::eval(Arrays &&... outputs)']]], + ['eval_5fcpu_29',['eval_cpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#acdc1965ad64ee9ee6328fe150a97902e',1,'mlx::core::distributed::AllReduce::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ab721fe0072fffbddbc3c4334dd033ba5',1,'mlx::core::distributed::AllGather::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#af2620837bfc1b97217d006ed6e374051',1,'mlx::core::distributed::Send::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a3be84b08122a939edd6062d26261358a',1,'mlx::core::distributed::Recv::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a7da6e0cfd630958d9633b2e2bd97a54f',1,'mlx::core::fast::RMSNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#adfc1d52bc266466ab29ee45fd8fab439',1,'mlx::core::fast::RMSNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a5d7a4c1c9ee84e327d1c371733108c05',1,'mlx::core::fast::LayerNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a0d8c4c6e7462befc38f7e08244fa1c2b',1,'mlx::core::fast::LayerNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a05a7d595c6b9dadf7ddfd6e3fd402f0e',1,'mlx::core::fast::RoPE::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ae20851e002f7fcb6d4f97817596f6328',1,'mlx::core::fast::ScaledDotProductAttention::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a3b5d628628d245b38911118d4a0ff9fd',1,'mlx::core::fast::AffineQuantize::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a4ad1b7a9919753c759093f3e21a15bad',1,'mlx::core::fast::CustomKernel::eval_cpu()'],['../classmlx_1_1core_1_1_primitive.html#a1596dc50b910538eae14878e98f07575',1,'mlx::core::Primitive::eval_cpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a7e8f6f5d6ae0a33f6abc0f5a46e0b132',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#aa0ed6e32c36200a3ff9bc592c9b300db',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0d3e697496ef8e842d21195cb3c14e60',1,'mlx::core::Abs::eval_cpu()'],['../classmlx_1_1core_1_1_add.html#a5bacfc51dfa2a5a931bad2dd7bdc7a5f',1,'mlx::core::Add::eval_cpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a15694e3bf2ed5c193237b2b9ca00867c',1,'mlx::core::AddMM::eval_cpu()'],['../classmlx_1_1core_1_1_arange.html#aba44432491cbd599bf72712f5f4267a1',1,'mlx::core::Arange::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a58dcba9e706cb12bab062bb7fa5fa006',1,'mlx::core::ArcCos::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#a0f6d989bcbbc38f15ef17a136879a9c9',1,'mlx::core::ArcCosh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sin.html#ab3542492c14021329788de8f2a9be1e4',1,'mlx::core::ArcSin::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a52574b24d8d16839c58673f51f8ac066',1,'mlx::core::ArcSinh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a1211bc31241227528f04435239ddb9a3',1,'mlx::core::ArcTan::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a13094e6b702769928ca0da468f5ce45c',1,'mlx::core::ArcTan2::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a5af9224e1f1ffec412b0baa0af7e1ecd',1,'mlx::core::ArcTanh::eval_cpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a896f75c5325798ac3f9093f6a4581828',1,'mlx::core::ArgPartition::eval_cpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#ad8d48725623ede1ff654fa13eccf2287',1,'mlx::core::ArgReduce::eval_cpu()'],['../classmlx_1_1core_1_1_arg_sort.html#a022079683774bfeb531b3a002cff16fa',1,'mlx::core::ArgSort::eval_cpu()'],['../classmlx_1_1core_1_1_as_type.html#aa89dbf4d73b00c6a44cffd04d5bb228d',1,'mlx::core::AsType::eval_cpu()'],['../classmlx_1_1core_1_1_as_strided.html#acdd4705e4503ff0b124215c4676b4193',1,'mlx::core::AsStrided::eval_cpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a2194bf585213bda1b2966aa02d2fe283',1,'mlx::core::BitwiseBinary::eval_cpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aa85da478cdc6d4a97be06e5d4abee1f2',1,'mlx::core::BlockMaskedMM::eval_cpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#a62352074a480df0e1f879b0bae425730',1,'mlx::core::GatherMM::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a6423095cd28b2f90893c03166257a568',1,'mlx::core::BroadcastAxes::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast.html#a53d48d9778e2d4c24a124cd767900780',1,'mlx::core::Broadcast::eval_cpu()'],['../classmlx_1_1core_1_1_ceil.html#a9791801fff3f8b79944e15ac2a45a035',1,'mlx::core::Ceil::eval_cpu()'],['../classmlx_1_1core_1_1_compiled.html#ac45b1d0fedd85feefbff7ce7e168b151',1,'mlx::core::Compiled::eval_cpu()'],['../classmlx_1_1core_1_1_concatenate.html#a609e76bede7fc5581ec84ddcb727a258',1,'mlx::core::Concatenate::eval_cpu()'],['../classmlx_1_1core_1_1_conjugate.html#ae39643e2178f442ffba05139f8609d61',1,'mlx::core::Conjugate::eval_cpu()'],['../classmlx_1_1core_1_1_contiguous.html#a742de24e6c0310cd85a606dec0cd8336',1,'mlx::core::Contiguous::eval_cpu()'],['../classmlx_1_1core_1_1_convolution.html#ac74256068da01730629109fa4fa8432b',1,'mlx::core::Convolution::eval_cpu()'],['../classmlx_1_1core_1_1_copy.html#af4a0ebec423e84ffe8083a5e9ed0d70c',1,'mlx::core::Copy::eval_cpu()'],['../classmlx_1_1core_1_1_cos.html#a061fc446268fe56237ae6b20ccf78152',1,'mlx::core::Cos::eval_cpu()'],['../classmlx_1_1core_1_1_cosh.html#ae8702df7e8f0e20cbeccb2a548961d3d',1,'mlx::core::Cosh::eval_cpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#adba1c40c77a2138df6b5f75483f62184',1,'mlx::core::CustomTransforms::eval_cpu()'],['../classmlx_1_1core_1_1_depends.html#a0c7ea6db97337591fa53c6e6bde41e5e',1,'mlx::core::Depends::eval_cpu()'],['../classmlx_1_1core_1_1_divide.html#a823443c2a8e8b81bbcaeee6ddbcdbf49',1,'mlx::core::Divide::eval_cpu()'],['../classmlx_1_1core_1_1_div_mod.html#ae350b7b93ad128e3133ee14f247193b3',1,'mlx::core::DivMod::eval_cpu()'],['../classmlx_1_1core_1_1_select.html#aa51aa36e0adbd69e0d23d7c7adf88de2',1,'mlx::core::Select::eval_cpu()'],['../classmlx_1_1core_1_1_remainder.html#ac6c6c86a0bf02e6e529eb87f6e617ccc',1,'mlx::core::Remainder::eval_cpu()'],['../classmlx_1_1core_1_1_equal.html#aabb8aa61fa581defddcdca1274b1b454',1,'mlx::core::Equal::eval_cpu()'],['../classmlx_1_1core_1_1_erf.html#a84ea16e43d5b7f83bbc2d5ece78a3fb6',1,'mlx::core::Erf::eval_cpu()'],['../classmlx_1_1core_1_1_erf_inv.html#af579627402af3249565134884701d39e',1,'mlx::core::ErfInv::eval_cpu()'],['../classmlx_1_1core_1_1_exp.html#a47934c5a5023bc7ae7ae89bff45ebb2c',1,'mlx::core::Exp::eval_cpu()'],['../classmlx_1_1core_1_1_expm1.html#ab9c8b7aa50fe4592d55f8957baac647a',1,'mlx::core::Expm1::eval_cpu()'],['../classmlx_1_1core_1_1_expand_dims.html#a34058a87582a6ab2e5d82a75bc713030',1,'mlx::core::ExpandDims::eval_cpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a6bc262a0c2b5d4fe655e3e2e0ff28635',1,'mlx::core::FFT::eval_cpu()'],['../classmlx_1_1core_1_1_flatten.html#a72ade7d22386b349712f6c7c1f619842',1,'mlx::core::Flatten::eval_cpu()'],['../classmlx_1_1core_1_1_floor.html#a1a7dc5f571b7b73e7ef3cbdc1dd1fcf7',1,'mlx::core::Floor::eval_cpu()'],['../classmlx_1_1core_1_1_full.html#a3dccd3756599d7fd018b2af0093b082c',1,'mlx::core::Full::eval_cpu()'],['../classmlx_1_1core_1_1_gather.html#a9ed5587f0d04b59a2b9186c0aac21290',1,'mlx::core::Gather::eval_cpu()'],['../classmlx_1_1core_1_1_gather_axis.html#a474eae1d024e676e668318bf10928e2a',1,'mlx::core::GatherAxis::eval_cpu()'],['../classmlx_1_1core_1_1_greater.html#abe1c03f311d0e0b610f3392a6566f2ae',1,'mlx::core::Greater::eval_cpu()'],['../classmlx_1_1core_1_1_greater_equal.html#a15469125b9bea89b64bfeac01590c075',1,'mlx::core::GreaterEqual::eval_cpu()'],['../classmlx_1_1core_1_1_hadamard.html#ab27d6a9df42b3aab41ace3073a4c880d',1,'mlx::core::Hadamard::eval_cpu()'],['../classmlx_1_1core_1_1_imag.html#a17d1f1f9f8528668fcdf39b636720829',1,'mlx::core::Imag::eval_cpu()'],['../classmlx_1_1core_1_1_less.html#a32624124ffece066f496b3299056bcef',1,'mlx::core::Less::eval_cpu()'],['../classmlx_1_1core_1_1_less_equal.html#a55d1352b0e97841a92503bc57c19ed16',1,'mlx::core::LessEqual::eval_cpu()'],['../classmlx_1_1core_1_1_load.html#ada026ac30566f3109d8182e35d307c0a',1,'mlx::core::Load::eval_cpu()'],['../classmlx_1_1core_1_1_log.html#aadc7bb4cb24f3ecbbb9ed54a699ab74f',1,'mlx::core::Log::eval_cpu()'],['../classmlx_1_1core_1_1_log1p.html#a8192e5438de99c4cda056987935cba23',1,'mlx::core::Log1p::eval_cpu()'],['../classmlx_1_1core_1_1_logical_not.html#acf3f7b3b20ca69533536e0e0a05725b3',1,'mlx::core::LogicalNot::eval_cpu()'],['../classmlx_1_1core_1_1_logical_and.html#adbe1c1785af1a8b827289d22b0d170b3',1,'mlx::core::LogicalAnd::eval_cpu()'],['../classmlx_1_1core_1_1_logical_or.html#a13cd4cbf26589287e85aeaaca42d7f62',1,'mlx::core::LogicalOr::eval_cpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#abef17fb590b1a8d356f2a580e45d41f0',1,'mlx::core::LogAddExp::eval_cpu()'],['../classmlx_1_1core_1_1_matmul.html#a357a7f57a2a220a91977f810a69413fc',1,'mlx::core::Matmul::eval_cpu()'],['../classmlx_1_1core_1_1_maximum.html#a62b38fbe5f96db58c2b60165ac4eadcf',1,'mlx::core::Maximum::eval_cpu()'],['../classmlx_1_1core_1_1_minimum.html#a6b93f493ee87089943a8085fe59dfc6e',1,'mlx::core::Minimum::eval_cpu()'],['../classmlx_1_1core_1_1_multiply.html#a624fce06c047cdc4dfdbdcaaddb25f34',1,'mlx::core::Multiply::eval_cpu()'],['../classmlx_1_1core_1_1_negative.html#af43553dc418c8ebe75fa9cdcba103c3b',1,'mlx::core::Negative::eval_cpu()'],['../classmlx_1_1core_1_1_not_equal.html#a8f95f8b5873850b875b1641df8196047',1,'mlx::core::NotEqual::eval_cpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#acc328321cf5300874ee884367cbede3f',1,'mlx::core::NumberOfElements::eval_cpu()'],['../classmlx_1_1core_1_1_pad.html#aaf82dd163cd536fbf97304f8b29080cb',1,'mlx::core::Pad::eval_cpu()'],['../classmlx_1_1core_1_1_partition.html#a784596ab567f9f3cb4fe1a69466523d8',1,'mlx::core::Partition::eval_cpu()'],['../classmlx_1_1core_1_1_power.html#a6783da16fb6ff393aaa57737f1973206',1,'mlx::core::Power::eval_cpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ab3dfa73b74d8f4f2e9ab4f0eb016b0e3',1,'mlx::core::QuantizedMatmul::eval_cpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a89aae98bfbdd6563df44ef7d70f0bf8c',1,'mlx::core::GatherQMM::eval_cpu()'],['../classmlx_1_1core_1_1_random_bits.html#a5752d051cd16cf5f8d4754c0a656f0d2',1,'mlx::core::RandomBits::eval_cpu()'],['../classmlx_1_1core_1_1_real.html#a365d046caac91b521f0f5a5518037934',1,'mlx::core::Real::eval_cpu()'],['../classmlx_1_1core_1_1_reshape.html#a658de2c5f710991b48e14b2bd19b229f',1,'mlx::core::Reshape::eval_cpu()'],['../classmlx_1_1core_1_1_reduce.html#aeb8a58b560c0a09ae3a695df7829acfa',1,'mlx::core::Reduce::eval_cpu()'],['../classmlx_1_1core_1_1_round.html#ad066b0944b437f64ab546025efa00007',1,'mlx::core::Round::eval_cpu()'],['../classmlx_1_1core_1_1_scan.html#a15676d9fd066e935782a923fba3e940b',1,'mlx::core::Scan::eval_cpu()'],['../classmlx_1_1core_1_1_scatter.html#a7623f590f8b77167b5ebb4f14bc9dc97',1,'mlx::core::Scatter::eval_cpu()'],['../classmlx_1_1core_1_1_scatter_axis.html#abf9d24565abdd7e1034daacac603cc54',1,'mlx::core::ScatterAxis::eval_cpu()'],['../classmlx_1_1core_1_1_sigmoid.html#aa930ce05734cca529ebcb8d0ca8e1255',1,'mlx::core::Sigmoid::eval_cpu()'],['../classmlx_1_1core_1_1_sign.html#a7498ec993b66879be30c5d9762c45a97',1,'mlx::core::Sign::eval_cpu()'],['../classmlx_1_1core_1_1_sin.html#ab34f9cebc2aed55a0b6ab4c991f02eb5',1,'mlx::core::Sin::eval_cpu()'],['../classmlx_1_1core_1_1_sinh.html#ab6d5f6f40d177f6435f6a51c71b939dd',1,'mlx::core::Sinh::eval_cpu()'],['../classmlx_1_1core_1_1_slice.html#a4b13503f5b2f5c6a90d394b020f9b3f2',1,'mlx::core::Slice::eval_cpu()'],['../classmlx_1_1core_1_1_slice_update.html#ad82ca0e3ab88a0e086431050deea831b',1,'mlx::core::SliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a4e8c22c24a587ea0648ce89f461ed1ee',1,'mlx::core::DynamicSlice::eval_cpu()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a379185914db0326a5d4839839fe4fc83',1,'mlx::core::DynamicSliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_softmax.html#ac9ebc2eab1683b682e689ed8f4622b79',1,'mlx::core::Softmax::eval_cpu()'],['../classmlx_1_1core_1_1_sort.html#a459769a0241b2620e55bedaba19827cd',1,'mlx::core::Sort::eval_cpu()'],['../classmlx_1_1core_1_1_split.html#aff2889cb9074f0fda53edf8fa40b1fd4',1,'mlx::core::Split::eval_cpu()'],['../classmlx_1_1core_1_1_square.html#a1f4d327a705950616da63b83c2829e59',1,'mlx::core::Square::eval_cpu()'],['../classmlx_1_1core_1_1_sqrt.html#a5a64ecc4eef1e30a2963435dca7cefd5',1,'mlx::core::Sqrt::eval_cpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a56207714d374b08f60e4d9cdbc7340b2',1,'mlx::core::StopGradient::eval_cpu()'],['../classmlx_1_1core_1_1_subtract.html#a47574258b6c95f8ad260c114d6d36a12',1,'mlx::core::Subtract::eval_cpu()'],['../classmlx_1_1core_1_1_squeeze.html#a9bcb7476041020f59ef816196ddb81cb',1,'mlx::core::Squeeze::eval_cpu()'],['../classmlx_1_1core_1_1_tan.html#a9c9a731158fa60eef30067fe0da9f3e9',1,'mlx::core::Tan::eval_cpu()'],['../classmlx_1_1core_1_1_tanh.html#af7ed4345f622da069e5b0284067923f5',1,'mlx::core::Tanh::eval_cpu()'],['../classmlx_1_1core_1_1_unflatten.html#a507c22306b7afcdd5970cfaa32188f0a',1,'mlx::core::Unflatten::eval_cpu()'],['../classmlx_1_1core_1_1_view.html#a0ad6deb11914a242f10e8039fcb02497',1,'mlx::core::View::eval_cpu()'],['../classmlx_1_1core_1_1_transpose.html#a1fbcfcca43f9ec06c63a3c14708c30f8',1,'mlx::core::Transpose::eval_cpu()'],['../classmlx_1_1core_1_1_q_r_f.html#a48493887395d65a27f04de1804d277d2',1,'mlx::core::QRF::eval_cpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a637f5c39fa8b10722c04a066f6c1ada6',1,'mlx::core::SVD::eval_cpu()'],['../classmlx_1_1core_1_1_inverse.html#aeb1d8dc9bc4052a616023f65b3c7bb81',1,'mlx::core::Inverse::eval_cpu()'],['../classmlx_1_1core_1_1_cholesky.html#a4bdec36c1cc99aadf9a4a39d4c57bea5',1,'mlx::core::Cholesky::eval_cpu()'],['../classmlx_1_1core_1_1_eigh.html#a894b32e17229394f6a43b4a0655fd8be',1,'mlx::core::Eigh::eval_cpu()']]], + ['eval_5fgpu_30',['eval_gpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a52df7155f56b8450581b2fd2747cad20',1,'mlx::core::distributed::AllReduce::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a4251ce0f2db2045226b66210b828af7a',1,'mlx::core::distributed::AllGather::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_send.html#a0c8dbd2a912be91be04ec701e29fba3d',1,'mlx::core::distributed::Send::eval_gpu()'],['../classmlx_1_1core_1_1distributed_1_1_recv.html#a932e39624bc3d234a7489c3decc4749e',1,'mlx::core::distributed::Recv::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae7955e8d43c097eecae264e804b4d8ca',1,'mlx::core::fast::RMSNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a48efb8fa84c4ba6cc9fb560ebbe01560',1,'mlx::core::fast::RMSNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a77abda7f47bffa2c037a5d60cccc1528',1,'mlx::core::fast::LayerNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a954a003a4a27c8c4c60a5a14142a9cc3',1,'mlx::core::fast::LayerNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a913b6b00fc518b25ac3947e4e15790f2',1,'mlx::core::fast::RoPE::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a505f38ba93a3499895f5312e0112e73d',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ad51666e69f670e286293aff96eb435a9',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, array &out)'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a63812b2abaf26ad7e7fa4c9e82db1628',1,'mlx::core::fast::AffineQuantize::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_custom_kernel.html#a2ed2a16b23053f8195068386a99fd6db',1,'mlx::core::fast::CustomKernel::eval_gpu()'],['../classmlx_1_1core_1_1_primitive.html#ad217376dcf5eff691d731566faec2ba2',1,'mlx::core::Primitive::eval_gpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a6b7f80abaf038d53ec6ffbb0dfac6adb',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#a971fe9ad47f6569118879ce1d0f41447',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0a976e636dd8505b473fbdddf949f514',1,'mlx::core::Abs::eval_gpu()'],['../classmlx_1_1core_1_1_add.html#aa0aacbc1e26b95a2f040f62aa4f69c3d',1,'mlx::core::Add::eval_gpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a5f933be14baebc32a0be0f9a69148aa9',1,'mlx::core::AddMM::eval_gpu()'],['../classmlx_1_1core_1_1_arange.html#a7a2e9787c6c3a78b4a6df91206974031',1,'mlx::core::Arange::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a46f72d4af89b0a0f5f203783fb44589c',1,'mlx::core::ArcCos::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#aa6a2587485a0e015ac2d5211d7d045fc',1,'mlx::core::ArcCosh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sin.html#a7fa4ae7a85bc8bed97ea258ae30762f3',1,'mlx::core::ArcSin::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79f648a86de4c10386a1ce3b5e38e8ac',1,'mlx::core::ArcSinh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a77866feb27028865d844070447c9a254',1,'mlx::core::ArcTan::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a76d3f0c29e0ff4642b8d39dac90d3f50',1,'mlx::core::ArcTan2::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a10566b9d3b2c7d090895b46d9040bc1d',1,'mlx::core::ArcTanh::eval_gpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a9a60995eaf85f63c877e86b23cbc15fc',1,'mlx::core::ArgPartition::eval_gpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#aafa982ce2abc0cd9e81e43aa2c823d29',1,'mlx::core::ArgReduce::eval_gpu()'],['../classmlx_1_1core_1_1_arg_sort.html#abc2d730850ec4ee8d7968b7417911709',1,'mlx::core::ArgSort::eval_gpu()'],['../classmlx_1_1core_1_1_as_type.html#a5b111b9d74c60d27b4a7ebaa49f96e0b',1,'mlx::core::AsType::eval_gpu()'],['../classmlx_1_1core_1_1_as_strided.html#ab6771a208323994927ca162ba7bb10ed',1,'mlx::core::AsStrided::eval_gpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#ac831a29fc46701b00bbe63ee33832afd',1,'mlx::core::BitwiseBinary::eval_gpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ab372b6df4de00a33795a052a23bb1df9',1,'mlx::core::BlockMaskedMM::eval_gpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#ad754c35f460a055cc383ad93a5f72da1',1,'mlx::core::GatherMM::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a56d16e75a0df867d2f1ba4e5198f15cb',1,'mlx::core::BroadcastAxes::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast.html#ab9bd9dbcedcefc9b29c84911b5ce69fe',1,'mlx::core::Broadcast::eval_gpu()'],['../classmlx_1_1core_1_1_ceil.html#abe178e0058e44b6618be414215e96887',1,'mlx::core::Ceil::eval_gpu()'],['../classmlx_1_1core_1_1_compiled.html#aa3d5ff0f2b3554ad48fbbf2a0f3336d5',1,'mlx::core::Compiled::eval_gpu()'],['../classmlx_1_1core_1_1_concatenate.html#a309a1c50e97f9925866433ee2841c474',1,'mlx::core::Concatenate::eval_gpu()'],['../classmlx_1_1core_1_1_conjugate.html#aff0a802166e3724db88ab5d3feb2d3de',1,'mlx::core::Conjugate::eval_gpu()'],['../classmlx_1_1core_1_1_contiguous.html#a519cd16fd0c55b371ea7625fbb37c70f',1,'mlx::core::Contiguous::eval_gpu()'],['../classmlx_1_1core_1_1_convolution.html#a30b64109eeb1778f002b99447dff9dd2',1,'mlx::core::Convolution::eval_gpu()'],['../classmlx_1_1core_1_1_copy.html#a1eda7b2ea771a168f67421f0d384b3a1',1,'mlx::core::Copy::eval_gpu()'],['../classmlx_1_1core_1_1_cos.html#a5ef41aafad595f6cdd8c535e36e12060',1,'mlx::core::Cos::eval_gpu()'],['../classmlx_1_1core_1_1_cosh.html#a23f71b43792934c3ec0ebe9b74f32559',1,'mlx::core::Cosh::eval_gpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#a7b3538681acbb20af3ed37b0877f6667',1,'mlx::core::CustomTransforms::eval_gpu()'],['../classmlx_1_1core_1_1_depends.html#ae5057f65e69490ad0add8eeda2b75e28',1,'mlx::core::Depends::eval_gpu()'],['../classmlx_1_1core_1_1_divide.html#abffda0ce37221ddc28dc9eea794f6bc7',1,'mlx::core::Divide::eval_gpu()'],['../classmlx_1_1core_1_1_div_mod.html#a003117c9ecf3c06a27248f72a76348dc',1,'mlx::core::DivMod::eval_gpu()'],['../classmlx_1_1core_1_1_select.html#a2a82b6cba4c386b2b87f225a4b08ea9b',1,'mlx::core::Select::eval_gpu()'],['../classmlx_1_1core_1_1_remainder.html#a7919ea9b84e42522d51bf0d5a396e161',1,'mlx::core::Remainder::eval_gpu()'],['../classmlx_1_1core_1_1_equal.html#ac3757001fec42ceb5ece2954df42161c',1,'mlx::core::Equal::eval_gpu()'],['../classmlx_1_1core_1_1_erf.html#ad8551be664d767dccc3c0d8cc1eca008',1,'mlx::core::Erf::eval_gpu()'],['../classmlx_1_1core_1_1_erf_inv.html#a4a2413d0634db1f3dae1806ddfa632db',1,'mlx::core::ErfInv::eval_gpu()'],['../classmlx_1_1core_1_1_exp.html#a7d63695a97a14760fd33b5d4e6590822',1,'mlx::core::Exp::eval_gpu()'],['../classmlx_1_1core_1_1_expm1.html#a82930071f4b77d883b300f77966aff5f',1,'mlx::core::Expm1::eval_gpu()'],['../classmlx_1_1core_1_1_expand_dims.html#ad350ede3abecc55371ddeb89fbba2b90',1,'mlx::core::ExpandDims::eval_gpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a1c21b26d1e9ad7c4da78ae845721b2dd',1,'mlx::core::FFT::eval_gpu()'],['../classmlx_1_1core_1_1_flatten.html#acb2219cc122d218b273af2cb9a882e7f',1,'mlx::core::Flatten::eval_gpu()'],['../classmlx_1_1core_1_1_floor.html#aaa29c83538099eb8f951c95a41f2eb65',1,'mlx::core::Floor::eval_gpu()'],['../classmlx_1_1core_1_1_full.html#aa54f99bb4cba12a551392dea56003872',1,'mlx::core::Full::eval_gpu()'],['../classmlx_1_1core_1_1_gather.html#aec48ee529cb2449915a7b27a3c4361e8',1,'mlx::core::Gather::eval_gpu()'],['../classmlx_1_1core_1_1_gather_axis.html#a1344749d33e4ea2cb80b69a5a4a21afc',1,'mlx::core::GatherAxis::eval_gpu()'],['../classmlx_1_1core_1_1_greater.html#ae8957cccf4c924d941f57a1bb751c878',1,'mlx::core::Greater::eval_gpu()'],['../classmlx_1_1core_1_1_greater_equal.html#ac246263b4548126c3d4ab7e392575d24',1,'mlx::core::GreaterEqual::eval_gpu()'],['../classmlx_1_1core_1_1_hadamard.html#a2470feb690f5463138490763c38b5733',1,'mlx::core::Hadamard::eval_gpu()'],['../classmlx_1_1core_1_1_imag.html#a247a4d059b0a99678c6be8c15e42c1e6',1,'mlx::core::Imag::eval_gpu()'],['../classmlx_1_1core_1_1_less.html#a353335ce06ddbe8498d86d129c835917',1,'mlx::core::Less::eval_gpu()'],['../classmlx_1_1core_1_1_less_equal.html#acf035a82b11e6f63742143ea540fedac',1,'mlx::core::LessEqual::eval_gpu()'],['../classmlx_1_1core_1_1_load.html#a06933e887ea94a4d01d81195c5e07a3d',1,'mlx::core::Load::eval_gpu()'],['../classmlx_1_1core_1_1_log.html#aaaa49e9455f3a197bc319646b5ca6390',1,'mlx::core::Log::eval_gpu()'],['../classmlx_1_1core_1_1_log1p.html#a1b97decae7338d46874e736c95fa7431',1,'mlx::core::Log1p::eval_gpu()'],['../classmlx_1_1core_1_1_logical_not.html#a1d0d2bc93f935eca6c85ef7bf67f2d6a',1,'mlx::core::LogicalNot::eval_gpu()'],['../classmlx_1_1core_1_1_logical_and.html#a132b2eedaa3978de5a5350da3c2ca40f',1,'mlx::core::LogicalAnd::eval_gpu()'],['../classmlx_1_1core_1_1_logical_or.html#a3be1da328f0f8620de2e4fc1d22a077a',1,'mlx::core::LogicalOr::eval_gpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#acace355b62ec00df649f9f99e8f2eb7a',1,'mlx::core::LogAddExp::eval_gpu()'],['../classmlx_1_1core_1_1_matmul.html#a8707a4e9b75c769e8f1dbca15c6a1ae7',1,'mlx::core::Matmul::eval_gpu()'],['../classmlx_1_1core_1_1_maximum.html#ade0f721b10a6b3a12bdadd34c48f72a7',1,'mlx::core::Maximum::eval_gpu()'],['../classmlx_1_1core_1_1_minimum.html#aadc68afa0afbe2103f19d161f5e0a2ba',1,'mlx::core::Minimum::eval_gpu()'],['../classmlx_1_1core_1_1_multiply.html#a634fcb4e981d8d3f4d94252caf25bee0',1,'mlx::core::Multiply::eval_gpu()'],['../classmlx_1_1core_1_1_negative.html#a97f1b316eace0c6d9e576d766940c75b',1,'mlx::core::Negative::eval_gpu()'],['../classmlx_1_1core_1_1_not_equal.html#a61179747e34e203150e9c660dfddb5f2',1,'mlx::core::NotEqual::eval_gpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#a2c98c42915fb2bfe12f5c99ea553eff5',1,'mlx::core::NumberOfElements::eval_gpu()'],['../classmlx_1_1core_1_1_pad.html#aefd4d3a5bd8b6b35b266c9e558ada153',1,'mlx::core::Pad::eval_gpu()'],['../classmlx_1_1core_1_1_partition.html#a8eca1be21ae9ccfda46e6f3e85f506ef',1,'mlx::core::Partition::eval_gpu()'],['../classmlx_1_1core_1_1_power.html#a80577d4c0853c24027777c90a1ec7e11',1,'mlx::core::Power::eval_gpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a2812ad007d695ed1aaf9cf706fb9c4b3',1,'mlx::core::QuantizedMatmul::eval_gpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a86eb048afc95646b2e96ec5493e3d887',1,'mlx::core::GatherQMM::eval_gpu()'],['../classmlx_1_1core_1_1_random_bits.html#a578756866665358577418e4cdd94aa3a',1,'mlx::core::RandomBits::eval_gpu()'],['../classmlx_1_1core_1_1_real.html#a1e209e88a43bdd1eea43ad0b03f9a7f2',1,'mlx::core::Real::eval_gpu()'],['../classmlx_1_1core_1_1_reshape.html#aa1e85f28471875750c47351520b56059',1,'mlx::core::Reshape::eval_gpu()'],['../classmlx_1_1core_1_1_reduce.html#ae9caaf42edadfe73ea208d98f526890f',1,'mlx::core::Reduce::eval_gpu()'],['../classmlx_1_1core_1_1_round.html#af7fe5ff8f3db166c203b4be4b07f13ec',1,'mlx::core::Round::eval_gpu()'],['../classmlx_1_1core_1_1_scan.html#aef22c6fc2b2cb2a907cd8965c7413dde',1,'mlx::core::Scan::eval_gpu()'],['../classmlx_1_1core_1_1_scatter.html#ab304345db3d8cfeea15e27461ae2e678',1,'mlx::core::Scatter::eval_gpu()'],['../classmlx_1_1core_1_1_scatter_axis.html#a715c3b959dc904faefb16edbb11f29d7',1,'mlx::core::ScatterAxis::eval_gpu()'],['../classmlx_1_1core_1_1_sigmoid.html#a7a6bd0222d51d7f25f2719a91ccdfeca',1,'mlx::core::Sigmoid::eval_gpu()'],['../classmlx_1_1core_1_1_sign.html#afa2b48b99a194106006b44af69ffda8b',1,'mlx::core::Sign::eval_gpu()'],['../classmlx_1_1core_1_1_sin.html#a6b59f1156cf8bdad8d45acd1d825cb5e',1,'mlx::core::Sin::eval_gpu()'],['../classmlx_1_1core_1_1_sinh.html#a5a1af2399f166d5b228b5e83a1837c75',1,'mlx::core::Sinh::eval_gpu()'],['../classmlx_1_1core_1_1_slice.html#aa53c21ff06a7c659e889af6b97d10a4a',1,'mlx::core::Slice::eval_gpu()'],['../classmlx_1_1core_1_1_slice_update.html#aac1a1d122e5697be057d63552141032b',1,'mlx::core::SliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_dynamic_slice.html#ab0a2e31c03f02a4f25700e240cf18e3e',1,'mlx::core::DynamicSlice::eval_gpu()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a249dab28690c45203c3995698de0cab7',1,'mlx::core::DynamicSliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_softmax.html#a35dac69ddcc7e2ec0e1a76fe93db85af',1,'mlx::core::Softmax::eval_gpu()'],['../classmlx_1_1core_1_1_sort.html#a4141c48f0e8670c728663f3722675382',1,'mlx::core::Sort::eval_gpu()'],['../classmlx_1_1core_1_1_split.html#a78ddda89c4daee73c74cfbc1e44656df',1,'mlx::core::Split::eval_gpu()'],['../classmlx_1_1core_1_1_square.html#a0ea2a78a5bb52daa4103263bf2f98045',1,'mlx::core::Square::eval_gpu()'],['../classmlx_1_1core_1_1_sqrt.html#a6d205e679a593d1ba20206c5c47ba501',1,'mlx::core::Sqrt::eval_gpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a907b96f0a1ce608e211d87ccf2b9ca89',1,'mlx::core::StopGradient::eval_gpu()'],['../classmlx_1_1core_1_1_subtract.html#a69021b23daf061764d97fabbc0f4f06c',1,'mlx::core::Subtract::eval_gpu()'],['../classmlx_1_1core_1_1_squeeze.html#a18d382c8bc59d60b38e9fd1cb70660fd',1,'mlx::core::Squeeze::eval_gpu()'],['../classmlx_1_1core_1_1_tan.html#aca7dbb4836507005a2032ac957a04d3f',1,'mlx::core::Tan::eval_gpu()'],['../classmlx_1_1core_1_1_tanh.html#a48df896599ae93dbce84a5c0f50cf761',1,'mlx::core::Tanh::eval_gpu()'],['../classmlx_1_1core_1_1_unflatten.html#adfbb8208355f9c3cb2e4cb1fd4fe788f',1,'mlx::core::Unflatten::eval_gpu()'],['../classmlx_1_1core_1_1_view.html#add6e12ff1e476fe1db7718b14f21b075',1,'mlx::core::View::eval_gpu()'],['../classmlx_1_1core_1_1_transpose.html#a38d25739c08aa594a6775015a1d7d92e',1,'mlx::core::Transpose::eval_gpu()'],['../classmlx_1_1core_1_1_q_r_f.html#ae5fa3482192f4713605cd07e7fc1c6c9',1,'mlx::core::QRF::eval_gpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a7067b2207f826a25549d571856b94e83',1,'mlx::core::SVD::eval_gpu()'],['../classmlx_1_1core_1_1_inverse.html#a086fbbc947ad232e01686ad063a78ed2',1,'mlx::core::Inverse::eval_gpu()'],['../classmlx_1_1core_1_1_cholesky.html#a8c918594bf129888044ef37fcae56795',1,'mlx::core::Cholesky::eval_gpu()'],['../classmlx_1_1core_1_1_eigh.html#a67775b41c0a15e356f08d51d9736baa2',1,'mlx::core::Eigh::eval_gpu()']]], + ['event_31',['Event',['../classmlx_1_1core_1_1_event.html#a833506419b2110ad1abd89b2dd238b4d',1,'mlx::core::Event::Event()=default'],['../classmlx_1_1core_1_1_event.html#a13e4835f2ffb2cc22e29148a448ea184',1,'mlx::core::Event::Event(const Stream &steam)']]], + ['event_32',['event',['../classmlx_1_1core_1_1array.html#a0a8e4d6e67e739a712876bb36f88f9bf',1,'mlx::core::array']]], + ['exec_33',['exec',['../classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2',1,'pocketfft::detail::cfftp::exec()'],['../classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac',1,'pocketfft::detail::rfftp::exec()'],['../classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c',1,'pocketfft::detail::fftblue::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d',1,'pocketfft::detail::pocketfft_c::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33',1,'pocketfft::detail::pocketfft_r::exec()'],['../classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058',1,'pocketfft::detail::T_dct1::exec()'],['../classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26',1,'pocketfft::detail::T_dst1::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac',1,'pocketfft::detail::T_dcst23::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708',1,'pocketfft::detail::T_dcst4::exec()'],['../classmlx_1_1core_1_1_jit_compiler.html#adcf98f940e1919388eaab907ea17a540',1,'mlx::core::JitCompiler::exec()']]], + ['exec_5fr_34',['exec_r',['../classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00',1,'pocketfft::detail::fftblue']]], + ['exp_35',['Exp',['../classmlx_1_1core_1_1_exp.html#a1d0a618cbb91ab29ef53b57ff6ed6e06',1,'mlx::core::Exp']]], + ['exp_36',['exp',['../namespacemlx_1_1core_1_1simd.html#a835d71dd0bb2f9494a397d9939696ec2',1,'mlx::core::simd::exp()'],['../namespacemetal.html#ac2a0b3618d922ac014baac8189d44650',1,'metal::exp()'],['../namespacemetal_1_1fast.html#ad3dbd387b63373c29e3449609f763ede',1,'metal::fast::exp()'],['../namespacemetal_1_1precise.html#a8d8d2d5700ce432b33cf47cf22528e8f',1,'metal::precise::exp()'],['../group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125',1,'mlx::core::exp()']]], + ['exp10_37',['exp10',['../namespacemetal.html#a4c63707d13c89364496a48906631c204',1,'metal::exp10()'],['../namespacemetal_1_1fast.html#a453122f982485cbb4e471b3ac282ee5e',1,'metal::fast::exp10()'],['../namespacemetal_1_1precise.html#af9addb343c967da3a83e9e123a8521fd',1,'metal::precise::exp10()']]], + ['exp2_38',['exp2',['../namespacemetal.html#a228201c20777848804a4d0589c1d33e7',1,'metal::exp2()'],['../namespacemetal_1_1fast.html#ac092b65a46720adaf22f6266671d2d71',1,'metal::fast::exp2()'],['../namespacemetal_1_1precise.html#a92a880bd2197efc0da0f8f0f7ec1e4c9',1,'metal::precise::exp2()']]], + ['expand_5fdims_39',['expand_dims',['../group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133',1,'mlx::core::expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a',1,'mlx::core::expand_dims(const array &a, int axis, StreamOrDevice s={})']]], + ['expanddims_40',['ExpandDims',['../classmlx_1_1core_1_1_expand_dims.html#aea2479ea4dd93941eb83a22e087983a8',1,'mlx::core::ExpandDims']]], + ['expm1_41',['Expm1',['../classmlx_1_1core_1_1_expm1.html#a47c2a1b2a4ef6bb07ba77c55ddddaec2',1,'mlx::core::Expm1']]], + ['expm1_42',['expm1',['../namespacemlx_1_1core_1_1simd.html#a9407980793ecff5d5eb19c9a2cbda1eb',1,'mlx::core::simd::expm1(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a464687a8809d0180035acc9af2943a94',1,'mlx::core::simd::expm1(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a8f73d1dac82177e0aeadaeda349c4f96',1,'mlx::core::simd::expm1(Simd< T, 1 > in)'],['../group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0',1,'mlx::core::expm1()']]], + ['expm1f_43',['expm1f',['../expm1f_8h.html#a87f66d30e185950f42ce3641783cdc40',1,'expm1f.h']]], + ['expm1f_5fscaled_5funchecked_44',['expm1f_scaled_unchecked',['../expm1f_8h.html#adf20e03405fba634ca8d01acac24592e',1,'expm1f.h']]], + ['export_5ffunction_45',['export_function',['../namespacemlx_1_1core.html#a2afa4ea816ac9317200fd5c964fc89dc',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, const Args &args, bool shapeless=false)'],['../namespacemlx_1_1core.html#ace51644e2aa72f8d56b86eaa0e1a68b7',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, const Kwargs &kwargs, bool shapeless=false)'],['../namespacemlx_1_1core.html#a9692d7bb6de3456abc535d0f4bac7a94',1,'mlx::core::export_function(const std::string &file, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, const Args &args, const Kwargs &kwargs, bool shapeless=false)']]], + ['export_5fto_5fdot_46',['export_to_dot',['../namespacemlx_1_1core.html#a30338cb7d259334e46dc7a4819716fa6',1,'mlx::core::export_to_dot(std::ostream &os, NodeNamer namer, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a57395bdf43d9c5c134e610c169222cca',1,'mlx::core::export_to_dot(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a839f94dbad44f0d37333006fc876b42e',1,'mlx::core::export_to_dot(std::ostream &os, Arrays &&... outputs)'],['../namespacemlx_1_1core.html#a12faa564e0d85a1daa77c08babb75261',1,'mlx::core::export_to_dot(std::ostream &os, NodeNamer namer, Arrays &&... outputs)']]], + ['exporter_47',['exporter',['../namespacemlx_1_1core.html#a0303e26b737c9fd197ed9caa90fd21a7',1,'mlx::core::exporter(const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#af2735df8513ecce88456585f5aea50f5',1,'mlx::core::exporter(const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, bool shapeless=false)'],['../namespacemlx_1_1core.html#af38d5718f517e50a590fdb3d63a90df1',1,'mlx::core::exporter(const std::string &path, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, bool shapeless=false)']]], + ['eye_48',['eye',['../group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3',1,'mlx::core::eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd',1,'mlx::core::eye(int n, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga61657db78ef35d41112d362c869c25d2',1,'mlx::core::eye(int n, int m, StreamOrDevice s={})'],['../group__ops.html#ga908a15b42834be498a46856c99dfc779',1,'mlx::core::eye(int n, int m, int k, StreamOrDevice s={})'],['../group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f',1,'mlx::core::eye(int n, StreamOrDevice s={})']]] ]; diff --git a/docs/build/html/search/functions_6.js b/docs/build/html/search/functions_6.js index b35378abb..de7906063 100644 --- a/docs/build/html/search/functions_6.js +++ b/docs/build/html/search/functions_6.js @@ -1,39 +1,36 @@ var searchData= [ ['fabs_0',['fabs',['../namespacemetal.html#a487eba718144be1325abcf66e109bb21',1,'metal::fabs()'],['../namespacemetal_1_1fast.html#a129fbd68c9df1a437e8959a25187f554',1,'metal::fast::fabs()'],['../namespacemetal_1_1precise.html#ae4c71d8bc8ef291036a7aaa05f8be3d1',1,'metal::precise::fabs()']]], - ['fast_5ferf_1',['fast_erf',['../namespacemlx_1_1core_1_1detail.html#a90c9f6149af5adf4e2a95608d5f7b790',1,'mlx::core::detail']]], - ['fast_5ferfinv_2',['fast_erfinv',['../namespacemlx_1_1core_1_1detail.html#ad0ff3975e4c96317df1a2de0f7a30c1d',1,'mlx::core::detail']]], - ['fast_5fexp_3',['fast_exp',['../namespacemlx_1_1core_1_1detail.html#a2726436fc72d4a3f0030c89579b4d374',1,'mlx::core::detail']]], - ['fdim_4',['fdim',['../namespacemetal.html#a85a560794be56d8116889c1ee2d78761',1,'metal::fdim()'],['../namespacemetal_1_1fast.html#a667df76100d5ea0ce5860ddae3e5a00b',1,'metal::fast::fdim()'],['../namespacemetal_1_1precise.html#af693e7c93de446e80dd1377f5e9e7260',1,'metal::precise::fdim()']]], - ['fence_5',['Fence',['../structmlx_1_1core_1_1metal_1_1_fence.html#a30bee4957ae595e04922952a8010fc79',1,'mlx::core::metal::Fence']]], - ['fft_6',['FFT',['../classmlx_1_1core_1_1_f_f_t.html#a0cdce626ed2c8eeeecc6949418437839',1,'mlx::core::FFT']]], - ['fft_7',['fft',['../namespacemlx_1_1core_1_1metal.html#a39f43360d9e916fcf7e86c919b419554',1,'mlx::core::metal::fft()'],['../backend_2metal_2kernels_2fft_8h.html#a4010b0e151e5f01e610e9c32234458c7',1,'fft(): fft.h'],['../namespacemlx_1_1core_1_1fft.html#ad672de5ca029a6925b05f03bbebe5ad3',1,'mlx::core::fft::fft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a3fe55b7b6eba32c4c8b2d206036216e0',1,'mlx::core::fft::fft(const array &a, int axis=-1, StreamOrDevice s={})']]], - ['fft2_8',['fft2',['../namespacemlx_1_1core_1_1fft.html#a7a318ed0ab6a600cd7cba96ecbd72a1d',1,'mlx::core::fft::fft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a6eb0c5f8b33694ddb56748a97d17e8b7',1,'mlx::core::fft::fft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], - ['fftblue_9',['fftblue',['../classpocketfft_1_1detail_1_1fftblue.html#a8613b2131e7843bbca9e1cd4cc15bc01',1,'pocketfft::detail::fftblue']]], - ['fftn_10',['fftn',['../namespacemlx_1_1core_1_1fft.html#a2c6685806eef1cae8d2da53567d23e5c',1,'mlx::core::fft::fftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#aaa116429c2cb5bab20b464be890252c8',1,'mlx::core::fft::fftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a039a44197ad299a15a5847639292800c',1,'mlx::core::fft::fftn(const array &a, StreamOrDevice s={})']]], - ['filewriter_11',['FileWriter',['../classmlx_1_1core_1_1io_1_1_file_writer.html#a40b241ad540ee4aadc3a19a6b1ccfb4d',1,'mlx::core::io::FileWriter::FileWriter(std::string file_path)'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#aee57db8516361f17de3cf2087d9a87d9',1,'mlx::core::io::FileWriter::FileWriter(const FileWriter &)=delete'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a12b148df8a52136628728b508ee9c55e',1,'mlx::core::io::FileWriter::FileWriter(FileWriter &&other)']]], - ['fill_5fgpu_12',['fill_gpu',['../namespacemlx_1_1core.html#ae789dbda2a0f4e21aa0984f6a5dc986c',1,'mlx::core']]], - ['finfo_13',['finfo',['../structmlx_1_1core_1_1finfo.html#a00dee158d75d12768d02a3e7b6709109',1,'mlx::core::finfo']]], - ['flags_14',['flags',['../classmlx_1_1core_1_1array.html#a0a20a6065ae71b64c1e3aa22a45fd8a1',1,'mlx::core::array']]], - ['flatten_15',['Flatten',['../classmlx_1_1core_1_1_flatten.html#ab9f72c6a90640b91f35a2bcc8dac8780',1,'mlx::core::Flatten']]], - ['flatten_16',['flatten',['../group__ops.html#ga50aa98754b412bb57c083f6e3e95061f',1,'mlx::core::flatten(const array &a, int start_axis, int end_axis=-1, StreamOrDevice s={})'],['../group__ops.html#gaa6adbc9c86f0ab27d8810a02e9e719fd',1,'mlx::core::flatten(const array &a, StreamOrDevice s={})']]], - ['float_5fto_5fbfloat_5fbits_17',['float_to_bfloat_bits',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31ce5e8e860295fa236e0d4b0befeae1',1,'bf16.h']]], - ['floor_18',['Floor',['../classmlx_1_1core_1_1_floor.html#ada4e979b784b732696313d7094e91340',1,'mlx::core::Floor']]], - ['floor_19',['floor',['../namespacemetal.html#a020790f30c28a9982c4a83deaa258277',1,'metal::floor()'],['../namespacemetal_1_1fast.html#ac012ce1701c2339914f15cce9f2c632f',1,'metal::fast::floor()'],['../namespacemetal_1_1precise.html#a66e02b028e3cecfe7c80773460dc7925',1,'metal::precise::floor()'],['../group__ops.html#ga8d656904aa2690b60955ae745aecfc30',1,'mlx::core::floor(const array &a, StreamOrDevice s={})']]], - ['floor_5fdivide_20',['floor_divide',['../group__ops.html#ga05b4c6054d028107869511f927da01cd',1,'mlx::core']]], - ['fma_21',['fma',['../namespacemetal.html#a6301a78d69ff14a06194ca85a0c7d326',1,'metal::fma()'],['../namespacemetal_1_1fast.html#aebcd6e951da6f7157ec219eb7a8f1ddd',1,'metal::fast::fma()'],['../namespacemetal_1_1precise.html#a49391a64d6b66fe3a212516b316a2144',1,'metal::precise::fma()']]], - ['fmax_22',['fmax',['../namespacemetal.html#a0558e56fdb94b456deea6a4eb53964ed',1,'metal::fmax()'],['../namespacemetal_1_1fast.html#a26e3257cf877154f8a0d434be0bdb034',1,'metal::fast::fmax()'],['../namespacemetal_1_1precise.html#ac7d49f921c2883caf9eec66efc4de1cd',1,'metal::precise::fmax()']]], - ['fmax3_23',['fmax3',['../namespacemetal.html#ae0c1a7ba1a7449adc64d00b2a29e67f6',1,'metal::fmax3()'],['../namespacemetal_1_1fast.html#a5c6a3a389f348e1f92e8392b765a32c7',1,'metal::fast::fmax3()'],['../namespacemetal_1_1precise.html#adf750e51bd83d569994d0967029e3bdc',1,'metal::precise::fmax3()']]], - ['fmedian3_24',['fmedian3',['../namespacemetal.html#aa35227450d943fb88cf43162aa9d8c49',1,'metal::fmedian3()'],['../namespacemetal_1_1fast.html#a923869181c3f576f2d86fba5bfa85633',1,'metal::fast::fmedian3()'],['../namespacemetal_1_1precise.html#a48d1d0be889de4043b775bb6b030a989',1,'metal::precise::fmedian3()']]], - ['fmin_25',['fmin',['../namespacemetal.html#a66ac19825ea79b8294e243ae6d0b3d3c',1,'metal::fmin()'],['../namespacemetal_1_1fast.html#a7e202ec52bf12bfabdf2265b300acbfa',1,'metal::fast::fmin()'],['../namespacemetal_1_1precise.html#a18df8eb481dfa56c92ad31b5bab8e069',1,'metal::precise::fmin()']]], - ['fmin3_26',['fmin3',['../namespacemetal.html#ae2acd25f2241f00aaf89ff48f132a879',1,'metal::fmin3()'],['../namespacemetal_1_1fast.html#a9531c6a4a520927523961e6eb6b94c1a',1,'metal::fast::fmin3()'],['../namespacemetal_1_1precise.html#a5bb710e6742996d32225a8f54a0f116c',1,'metal::precise::fmin3()']]], - ['fmod_27',['fmod',['../namespacemetal.html#a2ff952d4d596a7969b2a3035fc2fda58',1,'metal::fmod()'],['../namespacemetal_1_1fast.html#adbec09f18a89f773d7e368ef04a69526',1,'metal::fast::fmod()'],['../namespacemetal_1_1precise.html#aa99937178a1fc8158054e328eeeae648',1,'metal::precise::fmod()']]], - ['four_5fstep_5ffft_28',['four_step_fft',['../backend_2metal_2kernels_2fft_8h.html#a6558a8205ee4c3e4767bafa93f7606de',1,'fft.h']]], - ['fract_29',['fract',['../namespacemetal.html#a6b1c15d251aeaacb1f4338a5e152ae78',1,'metal::fract()'],['../namespacemetal_1_1fast.html#aa8bb448827503e485eb649eb3edb2d4c',1,'metal::fast::fract()'],['../namespacemetal_1_1precise.html#a0f21c19332a90df1a8ff507a813b5757',1,'metal::precise::fract()']]], - ['frag_5fat_30',['frag_at',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1a6b1446e8c8da46885bbaa8e8fdc7e4',1,'mlx::steel::MMATile::frag_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad476e1d9a12178fb35c207312339e485',1,'mlx::steel::MMATile::frag_at(const short i, const short j) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1a6b1446e8c8da46885bbaa8e8fdc7e4',1,'mlx::steel::MMATile::frag_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad476e1d9a12178fb35c207312339e485',1,'mlx::steel::MMATile::frag_at(const short i, const short j) const']]], - ['free_31',['free',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#ae963d551be646ae0e13df2c16f2beefb',1,'mlx::core::allocator::Allocator::free()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#a84b50d1a3cbffa12c1a6cf0ed8c71079',1,'mlx::core::allocator::CommonAllocator::free()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a109a0a37fb0b3be381a62dc3b1a54bf0',1,'mlx::core::metal::MetalAllocator::free()'],['../namespacemlx_1_1core_1_1allocator.html#a77f0a1215be242db6485612bcb273af5',1,'mlx::core::allocator::free()']]], - ['frexp_32',['frexp',['../namespacemetal.html#ac89d4ef524d21a301da6c37dbd95ff9f',1,'metal::frexp()'],['../namespacemetal_1_1fast.html#a23902df22aeaa859ef673a36381387c2',1,'metal::fast::frexp()'],['../namespacemetal_1_1precise.html#a0fbb1624c308b97380f894f92fd858b4',1,'metal::precise::frexp()']]], - ['full_33',['Full',['../classmlx_1_1core_1_1_full.html#aafcb86a2e41353853ec48c717e0c54d6',1,'mlx::core::Full']]], - ['full_34',['full',['../group__ops.html#ga1cf232308668fe3f4214c8b895ed4aee',1,'mlx::core::full(Shape shape, array vals, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga59f6c844cbb173e108c3eeb11801f8c6',1,'mlx::core::full(Shape shape, array vals, StreamOrDevice s={})'],['../group__ops.html#gaf073760b7b51fe35932da0d81c531a55',1,'mlx::core::full(Shape shape, T val, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaf6f2cce92aff9b71756a3cc3c961fd5a',1,'mlx::core::full(Shape shape, T val, StreamOrDevice s={})']]], - ['functionexporter_35',['FunctionExporter',['../structmlx_1_1core_1_1_function_exporter.html#a97ff954496a084d96e73a9c520c9dc0c',1,'mlx::core::FunctionExporter::FunctionExporter(const FunctionExporter &)=delete'],['../structmlx_1_1core_1_1_function_exporter.html#ac317e349139f8a6cd70d63ef65368fc2',1,'mlx::core::FunctionExporter::FunctionExporter(FunctionExporter &&other)=default']]] + ['fdim_1',['fdim',['../namespacemetal.html#a85a560794be56d8116889c1ee2d78761',1,'metal::fdim()'],['../namespacemetal_1_1fast.html#a667df76100d5ea0ce5860ddae3e5a00b',1,'metal::fast::fdim()'],['../namespacemetal_1_1precise.html#af693e7c93de446e80dd1377f5e9e7260',1,'metal::precise::fdim()']]], + ['fence_2',['Fence',['../structmlx_1_1core_1_1metal_1_1_fence.html#a30bee4957ae595e04922952a8010fc79',1,'mlx::core::metal::Fence::Fence()'],['../classmlx_1_1core_1_1_fence.html#a3e3ed08eb6a1025b051b5a435f6f95f7',1,'mlx::core::Fence::Fence()']]], + ['fft_3',['FFT',['../classmlx_1_1core_1_1_f_f_t.html#a0cdce626ed2c8eeeecc6949418437839',1,'mlx::core::FFT']]], + ['fft_4',['fft',['../namespacemlx_1_1core_1_1metal.html#a39f43360d9e916fcf7e86c919b419554',1,'mlx::core::metal::fft()'],['../backend_2metal_2kernels_2fft_8h.html#a4010b0e151e5f01e610e9c32234458c7',1,'fft(): fft.h'],['../namespacemlx_1_1core_1_1fft.html#ad672de5ca029a6925b05f03bbebe5ad3',1,'mlx::core::fft::fft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a3fe55b7b6eba32c4c8b2d206036216e0',1,'mlx::core::fft::fft(const array &a, int axis=-1, StreamOrDevice s={})']]], + ['fft2_5',['fft2',['../namespacemlx_1_1core_1_1fft.html#a7a318ed0ab6a600cd7cba96ecbd72a1d',1,'mlx::core::fft::fft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a6eb0c5f8b33694ddb56748a97d17e8b7',1,'mlx::core::fft::fft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], + ['fftblue_6',['fftblue',['../classpocketfft_1_1detail_1_1fftblue.html#a8613b2131e7843bbca9e1cd4cc15bc01',1,'pocketfft::detail::fftblue']]], + ['fftn_7',['fftn',['../namespacemlx_1_1core_1_1fft.html#a2c6685806eef1cae8d2da53567d23e5c',1,'mlx::core::fft::fftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#aaa116429c2cb5bab20b464be890252c8',1,'mlx::core::fft::fftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a039a44197ad299a15a5847639292800c',1,'mlx::core::fft::fftn(const array &a, StreamOrDevice s={})']]], + ['filewriter_8',['FileWriter',['../classmlx_1_1core_1_1io_1_1_file_writer.html#a40b241ad540ee4aadc3a19a6b1ccfb4d',1,'mlx::core::io::FileWriter::FileWriter(std::string file_path)'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#aee57db8516361f17de3cf2087d9a87d9',1,'mlx::core::io::FileWriter::FileWriter(const FileWriter &)=delete'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a12b148df8a52136628728b508ee9c55e',1,'mlx::core::io::FileWriter::FileWriter(FileWriter &&other)']]], + ['fill_5fgpu_9',['fill_gpu',['../namespacemlx_1_1core.html#ae789dbda2a0f4e21aa0984f6a5dc986c',1,'mlx::core']]], + ['finfo_10',['finfo',['../structmlx_1_1core_1_1finfo.html#a00dee158d75d12768d02a3e7b6709109',1,'mlx::core::finfo']]], + ['flags_11',['flags',['../classmlx_1_1core_1_1array.html#a0a20a6065ae71b64c1e3aa22a45fd8a1',1,'mlx::core::array']]], + ['flatten_12',['Flatten',['../classmlx_1_1core_1_1_flatten.html#ab9f72c6a90640b91f35a2bcc8dac8780',1,'mlx::core::Flatten']]], + ['flatten_13',['flatten',['../group__ops.html#ga50aa98754b412bb57c083f6e3e95061f',1,'mlx::core::flatten(const array &a, int start_axis, int end_axis=-1, StreamOrDevice s={})'],['../group__ops.html#gaa6adbc9c86f0ab27d8810a02e9e719fd',1,'mlx::core::flatten(const array &a, StreamOrDevice s={})']]], + ['float_5fto_5fbfloat_5fbits_14',['float_to_bfloat_bits',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31ce5e8e860295fa236e0d4b0befeae1',1,'bf16.h']]], + ['floor_15',['Floor',['../classmlx_1_1core_1_1_floor.html#ada4e979b784b732696313d7094e91340',1,'mlx::core::Floor']]], + ['floor_16',['floor',['../namespacemlx_1_1core_1_1simd.html#a8e22c484298d9af10b6604c835e52052',1,'mlx::core::simd::floor(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#aa396efa6e9c94f4ac1f8381d5e07f069',1,'mlx::core::simd::floor(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#ad6b89aecafefe57b6ce69bec143ccd6e',1,'mlx::core::simd::floor(Simd< float16_t, N > a)'],['../namespacemetal.html#a020790f30c28a9982c4a83deaa258277',1,'metal::floor()'],['../namespacemetal_1_1fast.html#ac012ce1701c2339914f15cce9f2c632f',1,'metal::fast::floor()'],['../namespacemetal_1_1precise.html#a66e02b028e3cecfe7c80773460dc7925',1,'metal::precise::floor()'],['../group__ops.html#ga8d656904aa2690b60955ae745aecfc30',1,'mlx::core::floor(const array &a, StreamOrDevice s={})']]], + ['floor_5fdivide_17',['floor_divide',['../group__ops.html#ga05b4c6054d028107869511f927da01cd',1,'mlx::core']]], + ['fma_18',['fma',['../namespacemlx_1_1core_1_1simd.html#a9ddc7f119cc1dc04372ec1adcaf55f70',1,'mlx::core::simd::fma(Simd< T, N > x, Simd< T, N > y, U z)'],['../namespacemlx_1_1core_1_1simd.html#a8aa81ebff4c26f21cae2253d885fd87a',1,'mlx::core::simd::fma(Simd< T, 1 > x, Simd< T, 1 > y, U z)'],['../namespacemlx_1_1core_1_1simd.html#a99099c338377518773b55d4042f9410d',1,'mlx::core::simd::fma(Simd< float16_t, N > x, Simd< float16_t, N > y, T z)'],['../namespacemetal.html#a6301a78d69ff14a06194ca85a0c7d326',1,'metal::fma()'],['../namespacemetal_1_1fast.html#aebcd6e951da6f7157ec219eb7a8f1ddd',1,'metal::fast::fma()'],['../namespacemetal_1_1precise.html#a49391a64d6b66fe3a212516b316a2144',1,'metal::precise::fma()']]], + ['fmax_19',['fmax',['../namespacemetal.html#a0558e56fdb94b456deea6a4eb53964ed',1,'metal::fmax()'],['../namespacemetal_1_1fast.html#a26e3257cf877154f8a0d434be0bdb034',1,'metal::fast::fmax()'],['../namespacemetal_1_1precise.html#ac7d49f921c2883caf9eec66efc4de1cd',1,'metal::precise::fmax()']]], + ['fmax3_20',['fmax3',['../namespacemetal.html#ae0c1a7ba1a7449adc64d00b2a29e67f6',1,'metal::fmax3()'],['../namespacemetal_1_1fast.html#a5c6a3a389f348e1f92e8392b765a32c7',1,'metal::fast::fmax3()'],['../namespacemetal_1_1precise.html#adf750e51bd83d569994d0967029e3bdc',1,'metal::precise::fmax3()']]], + ['fmedian3_21',['fmedian3',['../namespacemetal.html#aa35227450d943fb88cf43162aa9d8c49',1,'metal::fmedian3()'],['../namespacemetal_1_1fast.html#a923869181c3f576f2d86fba5bfa85633',1,'metal::fast::fmedian3()'],['../namespacemetal_1_1precise.html#a48d1d0be889de4043b775bb6b030a989',1,'metal::precise::fmedian3()']]], + ['fmin_22',['fmin',['../namespacemetal.html#a66ac19825ea79b8294e243ae6d0b3d3c',1,'metal::fmin()'],['../namespacemetal_1_1fast.html#a7e202ec52bf12bfabdf2265b300acbfa',1,'metal::fast::fmin()'],['../namespacemetal_1_1precise.html#a18df8eb481dfa56c92ad31b5bab8e069',1,'metal::precise::fmin()']]], + ['fmin3_23',['fmin3',['../namespacemetal.html#ae2acd25f2241f00aaf89ff48f132a879',1,'metal::fmin3()'],['../namespacemetal_1_1fast.html#a9531c6a4a520927523961e6eb6b94c1a',1,'metal::fast::fmin3()'],['../namespacemetal_1_1precise.html#a5bb710e6742996d32225a8f54a0f116c',1,'metal::precise::fmin3()']]], + ['fmod_24',['fmod',['../namespacemetal.html#a2ff952d4d596a7969b2a3035fc2fda58',1,'metal::fmod()'],['../namespacemetal_1_1fast.html#adbec09f18a89f773d7e368ef04a69526',1,'metal::fast::fmod()'],['../namespacemetal_1_1precise.html#aa99937178a1fc8158054e328eeeae648',1,'metal::precise::fmod()']]], + ['four_5fstep_5ffft_25',['four_step_fft',['../backend_2metal_2kernels_2fft_8h.html#a6558a8205ee4c3e4767bafa93f7606de',1,'fft.h']]], + ['fract_26',['fract',['../namespacemetal.html#a6b1c15d251aeaacb1f4338a5e152ae78',1,'metal::fract()'],['../namespacemetal_1_1fast.html#aa8bb448827503e485eb649eb3edb2d4c',1,'metal::fast::fract()'],['../namespacemetal_1_1precise.html#a0f21c19332a90df1a8ff507a813b5757',1,'metal::precise::fract()']]], + ['frag_5fat_27',['frag_at',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1a6b1446e8c8da46885bbaa8e8fdc7e4',1,'mlx::steel::MMATile::frag_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad476e1d9a12178fb35c207312339e485',1,'mlx::steel::MMATile::frag_at(const short i, const short j) const'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1a6b1446e8c8da46885bbaa8e8fdc7e4',1,'mlx::steel::MMATile::frag_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad476e1d9a12178fb35c207312339e485',1,'mlx::steel::MMATile::frag_at(const short i, const short j) const']]], + ['free_28',['free',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#ae963d551be646ae0e13df2c16f2beefb',1,'mlx::core::allocator::Allocator::free()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#a84b50d1a3cbffa12c1a6cf0ed8c71079',1,'mlx::core::allocator::CommonAllocator::free()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a109a0a37fb0b3be381a62dc3b1a54bf0',1,'mlx::core::metal::MetalAllocator::free()'],['../namespacemlx_1_1core_1_1allocator.html#a77f0a1215be242db6485612bcb273af5',1,'mlx::core::allocator::free()']]], + ['frexp_29',['frexp',['../namespacemetal.html#ac89d4ef524d21a301da6c37dbd95ff9f',1,'metal::frexp()'],['../namespacemetal_1_1fast.html#a23902df22aeaa859ef673a36381387c2',1,'metal::fast::frexp()'],['../namespacemetal_1_1precise.html#a0fbb1624c308b97380f894f92fd858b4',1,'metal::precise::frexp()']]], + ['full_30',['Full',['../classmlx_1_1core_1_1_full.html#aafcb86a2e41353853ec48c717e0c54d6',1,'mlx::core::Full']]], + ['full_31',['full',['../group__ops.html#ga1cf232308668fe3f4214c8b895ed4aee',1,'mlx::core::full(Shape shape, array vals, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga59f6c844cbb173e108c3eeb11801f8c6',1,'mlx::core::full(Shape shape, array vals, StreamOrDevice s={})'],['../group__ops.html#gaf073760b7b51fe35932da0d81c531a55',1,'mlx::core::full(Shape shape, T val, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaf6f2cce92aff9b71756a3cc3c961fd5a',1,'mlx::core::full(Shape shape, T val, StreamOrDevice s={})']]], + ['functionexporter_32',['FunctionExporter',['../structmlx_1_1core_1_1_function_exporter.html#a97ff954496a084d96e73a9c520c9dc0c',1,'mlx::core::FunctionExporter::FunctionExporter(const FunctionExporter &)=delete'],['../structmlx_1_1core_1_1_function_exporter.html#ac317e349139f8a6cd70d63ef65368fc2',1,'mlx::core::FunctionExporter::FunctionExporter(FunctionExporter &&other)=default']]] ]; diff --git a/docs/build/html/search/functions_7.js b/docs/build/html/search/functions_7.js index 4544fdc2e..c0a6872e0 100644 --- a/docs/build/html/search/functions_7.js +++ b/docs/build/html/search/functions_7.js @@ -2,80 +2,84 @@ var searchData= [ ['gather_0',['Gather',['../classmlx_1_1core_1_1_gather.html#af24220fde798f2ad17cdce297c0dbc43',1,'mlx::core::Gather']]], ['gather_1',['gather',['../namespacemlx_1_1core_1_1metal.html#a545de371fefba1feec2e70b7e9f4187c',1,'mlx::core::metal::gather()'],['../group__ops.html#ga8fcc3ad0677c834c36b72d5b2ebba6d0',1,'mlx::core::gather(const array &a, const std::vector< array > &indices, const std::vector< int > &axes, const Shape &slice_sizes, StreamOrDevice s={})'],['../group__ops.html#gafe2bd174c9953ed7f12664f7abaca0e6',1,'mlx::core::gather(const array &a, const array &indices, int axis, const Shape &slice_sizes, StreamOrDevice s={})']]], - ['gather_5fimpl_2',['gather_impl',['../gather_8h.html#a5cf4ff9849edd2d99ff24ab4a6c9cd5d',1,'gather.h']]], - ['gather_5fmm_3',['gather_mm',['../group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7',1,'mlx::core']]], - ['gather_5fqmm_4',['gather_qmm',['../group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0',1,'mlx::core']]], - ['gathermm_5',['GatherMM',['../classmlx_1_1core_1_1_gather_m_m.html#afd9bbc08138181b80e2fb86536ff3f2a',1,'mlx::core::GatherMM']]], - ['gatherqmm_6',['GatherQMM',['../classmlx_1_1core_1_1_gather_q_m_m.html#a60ed2ade7f10dd9c9314913a810f9360',1,'mlx::core::GatherQMM']]], - ['gemm_7',['gemm',['../namespacemlx_1_1core_1_1metal.html#ac46fd23516a61fc56d997910e4144281',1,'mlx::core::metal::gemm()'],['../steel__gemm__fused_8h.html#a70d5b54bfae6471242d4e85675c929fd',1,'gemm(): steel_gemm_fused.h']]], - ['gemm_5floop_8',['gemm_loop',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a756d7bbcc96e2919cd65eec4bc135780',1,'mlx::steel::GEMMKernel::gemm_loop(threadgroup T *As, threadgroup T *Bs, const int gemm_k_iterations, thread loader_a_t &loader_a, thread loader_b_t &loader_b, thread mma_t &mma_op, thread const short &tgp_bm, thread const short &tgp_bn, thread const short &lbk, LoopAlignment< M_aligned, N_aligned, K_aligned_ > l={})'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a756d7bbcc96e2919cd65eec4bc135780',1,'mlx::steel::GEMMKernel::gemm_loop(threadgroup T *As, threadgroup T *Bs, const int gemm_k_iterations, thread loader_a_t &loader_a, thread loader_b_t &loader_b, thread mma_t &mma_op, thread const short &tgp_bm, thread const short &tgp_bn, thread const short &lbk, LoopAlignment< M_aligned, N_aligned, K_aligned_ > l={})']]], - ['gemm_5fsplitk_9',['gemm_splitk',['../steel__gemm__splitk_8h.html#a3be6e095a0a026d3ecf57a3e67f76188',1,'steel_gemm_splitk.h']]], - ['gemm_5fsplitk_5faccum_10',['gemm_splitk_accum',['../steel__gemm__splitk_8h.html#abeb921bf1dc7941125188ddd390b0907',1,'steel_gemm_splitk.h']]], - ['gemm_5fsplitk_5faccum_5faxpby_11',['gemm_splitk_accum_axpby',['../steel__gemm__splitk_8h.html#acc33fdfaaf3eb3a0629b3d52c7043dc1',1,'steel_gemm_splitk.h']]], - ['gemv_5fmasked_12',['gemv_masked',['../namespacemlx_1_1core_1_1metal.html#abc055b75e6a059618f279c35f8de36e7',1,'mlx::core::metal::gemv_masked()'],['../kernels_2gemv__masked_8h.html#af890b6ac155165f8ee0c600363938341',1,'gemv_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant int64_t *vector_batch_stride, const constant int64_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant int64_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid): gemv_masked.h']]], - ['gemv_5ft_5fmasked_13',['gemv_t_masked',['../kernels_2gemv__masked_8h.html#ae5b4a5124ddf92a984258a0be1ff0f4f',1,'gemv_masked.h']]], - ['general_5fc2r_14',['general_c2r',['../namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc',1,'pocketfft::detail']]], - ['general_5fnd_15',['general_nd',['../namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0',1,'pocketfft::detail']]], - ['general_5fr2c_16',['general_r2c',['../namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed',1,'pocketfft::detail']]], - ['get_5f2d_5fgrid_5fdims_17',['get_2d_grid_dims',['../namespacemlx_1_1core.html#aa9692de582995fd3ce19493b45ab7144',1,'mlx::core::get_2d_grid_dims(const Shape &shape, const Strides &strides)'],['../namespacemlx_1_1core.html#a6ec5cdf3253a9f20ca5ea7a1590fb386',1,'mlx::core::get_2d_grid_dims(const Shape &shape, const Strides &strides, size_t divisor)']]], - ['get_5factive_5fmemory_18',['get_active_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a7a3ad4e33d57a47474c98e2f88e775d7',1,'mlx::core::metal::MetalAllocator::get_active_memory()'],['../namespacemlx_1_1core_1_1metal.html#a7b75c2639016ac4d350fa6c9da386667',1,'mlx::core::metal::get_active_memory()']]], - ['get_5farange_5fkernel_19',['get_arange_kernel',['../namespacemlx_1_1core.html#a76f614e9956a6ca05a9be4db5a483446',1,'mlx::core']]], - ['get_5farchitecture_20',['get_architecture',['../classmlx_1_1core_1_1metal_1_1_device.html#a65f64dd8bafdc704d871fc5be5e7bc0b',1,'mlx::core::metal::Device']]], - ['get_5fbinary_5fkernel_21',['get_binary_kernel',['../namespacemlx_1_1core.html#a4decd4a07d91487e6903f6e3c8b7513a',1,'mlx::core']]], - ['get_5fbinary_5ftwo_5fkernel_22',['get_binary_two_kernel',['../namespacemlx_1_1core.html#a4e809746f48e5dcf7fa63215d3f5e33e',1,'mlx::core']]], - ['get_5fblock_5fdims_23',['get_block_dims',['../namespacemlx_1_1core.html#a0f0f59d3ffe2d16a684e5fc093302e15',1,'mlx::core']]], - ['get_5fcache_5fmemory_24',['get_cache_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ad3cabbe638917ca4114eb74dcabe381f',1,'mlx::core::metal::MetalAllocator::get_cache_memory()'],['../namespacemlx_1_1core_1_1metal.html#a43307654f62ed7c58e014be7fb03909c',1,'mlx::core::metal::get_cache_memory()']]], - ['get_5fcolocated_5fmtllib_5fpath_25',['get_colocated_mtllib_path',['../namespacemlx_1_1core_1_1metal.html#a5fd6ba2040e53a254b9d71ae7ebd315f',1,'mlx::core::metal']]], - ['get_5fcommand_5fbuffer_26',['get_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a5fe3970fbe92ccc55fce4241ffbe5210',1,'mlx::core::metal::Device']]], - ['get_5fcommand_5fbuffer_5fops_27',['get_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a064e1cb6a16de7a0619f6447622350f8',1,'mlx::core::metal::Device']]], - ['get_5fcommand_5fencoder_28',['get_command_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#affa682ef612def4890f5152f81ffb7e6',1,'mlx::core::metal::Device']]], - ['get_5fcoord_29',['get_coord',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a7331fff1d12f2f8b72b0006a3ad0dd83',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord(ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a7331fff1d12f2f8b72b0006a3ad0dd83',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord(ushort simd_lane_id)']]], - ['get_5fcopy_5fkernel_30',['get_copy_kernel',['../namespacemlx_1_1core.html#a05a220cff45f12439fde775983c6df78',1,'mlx::core']]], - ['get_5fdefault_5fstream_31',['get_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a2366c7b888e433608e203752edc92282',1,'mlx::core::scheduler::Scheduler']]], - ['get_5fdynamic_5fcopy_5fkernel_32',['get_dynamic_copy_kernel',['../namespacemlx_1_1core.html#addb29b3e6771875f2aecd035ed560461',1,'mlx::core']]], - ['get_5ffft_5fkernel_33',['get_fft_kernel',['../namespacemlx_1_1core.html#a1d4cffc3c78067b3d9a62d64f3fb686f',1,'mlx::core']]], - ['get_5fgemv_5fmasked_5fkernel_34',['get_gemv_masked_kernel',['../namespacemlx_1_1core.html#a90c24e0d0b99b68fad9deefcf4d3e818',1,'mlx::core']]], - ['get_5fglobal_5fformatter_35',['get_global_formatter',['../namespacemlx_1_1core.html#afc71e62dc5757564486cea5ebb12500e',1,'mlx::core']]], - ['get_5fkernel_36',['get_kernel',['../classmlx_1_1core_1_1metal_1_1_device.html#a6810c4dcbcfbf93fc51d42aa5ff0fc3a',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, MTL::Library *mtl_lib, const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#afa0cac9d800c21a8a7f6cb224256abaf',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, const std::string &lib_name="mlx", const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})']]], - ['get_5fkernel_5fpreamble_37',['get_kernel_preamble',['../compiled__preamble_8h.html#a1dfa17a0369fb90ff615c7461f5013f3',1,'compiled_preamble.h']]], - ['get_5flibrary_38',['get_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a75ed55e73baf48013028796518723ff0',1,'mlx::core::metal::Device']]], - ['get_5fmb_5fsort_5fkernel_39',['get_mb_sort_kernel',['../namespacemlx_1_1core.html#afb57825bb763050cc9a9d194aa41ac36',1,'mlx::core']]], - ['get_5fname_40',['get_name',['../structmlx_1_1core_1_1_node_namer.html#a1690dd38de288c0aee2bb53156eb770e',1,'mlx::core::NodeNamer']]], - ['get_5fpeak_5fmemory_41',['get_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ac7972a3fe58e69489de775a0f152da17',1,'mlx::core::metal::MetalAllocator::get_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#a4b67d680cefa95f0ed5801f0e14e48ce',1,'mlx::core::metal::get_peak_memory()']]], - ['get_5fplan_42',['get_plan',['../namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b',1,'pocketfft::detail']]], - ['get_5fpool_43',['get_pool',['../namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a',1,'pocketfft::detail::threading']]], - ['get_5fprimitive_5fstring_44',['get_primitive_string',['../namespacemlx_1_1core.html#ad4be35b310a252edd80d9cf04f094a60',1,'mlx::core']]], - ['get_5fquantized_5fkernel_45',['get_quantized_kernel',['../namespacemlx_1_1core.html#aa3faeae5378bfaafe3ce3432a051e43e',1,'mlx::core']]], - ['get_5freduce_5finit_5fkernel_46',['get_reduce_init_kernel',['../namespacemlx_1_1core.html#ae0470605dc819efeb6510183619f0299',1,'mlx::core']]], - ['get_5freduce_5fkernel_47',['get_reduce_kernel',['../namespacemlx_1_1core.html#a1be32ba7d67137dde7ac191dfe83ff49',1,'mlx::core']]], - ['get_5freduction_5fplan_48',['get_reduction_plan',['../namespacemlx_1_1core.html#ac97b5a6f009ca3d99854ce9512c20dba',1,'mlx::core']]], - ['get_5fscan_5fkernel_49',['get_scan_kernel',['../namespacemlx_1_1core.html#aeefaff208444d3fa61ecc0946fe1de5f',1,'mlx::core']]], - ['get_5fshape_50',['get_shape',['../namespacemlx_1_1core.html#a30fb38e05feeee19ae2b87e62bff3acf',1,'mlx::core']]], - ['get_5fsoftmax_5fkernel_51',['get_softmax_kernel',['../namespacemlx_1_1core.html#a35a412f688d79eb47e42d20a7c8650ee',1,'mlx::core']]], - ['get_5fsort_5fkernel_52',['get_sort_kernel',['../namespacemlx_1_1core.html#a84ebe6275218070f0ea320f126f64e22',1,'mlx::core']]], - ['get_5fsteel_5fconv_5fgeneral_5fkernel_53',['get_steel_conv_general_kernel',['../namespacemlx_1_1core.html#abce2b67044ee06a7bbe7a91ec7c8c48d',1,'mlx::core']]], - ['get_5fsteel_5fconv_5fkernel_54',['get_steel_conv_kernel',['../namespacemlx_1_1core.html#adce79d220672f5f3c65cc31d145ca9c4',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5ffused_5fkernel_55',['get_steel_gemm_fused_kernel',['../namespacemlx_1_1core.html#a84fa8e0aee321a9d614433a0b933103b',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fmasked_5fkernel_56',['get_steel_gemm_masked_kernel',['../namespacemlx_1_1core.html#ab5f60614e965144b451930fdf935e08d',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fsplitk_5faccum_5fkernel_57',['get_steel_gemm_splitk_accum_kernel',['../namespacemlx_1_1core.html#a195b86cad5bb99aa1bcd23952305af6b',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fsplitk_5fkernel_58',['get_steel_gemm_splitk_kernel',['../namespacemlx_1_1core.html#af48c6f2f72b61dbd6766e4f5fea85df5',1,'mlx::core']]], - ['get_5fstream_59',['get_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#aa6726858b362c7cd1f8a846a63085dbc',1,'mlx::core::scheduler::Scheduler::get_stream()'],['../namespacemlx_1_1core.html#ab436b8c08be2be32ef61bd72f7df63cd',1,'mlx::core::get_stream(int index)']]], - ['get_5ftemplate_5fdefinition_60',['get_template_definition',['../namespacemlx_1_1core.html#aae0d19f0acdef2accd2428fb84c8a032',1,'mlx::core']]], - ['get_5fternary_5fkernel_61',['get_ternary_kernel',['../namespacemlx_1_1core.html#a54eb3b65375022428aab5f810e40624b',1,'mlx::core']]], - ['get_5ftwiddle_62',['get_twiddle',['../radix_8h.html#ac5cf950316b9445296ee9ecfc56a56bd',1,'radix.h']]], - ['get_5ftype_5fstring_63',['get_type_string',['../namespacemlx_1_1core.html#af776fd91dd60594dcfebbafd17f19068',1,'mlx::core']]], - ['get_5funary_5fkernel_64',['get_unary_kernel',['../namespacemlx_1_1core.html#afbb085188b563a54606d84f87a9bf5a6',1,'mlx::core']]], - ['get_5fvar_65',['get_var',['../namespacemlx_1_1core_1_1env.html#a0efecbf9efe695adafad12b5a4945df3',1,'mlx::core::env']]], - ['gguf_5fload_5fquantized_66',['gguf_load_quantized',['../namespacemlx_1_1core.html#a65dd68163bdaef3631e3724327782498',1,'mlx::core']]], - ['good_67',['good',['../classmlx_1_1core_1_1io_1_1_reader.html#a005d0b52c1f34866f7412b7f41dabec3',1,'mlx::core::io::Reader::good()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a0b050c2c27487007e250e2e19560ffe4',1,'mlx::core::io::Writer::good()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#ac54a2c693acc3d9e6e942412148ffcc9',1,'mlx::core::io::ParallelFileReader::good()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9ec4934b26fb358d699ddce1482b2d54',1,'mlx::core::io::FileWriter::good()']]], - ['good_5fsize_5fcmplx_68',['good_size_cmplx',['../structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624',1,'pocketfft::detail::util']]], - ['good_5fsize_5freal_69',['good_size_real',['../structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa',1,'pocketfft::detail::util']]], - ['grad_70',['grad',['../namespacemlx_1_1core.html#a3d2b2929ed4636e9e2b86e125b2e57d9',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#af482f6c64acd77c57ef5bb4b7be9726c',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a64bc619876b0f8cc81a2637ca81c99f7',1,'mlx::core::grad(const std::function< array(const array &)> &fun)']]], - ['greater_71',['Greater',['../classmlx_1_1core_1_1_greater.html#a1d5992a66c020cd97a70e8e3d8cd1a1b',1,'mlx::core::Greater']]], - ['greater_72',['greater',['../group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d',1,'mlx::core']]], - ['greater_5fequal_73',['greater_equal',['../group__ops.html#ga7153071bcfff6faad21332163fb9a430',1,'mlx::core']]], - ['greaterequal_74',['GreaterEqual',['../classmlx_1_1core_1_1_greater_equal.html#a19a3c49d5a9b40e17da0e56ef6908527',1,'mlx::core::GreaterEqual']]], - ['group_75',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html#a32e6e085a427b41ca3529c5e5db30a1b',1,'mlx::core::distributed::Group']]], - ['group_76',['group',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8831cb61ac633431b78b5fb99c0ea9ff',1,'mlx::core::distributed::DistPrimitive']]], - ['gumbel_77',['gumbel',['../namespacemlx_1_1core_1_1random.html#a76f81f8f9468039a0b941513b46cb825',1,'mlx::core::random']]] + ['gather_5faxis_2',['gather_axis',['../namespacemlx_1_1core_1_1metal.html#adc66b1b48b51ac2d2f1f5bfac1b95ee3',1,'mlx::core::metal::gather_axis()'],['../gather__axis_8h.html#a0d7d3866afe2007e280781502dc767c6',1,'gather_axis(const device T *src, const device IdxT *indices, device T *out, const constant int *shape, const constant int64_t *src_strides, const constant int64_t *idx_strides, const constant size_t &ndim, const constant int &axis, const constant int &axis_size, const constant size_t &src_ax_stride, const constant size_t &idx_ax_stride, uint3 index, uint3 grid_dim): gather_axis.h']]], + ['gather_5fimpl_3',['gather_impl',['../gather_8h.html#a5cf4ff9849edd2d99ff24ab4a6c9cd5d',1,'gather.h']]], + ['gather_5fmm_4',['gather_mm',['../group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7',1,'mlx::core']]], + ['gather_5fqmm_5',['gather_qmm',['../group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0',1,'mlx::core']]], + ['gatheraxis_6',['GatherAxis',['../classmlx_1_1core_1_1_gather_axis.html#ad8fc6400954c52079f0a2f2b711df060',1,'mlx::core::GatherAxis']]], + ['gathermm_7',['GatherMM',['../classmlx_1_1core_1_1_gather_m_m.html#afd9bbc08138181b80e2fb86536ff3f2a',1,'mlx::core::GatherMM']]], + ['gatherqmm_8',['GatherQMM',['../classmlx_1_1core_1_1_gather_q_m_m.html#a60ed2ade7f10dd9c9314913a810f9360',1,'mlx::core::GatherQMM']]], + ['gemm_9',['gemm',['../namespacemlx_1_1core_1_1metal.html#ac46fd23516a61fc56d997910e4144281',1,'mlx::core::metal::gemm()'],['../steel__gemm__fused_8h.html#a70d5b54bfae6471242d4e85675c929fd',1,'gemm(): steel_gemm_fused.h']]], + ['gemm_5floop_10',['gemm_loop',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a756d7bbcc96e2919cd65eec4bc135780',1,'mlx::steel::GEMMKernel::gemm_loop(threadgroup T *As, threadgroup T *Bs, const int gemm_k_iterations, thread loader_a_t &loader_a, thread loader_b_t &loader_b, thread mma_t &mma_op, thread const short &tgp_bm, thread const short &tgp_bn, thread const short &lbk, LoopAlignment< M_aligned, N_aligned, K_aligned_ > l={})'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a756d7bbcc96e2919cd65eec4bc135780',1,'mlx::steel::GEMMKernel::gemm_loop(threadgroup T *As, threadgroup T *Bs, const int gemm_k_iterations, thread loader_a_t &loader_a, thread loader_b_t &loader_b, thread mma_t &mma_op, thread const short &tgp_bm, thread const short &tgp_bn, thread const short &lbk, LoopAlignment< M_aligned, N_aligned, K_aligned_ > l={})']]], + ['gemm_5fsplitk_11',['gemm_splitk',['../steel__gemm__splitk_8h.html#a3be6e095a0a026d3ecf57a3e67f76188',1,'steel_gemm_splitk.h']]], + ['gemm_5fsplitk_5faccum_12',['gemm_splitk_accum',['../steel__gemm__splitk_8h.html#abeb921bf1dc7941125188ddd390b0907',1,'steel_gemm_splitk.h']]], + ['gemm_5fsplitk_5faccum_5faxpby_13',['gemm_splitk_accum_axpby',['../steel__gemm__splitk_8h.html#acc33fdfaaf3eb3a0629b3d52c7043dc1',1,'steel_gemm_splitk.h']]], + ['gemv_5fmasked_14',['gemv_masked',['../namespacemlx_1_1core_1_1metal.html#abc055b75e6a059618f279c35f8de36e7',1,'mlx::core::metal::gemv_masked()'],['../gemv__masked_8h.html#af890b6ac155165f8ee0c600363938341',1,'gemv_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant int64_t *vector_batch_stride, const constant int64_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant int64_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid): gemv_masked.h']]], + ['gemv_5ft_5fmasked_15',['gemv_t_masked',['../gemv__masked_8h.html#ae5b4a5124ddf92a984258a0be1ff0f4f',1,'gemv_masked.h']]], + ['general_5fc2r_16',['general_c2r',['../namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc',1,'pocketfft::detail']]], + ['general_5fnd_17',['general_nd',['../namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0',1,'pocketfft::detail']]], + ['general_5fr2c_18',['general_r2c',['../namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed',1,'pocketfft::detail']]], + ['get_5f2d_5fgrid_5fdims_19',['get_2d_grid_dims',['../namespacemlx_1_1core.html#aa9692de582995fd3ce19493b45ab7144',1,'mlx::core::get_2d_grid_dims(const Shape &shape, const Strides &strides)'],['../namespacemlx_1_1core.html#a6ec5cdf3253a9f20ca5ea7a1590fb386',1,'mlx::core::get_2d_grid_dims(const Shape &shape, const Strides &strides, size_t divisor)']]], + ['get_5factive_5fmemory_20',['get_active_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a7a3ad4e33d57a47474c98e2f88e775d7',1,'mlx::core::metal::MetalAllocator::get_active_memory()'],['../namespacemlx_1_1core_1_1metal.html#a7b75c2639016ac4d350fa6c9da386667',1,'mlx::core::metal::get_active_memory()']]], + ['get_5farange_5fkernel_21',['get_arange_kernel',['../namespacemlx_1_1core.html#a76f614e9956a6ca05a9be4db5a483446',1,'mlx::core']]], + ['get_5farchitecture_22',['get_architecture',['../classmlx_1_1core_1_1metal_1_1_device.html#a65f64dd8bafdc704d871fc5be5e7bc0b',1,'mlx::core::metal::Device']]], + ['get_5fbinary_5fkernel_23',['get_binary_kernel',['../namespacemlx_1_1core.html#a4decd4a07d91487e6903f6e3c8b7513a',1,'mlx::core']]], + ['get_5fbinary_5fop_5ftype_24',['get_binary_op_type',['../namespacemlx_1_1core.html#a24eef9908f164adeece3be7c6924919a',1,'mlx::core']]], + ['get_5fbinary_5ftwo_5fkernel_25',['get_binary_two_kernel',['../namespacemlx_1_1core.html#a4e809746f48e5dcf7fa63215d3f5e33e',1,'mlx::core']]], + ['get_5fblock_5fdims_26',['get_block_dims',['../namespacemlx_1_1core.html#a0f0f59d3ffe2d16a684e5fc093302e15',1,'mlx::core']]], + ['get_5fcache_5fmemory_27',['get_cache_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ad3cabbe638917ca4114eb74dcabe381f',1,'mlx::core::metal::MetalAllocator::get_cache_memory()'],['../namespacemlx_1_1core_1_1metal.html#a43307654f62ed7c58e014be7fb03909c',1,'mlx::core::metal::get_cache_memory()']]], + ['get_5fcolocated_5fmtllib_5fpath_28',['get_colocated_mtllib_path',['../namespacemlx_1_1core_1_1metal.html#a5fd6ba2040e53a254b9d71ae7ebd315f',1,'mlx::core::metal']]], + ['get_5fcommand_5fbuffer_29',['get_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a5fe3970fbe92ccc55fce4241ffbe5210',1,'mlx::core::metal::Device']]], + ['get_5fcommand_5fbuffer_5fops_30',['get_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a064e1cb6a16de7a0619f6447622350f8',1,'mlx::core::metal::Device']]], + ['get_5fcommand_5fencoder_31',['get_command_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#affa682ef612def4890f5152f81ffb7e6',1,'mlx::core::metal::Device']]], + ['get_5fcoord_32',['get_coord',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a7331fff1d12f2f8b72b0006a3ad0dd83',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord(ushort simd_lane_id)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a7331fff1d12f2f8b72b0006a3ad0dd83',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord(ushort simd_lane_id)']]], + ['get_5fcopy_5fkernel_33',['get_copy_kernel',['../namespacemlx_1_1core.html#a05a220cff45f12439fde775983c6df78',1,'mlx::core']]], + ['get_5fdefault_5fstream_34',['get_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a2366c7b888e433608e203752edc92282',1,'mlx::core::scheduler::Scheduler']]], + ['get_5fdynamic_5fcopy_5fkernel_35',['get_dynamic_copy_kernel',['../namespacemlx_1_1core.html#addb29b3e6771875f2aecd035ed560461',1,'mlx::core']]], + ['get_5ffft_5fkernel_36',['get_fft_kernel',['../namespacemlx_1_1core.html#a1d4cffc3c78067b3d9a62d64f3fb686f',1,'mlx::core']]], + ['get_5fgemv_5fmasked_5fkernel_37',['get_gemv_masked_kernel',['../namespacemlx_1_1core.html#a90c24e0d0b99b68fad9deefcf4d3e818',1,'mlx::core']]], + ['get_5fglobal_5fformatter_38',['get_global_formatter',['../namespacemlx_1_1core.html#afc71e62dc5757564486cea5ebb12500e',1,'mlx::core']]], + ['get_5fkernel_39',['get_kernel',['../classmlx_1_1core_1_1metal_1_1_device.html#a6810c4dcbcfbf93fc51d42aa5ff0fc3a',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, MTL::Library *mtl_lib, const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#afa0cac9d800c21a8a7f6cb224256abaf',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, const std::string &lib_name="mlx", const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})']]], + ['get_5fkernel_5fpreamble_40',['get_kernel_preamble',['../compiled__preamble_8h.html#a1dfa17a0369fb90ff615c7461f5013f3',1,'compiled_preamble.h']]], + ['get_5flibrary_41',['get_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a75ed55e73baf48013028796518723ff0',1,'mlx::core::metal::Device']]], + ['get_5fmb_5fsort_5fkernel_42',['get_mb_sort_kernel',['../namespacemlx_1_1core.html#afb57825bb763050cc9a9d194aa41ac36',1,'mlx::core']]], + ['get_5fname_43',['get_name',['../structmlx_1_1core_1_1_node_namer.html#a1690dd38de288c0aee2bb53156eb770e',1,'mlx::core::NodeNamer']]], + ['get_5fpeak_5fmemory_44',['get_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ac7972a3fe58e69489de775a0f152da17',1,'mlx::core::metal::MetalAllocator::get_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#a4b67d680cefa95f0ed5801f0e14e48ce',1,'mlx::core::metal::get_peak_memory()']]], + ['get_5fplan_45',['get_plan',['../namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b',1,'pocketfft::detail']]], + ['get_5fpool_46',['get_pool',['../namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a',1,'pocketfft::detail::threading']]], + ['get_5fprimitive_5fstring_47',['get_primitive_string',['../namespacemlx_1_1core.html#ad4be35b310a252edd80d9cf04f094a60',1,'mlx::core']]], + ['get_5fquantized_5fkernel_48',['get_quantized_kernel',['../namespacemlx_1_1core.html#aa3faeae5378bfaafe3ce3432a051e43e',1,'mlx::core']]], + ['get_5freduce_5finit_5fkernel_49',['get_reduce_init_kernel',['../namespacemlx_1_1core.html#ae0470605dc819efeb6510183619f0299',1,'mlx::core']]], + ['get_5freduce_5fkernel_50',['get_reduce_kernel',['../namespacemlx_1_1core.html#a1be32ba7d67137dde7ac191dfe83ff49',1,'mlx::core']]], + ['get_5freduction_5fplan_51',['get_reduction_plan',['../namespacemlx_1_1core.html#ac97b5a6f009ca3d99854ce9512c20dba',1,'mlx::core']]], + ['get_5fscan_5fkernel_52',['get_scan_kernel',['../namespacemlx_1_1core.html#aeefaff208444d3fa61ecc0946fe1de5f',1,'mlx::core']]], + ['get_5fshape_53',['get_shape',['../namespacemlx_1_1core.html#a30fb38e05feeee19ae2b87e62bff3acf',1,'mlx::core']]], + ['get_5fsoftmax_5fkernel_54',['get_softmax_kernel',['../namespacemlx_1_1core.html#a35a412f688d79eb47e42d20a7c8650ee',1,'mlx::core']]], + ['get_5fsort_5fkernel_55',['get_sort_kernel',['../namespacemlx_1_1core.html#a84ebe6275218070f0ea320f126f64e22',1,'mlx::core']]], + ['get_5fsteel_5fconv_5fgeneral_5fkernel_56',['get_steel_conv_general_kernel',['../namespacemlx_1_1core.html#abce2b67044ee06a7bbe7a91ec7c8c48d',1,'mlx::core']]], + ['get_5fsteel_5fconv_5fkernel_57',['get_steel_conv_kernel',['../namespacemlx_1_1core.html#adce79d220672f5f3c65cc31d145ca9c4',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5ffused_5fkernel_58',['get_steel_gemm_fused_kernel',['../namespacemlx_1_1core.html#a84fa8e0aee321a9d614433a0b933103b',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fmasked_5fkernel_59',['get_steel_gemm_masked_kernel',['../namespacemlx_1_1core.html#ab5f60614e965144b451930fdf935e08d',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fsplitk_5faccum_5fkernel_60',['get_steel_gemm_splitk_accum_kernel',['../namespacemlx_1_1core.html#a195b86cad5bb99aa1bcd23952305af6b',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fsplitk_5fkernel_61',['get_steel_gemm_splitk_kernel',['../namespacemlx_1_1core.html#af48c6f2f72b61dbd6766e4f5fea85df5',1,'mlx::core']]], + ['get_5fstream_62',['get_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#aa6726858b362c7cd1f8a846a63085dbc',1,'mlx::core::scheduler::Scheduler::get_stream()'],['../namespacemlx_1_1core.html#ab436b8c08be2be32ef61bd72f7df63cd',1,'mlx::core::get_stream(int index)']]], + ['get_5ftemplate_5fdefinition_63',['get_template_definition',['../namespacemlx_1_1core.html#aae0d19f0acdef2accd2428fb84c8a032',1,'mlx::core']]], + ['get_5fternary_5fkernel_64',['get_ternary_kernel',['../namespacemlx_1_1core.html#a54eb3b65375022428aab5f810e40624b',1,'mlx::core']]], + ['get_5fternary_5fop_5ftype_65',['get_ternary_op_type',['../namespacemlx_1_1core.html#a548b6f4a39e639c18896e50b1702c830',1,'mlx::core']]], + ['get_5ftwiddle_66',['get_twiddle',['../radix_8h.html#ac5cf950316b9445296ee9ecfc56a56bd',1,'radix.h']]], + ['get_5ftype_5fstring_67',['get_type_string',['../namespacemlx_1_1core.html#af776fd91dd60594dcfebbafd17f19068',1,'mlx::core']]], + ['get_5funary_5fkernel_68',['get_unary_kernel',['../namespacemlx_1_1core.html#afbb085188b563a54606d84f87a9bf5a6',1,'mlx::core']]], + ['get_5fvar_69',['get_var',['../namespacemlx_1_1core_1_1env.html#a0efecbf9efe695adafad12b5a4945df3',1,'mlx::core::env']]], + ['gguf_5fload_5fquantized_70',['gguf_load_quantized',['../namespacemlx_1_1core.html#a65dd68163bdaef3631e3724327782498',1,'mlx::core']]], + ['good_71',['good',['../classmlx_1_1core_1_1io_1_1_reader.html#a005d0b52c1f34866f7412b7f41dabec3',1,'mlx::core::io::Reader::good()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a0b050c2c27487007e250e2e19560ffe4',1,'mlx::core::io::Writer::good()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#ac54a2c693acc3d9e6e942412148ffcc9',1,'mlx::core::io::ParallelFileReader::good()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9ec4934b26fb358d699ddce1482b2d54',1,'mlx::core::io::FileWriter::good()']]], + ['good_5fsize_5fcmplx_72',['good_size_cmplx',['../structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624',1,'pocketfft::detail::util']]], + ['good_5fsize_5freal_73',['good_size_real',['../structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa',1,'pocketfft::detail::util']]], + ['grad_74',['grad',['../namespacemlx_1_1core.html#a3d2b2929ed4636e9e2b86e125b2e57d9',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#af482f6c64acd77c57ef5bb4b7be9726c',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a64bc619876b0f8cc81a2637ca81c99f7',1,'mlx::core::grad(const std::function< array(const array &)> &fun)']]], + ['greater_75',['Greater',['../classmlx_1_1core_1_1_greater.html#a1d5992a66c020cd97a70e8e3d8cd1a1b',1,'mlx::core::Greater']]], + ['greater_76',['greater',['../group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d',1,'mlx::core']]], + ['greater_5fequal_77',['greater_equal',['../group__ops.html#ga7153071bcfff6faad21332163fb9a430',1,'mlx::core']]], + ['greaterequal_78',['GreaterEqual',['../classmlx_1_1core_1_1_greater_equal.html#a19a3c49d5a9b40e17da0e56ef6908527',1,'mlx::core::GreaterEqual']]], + ['group_79',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html#a32e6e085a427b41ca3529c5e5db30a1b',1,'mlx::core::distributed::Group']]], + ['group_80',['group',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8831cb61ac633431b78b5fb99c0ea9ff',1,'mlx::core::distributed::DistPrimitive']]], + ['gumbel_81',['gumbel',['../namespacemlx_1_1core_1_1random.html#a76f81f8f9468039a0b941513b46cb825',1,'mlx::core::random']]] ]; diff --git a/docs/build/html/search/functions_9.js b/docs/build/html/search/functions_9.js index 58914bc67..66df6e28a 100644 --- a/docs/build/html/search/functions_9.js +++ b/docs/build/html/search/functions_9.js @@ -6,15 +6,15 @@ var searchData= ['ifft2_3',['ifft2',['../namespacemlx_1_1core_1_1fft.html#a9c83e5b0498c63cac9c882494b6b3eb6',1,'mlx::core::fft::ifft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a865adcb7d7fe35541ad8c21f963905e0',1,'mlx::core::fft::ifft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], ['ifftn_4',['ifftn',['../namespacemlx_1_1core_1_1fft.html#ae29b70549a1106cf7e70a31a0f80e66a',1,'mlx::core::fft::ifftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a3794c67262e4ab28d35fa89abfdfd063',1,'mlx::core::fft::ifftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a8adeca9b76277676390ec7d04dc0620b',1,'mlx::core::fft::ifftn(const array &a, StreamOrDevice s={})']]], ['imag_5',['Imag',['../classmlx_1_1core_1_1_imag.html#a284b7de34a316110fdc98e7b753372b2',1,'mlx::core::Imag']]], - ['imag_6',['imag',['../group__ops.html#ga7ff592a64d528f0cf4f3d098465da029',1,'mlx::core']]], - ['implicit_5fgemm_5fconv_5f2d_7',['implicit_gemm_conv_2d',['../kernels_2steel_2conv_2kernels_2steel__conv_8h.html#a5728711d1c2ee4038457babb7ac12888',1,'steel_conv.h']]], + ['imag_6',['imag',['../namespacemlx_1_1core_1_1simd.html#a97c69b04852ccba242f1348fda17ca20',1,'mlx::core::simd::imag()'],['../group__ops.html#ga7ff592a64d528f0cf4f3d098465da029',1,'mlx::core::imag()']]], + ['implicit_5fgemm_5fconv_5f2d_7',['implicit_gemm_conv_2d',['../steel__conv_8h.html#a5728711d1c2ee4038457babb7ac12888',1,'steel_conv.h']]], ['implicit_5fgemm_5fconv_5f2d_5fgeneral_8',['implicit_gemm_conv_2d_general',['../steel__conv__general_8h.html#ad620c0656ae92e36dcb4a285d9e790ee',1,'steel_conv_general.h']]], ['import_5ffunction_9',['import_function',['../namespacemlx_1_1core.html#a4fbb29691ee1ff22c3ee2a67cbc053d5',1,'mlx::core']]], ['in_5fdynamic_5ftracing_10',['in_dynamic_tracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#af7780f0017267567ad6e5c9271e8933e',1,'mlx::core::detail::InTracing::in_dynamic_tracing()'],['../namespacemlx_1_1core_1_1detail.html#a3cede3c723ea5766a87548140bc6728e',1,'mlx::core::detail::in_dynamic_tracing()']]], ['in_5ftracing_11',['in_tracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#ac52b8e2c3f808d3076c4e1ebaf9dc63d',1,'mlx::core::detail::InTracing::in_tracing()'],['../namespacemlx_1_1core_1_1detail.html#af974e1a6f06acfc949e67a330898ac11',1,'mlx::core::detail::in_tracing()']]], ['increment_5fcommand_5fbuffer_5fops_12',['increment_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a7a33d4d601423a3d3c23d5ad7072abb6',1,'mlx::core::metal::Device']]], - ['infinity_13',['infinity',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61eb741e7af49046beb863abf023b206',1,'metal::_numeric_limits_impl< bfloat16_t >::infinity()'],['../structmlx_1_1core_1_1numeric__limits.html#a2a8f3f489b47b7e8398bec9895ae0c27',1,'mlx::core::numeric_limits< float16_t >::infinity()'],['../structmlx_1_1core_1_1numeric__limits.html#a974982399d0211786599526abdb843b8',1,'mlx::core::numeric_limits< bfloat16_t >::infinity()']]], - ['init_14',['init',['../namespacemlx_1_1core_1_1distributed.html#a33633c058c7ec82cca4f237243c6810d',1,'mlx::core::distributed::init()'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#ab40a34a8837956e24fb9b9661104c8f9',1,'mlx::core::distributed::mpi::init()']]], + ['infinity_13',['infinity',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61eb741e7af49046beb863abf023b206',1,'metal::_numeric_limits_impl< bfloat16_t >::infinity()'],['../structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html#a2a8f3f489b47b7e8398bec9895ae0c27',1,'mlx::core::numeric_limits< float16_t >::infinity()'],['../structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html#a974982399d0211786599526abdb843b8',1,'mlx::core::numeric_limits< bfloat16_t >::infinity()']]], + ['init_14',['init',['../namespacemlx_1_1core_1_1distributed.html#a24cdcd2aa23a3410a8973753ade3f772',1,'mlx::core::distributed::init()'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#ab40a34a8837956e24fb9b9661104c8f9',1,'mlx::core::distributed::mpi::init()'],['../namespacemlx_1_1core_1_1distributed_1_1ring.html#a81a13abe6f334d2f6b058b39a2221e67',1,'mlx::core::distributed::ring::init()']]], ['init_5freduce_15',['init_reduce',['../reduce__init_8h.html#a0088604ac2eaa6940689ff12c4ba5fc2',1,'reduce_init.h']]], ['inner_16',['inner',['../group__ops.html#ga654fec16a9746b390916697a2ab2546e',1,'mlx::core']]], ['inputs_17',['inputs',['../classmlx_1_1core_1_1array.html#a2913abcdf71826827c8457f529825fff',1,'mlx::core::array::inputs() const'],['../classmlx_1_1core_1_1array.html#acffb082177f9b78f0c52e406adff972f',1,'mlx::core::array::inputs()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a27ded7e54bc1712063c874646b445509',1,'mlx::core::metal::CommandEncoder::inputs()'],['../structmlx_1_1core_1_1_command_encoder.html#a27ded7e54bc1712063c874646b445509',1,'mlx::core::CommandEncoder::inputs()']]], @@ -26,9 +26,9 @@ var searchData= ['irfft_23',['irfft',['../namespacemlx_1_1core_1_1fft.html#aafa721d0492e9f74913a6e86b4896ad8',1,'mlx::core::fft::irfft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#afbd0035a3cf91f428838de1fcf01a3a3',1,'mlx::core::fft::irfft(const array &a, int axis=-1, StreamOrDevice s={})']]], ['irfft2_24',['irfft2',['../namespacemlx_1_1core_1_1fft.html#ae2309d3a7a72c62dabdc16d5b38cc6b3',1,'mlx::core::fft::irfft2(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#af7c7bbbbce26c2775a77473502a8de02',1,'mlx::core::fft::irfft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], ['irfftn_25',['irfftn',['../namespacemlx_1_1core_1_1fft.html#a2c6abf48be3fcf5afd88c172a5f038ea',1,'mlx::core::fft::irfftn(const array &a, const Shape &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a1c9ad11121c5879d5c04bbde2ee238c3',1,'mlx::core::fft::irfftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#aaf5a7ef93b3426b94c2363a23a5a5b36',1,'mlx::core::fft::irfftn(const array &a, StreamOrDevice s={})']]], - ['is_5favailable_26',['is_available',['../classmlx_1_1core_1_1array.html#aebed1f37c19197be76105161102a8a40',1,'mlx::core::array::is_available()'],['../namespacemlx_1_1core_1_1metal.html#a0cdf2c08c7bc0927a86070adc206987f',1,'mlx::core::metal::is_available()'],['../namespacemlx_1_1core_1_1distributed.html#a95655473cd0032c06e5fe3fca85aeef3',1,'mlx::core::distributed::is_available()'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#a86d8a52e75b15bae8fb0992b418a41c7',1,'mlx::core::distributed::mpi::is_available()']]], + ['is_5favailable_26',['is_available',['../classmlx_1_1core_1_1array.html#aebed1f37c19197be76105161102a8a40',1,'mlx::core::array::is_available()'],['../namespacemlx_1_1core_1_1metal.html#a0cdf2c08c7bc0927a86070adc206987f',1,'mlx::core::metal::is_available()'],['../namespacemlx_1_1core_1_1distributed.html#a95655473cd0032c06e5fe3fca85aeef3',1,'mlx::core::distributed::is_available()'],['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#a86d8a52e75b15bae8fb0992b418a41c7',1,'mlx::core::distributed::mpi::is_available()'],['../namespacemlx_1_1core_1_1distributed_1_1ring.html#a1238e89ee95ba016741f0abe91b540ac',1,'mlx::core::distributed::ring::is_available()']]], ['is_5fdonatable_27',['is_donatable',['../classmlx_1_1core_1_1array.html#a4677a404b5d191af20b52649225de087',1,'mlx::core::array::is_donatable()'],['../namespacemlx_1_1core.html#af650e831ce21759da1ac103037d08d84',1,'mlx::core::is_donatable()']]], - ['is_5fequivalent_28',['is_equivalent',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#af08b1294f3f93505a96fdfa85b1edd62',1,'mlx::core::fast::ScaledDotProductAttention::is_equivalent()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a5936175e5923aec272d6f718785f57a1',1,'mlx::core::fast::AffineQuantize::is_equivalent()'],['../classmlx_1_1core_1_1_primitive.html#a6140a502af4c2bbbc776ab26e9afebcd',1,'mlx::core::Primitive::is_equivalent()'],['../classmlx_1_1core_1_1_abs.html#ab6f0ec56bc7c048382297e12dabadc67',1,'mlx::core::Abs::is_equivalent()'],['../classmlx_1_1core_1_1_add.html#aba0a35410c3aac53d0f7a0c283d9ee3f',1,'mlx::core::Add::is_equivalent()'],['../classmlx_1_1core_1_1_add_m_m.html#a6e37c6882dba995a63fb6d8dfb01754f',1,'mlx::core::AddMM::is_equivalent()'],['../classmlx_1_1core_1_1_arange.html#a7b6a45cf9c4b109d4e0373f3fe576c35',1,'mlx::core::Arange::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cos.html#a39557461e3235801886675a9b7d25bf5',1,'mlx::core::ArcCos::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6928e827b9ac2e86e7d5b02b78150eee',1,'mlx::core::ArcCosh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sin.html#a13b5e39eeccaf32d94b8eb85b3b753ab',1,'mlx::core::ArcSin::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sinh.html#a63c7a765c7906242dc3371deec094f0f',1,'mlx::core::ArcSinh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan.html#a0e5b5fc7218143ecd0a8666d9137c34c',1,'mlx::core::ArcTan::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan2.html#aeaee58cd803d3ebf0b76574a409682cc',1,'mlx::core::ArcTan2::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tanh.html#ac8ecdd640043dab0461d49d7650679a2',1,'mlx::core::ArcTanh::is_equivalent()'],['../classmlx_1_1core_1_1_arg_partition.html#ad87509ce70b51fb75dfb9c3a05a5b31a',1,'mlx::core::ArgPartition::is_equivalent()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03b81a670dcb1e39bf7279e4d4583b97',1,'mlx::core::ArgReduce::is_equivalent()'],['../classmlx_1_1core_1_1_arg_sort.html#a048cd09c557d29d1111726f97010a845',1,'mlx::core::ArgSort::is_equivalent()'],['../classmlx_1_1core_1_1_as_type.html#a8e6c8b2428ab15c4fb43f2e3a8fb38af',1,'mlx::core::AsType::is_equivalent()'],['../classmlx_1_1core_1_1_as_strided.html#a1738c6aa0a3a3eb68530f0d5b436e094',1,'mlx::core::AsStrided::is_equivalent()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8cd6b916b4838a6c329cf4df8530c3b8',1,'mlx::core::BitwiseBinary::is_equivalent()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aef1c303955f9b8f445296372cf181160',1,'mlx::core::BlockMaskedMM::is_equivalent()'],['../classmlx_1_1core_1_1_gather_m_m.html#a163f17f6ce2c002f22e81b302777342b',1,'mlx::core::GatherMM::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a44d14b22b995e643cb04cc43654d7b16',1,'mlx::core::BroadcastAxes::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast.html#a0e27692b0090ec451954649a36042616',1,'mlx::core::Broadcast::is_equivalent()'],['../classmlx_1_1core_1_1_ceil.html#aacd90acb56eb0649c1cef807aa21df52',1,'mlx::core::Ceil::is_equivalent()'],['../classmlx_1_1core_1_1_compiled.html#a63e5016458887813b4a59dee5a0a3f10',1,'mlx::core::Compiled::is_equivalent()'],['../classmlx_1_1core_1_1_concatenate.html#aaf8a72a0c30114460caf519580cc35d2',1,'mlx::core::Concatenate::is_equivalent()'],['../classmlx_1_1core_1_1_conjugate.html#af42f00a790c6bc5572bd8fe9e5b36c5e',1,'mlx::core::Conjugate::is_equivalent()'],['../classmlx_1_1core_1_1_contiguous.html#aa5d273a461fc6e64f3c9a67c24cb3372',1,'mlx::core::Contiguous::is_equivalent()'],['../classmlx_1_1core_1_1_convolution.html#afb87708a5e3aab2e9e663daa9d8863de',1,'mlx::core::Convolution::is_equivalent()'],['../classmlx_1_1core_1_1_copy.html#afcfa39465015f638e294aa954ea0f3da',1,'mlx::core::Copy::is_equivalent()'],['../classmlx_1_1core_1_1_cos.html#ab611ca38c987915659f7ffcce0370417',1,'mlx::core::Cos::is_equivalent()'],['../classmlx_1_1core_1_1_cosh.html#ae0bacccaf501f5349db0c13cca776ff9',1,'mlx::core::Cosh::is_equivalent()'],['../classmlx_1_1core_1_1_divide.html#a3dda091f05c4164c29bb8129e9712650',1,'mlx::core::Divide::is_equivalent()'],['../classmlx_1_1core_1_1_div_mod.html#af5fcf8ec8515d46844cbeeab6dafb38a',1,'mlx::core::DivMod::is_equivalent()'],['../classmlx_1_1core_1_1_select.html#afc3c333fac7f902c98839921ef2874c8',1,'mlx::core::Select::is_equivalent()'],['../classmlx_1_1core_1_1_remainder.html#a802039faaa2ed7b763ec3d7debcce814',1,'mlx::core::Remainder::is_equivalent()'],['../classmlx_1_1core_1_1_equal.html#a58c1c5003e43f47dc0788c1851deaa02',1,'mlx::core::Equal::is_equivalent()'],['../classmlx_1_1core_1_1_erf.html#abe99dfbc2954c3a7d5dec56ab165ee82',1,'mlx::core::Erf::is_equivalent()'],['../classmlx_1_1core_1_1_erf_inv.html#aaac9e3b454ba564f9c6e804ab6562832',1,'mlx::core::ErfInv::is_equivalent()'],['../classmlx_1_1core_1_1_exp.html#ac6e44bffe7a643ab4ca51e74c7328357',1,'mlx::core::Exp::is_equivalent()'],['../classmlx_1_1core_1_1_expand_dims.html#aef468da4027527afec7b24161ce1e1f3',1,'mlx::core::ExpandDims::is_equivalent()'],['../classmlx_1_1core_1_1_f_f_t.html#a0ede3bc8b6d77d560c0a750b68fddc06',1,'mlx::core::FFT::is_equivalent()'],['../classmlx_1_1core_1_1_flatten.html#a42499e796aac751fceb4628317cc58f4',1,'mlx::core::Flatten::is_equivalent()'],['../classmlx_1_1core_1_1_floor.html#a24b64feb026c4fcd02fc481cffdb1c94',1,'mlx::core::Floor::is_equivalent()'],['../classmlx_1_1core_1_1_full.html#afafcbcae1e28597fe8f7fde289105792',1,'mlx::core::Full::is_equivalent()'],['../classmlx_1_1core_1_1_gather.html#a23ff1406dbf0c770e75ad47440b467aa',1,'mlx::core::Gather::is_equivalent()'],['../classmlx_1_1core_1_1_greater.html#a6877a6888614a618dc64296763ccabb1',1,'mlx::core::Greater::is_equivalent()'],['../classmlx_1_1core_1_1_greater_equal.html#a3daef8596b963026b602019bc56fc5fc',1,'mlx::core::GreaterEqual::is_equivalent()'],['../classmlx_1_1core_1_1_hadamard.html#a8a528d8d69a7343bdfd704a3e74230b8',1,'mlx::core::Hadamard::is_equivalent()'],['../classmlx_1_1core_1_1_imag.html#a51c15ae82855edebba2ba779516465f5',1,'mlx::core::Imag::is_equivalent()'],['../classmlx_1_1core_1_1_less.html#a7d6ed6353a0dcefebd008026dbd3cd63',1,'mlx::core::Less::is_equivalent()'],['../classmlx_1_1core_1_1_less_equal.html#a76ee1438cf4bd109eae4e0b3472b26af',1,'mlx::core::LessEqual::is_equivalent()'],['../classmlx_1_1core_1_1_log.html#a2fc58ea4ca744db493b947d1136d05f8',1,'mlx::core::Log::is_equivalent()'],['../classmlx_1_1core_1_1_logical_not.html#aba53675da351cd9b71a73d475b4bbe99',1,'mlx::core::LogicalNot::is_equivalent()'],['../classmlx_1_1core_1_1_logical_and.html#a9572c35f72e0db2f7f86bbf42438a6be',1,'mlx::core::LogicalAnd::is_equivalent()'],['../classmlx_1_1core_1_1_logical_or.html#a9c8b10a5cf5c69fdc2362390197e4e71',1,'mlx::core::LogicalOr::is_equivalent()'],['../classmlx_1_1core_1_1_log_add_exp.html#a3cf9a202c05aff39919d713d6e2b32e4',1,'mlx::core::LogAddExp::is_equivalent()'],['../classmlx_1_1core_1_1_matmul.html#aab372b59eae0840fc4f75ef5719a2630',1,'mlx::core::Matmul::is_equivalent()'],['../classmlx_1_1core_1_1_maximum.html#a21fe93fbd7799682f481260aee8bdb46',1,'mlx::core::Maximum::is_equivalent()'],['../classmlx_1_1core_1_1_minimum.html#a56c54ee3293cc2cd84462b9ec7ac36b4',1,'mlx::core::Minimum::is_equivalent()'],['../classmlx_1_1core_1_1_multiply.html#ae288159fa2d6d35087a85aca8eafa9b2',1,'mlx::core::Multiply::is_equivalent()'],['../classmlx_1_1core_1_1_negative.html#ac2a4d8159c548639d6289980c8975823',1,'mlx::core::Negative::is_equivalent()'],['../classmlx_1_1core_1_1_not_equal.html#ac12fd6b3e2f2e7e4e622b59badf2c73d',1,'mlx::core::NotEqual::is_equivalent()'],['../classmlx_1_1core_1_1_number_of_elements.html#ad6a32565ccc64499e368e15bba0b438f',1,'mlx::core::NumberOfElements::is_equivalent()'],['../classmlx_1_1core_1_1_pad.html#aad7c3bfecafe435d6a8e807de4c7ea9b',1,'mlx::core::Pad::is_equivalent()'],['../classmlx_1_1core_1_1_partition.html#aabdf6ef4f2159b2bfe93e0e87d4772f8',1,'mlx::core::Partition::is_equivalent()'],['../classmlx_1_1core_1_1_power.html#a76b4ec9d1ff07f06189e414480453d68',1,'mlx::core::Power::is_equivalent()'],['../classmlx_1_1core_1_1_quantized_matmul.html#af28b36e3f40ea41785387800326cc8e1',1,'mlx::core::QuantizedMatmul::is_equivalent()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a6a7da6bcf657fcdb157c45bf35fdec11',1,'mlx::core::GatherQMM::is_equivalent()'],['../classmlx_1_1core_1_1_random_bits.html#a72ec915debf5823e7c0463045b2894e6',1,'mlx::core::RandomBits::is_equivalent()'],['../classmlx_1_1core_1_1_real.html#a6d9bed396862a9e9d6abfbdcd8d8d239',1,'mlx::core::Real::is_equivalent()'],['../classmlx_1_1core_1_1_reshape.html#abd07c53af476777a04307e0423784cf3',1,'mlx::core::Reshape::is_equivalent()'],['../classmlx_1_1core_1_1_reduce.html#abe8f3327d617d0dd7438f066497ae08e',1,'mlx::core::Reduce::is_equivalent()'],['../classmlx_1_1core_1_1_round.html#aeb3d8607bbba7345a3142d4cbd4e6927',1,'mlx::core::Round::is_equivalent()'],['../classmlx_1_1core_1_1_scan.html#a54445a4d677ca4fe2a58d08eb5223ac6',1,'mlx::core::Scan::is_equivalent()'],['../classmlx_1_1core_1_1_scatter.html#a0208172562abdc90472e6eb5f84c987f',1,'mlx::core::Scatter::is_equivalent()'],['../classmlx_1_1core_1_1_sigmoid.html#a04814ba1b0edf8299d5ca1bcb8749d8e',1,'mlx::core::Sigmoid::is_equivalent()'],['../classmlx_1_1core_1_1_sign.html#a8c0934acbcc4b146e5aacd35a8c445bb',1,'mlx::core::Sign::is_equivalent()'],['../classmlx_1_1core_1_1_sin.html#af00b0e5516f884996ce7a97e6c1e3e6a',1,'mlx::core::Sin::is_equivalent()'],['../classmlx_1_1core_1_1_sinh.html#adcb1878996fd4902cd550042dd6ad70d',1,'mlx::core::Sinh::is_equivalent()'],['../classmlx_1_1core_1_1_slice.html#a43202c3b8966ae1db9ab82072e4918b0',1,'mlx::core::Slice::is_equivalent()'],['../classmlx_1_1core_1_1_slice_update.html#a60f588acced42391e6e5615ae8d16119',1,'mlx::core::SliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a0325271def8d9ea9ed21eb27e51994b4',1,'mlx::core::DynamicSlice::is_equivalent()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ae6292d2b1f3221a7c8ef6b77cb466481',1,'mlx::core::DynamicSliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_softmax.html#a9215ed7bd36bc11276c58dfb9808d728',1,'mlx::core::Softmax::is_equivalent()'],['../classmlx_1_1core_1_1_sort.html#ae48f07cf641d54234fc4fb6529a33511',1,'mlx::core::Sort::is_equivalent()'],['../classmlx_1_1core_1_1_split.html#af25a0cc259573b9dce60d285eee18345',1,'mlx::core::Split::is_equivalent()'],['../classmlx_1_1core_1_1_square.html#a6abc881d44071019aa15481e5ea75ab2',1,'mlx::core::Square::is_equivalent()'],['../classmlx_1_1core_1_1_sqrt.html#ab871c2b8ab4a27a3f782a005d0e87c46',1,'mlx::core::Sqrt::is_equivalent()'],['../classmlx_1_1core_1_1_stop_gradient.html#a327539298b21d800d26482b94fce41b3',1,'mlx::core::StopGradient::is_equivalent()'],['../classmlx_1_1core_1_1_subtract.html#af1c05e1e3f703ba916d54f8ccbbd102b',1,'mlx::core::Subtract::is_equivalent()'],['../classmlx_1_1core_1_1_squeeze.html#a08f35991d36e30fa4c05a5c9e91feb93',1,'mlx::core::Squeeze::is_equivalent()'],['../classmlx_1_1core_1_1_tan.html#afdf46288e7f60ea7f878688347dff7e4',1,'mlx::core::Tan::is_equivalent()'],['../classmlx_1_1core_1_1_tanh.html#a0692a1de2373b86eb394252ed4fecfda',1,'mlx::core::Tanh::is_equivalent()'],['../classmlx_1_1core_1_1_unflatten.html#a6a89fc709aae0fb3e17035e39b5ccd58',1,'mlx::core::Unflatten::is_equivalent()'],['../classmlx_1_1core_1_1_view.html#a7cb8403a96a47cb258caac4e3b850f64',1,'mlx::core::View::is_equivalent()'],['../classmlx_1_1core_1_1_transpose.html#a799ec3c3fa9f1b9e6177c755252a3eab',1,'mlx::core::Transpose::is_equivalent()'],['../classmlx_1_1core_1_1_eigh.html#a09414e3fe88a952408d164d6dd0af381',1,'mlx::core::Eigh::is_equivalent()']]], + ['is_5fequivalent_28',['is_equivalent',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#af08b1294f3f93505a96fdfa85b1edd62',1,'mlx::core::fast::ScaledDotProductAttention::is_equivalent()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a5936175e5923aec272d6f718785f57a1',1,'mlx::core::fast::AffineQuantize::is_equivalent()'],['../classmlx_1_1core_1_1_primitive.html#a6140a502af4c2bbbc776ab26e9afebcd',1,'mlx::core::Primitive::is_equivalent()'],['../classmlx_1_1core_1_1_abs.html#ab6f0ec56bc7c048382297e12dabadc67',1,'mlx::core::Abs::is_equivalent()'],['../classmlx_1_1core_1_1_add.html#aba0a35410c3aac53d0f7a0c283d9ee3f',1,'mlx::core::Add::is_equivalent()'],['../classmlx_1_1core_1_1_add_m_m.html#a6e37c6882dba995a63fb6d8dfb01754f',1,'mlx::core::AddMM::is_equivalent()'],['../classmlx_1_1core_1_1_arange.html#a7b6a45cf9c4b109d4e0373f3fe576c35',1,'mlx::core::Arange::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cos.html#a39557461e3235801886675a9b7d25bf5',1,'mlx::core::ArcCos::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6928e827b9ac2e86e7d5b02b78150eee',1,'mlx::core::ArcCosh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sin.html#a13b5e39eeccaf32d94b8eb85b3b753ab',1,'mlx::core::ArcSin::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sinh.html#a63c7a765c7906242dc3371deec094f0f',1,'mlx::core::ArcSinh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan.html#a0e5b5fc7218143ecd0a8666d9137c34c',1,'mlx::core::ArcTan::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan2.html#aeaee58cd803d3ebf0b76574a409682cc',1,'mlx::core::ArcTan2::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tanh.html#ac8ecdd640043dab0461d49d7650679a2',1,'mlx::core::ArcTanh::is_equivalent()'],['../classmlx_1_1core_1_1_arg_partition.html#ad87509ce70b51fb75dfb9c3a05a5b31a',1,'mlx::core::ArgPartition::is_equivalent()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03b81a670dcb1e39bf7279e4d4583b97',1,'mlx::core::ArgReduce::is_equivalent()'],['../classmlx_1_1core_1_1_arg_sort.html#a048cd09c557d29d1111726f97010a845',1,'mlx::core::ArgSort::is_equivalent()'],['../classmlx_1_1core_1_1_as_type.html#a8e6c8b2428ab15c4fb43f2e3a8fb38af',1,'mlx::core::AsType::is_equivalent()'],['../classmlx_1_1core_1_1_as_strided.html#a1738c6aa0a3a3eb68530f0d5b436e094',1,'mlx::core::AsStrided::is_equivalent()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8cd6b916b4838a6c329cf4df8530c3b8',1,'mlx::core::BitwiseBinary::is_equivalent()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aef1c303955f9b8f445296372cf181160',1,'mlx::core::BlockMaskedMM::is_equivalent()'],['../classmlx_1_1core_1_1_gather_m_m.html#a163f17f6ce2c002f22e81b302777342b',1,'mlx::core::GatherMM::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a44d14b22b995e643cb04cc43654d7b16',1,'mlx::core::BroadcastAxes::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast.html#a0e27692b0090ec451954649a36042616',1,'mlx::core::Broadcast::is_equivalent()'],['../classmlx_1_1core_1_1_ceil.html#aacd90acb56eb0649c1cef807aa21df52',1,'mlx::core::Ceil::is_equivalent()'],['../classmlx_1_1core_1_1_compiled.html#a63e5016458887813b4a59dee5a0a3f10',1,'mlx::core::Compiled::is_equivalent()'],['../classmlx_1_1core_1_1_concatenate.html#aaf8a72a0c30114460caf519580cc35d2',1,'mlx::core::Concatenate::is_equivalent()'],['../classmlx_1_1core_1_1_conjugate.html#af42f00a790c6bc5572bd8fe9e5b36c5e',1,'mlx::core::Conjugate::is_equivalent()'],['../classmlx_1_1core_1_1_contiguous.html#aa5d273a461fc6e64f3c9a67c24cb3372',1,'mlx::core::Contiguous::is_equivalent()'],['../classmlx_1_1core_1_1_convolution.html#afb87708a5e3aab2e9e663daa9d8863de',1,'mlx::core::Convolution::is_equivalent()'],['../classmlx_1_1core_1_1_copy.html#afcfa39465015f638e294aa954ea0f3da',1,'mlx::core::Copy::is_equivalent()'],['../classmlx_1_1core_1_1_cos.html#ab611ca38c987915659f7ffcce0370417',1,'mlx::core::Cos::is_equivalent()'],['../classmlx_1_1core_1_1_cosh.html#ae0bacccaf501f5349db0c13cca776ff9',1,'mlx::core::Cosh::is_equivalent()'],['../classmlx_1_1core_1_1_divide.html#a3dda091f05c4164c29bb8129e9712650',1,'mlx::core::Divide::is_equivalent()'],['../classmlx_1_1core_1_1_div_mod.html#af5fcf8ec8515d46844cbeeab6dafb38a',1,'mlx::core::DivMod::is_equivalent()'],['../classmlx_1_1core_1_1_select.html#afc3c333fac7f902c98839921ef2874c8',1,'mlx::core::Select::is_equivalent()'],['../classmlx_1_1core_1_1_remainder.html#a802039faaa2ed7b763ec3d7debcce814',1,'mlx::core::Remainder::is_equivalent()'],['../classmlx_1_1core_1_1_equal.html#a58c1c5003e43f47dc0788c1851deaa02',1,'mlx::core::Equal::is_equivalent()'],['../classmlx_1_1core_1_1_erf.html#abe99dfbc2954c3a7d5dec56ab165ee82',1,'mlx::core::Erf::is_equivalent()'],['../classmlx_1_1core_1_1_erf_inv.html#aaac9e3b454ba564f9c6e804ab6562832',1,'mlx::core::ErfInv::is_equivalent()'],['../classmlx_1_1core_1_1_exp.html#ac6e44bffe7a643ab4ca51e74c7328357',1,'mlx::core::Exp::is_equivalent()'],['../classmlx_1_1core_1_1_expand_dims.html#aef468da4027527afec7b24161ce1e1f3',1,'mlx::core::ExpandDims::is_equivalent()'],['../classmlx_1_1core_1_1_f_f_t.html#a0ede3bc8b6d77d560c0a750b68fddc06',1,'mlx::core::FFT::is_equivalent()'],['../classmlx_1_1core_1_1_flatten.html#a42499e796aac751fceb4628317cc58f4',1,'mlx::core::Flatten::is_equivalent()'],['../classmlx_1_1core_1_1_floor.html#a24b64feb026c4fcd02fc481cffdb1c94',1,'mlx::core::Floor::is_equivalent()'],['../classmlx_1_1core_1_1_full.html#afafcbcae1e28597fe8f7fde289105792',1,'mlx::core::Full::is_equivalent()'],['../classmlx_1_1core_1_1_gather.html#a23ff1406dbf0c770e75ad47440b467aa',1,'mlx::core::Gather::is_equivalent()'],['../classmlx_1_1core_1_1_gather_axis.html#a8f603c5c46d566654bd8a615d24c1089',1,'mlx::core::GatherAxis::is_equivalent()'],['../classmlx_1_1core_1_1_greater.html#a6877a6888614a618dc64296763ccabb1',1,'mlx::core::Greater::is_equivalent()'],['../classmlx_1_1core_1_1_greater_equal.html#a3daef8596b963026b602019bc56fc5fc',1,'mlx::core::GreaterEqual::is_equivalent()'],['../classmlx_1_1core_1_1_hadamard.html#a8a528d8d69a7343bdfd704a3e74230b8',1,'mlx::core::Hadamard::is_equivalent()'],['../classmlx_1_1core_1_1_imag.html#a51c15ae82855edebba2ba779516465f5',1,'mlx::core::Imag::is_equivalent()'],['../classmlx_1_1core_1_1_less.html#a7d6ed6353a0dcefebd008026dbd3cd63',1,'mlx::core::Less::is_equivalent()'],['../classmlx_1_1core_1_1_less_equal.html#a76ee1438cf4bd109eae4e0b3472b26af',1,'mlx::core::LessEqual::is_equivalent()'],['../classmlx_1_1core_1_1_log.html#a2fc58ea4ca744db493b947d1136d05f8',1,'mlx::core::Log::is_equivalent()'],['../classmlx_1_1core_1_1_logical_not.html#aba53675da351cd9b71a73d475b4bbe99',1,'mlx::core::LogicalNot::is_equivalent()'],['../classmlx_1_1core_1_1_logical_and.html#a9572c35f72e0db2f7f86bbf42438a6be',1,'mlx::core::LogicalAnd::is_equivalent()'],['../classmlx_1_1core_1_1_logical_or.html#a9c8b10a5cf5c69fdc2362390197e4e71',1,'mlx::core::LogicalOr::is_equivalent()'],['../classmlx_1_1core_1_1_log_add_exp.html#a3cf9a202c05aff39919d713d6e2b32e4',1,'mlx::core::LogAddExp::is_equivalent()'],['../classmlx_1_1core_1_1_matmul.html#aab372b59eae0840fc4f75ef5719a2630',1,'mlx::core::Matmul::is_equivalent()'],['../classmlx_1_1core_1_1_maximum.html#a21fe93fbd7799682f481260aee8bdb46',1,'mlx::core::Maximum::is_equivalent()'],['../classmlx_1_1core_1_1_minimum.html#a56c54ee3293cc2cd84462b9ec7ac36b4',1,'mlx::core::Minimum::is_equivalent()'],['../classmlx_1_1core_1_1_multiply.html#ae288159fa2d6d35087a85aca8eafa9b2',1,'mlx::core::Multiply::is_equivalent()'],['../classmlx_1_1core_1_1_negative.html#ac2a4d8159c548639d6289980c8975823',1,'mlx::core::Negative::is_equivalent()'],['../classmlx_1_1core_1_1_not_equal.html#ac12fd6b3e2f2e7e4e622b59badf2c73d',1,'mlx::core::NotEqual::is_equivalent()'],['../classmlx_1_1core_1_1_number_of_elements.html#ad6a32565ccc64499e368e15bba0b438f',1,'mlx::core::NumberOfElements::is_equivalent()'],['../classmlx_1_1core_1_1_pad.html#aad7c3bfecafe435d6a8e807de4c7ea9b',1,'mlx::core::Pad::is_equivalent()'],['../classmlx_1_1core_1_1_partition.html#aabdf6ef4f2159b2bfe93e0e87d4772f8',1,'mlx::core::Partition::is_equivalent()'],['../classmlx_1_1core_1_1_power.html#a76b4ec9d1ff07f06189e414480453d68',1,'mlx::core::Power::is_equivalent()'],['../classmlx_1_1core_1_1_quantized_matmul.html#af28b36e3f40ea41785387800326cc8e1',1,'mlx::core::QuantizedMatmul::is_equivalent()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a6a7da6bcf657fcdb157c45bf35fdec11',1,'mlx::core::GatherQMM::is_equivalent()'],['../classmlx_1_1core_1_1_random_bits.html#a72ec915debf5823e7c0463045b2894e6',1,'mlx::core::RandomBits::is_equivalent()'],['../classmlx_1_1core_1_1_real.html#a6d9bed396862a9e9d6abfbdcd8d8d239',1,'mlx::core::Real::is_equivalent()'],['../classmlx_1_1core_1_1_reshape.html#abd07c53af476777a04307e0423784cf3',1,'mlx::core::Reshape::is_equivalent()'],['../classmlx_1_1core_1_1_reduce.html#abe8f3327d617d0dd7438f066497ae08e',1,'mlx::core::Reduce::is_equivalent()'],['../classmlx_1_1core_1_1_round.html#aeb3d8607bbba7345a3142d4cbd4e6927',1,'mlx::core::Round::is_equivalent()'],['../classmlx_1_1core_1_1_scan.html#a54445a4d677ca4fe2a58d08eb5223ac6',1,'mlx::core::Scan::is_equivalent()'],['../classmlx_1_1core_1_1_scatter.html#a0208172562abdc90472e6eb5f84c987f',1,'mlx::core::Scatter::is_equivalent()'],['../classmlx_1_1core_1_1_scatter_axis.html#af511c39926d5b85ca59558d64e0608fb',1,'mlx::core::ScatterAxis::is_equivalent()'],['../classmlx_1_1core_1_1_sigmoid.html#a04814ba1b0edf8299d5ca1bcb8749d8e',1,'mlx::core::Sigmoid::is_equivalent()'],['../classmlx_1_1core_1_1_sign.html#a8c0934acbcc4b146e5aacd35a8c445bb',1,'mlx::core::Sign::is_equivalent()'],['../classmlx_1_1core_1_1_sin.html#af00b0e5516f884996ce7a97e6c1e3e6a',1,'mlx::core::Sin::is_equivalent()'],['../classmlx_1_1core_1_1_sinh.html#adcb1878996fd4902cd550042dd6ad70d',1,'mlx::core::Sinh::is_equivalent()'],['../classmlx_1_1core_1_1_slice.html#a43202c3b8966ae1db9ab82072e4918b0',1,'mlx::core::Slice::is_equivalent()'],['../classmlx_1_1core_1_1_slice_update.html#a60f588acced42391e6e5615ae8d16119',1,'mlx::core::SliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a0325271def8d9ea9ed21eb27e51994b4',1,'mlx::core::DynamicSlice::is_equivalent()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ae6292d2b1f3221a7c8ef6b77cb466481',1,'mlx::core::DynamicSliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_softmax.html#a9215ed7bd36bc11276c58dfb9808d728',1,'mlx::core::Softmax::is_equivalent()'],['../classmlx_1_1core_1_1_sort.html#ae48f07cf641d54234fc4fb6529a33511',1,'mlx::core::Sort::is_equivalent()'],['../classmlx_1_1core_1_1_split.html#af25a0cc259573b9dce60d285eee18345',1,'mlx::core::Split::is_equivalent()'],['../classmlx_1_1core_1_1_square.html#a6abc881d44071019aa15481e5ea75ab2',1,'mlx::core::Square::is_equivalent()'],['../classmlx_1_1core_1_1_sqrt.html#ab871c2b8ab4a27a3f782a005d0e87c46',1,'mlx::core::Sqrt::is_equivalent()'],['../classmlx_1_1core_1_1_stop_gradient.html#a327539298b21d800d26482b94fce41b3',1,'mlx::core::StopGradient::is_equivalent()'],['../classmlx_1_1core_1_1_subtract.html#af1c05e1e3f703ba916d54f8ccbbd102b',1,'mlx::core::Subtract::is_equivalent()'],['../classmlx_1_1core_1_1_squeeze.html#a08f35991d36e30fa4c05a5c9e91feb93',1,'mlx::core::Squeeze::is_equivalent()'],['../classmlx_1_1core_1_1_tan.html#afdf46288e7f60ea7f878688347dff7e4',1,'mlx::core::Tan::is_equivalent()'],['../classmlx_1_1core_1_1_tanh.html#a0692a1de2373b86eb394252ed4fecfda',1,'mlx::core::Tanh::is_equivalent()'],['../classmlx_1_1core_1_1_unflatten.html#a6a89fc709aae0fb3e17035e39b5ccd58',1,'mlx::core::Unflatten::is_equivalent()'],['../classmlx_1_1core_1_1_view.html#a7cb8403a96a47cb258caac4e3b850f64',1,'mlx::core::View::is_equivalent()'],['../classmlx_1_1core_1_1_transpose.html#a799ec3c3fa9f1b9e6177c755252a3eab',1,'mlx::core::Transpose::is_equivalent()'],['../classmlx_1_1core_1_1_eigh.html#a09414e3fe88a952408d164d6dd0af381',1,'mlx::core::Eigh::is_equivalent()']]], ['is_5fopen_29',['is_open',['../classmlx_1_1core_1_1io_1_1_reader.html#a780f504058bd9c80cb3d105046a9f985',1,'mlx::core::io::Reader::is_open()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a85aa36bdb0dbfb8c5b6cfd955b03417a',1,'mlx::core::io::Writer::is_open()'],['../classmlx_1_1core_1_1io_1_1_parallel_file_reader.html#a653009adbcbce8248bc666df502fdbde',1,'mlx::core::io::ParallelFileReader::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#ad5d2ee671a81700cb1658c41309d6676',1,'mlx::core::io::FileWriter::is_open()']]], ['is_5fpower_5fof_5f2_30',['is_power_of_2',['../namespacemlx_1_1core.html#adacbc4526e8964b267a8ec3eb1bc1a32',1,'mlx::core']]], ['is_5fready_31',['is_ready',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3',1,'pocketfft::detail::threading::latch']]], @@ -39,7 +39,7 @@ var searchData= ['isclose_36',['isclose',['../group__ops.html#ga51eac95c04400921c54716de14b52491',1,'mlx::core']]], ['isfinite_37',['isfinite',['../group__ops.html#ga725ff0789f934b1fdd54ee29e47022ff',1,'mlx::core']]], ['isinf_38',['isinf',['../group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844',1,'mlx::core']]], - ['isnan_39',['isnan',['../namespacemetal.html#a83320ba983d90dd1fa5847b6940dc0bb',1,'metal::isnan()'],['../group__ops.html#ga175592792471b0ffb45196dca4711ba6',1,'mlx::core::isnan(const array &a, StreamOrDevice s={})']]], + ['isnan_39',['isnan',['../namespacemlx_1_1core_1_1simd.html#a05f4422a037c3bef343fb11f71363b65',1,'mlx::core::simd::isnan(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ac1c6c9b8bc7f3cd32ae39fa84975194d',1,'mlx::core::simd::isnan(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#a20ffdefe25beda96860a1dc9a6f4aa02',1,'mlx::core::simd::isnan(Simd< float16_t, N > v)'],['../namespacemetal.html#a83320ba983d90dd1fa5847b6940dc0bb',1,'metal::isnan()'],['../group__ops.html#ga175592792471b0ffb45196dca4711ba6',1,'mlx::core::isnan(const array &a, StreamOrDevice s={})']]], ['isneginf_40',['isneginf',['../group__ops.html#ga1940523da381ed7be50656a3bc465ff3',1,'mlx::core']]], ['isposinf_41',['isposinf',['../group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993',1,'mlx::core']]], ['issubdtype_42',['issubdtype',['../namespacemlx_1_1core.html#aca9e69b06f4212eba44bf0ce6711d5f7',1,'mlx::core::issubdtype(const Dtype &a, const Dtype &b)'],['../namespacemlx_1_1core.html#aca39f224c1d17bde35dfcb9088430704',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype &b)'],['../namespacemlx_1_1core.html#ae9ee4a7c205df061c1caa7e62b7504e8',1,'mlx::core::issubdtype(const Dtype &a, const Dtype::Category &b)'],['../namespacemlx_1_1core.html#ab5b1a5a3d545a5de00c3117f76d71a1d',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype::Category &b)']]], diff --git a/docs/build/html/search/functions_a.js b/docs/build/html/search/functions_a.js index b378c052c..af542a3a9 100644 --- a/docs/build/html/search/functions_a.js +++ b/docs/build/html/search/functions_a.js @@ -1,4 +1,4 @@ var searchData= [ - ['jvp_0',['jvp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#aeaf6f2b5955e7417cd1e36db42c45a80',1,'mlx::core::distributed::AllReduce::jvp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a96f08a4ea8453d0b4b737c7b07972913',1,'mlx::core::distributed::AllGather::jvp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#ac77b28702654df8e7d882a49357a9584',1,'mlx::core::fast::Custom::jvp()'],['../classmlx_1_1core_1_1_primitive.html#a9fecf38f53da08ba1947543c2b3158c2',1,'mlx::core::Primitive::jvp()'],['../classmlx_1_1core_1_1_abs.html#a6c1e6eeaf4f5e63898c3487106e88e11',1,'mlx::core::Abs::jvp()'],['../classmlx_1_1core_1_1_add.html#a77230069f76fe60a2fe1007822a277b7',1,'mlx::core::Add::jvp()'],['../classmlx_1_1core_1_1_arc_cos.html#a240079c616f1a1f127aa783308096fe9',1,'mlx::core::ArcCos::jvp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a80fcb790649219c30260af903b76a1d7',1,'mlx::core::ArcCosh::jvp()'],['../classmlx_1_1core_1_1_arc_sin.html#a37affc8c5e84e5c54e73a71fc0821ea4',1,'mlx::core::ArcSin::jvp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79ebf2f6dfecbfbb93170fdd1ca87bf4',1,'mlx::core::ArcSinh::jvp()'],['../classmlx_1_1core_1_1_arc_tan.html#a0f5590a2297fc133b4b0a15f9dd0c760',1,'mlx::core::ArcTan::jvp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a01675433f2a4fa466b2f48272dbca738',1,'mlx::core::ArcTan2::jvp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a534ebdbfe77241884630d25021274c4a',1,'mlx::core::ArcTanh::jvp()'],['../classmlx_1_1core_1_1_arg_partition.html#aedea4b47f947a6fe358dd1238cdfb595',1,'mlx::core::ArgPartition::jvp()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03bb925e1b488c560bc3d67ce62ba6fa',1,'mlx::core::ArgReduce::jvp()'],['../classmlx_1_1core_1_1_as_type.html#a213400967150c57da35795e1c9f65ca0',1,'mlx::core::AsType::jvp()'],['../classmlx_1_1core_1_1_as_strided.html#a8ff0a398c47b42e08bc1122e07a02b53',1,'mlx::core::AsStrided::jvp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a1dae6ce5dc0498d20530403fe5c5531d',1,'mlx::core::BitwiseBinary::jvp()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a830bae1f3f9078bd5b422ce6e46685a7',1,'mlx::core::BroadcastAxes::jvp()'],['../classmlx_1_1core_1_1_broadcast.html#ae2fc3851a117079244708864be770ece',1,'mlx::core::Broadcast::jvp()'],['../classmlx_1_1core_1_1_ceil.html#a7ad74b27d9f26c886c2af516b845f066',1,'mlx::core::Ceil::jvp()'],['../classmlx_1_1core_1_1_compiled.html#aa385fe28626856ca5f57161b47a3c205',1,'mlx::core::Compiled::jvp()'],['../classmlx_1_1core_1_1_concatenate.html#a9f9e7a9dc3a00e02b84c94e1868baff1',1,'mlx::core::Concatenate::jvp()'],['../classmlx_1_1core_1_1_contiguous.html#a1f9fcae7235e0ae9217825b78cb0f991',1,'mlx::core::Contiguous::jvp()'],['../classmlx_1_1core_1_1_copy.html#a5acf02aa360cbefd86749fe9877b29cc',1,'mlx::core::Copy::jvp()'],['../classmlx_1_1core_1_1_cos.html#a99dd0b7e4aa2c838b77736f1fd539ee1',1,'mlx::core::Cos::jvp()'],['../classmlx_1_1core_1_1_cosh.html#a79facb0882443533f36a0a18407f5863',1,'mlx::core::Cosh::jvp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa9f695100170d5cae999b3da138ce720',1,'mlx::core::CustomTransforms::jvp()'],['../classmlx_1_1core_1_1_divide.html#ae1f408c447b17b3c84fe7f951d95559c',1,'mlx::core::Divide::jvp()'],['../classmlx_1_1core_1_1_div_mod.html#a1267401f25f25847888dd0a00b3fe3b9',1,'mlx::core::DivMod::jvp()'],['../classmlx_1_1core_1_1_select.html#a172df6812c2ea3e9d3c3fc5d527548d6',1,'mlx::core::Select::jvp()'],['../classmlx_1_1core_1_1_remainder.html#a972002173fc00ee86029d12bf1a9ba79',1,'mlx::core::Remainder::jvp()'],['../classmlx_1_1core_1_1_equal.html#a659d484589d7cd96d038922a1a98730f',1,'mlx::core::Equal::jvp()'],['../classmlx_1_1core_1_1_erf.html#ac733d605d80277d613954794eb8c46fe',1,'mlx::core::Erf::jvp()'],['../classmlx_1_1core_1_1_erf_inv.html#aa52710297ab6f7cd6826418c303e64be',1,'mlx::core::ErfInv::jvp()'],['../classmlx_1_1core_1_1_exp.html#aef6721832fcc283b082e35a7d436fa59',1,'mlx::core::Exp::jvp()'],['../classmlx_1_1core_1_1_expm1.html#ad463730632a00945d3a8addfdaec67b1',1,'mlx::core::Expm1::jvp()'],['../classmlx_1_1core_1_1_expand_dims.html#a2ccfe836a715dd8fa908731523f3ce5f',1,'mlx::core::ExpandDims::jvp()'],['../classmlx_1_1core_1_1_f_f_t.html#a34578814b6576f7b7b447541984ecba6',1,'mlx::core::FFT::jvp()'],['../classmlx_1_1core_1_1_flatten.html#ae0351b8d8088baa09db927ac8546186b',1,'mlx::core::Flatten::jvp()'],['../classmlx_1_1core_1_1_floor.html#aa47bc360ec563b6e7d93e8b50626d8af',1,'mlx::core::Floor::jvp()'],['../classmlx_1_1core_1_1_full.html#a281a865d0664596ac8d05ea8e7f26407',1,'mlx::core::Full::jvp()'],['../classmlx_1_1core_1_1_gather.html#ac54ef8fac92ab190f1793f3dd95b9e8d',1,'mlx::core::Gather::jvp()'],['../classmlx_1_1core_1_1_greater.html#aa47a9f80f45daf6a405e34f6dc7c99c1',1,'mlx::core::Greater::jvp()'],['../classmlx_1_1core_1_1_greater_equal.html#ac7346080aaaa01d52896127f383f9d20',1,'mlx::core::GreaterEqual::jvp()'],['../classmlx_1_1core_1_1_hadamard.html#a22b9d55ae3ba5eef63505124696e712a',1,'mlx::core::Hadamard::jvp()'],['../classmlx_1_1core_1_1_imag.html#ac01c5ed9b886983450ed9f049ddac55a',1,'mlx::core::Imag::jvp()'],['../classmlx_1_1core_1_1_less.html#af1493d566f6d940b8f674aac17f5dfce',1,'mlx::core::Less::jvp()'],['../classmlx_1_1core_1_1_less_equal.html#addfe62d3557d216f8307bdf1cbff6a8f',1,'mlx::core::LessEqual::jvp()'],['../classmlx_1_1core_1_1_log.html#ac646d4155322c34f58183d97301e3832',1,'mlx::core::Log::jvp()'],['../classmlx_1_1core_1_1_log1p.html#a537e44c7c993daf48698082e75e71ba2',1,'mlx::core::Log1p::jvp()'],['../classmlx_1_1core_1_1_logical_not.html#a4838c483ced707cfda3d6cd24bf4667c',1,'mlx::core::LogicalNot::jvp()'],['../classmlx_1_1core_1_1_logical_and.html#a78d3be71da224ea19158cf9e8c4cf434',1,'mlx::core::LogicalAnd::jvp()'],['../classmlx_1_1core_1_1_logical_or.html#a292de6001c551214c8152a7a5b0e6bd4',1,'mlx::core::LogicalOr::jvp()'],['../classmlx_1_1core_1_1_log_add_exp.html#aea2d1d58794e86f3488219ed3fa14329',1,'mlx::core::LogAddExp::jvp()'],['../classmlx_1_1core_1_1_maximum.html#a25ac5d5b453e571bf7240aa8de103c39',1,'mlx::core::Maximum::jvp()'],['../classmlx_1_1core_1_1_minimum.html#a10acf4fef35eed7ca55d131b5ae2d038',1,'mlx::core::Minimum::jvp()'],['../classmlx_1_1core_1_1_multiply.html#a79f7f0bb70de2e3e41a66c96285325b4',1,'mlx::core::Multiply::jvp()'],['../classmlx_1_1core_1_1_negative.html#a7d918f9b26b8fb7b047a27d85ebab979',1,'mlx::core::Negative::jvp()'],['../classmlx_1_1core_1_1_not_equal.html#ae2d3e5776efaefed7f4c73f679b02f17',1,'mlx::core::NotEqual::jvp()'],['../classmlx_1_1core_1_1_pad.html#a6e43a42032ef11497e8d91290574ec72',1,'mlx::core::Pad::jvp()'],['../classmlx_1_1core_1_1_partition.html#a310f569a163958940ed02cf52079746a',1,'mlx::core::Partition::jvp()'],['../classmlx_1_1core_1_1_power.html#a3e78b06453faa4fd149fd19c0e7a300a',1,'mlx::core::Power::jvp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ae51fdd0b81dd26c6687577567c126e23',1,'mlx::core::QuantizedMatmul::jvp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#adc579058752b927c71b45a962d4869e0',1,'mlx::core::GatherQMM::jvp()'],['../classmlx_1_1core_1_1_real.html#adff418a54970e2344bd3c2885aae5526',1,'mlx::core::Real::jvp()'],['../classmlx_1_1core_1_1_reshape.html#ab8fc28748991017cc3e29f93c91087a5',1,'mlx::core::Reshape::jvp()'],['../classmlx_1_1core_1_1_round.html#a032075a7d0dde2dba6189636d216c5e7',1,'mlx::core::Round::jvp()'],['../classmlx_1_1core_1_1_scan.html#a6f9c862f4fbc7eaf430a361cdd8933ee',1,'mlx::core::Scan::jvp()'],['../classmlx_1_1core_1_1_scatter.html#a270fa8ccf36ce4bbbc23875139223934',1,'mlx::core::Scatter::jvp()'],['../classmlx_1_1core_1_1_sigmoid.html#a62ca1c440896e32958c77af3340847db',1,'mlx::core::Sigmoid::jvp()'],['../classmlx_1_1core_1_1_sign.html#a957992c7aa0e86cf06f861a94372086b',1,'mlx::core::Sign::jvp()'],['../classmlx_1_1core_1_1_sin.html#af662d10180967399820496477ff050de',1,'mlx::core::Sin::jvp()'],['../classmlx_1_1core_1_1_sinh.html#a86e2b37823daf20a4c74c9f273215f9c',1,'mlx::core::Sinh::jvp()'],['../classmlx_1_1core_1_1_slice.html#a8288324045ab21d6c97b1695ce86ef36',1,'mlx::core::Slice::jvp()'],['../classmlx_1_1core_1_1_slice_update.html#a0ce3248cc61dae2b51d7aa8ee4197611',1,'mlx::core::SliceUpdate::jvp()'],['../classmlx_1_1core_1_1_dynamic_slice.html#acd0d2d6d83d4112e9e6fdd9ca8072ac3',1,'mlx::core::DynamicSlice::jvp()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ad1eae28869ebc2ecad87a9a01e314d56',1,'mlx::core::DynamicSliceUpdate::jvp()'],['../classmlx_1_1core_1_1_softmax.html#af96172634a24332b0fc8d7ca7e73f19f',1,'mlx::core::Softmax::jvp()'],['../classmlx_1_1core_1_1_sort.html#af113ac983473433eec851c8fddfcba62',1,'mlx::core::Sort::jvp()'],['../classmlx_1_1core_1_1_split.html#ab8a8d30fd1ebf0891f41f3c32eabe282',1,'mlx::core::Split::jvp()'],['../classmlx_1_1core_1_1_square.html#a822629b93b91e2bef29959431d95e22d',1,'mlx::core::Square::jvp()'],['../classmlx_1_1core_1_1_sqrt.html#a78544b1fb5da0c14bce3051ffd177818',1,'mlx::core::Sqrt::jvp()'],['../classmlx_1_1core_1_1_subtract.html#a8100081a99df5166f02efc76d6641220',1,'mlx::core::Subtract::jvp()'],['../classmlx_1_1core_1_1_squeeze.html#a65ac5f63f98d85453ad884e9fa6e8083',1,'mlx::core::Squeeze::jvp()'],['../classmlx_1_1core_1_1_tan.html#a5d7c76122d63619df17b0e45450bc8f2',1,'mlx::core::Tan::jvp()'],['../classmlx_1_1core_1_1_tanh.html#ae0fbb5370dc1c3a4fb0dd02ca28a832a',1,'mlx::core::Tanh::jvp()'],['../classmlx_1_1core_1_1_unflatten.html#aa3da5fc9920581931d6f9d4236a6d8e5',1,'mlx::core::Unflatten::jvp()'],['../classmlx_1_1core_1_1_transpose.html#ac1a523e25ab7fd9df4da363a922afbe1',1,'mlx::core::Transpose::jvp()'],['../namespacemlx_1_1core.html#a179a632200366c223d6ab56d3e032592',1,'mlx::core::jvp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &tangents)'],['../namespacemlx_1_1core.html#af38e7582db29519bb39326f6fa531d20',1,'mlx::core::jvp(const std::function< array(const array &)> &fun, const array &primal, const array &tangent)']]] + ['jvp_0',['jvp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#aeaf6f2b5955e7417cd1e36db42c45a80',1,'mlx::core::distributed::AllReduce::jvp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a96f08a4ea8453d0b4b737c7b07972913',1,'mlx::core::distributed::AllGather::jvp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#ac77b28702654df8e7d882a49357a9584',1,'mlx::core::fast::Custom::jvp()'],['../classmlx_1_1core_1_1_primitive.html#a9fecf38f53da08ba1947543c2b3158c2',1,'mlx::core::Primitive::jvp()'],['../classmlx_1_1core_1_1_abs.html#a6c1e6eeaf4f5e63898c3487106e88e11',1,'mlx::core::Abs::jvp()'],['../classmlx_1_1core_1_1_add.html#a77230069f76fe60a2fe1007822a277b7',1,'mlx::core::Add::jvp()'],['../classmlx_1_1core_1_1_add_m_m.html#a6572a4ffdd09ab857d3c09d9c5103f66',1,'mlx::core::AddMM::jvp()'],['../classmlx_1_1core_1_1_arc_cos.html#a240079c616f1a1f127aa783308096fe9',1,'mlx::core::ArcCos::jvp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a80fcb790649219c30260af903b76a1d7',1,'mlx::core::ArcCosh::jvp()'],['../classmlx_1_1core_1_1_arc_sin.html#a37affc8c5e84e5c54e73a71fc0821ea4',1,'mlx::core::ArcSin::jvp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79ebf2f6dfecbfbb93170fdd1ca87bf4',1,'mlx::core::ArcSinh::jvp()'],['../classmlx_1_1core_1_1_arc_tan.html#a0f5590a2297fc133b4b0a15f9dd0c760',1,'mlx::core::ArcTan::jvp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a01675433f2a4fa466b2f48272dbca738',1,'mlx::core::ArcTan2::jvp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a534ebdbfe77241884630d25021274c4a',1,'mlx::core::ArcTanh::jvp()'],['../classmlx_1_1core_1_1_arg_partition.html#aedea4b47f947a6fe358dd1238cdfb595',1,'mlx::core::ArgPartition::jvp()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03bb925e1b488c560bc3d67ce62ba6fa',1,'mlx::core::ArgReduce::jvp()'],['../classmlx_1_1core_1_1_as_type.html#a213400967150c57da35795e1c9f65ca0',1,'mlx::core::AsType::jvp()'],['../classmlx_1_1core_1_1_as_strided.html#a8ff0a398c47b42e08bc1122e07a02b53',1,'mlx::core::AsStrided::jvp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a1dae6ce5dc0498d20530403fe5c5531d',1,'mlx::core::BitwiseBinary::jvp()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a830bae1f3f9078bd5b422ce6e46685a7',1,'mlx::core::BroadcastAxes::jvp()'],['../classmlx_1_1core_1_1_broadcast.html#ae2fc3851a117079244708864be770ece',1,'mlx::core::Broadcast::jvp()'],['../classmlx_1_1core_1_1_ceil.html#a7ad74b27d9f26c886c2af516b845f066',1,'mlx::core::Ceil::jvp()'],['../classmlx_1_1core_1_1_compiled.html#aa385fe28626856ca5f57161b47a3c205',1,'mlx::core::Compiled::jvp()'],['../classmlx_1_1core_1_1_concatenate.html#a9f9e7a9dc3a00e02b84c94e1868baff1',1,'mlx::core::Concatenate::jvp()'],['../classmlx_1_1core_1_1_contiguous.html#a1f9fcae7235e0ae9217825b78cb0f991',1,'mlx::core::Contiguous::jvp()'],['../classmlx_1_1core_1_1_copy.html#a5acf02aa360cbefd86749fe9877b29cc',1,'mlx::core::Copy::jvp()'],['../classmlx_1_1core_1_1_cos.html#a99dd0b7e4aa2c838b77736f1fd539ee1',1,'mlx::core::Cos::jvp()'],['../classmlx_1_1core_1_1_cosh.html#a79facb0882443533f36a0a18407f5863',1,'mlx::core::Cosh::jvp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa9f695100170d5cae999b3da138ce720',1,'mlx::core::CustomTransforms::jvp()'],['../classmlx_1_1core_1_1_divide.html#ae1f408c447b17b3c84fe7f951d95559c',1,'mlx::core::Divide::jvp()'],['../classmlx_1_1core_1_1_div_mod.html#a1267401f25f25847888dd0a00b3fe3b9',1,'mlx::core::DivMod::jvp()'],['../classmlx_1_1core_1_1_select.html#a172df6812c2ea3e9d3c3fc5d527548d6',1,'mlx::core::Select::jvp()'],['../classmlx_1_1core_1_1_remainder.html#a972002173fc00ee86029d12bf1a9ba79',1,'mlx::core::Remainder::jvp()'],['../classmlx_1_1core_1_1_equal.html#a659d484589d7cd96d038922a1a98730f',1,'mlx::core::Equal::jvp()'],['../classmlx_1_1core_1_1_erf.html#ac733d605d80277d613954794eb8c46fe',1,'mlx::core::Erf::jvp()'],['../classmlx_1_1core_1_1_erf_inv.html#aa52710297ab6f7cd6826418c303e64be',1,'mlx::core::ErfInv::jvp()'],['../classmlx_1_1core_1_1_exp.html#aef6721832fcc283b082e35a7d436fa59',1,'mlx::core::Exp::jvp()'],['../classmlx_1_1core_1_1_expm1.html#ad463730632a00945d3a8addfdaec67b1',1,'mlx::core::Expm1::jvp()'],['../classmlx_1_1core_1_1_expand_dims.html#a2ccfe836a715dd8fa908731523f3ce5f',1,'mlx::core::ExpandDims::jvp()'],['../classmlx_1_1core_1_1_f_f_t.html#a34578814b6576f7b7b447541984ecba6',1,'mlx::core::FFT::jvp()'],['../classmlx_1_1core_1_1_flatten.html#ae0351b8d8088baa09db927ac8546186b',1,'mlx::core::Flatten::jvp()'],['../classmlx_1_1core_1_1_floor.html#aa47bc360ec563b6e7d93e8b50626d8af',1,'mlx::core::Floor::jvp()'],['../classmlx_1_1core_1_1_full.html#a281a865d0664596ac8d05ea8e7f26407',1,'mlx::core::Full::jvp()'],['../classmlx_1_1core_1_1_gather.html#ac54ef8fac92ab190f1793f3dd95b9e8d',1,'mlx::core::Gather::jvp()'],['../classmlx_1_1core_1_1_gather_axis.html#a4f6015bf2c9bb8773118eb51be45b378',1,'mlx::core::GatherAxis::jvp()'],['../classmlx_1_1core_1_1_greater.html#aa47a9f80f45daf6a405e34f6dc7c99c1',1,'mlx::core::Greater::jvp()'],['../classmlx_1_1core_1_1_greater_equal.html#ac7346080aaaa01d52896127f383f9d20',1,'mlx::core::GreaterEqual::jvp()'],['../classmlx_1_1core_1_1_hadamard.html#a22b9d55ae3ba5eef63505124696e712a',1,'mlx::core::Hadamard::jvp()'],['../classmlx_1_1core_1_1_imag.html#ac01c5ed9b886983450ed9f049ddac55a',1,'mlx::core::Imag::jvp()'],['../classmlx_1_1core_1_1_less.html#af1493d566f6d940b8f674aac17f5dfce',1,'mlx::core::Less::jvp()'],['../classmlx_1_1core_1_1_less_equal.html#addfe62d3557d216f8307bdf1cbff6a8f',1,'mlx::core::LessEqual::jvp()'],['../classmlx_1_1core_1_1_log.html#ac646d4155322c34f58183d97301e3832',1,'mlx::core::Log::jvp()'],['../classmlx_1_1core_1_1_log1p.html#a537e44c7c993daf48698082e75e71ba2',1,'mlx::core::Log1p::jvp()'],['../classmlx_1_1core_1_1_logical_not.html#a4838c483ced707cfda3d6cd24bf4667c',1,'mlx::core::LogicalNot::jvp()'],['../classmlx_1_1core_1_1_logical_and.html#a78d3be71da224ea19158cf9e8c4cf434',1,'mlx::core::LogicalAnd::jvp()'],['../classmlx_1_1core_1_1_logical_or.html#a292de6001c551214c8152a7a5b0e6bd4',1,'mlx::core::LogicalOr::jvp()'],['../classmlx_1_1core_1_1_log_add_exp.html#aea2d1d58794e86f3488219ed3fa14329',1,'mlx::core::LogAddExp::jvp()'],['../classmlx_1_1core_1_1_matmul.html#a6d949d8ab0fab0395532706c174686d5',1,'mlx::core::Matmul::jvp()'],['../classmlx_1_1core_1_1_maximum.html#a25ac5d5b453e571bf7240aa8de103c39',1,'mlx::core::Maximum::jvp()'],['../classmlx_1_1core_1_1_minimum.html#a10acf4fef35eed7ca55d131b5ae2d038',1,'mlx::core::Minimum::jvp()'],['../classmlx_1_1core_1_1_multiply.html#a79f7f0bb70de2e3e41a66c96285325b4',1,'mlx::core::Multiply::jvp()'],['../classmlx_1_1core_1_1_negative.html#a7d918f9b26b8fb7b047a27d85ebab979',1,'mlx::core::Negative::jvp()'],['../classmlx_1_1core_1_1_not_equal.html#ae2d3e5776efaefed7f4c73f679b02f17',1,'mlx::core::NotEqual::jvp()'],['../classmlx_1_1core_1_1_pad.html#a6e43a42032ef11497e8d91290574ec72',1,'mlx::core::Pad::jvp()'],['../classmlx_1_1core_1_1_partition.html#a310f569a163958940ed02cf52079746a',1,'mlx::core::Partition::jvp()'],['../classmlx_1_1core_1_1_power.html#a3e78b06453faa4fd149fd19c0e7a300a',1,'mlx::core::Power::jvp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ae51fdd0b81dd26c6687577567c126e23',1,'mlx::core::QuantizedMatmul::jvp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#adc579058752b927c71b45a962d4869e0',1,'mlx::core::GatherQMM::jvp()'],['../classmlx_1_1core_1_1_real.html#adff418a54970e2344bd3c2885aae5526',1,'mlx::core::Real::jvp()'],['../classmlx_1_1core_1_1_reshape.html#ab8fc28748991017cc3e29f93c91087a5',1,'mlx::core::Reshape::jvp()'],['../classmlx_1_1core_1_1_round.html#a032075a7d0dde2dba6189636d216c5e7',1,'mlx::core::Round::jvp()'],['../classmlx_1_1core_1_1_scan.html#a6f9c862f4fbc7eaf430a361cdd8933ee',1,'mlx::core::Scan::jvp()'],['../classmlx_1_1core_1_1_scatter.html#a270fa8ccf36ce4bbbc23875139223934',1,'mlx::core::Scatter::jvp()'],['../classmlx_1_1core_1_1_scatter_axis.html#a77129b601e5ca9d97669a8b0fdc69805',1,'mlx::core::ScatterAxis::jvp()'],['../classmlx_1_1core_1_1_sigmoid.html#a62ca1c440896e32958c77af3340847db',1,'mlx::core::Sigmoid::jvp()'],['../classmlx_1_1core_1_1_sign.html#a957992c7aa0e86cf06f861a94372086b',1,'mlx::core::Sign::jvp()'],['../classmlx_1_1core_1_1_sin.html#af662d10180967399820496477ff050de',1,'mlx::core::Sin::jvp()'],['../classmlx_1_1core_1_1_sinh.html#a86e2b37823daf20a4c74c9f273215f9c',1,'mlx::core::Sinh::jvp()'],['../classmlx_1_1core_1_1_slice.html#a8288324045ab21d6c97b1695ce86ef36',1,'mlx::core::Slice::jvp()'],['../classmlx_1_1core_1_1_slice_update.html#a0ce3248cc61dae2b51d7aa8ee4197611',1,'mlx::core::SliceUpdate::jvp()'],['../classmlx_1_1core_1_1_dynamic_slice.html#acd0d2d6d83d4112e9e6fdd9ca8072ac3',1,'mlx::core::DynamicSlice::jvp()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#ad1eae28869ebc2ecad87a9a01e314d56',1,'mlx::core::DynamicSliceUpdate::jvp()'],['../classmlx_1_1core_1_1_softmax.html#af96172634a24332b0fc8d7ca7e73f19f',1,'mlx::core::Softmax::jvp()'],['../classmlx_1_1core_1_1_sort.html#af113ac983473433eec851c8fddfcba62',1,'mlx::core::Sort::jvp()'],['../classmlx_1_1core_1_1_split.html#ab8a8d30fd1ebf0891f41f3c32eabe282',1,'mlx::core::Split::jvp()'],['../classmlx_1_1core_1_1_square.html#a822629b93b91e2bef29959431d95e22d',1,'mlx::core::Square::jvp()'],['../classmlx_1_1core_1_1_sqrt.html#a78544b1fb5da0c14bce3051ffd177818',1,'mlx::core::Sqrt::jvp()'],['../classmlx_1_1core_1_1_subtract.html#a8100081a99df5166f02efc76d6641220',1,'mlx::core::Subtract::jvp()'],['../classmlx_1_1core_1_1_squeeze.html#a65ac5f63f98d85453ad884e9fa6e8083',1,'mlx::core::Squeeze::jvp()'],['../classmlx_1_1core_1_1_tan.html#a5d7c76122d63619df17b0e45450bc8f2',1,'mlx::core::Tan::jvp()'],['../classmlx_1_1core_1_1_tanh.html#ae0fbb5370dc1c3a4fb0dd02ca28a832a',1,'mlx::core::Tanh::jvp()'],['../classmlx_1_1core_1_1_unflatten.html#aa3da5fc9920581931d6f9d4236a6d8e5',1,'mlx::core::Unflatten::jvp()'],['../classmlx_1_1core_1_1_transpose.html#ac1a523e25ab7fd9df4da363a922afbe1',1,'mlx::core::Transpose::jvp()'],['../namespacemlx_1_1core.html#a179a632200366c223d6ab56d3e032592',1,'mlx::core::jvp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &tangents)'],['../namespacemlx_1_1core.html#af38e7582db29519bb39326f6fa531d20',1,'mlx::core::jvp(const std::function< array(const array &)> &fun, const array &primal, const array &tangent)']]] ]; diff --git a/docs/build/html/search/functions_c.js b/docs/build/html/search/functions_c.js index 6836b835d..a9b1926de 100644 --- a/docs/build/html/search/functions_c.js +++ b/docs/build/html/search/functions_c.js @@ -19,22 +19,22 @@ var searchData= ['lib_5fname_16',['lib_name',['../classmlx_1_1core_1_1_compiled.html#ae5c16cb91ac31b97e7652cc526c07439',1,'mlx::core::Compiled']]], ['linspace_17',['linspace',['../group__ops.html#ga968bcabed902311dcfbd903b0fb886ec',1,'mlx::core']]], ['load_18',['Load',['../classmlx_1_1core_1_1_load.html#a3aa8a537cd90bab048df47dca1ed526a',1,'mlx::core::Load']]], - ['load_19',['load',['../struct_read_writer.html#a120eaf4b5f32e80972a18d14e82a2d75',1,'ReadWriter::load()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#ac73006b36fc710feda3a7c796e21415c',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa5426c6beabfb3ee41b58f01b3392a96',1,'mlx::steel::MMATile::load(const threadgroup U *src)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa9e484d8cae936503898d5b772c573f9',1,'mlx::steel::MMATile::load(const device U *src, const int ld)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#ac73006b36fc710feda3a7c796e21415c',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa5426c6beabfb3ee41b58f01b3392a96',1,'mlx::steel::MMATile::load(const threadgroup U *src)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa9e484d8cae936503898d5b772c573f9',1,'mlx::steel::MMATile::load(const device U *src, const int ld)'],['../struct_read_writer.html#a8a97ba42db5692898ef7391db08d8fd0',1,'ReadWriter::load() const'],['../struct_read_writer.html#a2506ee61be67826ac9494efb12a81900',1,'ReadWriter::load() const'],['../namespacemlx_1_1core.html#a954de19249da7c1fa39b89bdc47368aa',1,'mlx::core::load(array &out, size_t offset, const std::shared_ptr< io::Reader > &reader, bool swap_endianess)'],['../namespacemlx_1_1core.html#abada9bfa834d7423959362386720f3db',1,'mlx::core::load(std::shared_ptr< io::Reader > in_stream, StreamOrDevice s={})'],['../namespacemlx_1_1core.html#ac71a08bf4c052ae3c77e9e89cbea071d',1,'mlx::core::load(std::string file, StreamOrDevice s={})']]], + ['load_19',['load',['../struct_read_writer.html#a120eaf4b5f32e80972a18d14e82a2d75',1,'ReadWriter::load()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ac73006b36fc710feda3a7c796e21415c',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa5426c6beabfb3ee41b58f01b3392a96',1,'mlx::steel::MMATile::load(const threadgroup U *src)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa9e484d8cae936503898d5b772c573f9',1,'mlx::steel::MMATile::load(const device U *src, const int ld)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ac73006b36fc710feda3a7c796e21415c',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa5426c6beabfb3ee41b58f01b3392a96',1,'mlx::steel::MMATile::load(const threadgroup U *src)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa9e484d8cae936503898d5b772c573f9',1,'mlx::steel::MMATile::load(const device U *src, const int ld)'],['../struct_read_writer.html#a8a97ba42db5692898ef7391db08d8fd0',1,'ReadWriter::load() const'],['../struct_read_writer.html#a2506ee61be67826ac9494efb12a81900',1,'ReadWriter::load() const'],['../namespacemlx_1_1core.html#a954de19249da7c1fa39b89bdc47368aa',1,'mlx::core::load()'],['../namespacemlx_1_1core_1_1simd.html#a4041676517d96870293e5448c7e2b5a4',1,'mlx::core::simd::load()'],['../namespacemlx_1_1core.html#abada9bfa834d7423959362386720f3db',1,'mlx::core::load(std::shared_ptr< io::Reader > in_stream, StreamOrDevice s={})'],['../namespacemlx_1_1core.html#ac71a08bf4c052ae3c77e9e89cbea071d',1,'mlx::core::load(std::string file, StreamOrDevice s={})']]], ['load_5fgguf_20',['load_gguf',['../namespacemlx_1_1core.html#a2aa12b351ce559deb14cda0a5292c2ce',1,'mlx::core']]], ['load_5fpadded_21',['load_padded',['../struct_read_writer.html#add5bd3f647793a5a19d63197a19df73c',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#af3ce6bbb1a8dfb3bab1ae18d3eb45bc0',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#ab116f4569bb9dc6eaef0d8d08472e239',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const']]], - ['load_5fsafe_22',['load_safe',['../struct_g_e_m_v_kernel.html#a04bb72da9a93d6d1eba468fa311bbba7',1,'GEMVKernel::load_safe()'],['../struct_quantized_block_loader.html#a699dc9aa284b8fbf870310bbb224465b',1,'QuantizedBlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader_t.html#ac2d95e35ba39e0984e6f1e58ca935f7d',1,'mlx::steel::BlockLoaderT::load_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#ad22aaee4a2938cbdd315b39eda84e07d',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3a4af67813908109da08ce7352f82da',1,'mlx::steel::MMATile::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#ad22aaee4a2938cbdd315b39eda84e07d',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3a4af67813908109da08ce7352f82da',1,'mlx::steel::MMATile::load_safe()'],['../scan_8h.html#ae8eb101e538b85f8a4bcf451489ae0ac',1,'load_safe(): scan.h']]], + ['load_5fsafe_22',['load_safe',['../struct_g_e_m_v_kernel.html#a04bb72da9a93d6d1eba468fa311bbba7',1,'GEMVKernel::load_safe()'],['../struct_quantized_block_loader.html#a699dc9aa284b8fbf870310bbb224465b',1,'QuantizedBlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader_t.html#ac2d95e35ba39e0984e6f1e58ca935f7d',1,'mlx::steel::BlockLoaderT::load_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ad22aaee4a2938cbdd315b39eda84e07d',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3a4af67813908109da08ce7352f82da',1,'mlx::steel::MMATile::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ad22aaee4a2938cbdd315b39eda84e07d',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe()'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3a4af67813908109da08ce7352f82da',1,'mlx::steel::MMATile::load_safe()'],['../scan_8h.html#ae8eb101e538b85f8a4bcf451489ae0ac',1,'load_safe(): scan.h']]], ['load_5fsafetensors_23',['load_safetensors',['../namespacemlx_1_1core.html#a96cc40e1af8c4626c813ce4859f70a5c',1,'mlx::core::load_safetensors(std::shared_ptr< io::Reader > in_stream, StreamOrDevice s={})'],['../namespacemlx_1_1core.html#af7eea1682a38d363c56a066321e6d526',1,'mlx::core::load_safetensors(const std::string &file, StreamOrDevice s={})']]], ['load_5fstrided_24',['load_strided',['../struct_read_writer.html#a998ef484bade81f726b9edfc6b878197',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a3d9c8cbc582cad6b5218339d0f721559',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a795a71a8e1f154a5af415ebe1b3f0713',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a0935b946b8bf2e769427fcbf2da2f7be',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a7d45368c74a8b7c632659504b3273a13',1,'ReadWriter::load_strided(int stride, int overall_n)']]], ['load_5funsafe_25',['load_unsafe',['../struct_g_e_m_v_kernel.html#a6013e9c5b2f72fa1311dd038172df0ce',1,'GEMVKernel::load_unsafe()'],['../struct_quantized_block_loader.html#a86009527cb4b53e4c21fd6b1f78cfefc',1,'QuantizedBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader.html#a6c9e27f11f48b34580ed2c7e9cad9a27',1,'mlx::steel::BlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader_t.html#acb743f32146fdc7986264b7beb35fb38',1,'mlx::steel::BlockLoaderT::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a961836be363409744e48e595d5e0c2ec',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8034abc10483487fc94313e3674d1111',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a69e2f7c9814d1cc1c5c267be8618dc55',1,'mlx::steel::Conv2DWeightBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#aa11d1a142bc868df462f48a7102147f3',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a0e262b003ac0e7ee6272585eac921704',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3859ca11b5991ef6ee9b99afdc3ea30a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8f078982186421f5b484c0b53af9c655',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader.html#a6c9e27f11f48b34580ed2c7e9cad9a27',1,'mlx::steel::BlockLoader::load_unsafe()'],['../scan_8h.html#a9c415d07921f3961bad0a00a34f4a9a3',1,'load_unsafe(U values[N_READS], const device T *input): scan.h']]], ['load_5fvector_26',['load_vector',['../quantized_8h.html#a8dbace41de9e1e21dd59d016db11b3e9',1,'quantized.h']]], ['load_5fvector_5fsafe_27',['load_vector_safe',['../quantized_8h.html#aa69e143d646fad332c1a53e8c9b337b7',1,'quantized.h']]], - ['location_28',['location',['../struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e',1,'LoopedElemToLoc::location()'],['../struct_looped_elem_to_loc.html#a66b84b12f6c1494e5908989ed2849a9f',1,'LoopedElemToLoc< 1, OffsetT, true >::location()'],['../struct_looped_elem_to_loc.html#a89d9ec4dc2f2f0d77e27aa0c05f261ef',1,'LoopedElemToLoc< 1, OffsetT, false >::location()']]], + ['location_28',['location',['../struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e',1,'LoopedElemToLoc::location()'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a66b84b12f6c1494e5908989ed2849a9f',1,'LoopedElemToLoc< 1, OffsetT, true >::location()'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a89d9ec4dc2f2f0d77e27aa0c05f261ef',1,'LoopedElemToLoc< 1, OffsetT, false >::location()'],['../struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e',1,'LoopedElemToLoc< 1, OffsetT, false >::location()'],['../struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e',1,'LoopedElemToLoc< 1, OffsetT, true >::location()']]], ['log_29',['Log',['../classmlx_1_1core_1_1_log.html#a663e54790c60b56eb0ff09f4f6635fb9',1,'mlx::core::Log']]], - ['log_30',['log',['../namespacemetal.html#a423a9f4f2fc7ef5ec7eda061277b51b6',1,'metal::log()'],['../namespacemetal_1_1fast.html#aef942e7f9e5c2e58c58644ab1bdd58d1',1,'metal::fast::log()'],['../namespacemetal_1_1precise.html#a341c2b8c27d1bed860f85f8b355023d4',1,'metal::precise::log()'],['../group__ops.html#ga6fb22d4926133573e430fcc92f4eef31',1,'mlx::core::log()']]], - ['log10_31',['log10',['../namespacemetal.html#a042b98827baa910e9d726227cec55a80',1,'metal::log10()'],['../namespacemetal_1_1fast.html#a0d1150cf2deee5100a7ea2988b3bb39e',1,'metal::fast::log10()'],['../namespacemetal_1_1precise.html#a44239067e8e9248b1574353f98e94d72',1,'metal::precise::log10()'],['../group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e',1,'mlx::core::log10()']]], + ['log_30',['log',['../namespacemlx_1_1core_1_1simd.html#a8cec82f4fb15bfd31d7554c6c09ceed4',1,'mlx::core::simd::log(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a97a8ca857fe0edd84c68dc0f3dc2c6c4',1,'mlx::core::simd::log(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#abd09d3f5989558ce5156549a94d0fb04',1,'mlx::core::simd::log(Simd< T, 1 > in)'],['../namespacemetal.html#a423a9f4f2fc7ef5ec7eda061277b51b6',1,'metal::log()'],['../namespacemetal_1_1fast.html#aef942e7f9e5c2e58c58644ab1bdd58d1',1,'metal::fast::log()'],['../namespacemetal_1_1precise.html#a341c2b8c27d1bed860f85f8b355023d4',1,'metal::precise::log()'],['../group__ops.html#ga6fb22d4926133573e430fcc92f4eef31',1,'mlx::core::log()']]], + ['log10_31',['log10',['../namespacemlx_1_1core_1_1simd.html#a3cb6ea94836e999c07329b34c501ed85',1,'mlx::core::simd::log10(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#aba81b735e8f99cedf8b4846b2ab4e236',1,'mlx::core::simd::log10(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#adf754ade6cc1dd0e0bae0e31c7b513a2',1,'mlx::core::simd::log10(Simd< T, 1 > in)'],['../namespacemetal.html#a042b98827baa910e9d726227cec55a80',1,'metal::log10()'],['../namespacemetal_1_1fast.html#a0d1150cf2deee5100a7ea2988b3bb39e',1,'metal::fast::log10()'],['../namespacemetal_1_1precise.html#a44239067e8e9248b1574353f98e94d72',1,'metal::precise::log10()'],['../group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e',1,'mlx::core::log10()']]], ['log1p_32',['Log1p',['../classmlx_1_1core_1_1_log1p.html#ab0d6eb90c6f98775fce56f3446ff127a',1,'mlx::core::Log1p']]], - ['log1p_33',['log1p',['../backend_2metal_2kernels_2utils_8h.html#a27c03f2f90ab56db2e4d59559a3d2e9a',1,'log1p(float x): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a3501b665c8837eabf9789ea27a7d6946',1,'log1p(bfloat16_t x): utils.h'],['../group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604',1,'mlx::core::log1p()']]], - ['log2_34',['log2',['../namespacemetal.html#ae894dd5fc13799f120b55cab6267c89c',1,'metal::log2()'],['../namespacemetal_1_1fast.html#a986ef245dd433ae62af864f5cbb07118',1,'metal::fast::log2()'],['../namespacemetal_1_1precise.html#a632dbbdcc1a465cf4739a14306147573',1,'metal::precise::log2()'],['../group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b',1,'mlx::core::log2()']]], + ['log1p_33',['log1p',['../namespacemlx_1_1core_1_1simd.html#ad06680bbc041e76efe2dbff4e11b9a13',1,'mlx::core::simd::log1p(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a830591eb3007fef5d87dc296f5615108',1,'mlx::core::simd::log1p(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a92fcc8037ddb767bff517814ab55c259',1,'mlx::core::simd::log1p(Simd< T, 1 > in)'],['../backend_2metal_2kernels_2utils_8h.html#a27c03f2f90ab56db2e4d59559a3d2e9a',1,'log1p(float x): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a3501b665c8837eabf9789ea27a7d6946',1,'log1p(bfloat16_t x): utils.h'],['../group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604',1,'mlx::core::log1p()']]], + ['log2_34',['log2',['../namespacemlx_1_1core_1_1simd.html#a66426c28a4324b9f617b7018d9354ea1',1,'mlx::core::simd::log2(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#acb1c49b90d029bc4a7eed257ec52791d',1,'mlx::core::simd::log2(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#af9eafa15692dec783860ddae3dd8c072',1,'mlx::core::simd::log2(Simd< T, 1 > in)'],['../namespacemetal.html#ae894dd5fc13799f120b55cab6267c89c',1,'metal::log2()'],['../namespacemetal_1_1fast.html#a986ef245dd433ae62af864f5cbb07118',1,'metal::fast::log2()'],['../namespacemetal_1_1precise.html#a632dbbdcc1a465cf4739a14306147573',1,'metal::precise::log2()'],['../group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b',1,'mlx::core::log2()']]], ['logaddexp_35',['LogAddExp',['../classmlx_1_1core_1_1_log_add_exp.html#ad8938ca90ccf1a3259973fc68902975a',1,'mlx::core::LogAddExp']]], ['logaddexp_36',['logaddexp',['../group__ops.html#gaf985df6609c6bd75a14a844655d89eaa',1,'mlx::core']]], ['logical_5fand_37',['logical_and',['../group__ops.html#ga768977cda8d68cf23f464a6af9907876',1,'mlx::core']]], @@ -44,6 +44,6 @@ var searchData= ['logicalnot_41',['LogicalNot',['../classmlx_1_1core_1_1_logical_not.html#a6f5850b4c78b83d5e2c0d37437fc79b7',1,'mlx::core::LogicalNot']]], ['logicalor_42',['LogicalOr',['../classmlx_1_1core_1_1_logical_or.html#a269c22daca1c15ad010bb860bce93918',1,'mlx::core::LogicalOr']]], ['logsumexp_43',['logsumexp',['../group__ops.html#gacff4eb57c085d571e722083680267ac5',1,'mlx::core::logsumexp(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga59be50b4e92f1dc20b53460cefa3910d',1,'mlx::core::logsumexp(const array &a, StreamOrDevice s={})'],['../group__ops.html#gae3969c7bd24c4f3ab97831df28239689',1,'mlx::core::logsumexp(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafef5cb2159c16a60a95470cc823bdd44',1,'mlx::core::logsumexp(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['loopedelemtoloc_44',['LoopedElemToLoc',['../struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b',1,'LoopedElemToLoc::LoopedElemToLoc()'],['../struct_looped_elem_to_loc.html#abf536c7162d36af7367e390789944c86',1,'LoopedElemToLoc< 1, OffsetT, true >::LoopedElemToLoc()'],['../struct_looped_elem_to_loc.html#a0e21977d9f23b6994773e8e4f3ee70de',1,'LoopedElemToLoc< 1, OffsetT, false >::LoopedElemToLoc()']]], - ['lowest_45',['lowest',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ae81c58b8223e504965183c99d19a2116',1,'metal::_numeric_limits_impl< bfloat16_t >::lowest()'],['../structmlx_1_1core_1_1numeric__limits.html#abc2d9cd8d0a90219f7eb6fd05b98e4ac',1,'mlx::core::numeric_limits< float16_t >::lowest()'],['../structmlx_1_1core_1_1numeric__limits.html#a3623580fbfd92ceb69cdd8e329e18fa8',1,'mlx::core::numeric_limits< bfloat16_t >::lowest()']]] + ['loopedelemtoloc_44',['LoopedElemToLoc',['../struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b',1,'LoopedElemToLoc::LoopedElemToLoc()'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#abf536c7162d36af7367e390789944c86',1,'LoopedElemToLoc< 1, OffsetT, true >::LoopedElemToLoc()'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a0e21977d9f23b6994773e8e4f3ee70de',1,'LoopedElemToLoc< 1, OffsetT, false >::LoopedElemToLoc(int)'],['../struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b',1,'LoopedElemToLoc< 1, OffsetT, false >::LoopedElemToLoc(int dim)'],['../struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b',1,'LoopedElemToLoc< 1, OffsetT, true >::LoopedElemToLoc()']]], + ['lowest_45',['lowest',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ae81c58b8223e504965183c99d19a2116',1,'metal::_numeric_limits_impl< bfloat16_t >::lowest()'],['../structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html#abc2d9cd8d0a90219f7eb6fd05b98e4ac',1,'mlx::core::numeric_limits< float16_t >::lowest()'],['../structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html#a3623580fbfd92ceb69cdd8e329e18fa8',1,'mlx::core::numeric_limits< bfloat16_t >::lowest()']]] ]; diff --git a/docs/build/html/search/functions_d.js b/docs/build/html/search/functions_d.js index 505c8020b..e93d02272 100644 --- a/docs/build/html/search/functions_d.js +++ b/docs/build/html/search/functions_d.js @@ -9,47 +9,48 @@ var searchData= ['malloc_5for_5fwait_6',['malloc_or_wait',['../namespacemlx_1_1core_1_1allocator.html#a86ac0a11ff78f21e717f641716c34abc',1,'mlx::core::allocator']]], ['mat_5fat_7',['mat_at',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a323a4f38cd0693bf333832bb4258b28e',1,'mlx::steel::MMATile::mat_at(const short i, const short j)'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a323a4f38cd0693bf333832bb4258b28e',1,'mlx::steel::MMATile::mat_at(const short i, const short j)']]], ['matmul_8',['Matmul',['../classmlx_1_1core_1_1_matmul.html#adef92f30ab35e540ccb316ea6b94e6f7',1,'mlx::core::Matmul']]], - ['matmul_9',['matmul',['../group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a',1,'mlx::core']]], - ['max_10',['max',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a92320d40a58218e40cc414986ac95c50',1,'metal::_numeric_limits_impl< bfloat16_t >::max()'],['../structmlx_1_1core_1_1numeric__limits.html#a6dd1fadd4cc7c2cec6223977c238c334',1,'mlx::core::numeric_limits< float16_t >::max()'],['../structmlx_1_1core_1_1numeric__limits.html#a01712fcb04266320225c168a0e6f619a',1,'mlx::core::numeric_limits< bfloat16_t >::max()'],['../namespacemetal.html#a853c80479ab2264d9c4587c7bcac767b',1,'metal::max()'],['../namespacemetal_1_1fast.html#a747e2e58092a27fb8b4dd3d16934fb52',1,'metal::fast::max()'],['../namespacemetal_1_1precise.html#a6a954a4e4e3753303d1dc734855a185f',1,'metal::precise::max()'],['../group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7',1,'mlx::core::max(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc',1,'mlx::core::max(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f',1,'mlx::core::max(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga7b638050e03a93f2896c981bc2850a47',1,'mlx::core::max(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['matmul_9',['matmul',['../namespacemlx_1_1core.html#aaacf0afe13d77a5c49ce96f1e833eb2d',1,'mlx::core::matmul(const array &a, const array &b, array &out, bool a_transposed, bool b_transposed, size_t lda, size_t ldb, float alpha, float beta)'],['../group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a',1,'mlx::core::matmul(const array &a, const array &b, StreamOrDevice s={})']]], + ['max_10',['max',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a92320d40a58218e40cc414986ac95c50',1,'metal::_numeric_limits_impl< bfloat16_t >::max()'],['../structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html#a6dd1fadd4cc7c2cec6223977c238c334',1,'mlx::core::numeric_limits< float16_t >::max()'],['../structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html#a01712fcb04266320225c168a0e6f619a',1,'mlx::core::numeric_limits< bfloat16_t >::max()'],['../namespacemlx_1_1core_1_1simd.html#a6fcea259041cecfd042d0c4e6afc4b8f',1,'mlx::core::simd::max(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#a312ecd0ae1c38d32147cee71fd8539d7',1,'mlx::core::simd::max(Simd< T, 1 > x)'],['../namespacemlx_1_1core_1_1simd.html#a995da0f1b4ca8077abbbc6f6a6dfd663',1,'mlx::core::simd::max(Simd< float16_t, N > x)'],['../namespacemetal.html#a853c80479ab2264d9c4587c7bcac767b',1,'metal::max()'],['../namespacemetal_1_1fast.html#a747e2e58092a27fb8b4dd3d16934fb52',1,'metal::fast::max()'],['../namespacemetal_1_1precise.html#a6a954a4e4e3753303d1dc734855a185f',1,'metal::precise::max()'],['../group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7',1,'mlx::core::max(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc',1,'mlx::core::max(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f',1,'mlx::core::max(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga7b638050e03a93f2896c981bc2850a47',1,'mlx::core::max(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], ['max3_11',['max3',['../namespacemetal.html#a00f9c0ad66d969794614f56912eed9c9',1,'metal::max3()'],['../namespacemetal_1_1fast.html#a6fc2cf18ffa8149561864c86dba0f803',1,'metal::fast::max3()'],['../namespacemetal_1_1precise.html#ac490e8614ebd2c9343af1ae6c0d4e82c',1,'metal::precise::max3()']]], ['max_5fops_5fper_5fbuffer_12',['max_ops_per_buffer',['../namespacemlx_1_1core_1_1env.html#aedbf4e739553024c33dd0094dd9107aa',1,'mlx::core::env']]], ['maximum_13',['Maximum',['../classmlx_1_1core_1_1_maximum.html#a28389307e385efe1b2955b86b115e816',1,'mlx::core::Maximum']]], - ['maximum_14',['maximum',['../group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d',1,'mlx::core']]], + ['maximum_14',['maximum',['../namespacemlx_1_1core_1_1simd.html#a7f7a298284e71ddbd2ba0bb6d98b0d16',1,'mlx::core::simd::maximum(Simd< T, N > a, Simd< T, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ab54ff0f073be504e8428912f8e21effd',1,'mlx::core::simd::maximum(Simd< T, 1 > a_, Simd< T, 1 > b_)'],['../namespacemlx_1_1core_1_1simd.html#ae1f11d9c2c15ebecf001d11b3fca5da2',1,'mlx::core::simd::maximum(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aa78385c9cf0b87aabc377b1b47b2929d',1,'mlx::core::simd::maximum(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a0ff63db5f193a57ef3b1fffa374eb15a',1,'mlx::core::simd::maximum(T a, Simd< float16_t, N > b)'],['../group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d',1,'mlx::core::maximum()']]], ['maybeinsertbarrier_15',['maybeInsertBarrier',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ad538ae88f90560063f9ba502e2795991',1,'mlx::core::metal::CommandEncoder::maybeInsertBarrier()'],['../structmlx_1_1core_1_1_command_encoder.html#ad538ae88f90560063f9ba502e2795991',1,'mlx::core::CommandEncoder::maybeInsertBarrier()']]], - ['mb_5fblock_5fmerge_16',['mb_block_merge',['../sort_8h.html#ab381cd57f344bc7304ab580bfdc78807',1,'sort.h']]], - ['mb_5fblock_5fpartition_17',['mb_block_partition',['../sort_8h.html#a32cbe4163b8b0f5cb2c97b256119a4b2',1,'sort.h']]], - ['mb_5fblock_5fsort_18',['mb_block_sort',['../sort_8h.html#a7b5b1180af2b8c6ab814eadd701ce4ba',1,'sort.h']]], + ['mb_5fblock_5fmerge_16',['mb_block_merge',['../sort_8h.html#a9cd2751d251acde874a95330d35fac5f',1,'sort.h']]], + ['mb_5fblock_5fpartition_17',['mb_block_partition',['../sort_8h.html#a812f19ed1db562026edc24e29185fe8c',1,'sort.h']]], + ['mb_5fblock_5fsort_18',['mb_block_sort',['../sort_8h.html#ad1ebc6ed8452f970c37c8aad5414551f',1,'sort.h']]], ['mean_19',['mean',['../group__ops.html#gade46e768fd46b8b640eb16f26abeecef',1,'mlx::core::mean(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga52b59fdd8e8430538e564f5bbcfa31e6',1,'mlx::core::mean(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga066161f3d3e395a1d76c638cb680d444',1,'mlx::core::mean(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga45fba73eab0e3b6e128ed3ce2f43a5da',1,'mlx::core::mean(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], ['median3_20',['median3',['../namespacemetal.html#aa3ff49457ce3c93fc1c0897fd1525157',1,'metal::median3()'],['../namespacemetal_1_1fast.html#a742b55f1e4369921ee7f60d70185bfbc',1,'metal::fast::median3()'],['../namespacemetal_1_1precise.html#a14555ff99c4388493fec48e070144ae2',1,'metal::precise::median3()']]], - ['merge_5fpartition_21',['merge_partition',['../struct_block_merge_sort.html#ab2300cbecb23f3433bad888924c831ca',1,'BlockMergeSort::merge_partition()'],['../struct_kernel_multi_block_merge_sort.html#ab15895b4233aba0e279cc44a07a201fe',1,'KernelMultiBlockMergeSort::merge_partition()']]], - ['merge_5fstep_22',['merge_step',['../struct_block_merge_sort.html#ab65f190edf1851b37c39ad49ce99a43c',1,'BlockMergeSort']]], - ['meshgrid_23',['meshgrid',['../group__ops.html#ga577c911618575314de63d1060656a26e',1,'mlx::core']]], - ['metal_5fkernel_24',['metal_kernel',['../namespacemlx_1_1core_1_1fast.html#ab16436b465dc10ce472193d541d8426e',1,'mlx::core::fast']]], - ['min_25',['min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#adaed80031f5ca0ff69d30ec4c5d0c98f',1,'metal::_numeric_limits_impl< bfloat16_t >::min()'],['../namespacemetal.html#a6653b28c9473087141eddce39878d4d3',1,'metal::min()'],['../namespacemetal_1_1fast.html#a3e958e56a4712687c381a0b64d123e61',1,'metal::fast::min()'],['../namespacemetal_1_1precise.html#afed0da2f7df3505b5dffa2389c3cb36e',1,'metal::precise::min()'],['../group__ops.html#gab27599802617a4c8f9964ab5f4ffee12',1,'mlx::core::min(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga0140b91e9cdfc3fef0da8e332f65a9e8',1,'mlx::core::min(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga6efb83cd46436678c8f8c4af15cc00f5',1,'mlx::core::min(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga36fa315eef677f4143868f552cd26d03',1,'mlx::core::min(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['min3_26',['min3',['../namespacemetal.html#a005510c8c0f964ce2b8aad3ba76a7a3f',1,'metal::min3()'],['../namespacemetal_1_1fast.html#a606a4c1b34ce05ea89ca5af81724036f',1,'metal::fast::min3()'],['../namespacemetal_1_1precise.html#a4d37ce31c3549ca4772a4ee29798e231',1,'metal::precise::min3()']]], - ['minimum_27',['Minimum',['../classmlx_1_1core_1_1_minimum.html#ab0f2ce17108df44b82cff68886b0f6f5',1,'mlx::core::Minimum']]], - ['minimum_28',['minimum',['../group__ops.html#ga49ba00c090f81f331c91b0c97040bce0',1,'mlx::core']]], - ['mlx_5fatomic_5fcompare_5fexchange_5fweak_5fexplicit_29',['mlx_atomic_compare_exchange_weak_explicit',['../atomic_8h.html#ad7f32327ff66354cfa2f0cfdac79316f',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread T *expected, T val, size_t offset): atomic.h'],['../atomic_8h.html#aa8f47b2e9b95d4b00ad51f08b070deb5',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread uint *expected, uint val, size_t offset): atomic.h']]], - ['mlx_5fatomic_5ffetch_5fadd_5fexplicit_30',['mlx_atomic_fetch_add_explicit',['../atomic_8h.html#aad448d9e06e001700b65ca8317216a3b',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fand_5fexplicit_31',['mlx_atomic_fetch_and_explicit',['../atomic_8h.html#a253e3c870c0ddc7c28ab2f6ca2c3eae5',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_32',['mlx_atomic_fetch_max_explicit',['../atomic_8h.html#ac480f2b459a8ad9095cee353e152d00c',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_3c_20float_20_3e_33',['mlx_atomic_fetch_max_explicit< float >',['../atomic_8h.html#a1dce2abfa16417122c4d2bf261129ae4',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_34',['mlx_atomic_fetch_min_explicit',['../atomic_8h.html#a2ec33dca0039bd944d73d1c2b378cc19',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_3c_20float_20_3e_35',['mlx_atomic_fetch_min_explicit< float >',['../atomic_8h.html#ab7d1dc49f319f239b7ee0b7c72976dd0',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmul_5fexplicit_36',['mlx_atomic_fetch_mul_explicit',['../atomic_8h.html#adfdbea60436f14f1af9ce36e2a0a77a3',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5for_5fexplicit_37',['mlx_atomic_fetch_or_explicit',['../atomic_8h.html#ab7391f197001471e4788312bdb6ab37a',1,'atomic.h']]], - ['mlx_5fatomic_5fload_5fexplicit_38',['mlx_atomic_load_explicit',['../atomic_8h.html#a253a4e8c2c5768a069e2791b627dfc99',1,'atomic.h']]], - ['mlx_5fatomic_5fstore_5fexplicit_39',['mlx_atomic_store_explicit',['../atomic_8h.html#a0ae453140b0819a4c02f265334de98c0',1,'atomic.h']]], - ['mma_40',['mma',['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a8028512f5a3d2b6acaf966be529627a3',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a1868f57d57c8adedab2c58492ec76946',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA::mma()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a8028512f5a3d2b6acaf966be529627a3',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag.html#a1868f57d57c8adedab2c58492ec76946',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA::mma()']]], - ['mmatile_41',['MMATile',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3fb310dd08ec23c334511f7b316d1b6',1,'mlx::steel::MMATile::MMATile() thread'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3fb310dd08ec23c334511f7b316d1b6',1,'mlx::steel::MMATile::MMATile() thread']]], - ['move_5for_5fcopy_42',['move_or_copy',['../namespacemlx_1_1core.html#a830a47d8a317dffb0c88e5a7afe6aee2',1,'mlx::core::move_or_copy(const array &in, array &out)'],['../namespacemlx_1_1core.html#a9fcb3711b150cb65c7778a35c51284b2',1,'mlx::core::move_or_copy(const array &in, array &out, const Strides &strides, array::Flags flags, size_t data_size, size_t offset=0)']]], - ['move_5fshared_5fbuffer_43',['move_shared_buffer',['../classmlx_1_1core_1_1array.html#ad41cc5e7aebfcad849ad15d697584cf8',1,'mlx::core::array::move_shared_buffer(array other, const Strides &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a38d7ad605f8282e5e49d0c09e0555c78',1,'mlx::core::array::move_shared_buffer(array other)']]], - ['moveaxis_44',['moveaxis',['../group__ops.html#ga24067d10a842db2c9d509ea48135a2c3',1,'mlx::core']]], - ['mpinplace_45',['MPINPLACE',['../namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2',1,'pocketfft::detail']]], - ['mtl_5fdevice_46',['mtl_device',['../classmlx_1_1core_1_1metal_1_1_device.html#a31dba377f2be44a746db10d1b9367653',1,'mlx::core::metal::Device']]], - ['mtl_5fresidency_5fset_47',['mtl_residency_set',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#ac4bfe5ef5e2eaebc458a1ed1953d15e9',1,'mlx::core::metal::ResidencySet']]], - ['multi_5fiter_48',['multi_iter',['../classpocketfft_1_1detail_1_1multi__iter.html#a9be43bb18840202da6d17988fccc64b9',1,'pocketfft::detail::multi_iter']]], - ['multiply_49',['Multiply',['../classmlx_1_1core_1_1_multiply.html#aca5c50f900321f3eb4d6fbcbc225c00c',1,'mlx::core::Multiply']]], - ['multiply_50',['multiply',['../group__ops.html#gaf57392e641640b5d06e4c99518391c38',1,'mlx::core']]], - ['multivariate_5fnormal_51',['multivariate_normal',['../namespacemlx_1_1core_1_1random.html#ae6a8407fbca0817a4b8c94e02952f77d',1,'mlx::core::random']]] + ['merge_5fpartition_21',['merge_partition',['../struct_block_merge_sort.html#ad5bd0d853e9b4352ecfd902a706d7178',1,'BlockMergeSort::merge_partition()'],['../struct_kernel_multi_block_merge_sort.html#a811e72376de254af2bf5303133562a9a',1,'KernelMultiBlockMergeSort::merge_partition()']]], + ['merge_5fstep_22',['merge_step',['../struct_block_merge_sort.html#a0386ce33d7bcfd12dbb17558d26da1bb',1,'BlockMergeSort']]], + ['meshgrid_23',['meshgrid',['../group__ops.html#ga5ecddb74ba7861eb82eca8653501d5dc',1,'mlx::core']]], + ['metal_5ffast_5fsynch_24',['metal_fast_synch',['../namespacemlx_1_1core_1_1env.html#afa1ecf087fe0c633d5460ddb2c31c945',1,'mlx::core::env']]], + ['metal_5fkernel_25',['metal_kernel',['../namespacemlx_1_1core_1_1fast.html#ab16436b465dc10ce472193d541d8426e',1,'mlx::core::fast']]], + ['min_26',['min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#adaed80031f5ca0ff69d30ec4c5d0c98f',1,'metal::_numeric_limits_impl< bfloat16_t >::min()'],['../namespacemlx_1_1core_1_1simd.html#acd4196d0c66204cfae70b064c305e146',1,'mlx::core::simd::min(Simd< T, N > x)'],['../namespacemlx_1_1core_1_1simd.html#a96db878d780a8da6abad19ac772d08ca',1,'mlx::core::simd::min(Simd< T, 1 > x)'],['../namespacemlx_1_1core_1_1simd.html#a160075943b92d541f2e7f7472eaa5167',1,'mlx::core::simd::min(Simd< float16_t, N > x)'],['../namespacemetal.html#a6653b28c9473087141eddce39878d4d3',1,'metal::min()'],['../namespacemetal_1_1fast.html#a3e958e56a4712687c381a0b64d123e61',1,'metal::fast::min()'],['../namespacemetal_1_1precise.html#afed0da2f7df3505b5dffa2389c3cb36e',1,'metal::precise::min()'],['../group__ops.html#gab27599802617a4c8f9964ab5f4ffee12',1,'mlx::core::min(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga0140b91e9cdfc3fef0da8e332f65a9e8',1,'mlx::core::min(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga6efb83cd46436678c8f8c4af15cc00f5',1,'mlx::core::min(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga36fa315eef677f4143868f552cd26d03',1,'mlx::core::min(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['min3_27',['min3',['../namespacemetal.html#a005510c8c0f964ce2b8aad3ba76a7a3f',1,'metal::min3()'],['../namespacemetal_1_1fast.html#a606a4c1b34ce05ea89ca5af81724036f',1,'metal::fast::min3()'],['../namespacemetal_1_1precise.html#a4d37ce31c3549ca4772a4ee29798e231',1,'metal::precise::min3()']]], + ['minimum_28',['Minimum',['../classmlx_1_1core_1_1_minimum.html#ab0f2ce17108df44b82cff68886b0f6f5',1,'mlx::core::Minimum']]], + ['minimum_29',['minimum',['../namespacemlx_1_1core_1_1simd.html#a1996e77a8c3c24b1ba706113ed9028c4',1,'mlx::core::simd::minimum(Simd< T, N > a, Simd< T, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ac836568622a3e5957c275e115e2fcaf3',1,'mlx::core::simd::minimum(Simd< T, 1 > a_, Simd< T, 1 > b_)'],['../namespacemlx_1_1core_1_1simd.html#abaa09259e92f0fe758dc979d54c327e8',1,'mlx::core::simd::minimum(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ae9ce2f34c97aba7b99223792a86d5c83',1,'mlx::core::simd::minimum(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a17f7baec6300f2ff96ec53fb1943cb49',1,'mlx::core::simd::minimum(T a, Simd< float16_t, N > b)'],['../group__ops.html#ga49ba00c090f81f331c91b0c97040bce0',1,'mlx::core::minimum()']]], + ['mlx_5fatomic_5fcompare_5fexchange_5fweak_5fexplicit_30',['mlx_atomic_compare_exchange_weak_explicit',['../atomic_8h.html#ad7f32327ff66354cfa2f0cfdac79316f',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread T *expected, T val, size_t offset): atomic.h'],['../atomic_8h.html#aa8f47b2e9b95d4b00ad51f08b070deb5',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread uint *expected, uint val, size_t offset): atomic.h']]], + ['mlx_5fatomic_5ffetch_5fadd_5fexplicit_31',['mlx_atomic_fetch_add_explicit',['../atomic_8h.html#aad448d9e06e001700b65ca8317216a3b',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fand_5fexplicit_32',['mlx_atomic_fetch_and_explicit',['../atomic_8h.html#a253e3c870c0ddc7c28ab2f6ca2c3eae5',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_33',['mlx_atomic_fetch_max_explicit',['../atomic_8h.html#ac480f2b459a8ad9095cee353e152d00c',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_3c_20float_20_3e_34',['mlx_atomic_fetch_max_explicit< float >',['../atomic_8h.html#a1dce2abfa16417122c4d2bf261129ae4',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_35',['mlx_atomic_fetch_min_explicit',['../atomic_8h.html#a2ec33dca0039bd944d73d1c2b378cc19',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_3c_20float_20_3e_36',['mlx_atomic_fetch_min_explicit< float >',['../atomic_8h.html#ab7d1dc49f319f239b7ee0b7c72976dd0',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmul_5fexplicit_37',['mlx_atomic_fetch_mul_explicit',['../atomic_8h.html#adfdbea60436f14f1af9ce36e2a0a77a3',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5for_5fexplicit_38',['mlx_atomic_fetch_or_explicit',['../atomic_8h.html#ab7391f197001471e4788312bdb6ab37a',1,'atomic.h']]], + ['mlx_5fatomic_5fload_5fexplicit_39',['mlx_atomic_load_explicit',['../atomic_8h.html#a253a4e8c2c5768a069e2791b627dfc99',1,'atomic.h']]], + ['mlx_5fatomic_5fstore_5fexplicit_40',['mlx_atomic_store_explicit',['../atomic_8h.html#a0ae453140b0819a4c02f265334de98c0',1,'atomic.h']]], + ['mma_41',['mma',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8028512f5a3d2b6acaf966be529627a3',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1868f57d57c8adedab2c58492ec76946',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA::mma()'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8028512f5a3d2b6acaf966be529627a3',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1868f57d57c8adedab2c58492ec76946',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA::mma()']]], + ['mmatile_42',['MMATile',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3fb310dd08ec23c334511f7b316d1b6',1,'mlx::steel::MMATile::MMATile() thread'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aa3fb310dd08ec23c334511f7b316d1b6',1,'mlx::steel::MMATile::MMATile() thread']]], + ['move_5for_5fcopy_43',['move_or_copy',['../namespacemlx_1_1core.html#a830a47d8a317dffb0c88e5a7afe6aee2',1,'mlx::core::move_or_copy(const array &in, array &out)'],['../namespacemlx_1_1core.html#a9fcb3711b150cb65c7778a35c51284b2',1,'mlx::core::move_or_copy(const array &in, array &out, const Strides &strides, array::Flags flags, size_t data_size, size_t offset=0)']]], + ['move_5fshared_5fbuffer_44',['move_shared_buffer',['../classmlx_1_1core_1_1array.html#ad41cc5e7aebfcad849ad15d697584cf8',1,'mlx::core::array::move_shared_buffer(array other, const Strides &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a38d7ad605f8282e5e49d0c09e0555c78',1,'mlx::core::array::move_shared_buffer(array other)']]], + ['moveaxis_45',['moveaxis',['../group__ops.html#ga24067d10a842db2c9d509ea48135a2c3',1,'mlx::core']]], + ['mpinplace_46',['MPINPLACE',['../namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2',1,'pocketfft::detail']]], + ['mtl_5fdevice_47',['mtl_device',['../classmlx_1_1core_1_1metal_1_1_device.html#a31dba377f2be44a746db10d1b9367653',1,'mlx::core::metal::Device']]], + ['mtl_5fresidency_5fset_48',['mtl_residency_set',['../classmlx_1_1core_1_1metal_1_1_residency_set.html#ac4bfe5ef5e2eaebc458a1ed1953d15e9',1,'mlx::core::metal::ResidencySet']]], + ['multi_5fiter_49',['multi_iter',['../classpocketfft_1_1detail_1_1multi__iter.html#a9be43bb18840202da6d17988fccc64b9',1,'pocketfft::detail::multi_iter']]], + ['multiply_50',['Multiply',['../classmlx_1_1core_1_1_multiply.html#aca5c50f900321f3eb4d6fbcbc225c00c',1,'mlx::core::Multiply']]], + ['multiply_51',['multiply',['../group__ops.html#gaf57392e641640b5d06e4c99518391c38',1,'mlx::core']]], + ['multivariate_5fnormal_52',['multivariate_normal',['../namespacemlx_1_1core_1_1random.html#ae6a8407fbca0817a4b8c94e02952f77d',1,'mlx::core::random']]] ]; diff --git a/docs/build/html/search/functions_e.js b/docs/build/html/search/functions_e.js index 4cbc0f20c..67c44bb4b 100644 --- a/docs/build/html/search/functions_e.js +++ b/docs/build/html/search/functions_e.js @@ -3,25 +3,24 @@ var searchData= ['n_5factive_5ftasks_0',['n_active_tasks',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a3c9fa21442974acba3409d49bb033131',1,'mlx::core::scheduler::Scheduler::n_active_tasks()'],['../namespacemlx_1_1core_1_1scheduler.html#a9bf641981df5fc16b0fb0dbacc0c3afd',1,'mlx::core::scheduler::n_active_tasks()']]], ['nan_5fto_5fnum_1',['nan_to_num',['../group__ops.html#gab1467c6a9e675152e768afd6dcfb61de',1,'mlx::core']]], ['nbytes_2',['nbytes',['../classmlx_1_1core_1_1array.html#a387b67cd3ef5cfc1e749c371766c4a05',1,'mlx::core::array']]], - ['nd_5floop_3',['nd_loop',['../namespacemlx_1_1core.html#adbe9ca762e3e6cea8286c4636347c4b5',1,'mlx::core']]], - ['ndarr_4',['ndarr',['../classpocketfft_1_1detail_1_1ndarr.html#a8f0037a172d96cb1ad915a5069175fa2',1,'pocketfft::detail::ndarr']]], - ['ndim_5',['ndim',['../classpocketfft_1_1detail_1_1arr__info.html#ac608c8af2a59a28a0012e308be7ee414',1,'pocketfft::detail::arr_info::ndim()'],['../classmlx_1_1core_1_1array.html#a53006e77d13d9d88b525ef577748939f',1,'mlx::core::array::ndim()']]], - ['negative_6',['Negative',['../classmlx_1_1core_1_1_negative.html#aa3b73395d9fa5b7215dca488bc0d3c70',1,'mlx::core::Negative']]], - ['negative_7',['negative',['../group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6',1,'mlx::core']]], - ['new_5fqueue_8',['new_queue',['../classmlx_1_1core_1_1metal_1_1_device.html#a8135ae2a8c1e6f3861e84d4e60c28b67',1,'mlx::core::metal::Device']]], - ['new_5fscoped_5fmemory_5fpool_9',['new_scoped_memory_pool',['../namespacemlx_1_1core_1_1metal.html#a46583a1aba89449fa72e6cb3a7090981',1,'mlx::core::metal']]], - ['new_5fstream_10',['new_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a157c8da85fa1bddb8eacf8515a3cc879',1,'mlx::core::scheduler::Scheduler::new_stream()'],['../namespacemlx_1_1core_1_1metal.html#a8b4188f9a090a1da42d62b8a369bf106',1,'mlx::core::metal::new_stream()'],['../namespacemlx_1_1core.html#a6f7c63a9be10337b3b96d527e1db3c2f',1,'mlx::core::new_stream()']]], - ['next_11',['next',['../struct_quantized_block_loader.html#a674138ef7c43cc45586ea9f8fd6f6bd9',1,'QuantizedBlockLoader::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../structmlx_1_1steel_1_1_block_loader_t.html#a6008ef45ff980dbe1119da0630f6c697',1,'mlx::steel::BlockLoaderT::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a3b71f379ff9baf39830c92f4f1ecde52',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a78d2b0098311a278be8394edbd5fc731',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aae56c19bb562219770fec38e5666c6ce',1,'mlx::steel::Conv2DWeightBlockLoader::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af9ce1a767266664bea131a5437002c80',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a30b10bebde7f08b89d03bdd9ea0f48da',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3e5ee68ed0ee43f7e979dd4222f76a8c',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a11743cb1c108f42ccdc6e59204a5b3e8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531',1,'LoopedElemToLoc::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394',1,'LoopedElemToLoc::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#adc8113de3152d291331f219bfe5970c2',1,'LoopedElemToLoc< 1, OffsetT, true >::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#a70d92070097fe0622b836afc778bf8d2',1,'LoopedElemToLoc< 1, OffsetT, true >::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#a9d85f32e5460707827df3ba74b905c23',1,'LoopedElemToLoc< 1, OffsetT, false >::next(const constant int *, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#a634f4fa8f4b5f1fdd47a26754ef2eba7',1,'LoopedElemToLoc< 1, OffsetT, false >::next(int n, const constant int *, const constant int64_t *strides)'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a4193c5eac3ef093a740d5305b25d3e18',1,'mlx::core::random::KeySequence::next()']]], - ['next_5fpower_5fof_5f2_12',['next_power_of_2',['../namespacemlx_1_1core.html#a685c0530e338aabc622325685846ce93',1,'mlx::core']]], - ['nextafter_13',['nextafter',['../namespacemetal.html#a9547fd7b09164931986f6db4813bd72d',1,'metal::nextafter()'],['../namespacemetal_1_1fast.html#a4583e8be04fc0bd475b97b0934604f23',1,'metal::fast::nextafter()'],['../namespacemetal_1_1precise.html#ad012ceeb55b77f1533749b351331e026',1,'metal::precise::nextafter()']]], - ['norm_14',['norm',['../namespacemlx_1_1core_1_1linalg.html#aba765b8e95e9a1d33d31f727a185919d',1,'mlx::core::linalg::norm(const array &a, const double ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#acaa85b4146821c268abecec2422c02d2',1,'mlx::core::linalg::norm(const array &a, const double ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#af1ebe0c6dcba9a1c49b5e397dddf3264',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#ae8da67e4c6e073f93889f1051203cd9e',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a229018071d5602e38d6248230f334a10',1,'mlx::core::linalg::norm(const array &a, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a44250cff34238f01471fd61e76036f03',1,'mlx::core::linalg::norm(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['normal_15',['normal',['../namespacemlx_1_1core_1_1random.html#a6ec2c08e63d379594b0744cfc7d8dc41',1,'mlx::core::random::normal(const Shape &shape, Dtype dtype, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a34c5efe846c68b62e65cdb26803d609c',1,'mlx::core::random::normal(const Shape &shape, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a980b9805dac939e80c2915f0046e6dee',1,'mlx::core::random::normal(const Shape &shape, const Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a1df350be44a32e2eefdd60bb21811246',1,'mlx::core::random::normal(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['normalize_5faxis_5findex_16',['normalize_axis_index',['../namespacemlx_1_1core.html#a1cc130b06d9cdd03dddc74a3b1db0167',1,'mlx::core']]], - ['not_5fequal_17',['not_equal',['../group__ops.html#ga28f22c5d2c399eee53be7b3facc11103',1,'mlx::core']]], - ['notequal_18',['NotEqual',['../classmlx_1_1core_1_1_not_equal.html#ac568397bd17b5d9f25ad1a0ebadedbb9',1,'mlx::core::NotEqual']]], - ['notify_5fnew_5ftask_19',['notify_new_task',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ae8aa34a9be8bc73508dd500000421173',1,'mlx::core::scheduler::Scheduler::notify_new_task()'],['../namespacemlx_1_1core_1_1scheduler.html#a6b7289e33cef665178fe614aac75c1b2',1,'mlx::core::scheduler::notify_new_task()']]], - ['notify_5ftask_5fcompletion_20',['notify_task_completion',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#abbb2b1c2f8bae2b9c7cc51db65f18a3b',1,'mlx::core::scheduler::Scheduler::notify_task_completion()'],['../namespacemlx_1_1core_1_1scheduler.html#a1d06ffdbab36790b78deb6e34adc737f',1,'mlx::core::scheduler::notify_task_completion()']]], - ['num_5fthreads_21',['num_threads',['../namespacepocketfft_1_1detail_1_1threading.html#af5432c2e25aed679a73fe7b29534c833',1,'pocketfft::detail::threading']]], - ['number_5fof_5felements_22',['number_of_elements',['../group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636',1,'mlx::core']]], - ['numberofelements_23',['NumberOfElements',['../classmlx_1_1core_1_1_number_of_elements.html#ac64d7c40ae29d687f8b7d2fa33e13b06',1,'mlx::core::NumberOfElements']]] + ['ndarr_3',['ndarr',['../classpocketfft_1_1detail_1_1ndarr.html#a8f0037a172d96cb1ad915a5069175fa2',1,'pocketfft::detail::ndarr']]], + ['ndim_4',['ndim',['../classpocketfft_1_1detail_1_1arr__info.html#ac608c8af2a59a28a0012e308be7ee414',1,'pocketfft::detail::arr_info::ndim()'],['../classmlx_1_1core_1_1array.html#a53006e77d13d9d88b525ef577748939f',1,'mlx::core::array::ndim()']]], + ['negative_5',['Negative',['../classmlx_1_1core_1_1_negative.html#aa3b73395d9fa5b7215dca488bc0d3c70',1,'mlx::core::Negative']]], + ['negative_6',['negative',['../group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6',1,'mlx::core']]], + ['new_5fqueue_7',['new_queue',['../classmlx_1_1core_1_1metal_1_1_device.html#a8135ae2a8c1e6f3861e84d4e60c28b67',1,'mlx::core::metal::Device']]], + ['new_5fscoped_5fmemory_5fpool_8',['new_scoped_memory_pool',['../namespacemlx_1_1core_1_1metal.html#a46583a1aba89449fa72e6cb3a7090981',1,'mlx::core::metal']]], + ['new_5fstream_9',['new_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a157c8da85fa1bddb8eacf8515a3cc879',1,'mlx::core::scheduler::Scheduler::new_stream()'],['../namespacemlx_1_1core_1_1metal.html#a8b4188f9a090a1da42d62b8a369bf106',1,'mlx::core::metal::new_stream()'],['../namespacemlx_1_1core.html#a6f7c63a9be10337b3b96d527e1db3c2f',1,'mlx::core::new_stream()']]], + ['next_10',['next',['../struct_quantized_block_loader.html#a674138ef7c43cc45586ea9f8fd6f6bd9',1,'QuantizedBlockLoader::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../structmlx_1_1steel_1_1_block_loader_t.html#a6008ef45ff980dbe1119da0630f6c697',1,'mlx::steel::BlockLoaderT::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a3b71f379ff9baf39830c92f4f1ecde52',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a78d2b0098311a278be8394edbd5fc731',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aae56c19bb562219770fec38e5666c6ce',1,'mlx::steel::Conv2DWeightBlockLoader::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af9ce1a767266664bea131a5437002c80',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a30b10bebde7f08b89d03bdd9ea0f48da',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3e5ee68ed0ee43f7e979dd4222f76a8c',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a11743cb1c108f42ccdc6e59204a5b3e8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531',1,'LoopedElemToLoc::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394',1,'LoopedElemToLoc::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#adc8113de3152d291331f219bfe5970c2',1,'LoopedElemToLoc< 1, OffsetT, true >::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a70d92070097fe0622b836afc778bf8d2',1,'LoopedElemToLoc< 1, OffsetT, true >::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a9d85f32e5460707827df3ba74b905c23',1,'LoopedElemToLoc< 1, OffsetT, false >::next(const constant int *, const constant int64_t *strides)'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a634f4fa8f4b5f1fdd47a26754ef2eba7',1,'LoopedElemToLoc< 1, OffsetT, false >::next(int n, const constant int *, const constant int64_t *strides)'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a4193c5eac3ef093a740d5305b25d3e18',1,'mlx::core::random::KeySequence::next()'],['../struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531',1,'LoopedElemToLoc< 1, OffsetT, false >::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394',1,'LoopedElemToLoc< 1, OffsetT, false >::next(int n, const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531',1,'LoopedElemToLoc< 1, OffsetT, true >::next(const constant int *shape, const constant int64_t *strides)'],['../struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394',1,'LoopedElemToLoc< 1, OffsetT, true >::next(int n, const constant int *shape, const constant int64_t *strides)']]], + ['next_5fpower_5fof_5f2_11',['next_power_of_2',['../namespacemlx_1_1core.html#a685c0530e338aabc622325685846ce93',1,'mlx::core']]], + ['nextafter_12',['nextafter',['../namespacemetal.html#a9547fd7b09164931986f6db4813bd72d',1,'metal::nextafter()'],['../namespacemetal_1_1fast.html#a4583e8be04fc0bd475b97b0934604f23',1,'metal::fast::nextafter()'],['../namespacemetal_1_1precise.html#ad012ceeb55b77f1533749b351331e026',1,'metal::precise::nextafter()']]], + ['norm_13',['norm',['../namespacemlx_1_1core_1_1linalg.html#aba765b8e95e9a1d33d31f727a185919d',1,'mlx::core::linalg::norm(const array &a, const double ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#acaa85b4146821c268abecec2422c02d2',1,'mlx::core::linalg::norm(const array &a, const double ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#af1ebe0c6dcba9a1c49b5e397dddf3264',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#ae8da67e4c6e073f93889f1051203cd9e',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a229018071d5602e38d6248230f334a10',1,'mlx::core::linalg::norm(const array &a, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a44250cff34238f01471fd61e76036f03',1,'mlx::core::linalg::norm(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['normal_14',['normal',['../namespacemlx_1_1core_1_1random.html#a6ec2c08e63d379594b0744cfc7d8dc41',1,'mlx::core::random::normal(const Shape &shape, Dtype dtype, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a34c5efe846c68b62e65cdb26803d609c',1,'mlx::core::random::normal(const Shape &shape, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a980b9805dac939e80c2915f0046e6dee',1,'mlx::core::random::normal(const Shape &shape, const Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a1df350be44a32e2eefdd60bb21811246',1,'mlx::core::random::normal(const Shape &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['normalize_5faxis_5findex_15',['normalize_axis_index',['../namespacemlx_1_1core.html#a1cc130b06d9cdd03dddc74a3b1db0167',1,'mlx::core']]], + ['not_5fequal_16',['not_equal',['../group__ops.html#ga28f22c5d2c399eee53be7b3facc11103',1,'mlx::core']]], + ['notequal_17',['NotEqual',['../classmlx_1_1core_1_1_not_equal.html#ac568397bd17b5d9f25ad1a0ebadedbb9',1,'mlx::core::NotEqual']]], + ['notify_5fnew_5ftask_18',['notify_new_task',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ae8aa34a9be8bc73508dd500000421173',1,'mlx::core::scheduler::Scheduler::notify_new_task()'],['../namespacemlx_1_1core_1_1scheduler.html#a6b7289e33cef665178fe614aac75c1b2',1,'mlx::core::scheduler::notify_new_task()']]], + ['notify_5ftask_5fcompletion_19',['notify_task_completion',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#abbb2b1c2f8bae2b9c7cc51db65f18a3b',1,'mlx::core::scheduler::Scheduler::notify_task_completion()'],['../namespacemlx_1_1core_1_1scheduler.html#a1d06ffdbab36790b78deb6e34adc737f',1,'mlx::core::scheduler::notify_task_completion()']]], + ['num_5fthreads_20',['num_threads',['../namespacepocketfft_1_1detail_1_1threading.html#af5432c2e25aed679a73fe7b29534c833',1,'pocketfft::detail::threading']]], + ['number_5fof_5felements_21',['number_of_elements',['../group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636',1,'mlx::core']]], + ['numberofelements_22',['NumberOfElements',['../classmlx_1_1core_1_1_number_of_elements.html#ac64d7c40ae29d687f8b7d2fa33e13b06',1,'mlx::core::NumberOfElements']]] ]; diff --git a/docs/build/html/search/functions_f.js b/docs/build/html/search/functions_f.js index 4a6faa098..818539e8b 100644 --- a/docs/build/html/search/functions_f.js +++ b/docs/build/html/search/functions_f.js @@ -8,42 +8,45 @@ var searchData= ['operator_20bool_5',['operator bool',['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant'],['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant']]], ['operator_20dtype_6',['operator Dtype',['../structmlx_1_1core_1_1_type_to_dtype.html#aefdd0fd6a5bbf0197a3996ccd4adea13',1,'mlx::core::TypeToDtype']]], ['operator_20float_7',['operator float',['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aaae72e5340ce91325f1925be36ba46cb',1,'mlx::core::_MLX_BFloat16::operator float()'],['../structmlx_1_1core_1_1complex128__t.html#a3e2faf180c0b785646a0e4296f709a5e',1,'mlx::core::complex128_t::operator float()'],['../structmlx_1_1core_1_1complex64__t.html#a90d224dd37308345086bb9cc882ef6fc',1,'mlx::core::complex64_t::operator float()'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a363de5054f3673bddc90293fc3c9bb99',1,'mlx::core::_MLX_Float16::operator float()']]], - ['operator_20t_8',['operator T',['../structcomplex64__t.html#a70e9b16031eeaff3baa601f400023fcd',1,'complex64_t::operator T() const thread'],['../structcomplex64__t.html#a4f3beea7ab6001189b782a74d1746b67',1,'complex64_t::operator T() const threadgroup'],['../structcomplex64__t.html#a9f4f7eca89ffe6c8d126a4145df6d9f2',1,'complex64_t::operator T() const device'],['../structcomplex64__t.html#ac33e2e5263fec76a4fb4418c6e1d8d14',1,'complex64_t::operator T() const constant'],['../struct___m_l_x___b_float16.html#aa7dfefdf0d15e102d2b8258c9ab01836',1,'_MLX_BFloat16::operator T() const thread'],['../struct___m_l_x___b_float16.html#a2546a8afa77e14ed5b3c5da79a281260',1,'_MLX_BFloat16::operator T() const threadgroup'],['../struct___m_l_x___b_float16.html#a1d523f87740fcb852db6ab57896c245a',1,'_MLX_BFloat16::operator T() const device'],['../struct___m_l_x___b_float16.html#a95acd29283024d7093a0bc58c9468a0a',1,'_MLX_BFloat16::operator T() const constant']]], - ['operator_20val_9',['operator Val',['../structmlx_1_1core_1_1_dtype.html#a3b3bc059be5836476da3cb88a4f5e9fd',1,'mlx::core::Dtype']]], - ['operator_20value_5ftype_10',['operator value_type',['../structmlx_1_1steel_1_1integral__constant.html#a0c11203bed44a6a2c387b365134dcd64',1,'mlx::steel::integral_constant']]], - ['operator_21_3d_11',['operator!=',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afc6e4fc5589bbf30f978f34868dd4e55',1,'operator!=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6baa722c22d66c7510786bb275cb8cc2',1,'operator!=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa8d9f01582a0a9f01a666d110c74db2a',1,'operator!=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa504a474ab6e00ebe2b1b7ed2f7d1ffb',1,'operator!=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abf5f3040227f021a5b84cf2eda248b2f',1,'operator!=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a347c9bbf816bad2e9e5e91aa448f8b65',1,'operator!=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a33ea086b561c652f25833a5e1ded34dd',1,'operator!=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2bbdcece13148826d3fe33af727bb79b',1,'operator!=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeb1efa47c5f22cc0b35d49ccce73c406',1,'operator!=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa6b99cde403405df1865c989e4ce845a',1,'operator!=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a204d13a881ae8d337f6efbb98673790c',1,'operator!=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3602117b4c61d5cd4fd72fb8e5f68bd6',1,'operator!=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2721c088adfc9d73cde442d6badd2a6c',1,'operator!=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa4364eda56525cf7576ff00e550175e6',1,'mlx::steel::operator!=()'],['../namespacemlx_1_1core.html#a94d00a1b7f8a4717ab3f26f45e4da655',1,'mlx::core::operator!=(const Device &lhs, const Device &rhs)'],['../group__ops.html#ga0ac483d85f23252ca8757e9926d5a3c5',1,'mlx::core::operator!=(const array &a, const array &b)'],['../group__ops.html#ga3fecba9f3cb9a19afd8ca492cf509ce0',1,'mlx::core::operator!=(T a, const array &b)'],['../group__ops.html#gaebbf1cfde388c7480159a03c92c9a385',1,'mlx::core::operator!=(const array &a, T b)'],['../namespacemlx_1_1core.html#a164f109bc19c927b2b3bcc47a5021419',1,'mlx::core::operator!=(const Stream &lhs, const Stream &rhs)'],['../namespacemlx_1_1core.html#ad2f9e1c230ec35d5c406dd616e8f4dea',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af5899b4d5644682cb0ac2a488f630d55',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a72ac8edd190601d7a46782582cedecd8',1,'mlx::core::operator!=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a8084162ba2dd3f9b89195d2bebc3fbb0',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a514263e63f6825b490203ca586864687',1,'mlx::core::operator!=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a1c482bb3d9f9d4c62dee5865892c1f96',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a0030fe7ad09837c670cdfb7d51279519',1,'mlx::core::operator!=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ade3791bc723b8f10fbab22eadb0f705a',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ad78c664f242cd36247c13868547e3dd4',1,'mlx::core::operator!=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab0743a1a1dcb92d40f41ca42d36f242c',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ae7a0f810e546a166c7d05849b5d41f30',1,'mlx::core::operator!=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a676a40637a563f013c725d24fa33fdc8',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a9fcb662b1561e4136bac0106cfb63b6c',1,'mlx::core::operator!=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abcca7fd43590c4347e0f5df8f134030c',1,'mlx::core::operator!=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af3ede3688a2e3b3ba8cb2da180ffe151',1,'mlx::core::operator!=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a54f48469fabd1414bef5097bcded0002',1,'mlx::core::operator!=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af8c648e892cbc6973de535aa17dc2cfe',1,'mlx::core::operator!=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#abc855e1c0584b64d7d995e33211361ab',1,'mlx::core::operator!=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad3684d660d18a54505c759ab286bd936',1,'mlx::core::operator!=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a8afdda14b14262ab5ce0a00c7745d7e8',1,'mlx::core::operator!=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7ccc479be236f2bf3f7725729c5ba201',1,'mlx::core::operator!=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a26a721b8111fce3a1dec9bf724034cd4',1,'mlx::core::operator!=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad5f8c221a53a89e8095aa39fd1f61867',1,'mlx::core::operator!=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a017b52ecf30b33da4aa8da35ccc43220',1,'mlx::core::operator!=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a43c10ca5fb05ee7d0ee63ba56f8a08a3',1,'mlx::core::operator!=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a81284b6ac737f91a8d1ffbbbbf938fe5',1,'mlx::core::operator!=(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_25_12',['operator%',['../backend_2metal_2kernels_2complex_8h.html#aaf53122a07c8eca858b5a8e38ae280e0',1,'operator%(): complex.h'],['../group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde',1,'mlx::core::operator%(const array &a, const array &b)'],['../group__ops.html#ga50817666f0b82afcbf4a123486af9908',1,'mlx::core::operator%(T a, const array &b)'],['../group__ops.html#ga46c01daa07433542a477d216e13a8480',1,'mlx::core::operator%(const array &a, T b)'],['../namespacemlx_1_1core.html#a8723d145dd49021bfcb8e6c99e1c91a5',1,'mlx::core::operator%(complex64_t a, complex64_t b)']]], - ['operator_26_13',['operator&',['../group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff',1,'mlx::core::operator&(const array &a, const array &b)'],['../namespacemlx_1_1core.html#a9ee95f97bbd69262d99d7bea3bf77631',1,'mlx::core::operator&(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0fefc3ae4f1350ebe05ec6098fd6bae3',1,'mlx::core::operator&(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a1e4cb758ccfe5c267baed9aeb0044834',1,'mlx::core::operator&(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab9d0f9910070231695d61de08cadb930',1,'mlx::core::operator&(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a889d401f425db79d1868aa3beea4829b',1,'mlx::core::operator&(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a76dcd1fa3c68b386bc1d1d899a68a120',1,'mlx::core::operator&(uint16_t lhs, _MLX_Float16 rhs)']]], - ['operator_26_26_14',['operator&&',['../namespacemlx_1_1steel.html#a6353bf11881842e25c46b56f92b7044f',1,'mlx::steel::operator&&()'],['../group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3',1,'mlx::core::operator&&(const array &a, const array &b)']]], - ['operator_26_3d_15',['operator&=',['../namespacemlx_1_1core.html#a60c263ef46e552c3954688869734b513',1,'mlx::core::operator&=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af9670fc8088339669c54c68b3a320e25',1,'mlx::core::operator&=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ad1f96f0a02024f347b4c4431629407fc',1,'mlx::core::operator&=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae0540f16c4e7bd55d0e86a88495e4967',1,'mlx::core::operator&=(_MLX_Float16 &lhs, uint16_t rhs)']]], - ['operator_28_29_16',['operator()',['../structpocketfft_1_1detail_1_1_exec_c2_c.html#a4fd637f1a6d335826789af28ac089ecb',1,'pocketfft::detail::ExecC2C::operator()()'],['../structpocketfft_1_1detail_1_1_exec_hartley.html#a67c98b38d12440781053552b9a33bba1',1,'pocketfft::detail::ExecHartley::operator()()'],['../structpocketfft_1_1detail_1_1_exec_dcst.html#a67f4f56e3574c491695f8cb8a1e983d8',1,'pocketfft::detail::ExecDcst::operator()()'],['../structpocketfft_1_1detail_1_1_exec_r2_r.html#acdba1650962714e6afff51e9ca456970',1,'pocketfft::detail::ExecR2R::operator()()'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a0d657bc9a381dca1b5860b9a1b5a5702',1,'mlx::core::detail::Abs::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a564232db7d32811e2ae126c86de104f0',1,'mlx::core::detail::Abs::operator()(uint8_t x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a5fac7e6c8277d8706535a52820503c9d',1,'mlx::core::detail::Abs::operator()(uint16_t x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#af2c3723e648bd5ed2fe558cc20b7f5eb',1,'mlx::core::detail::Abs::operator()(uint32_t x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a57312cd8530dd0ede3b8037f9c401883',1,'mlx::core::detail::Abs::operator()(uint64_t x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#ab3b5e3853ed56bfbfa577d965c21112e',1,'mlx::core::detail::Abs::operator()(bool x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html#a04b4c9d1fc0160973aa28b1f809b9d51',1,'mlx::core::detail::ArcCos::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html#a767d354bec863942822ee0b9b6742a88',1,'mlx::core::detail::ArcCosh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html#ac69091929815e5317308b4088f5c2f46',1,'mlx::core::detail::ArcSin::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html#ac7bf9bac66fef917f75494b2345e6aaf',1,'mlx::core::detail::ArcSinh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html#aee87bf10c278a70ca788085d1b499afe',1,'mlx::core::detail::ArcTan::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html#a9040b7afcdb4969924aa782fa67f03ac',1,'mlx::core::detail::ArcTan2::operator()()'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html#a601e8c52bb938eb3a616756a35419e8b',1,'mlx::core::detail::ArcTanh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a672f65e47d65e4e8d88be252bce0164b',1,'mlx::core::detail::Ceil::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a72a2cab2728fb5e1cc6329a539e5d573',1,'mlx::core::detail::Ceil::operator()(int8_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#aa34590f6a41331be92988558a90dc6fa',1,'mlx::core::detail::Ceil::operator()(int16_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#af14120f3dd98f6198ea257d75be223f7',1,'mlx::core::detail::Ceil::operator()(int32_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#af263ce7743cf7319387baba616c375b5',1,'mlx::core::detail::Ceil::operator()(int64_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a48f00affcd5c2ea1f81d821e019fec29',1,'mlx::core::detail::Ceil::operator()(uint8_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#ad4d24a44e8a328948393701dacb0ceac',1,'mlx::core::detail::Ceil::operator()(uint16_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a70da19b5c9c69f04b9f196bdf266f93c',1,'mlx::core::detail::Ceil::operator()(uint32_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#af0e7e806b73c664ada837476f9d4d43b',1,'mlx::core::detail::Ceil::operator()(uint64_t x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#acc1bfc84a9b91f6e9764234cbe3b9687',1,'mlx::core::detail::Ceil::operator()(bool x)'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html#a7e662d05c6998bd6ced8ad9c187324a5',1,'mlx::core::detail::Conjugate::operator()()'],['../structmlx_1_1core_1_1detail_1_1_cos.html#ad4caef573f9d9071f8945a8efed231ad',1,'mlx::core::detail::Cos::operator()()'],['../structmlx_1_1core_1_1detail_1_1_cosh.html#a63591f49776d9aadc02200036ae38317',1,'mlx::core::detail::Cosh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_erf.html#a168f8ccc6c8053b05dd1a48904ca8fd4',1,'mlx::core::detail::Erf::operator()()'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html#acc93c0511141404208b35f302f8c1fcb',1,'mlx::core::detail::ErfInv::operator()()'],['../structmlx_1_1core_1_1detail_1_1_exp.html#a0846300cee28315e5b42f74acafbd1a1',1,'mlx::core::detail::Exp::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_exp.html#af247c0d19d59f3310fd0a081eb92cf8b',1,'mlx::core::detail::Exp::operator()(complex64_t x)'],['../structmlx_1_1core_1_1detail_1_1_expm1.html#abf7e61b8387521e9d44334ce88d833a0',1,'mlx::core::detail::Expm1::operator()()'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a16c13cfe736098bffc81d655e172294a',1,'mlx::core::detail::Floor::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a9b6c4c34b6594b8c413abe31f34a73df',1,'mlx::core::detail::Floor::operator()(int8_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#aca4c71204b3ceeca6329f7ea2b041f4c',1,'mlx::core::detail::Floor::operator()(int16_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a3c3ab9e00d1fbd124802517e8c35fe02',1,'mlx::core::detail::Floor::operator()(int32_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a4b5954ffc59c741dd7b86bafda69d5cc',1,'mlx::core::detail::Floor::operator()(int64_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a2e33b10bd5b04551054a87c601440bc7',1,'mlx::core::detail::Floor::operator()(uint8_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a2500f971100919a694f78669a5e4f438',1,'mlx::core::detail::Floor::operator()(uint16_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a23df818301d68389e6e12f5a9ec1fbd7',1,'mlx::core::detail::Floor::operator()(uint32_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#ac988b4f265cf46c68609c9c8787c15fb',1,'mlx::core::detail::Floor::operator()(uint64_t x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a7f936e3fd53772bc189d845c73b53202',1,'mlx::core::detail::Floor::operator()(bool x)'],['../structmlx_1_1core_1_1detail_1_1_imag.html#a5bd82e2185f3779e398c179d42a3e782',1,'mlx::core::detail::Imag::operator()()'],['../structmlx_1_1core_1_1detail_1_1_log.html#a0012a4e1744dbe9a28c3b5652be6e1c6',1,'mlx::core::detail::Log::operator()()'],['../structmlx_1_1core_1_1detail_1_1_log2.html#a467bd4c995674721ff5fff6df33aead8',1,'mlx::core::detail::Log2::operator()()'],['../structmlx_1_1core_1_1detail_1_1_log10.html#a2633c5b772bbc9f8b66cffd4a3e01a3f',1,'mlx::core::detail::Log10::operator()()'],['../structmlx_1_1core_1_1detail_1_1_log1p.html#a3220de8c6090c44aa2070b1fbb2dc340',1,'mlx::core::detail::Log1p::operator()()'],['../structmlx_1_1core_1_1detail_1_1_logical_not.html#a79799668ea5c364b0b4e2bc330e76253',1,'mlx::core::detail::LogicalNot::operator()()'],['../structmlx_1_1core_1_1detail_1_1_negative.html#afc4595c70ef7196df374cf4b2cc5e526',1,'mlx::core::detail::Negative::operator()()'],['../structmlx_1_1core_1_1detail_1_1_real.html#ae84a939fdb5916257a7731cda66d4d61',1,'mlx::core::detail::Real::operator()()'],['../structmlx_1_1core_1_1detail_1_1_round.html#a653f29c059bbfa6192378732a8a23351',1,'mlx::core::detail::Round::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_round.html#a82a984f13568051009e257fe85227da6',1,'mlx::core::detail::Round::operator()(complex64_t x)'],['../structmlx_1_1core_1_1detail_1_1_sigmoid.html#a64b72561bfaf758632167f00648f4c89',1,'mlx::core::detail::Sigmoid::operator()()'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a64ed5013cee7ff18c7fe70bc04737e7b',1,'mlx::core::detail::Sign::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a7106ed1f2f98a365fcb3e6ee39084748',1,'mlx::core::detail::Sign::operator()(uint8_t x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a7163e8c068dcc460600ed04014dc9945',1,'mlx::core::detail::Sign::operator()(uint16_t x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#ae8f56c7134721c846240830169424c22',1,'mlx::core::detail::Sign::operator()(uint32_t x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a10ae519e9a74a327fc72c410e9ab2936',1,'mlx::core::detail::Sign::operator()(uint64_t x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a91be4e273f6c7ea5d44cfab380b77603',1,'mlx::core::detail::Sign::operator()(complex64_t x)'],['../structmlx_1_1core_1_1detail_1_1_sin.html#ae95671816529cc2188389af37a2f1a13',1,'mlx::core::detail::Sin::operator()()'],['../structmlx_1_1core_1_1detail_1_1_sinh.html#a9663ddf0fa4c0003576b48f3d5385f00',1,'mlx::core::detail::Sinh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_square.html#a54e9e3c0d0896e142289e8282eab1099',1,'mlx::core::detail::Square::operator()()'],['../structmlx_1_1core_1_1detail_1_1_sqrt.html#aa5a4830b3ef7efab20ea88a110667efd',1,'mlx::core::detail::Sqrt::operator()()'],['../structmlx_1_1core_1_1detail_1_1_rsqrt.html#a9af247be16bab83243038aac54446b79',1,'mlx::core::detail::Rsqrt::operator()()'],['../structmlx_1_1core_1_1detail_1_1_tan.html#aba397cd7ac05bbe06dfa9e3a64bdb05f',1,'mlx::core::detail::Tan::operator()()'],['../structmlx_1_1core_1_1detail_1_1_tanh.html#a1749ba1edfd53095ed7d45c0e53bab61',1,'mlx::core::detail::Tanh::operator()()'],['../structmlx_1_1core_1_1detail_1_1_add.html#a2d6011c35768b5fcd2bb75747b944353',1,'mlx::core::detail::Add::operator()()'],['../structmlx_1_1core_1_1detail_1_1_divide.html#a5e0d22e2084c4ca81bec0d457a46c662',1,'mlx::core::detail::Divide::operator()()'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a3bdaf1095ad883ecc0fecc455f02cbf3',1,'mlx::core::detail::Remainder::operator()(T numerator, T denominator)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a52c3a2ba86fccb24d37d218ae8328954',1,'mlx::core::detail::Remainder::operator()(T numerator, T denominator)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a0b0dd6ef5b08585fdf8355770da8d747',1,'mlx::core::detail::Remainder::operator()(T numerator, T denominator)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a68fe542084fb94d9a5abd740fe07832b',1,'mlx::core::detail::Remainder::operator()(complex64_t numerator, complex64_t denominator)'],['../structmlx_1_1core_1_1detail_1_1_equal.html#a2994cf1884e7126e76d0a20b215fe3ab',1,'mlx::core::detail::Equal::operator()()'],['../structmlx_1_1core_1_1detail_1_1_na_n_equal.html#a073b20b0d8d41ec8364b7c477421b9bf',1,'mlx::core::detail::NaNEqual::operator()()'],['../structmlx_1_1core_1_1detail_1_1_greater.html#aa3844c2bae3c7a981739f642aa0dd094',1,'mlx::core::detail::Greater::operator()()'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html#a3b005f85522ad0e4b57044eed930ac30',1,'mlx::core::detail::GreaterEqual::operator()()'],['../structmlx_1_1core_1_1detail_1_1_less.html#a0b4032dff1ad2b387745cb000aabdcbb',1,'mlx::core::detail::Less::operator()()'],['../structmlx_1_1core_1_1detail_1_1_less_equal.html#a31e70f8830a07557697541301555a7a7',1,'mlx::core::detail::LessEqual::operator()()'],['../structmlx_1_1core_1_1detail_1_1_maximum.html#a3eb37abec8426ebc42b8c685075c523a',1,'mlx::core::detail::Maximum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_maximum.html#af99345c7c8bc95ccab1b22c0792ac6fd',1,'mlx::core::detail::Maximum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_minimum.html#afca0861556416a8547dd8574528feb69',1,'mlx::core::detail::Minimum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_minimum.html#a64b2eecfbc56aaef7deb939423bac3f8',1,'mlx::core::detail::Minimum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_log_add_exp.html#ad1663fd809acaa4038f90666436599e5',1,'mlx::core::detail::LogAddExp::operator()()'],['../structmlx_1_1core_1_1detail_1_1_multiply.html#a898b090966b047723513224b8d3b22f1',1,'mlx::core::detail::Multiply::operator()()'],['../structmlx_1_1core_1_1detail_1_1_not_equal.html#a23d662b5fd968dc17d3bee2595b5f99d',1,'mlx::core::detail::NotEqual::operator()()'],['../structmlx_1_1core_1_1detail_1_1_power.html#a2c047e1b488e6525447a224975a75db8',1,'mlx::core::detail::Power::operator()(T base, T exp)'],['../structmlx_1_1core_1_1detail_1_1_power.html#a9967db24b8f67d54b6aa3810e274f28c',1,'mlx::core::detail::Power::operator()(T base, T exp)'],['../structmlx_1_1core_1_1detail_1_1_subtract.html#a72ef05830615a2d5d9662926ed82672a',1,'mlx::core::detail::Subtract::operator()()'],['../structmlx_1_1core_1_1detail_1_1_logical_and.html#a046536c1f2f9367983f052a213d7b7d8',1,'mlx::core::detail::LogicalAnd::operator()()'],['../structmlx_1_1core_1_1detail_1_1_logical_or.html#afb134dbab79307d4ba597843c61d0b1a',1,'mlx::core::detail::LogicalOr::operator()()'],['../structmlx_1_1core_1_1detail_1_1_select.html#a930f9da2e6b3453e04f21382435a2cfb',1,'mlx::core::detail::Select::operator()()'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html#ae0bed77f95fe2b2f0b594addddd04700',1,'mlx::core::detail::BitwiseAnd::operator()()'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html#a5ab05734c5000b454975de6647a08d20',1,'mlx::core::detail::BitwiseOr::operator()()'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html#a0989e3bcd064ae06c33f660696a869a0',1,'mlx::core::detail::BitwiseXor::operator()()'],['../structmlx_1_1core_1_1detail_1_1_left_shift.html#a9385f580830a6ad163dd9bb8c4905e7a',1,'mlx::core::detail::LeftShift::operator()()'],['../structmlx_1_1core_1_1detail_1_1_right_shift.html#a154528ba50e89a4c532a181f135b1620',1,'mlx::core::detail::RightShift::operator()()'],['../structmlx_1_1core_1_1_default_strided_reduce.html#a024682ab93b84e544a07e3a9c3c51fba',1,'mlx::core::DefaultStridedReduce::operator()()'],['../structmlx_1_1core_1_1_default_contiguous_reduce.html#a08144c7a3cdf10af5e47f4575da3694f',1,'mlx::core::DefaultContiguousReduce::operator()()'],['../struct_add.html#ac5c66b63d63a222d3ae0ab8cc7c90eb5',1,'Add::operator()()'],['../struct_floor_divide.html#a2b328e4d768e718fa439f955c524666a',1,'FloorDivide::operator()(T x, T y)'],['../struct_floor_divide.html#afc16a2b2a745225e0bc95640f3fc0219',1,'FloorDivide::operator()(float x, float y)'],['../struct_floor_divide.html#ae91719a15f7e643d552129f476089c6a',1,'FloorDivide::operator()(half x, half y)'],['../struct_floor_divide.html#a4aa9f858626583e02bd79f747229bbca',1,'FloorDivide::operator()(bfloat16_t x, bfloat16_t y)'],['../struct_divide.html#a0a16b9194abc2ab7c61129f81a9bbb3d',1,'Divide::operator()()'],['../struct_remainder.html#ab7875512ff4341c580c6dc372e64fc58',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#a18150b5f4425e30b95ffabc6bb25cede',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#ab3b75f54b56fd357c9755daadb2cafc2',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#ae918ce0e246937d4fe04e2ea36e4b2c1',1,'Remainder::operator()(complex64_t x, complex64_t y)'],['../struct_equal.html#aa498087080900d4428ba428a6496a769',1,'Equal::operator()()'],['../struct_na_n_equal.html#a00220898e02db656d21dde9e9354a8dc',1,'NaNEqual::operator()(T x, T y)'],['../struct_na_n_equal.html#a6185e4554dce5b4659d21673c576be51',1,'NaNEqual::operator()(complex64_t x, complex64_t y)'],['../struct_greater.html#a98d7d8ee360cd0f469c6eb9a017560f5',1,'Greater::operator()()'],['../struct_greater_equal.html#ae69a3bccc567a46506cf0d296294ce80',1,'GreaterEqual::operator()()'],['../struct_less.html#a5ee0b31b2d9123dc4504f2979a5854d3',1,'Less::operator()()'],['../struct_less_equal.html#ae9f9a1b2eae548977139704f0044acfe',1,'LessEqual::operator()()'],['../struct_log_add_exp.html#ab32417f18e8ff68c15f78aceeb624edf',1,'LogAddExp::operator()()'],['../struct_maximum.html#a3ea0f42bc4cd80b68a98f189f9fa859c',1,'Maximum::operator()(T x, T y)'],['../struct_maximum.html#a0bc8fadc87f2c49fc440d625bfc97ca6',1,'Maximum::operator()(T x, T y)'],['../struct_maximum.html#a907e8793900be5927625377dab199644',1,'Maximum::operator()(complex64_t x, complex64_t y)'],['../struct_minimum.html#aa6113dfac3986c0f571fa53f65c5330e',1,'Minimum::operator()(T x, T y)'],['../struct_minimum.html#a0c939921de87ab9c6959238aac81a059',1,'Minimum::operator()(T x, T y)'],['../struct_minimum.html#a800fba087280f79c2f7e9aff75bed093',1,'Minimum::operator()(complex64_t x, complex64_t y)'],['../struct_multiply.html#a1327fc5a0713931afe997b0d4d2988e0',1,'Multiply::operator()()'],['../struct_not_equal.html#af008d73a5d9cde0b8309b7e8ee7438b2',1,'NotEqual::operator()(T x, T y)'],['../struct_not_equal.html#a14de494cea4e4869351202cad1149f17',1,'NotEqual::operator()(complex64_t x, complex64_t y)'],['../struct_power.html#a2b6df2a9e48155ff9734caca8504a79f',1,'Power::operator()(T base, T exp)'],['../struct_power.html#a36829163d42973034a1f8a7ecc57a1de',1,'Power::operator()(T base, T exp)'],['../struct_power.html#a27cdfb313c4e82b63bdcdaee923cbbef',1,'Power::operator()(complex64_t x, complex64_t y)'],['../struct_subtract.html#ae0856cd8d449074ca287baa7e460f68a',1,'Subtract::operator()()'],['../struct_logical_and.html#a8bc6bdabc0ea0678a46e2cf6217cb3a6',1,'LogicalAnd::operator()()'],['../struct_logical_or.html#ade6a931324a604a3119d2220d6f5460d',1,'LogicalOr::operator()()'],['../struct_bitwise_and.html#afb48af090b01dd0200963bc12d842e36',1,'BitwiseAnd::operator()()'],['../struct_bitwise_or.html#a41f847463daafa99ee56f4035578390f',1,'BitwiseOr::operator()()'],['../struct_bitwise_xor.html#a3a3e8a56caab739d40262d9349c9c485',1,'BitwiseXor::operator()()'],['../struct_left_shift.html#aa729747784c38bfdbba34794fcf5175b',1,'LeftShift::operator()()'],['../struct_right_shift.html#a2cc59b400c68342b0e43050431323c17',1,'RightShift::operator()()'],['../struct_arc_tan2.html#ac9b7729753e13be293ab700231d061ac',1,'ArcTan2::operator()()'],['../struct_div_mod.html#a8b5758f2ea18d4c903b462331b25abfe',1,'DivMod::operator()()'],['../struct_cum_prod.html#ad634be0b139d10ce6d21332eef0d936b',1,'CumProd< bool >::operator()()'],['../struct_cum_max.html#a781b9b955c5412466da6af6c70d73c06',1,'CumMax::operator()()'],['../struct_cum_min.html#ae0b8c3761e04fa538d304ca842281a66',1,'CumMin::operator()()'],['../struct_less_than.html#a2798eb377b411c93a4ed30cf35caade2',1,'LessThan::operator()()'],['../struct_select.html#adb51692aae3038de07dd745891bf9848',1,'Select::operator()()'],['../struct_abs.html#a9e7481dfcc162509769852026ff4a344',1,'Abs::operator()(T x)'],['../struct_abs.html#a0ca113fd036151c443df3f83cc667f28',1,'Abs::operator()(uint8_t x)'],['../struct_abs.html#adaeab32a7e377dc990077ab15f3dc4c2',1,'Abs::operator()(uint16_t x)'],['../struct_abs.html#a99d2a2f37a6cddd3168b0224f2a9b963',1,'Abs::operator()(uint32_t x)'],['../struct_abs.html#ac9cbc02422d930479303f240a7ea6c71',1,'Abs::operator()(uint64_t x)'],['../struct_abs.html#ac30835b27784d451bd2e4524c8eb9e11',1,'Abs::operator()(bool x)'],['../struct_abs.html#ab82917d6b30a2c579e7eb879d305c5fc',1,'Abs::operator()(complex64_t x)'],['../struct_arc_cos.html#a5553cecf58511e24e76ac97f2d90b9ac',1,'ArcCos::operator()()'],['../struct_arc_cosh.html#a5c9e7712c14c97298b23ec48e19abc58',1,'ArcCosh::operator()()'],['../struct_arc_sin.html#a0343872f2da93bae2bb0baadf49da022',1,'ArcSin::operator()()'],['../struct_arc_sinh.html#a3066fb7dc7c3180100fb55ff94af6a7a',1,'ArcSinh::operator()()'],['../struct_arc_tan.html#af3a0aec6acec8ae8f5e4c4d5cf8c91ba',1,'ArcTan::operator()()'],['../struct_arc_tanh.html#a37dc3e01ec2830de7e82ed6c6363ac88',1,'ArcTanh::operator()()'],['../struct_ceil.html#a5e2a4ef1b012f5d352064489156e5e44',1,'Ceil::operator()(T x)'],['../struct_ceil.html#a455cd8083ba859993077f2e078ae165b',1,'Ceil::operator()(int8_t x)'],['../struct_ceil.html#a2acb61bc658c7a216795e7f76ebcf98a',1,'Ceil::operator()(int16_t x)'],['../struct_ceil.html#aef8c37f7a8ee3fc80700d605a09891fb',1,'Ceil::operator()(int32_t x)'],['../struct_ceil.html#a93d0110511ad5dd200e12d37a3d7d6e3',1,'Ceil::operator()(int64_t x)'],['../struct_ceil.html#aa335b745fa26e0f443cdb36298105484',1,'Ceil::operator()(uint8_t x)'],['../struct_ceil.html#ade17e13b7f30f5c590fae1581a2013ac',1,'Ceil::operator()(uint16_t x)'],['../struct_ceil.html#a411c75cc35cdc088402e176a1defd22d',1,'Ceil::operator()(uint32_t x)'],['../struct_ceil.html#a9ac660ca29eef7a7429fceb7b917a68a',1,'Ceil::operator()(uint64_t x)'],['../struct_ceil.html#a40de367e62f06ebd7e1330afa93a9ad9',1,'Ceil::operator()(bool x)'],['../struct_cos.html#ae222f8710f6b8254c471ebd475aa5bda',1,'Cos::operator()(T x)'],['../struct_cos.html#a5f26feb1dcc4bec5f59a9ff511c5b163',1,'Cos::operator()(complex64_t x)'],['../struct_cosh.html#a5847ebeebb236fdc926798ddc16475ba',1,'Cosh::operator()(T x)'],['../struct_cosh.html#aefdd91298dac16d528d29ee47e2f7252',1,'Cosh::operator()(complex64_t x)'],['../struct_conjugate.html#acb0a2694285f1f57c7654b371ce8cbd8',1,'Conjugate::operator()()'],['../struct_erf.html#a80719402ad7f7d418859a6677d7b604d',1,'Erf::operator()()'],['../struct_erf_inv.html#afbf3668d1a512e889f093a0bc7673309',1,'ErfInv::operator()()'],['../struct_exp.html#a5ef395868e055348c0802fd5fe45669c',1,'Exp::operator()(T x)'],['../struct_exp.html#a2b341ac400c4d145397950eb60734336',1,'Exp::operator()(complex64_t x)'],['../struct_expm1.html#a4b834d42cf0b84daf03fec62c222091a',1,'Expm1::operator()()'],['../struct_floor.html#ace3551f28429081e9f3a3dab0c84212b',1,'Floor::operator()(T x)'],['../struct_floor.html#a10d7fd05b4c224c9f135451246d13014',1,'Floor::operator()(int8_t x)'],['../struct_floor.html#a2865a04a492e3590302f4bd3215a10d7',1,'Floor::operator()(int16_t x)'],['../struct_floor.html#a41012343ff0463ec44b4d06196f41182',1,'Floor::operator()(int32_t x)'],['../struct_floor.html#aae3181d15856796aa0628cf30c92aa2e',1,'Floor::operator()(int64_t x)'],['../struct_floor.html#ac6cf38d82c8e270911afdca4c69ad51b',1,'Floor::operator()(uint8_t x)'],['../struct_floor.html#a78969b9e2b53ae248e72a67259eea5d8',1,'Floor::operator()(uint16_t x)'],['../struct_floor.html#a959009320ed622ed45b39becab1d5b98',1,'Floor::operator()(uint32_t x)'],['../struct_floor.html#a7d04b83c3345cd867315cae2d7ff68ab',1,'Floor::operator()(uint64_t x)'],['../struct_floor.html#abea845fe5e8e6b93bd4bca8717337e0b',1,'Floor::operator()(bool x)'],['../struct_imag.html#a3b29e9f8a46c194d683f6a9938314400',1,'Imag::operator()()'],['../struct_log.html#a32a383cb6be06e616a75f23bf49089c3',1,'Log::operator()()'],['../struct_log2.html#ac1e067ecdcbdbffb6106e789c2b98b64',1,'Log2::operator()()'],['../struct_log10.html#ac596a74c1642a00f3eced07ee3334122',1,'Log10::operator()()'],['../struct_log1p.html#a4464c6e7bdbe55ffd7d961c695cd13ce',1,'Log1p::operator()()'],['../struct_logical_not.html#a8a620bac957ab8c09ac85adfddd96708',1,'LogicalNot::operator()()'],['../struct_negative.html#af6879b374314a559faa321e8cce3d710',1,'Negative::operator()()'],['../struct_real.html#a85b9c5b9e65297994fa26ff68e19e809',1,'Real::operator()()'],['../struct_round.html#aa06a0195867e2ceb679c403b6909a1c4',1,'Round::operator()(T x)'],['../struct_round.html#ad3a08f2276ff1033900bc0a7da812655',1,'Round::operator()(complex64_t x)'],['../struct_sigmoid.html#a75a24cd75cb4d4c9a072811b2d70ad55',1,'Sigmoid::operator()()'],['../struct_sign.html#aa3304c6b43bcad53061614b741d8403c',1,'Sign::operator()(T x)'],['../struct_sign.html#ac48992b675b8b28be1e27e1f2ec5d2f7',1,'Sign::operator()(uint32_t x)'],['../struct_sign.html#ae07a4249e1b61419a3b9ca6c337b7bb5',1,'Sign::operator()(complex64_t x)'],['../struct_sin.html#a7caf98c777521fa5d5c6ddaaa3b779fd',1,'Sin::operator()(T x)'],['../struct_sin.html#aa510cf4595b6d49065ab6b602d8fcb14',1,'Sin::operator()(complex64_t x)'],['../struct_sinh.html#a02cf32bcf560657b9ee34fb1affed8e2',1,'Sinh::operator()(T x)'],['../struct_sinh.html#a1f8ba1858d352ee68861cd6ea861af43',1,'Sinh::operator()(complex64_t x)'],['../struct_square.html#afde739fc544e45dd30964c02dca94310',1,'Square::operator()()'],['../struct_sqrt.html#ab9b16d2b9b03a1c54190f4479a56a4ad',1,'Sqrt::operator()()'],['../struct_rsqrt.html#ae16699fd829e40416436247a39233fda',1,'Rsqrt::operator()()'],['../struct_tan.html#a1e6fb8c691621c69cb9bd393de4f6e78',1,'Tan::operator()(T x)'],['../struct_tan.html#a2ef120c9f92b0d2e9cec8389eda05724',1,'Tan::operator()(complex64_t x)'],['../struct_tanh.html#adce11a7ad33226c6ecff34f46f5c45d7',1,'Tanh::operator()(T x)'],['../struct_tanh.html#aa8423b43c725bb4b88965a11e8cf20f6',1,'Tanh::operator()(complex64_t x)'],['../structmlx_1_1core_1_1_function_exporter.html#ada4e13daeb3ba0f5ebe20ec0663727b3',1,'mlx::core::FunctionExporter::operator()(const std::initializer_list< array > &args)'],['../structmlx_1_1core_1_1_function_exporter.html#a82aeb5fa32ef5638f42dc2372278427e',1,'mlx::core::FunctionExporter::operator()(const Args &args)'],['../structmlx_1_1core_1_1_function_exporter.html#ac8b8fa0a23d58a94e2e9b923dc7324e8',1,'mlx::core::FunctionExporter::operator()(const Kwargs &kwargs)'],['../structmlx_1_1core_1_1_function_exporter.html#a35a3c1d94249ce0fe0e82b0ea047d441',1,'mlx::core::FunctionExporter::operator()(const Args &args, const Kwargs &kwargs)'],['../structmlx_1_1core_1_1_imported_function.html#a3555db23026d30eaeee265fed99947b2',1,'mlx::core::ImportedFunction::operator()(const std::initializer_list< array > &args) const'],['../structmlx_1_1core_1_1_imported_function.html#a5953b3f47c094cc47bcbb0845379ca8d',1,'mlx::core::ImportedFunction::operator()(const Args &args) const'],['../structmlx_1_1core_1_1_imported_function.html#a10fec4eab5851ed825a9b46a31cedcc9',1,'mlx::core::ImportedFunction::operator()(const Kwargs &kwargs) const'],['../structmlx_1_1core_1_1_imported_function.html#a7d1accece61230eec256e0f70610776d',1,'mlx::core::ImportedFunction::operator()(const Args &args, const Kwargs &kwargs) const']]], - ['operator_2a_17',['operator*',['../structpocketfft_1_1detail_1_1cmplx.html#a26bf3d709a58f06228e502af6db8e5ac',1,'pocketfft::detail::cmplx::operator*(const T2 &other) const -> cmplx< decltype(r *other)>'],['../structpocketfft_1_1detail_1_1cmplx.html#ad9c591ef8ae976293f207937d273e9a1',1,'pocketfft::detail::cmplx::operator*(const cmplx< T2 > &other) const -> cmplx< decltype(r+other.r)>'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a153756072fda6d3e53bcca11b46a1238',1,'mlx::core::array::ArrayIterator::operator*()'],['../backend_2metal_2kernels_2complex_8h.html#a681d4fb076973f58f7dac894ec62a385',1,'operator*(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8f06316063fc91747533105f256b55b5',1,'operator*(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7b3bce3f6f17089d87e13e91f580a581',1,'operator*(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a54ae7216b82c5cea362f6b83e1df3a9b',1,'operator*(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a852689073c17596de4fb545bc046b380',1,'operator*(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a168300bbd04d8e97c5e4218cb14ae378',1,'operator*(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6278bd2e0e2805090b33ef666bf7f6bb',1,'operator*(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aecf703522d9ce32dfeefe1e6e903db06',1,'operator*(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7cd44d27fa9a4f13df39894c34fdb348',1,'operator*(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee64dc1890abb6d1035361cb8c751f96',1,'operator*(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1a559ab88dbbb4fd2c7509d2c94e55b',1,'operator*(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a495ae2d9be5d97c4c6448fc4e50a03e1',1,'operator*(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a87ab4b7a502430da664ccb8abd383058',1,'operator*(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5f997839cf49c24ab594a0dff486a7bc',1,'operator*(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa0c2d29950926ae579adf6337fbea64b',1,'mlx::steel::operator*()'],['../group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0',1,'mlx::core::operator*(const array &a, const array &b)'],['../group__ops.html#gac22a67f7de797b1ae59029843cbdcab6',1,'mlx::core::operator*(T a, const array &b)'],['../group__ops.html#ga6f2369ed5fae8ff9b1528670a004dde2',1,'mlx::core::operator*(const array &a, T b)'],['../namespacemlx_1_1core.html#a0cc824d6318f97f7058918ab64ddfc25',1,'mlx::core::operator*(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a81e1c727c3fc48910b030cb65a9e7afa',1,'mlx::core::operator*(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a861d948220d8f48d46c68d2ddb16a096',1,'mlx::core::operator*(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a13d16561812679b36e68185dc4b2d04d',1,'mlx::core::operator*(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a5287610200ff573730c9c92413f48881',1,'mlx::core::operator*(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a377ccc6b4ef36767abca102dca56dc10',1,'mlx::core::operator*(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a5d696b63635ce6967526d6a410f7f6b1',1,'mlx::core::operator*(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abe90e9527bfa3e1c813d41df4a2372e7',1,'mlx::core::operator*(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a5f14963c77f96bcb5a3bef5661a86ba4',1,'mlx::core::operator*(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#acfb06fe9f5fee01dbb5a2b23bccfd0d3',1,'mlx::core::operator*(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#afc9a87f1fccbac05242b91bfbb35c24d',1,'mlx::core::operator*(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0b9678af9b487900cacf6639a4693de0',1,'mlx::core::operator*(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ad5950619081389e6ed7512f38358d33d',1,'mlx::core::operator*(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a65d25d082374761c05b056e1046d1d4e',1,'mlx::core::operator*(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a759191fb984e7737f0ef529c2053ad73',1,'mlx::core::operator*(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3a52675c3d4552b319dd9707844abdec',1,'mlx::core::operator*(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a45d67f5d80fba4d42e34c682a8d22beb',1,'mlx::core::operator*(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ad25880c67bbcbfafbe54dc16418bf736',1,'mlx::core::operator*(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a63c836e1141e07ae72cee770bad01200',1,'mlx::core::operator*(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a265a37b8ee4a97390213e9ec49693e66',1,'mlx::core::operator*(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab5a457da04dcb157a0b5172c4b2244b6',1,'mlx::core::operator*(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#aa56a8bda08be9ef3711496e216a75c95',1,'mlx::core::operator*(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af89612098dd355b1eefb841c753b36ab',1,'mlx::core::operator*(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a4552687a0637f710b5d55bb6378fcabe',1,'mlx::core::operator*(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af69db7def588d7da430434a69456e29c',1,'mlx::core::operator*(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a00af6e5095888f00791ee0ab6d993ad6',1,'mlx::core::operator*(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab48feddc1aa304383e5493923506ad7a',1,'mlx::core::operator*(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a0367b582e85162b4180e086f725e49e9',1,'mlx::core::operator*(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a45f0479526fbccdb00bc73ea7f3b7625',1,'mlx::core::operator*(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a394797646010ba9ef2a1f9b9a4b8ddd9',1,'mlx::core::operator*(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acaaa86b59c7ceb2e092ac07f2a75225c',1,'mlx::core::operator*(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a067d47823a322b88043cce7ce4a3ec78',1,'mlx::core::operator*(bfloat16_t lhs, float16_t rhs)']]], - ['operator_2a_3d_18',['operator*=',['../structpocketfft_1_1detail_1_1cmplx.html#a683fd490182c9189fa2c05b1823edd93',1,'pocketfft::detail::cmplx::operator*=(T2 other)'],['../structpocketfft_1_1detail_1_1cmplx.html#a06f2c26c6fc4722e61b44da4c242ed87',1,'pocketfft::detail::cmplx::operator*=(const cmplx< T2 > &other)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7232b0a0e193b3c6172d6fc2578bf419',1,'operator*=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ade65ebca11e38d56408c512df89b99f4',1,'operator*=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af4348ce3425dd99d069e8fdf06e25a3c',1,'operator*=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2c3c5f793b3d957d7295d7f1faabebee',1,'operator*=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac66657077d55e94197b52b63acb50b7d',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a383165ea838cc3feeee4d9cf54aa77cc',1,'operator*=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab706af260b61f735b28464877d02137c',1,'operator*=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a979374b1dd4e0eaf602326fa901336d1',1,'operator*=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac815eec2c1b15a47b1c6ea6790e77d24',1,'operator*=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8110fae7bcc34a0de5927546b24aa935',1,'operator*=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae4acef3e7ae7dfe359422503f894e885',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc268cdbc30500f3009f5de2b2f0f67a',1,'operator*=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a81f65b04a87a25c7eb1a751d1be9fa55',1,'operator*=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08c1f916302eb9d48c93f8b7260538fe',1,'operator*=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc8e82b8f593b12c6d405e2250ab0f62',1,'operator*=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4611728172afea51860a77fdb06cafa0',1,'operator*=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0b8736e2ae24758b6e24ea72668df5b4',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad920df9579603f0b0ee2689eba330617',1,'operator*=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae97ab6c3ddcc2754b24f86319a5398be',1,'operator*=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3ff4ff59f411010ac8502cfabda4bd6f',1,'operator*=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abd3d82e2dec1847e97eb8fc3bab2985a',1,'operator*=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a738078eb7d5ff94ff48156a555d763a5',1,'operator*=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a435f2f4256aadb1b57fd62bb7f733cf7',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0e4377b120d6305335d296e031ee5b30',1,'operator*=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a917354f77eac26189da8a2f610a00074',1,'operator*=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af725f935bfa0405e5ff17ede3ac47283',1,'operator*=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7c56980c234a04260b8b19298085e526',1,'operator*=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab840ff9de0cdd0e9afffb8baa2a850a3',1,'operator*=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a73416a7415f3fe31525e33419e5e8aab',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a16978f4b16d954ef4d4cf0f32f6c0b94',1,'operator*=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a99aa4cc110d1c7aa3b4c8c5cbf9235b7',1,'operator*=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2179abbc91ce8763e96e39e1917bfa6e',1,'operator*=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab070ea4676d10a10ff3e9379a4068a57',1,'operator*=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0197e039d4c65bf49649a6f250c2d436',1,'operator*=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad3565cc6fd1e088d052b1108aa065851',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a711693988c437c2fb4d7da505982fe21',1,'operator*=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeff4c28986f98c23de1df17043edb0f5',1,'operator*=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7dbf0c75df4817cb4ef8b60c417a89d0',1,'operator*=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a323a80492cd17a49e2c3dd18f8c8b5cc',1,'operator*=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adb465776d3868bda0525d632ffc4d129',1,'operator*=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a12a98d71d670b409b8065e0d61672d55',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5d00eb2ec2b0e15b2753d100694c45ae',1,'operator*=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1a2a683ff40490226eb1371fb905023d',1,'operator*=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4126fb7ed5bbb27a2332c543cf56a337',1,'operator*=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab092d9790ef20fc0386707530aee89db',1,'operator*=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abff1fd2439e31e6e64a3d2fdee3c7821',1,'operator*=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a625dcb133f1f953f263e6200399866c6',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08b6071245513e1726ec68e3b63edc53',1,'operator*=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a13aa79165ec87710e977f33fe0361e91',1,'operator*=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3796dcf819adb1ef8152f57ba63ff6b1',1,'operator*=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aaab79d0b4c9e9bdc059ace6ec58c5b00',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a0dd3893abc8986901872c8365ab1509d',1,'mlx::core::operator*=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a3cc5c154e4ad9a83ad43da8513146fdc',1,'mlx::core::operator*=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a600e77dbc72e78207b5f5dbf4b298781',1,'mlx::core::operator*=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a54833be1d44bc3adfc9ea218fc3685bd',1,'mlx::core::operator*=(float &lhs, _MLX_Float16 rhs)']]], - ['operator_2b_19',['operator+',['../structpocketfft_1_1detail_1_1cmplx.html#a76447ef141c8732d57421749fc81b236',1,'pocketfft::detail::cmplx::operator+()'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ae2adde594b5a4853f6bc78263a957d85',1,'mlx::core::array::ArrayIterator::operator+()'],['../backend_2metal_2kernels_2complex_8h.html#ad6af5c6c5ed4898b49758618e5aee189',1,'operator+(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a09c1a797eb7f43742578680899932f50',1,'operator+(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a551b970f73bb4a3b287653021d000b60',1,'operator+(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a43a225e7e548bb041f3a5d844faaf0da',1,'operator+(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8b6c3fd9d068a2159084359df8b9b449',1,'operator+(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0a5bfe15d95ba540795f4c25ebfa4f07',1,'operator+(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa415ce182fe7582d885fe633fc3527ce',1,'operator+(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a62f891b7dbba0000749cf338f594bedb',1,'operator+(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab43932322f81bf322aa1b0deeee9a987',1,'operator+(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acd15d46ea5827a2a39898ccbb8352eb8',1,'operator+(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a006763fae6e0577fc168ec9446f0f747',1,'operator+(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a12a47e8ac0be788edff57ae0a96d7830',1,'operator+(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af87dfa2122e9c76042dc41fb7f338a87',1,'operator+(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af2737d09c887ee8cd43fdeabceddbe82',1,'operator+(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a12ff4f38aa8474bf76770c7b8e3e18cb',1,'mlx::steel::operator+()'],['../group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c',1,'mlx::core::operator+(const array &a, const array &b)'],['../group__ops.html#ga7d0ec8d01e7cefa6a6b25f11876761b5',1,'mlx::core::operator+(T a, const array &b)'],['../group__ops.html#ga7cc080a4f9d4a667f2099aa0dbfefadd',1,'mlx::core::operator+(const array &a, T b)'],['../namespacemlx_1_1core.html#ac14b984970cafd8fbe24d080949515cc',1,'mlx::core::operator+(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab076069c6f0047c548a8dc29d35dd36a',1,'mlx::core::operator+(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aab9d96b0a168f4d05146000a6212b5d8',1,'mlx::core::operator+(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac4e6f03d7e4ae701b4eefa784f36185b',1,'mlx::core::operator+(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a4cabd600a5271b0d416c91e8d31dd9c1',1,'mlx::core::operator+(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af26df9dc279d71b7cc10892c72162b58',1,'mlx::core::operator+(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#ac3b97eecec9bd8efb313f8f201560343',1,'mlx::core::operator+(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2e3bb121cbde30c2e6d806df0d41ff59',1,'mlx::core::operator+(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ac87ecce4b44b0826e666a169ddc6f878',1,'mlx::core::operator+(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aed3d9cd32698ef0fe65b1280f103b3f5',1,'mlx::core::operator+(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6fa13b9359cf3f575fbda5260e6e035d',1,'mlx::core::operator+(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af240a6471ff827819192808bffeb857a',1,'mlx::core::operator+(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ac25a05679f312b724c406d8b282803c9',1,'mlx::core::operator+(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a54863a54f258acf2b5c734950618e4e1',1,'mlx::core::operator+(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a9f81f5ea8909db9660197217612ee446',1,'mlx::core::operator+(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a13e26c38da0a4e332e0ae4eb0aed9cb8',1,'mlx::core::operator+(const std::complex< float > &x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a59bb13a0bb7f748c8de34415b248bc57',1,'mlx::core::operator+(const complex64_t &x, const std::complex< float > &y)'],['../namespacemlx_1_1core.html#a38a44c412c8be4c8b952d3082cc7db74',1,'mlx::core::operator+(const complex64_t &x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a011dbdbd2413e59e744cf82b05431340',1,'mlx::core::operator+(bool x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a230e3b7c479add1b171fa0aaa3a8b13c',1,'mlx::core::operator+(const complex64_t &x, bool y)'],['../namespacemlx_1_1core.html#a3a6f43c2485f0d42293184f1aecbeaee',1,'mlx::core::operator+(uint32_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a766157c5d5d00fdf3da95eb7cb2981b9',1,'mlx::core::operator+(const complex64_t &x, uint32_t y)'],['../namespacemlx_1_1core.html#a64dceec2bb03eee963a2a1bc1ac69284',1,'mlx::core::operator+(uint64_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#ae36badb78a17cd7d13663a69645fc328',1,'mlx::core::operator+(const complex64_t &x, uint64_t y)'],['../namespacemlx_1_1core.html#ac1afa5d4c856e4b58109eff086e70ffd',1,'mlx::core::operator+(int32_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a8978def3c2cfe2a96314d564613b80db',1,'mlx::core::operator+(const complex64_t &x, int32_t y)'],['../namespacemlx_1_1core.html#a5b8af5ca4c0e37aba0b7530542bd64c2',1,'mlx::core::operator+(int64_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a3eaa72850205c18450c3af9a01cda219',1,'mlx::core::operator+(const complex64_t &x, int64_t y)'],['../namespacemlx_1_1core.html#ad38b38a3faf050735d45eed4438ee27a',1,'mlx::core::operator+(float16_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a358e66ff205bda3e8542427b6d2edadc',1,'mlx::core::operator+(const complex64_t &x, float16_t y)'],['../namespacemlx_1_1core.html#af56d4b85e329e39a825c01a50e3a2522',1,'mlx::core::operator+(bfloat16_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a806a495a129ebaab69cc57ca7db831d6',1,'mlx::core::operator+(const complex64_t &x, bfloat16_t y)'],['../namespacemlx_1_1core.html#a09fc6ebda917969383783a112a8547e7',1,'mlx::core::operator+(float x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a7ed0e2cdb65612f54e67166762cb6408',1,'mlx::core::operator+(const complex64_t &x, float y)'],['../namespacemlx_1_1core.html#af7577c91b8c43682f0ebc9eb9758aae4',1,'mlx::core::operator+(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#abe36af9951afd8dd3ffe90ceedeb7f2b',1,'mlx::core::operator+(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#afb9f780dd056a4f975518f71a3b021ee',1,'mlx::core::operator+(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a6a8e093b24c4c789b7cd160f7e7f7de9',1,'mlx::core::operator+(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#af3a603690fd3de9e4f7f2035a4d25621',1,'mlx::core::operator+(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afa2a4bccfeea9688ac922cb638341511',1,'mlx::core::operator+(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a6111e94d51de12391e5d68b765f28fc3',1,'mlx::core::operator+(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7c7dd6d346e0cdf398a896f2c6958258',1,'mlx::core::operator+(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a00872a443f462b0ae0a30c84fb001bc0',1,'mlx::core::operator+(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4f5d80d03bae6d8d90455d3c47a8c116',1,'mlx::core::operator+(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a78f1f388f9d81ed93f60311f4645d8d0',1,'mlx::core::operator+(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aa43e1d6958c5d5a6fa9a625a1660e741',1,'mlx::core::operator+(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ae877e1d5e3cf57734da8b49535fe3fb3',1,'mlx::core::operator+(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a9a5ae769f67f886d59c8e292a8218550',1,'mlx::core::operator+(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a058878237ce50baa4c909d8d15448d7e',1,'mlx::core::operator+(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a95fd207028f125eefbafe9e0522407fe',1,'mlx::core::operator+(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#abc6425a3fbb386f5ea5964b42507e989',1,'mlx::core::operator+(bfloat16_t lhs, float16_t rhs)']]], - ['operator_2b_2b_20',['operator++',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a3efe69356a84d0d4438f033992fcbd9d',1,'mlx::core::array::ArrayIterator']]], - ['operator_2b_3d_21',['operator+=',['../structpocketfft_1_1detail_1_1cmplx.html#ad4e69dcd89bdb7764c9c5807168f911e',1,'pocketfft::detail::cmplx::operator+=(const cmplx &other)'],['../structpocketfft_1_1detail_1_1cmplx.html#affa618d8850a7c232793b7c61db6d184',1,'pocketfft::detail::cmplx::operator+=(const cmplx< T2 > &other)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab04f480aea9fbba0895068c7558dd400',1,'operator+=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a251780ac4592cc2b1a543e417ff57770',1,'operator+=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24381d991c2d570aa953694f396a69b5',1,'operator+=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7595740d4cc12924905d6bd1b99ee4da',1,'operator+=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac1498acb8c3623b5f412f70ab6a6528b',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abce5ab327110c164f054b43ed47f79a0',1,'operator+=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae0c70198e236ffe1a98f79987c686419',1,'operator+=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a13b3338935440ae51ecc4a356093efc5',1,'operator+=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a0cb8544b4ebd2906ba8e7f2868e8de',1,'operator+=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7b134429ea0c8493800ff8b465410f9c',1,'operator+=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4154f90ab7857ca856f9e15fe1bf5acf',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab9ae6a51e2027b02cac9966e05f3ba68',1,'operator+=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab93ce536eb7998bee00de4af868e31a9',1,'operator+=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad0ae9e2b4874f991a2c853e1c1fe735d',1,'operator+=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a194a6670cc25ade35a24b566f31af785',1,'operator+=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3d0d689516c99003659c5d026847bd2e',1,'operator+=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a007f58508b98bb79e5c323ed0dec89b6',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa7198e580e2a83c1fd01a4b6fdf86a80',1,'operator+=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a15573fefd880adefbba079b1c1bd8082',1,'operator+=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a104cf94cb9e359d1b6ef92ced2ce0c27',1,'operator+=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa04cfcb52191fd23205a1a3572b46ae0',1,'operator+=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad684bc2ae1a2a627cd3e4a4c641e2d77',1,'operator+=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1e28448e35f4934075b397c34ba3d66',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8ad16afd7f1711de83c0cec5af868f76',1,'operator+=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac45e9ca0c7155caebe3d0f7261518077',1,'operator+=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3c62ac679d6aa515144d40ebafe4a188',1,'operator+=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9ff5ab3aef1057fa083b53a65c8aba03',1,'operator+=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae74bb0a3c12cd1a23f3d29ce307d6fb1',1,'operator+=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac188bd19f236b098d603b0d8acd08921',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef9fa600d107b509f2e3df7d6b080e01',1,'operator+=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af5713afb3a62967a02c3c20661951ee4',1,'operator+=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7f1b84352a3ed6171444a43da1fc7e92',1,'operator+=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af1983edd26245e6e51c6e47354095e32',1,'operator+=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8cd55d1a579540eb450e12a8a8a950be',1,'operator+=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a588ef0f7e03f306758524d378278976f',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a74751abec7086f85f4f26ced44f1ca1f',1,'operator+=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4dd3cf0e5aa116ff330352a50c18cde7',1,'operator+=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afb9a0e18c0e40c77e6143fb7d84ebfba',1,'operator+=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adf0cfd9a608a6fb3d57933e32e7d81d2',1,'operator+=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4bd92db6c8b9b5dc96332c7ae3eff8c7',1,'operator+=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5d628a5bc4fa755610392f47a523a1f1',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7c790442f77f2437b482c4a55e224fc3',1,'operator+=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a77bab4481b41be50297b257e95058706',1,'operator+=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7816a97d16b1d2f8a90227bb1da2f6ac',1,'operator+=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac244d140c6149726ea44174d3e836ca3',1,'operator+=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af802541c4c65ee4442acd495de4d27fe',1,'operator+=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac06eb2fea47a09a8a8abdaa1aa9b4603',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5263b2463fecdc97f9521d00bffea059',1,'operator+=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24ca436ab299a710263d65302532dd3b',1,'operator+=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee1bdf0ab2e445293708b476e8cfde3b',1,'operator+=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a190e27077f0fba642a86f5c8f488bcc2',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a9f2c9d2f21fbf9fbbacd940c6967c9d1',1,'mlx::core::operator+=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a0b1b3c48afc0a785282e43435bba8418',1,'mlx::core::operator+=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7b763db8194e6fcb1b87eab143dfa47a',1,'mlx::core::operator+=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a827167f6a1ae55428fd218ddd51ec3b6',1,'mlx::core::operator+=(float &lhs, _MLX_Float16 rhs)']]], - ['operator_2d_22',['operator-',['../structpocketfft_1_1detail_1_1cmplx.html#a460da5db36d1c72fb1ed3496fd3abde4',1,'pocketfft::detail::cmplx::operator-()'],['../backend_2metal_2kernels_2complex_8h.html#a226cfd54d49f02e35c5aab3139c7596b',1,'operator-(complex64_t x): complex.h'],['../backend_2metal_2kernels_2complex_8h.html#af5608264cf920688607059b4e8cd3117',1,'operator-(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6aedc8d6d0980134ac69b96f22d9a855',1,'operator-(_MLX_BFloat16 x): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a333f67614dbf8027439a7e124052cb85',1,'operator-(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a891aa4bf46c20a26a55061736aba25f1',1,'operator-(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7ad7ff44a3200853711869f7a577d931',1,'operator-(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af69ef8f1d8ecae0e6f755bf1c46cf075',1,'operator-(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5bd875a54b79b2dcedf674807c3e53c5',1,'operator-(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab02f8646b47806e1d2038f248df03f06',1,'operator-(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab27b26182c7c6e08af37e6d511fd9253',1,'operator-(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5868c85c988ec3432cf86d7df40e464d',1,'operator-(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad03ef47e6cc7521bbfb45740dee20f88',1,'operator-(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab789f8a400512ff27e36b3373170f0c5',1,'operator-(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7f601b22ecc480132d82ad782e5363bf',1,'operator-(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a152366ab4e2ccc867e919af6c74ced91',1,'operator-(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a42bead8ef0beb9f3452128d64cd4df9d',1,'operator-(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aca8ef21c16984ccb329b3bd0c1e4be48',1,'mlx::steel::operator-()'],['../group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816',1,'mlx::core::operator-(const array &a)'],['../group__ops.html#ga0c7f3cb36d4ca516c7a33142f88b9181',1,'mlx::core::operator-(const array &a, const array &b)'],['../group__ops.html#gae68d3d0691ba951501218e98439f3465',1,'mlx::core::operator-(T a, const array &b)'],['../group__ops.html#gaf5e5d882c51ad0a0ea315c274d5439b2',1,'mlx::core::operator-(const array &a, T b)'],['../namespacemlx_1_1core.html#a622ce842fe44e4b6a95e03242341b459',1,'mlx::core::operator-(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af32a99d930d49e9b178472d7a65531ab',1,'mlx::core::operator-(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a3555a2b31fc0925850d3240e85e03ec5',1,'mlx::core::operator-(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a46080889fd9e5c3f9916508e97dff5ad',1,'mlx::core::operator-(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a9ca27fd1e512c8ed126342e565da12ae',1,'mlx::core::operator-(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3803f8d36558d32bb7dd6e580ea683b4',1,'mlx::core::operator-(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#af5d865528989ca66b3d357e5ce4e0300',1,'mlx::core::operator-(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#afb784b960f55aeb4edd7f567fa74d443',1,'mlx::core::operator-(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a29cbacf4b399c24728fb0808fad498f9',1,'mlx::core::operator-(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aececc0e451237aa6c0d1a2c3d828c86e',1,'mlx::core::operator-(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a383a26cc2689c98fd6c4435ade8dc669',1,'mlx::core::operator-(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad6311ef8df59bdfb212b5cf8169246b2',1,'mlx::core::operator-(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a23b7329bc1c93c8ac0a1f576565fefb0',1,'mlx::core::operator-(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad8d650bf63998abd716ee0ca28e1cbb9',1,'mlx::core::operator-(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a7339b33201254e9119d99d3a728ded72',1,'mlx::core::operator-(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a064318b7a16e5cb6d0a6407501b5c7dc',1,'mlx::core::operator-(_MLX_BFloat16 lhs)'],['../namespacemlx_1_1core.html#a7bae3ff296d9a60ff3c7e448f7fbc6bd',1,'mlx::core::operator-(const complex64_t &v)'],['../namespacemlx_1_1core.html#afb5069ecebdfd9d388c26f83df12c93c',1,'mlx::core::operator-(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a8d126e3f3fa9f8c1c1ae1b09f94df487',1,'mlx::core::operator-(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ad04f1ccd2cd7c487a2f2aaa055939f64',1,'mlx::core::operator-(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a15eb2ea76508ff823fa0591e811d0b7d',1,'mlx::core::operator-(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a96d9577db38d6809d022893e32feeda1',1,'mlx::core::operator-(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a5d9c02765c1672930757416411567bf2',1,'mlx::core::operator-(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a6105d3b5266666b7c6bb9469285a9ec3',1,'mlx::core::operator-(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a777aa772dfb205b25d26f3180d98a2f6',1,'mlx::core::operator-(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a085eb092f4ada47f8169de62886cff90',1,'mlx::core::operator-(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab25e5d211e2c8785b45c3a81a6282e2b',1,'mlx::core::operator-(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#abf5d09561a81b0f0b32d59d77e32e16f',1,'mlx::core::operator-(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4ce6867dbb4d1631d1870dac14022dbb',1,'mlx::core::operator-(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a8a049e646e0442064cfe9e202d7047c5',1,'mlx::core::operator-(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a78e2a1cfc65453185bcca13bd4f523cf',1,'mlx::core::operator-(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#af143cf68673e06390d4bb2ec2892bd22',1,'mlx::core::operator-(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a46d502dfe0b027955950d4e716c2eb26',1,'mlx::core::operator-(_MLX_Float16 lhs)'],['../namespacemlx_1_1core.html#a2631e78c6f0a602f6754ac577ec75f83',1,'mlx::core::operator-(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a73d79cbd75d543d0837b8a51bf103f9e',1,'mlx::core::operator-(bfloat16_t lhs, float16_t rhs)']]], - ['operator_2d_3d_23',['operator-=',['../structpocketfft_1_1detail_1_1cmplx.html#a12441ff423274bd1b54245933d69ad7e',1,'pocketfft::detail::cmplx::operator-=()'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab225043bd02bb423930bc98aae9c2bca',1,'operator-=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac2f1e1f2365cfa531b1519aa9ff67695',1,'operator-=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a513501355a5912a1263fd8b10864142b',1,'operator-=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab4f4ecd62c3d8b3363d02019573dc9f1',1,'operator-=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a92d1348f201d78fcd474f75d5b23ef68',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3eefe9a7f5fb226335ea687012f32d5c',1,'operator-=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef62c7e3e494b6a511a7833c0d942a60',1,'operator-=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad30726cc8b69fd300d33c2a46e123c28',1,'operator-=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8859b5b8dc241e4f58243c85d2630cc8',1,'operator-=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7003e1e5881e3d106257f22b6a3e59fe',1,'operator-=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3165e37d393be50c2cfa9ddcba153684',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a76f5bd895b7214cbc3cea3440992718a',1,'operator-=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7167343d90eb70e5a0d5fa9ec5398e94',1,'operator-=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9b31c363ebc93d592b6fa0e27b00335a',1,'operator-=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a967a1d7b5664f616e5b6f2d257367f0c',1,'operator-=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aff19193e1b2cee29a8737318e95cc74a',1,'operator-=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aede0cc4179507b739849948f1a2fed4b',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7e1a6056f9c96f3c89fe204dbf103be5',1,'operator-=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9d06cceea5c179bcc608452188bd7d6a',1,'operator-=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0aa9ffe056f49fda181bbacbd60556ea',1,'operator-=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ada5685d99c2d6708d1c4ef826d68e879',1,'operator-=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a726cecf778b8584b6f7c37db1b064576',1,'operator-=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3816a35f8468156d59c239256c12dcf3',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa332fae098e7c6dc23b98bc0026f1070',1,'operator-=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afb3cd302e0b78902c62111dce4494fe8',1,'operator-=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abb884888f14086cc674657677cb4b8bc',1,'operator-=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a38bb89f925eca4f9c042f6ee7a2c0193',1,'operator-=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac30c580713f354916088a7dc049ae4cd',1,'operator-=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a715c824ee8c87e0256114a85624d9949',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7bc91aaaf476a37063264d1d53d862cc',1,'operator-=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab155f418f15cabd86ff942c6f9472ddb',1,'operator-=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aaa66dc6d7b2c5efbfaa97ca9c7872bd8',1,'operator-=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a696978d9401e09200045b2d8aad045c2',1,'operator-=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae998d8f423a9fb73405cfbd4b836bc72',1,'operator-=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a279d09ab8542f1c1a8dc8173b65946b6',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a491dadfae957cd7cc0c36188d910f6f6',1,'operator-=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9a837c3b9c4e42f53d7cd1ed0d266e2f',1,'operator-=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acf7af2284269544064b68e807064bba4',1,'operator-=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a28d297705e29009197418546ef435393',1,'operator-=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a948579a4d9ba276523190b03b09578fb',1,'operator-=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a4b98a0a11db5b77cf9168df37c8bc7',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31a3d8f2ff8038f7e0d717845c039808',1,'operator-=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1dac193d9f1c8c0eb4473441895f8c58',1,'operator-=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad2817d53fdd4b112babfb6f0b38c8f39',1,'operator-=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa705d87cf4b78e9d7c6b07dd0c66cac6',1,'operator-=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a542affc376726840647a6e93acf2c1a7',1,'operator-=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#add18cfe4c0d38e95c6dff6bab3e7a932',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab1de7e7e7304ff3598925d2e69134764',1,'operator-=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0d3fb52437c677c5d0f1a3642384b15c',1,'operator-=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adda64cae388baac1f138b06dc8595237',1,'operator-=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af20874a61c6c3f4c3fd045a96e806644',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a8b8a55690df46d97fcfc2a60120783af',1,'mlx::core::operator-=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#ab03949b1f60fa035ce454a894cd73ae9',1,'mlx::core::operator-=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#adaf70bbfb3667df0d08fd3c99896e20a',1,'mlx::core::operator-=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a321c98e5a78621d3c9a3895f707f2f1c',1,'mlx::core::operator-=(float &lhs, _MLX_Float16 rhs)']]], - ['operator_2f_24',['operator/',['../backend_2metal_2kernels_2complex_8h.html#ae6a708f67d6fd9b0962aa8877cec6d35',1,'operator/(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9f16a44e1c9836ca57edc1d7b93b5d7c',1,'operator/(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aacaedf12f862c76457133336dd6fc446',1,'operator/(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a584a513596de20663dad951a5b81695e',1,'operator/(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad8f7b11669736fbd6ed2e28211d877d4',1,'operator/(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a59515695ebc48844345fa5120511aed1',1,'operator/(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8c8ac6736440fdca366ebdefe2a12b9f',1,'operator/(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad6859b04680d0d26d75fd6c4dd74ee24',1,'operator/(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4720cc79ab2b8e39952ea9ef20e51250',1,'operator/(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a72d10ec0e62949247da129eb3a83fb9b',1,'operator/(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad6399ba2b8708899739b4cdbb44add8d',1,'operator/(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a998b1ba877a606aedf722ab46b290403',1,'operator/(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa3277ae33976c70f7bd937ddff027b72',1,'operator/(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa708a970a200822c99c0489f389469fa',1,'operator/(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a6bde717aca2051499f73a3eee199bfdd',1,'mlx::steel::operator/()'],['../group__ops.html#gaeedf77f722b394429f1a7f6c367883bf',1,'mlx::core::operator/(const array &a, const array &b)'],['../group__ops.html#ga7366ec7f453be2a4dc449f0faa1bf554',1,'mlx::core::operator/(double a, const array &b)'],['../group__ops.html#gadfb324ae9b4feb2c7ea0ac6ade639f38',1,'mlx::core::operator/(const array &a, double b)'],['../namespacemlx_1_1core.html#a7573ac3b93ddecd69e9c88a26fc84ba9',1,'mlx::core::operator/(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a40e868dad70401d9aa9ee9c32235c315',1,'mlx::core::operator/(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a7587c28fbd2023b134e5fc12bb0dde23',1,'mlx::core::operator/(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a92cdd377c408becf4cf83c1ee9b7085d',1,'mlx::core::operator/(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aef89566301cb133d98c8e7bdd2b7bec6',1,'mlx::core::operator/(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a94e7b51185590492b46916685641276f',1,'mlx::core::operator/(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a04584788c08180835219d0ea1e2b97b1',1,'mlx::core::operator/(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad5af96e2ff09d207eb1e1980fe3e7c2d',1,'mlx::core::operator/(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ac2217bf760038cd011781158923149ed',1,'mlx::core::operator/(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aea414c04bddc4b9b609262e97398f1b4',1,'mlx::core::operator/(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a27fe23230cd082c0363b9451b731ce6b',1,'mlx::core::operator/(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abdd9bb8fb4411e5924f3eb7ef1bb52f8',1,'mlx::core::operator/(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a50bae338a7353f8b0ed3441071bb0cf6',1,'mlx::core::operator/(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aab26a3284dd3ac7d47c8b5b3a3290ce3',1,'mlx::core::operator/(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a749f48db01de38f259a0c6750a97fa77',1,'mlx::core::operator/(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a32a6a08a2a4652975b0a1bd1fcf3eafd',1,'mlx::core::operator/(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4b66fb38ddc5cc0c2489583d5c499602',1,'mlx::core::operator/(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a45726f1905b709cf8253e6efa046027b',1,'mlx::core::operator/(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afd4170c1e364384f30e6bae341146fa6',1,'mlx::core::operator/(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aef85739d150b9d5609973da8a3f1086a',1,'mlx::core::operator/(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af52a941f8ed9b25eec91402c7b9e281f',1,'mlx::core::operator/(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a477cade78296bc85894170f62db68870',1,'mlx::core::operator/(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a22f5a2257e11423fc2fe18e2dce91590',1,'mlx::core::operator/(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a640d3574dfe6ad934c720ae8bdd78bfa',1,'mlx::core::operator/(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a6f65d8fd0cdddc96fc01f6af95804873',1,'mlx::core::operator/(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a517019d42d4e426b7b98e1c719bb47ce',1,'mlx::core::operator/(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a0beb7a223c542015a4eff4aed814a9dd',1,'mlx::core::operator/(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#abc9b1bd5018d46514bc19d23db2e5063',1,'mlx::core::operator/(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af22937df654ddbd6e398ef12764d18c0',1,'mlx::core::operator/(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a775aed5f49b530c57e71cbac81404d45',1,'mlx::core::operator/(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a97efcd96d6be666e5608034ae77289ef',1,'mlx::core::operator/(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a899851f85dbddd96f9d36319b82542a0',1,'mlx::core::operator/(bfloat16_t lhs, float16_t rhs)']]], - ['operator_2f_3d_25',['operator/=',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5aa3b8c68a2b58d41ea33eaabbf83095',1,'operator/=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a90a1c5130db515db48624d8587edbb91',1,'operator/=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a65f30a2dc199134e35bc7c5d431b2263',1,'operator/=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7172d84db640e6c49dff0d08dd64b53e',1,'operator/=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acf7cb9927bf09022088401923f2e1916',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a86b2a001cbec0d3a8d762a3c7ff47b0b',1,'operator/=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a744f72ba83522fe3cc2a49a007b42543',1,'operator/=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a77c678665b34df7652dcde053ca73185',1,'operator/=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae0614b6b199d8a65ae95d4621b118b82',1,'operator/=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa846fde89c7d2d18b18ef180a8a9c8a3',1,'operator/=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08e778be18e4a291c108fcc528b981d3',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6b9e49ad9ea256d2d0220c0d81552602',1,'operator/=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab933bc3cdf9adfea10ab9dba5292c812',1,'operator/=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a25e7c5d2ecf3375756d59074f333858f',1,'operator/=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ae4a80fde67eea9a0a37b2803946544',1,'operator/=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a912393b7208fa45bd1e87f30b218b68b',1,'operator/=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a18963246f2b640874bef6dca7049f64d',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0e2c2c2cb50b3a55ff213f18978aca35',1,'operator/=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a64f1136b17006f168ef837e17240814f',1,'operator/=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae46d75b8046d557452d74513f1106710',1,'operator/=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08d2460e259b9106d90d889481ad60d5',1,'operator/=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f7fd418408806ef498745c6fdb2c062',1,'operator/=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac703495cb370b52526a5a2d36ae26038',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ca11d43174baf0a729f93b35eabcbea',1,'operator/=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9f835a0a80c411580c97b65fdc5bdfd3',1,'operator/=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a17f47ec9cff60f8e1b3477a2793b7ac0',1,'operator/=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5be23e296bbed3a885586a6424b1666e',1,'operator/=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afba39221eb54e272aae79910b3cd7ef5',1,'operator/=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac057d95a2bf087575584aa6f9a2c6bf5',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab986ae2cec780a1f494b7b4468b7ba11',1,'operator/=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a44522c2304c6396bbe6b9d32000f4b6f',1,'operator/=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef8e7e499ea9d432aa743d83c076f945',1,'operator/=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3a0a3edbf1ba2314551454059c3f422b',1,'operator/=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acb9f0aef9fbdfde8a4f46e33b0d6c52f',1,'operator/=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a303dfcc81ffd355f866f863d7d9f0fa5',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a359edd4bcb8776861ceb26a3005624c0',1,'operator/=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc9f32cc6f40768df4285fba2e4783c7',1,'operator/=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae71f66d814a03f6377c9d86cf0a2b5d7',1,'operator/=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad0125b6baba3065a87a174ec27aa9a61',1,'operator/=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5cc74ad3e522d7104e6e2117751151ad',1,'operator/=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab3b594321fb42b0c2da99954d1e0976c',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4a0023e2fd08875156cd6ef747fbb5cd',1,'operator/=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4358ee606e66ba2081fcf94f9c3b5915',1,'operator/=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1e7ef6f065695d4b1d017547b60ef62',1,'operator/=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a284dfc702f0f67b9c233b87162eeabdd',1,'operator/=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab8f211ea896fc5190004f3ad6ad8932f',1,'operator/=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7e1bcf3bc06cbcbc304c0cdf729802bc',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abbe42648a46092137b303ccd08f7df86',1,'operator/=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af1a12a1efb618a57da6dd41ae18cb53c',1,'operator/=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a94686039356dfa9aa45608a8b0562fdc',1,'operator/=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa251d6483d3b099d1b5311fbe6f0bce2',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a045ff27257cb6d8ab7a94771ba5a17e6',1,'mlx::core::operator/=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a58112951a56a0f9f8c90b60fe74f9508',1,'mlx::core::operator/=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae736defc89a04fbaf7627ad2695bb838',1,'mlx::core::operator/=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#ab1f260710251256ef737dd59be9e143c',1,'mlx::core::operator/=(float &lhs, _MLX_Float16 rhs)']]], - ['operator_3c_26',['operator<',['../backend_2metal_2kernels_2complex_8h.html#a67674e32596a9dae2258bb8e0e6a2058',1,'operator<(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9ef6a57b7185e9ca49e255fec1a44e25',1,'operator<(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aab02c65bc38ea66335b2192ead4095a8',1,'operator<(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae91686513e284bcc9635833744bbdda1',1,'operator<(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2486f3b5de85b0d57f458d8f21f82b42',1,'operator<(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a435a2aec4c777b4b184ff5d24992e8a1',1,'operator<(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abdd04257e6a73883b5f56f1186d0e906',1,'operator<(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a69984aaa05ae1d4fccccf7f57e8ecb4a',1,'operator<(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a501cc01d5bf15d9f03aa28545f9624ea',1,'operator<(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1b029e4ca72125a5f9471f582c819705',1,'operator<(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0736a76f56578d26ba1422dc8b744a18',1,'operator<(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24b1fa8998c892f90f8dde7c34fb10a5',1,'operator<(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af80ff2020ec2c4b406c5fdae3fe55e63',1,'operator<(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac03f6eefb836373d37dc280b0d813d78',1,'operator<(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#adb5f24b57d98214fc215a06475f21412',1,'mlx::steel::operator<()'],['../group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3',1,'mlx::core::operator<(const array &a, const array &b)'],['../group__ops.html#ga1ef8ea11cf15ce628c54201fa42748ef',1,'mlx::core::operator<(T a, const array &b)'],['../group__ops.html#ga95e72226dc7a79c40b3d16f990922050',1,'mlx::core::operator<(const array &a, T b)'],['../namespacemlx_1_1core.html#a987d631e1508e8df55d98ddd57e4d086',1,'mlx::core::operator<(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad3fb46370cd8f0992866fad9e2c64a3c',1,'mlx::core::operator<(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a3026691bf7ee5095243a8611bf3411aa',1,'mlx::core::operator<(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0d42d6c1d5f77a96e2f296b8ebd79ee6',1,'mlx::core::operator<(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ab5ce08a7de0a0ca00d61f7a7f8ea3ab4',1,'mlx::core::operator<(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abce8b7f24b61e5ec0f9a3afe20845caf',1,'mlx::core::operator<(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#aff97612627ae1ed260c43c0a7af0d306',1,'mlx::core::operator<(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a9119e518234df7923cae2b3802d59bf2',1,'mlx::core::operator<(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#aefb9b05ce8864ada99a920ab32017b89',1,'mlx::core::operator<(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abc55f3676c2d112a6e9ab276bd6b1796',1,'mlx::core::operator<(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#afe6581a2c45f24d7fab1e4006c1e3c70',1,'mlx::core::operator<(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aca1d50cdd9506481dcc4cd1ad4a4f734',1,'mlx::core::operator<(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a310720f513b6a2490e9df80c65f1bfb3',1,'mlx::core::operator<(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a29e457a170b6cefb6ba1e394c96c6f7b',1,'mlx::core::operator<(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#afd4519985b6b207ec41ad8530d1036df',1,'mlx::core::operator<(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae1e41ca94022e43a00cdfc5845102daa',1,'mlx::core::operator<(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ac80f4022bffd95b57526685ce8e1cbc1',1,'mlx::core::operator<(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a3a8f6f0af477788c4f0aa98abfc5f1ab',1,'mlx::core::operator<(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a3728ed9b6cbd152bf675251a0501b466',1,'mlx::core::operator<(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a5b9ad811a5e1358100c5423dd70ea387',1,'mlx::core::operator<(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a5c77e1db83995d3e06a8a26265bce5d6',1,'mlx::core::operator<(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab8a0a3f70664049b35ce1887bd8ff5c2',1,'mlx::core::operator<(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6652d93bfb2d426e261a1712a181a4d2',1,'mlx::core::operator<(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a03758b8d13da2de07cc4f4fc45d2854b',1,'mlx::core::operator<(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a325161b81a9ff179fd37d949780a17ba',1,'mlx::core::operator<(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a92eca79fce8233e4299343eee3996511',1,'mlx::core::operator<(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#adb016662b8f7eb680abfe1a421eabe72',1,'mlx::core::operator<(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3c_3c_27',['operator<<',['../group__ops.html#gad656c30f9fd7d9467e405657b325aa7e',1,'mlx::core::operator<<(const array &a, const array &b)'],['../namespacemlx_1_1core.html#a1e5c30e316afa30c14bc48b92afdb794',1,'mlx::core::operator<<(std::ostream &os, const Device &d)'],['../namespacemlx_1_1core.html#a4ddd07021b36c848d6fb1dd9ac276822',1,'mlx::core::operator<<(std::ostream &os, const Stream &s)'],['../namespacemlx_1_1core.html#a0023c267cf81345fad65e7a797954cd3',1,'mlx::core::operator<<(std::ostream &os, const Dtype &d)'],['../namespacemlx_1_1core.html#a1fd58658474fb842d648dcf8f7d9f078',1,'mlx::core::operator<<(std::ostream &os, const Dtype::Kind &k)'],['../namespacemlx_1_1core.html#a123331f01188bd76e37623b63b6b4340',1,'mlx::core::operator<<(std::ostream &os, array a)'],['../namespacemlx_1_1core.html#a4e733bba89760abed32393e085812b22',1,'mlx::core::operator<<(std::ostream &os, const std::vector< int > &v)'],['../namespacemlx_1_1core.html#a5e5bd5c57b1cf19776bdb41e732861d9',1,'mlx::core::operator<<(std::ostream &os, const std::vector< int64_t > &v)'],['../namespacemlx_1_1core.html#a42a19c8442b173606e714364227e7d45',1,'mlx::core::operator<<(std::ostream &os, const complex64_t &v)'],['../namespacemlx_1_1core.html#a57eb97a5eba99a846ac429795e407574',1,'mlx::core::operator<<(std::ostream &os, const float16_t &v)'],['../namespacemlx_1_1core.html#a7db909d54cf07375e89424c32c07a29c',1,'mlx::core::operator<<(std::ostream &os, const bfloat16_t &v)']]], - ['operator_3c_3d_28',['operator<=',['../backend_2metal_2kernels_2complex_8h.html#aee04c9a63c6716a99a027418354debb0',1,'operator<=(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af469c58cffeab488c681f4b33f02cd05',1,'operator<=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a81eae168dfafd299c2b94e3e8558cf',1,'operator<=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f486bf02c6ad5b9b6a96d3450f03e47',1,'operator<=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acba9efe192d22b7781b4622103c7a944',1,'operator<=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aff100489cc40ad276c2d5d67a9df67db',1,'operator<=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7eac96f64ca42991caf819c8e8c8d2bc',1,'operator<=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a88c11cd37600de5480570da3d2ae5732',1,'operator<=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08c7d12a0d16565fbf052dba2db8b22d',1,'operator<=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2b9de9624c0a507b4ead85f898ad9daf',1,'operator<=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a28f8d21c5eef047c701cf690ce9c2ef0',1,'operator<=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a14b56c687053ee2432398a25663c068f',1,'operator<=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f360806708b95a3be400af0b8871b57',1,'operator<=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a80d288f22cadfdf5e904410349e616a1',1,'operator<=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a6cc3bab5e7f6e7c719c82afa90ad2827',1,'mlx::steel::operator<=()'],['../group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6',1,'mlx::core::operator<=(const array &a, const array &b)'],['../group__ops.html#ga150a9be467c9f91482a6d6fc13504bc4',1,'mlx::core::operator<=(T a, const array &b)'],['../group__ops.html#ga624eeccef0cc4b130e1325abfea057cb',1,'mlx::core::operator<=(const array &a, T b)'],['../namespacemlx_1_1core.html#a0066a47cb21223ddebc77992ee874fb9',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2593dbace3ce50e7146d9514726a543f',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a88654bcf6c9728517a2933ca2e29a7c1',1,'mlx::core::operator<=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a5d4f449e9c1699b99fcf894dd15e8af3',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a6b678bea8fdcda1f11c6691b56a15211',1,'mlx::core::operator<=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae8aacc606ea16f018a90eae758830a35',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a25668dea4ffb51c7c00eeecb9530d1d8',1,'mlx::core::operator<=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a084558b6a5487549799c49c37c9e9652',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ade2e2a0daa79d5c52f278f85f03dde2e',1,'mlx::core::operator<=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a750a2d2b4976ad94b08994d081f83445',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ade5a175ff45347689ac4c798d04c8ffc',1,'mlx::core::operator<=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae25e0c01b46612f039313a4825ba6428',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a5c90f16d8f6edf4b75c96b945b9fa591',1,'mlx::core::operator<=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a8cd6583fa0fc9957f993e00b2ec01d91',1,'mlx::core::operator<=(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a012130a0458cbc30b88365e0e0eab232',1,'mlx::core::operator<=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae8c890bdcffadee8c5dab85c907f57eb',1,'mlx::core::operator<=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a43cb070553c1f2fffb32ef6670e30980',1,'mlx::core::operator<=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ac759b7798d668a99535e59e26d6ba192',1,'mlx::core::operator<=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a70e528a789b5660d98e783b045aaa379',1,'mlx::core::operator<=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a40bd8abb8a4d989ddabbb298518bd7f5',1,'mlx::core::operator<=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a4155d4b0c76f37ab5e0b54f9cd683f35',1,'mlx::core::operator<=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad8bb648d0603a206e0392990c911ca0b',1,'mlx::core::operator<=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ace72a5853f2afd6510dcb97d54fa650d',1,'mlx::core::operator<=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab38f7a0d3c0809071ff5d3af859018d6',1,'mlx::core::operator<=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a7904b886d7b535a6af0a885d00597323',1,'mlx::core::operator<=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a57952168bd0b54c2677204d4ab1cb6e5',1,'mlx::core::operator<=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a6235dc5f4db517618bb3449b08c96e8b',1,'mlx::core::operator<=(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3d_29',['operator=',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a027b84cddc8d476f736ac1f1a9991fe4',1,'mlx::core::allocator::Allocator::operator=(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a2e971b47339b1d0849a334a902a9df3c',1,'mlx::core::allocator::Allocator::operator=(Allocator &&other)=delete'],['../classmlx_1_1core_1_1array.html#a8acf2b4c75f9b7f79da6675dbc36cf36',1,'mlx::core::array::operator=(const array &other) &&=delete'],['../classmlx_1_1core_1_1array.html#a5c89c2406a610b32943955f9a5060fbd',1,'mlx::core::array::operator=(array &&other) &&=delete'],['../classmlx_1_1core_1_1array.html#ad3277ff68f1336aa217f9cbe40181479',1,'mlx::core::array::operator=(array &&other) &=default'],['../classmlx_1_1core_1_1array.html#a5da41aabecf4c8055b7515341bf57147',1,'mlx::core::array::operator=(const array &other) &'],['../structmlx_1_1core_1_1array_1_1_data.html#a68e9417954fe811b5e41e6317a526748',1,'mlx::core::array::Data::operator=()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a3f42a1362b4a513fa89e7b3dcc570a8e',1,'mlx::core::metal::CommandEncoder::operator=()'],['../classmlx_1_1core_1_1metal_1_1_device.html#ad1d6382fd18a46b1906e1b43e0bd2e73',1,'mlx::core::metal::Device::operator=()'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aef97dbbc755940789f99a26164591c45',1,'mlx::core::metal::ResidencySet::operator=()'],['../structmlx_1_1core_1_1_function_exporter.html#a7ec0f53eb2783d5b1953be612e36d5c7',1,'mlx::core::FunctionExporter::operator=()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a957211656a13b4c0d126989a9aba3e25',1,'mlx::core::io::FileWriter::operator=()'],['../classmlx_1_1core_1_1_primitive.html#a6b1be7ea92f3a7bb19875c70259dad6b',1,'mlx::core::Primitive::operator=(const Primitive &other)=delete'],['../classmlx_1_1core_1_1_primitive.html#a50bbddd43e1ba0cf5f127cd7aa756a9e',1,'mlx::core::Primitive::operator=(Primitive &&other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a0a859309a4f192f2679e07f2e4ff4d22',1,'mlx::core::UnaryPrimitive::operator=(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#ab90b2ea80f1d914be03cf44def5db5a5',1,'mlx::core::UnaryPrimitive::operator=(UnaryPrimitive &&other)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ab170dbd2ce34c51e2eeebf5d08e7e2db',1,'mlx::core::scheduler::Scheduler::operator=(const Scheduler &)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a035ea35f4dd8ee985973080f14029379',1,'mlx::core::scheduler::Scheduler::operator=(Scheduler &&)=delete'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#a0f65b0523b8ddd989f338da6cb2860e3',1,'mlx::core::_MLX_BFloat16::operator=(std::vector< bool >::reference x)'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#abb8cd44ee22b17c55333ff2eb4e13a14',1,'mlx::core::_MLX_BFloat16::operator=(const float &x)'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a608a099bf7116ee608dcfd31ea3ade2c',1,'mlx::core::_MLX_Float16::operator=(std::vector< bool >::reference x)'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a35543c3653d477c46350697fb808373d',1,'mlx::core::_MLX_Float16::operator=(const float &x)'],['../structmlx_1_1core_1_1_command_encoder.html#a3f42a1362b4a513fa89e7b3dcc570a8e',1,'mlx::core::CommandEncoder::operator=()']]], - ['operator_3d_3d_30',['operator==',['../backend_2metal_2kernels_2complex_8h.html#abfc19f03616441245dfc7726b278f190',1,'operator==(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a49a13b06a325ed3cca4004b6a0cde065',1,'operator==(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0aa3bfcfab53700488e5f386e6de60d5',1,'operator==(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3936148781ab1c4f33f58d12c116f370',1,'operator==(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae753526b669fba27771089dc809abd66',1,'operator==(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a05a4f197a71d0f16879032f44492bb79',1,'operator==(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae86f5917847b1ec9f313996250f2e0be',1,'operator==(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aab74ec4d33a64b92b908717d500f1ecf',1,'operator==(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac30a2c1fa6f172af903fdeb6a8632606',1,'operator==(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab4e9ad547aa23daa351075e0ecc58fa2',1,'operator==(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa5fa1a8f2b39c3508fe38205469756d1',1,'operator==(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeadc1f36c6bdc219294ce9341d80afa5',1,'operator==(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3ae2091ada1e39e857fbc53c97bdb79f',1,'operator==(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac7b4d295f3c7b1e09964f24f306422da',1,'operator==(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#abcc797f27e87e857b41c1a8d33ee2c78',1,'mlx::steel::operator==()'],['../namespacemlx_1_1core.html#a937503d72b66c661bf3f5fdcd98ef97c',1,'mlx::core::operator==(const Device &lhs, const Device &rhs)'],['../group__ops.html#gaa30cf69f3d22f65615f5e1696dd5703f',1,'mlx::core::operator==(const array &a, const array &b)'],['../group__ops.html#gaf115782d009ac2a547fcca395c9ec797',1,'mlx::core::operator==(T a, const array &b)'],['../group__ops.html#ga3ad3ed7aece2650943a35082dbe3a0a5',1,'mlx::core::operator==(const array &a, T b)'],['../namespacemlx_1_1core.html#ac470f937a379d6356c8f567c97cd7481',1,'mlx::core::operator==(const Stream &lhs, const Stream &rhs)'],['../namespacemlx_1_1core.html#aec63a0472cb943fe39f31e7678555572',1,'mlx::core::operator==(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad05311ca8e2f19ffe5849e963837cec7',1,'mlx::core::operator==(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aaaf591cb2188381e6cbd857132d04eb7',1,'mlx::core::operator==(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7ef33c33509ccccf1ab217500e8b3c1a',1,'mlx::core::operator==(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#abec4200a718b7c5ed80b7abcc4447260',1,'mlx::core::operator==(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad853981b1c5ba69b07d54c7b77055d22',1,'mlx::core::operator==(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a752d6cb4172a9cb91e5da19582329c6d',1,'mlx::core::operator==(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0175beb3de139faa08479a88215b35ea',1,'mlx::core::operator==(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a61da2851cb3beeef28049228346c28b5',1,'mlx::core::operator==(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aa24713cb9e39bacb516c992eb03d2b2b',1,'mlx::core::operator==(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a6d565dd93c46259f9486d9fdf0969589',1,'mlx::core::operator==(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a24e79a82557861de64dad66d36e6ff30',1,'mlx::core::operator==(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#af27d515ac390d62bd852b73ea759a947',1,'mlx::core::operator==(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae3e1e8b7a5410e0edf35f31f74295e2f',1,'mlx::core::operator==(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aaa22230a66b15c3e774d8ce45783a746',1,'mlx::core::operator==(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ae2a0bcdc171d7e9745d33e1d9aac4f8a',1,'mlx::core::operator==(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a331ec62442a8d3eb8ccba7b4de5168d1',1,'mlx::core::operator==(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#acfcaefe0990eb3533e2b11a6f2657492',1,'mlx::core::operator==(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a8d48dbd49cccff07777affb2a412058c',1,'mlx::core::operator==(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a88eae27edd22fa4418776672023cb276',1,'mlx::core::operator==(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a188b363f633ea360407b3f9cf4e1f1a6',1,'mlx::core::operator==(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ae065fe5c42c1a333d7858d19f6434fa9',1,'mlx::core::operator==(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a2f98db199deb6d7a82551fa4afec655a',1,'mlx::core::operator==(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a85f83add412cb320b5cd1c3da6aadbd5',1,'mlx::core::operator==(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7e2cee66c3ca1b56f4f3d7fd1d6e0be1',1,'mlx::core::operator==(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#ad436557da5c7fea71fc58182a876cfe5',1,'mlx::core::operator==(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3e_31',['operator>',['../backend_2metal_2kernels_2complex_8h.html#a032a8d3eec2384c9f03066f7fd945995',1,'operator>(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae394c0a10e47d1d047854a888402eb57',1,'operator>(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab9cd098786d2f4c855c42e4a6f30ab3e',1,'operator>(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a55600f3b9859e2891e0e0b5690867b72',1,'operator>(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afd7cdb8ed2a9820efe9cf322c06f188c',1,'operator>(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31bbdbe0b62b90a4d6ea4bb0a7db586b',1,'operator>(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a68125e66f74eaffe5ea9267638ce870d',1,'operator>(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac89eb6b29edad8cca63727ab97171c29',1,'operator>(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a74e477567c9477c2cf0684f81ef4498f',1,'operator>(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2d37130b6fd79b425f5ba92b65e36bed',1,'operator>(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a41d55d167e9dc63bf29d15e0ff004869',1,'operator>(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa95f9ebfdab3c5f524775651362ce914',1,'operator>(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2826bd301bb5393473ccd363f2052c0d',1,'operator>(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a62a512d0edd894759c69f724b970fbdb',1,'operator>(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a7512eadda6160e4c9d9e6aa4049fac20',1,'mlx::steel::operator>()'],['../group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb',1,'mlx::core::operator>(const array &a, const array &b)'],['../group__ops.html#ga32e106e794e2c32e4e7decee2df2477f',1,'mlx::core::operator>(T a, const array &b)'],['../group__ops.html#ga96552b90e89923c5d2064cc427775ec5',1,'mlx::core::operator>(const array &a, T b)'],['../namespacemlx_1_1core.html#aedc4e9df4bf71c0ac34fcfae60cdf550',1,'mlx::core::operator>(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a14c188303d09b97867bcfd34519aa4a6',1,'mlx::core::operator>(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ac97736fadafa7efa201624d0e1128ee8',1,'mlx::core::operator>(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3c41a304126bc225bdc68062d1eb6e7e',1,'mlx::core::operator>(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ab594f3ae1ee13227fae940fef0d00cb9',1,'mlx::core::operator>(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a01dabc077a872c115a9a9ccd95f1acec',1,'mlx::core::operator>(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#adabbd8768d216873617768249473a5c7',1,'mlx::core::operator>(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#adae1b14669d27ce1fe0c214771c07b77',1,'mlx::core::operator>(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ab03a22961d99fa12d3e74b3116e94e8f',1,'mlx::core::operator>(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a42011a27a3d23a60be5be44ee7cac87c',1,'mlx::core::operator>(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a50f6a94bb36d89cf28817aff88ab89c8',1,'mlx::core::operator>(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac173de50ee57b1b066d49363ba978c53',1,'mlx::core::operator>(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#ab09f1b4879aa3190c2f66c9bd1224021',1,'mlx::core::operator>(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a91eb6ca854217424129a55ae95a123b5',1,'mlx::core::operator>(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a58d5795d8312599d101ae16f194e4a2a',1,'mlx::core::operator>(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aafa3bbeda78610c4285f3e57042268f3',1,'mlx::core::operator>(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a8a928d76a6fbf3d336296401e14617a4',1,'mlx::core::operator>(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ade2f9222fd433cd4d673c6182f256235',1,'mlx::core::operator>(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ae24c337810c841ff23e327efde7045e1',1,'mlx::core::operator>(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acf401ede354fcc998b13ea6442994d7e',1,'mlx::core::operator>(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a2bb28a9a0894a73ae1b27e7f4da0841a',1,'mlx::core::operator>(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a09d631e8a85fd7ae72e1a868b8f9b9cb',1,'mlx::core::operator>(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a49421ea65b5a98df080d75b1636b2157',1,'mlx::core::operator>(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a692ce931b660415e17f92d18a8e0d446',1,'mlx::core::operator>(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a579bb87b3ede5663d7cd68c7c0f6fb9e',1,'mlx::core::operator>(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af810587a17e692f4eec256d3c3cd27de',1,'mlx::core::operator>(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a50f4177d3ca03a95fc2614e100c7391d',1,'mlx::core::operator>(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3e_3d_32',['operator>=',['../backend_2metal_2kernels_2complex_8h.html#aafbd686c180398c98b33d7643f893a46',1,'operator>=(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a430dd11fbf4c6f39bc1506ab43b2341f',1,'operator>=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a64f6787a96386246f83a8981d274150e',1,'operator>=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1a788f82212afad30e4c2ee40f1c313c',1,'operator>=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae88617c4a012c5dc12781a349a28c886',1,'operator>=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a467a88531150a4d9d30fce07c49c126e',1,'operator>=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9e21c5ea9dd724dc2ca8c54ad908f09c',1,'operator>=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2f6286d222e2176bcbdc824c5d598100',1,'operator>=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abec53064aa96265385ecc57de5fbc74c',1,'operator>=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac766839f8f9e4863e8e18418c342c875',1,'operator>=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2807fa6862b0f9689c81199b1e695ed8',1,'operator>=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee3ae0d0d1f941463b06eca0bf041b2b',1,'operator>=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a523eda93c809733368e2b45382d2add6',1,'operator>=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1f4e90909ac1c7280f4c7d1977c55fb7',1,'operator>=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa3c95c60cf69603705bb4636de547bcb',1,'mlx::steel::operator>=()'],['../group__ops.html#ga3a41895f25ed083a36994d95fa102546',1,'mlx::core::operator>=(const array &a, const array &b)'],['../group__ops.html#gaf509f2cb3b18963232f20d6c3bd229b2',1,'mlx::core::operator>=(T a, const array &b)'],['../group__ops.html#gafa0eb25d5978674bfc9e59d4145ec590',1,'mlx::core::operator>=(const array &a, T b)'],['../namespacemlx_1_1core.html#a8494764f5c686743ede66dc76d85d955',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a019df48807b506d9995856684bf7797a',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a96ab6405430efb887cdb5c828cb67d6e',1,'mlx::core::operator>=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac18be72269b1bcfb0249cc00a0600681',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aeb879815228efbd2c8f80986e1c8d41f',1,'mlx::core::operator>=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0051156f6a568f58cd54850f746fb507',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ae93556906e115625ed1b62d36cf21b70',1,'mlx::core::operator>=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab81ad16e3be591dfc9e42ac3c19b055f',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6cfe9b03e7c5f1eb9374208a552c3cc9',1,'mlx::core::operator>=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2f5add83812fb137dd9226c6c01e45d5',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ad1014a836e7ce9301de8588eef1e89ee',1,'mlx::core::operator>=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a17791561434dc995de9f268d145c0ed1',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a3755925b24a903045937464be117de2f',1,'mlx::core::operator>=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a6262aeb513d27fc8313293b261e72abb',1,'mlx::core::operator>=(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a6feb4b3ea511b0eda4d1ec9725f3fb4c',1,'mlx::core::operator>=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a03b3f7fcb755ec075985ab26336926f0',1,'mlx::core::operator>=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aecfbf5ef4872ae447eb4a374e4db28e4',1,'mlx::core::operator>=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae4690f349b2483f5d1a4b75aba67399f',1,'mlx::core::operator>=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a667e95146dd5199e67bcb121b984b1f0',1,'mlx::core::operator>=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a3375f1562f148bdc07451f2b6e54e6df',1,'mlx::core::operator>=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ae83df12368cb07ccb1c10c1117ff3922',1,'mlx::core::operator>=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad41251938cf852b5560c1180944ebb49',1,'mlx::core::operator>=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a4ddb5ef0b88929086f9b09729fda0dde',1,'mlx::core::operator>=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a0908a61ab261aff726922b33fa6ed159',1,'mlx::core::operator>=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a0fdadf87edd8a0a57c63953fb0ebe053',1,'mlx::core::operator>=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a47c82778e43032c0bbf5d59407e81dc9',1,'mlx::core::operator>=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a14e6c43b924eacca1b2dac1d5d00ca2b',1,'mlx::core::operator>=(uint64_t lhs, _MLX_Float16 rhs)']]], - ['operator_3e_3e_33',['operator>>',['../group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a',1,'mlx::core']]], - ['operator_5b_5d_34',['operator[]',['../classpocketfft_1_1detail_1_1arr.html#aea0bd899b19e03f54dfd6c188727061a',1,'pocketfft::detail::arr::operator[](size_t idx)'],['../classpocketfft_1_1detail_1_1arr.html#a99c54f96bc79c7cdd8925c1663462842',1,'pocketfft::detail::arr::operator[](size_t idx) const'],['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a71b02f67c47b24adb296eafd2c7a3598',1,'pocketfft::detail::sincos_2pibyn::operator[]()'],['../classpocketfft_1_1detail_1_1cndarr.html#ae4852d1fe936a5d61832b507816c7054',1,'pocketfft::detail::cndarr::operator[]()'],['../classpocketfft_1_1detail_1_1ndarr.html#a2b2c4e205e8b5c32c9fe55dfd7b8c8d8',1,'pocketfft::detail::ndarr::operator[]()']]], - ['operator_5e_35',['operator^',['../group__ops.html#gac3a6fe18694e84b3d63458e9553ac181',1,'mlx::core::operator^(const array &a, const array &b)'],['../namespacemlx_1_1core.html#ae36ea40b8477bfa12d41aae8245225c9',1,'mlx::core::operator^(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a03fc96696f5c6d9411841889d05f4670',1,'mlx::core::operator^(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a55130edf926366db0d6207989e609b7c',1,'mlx::core::operator^(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0b75198f364d742a1c25dd13e398f2c2',1,'mlx::core::operator^(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7f205f1b10b23180a23bf2be4bb726b1',1,'mlx::core::operator^(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a9edfe65f3c6da583c7b109290ec94b22',1,'mlx::core::operator^(uint16_t lhs, _MLX_Float16 rhs)']]], - ['operator_5e_3d_36',['operator^=',['../namespacemlx_1_1core.html#a97cb7d3eac404a442e84656cefe7cfb4',1,'mlx::core::operator^=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abcfd2d9615c96561fd44dfb9c341cf8e',1,'mlx::core::operator^=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ae78083d766b9cf6f87cded341bbcd63e',1,'mlx::core::operator^=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acf36c10779fbf1efbe1e6a7fd41176cd',1,'mlx::core::operator^=(_MLX_Float16 &lhs, uint16_t rhs)']]], - ['operator_7c_37',['operator|',['../group__ops.html#ga52392a2a98f09a80da8d338c4908bd02',1,'mlx::core::operator|(const array &a, const array &b)'],['../namespacemlx_1_1core.html#af84ed854132c1514dca5a524fdb7ed05',1,'mlx::core::operator|(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7423aac70f9f2e3fb6a5c9a3fc96f703',1,'mlx::core::operator|(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a19805f505cb7ac72bfab66c339ea7900',1,'mlx::core::operator|(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2d933573edf4ed305fddd8a0caef1ee8',1,'mlx::core::operator|(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afab3d4eb1b36a276922879ce6e44b7f5',1,'mlx::core::operator|(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ab132729fa6912d22a8e402057eb4ba12',1,'mlx::core::operator|(uint16_t lhs, _MLX_Float16 rhs)']]], - ['operator_7c_3d_38',['operator|=',['../namespacemlx_1_1core.html#a8e1d21375ae4b89b3cbea3a46d262abd',1,'mlx::core::operator|=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a28d6c2f89e73b7b874dd1f67f853a96f',1,'mlx::core::operator|=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a2d8470b69cbbeefece08d3ffd46c0082',1,'mlx::core::operator|=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a359c6257097a304c00d41d64296ef4c9',1,'mlx::core::operator|=(_MLX_Float16 &lhs, uint16_t rhs)']]], - ['operator_7c_7c_39',['operator||',['../namespacemlx_1_1steel.html#a1bb3ac5061a04e407fc4cdcc9f6ea03f',1,'mlx::steel::operator||()'],['../group__ops.html#ga27af56a98270d4d76d139f0f9171b83a',1,'mlx::core::operator||()']]], - ['out_5fof_5fbounds_40',['out_of_bounds',['../struct_read_writer.html#a08e10626fbc789b6dff9172fd6c36f7c',1,'ReadWriter::out_of_bounds() const'],['../struct_read_writer.html#a6f946aea5452109dca7fc70ed39c6efe',1,'ReadWriter::out_of_bounds() const'],['../struct_read_writer.html#a8f40d7f343d32134fe27a694abfde6bf',1,'ReadWriter::out_of_bounds() const']]], - ['outer_41',['outer',['../group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d',1,'mlx::core']]], - ['output_5fshape_42',['output_shape',['../classmlx_1_1core_1_1_broadcast_axes.html#aaa495110c16fbbc642fbb224ef8dfae6',1,'mlx::core::BroadcastAxes::output_shape()'],['../classmlx_1_1core_1_1_broadcast.html#a00c39c113fe3e698771e2e6b595c32cd',1,'mlx::core::Broadcast::output_shape()'],['../classmlx_1_1core_1_1_expand_dims.html#a3814ad4697eccb75fdb9275017a3fd67',1,'mlx::core::ExpandDims::output_shape()'],['../classmlx_1_1core_1_1_flatten.html#a2f8e1defb9c33af2dec29ff8697132aa',1,'mlx::core::Flatten::output_shape()'],['../classmlx_1_1core_1_1_reshape.html#aa15020d7d844d714d42bc60b44aeefc1',1,'mlx::core::Reshape::output_shape()'],['../classmlx_1_1core_1_1_squeeze.html#aadf1d3b85839390a2ec560603aeed04a',1,'mlx::core::Squeeze::output_shape()'],['../classmlx_1_1core_1_1_unflatten.html#a4c760c8fe981fd2ac17a31ff9faff10a',1,'mlx::core::Unflatten::output_shape()']]], - ['output_5fshapes_43',['output_shapes',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a50934862ccdb16a3dcce6626c5727080',1,'mlx::core::fast::AffineQuantize::output_shapes()'],['../classmlx_1_1core_1_1_primitive.html#aa5b443d71db1c7ed31a5ae6e31b7fe29',1,'mlx::core::Primitive::output_shapes()'],['../classmlx_1_1core_1_1_abs.html#ac9d55481e5490423e4aaf02b95cafc75',1,'mlx::core::Abs::output_shapes()'],['../classmlx_1_1core_1_1_add.html#a50877893083fd78b31aa25152f750418',1,'mlx::core::Add::output_shapes()'],['../classmlx_1_1core_1_1_arange.html#a447083a1403d3d42a7ad9c307a666946',1,'mlx::core::Arange::output_shapes()'],['../classmlx_1_1core_1_1_arc_cos.html#a67a5025f8d7e5bac22888ad4bf813679',1,'mlx::core::ArcCos::output_shapes()'],['../classmlx_1_1core_1_1_arc_cosh.html#a3ab82e9f0452faea735338abccb5f0ac',1,'mlx::core::ArcCosh::output_shapes()'],['../classmlx_1_1core_1_1_arc_sin.html#a0217b9a4e18196ed65ba96b4ad096ecd',1,'mlx::core::ArcSin::output_shapes()'],['../classmlx_1_1core_1_1_arc_sinh.html#a2f668f230d93c7b90e62200a0b7cb6f6',1,'mlx::core::ArcSinh::output_shapes()'],['../classmlx_1_1core_1_1_arc_tan.html#a2ebabfd1c2963199df0d7610b7ddf422',1,'mlx::core::ArcTan::output_shapes()'],['../classmlx_1_1core_1_1_arc_tan2.html#acb8e5cf85c4bc58f909ce2e8b83c3619',1,'mlx::core::ArcTan2::output_shapes()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6806f04142d850f107a18a71900759c6',1,'mlx::core::ArcTanh::output_shapes()'],['../classmlx_1_1core_1_1_arg_partition.html#a587ce69b0639683ba646652f887d0239',1,'mlx::core::ArgPartition::output_shapes()'],['../classmlx_1_1core_1_1_arg_reduce.html#a81a70885480c1d436329025091b2fa4c',1,'mlx::core::ArgReduce::output_shapes()'],['../classmlx_1_1core_1_1_arg_sort.html#a219ce04a811397a900c3235d8e6aef5c',1,'mlx::core::ArgSort::output_shapes()'],['../classmlx_1_1core_1_1_as_type.html#a3975b31cfd86d6eb33dc73554b357b88',1,'mlx::core::AsType::output_shapes()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a49c9d2688d3cca8abf5698a250d57d56',1,'mlx::core::BitwiseBinary::output_shapes()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a42c4385e65851d58e4411a4afe73f58e',1,'mlx::core::BroadcastAxes::output_shapes()'],['../classmlx_1_1core_1_1_broadcast.html#adef65b1ec75efbe43e5574ec81b7c0ac',1,'mlx::core::Broadcast::output_shapes()'],['../classmlx_1_1core_1_1_ceil.html#a3bf7db5178ed26e23d9ba360ba34ab85',1,'mlx::core::Ceil::output_shapes()'],['../classmlx_1_1core_1_1_compiled.html#a15cb081590ee024ba11476494581a4d4',1,'mlx::core::Compiled::output_shapes()'],['../classmlx_1_1core_1_1_concatenate.html#af8415a2fe28804a1437d0876ba15615f',1,'mlx::core::Concatenate::output_shapes()'],['../classmlx_1_1core_1_1_conjugate.html#afd68332463d12e69c47388f6b81ae96c',1,'mlx::core::Conjugate::output_shapes()'],['../classmlx_1_1core_1_1_contiguous.html#afff58fbf61f0c26b3606208dd2fa2072',1,'mlx::core::Contiguous::output_shapes()'],['../classmlx_1_1core_1_1_copy.html#a6bbe5fd9ce3cb5a39853b316106d2674',1,'mlx::core::Copy::output_shapes()'],['../classmlx_1_1core_1_1_cos.html#a923312e71c5a003a38b37ab67ec82580',1,'mlx::core::Cos::output_shapes()'],['../classmlx_1_1core_1_1_cosh.html#adf58c7e24b5059e66007132bc16dfe49',1,'mlx::core::Cosh::output_shapes()'],['../classmlx_1_1core_1_1_divide.html#a9563d9ee243204cfdaac6aca34853cd7',1,'mlx::core::Divide::output_shapes()'],['../classmlx_1_1core_1_1_div_mod.html#a1b7f104346cb5423ac15371b45c7ef86',1,'mlx::core::DivMod::output_shapes()'],['../classmlx_1_1core_1_1_select.html#a10e837a391542b364186288a87e11513',1,'mlx::core::Select::output_shapes()'],['../classmlx_1_1core_1_1_remainder.html#ab4de49818d1fdea8cdfef502f519b255',1,'mlx::core::Remainder::output_shapes()'],['../classmlx_1_1core_1_1_equal.html#ae714c2b0641fc9c339a2f8483bb4e257',1,'mlx::core::Equal::output_shapes()'],['../classmlx_1_1core_1_1_erf.html#ace70b96c48419e29243982ed697f6411',1,'mlx::core::Erf::output_shapes()'],['../classmlx_1_1core_1_1_erf_inv.html#a067cac7a7244b4dae6629c7e4466589f',1,'mlx::core::ErfInv::output_shapes()'],['../classmlx_1_1core_1_1_exp.html#aef2b3c24dba3ca3a63a210d3bd8e39b6',1,'mlx::core::Exp::output_shapes()'],['../classmlx_1_1core_1_1_expm1.html#ae78f03a204687f16164ed702cfc0d5cc',1,'mlx::core::Expm1::output_shapes()'],['../classmlx_1_1core_1_1_expand_dims.html#af64bd4bc2cc5f5c58869f34cd974bb3c',1,'mlx::core::ExpandDims::output_shapes()'],['../classmlx_1_1core_1_1_flatten.html#a5069a73ba1e7b52b7b051f692db6d0d2',1,'mlx::core::Flatten::output_shapes()'],['../classmlx_1_1core_1_1_floor.html#a0a62dee6df6a82fcd955bf7670be2cd5',1,'mlx::core::Floor::output_shapes()'],['../classmlx_1_1core_1_1_gather.html#a53d89a6c4ebb634bc208bd85aa2fcda1',1,'mlx::core::Gather::output_shapes()'],['../classmlx_1_1core_1_1_greater.html#af798a7cd704a2a9a8b3ecb6ef49583b0',1,'mlx::core::Greater::output_shapes()'],['../classmlx_1_1core_1_1_greater_equal.html#a1a77c18d89ee227171ff38efef6cacf6',1,'mlx::core::GreaterEqual::output_shapes()'],['../classmlx_1_1core_1_1_hadamard.html#aa709166de3c493308689769579d665e8',1,'mlx::core::Hadamard::output_shapes()'],['../classmlx_1_1core_1_1_imag.html#ad4f847483ba07d20aba5b927c2689be8',1,'mlx::core::Imag::output_shapes()'],['../classmlx_1_1core_1_1_less.html#ad7604a75b79260d263ac0c7d959cadd5',1,'mlx::core::Less::output_shapes()'],['../classmlx_1_1core_1_1_less_equal.html#a5598c700e881673098928e47b4da9ff8',1,'mlx::core::LessEqual::output_shapes()'],['../classmlx_1_1core_1_1_log.html#ab2cae6889352ca0674f6463f8f52d77d',1,'mlx::core::Log::output_shapes()'],['../classmlx_1_1core_1_1_log1p.html#a73a02ddf0f125fff83462d97146a0a08',1,'mlx::core::Log1p::output_shapes()'],['../classmlx_1_1core_1_1_logical_not.html#ad3889969521c6a040aa2f26caee219b7',1,'mlx::core::LogicalNot::output_shapes()'],['../classmlx_1_1core_1_1_logical_and.html#a266f1eaced19b8b11e273de9219cf9ed',1,'mlx::core::LogicalAnd::output_shapes()'],['../classmlx_1_1core_1_1_logical_or.html#a931b98fca3e19085af9fa97a43db8ced',1,'mlx::core::LogicalOr::output_shapes()'],['../classmlx_1_1core_1_1_log_add_exp.html#a234f8c8ea5f5bf2fb7e371588fea98b9',1,'mlx::core::LogAddExp::output_shapes()'],['../classmlx_1_1core_1_1_matmul.html#abfabe69f428f7f125bf5665713a0eb5c',1,'mlx::core::Matmul::output_shapes()'],['../classmlx_1_1core_1_1_maximum.html#a888a69fb68726c3c18973f3ea38cfd2b',1,'mlx::core::Maximum::output_shapes()'],['../classmlx_1_1core_1_1_minimum.html#af921b5202ebf9716972bcf0e3056742a',1,'mlx::core::Minimum::output_shapes()'],['../classmlx_1_1core_1_1_multiply.html#adfd4c7f89660b42ab58e088b1ae19435',1,'mlx::core::Multiply::output_shapes()'],['../classmlx_1_1core_1_1_negative.html#a606fb13a48d10c88707f1a2c41bee9e8',1,'mlx::core::Negative::output_shapes()'],['../classmlx_1_1core_1_1_not_equal.html#ad1e8a577dc103d96f1ab65bf3b389d35',1,'mlx::core::NotEqual::output_shapes()'],['../classmlx_1_1core_1_1_number_of_elements.html#a6cdf307348ba22b3dc8f90f1fb1e0757',1,'mlx::core::NumberOfElements::output_shapes()'],['../classmlx_1_1core_1_1_partition.html#a5e62aa0109e53fb4acb861ef39787b4a',1,'mlx::core::Partition::output_shapes()'],['../classmlx_1_1core_1_1_power.html#af23ed795bdcdc4c3f91f0d4c1bb1d928',1,'mlx::core::Power::output_shapes()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a7d57a31d41c58e1bd88ffe9c6b0dbf52',1,'mlx::core::QuantizedMatmul::output_shapes()'],['../classmlx_1_1core_1_1_real.html#a75999bd0b97d97a5675b9cdbab27dcff',1,'mlx::core::Real::output_shapes()'],['../classmlx_1_1core_1_1_reshape.html#aed3a83606d6917b2c344607101a2c43d',1,'mlx::core::Reshape::output_shapes()'],['../classmlx_1_1core_1_1_reduce.html#aaf3da1c98cdf530803118b382c5f58bc',1,'mlx::core::Reduce::output_shapes()'],['../classmlx_1_1core_1_1_round.html#a61821399e177e142723fc986e437d459',1,'mlx::core::Round::output_shapes()'],['../classmlx_1_1core_1_1_sigmoid.html#aff024a3309584724c9842f172a4e440b',1,'mlx::core::Sigmoid::output_shapes()'],['../classmlx_1_1core_1_1_sign.html#a2260f2e8e081010192eb8a6f90acde6e',1,'mlx::core::Sign::output_shapes()'],['../classmlx_1_1core_1_1_sin.html#abdd433ecbb54898161b43aa9e14ec7f1',1,'mlx::core::Sin::output_shapes()'],['../classmlx_1_1core_1_1_sinh.html#ae04d8f6175c691a8f0d2a9fdd15af0ad',1,'mlx::core::Sinh::output_shapes()'],['../classmlx_1_1core_1_1_slice_update.html#abb6376f13c4269bd9e739e131893da53',1,'mlx::core::SliceUpdate::output_shapes()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a920dc4d1ee4976065e6d91fe3ecfbbf3',1,'mlx::core::DynamicSlice::output_shapes()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a804c03c745fc563e209a7bfb3d425a91',1,'mlx::core::DynamicSliceUpdate::output_shapes()'],['../classmlx_1_1core_1_1_softmax.html#a1a798a4dcd62486362d4b58582357490',1,'mlx::core::Softmax::output_shapes()'],['../classmlx_1_1core_1_1_sort.html#acc0a3f078b3f4c83e6e1137cb81ee62c',1,'mlx::core::Sort::output_shapes()'],['../classmlx_1_1core_1_1_square.html#a0513541766bb997ed166643fe95a6d38',1,'mlx::core::Square::output_shapes()'],['../classmlx_1_1core_1_1_sqrt.html#ae45215d61e2e99749d9a0bae291edd45',1,'mlx::core::Sqrt::output_shapes()'],['../classmlx_1_1core_1_1_stop_gradient.html#a8af7641d478505d1dc39c75ba7d5a3cf',1,'mlx::core::StopGradient::output_shapes()'],['../classmlx_1_1core_1_1_subtract.html#aaaff4872bde70ad40cf90e6131ea0489',1,'mlx::core::Subtract::output_shapes()'],['../classmlx_1_1core_1_1_squeeze.html#a839d9d72ac0a19e1146b5b470292a174',1,'mlx::core::Squeeze::output_shapes()'],['../classmlx_1_1core_1_1_tan.html#a9e4bba311bb24617dbb5ca591bc2868e',1,'mlx::core::Tan::output_shapes()'],['../classmlx_1_1core_1_1_tanh.html#a8873286b69b805486fa83c4806843f3d',1,'mlx::core::Tanh::output_shapes()'],['../classmlx_1_1core_1_1_unflatten.html#a068cf053b5b0612fafd4a2d53d42f9fa',1,'mlx::core::Unflatten::output_shapes()'],['../classmlx_1_1core_1_1_transpose.html#ac9328f43900bedec555909d09202ccd7',1,'mlx::core::Transpose::output_shapes()'],['../classmlx_1_1core_1_1_eigh.html#a9892f5b72dec19a5a2f7af5efcf2a952',1,'mlx::core::Eigh::output_shapes()']]], - ['outputs_44',['outputs',['../classmlx_1_1core_1_1array.html#a2c186fd527f984f0589d4183b4976289',1,'mlx::core::array::outputs()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aefa48740fdee884f02e2d379bca4e78f',1,'mlx::core::metal::CommandEncoder::outputs()'],['../structmlx_1_1core_1_1_command_encoder.html#aefa48740fdee884f02e2d379bca4e78f',1,'mlx::core::CommandEncoder::outputs()']]], - ['overwrite_5fdescriptor_45',['overwrite_descriptor',['../classmlx_1_1core_1_1array.html#a95e6b156c8e05439f076b85c05079387',1,'mlx::core::array']]] + ['operator_20simd_3c_20float_2c_20n_20_3e_8',['operator Simd< float, N >',['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a98affc184d83627d8654e3530ab52d75',1,'mlx::core::simd::Simd< float16_t, N >']]], + ['operator_20simd_3c_20int16_5ft_2c_20n_20_3e_9',['operator Simd< int16_t, N >',['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a823af21442333505114fd3fdac9f24de',1,'mlx::core::simd::Simd< float16_t, N >']]], + ['operator_20t_10',['operator T',['../structcomplex64__t.html#a70e9b16031eeaff3baa601f400023fcd',1,'complex64_t::operator T() const thread'],['../structcomplex64__t.html#a4f3beea7ab6001189b782a74d1746b67',1,'complex64_t::operator T() const threadgroup'],['../structcomplex64__t.html#a9f4f7eca89ffe6c8d126a4145df6d9f2',1,'complex64_t::operator T() const device'],['../structcomplex64__t.html#ac33e2e5263fec76a4fb4418c6e1d8d14',1,'complex64_t::operator T() const constant'],['../struct___m_l_x___b_float16.html#aa7dfefdf0d15e102d2b8258c9ab01836',1,'_MLX_BFloat16::operator T() const thread'],['../struct___m_l_x___b_float16.html#a2546a8afa77e14ed5b3c5da79a281260',1,'_MLX_BFloat16::operator T() const threadgroup'],['../struct___m_l_x___b_float16.html#a1d523f87740fcb852db6ab57896c245a',1,'_MLX_BFloat16::operator T() const device'],['../struct___m_l_x___b_float16.html#a95acd29283024d7093a0bc58c9468a0a',1,'_MLX_BFloat16::operator T() const constant']]], + ['operator_20val_11',['operator Val',['../structmlx_1_1core_1_1_dtype.html#a3b3bc059be5836476da3cb88a4f5e9fd',1,'mlx::core::Dtype']]], + ['operator_20value_5ftype_12',['operator value_type',['../structmlx_1_1steel_1_1integral__constant.html#a0c11203bed44a6a2c387b365134dcd64',1,'mlx::steel::integral_constant']]], + ['operator_21_13',['operator!',['../namespacemlx_1_1core_1_1simd.html#a745e05627c77152ec13d8d90c19cc9bf',1,'mlx::core::simd::operator!(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#afaa6ce61de4d80a4b7e9b2ab7454fff4',1,'mlx::core::simd::operator!(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#aadb0ed44c238d8d643c056298d5b20ca',1,'mlx::core::simd::operator!(Simd< float16_t, N > v)']]], + ['operator_21_3d_14',['operator!=',['../namespacemlx_1_1core_1_1simd.html#a4971bfe7f9f9319f859b3040c18f39ca',1,'mlx::core::simd::operator!=(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#a5c49123bf2647a5ca4f0579a54f3e53a',1,'mlx::core::simd::operator!=(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a125cbaa7c5dd0931b0abd11003ab584a',1,'mlx::core::simd::operator!=(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a757838b9d56e132e797a381d3bb0dc86',1,'mlx::core::simd::operator!=(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#ae8ca6615d51866d876b5efb3425600ed',1,'mlx::core::simd::operator!=(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a7f1cebaff9cb88df59b5ec7557b5d167',1,'mlx::core::simd::operator!=(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#a6cce6db46c391a5d06dcb262e21b81fc',1,'mlx::core::simd::operator!=(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a3699410174385f5e597cfccad57fc736',1,'mlx::core::simd::operator!=(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#abc6a26b6e28d3d532fc356f96c97df1d',1,'mlx::core::simd::operator!=(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afc6e4fc5589bbf30f978f34868dd4e55',1,'operator!=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6baa722c22d66c7510786bb275cb8cc2',1,'operator!=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa8d9f01582a0a9f01a666d110c74db2a',1,'operator!=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa504a474ab6e00ebe2b1b7ed2f7d1ffb',1,'operator!=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abf5f3040227f021a5b84cf2eda248b2f',1,'operator!=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a347c9bbf816bad2e9e5e91aa448f8b65',1,'operator!=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a33ea086b561c652f25833a5e1ded34dd',1,'operator!=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2bbdcece13148826d3fe33af727bb79b',1,'operator!=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeb1efa47c5f22cc0b35d49ccce73c406',1,'operator!=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa6b99cde403405df1865c989e4ce845a',1,'operator!=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a204d13a881ae8d337f6efbb98673790c',1,'operator!=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3602117b4c61d5cd4fd72fb8e5f68bd6',1,'operator!=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2721c088adfc9d73cde442d6badd2a6c',1,'operator!=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa4364eda56525cf7576ff00e550175e6',1,'mlx::steel::operator!=()'],['../namespacemlx_1_1core.html#a94d00a1b7f8a4717ab3f26f45e4da655',1,'mlx::core::operator!=(const Device &lhs, const Device &rhs)'],['../group__ops.html#ga0ac483d85f23252ca8757e9926d5a3c5',1,'mlx::core::operator!=(const array &a, const array &b)'],['../group__ops.html#ga3fecba9f3cb9a19afd8ca492cf509ce0',1,'mlx::core::operator!=(T a, const array &b)'],['../group__ops.html#gaebbf1cfde388c7480159a03c92c9a385',1,'mlx::core::operator!=(const array &a, T b)'],['../namespacemlx_1_1core.html#a164f109bc19c927b2b3bcc47a5021419',1,'mlx::core::operator!=(const Stream &lhs, const Stream &rhs)'],['../namespacemlx_1_1core.html#ad2f9e1c230ec35d5c406dd616e8f4dea',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af5899b4d5644682cb0ac2a488f630d55',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a72ac8edd190601d7a46782582cedecd8',1,'mlx::core::operator!=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a8084162ba2dd3f9b89195d2bebc3fbb0',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a514263e63f6825b490203ca586864687',1,'mlx::core::operator!=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a1c482bb3d9f9d4c62dee5865892c1f96',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a0030fe7ad09837c670cdfb7d51279519',1,'mlx::core::operator!=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ade3791bc723b8f10fbab22eadb0f705a',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ad78c664f242cd36247c13868547e3dd4',1,'mlx::core::operator!=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab0743a1a1dcb92d40f41ca42d36f242c',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ae7a0f810e546a166c7d05849b5d41f30',1,'mlx::core::operator!=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a676a40637a563f013c725d24fa33fdc8',1,'mlx::core::operator!=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a9fcb662b1561e4136bac0106cfb63b6c',1,'mlx::core::operator!=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abcca7fd43590c4347e0f5df8f134030c',1,'mlx::core::operator!=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af3ede3688a2e3b3ba8cb2da180ffe151',1,'mlx::core::operator!=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a54f48469fabd1414bef5097bcded0002',1,'mlx::core::operator!=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af8c648e892cbc6973de535aa17dc2cfe',1,'mlx::core::operator!=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#abc855e1c0584b64d7d995e33211361ab',1,'mlx::core::operator!=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad3684d660d18a54505c759ab286bd936',1,'mlx::core::operator!=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a8afdda14b14262ab5ce0a00c7745d7e8',1,'mlx::core::operator!=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7ccc479be236f2bf3f7725729c5ba201',1,'mlx::core::operator!=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a26a721b8111fce3a1dec9bf724034cd4',1,'mlx::core::operator!=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad5f8c221a53a89e8095aa39fd1f61867',1,'mlx::core::operator!=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a017b52ecf30b33da4aa8da35ccc43220',1,'mlx::core::operator!=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a43c10ca5fb05ee7d0ee63ba56f8a08a3',1,'mlx::core::operator!=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a81284b6ac737f91a8d1ffbbbbf938fe5',1,'mlx::core::operator!=(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_25_15',['operator%',['../backend_2metal_2kernels_2complex_8h.html#aaf53122a07c8eca858b5a8e38ae280e0',1,'operator%(): complex.h'],['../group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde',1,'mlx::core::operator%(const array &a, const array &b)'],['../group__ops.html#ga50817666f0b82afcbf4a123486af9908',1,'mlx::core::operator%(T a, const array &b)'],['../group__ops.html#ga46c01daa07433542a477d216e13a8480',1,'mlx::core::operator%(const array &a, T b)'],['../namespacemlx_1_1core.html#a8723d145dd49021bfcb8e6c99e1c91a5',1,'mlx::core::operator%(complex64_t a, complex64_t b)']]], + ['operator_26_16',['operator&',['../namespacemlx_1_1core_1_1simd.html#a0727c897502944659b3e32b3cde9ee9b',1,'mlx::core::simd::operator&(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a832bbc02ed5589e70106c831c04500f1',1,'mlx::core::simd::operator&(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ac790406f4cf51cbc40d750d377dd741b',1,'mlx::core::simd::operator&(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a3c42ac1dc74f6c0bb934dfa45986875b',1,'mlx::core::simd::operator&(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value &b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a8beb567724ab9735b616afb777b93abd',1,'mlx::core::simd::operator&(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a &b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a3a060a225b6ead483ca93247c9ad8e4d',1,'mlx::core::simd::operator&(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value &b), 1 >'],['../group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff',1,'mlx::core::operator&(const array &a, const array &b)'],['../namespacemlx_1_1core.html#a9ee95f97bbd69262d99d7bea3bf77631',1,'mlx::core::operator&(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0fefc3ae4f1350ebe05ec6098fd6bae3',1,'mlx::core::operator&(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a1e4cb758ccfe5c267baed9aeb0044834',1,'mlx::core::operator&(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab9d0f9910070231695d61de08cadb930',1,'mlx::core::operator&(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a889d401f425db79d1868aa3beea4829b',1,'mlx::core::operator&(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a76dcd1fa3c68b386bc1d1d899a68a120',1,'mlx::core::operator&(uint16_t lhs, _MLX_Float16 rhs)']]], + ['operator_26_26_17',['operator&&',['../namespacemlx_1_1core_1_1simd.html#a85c23e7ed6fe0ec6dfe4c61f7412a362',1,'mlx::core::simd::operator&&(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a8a2c8aea209236b06c594c8451017ecb',1,'mlx::core::simd::operator&&(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a070f1fa094cf2da5ab7d6baecbbf4f56',1,'mlx::core::simd::operator&&(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a82676bd32059d1172296f8074a841de6',1,'mlx::core::simd::operator&&(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value &&b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#af97917ef704103c6ea1d0e44f22ec0d3',1,'mlx::core::simd::operator&&(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a &&b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a1eca7cf07b2a238307459c28204319fb',1,'mlx::core::simd::operator&&(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value &&b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a2a381e5ec89406074b8d1921304238bb',1,'mlx::core::simd::operator&&(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#af9d5f107ce0c40c3b6a2f176cbb70cd7',1,'mlx::core::simd::operator&&(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#af8f245dfc5154c04c0865a208ab1cfe9',1,'mlx::core::simd::operator&&(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1steel.html#a6353bf11881842e25c46b56f92b7044f',1,'mlx::steel::operator&&()'],['../group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3',1,'mlx::core::operator&&(const array &a, const array &b)']]], + ['operator_26_3d_18',['operator&=',['../namespacemlx_1_1core.html#a60c263ef46e552c3954688869734b513',1,'mlx::core::operator&=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af9670fc8088339669c54c68b3a320e25',1,'mlx::core::operator&=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ad1f96f0a02024f347b4c4431629407fc',1,'mlx::core::operator&=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae0540f16c4e7bd55d0e86a88495e4967',1,'mlx::core::operator&=(_MLX_Float16 &lhs, uint16_t rhs)']]], + ['operator_28_29_19',['operator()',['../structpocketfft_1_1detail_1_1_exec_c2_c.html#a4fd637f1a6d335826789af28ac089ecb',1,'pocketfft::detail::ExecC2C::operator()()'],['../structpocketfft_1_1detail_1_1_exec_hartley.html#a67c98b38d12440781053552b9a33bba1',1,'pocketfft::detail::ExecHartley::operator()()'],['../structpocketfft_1_1detail_1_1_exec_dcst.html#a67f4f56e3574c491695f8cb8a1e983d8',1,'pocketfft::detail::ExecDcst::operator()()'],['../structpocketfft_1_1detail_1_1_exec_r2_r.html#acdba1650962714e6afff51e9ca456970',1,'pocketfft::detail::ExecR2R::operator()()'],['../structmlx_1_1core_1_1_vector_scalar.html#a1af3ff644ce023a7e4f92a7c3634c44f',1,'mlx::core::VectorScalar::operator()()'],['../structmlx_1_1core_1_1_scalar_vector.html#ab174fe55970fb4ee1c6a2b7628a24df1',1,'mlx::core::ScalarVector::operator()()'],['../structmlx_1_1core_1_1_vector_vector.html#a97a0bed419933d7685238a962f2e4215',1,'mlx::core::VectorVector::operator()()'],['../structmlx_1_1core_1_1detail_1_1_add.html#a95cf053f89883d82f31ec53154b430a0',1,'mlx::core::detail::Add::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_add.html#a2d6011c35768b5fcd2bb75747b944353',1,'mlx::core::detail::Add::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html#a01da277adf65232bd67b252a31baedd7',1,'mlx::core::detail::ArcTan2::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html#af0cfd2ea4d541379b9c427fd4054828d',1,'mlx::core::detail::ArcTan2::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_divide.html#a9a3eab9eaf77b5a94ede2db8c7cef9f2',1,'mlx::core::detail::Divide::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_divide.html#a5e0d22e2084c4ca81bec0d457a46c662',1,'mlx::core::detail::Divide::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_multiply.html#a9dda09d0bf0f4153abf37ba894df37d4',1,'mlx::core::detail::Multiply::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_multiply.html#a898b090966b047723513224b8d3b22f1',1,'mlx::core::detail::Multiply::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_subtract.html#a48913052e0a051648b7a69376ec3e3e1',1,'mlx::core::detail::Subtract::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_subtract.html#a72ef05830615a2d5d9662926ed82672a',1,'mlx::core::detail::Subtract::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_logical_and.html#a5fb547e51ea53517deb54d89c76b4860',1,'mlx::core::detail::LogicalAnd::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_logical_and.html#a046536c1f2f9367983f052a213d7b7d8',1,'mlx::core::detail::LogicalAnd::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_logical_or.html#a4701821e656931d808815753ee529bad',1,'mlx::core::detail::LogicalOr::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_logical_or.html#afb134dbab79307d4ba597843c61d0b1a',1,'mlx::core::detail::LogicalOr::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html#a91cff5472e47b13fd9d291b17d2e877b',1,'mlx::core::detail::BitwiseAnd::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html#ae0bed77f95fe2b2f0b594addddd04700',1,'mlx::core::detail::BitwiseAnd::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html#abd39ee9af548b16e3fabe4ae956b6f1c',1,'mlx::core::detail::BitwiseOr::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html#a5ab05734c5000b454975de6647a08d20',1,'mlx::core::detail::BitwiseOr::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html#a8ed25d90a73141938a71ddddfd40b83d',1,'mlx::core::detail::BitwiseXor::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html#a0989e3bcd064ae06c33f660696a869a0',1,'mlx::core::detail::BitwiseXor::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_left_shift.html#a50bcbc53e2278483d9063decf7ad78d8',1,'mlx::core::detail::LeftShift::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_left_shift.html#a9385f580830a6ad163dd9bb8c4905e7a',1,'mlx::core::detail::LeftShift::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_right_shift.html#aa86d02e4ca59bc7ffacdc342841a0ea9',1,'mlx::core::detail::RightShift::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_right_shift.html#a154528ba50e89a4c532a181f135b1620',1,'mlx::core::detail::RightShift::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#a8b672df71eea3f31f5e2aa50662f3b19',1,'mlx::core::detail::Remainder::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_remainder.html#ac1bcf314046fa1c76e5491336cf68e02',1,'mlx::core::detail::Remainder::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_maximum.html#a1edfed0e0b33227b67c7709691f846c7',1,'mlx::core::detail::Maximum::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_maximum.html#a1a3bd09f6c4e61982ebf1a9bfaa38059',1,'mlx::core::detail::Maximum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_minimum.html#a28b51060b9345fb2021d5176cd607778',1,'mlx::core::detail::Minimum::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_minimum.html#a5cdc82cc78adbc9854aa9b1c4417d6d3',1,'mlx::core::detail::Minimum::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_power.html#ad047c7d25e1b0f32dc17a03d826cf0a0',1,'mlx::core::detail::Power::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_power.html#a5d3c31365fcf2de52f78c3695da83152',1,'mlx::core::detail::Power::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_equal.html#a5d3f7423078444e5d690fb6d50fcce23',1,'mlx::core::detail::Equal::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_equal.html#a2994cf1884e7126e76d0a20b215fe3ab',1,'mlx::core::detail::Equal::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_greater.html#a9186b3e29c84700ea93ca9470556b0b3',1,'mlx::core::detail::Greater::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_greater.html#aa3844c2bae3c7a981739f642aa0dd094',1,'mlx::core::detail::Greater::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html#a8da40f79562ef8ffbd30ddcf40d83e0f',1,'mlx::core::detail::GreaterEqual::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html#a3b005f85522ad0e4b57044eed930ac30',1,'mlx::core::detail::GreaterEqual::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_less.html#a8e9c159887284420b1161421e58a0bda',1,'mlx::core::detail::Less::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_less.html#a0b4032dff1ad2b387745cb000aabdcbb',1,'mlx::core::detail::Less::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_less_equal.html#a5f7f700be5fdf4629a96ab271caf5440',1,'mlx::core::detail::LessEqual::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_less_equal.html#a31e70f8830a07557697541301555a7a7',1,'mlx::core::detail::LessEqual::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_not_equal.html#a99d16a3d7f637901869bf650b1ea6e13',1,'mlx::core::detail::NotEqual::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_not_equal.html#a23d662b5fd968dc17d3bee2595b5f99d',1,'mlx::core::detail::NotEqual::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_na_n_equal.html#a441e5e8552be45ced34001b465d251e1',1,'mlx::core::detail::NaNEqual::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_na_n_equal.html#a073b20b0d8d41ec8364b7c477421b9bf',1,'mlx::core::detail::NaNEqual::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_log_add_exp.html#a434da15bcb95dc979c73ec795cfec339',1,'mlx::core::detail::LogAddExp::operator()(Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_log_add_exp.html#ad1663fd809acaa4038f90666436599e5',1,'mlx::core::detail::LogAddExp::operator()(T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_select.html#a930f9da2e6b3453e04f21382435a2cfb',1,'mlx::core::detail::Select::operator()(bool condition, T x, T y)'],['../structmlx_1_1core_1_1detail_1_1_select.html#a8c5135e3098cfd2521a2a266ba08f1e4',1,'mlx::core::detail::Select::operator()(Simd< bool, N > condition, Simd< T, N > x, Simd< T, N > y)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#acb9168d40f09d73a2243f75f13bbadc2',1,'mlx::core::detail::Abs::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_abs.html#a0d657bc9a381dca1b5860b9a1b5a5702',1,'mlx::core::detail::Abs::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html#a1b927a97bbef1478c768bb85cb764c94',1,'mlx::core::detail::ArcCos::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html#a04b4c9d1fc0160973aa28b1f809b9d51',1,'mlx::core::detail::ArcCos::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html#a4436be0278ceaced10ef98eb6f30f789',1,'mlx::core::detail::ArcCosh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html#a767d354bec863942822ee0b9b6742a88',1,'mlx::core::detail::ArcCosh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html#ab1ad6339c662305bd682b14f8d8afd6c',1,'mlx::core::detail::ArcSin::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html#ac69091929815e5317308b4088f5c2f46',1,'mlx::core::detail::ArcSin::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html#ac6e45e41f931f556697c060a2a858816',1,'mlx::core::detail::ArcSinh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html#ac7bf9bac66fef917f75494b2345e6aaf',1,'mlx::core::detail::ArcSinh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html#a697b7f12f30d642ee5f0c54aaf86a8ec',1,'mlx::core::detail::ArcTan::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html#aee87bf10c278a70ca788085d1b499afe',1,'mlx::core::detail::ArcTan::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html#a93a660ea073526e1f75b2d3c4ac6c366',1,'mlx::core::detail::ArcTanh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html#a601e8c52bb938eb3a616756a35419e8b',1,'mlx::core::detail::ArcTanh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a2354e9fa1502d1743834b98cdec17653',1,'mlx::core::detail::Ceil::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_ceil.html#a672f65e47d65e4e8d88be252bce0164b',1,'mlx::core::detail::Ceil::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html#a33bbfcc195781eb33df0a4efc50569ed',1,'mlx::core::detail::Conjugate::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html#a386b583d24a2cf1ba8dcc3ba52c226f5',1,'mlx::core::detail::Conjugate::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_cos.html#a663065fd41e5d85e8f044e9f81070568',1,'mlx::core::detail::Cos::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_cos.html#ad4caef573f9d9071f8945a8efed231ad',1,'mlx::core::detail::Cos::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_cosh.html#ae94b6da9ceb47e9d4aaf61451126f58d',1,'mlx::core::detail::Cosh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_cosh.html#a63591f49776d9aadc02200036ae38317',1,'mlx::core::detail::Cosh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_erf.html#a4f5986391863d30e0e7b17bd1996a5f6',1,'mlx::core::detail::Erf::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_erf.html#a168f8ccc6c8053b05dd1a48904ca8fd4',1,'mlx::core::detail::Erf::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html#a0cdd8d6e71222695d0f148b9ad048429',1,'mlx::core::detail::ErfInv::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html#acc93c0511141404208b35f302f8c1fcb',1,'mlx::core::detail::ErfInv::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_exp.html#aad7fb8de7561479c7aa3c741322a3101',1,'mlx::core::detail::Exp::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_exp.html#a0846300cee28315e5b42f74acafbd1a1',1,'mlx::core::detail::Exp::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_expm1.html#a2c78a15f0dd01d13f3a78ac45347ed3e',1,'mlx::core::detail::Expm1::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_expm1.html#abf7e61b8387521e9d44334ce88d833a0',1,'mlx::core::detail::Expm1::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a5c41fb72ec3da9289c24b92802e28f2e',1,'mlx::core::detail::Floor::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_floor.html#a16c13cfe736098bffc81d655e172294a',1,'mlx::core::detail::Floor::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_log.html#a0041795bfd063a9769a3747bd7a91d61',1,'mlx::core::detail::Log::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_log.html#a0012a4e1744dbe9a28c3b5652be6e1c6',1,'mlx::core::detail::Log::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_log2.html#a83258d8a3fe12e082d0b317fcfafb28b',1,'mlx::core::detail::Log2::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_log2.html#a467bd4c995674721ff5fff6df33aead8',1,'mlx::core::detail::Log2::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_log10.html#ade464425f69e5b76bf61b5ba3da75089',1,'mlx::core::detail::Log10::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_log10.html#a2633c5b772bbc9f8b66cffd4a3e01a3f',1,'mlx::core::detail::Log10::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_log1p.html#abed96d56b07c6a96666b770c9711e52e',1,'mlx::core::detail::Log1p::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_log1p.html#a3220de8c6090c44aa2070b1fbb2dc340',1,'mlx::core::detail::Log1p::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_logical_not.html#a4978cc3a63e70a1a4fee6470764ae9d9',1,'mlx::core::detail::LogicalNot::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_logical_not.html#a79799668ea5c364b0b4e2bc330e76253',1,'mlx::core::detail::LogicalNot::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_negative.html#a93a1dfb47eba54aff44b2945d131c97e',1,'mlx::core::detail::Negative::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_negative.html#afc4595c70ef7196df374cf4b2cc5e526',1,'mlx::core::detail::Negative::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_round.html#acd099ba81c8c281e9660cf8c0fed0cd1',1,'mlx::core::detail::Round::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_round.html#a653f29c059bbfa6192378732a8a23351',1,'mlx::core::detail::Round::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sin.html#a07c357c49dbf6b0579b1e771c6eb5766',1,'mlx::core::detail::Sin::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sin.html#ae95671816529cc2188389af37a2f1a13',1,'mlx::core::detail::Sin::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sinh.html#a1e299cd64bc0c7aaa1ceeac35dfe7831',1,'mlx::core::detail::Sinh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sinh.html#a9663ddf0fa4c0003576b48f3d5385f00',1,'mlx::core::detail::Sinh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sqrt.html#acac518e8e7cf3dd103f4f72f22b23221',1,'mlx::core::detail::Sqrt::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sqrt.html#aa5a4830b3ef7efab20ea88a110667efd',1,'mlx::core::detail::Sqrt::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_rsqrt.html#ac6720a6270393152ab2924a77bfb17b2',1,'mlx::core::detail::Rsqrt::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_rsqrt.html#a9af247be16bab83243038aac54446b79',1,'mlx::core::detail::Rsqrt::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_tan.html#a9c8d3570a1e4daa054bb41999043d9e9',1,'mlx::core::detail::Tan::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_tan.html#aba397cd7ac05bbe06dfa9e3a64bdb05f',1,'mlx::core::detail::Tan::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_tanh.html#a79eeba686f3dd5dce097ff5b9b27dd7c',1,'mlx::core::detail::Tanh::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_tanh.html#a1749ba1edfd53095ed7d45c0e53bab61',1,'mlx::core::detail::Tanh::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_imag.html#a070cf43bc4e30871f8f32d4b84be05c8',1,'mlx::core::detail::Imag::operator()(Simd< complex64_t, N > x)'],['../structmlx_1_1core_1_1detail_1_1_imag.html#a5bd82e2185f3779e398c179d42a3e782',1,'mlx::core::detail::Imag::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_real.html#a7c6c6c188d611e2084dba66b7489c21f',1,'mlx::core::detail::Real::operator()(Simd< complex64_t, N > x)'],['../structmlx_1_1core_1_1detail_1_1_real.html#ae84a939fdb5916257a7731cda66d4d61',1,'mlx::core::detail::Real::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sigmoid.html#a12a3d53f0fd797b5cdd9d04d048ce1a4',1,'mlx::core::detail::Sigmoid::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sigmoid.html#a64b72561bfaf758632167f00648f4c89',1,'mlx::core::detail::Sigmoid::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a913c095e25668c8a6bb6e3243e150606',1,'mlx::core::detail::Sign::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_sign.html#a64ed5013cee7ff18c7fe70bc04737e7b',1,'mlx::core::detail::Sign::operator()(T x)'],['../structmlx_1_1core_1_1detail_1_1_square.html#abab2378a94c4c38dffeb06a74b0f81ee',1,'mlx::core::detail::Square::operator()(Simd< T, N > x)'],['../structmlx_1_1core_1_1detail_1_1_square.html#a54e9e3c0d0896e142289e8282eab1099',1,'mlx::core::detail::Square::operator()(T x)'],['../struct_add.html#ac5c66b63d63a222d3ae0ab8cc7c90eb5',1,'Add::operator()()'],['../struct_floor_divide.html#a2b328e4d768e718fa439f955c524666a',1,'FloorDivide::operator()(T x, T y)'],['../struct_floor_divide.html#afc16a2b2a745225e0bc95640f3fc0219',1,'FloorDivide::operator()(float x, float y)'],['../struct_floor_divide.html#ae91719a15f7e643d552129f476089c6a',1,'FloorDivide::operator()(half x, half y)'],['../struct_floor_divide.html#a4aa9f858626583e02bd79f747229bbca',1,'FloorDivide::operator()(bfloat16_t x, bfloat16_t y)'],['../struct_divide.html#a0a16b9194abc2ab7c61129f81a9bbb3d',1,'Divide::operator()()'],['../struct_remainder.html#ab7875512ff4341c580c6dc372e64fc58',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#a18150b5f4425e30b95ffabc6bb25cede',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#ab3b75f54b56fd357c9755daadb2cafc2',1,'Remainder::operator()(T x, T y)'],['../struct_remainder.html#ae918ce0e246937d4fe04e2ea36e4b2c1',1,'Remainder::operator()(complex64_t x, complex64_t y)'],['../struct_equal.html#aa498087080900d4428ba428a6496a769',1,'Equal::operator()()'],['../struct_na_n_equal.html#a00220898e02db656d21dde9e9354a8dc',1,'NaNEqual::operator()(T x, T y)'],['../struct_na_n_equal.html#a6185e4554dce5b4659d21673c576be51',1,'NaNEqual::operator()(complex64_t x, complex64_t y)'],['../struct_greater.html#a98d7d8ee360cd0f469c6eb9a017560f5',1,'Greater::operator()()'],['../struct_greater_equal.html#ae69a3bccc567a46506cf0d296294ce80',1,'GreaterEqual::operator()()'],['../struct_less.html#a5ee0b31b2d9123dc4504f2979a5854d3',1,'Less::operator()()'],['../struct_less_equal.html#ae9f9a1b2eae548977139704f0044acfe',1,'LessEqual::operator()()'],['../struct_log_add_exp.html#ab32417f18e8ff68c15f78aceeb624edf',1,'LogAddExp::operator()()'],['../struct_maximum.html#a3ea0f42bc4cd80b68a98f189f9fa859c',1,'Maximum::operator()(T x, T y)'],['../struct_maximum.html#a0bc8fadc87f2c49fc440d625bfc97ca6',1,'Maximum::operator()(T x, T y)'],['../struct_maximum.html#a907e8793900be5927625377dab199644',1,'Maximum::operator()(complex64_t x, complex64_t y)'],['../struct_minimum.html#aa6113dfac3986c0f571fa53f65c5330e',1,'Minimum::operator()(T x, T y)'],['../struct_minimum.html#a0c939921de87ab9c6959238aac81a059',1,'Minimum::operator()(T x, T y)'],['../struct_minimum.html#a800fba087280f79c2f7e9aff75bed093',1,'Minimum::operator()(complex64_t x, complex64_t y)'],['../struct_multiply.html#a1327fc5a0713931afe997b0d4d2988e0',1,'Multiply::operator()()'],['../struct_not_equal.html#af008d73a5d9cde0b8309b7e8ee7438b2',1,'NotEqual::operator()(T x, T y)'],['../struct_not_equal.html#a14de494cea4e4869351202cad1149f17',1,'NotEqual::operator()(complex64_t x, complex64_t y)'],['../struct_power.html#a2b6df2a9e48155ff9734caca8504a79f',1,'Power::operator()(T base, T exp)'],['../struct_power.html#a36829163d42973034a1f8a7ecc57a1de',1,'Power::operator()(T base, T exp)'],['../struct_power.html#a27cdfb313c4e82b63bdcdaee923cbbef',1,'Power::operator()(complex64_t x, complex64_t y)'],['../struct_subtract.html#ae0856cd8d449074ca287baa7e460f68a',1,'Subtract::operator()()'],['../struct_logical_and.html#a8bc6bdabc0ea0678a46e2cf6217cb3a6',1,'LogicalAnd::operator()()'],['../struct_logical_or.html#ade6a931324a604a3119d2220d6f5460d',1,'LogicalOr::operator()()'],['../struct_bitwise_and.html#afb48af090b01dd0200963bc12d842e36',1,'BitwiseAnd::operator()()'],['../struct_bitwise_or.html#a41f847463daafa99ee56f4035578390f',1,'BitwiseOr::operator()()'],['../struct_bitwise_xor.html#a3a3e8a56caab739d40262d9349c9c485',1,'BitwiseXor::operator()()'],['../struct_left_shift.html#aa729747784c38bfdbba34794fcf5175b',1,'LeftShift::operator()()'],['../struct_right_shift.html#a2cc59b400c68342b0e43050431323c17',1,'RightShift::operator()()'],['../struct_arc_tan2.html#ac9b7729753e13be293ab700231d061ac',1,'ArcTan2::operator()()'],['../struct_div_mod.html#a8b5758f2ea18d4c903b462331b25abfe',1,'DivMod::operator()()'],['../struct_cum_prod_3_01bool_01_4.html#ad634be0b139d10ce6d21332eef0d936b',1,'CumProd< bool >::operator()()'],['../struct_cum_max.html#a781b9b955c5412466da6af6c70d73c06',1,'CumMax::operator()()'],['../struct_cum_min.html#ae0b8c3761e04fa538d304ca842281a66',1,'CumMin::operator()()'],['../struct_less_than.html#a2798eb377b411c93a4ed30cf35caade2',1,'LessThan::operator()()'],['../struct_select.html#adb51692aae3038de07dd745891bf9848',1,'Select::operator()()'],['../struct_abs.html#a9e7481dfcc162509769852026ff4a344',1,'Abs::operator()(T x)'],['../struct_abs.html#a0ca113fd036151c443df3f83cc667f28',1,'Abs::operator()(uint8_t x)'],['../struct_abs.html#adaeab32a7e377dc990077ab15f3dc4c2',1,'Abs::operator()(uint16_t x)'],['../struct_abs.html#a99d2a2f37a6cddd3168b0224f2a9b963',1,'Abs::operator()(uint32_t x)'],['../struct_abs.html#ac9cbc02422d930479303f240a7ea6c71',1,'Abs::operator()(uint64_t x)'],['../struct_abs.html#ac30835b27784d451bd2e4524c8eb9e11',1,'Abs::operator()(bool x)'],['../struct_abs.html#ab82917d6b30a2c579e7eb879d305c5fc',1,'Abs::operator()(complex64_t x)'],['../struct_arc_cos.html#a5553cecf58511e24e76ac97f2d90b9ac',1,'ArcCos::operator()()'],['../struct_arc_cosh.html#a5c9e7712c14c97298b23ec48e19abc58',1,'ArcCosh::operator()()'],['../struct_arc_sin.html#a0343872f2da93bae2bb0baadf49da022',1,'ArcSin::operator()()'],['../struct_arc_sinh.html#a3066fb7dc7c3180100fb55ff94af6a7a',1,'ArcSinh::operator()()'],['../struct_arc_tan.html#af3a0aec6acec8ae8f5e4c4d5cf8c91ba',1,'ArcTan::operator()()'],['../struct_arc_tanh.html#a37dc3e01ec2830de7e82ed6c6363ac88',1,'ArcTanh::operator()()'],['../struct_ceil.html#a5e2a4ef1b012f5d352064489156e5e44',1,'Ceil::operator()(T x)'],['../struct_ceil.html#a455cd8083ba859993077f2e078ae165b',1,'Ceil::operator()(int8_t x)'],['../struct_ceil.html#a2acb61bc658c7a216795e7f76ebcf98a',1,'Ceil::operator()(int16_t x)'],['../struct_ceil.html#aef8c37f7a8ee3fc80700d605a09891fb',1,'Ceil::operator()(int32_t x)'],['../struct_ceil.html#a93d0110511ad5dd200e12d37a3d7d6e3',1,'Ceil::operator()(int64_t x)'],['../struct_ceil.html#aa335b745fa26e0f443cdb36298105484',1,'Ceil::operator()(uint8_t x)'],['../struct_ceil.html#ade17e13b7f30f5c590fae1581a2013ac',1,'Ceil::operator()(uint16_t x)'],['../struct_ceil.html#a411c75cc35cdc088402e176a1defd22d',1,'Ceil::operator()(uint32_t x)'],['../struct_ceil.html#a9ac660ca29eef7a7429fceb7b917a68a',1,'Ceil::operator()(uint64_t x)'],['../struct_ceil.html#a40de367e62f06ebd7e1330afa93a9ad9',1,'Ceil::operator()(bool x)'],['../struct_cos.html#ae222f8710f6b8254c471ebd475aa5bda',1,'Cos::operator()(T x)'],['../struct_cos.html#a5f26feb1dcc4bec5f59a9ff511c5b163',1,'Cos::operator()(complex64_t x)'],['../struct_cosh.html#a5847ebeebb236fdc926798ddc16475ba',1,'Cosh::operator()(T x)'],['../struct_cosh.html#aefdd91298dac16d528d29ee47e2f7252',1,'Cosh::operator()(complex64_t x)'],['../struct_conjugate.html#acb0a2694285f1f57c7654b371ce8cbd8',1,'Conjugate::operator()()'],['../struct_erf.html#a80719402ad7f7d418859a6677d7b604d',1,'Erf::operator()()'],['../struct_erf_inv.html#afbf3668d1a512e889f093a0bc7673309',1,'ErfInv::operator()()'],['../struct_exp.html#a5ef395868e055348c0802fd5fe45669c',1,'Exp::operator()(T x)'],['../struct_exp.html#a2b341ac400c4d145397950eb60734336',1,'Exp::operator()(complex64_t x)'],['../struct_expm1.html#a4b834d42cf0b84daf03fec62c222091a',1,'Expm1::operator()()'],['../struct_floor.html#ace3551f28429081e9f3a3dab0c84212b',1,'Floor::operator()(T x)'],['../struct_floor.html#a10d7fd05b4c224c9f135451246d13014',1,'Floor::operator()(int8_t x)'],['../struct_floor.html#a2865a04a492e3590302f4bd3215a10d7',1,'Floor::operator()(int16_t x)'],['../struct_floor.html#a41012343ff0463ec44b4d06196f41182',1,'Floor::operator()(int32_t x)'],['../struct_floor.html#aae3181d15856796aa0628cf30c92aa2e',1,'Floor::operator()(int64_t x)'],['../struct_floor.html#ac6cf38d82c8e270911afdca4c69ad51b',1,'Floor::operator()(uint8_t x)'],['../struct_floor.html#a78969b9e2b53ae248e72a67259eea5d8',1,'Floor::operator()(uint16_t x)'],['../struct_floor.html#a959009320ed622ed45b39becab1d5b98',1,'Floor::operator()(uint32_t x)'],['../struct_floor.html#a7d04b83c3345cd867315cae2d7ff68ab',1,'Floor::operator()(uint64_t x)'],['../struct_floor.html#abea845fe5e8e6b93bd4bca8717337e0b',1,'Floor::operator()(bool x)'],['../struct_imag.html#a3b29e9f8a46c194d683f6a9938314400',1,'Imag::operator()()'],['../struct_log.html#a32a383cb6be06e616a75f23bf49089c3',1,'Log::operator()()'],['../struct_log2.html#ac1e067ecdcbdbffb6106e789c2b98b64',1,'Log2::operator()()'],['../struct_log10.html#ac596a74c1642a00f3eced07ee3334122',1,'Log10::operator()()'],['../struct_log1p.html#a4464c6e7bdbe55ffd7d961c695cd13ce',1,'Log1p::operator()()'],['../struct_logical_not.html#a8a620bac957ab8c09ac85adfddd96708',1,'LogicalNot::operator()()'],['../struct_negative.html#af6879b374314a559faa321e8cce3d710',1,'Negative::operator()()'],['../struct_real.html#a85b9c5b9e65297994fa26ff68e19e809',1,'Real::operator()()'],['../struct_round.html#aa06a0195867e2ceb679c403b6909a1c4',1,'Round::operator()(T x)'],['../struct_round.html#ad3a08f2276ff1033900bc0a7da812655',1,'Round::operator()(complex64_t x)'],['../struct_sigmoid.html#a75a24cd75cb4d4c9a072811b2d70ad55',1,'Sigmoid::operator()()'],['../struct_sign.html#aa3304c6b43bcad53061614b741d8403c',1,'Sign::operator()(T x)'],['../struct_sign.html#ac48992b675b8b28be1e27e1f2ec5d2f7',1,'Sign::operator()(uint32_t x)'],['../struct_sign.html#ae07a4249e1b61419a3b9ca6c337b7bb5',1,'Sign::operator()(complex64_t x)'],['../struct_sin.html#a7caf98c777521fa5d5c6ddaaa3b779fd',1,'Sin::operator()(T x)'],['../struct_sin.html#aa510cf4595b6d49065ab6b602d8fcb14',1,'Sin::operator()(complex64_t x)'],['../struct_sinh.html#a02cf32bcf560657b9ee34fb1affed8e2',1,'Sinh::operator()(T x)'],['../struct_sinh.html#a1f8ba1858d352ee68861cd6ea861af43',1,'Sinh::operator()(complex64_t x)'],['../struct_square.html#afde739fc544e45dd30964c02dca94310',1,'Square::operator()()'],['../struct_sqrt.html#ab9b16d2b9b03a1c54190f4479a56a4ad',1,'Sqrt::operator()()'],['../struct_rsqrt.html#ae16699fd829e40416436247a39233fda',1,'Rsqrt::operator()()'],['../struct_tan.html#a1e6fb8c691621c69cb9bd393de4f6e78',1,'Tan::operator()(T x)'],['../struct_tan.html#a2ef120c9f92b0d2e9cec8389eda05724',1,'Tan::operator()(complex64_t x)'],['../struct_tanh.html#adce11a7ad33226c6ecff34f46f5c45d7',1,'Tanh::operator()(T x)'],['../struct_tanh.html#aa8423b43c725bb4b88965a11e8cf20f6',1,'Tanh::operator()(complex64_t x)'],['../structmlx_1_1core_1_1_function_exporter.html#ada4e13daeb3ba0f5ebe20ec0663727b3',1,'mlx::core::FunctionExporter::operator()(const std::initializer_list< array > &args)'],['../structmlx_1_1core_1_1_function_exporter.html#a82aeb5fa32ef5638f42dc2372278427e',1,'mlx::core::FunctionExporter::operator()(const Args &args)'],['../structmlx_1_1core_1_1_function_exporter.html#ac8b8fa0a23d58a94e2e9b923dc7324e8',1,'mlx::core::FunctionExporter::operator()(const Kwargs &kwargs)'],['../structmlx_1_1core_1_1_function_exporter.html#a35a3c1d94249ce0fe0e82b0ea047d441',1,'mlx::core::FunctionExporter::operator()(const Args &args, const Kwargs &kwargs)'],['../structmlx_1_1core_1_1_imported_function.html#a3555db23026d30eaeee265fed99947b2',1,'mlx::core::ImportedFunction::operator()(const std::initializer_list< array > &args) const'],['../structmlx_1_1core_1_1_imported_function.html#a5953b3f47c094cc47bcbb0845379ca8d',1,'mlx::core::ImportedFunction::operator()(const Args &args) const'],['../structmlx_1_1core_1_1_imported_function.html#a10fec4eab5851ed825a9b46a31cedcc9',1,'mlx::core::ImportedFunction::operator()(const Kwargs &kwargs) const'],['../structmlx_1_1core_1_1_imported_function.html#a7d1accece61230eec256e0f70610776d',1,'mlx::core::ImportedFunction::operator()(const Args &args, const Kwargs &kwargs) const']]], + ['operator_2a_20',['operator*',['../structpocketfft_1_1detail_1_1cmplx.html#a26bf3d709a58f06228e502af6db8e5ac',1,'pocketfft::detail::cmplx::operator*(const T2 &other) const -> cmplx< decltype(r *other)>'],['../structpocketfft_1_1detail_1_1cmplx.html#ad9c591ef8ae976293f207937d273e9a1',1,'pocketfft::detail::cmplx::operator*(const cmplx< T2 > &other) const -> cmplx< decltype(r+other.r)>'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a153756072fda6d3e53bcca11b46a1238',1,'mlx::core::array::ArrayIterator::operator*()'],['../namespacemlx_1_1core_1_1simd.html#a08c1e7a00b1b4bc60e30d1554f4f46f2',1,'mlx::core::simd::operator*(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#ae4ec5f1f081d20b46b13eb83eb1b6431',1,'mlx::core::simd::operator*(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a4555cd6a3b50af00700f97fdf00f63a7',1,'mlx::core::simd::operator*(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ab6a73491bcb185cd91ae4db6b0f21e49',1,'mlx::core::simd::operator*(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value *b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a4030444ea38ce1529a8cbb8c183a28bd',1,'mlx::core::simd::operator*(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a *b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#acd5ac48dc7895f06daf55f0a7e0667fb',1,'mlx::core::simd::operator*(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value *b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a6f6d26e3fe39ee1ba0a7380d0ecf7b45',1,'mlx::core::simd::operator*(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a5373c1af09825b5f701ebd106508fa6b',1,'mlx::core::simd::operator*(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#ac50da923a4b7ac682554bd1d74c306d9',1,'mlx::core::simd::operator*(T a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#a681d4fb076973f58f7dac894ec62a385',1,'operator*(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8f06316063fc91747533105f256b55b5',1,'operator*(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7b3bce3f6f17089d87e13e91f580a581',1,'operator*(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a54ae7216b82c5cea362f6b83e1df3a9b',1,'operator*(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a852689073c17596de4fb545bc046b380',1,'operator*(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a168300bbd04d8e97c5e4218cb14ae378',1,'operator*(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6278bd2e0e2805090b33ef666bf7f6bb',1,'operator*(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aecf703522d9ce32dfeefe1e6e903db06',1,'operator*(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7cd44d27fa9a4f13df39894c34fdb348',1,'operator*(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee64dc1890abb6d1035361cb8c751f96',1,'operator*(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1a559ab88dbbb4fd2c7509d2c94e55b',1,'operator*(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a495ae2d9be5d97c4c6448fc4e50a03e1',1,'operator*(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a87ab4b7a502430da664ccb8abd383058',1,'operator*(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5f997839cf49c24ab594a0dff486a7bc',1,'operator*(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa0c2d29950926ae579adf6337fbea64b',1,'mlx::steel::operator*()'],['../group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0',1,'mlx::core::operator*(const array &a, const array &b)'],['../group__ops.html#gac22a67f7de797b1ae59029843cbdcab6',1,'mlx::core::operator*(T a, const array &b)'],['../group__ops.html#ga6f2369ed5fae8ff9b1528670a004dde2',1,'mlx::core::operator*(const array &a, T b)'],['../namespacemlx_1_1core.html#a0cc824d6318f97f7058918ab64ddfc25',1,'mlx::core::operator*(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a81e1c727c3fc48910b030cb65a9e7afa',1,'mlx::core::operator*(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a861d948220d8f48d46c68d2ddb16a096',1,'mlx::core::operator*(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a13d16561812679b36e68185dc4b2d04d',1,'mlx::core::operator*(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a5287610200ff573730c9c92413f48881',1,'mlx::core::operator*(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a377ccc6b4ef36767abca102dca56dc10',1,'mlx::core::operator*(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a5d696b63635ce6967526d6a410f7f6b1',1,'mlx::core::operator*(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abe90e9527bfa3e1c813d41df4a2372e7',1,'mlx::core::operator*(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a5f14963c77f96bcb5a3bef5661a86ba4',1,'mlx::core::operator*(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#acfb06fe9f5fee01dbb5a2b23bccfd0d3',1,'mlx::core::operator*(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#afc9a87f1fccbac05242b91bfbb35c24d',1,'mlx::core::operator*(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0b9678af9b487900cacf6639a4693de0',1,'mlx::core::operator*(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ad5950619081389e6ed7512f38358d33d',1,'mlx::core::operator*(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a65d25d082374761c05b056e1046d1d4e',1,'mlx::core::operator*(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a759191fb984e7737f0ef529c2053ad73',1,'mlx::core::operator*(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3a52675c3d4552b319dd9707844abdec',1,'mlx::core::operator*(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a45d67f5d80fba4d42e34c682a8d22beb',1,'mlx::core::operator*(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ad25880c67bbcbfafbe54dc16418bf736',1,'mlx::core::operator*(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a63c836e1141e07ae72cee770bad01200',1,'mlx::core::operator*(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a265a37b8ee4a97390213e9ec49693e66',1,'mlx::core::operator*(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab5a457da04dcb157a0b5172c4b2244b6',1,'mlx::core::operator*(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#aa56a8bda08be9ef3711496e216a75c95',1,'mlx::core::operator*(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af89612098dd355b1eefb841c753b36ab',1,'mlx::core::operator*(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a4552687a0637f710b5d55bb6378fcabe',1,'mlx::core::operator*(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af69db7def588d7da430434a69456e29c',1,'mlx::core::operator*(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a00af6e5095888f00791ee0ab6d993ad6',1,'mlx::core::operator*(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab48feddc1aa304383e5493923506ad7a',1,'mlx::core::operator*(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a0367b582e85162b4180e086f725e49e9',1,'mlx::core::operator*(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a45f0479526fbccdb00bc73ea7f3b7625',1,'mlx::core::operator*(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a394797646010ba9ef2a1f9b9a4b8ddd9',1,'mlx::core::operator*(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acaaa86b59c7ceb2e092ac07f2a75225c',1,'mlx::core::operator*(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a067d47823a322b88043cce7ce4a3ec78',1,'mlx::core::operator*(bfloat16_t lhs, float16_t rhs)']]], + ['operator_2a_3d_21',['operator*=',['../structpocketfft_1_1detail_1_1cmplx.html#a683fd490182c9189fa2c05b1823edd93',1,'pocketfft::detail::cmplx::operator*=(T2 other)'],['../structpocketfft_1_1detail_1_1cmplx.html#a06f2c26c6fc4722e61b44da4c242ed87',1,'pocketfft::detail::cmplx::operator*=(const cmplx< T2 > &other)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7232b0a0e193b3c6172d6fc2578bf419',1,'operator*=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ade65ebca11e38d56408c512df89b99f4',1,'operator*=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af4348ce3425dd99d069e8fdf06e25a3c',1,'operator*=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2c3c5f793b3d957d7295d7f1faabebee',1,'operator*=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac66657077d55e94197b52b63acb50b7d',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a383165ea838cc3feeee4d9cf54aa77cc',1,'operator*=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab706af260b61f735b28464877d02137c',1,'operator*=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a979374b1dd4e0eaf602326fa901336d1',1,'operator*=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac815eec2c1b15a47b1c6ea6790e77d24',1,'operator*=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8110fae7bcc34a0de5927546b24aa935',1,'operator*=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae4acef3e7ae7dfe359422503f894e885',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc268cdbc30500f3009f5de2b2f0f67a',1,'operator*=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a81f65b04a87a25c7eb1a751d1be9fa55',1,'operator*=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08c1f916302eb9d48c93f8b7260538fe',1,'operator*=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc8e82b8f593b12c6d405e2250ab0f62',1,'operator*=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4611728172afea51860a77fdb06cafa0',1,'operator*=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0b8736e2ae24758b6e24ea72668df5b4',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad920df9579603f0b0ee2689eba330617',1,'operator*=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae97ab6c3ddcc2754b24f86319a5398be',1,'operator*=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3ff4ff59f411010ac8502cfabda4bd6f',1,'operator*=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abd3d82e2dec1847e97eb8fc3bab2985a',1,'operator*=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a738078eb7d5ff94ff48156a555d763a5',1,'operator*=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a435f2f4256aadb1b57fd62bb7f733cf7',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0e4377b120d6305335d296e031ee5b30',1,'operator*=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a917354f77eac26189da8a2f610a00074',1,'operator*=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af725f935bfa0405e5ff17ede3ac47283',1,'operator*=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7c56980c234a04260b8b19298085e526',1,'operator*=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab840ff9de0cdd0e9afffb8baa2a850a3',1,'operator*=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a73416a7415f3fe31525e33419e5e8aab',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a16978f4b16d954ef4d4cf0f32f6c0b94',1,'operator*=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a99aa4cc110d1c7aa3b4c8c5cbf9235b7',1,'operator*=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2179abbc91ce8763e96e39e1917bfa6e',1,'operator*=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab070ea4676d10a10ff3e9379a4068a57',1,'operator*=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0197e039d4c65bf49649a6f250c2d436',1,'operator*=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad3565cc6fd1e088d052b1108aa065851',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a711693988c437c2fb4d7da505982fe21',1,'operator*=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeff4c28986f98c23de1df17043edb0f5',1,'operator*=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7dbf0c75df4817cb4ef8b60c417a89d0',1,'operator*=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a323a80492cd17a49e2c3dd18f8c8b5cc',1,'operator*=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adb465776d3868bda0525d632ffc4d129',1,'operator*=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a12a98d71d670b409b8065e0d61672d55',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5d00eb2ec2b0e15b2753d100694c45ae',1,'operator*=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1a2a683ff40490226eb1371fb905023d',1,'operator*=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4126fb7ed5bbb27a2332c543cf56a337',1,'operator*=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab092d9790ef20fc0386707530aee89db',1,'operator*=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abff1fd2439e31e6e64a3d2fdee3c7821',1,'operator*=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a625dcb133f1f953f263e6200399866c6',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08b6071245513e1726ec68e3b63edc53',1,'operator*=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a13aa79165ec87710e977f33fe0361e91',1,'operator*=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3796dcf819adb1ef8152f57ba63ff6b1',1,'operator*=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aaab79d0b4c9e9bdc059ace6ec58c5b00',1,'operator*=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a0dd3893abc8986901872c8365ab1509d',1,'mlx::core::operator*=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a3cc5c154e4ad9a83ad43da8513146fdc',1,'mlx::core::operator*=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a600e77dbc72e78207b5f5dbf4b298781',1,'mlx::core::operator*=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a54833be1d44bc3adfc9ea218fc3685bd',1,'mlx::core::operator*=(float &lhs, _MLX_Float16 rhs)']]], + ['operator_2b_22',['operator+',['../structpocketfft_1_1detail_1_1cmplx.html#a76447ef141c8732d57421749fc81b236',1,'pocketfft::detail::cmplx::operator+()'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ae2adde594b5a4853f6bc78263a957d85',1,'mlx::core::array::ArrayIterator::operator+()'],['../namespacemlx_1_1core_1_1simd.html#aac6acd134f1498b4fb45fdbc882335bf',1,'mlx::core::simd::operator+(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a8b622c47d07b171b2303ea744bf72284',1,'mlx::core::simd::operator+(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#aed655ffa017ade5e0f954f906d9f7ae6',1,'mlx::core::simd::operator+(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a27dfc3843dbefbbebed5b7137bacbb59',1,'mlx::core::simd::operator+(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value+b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#aa78806bf6a3be64b44e9a1f04bad3862',1,'mlx::core::simd::operator+(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a+b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a98b77f1ca24bff373f48ef62f0013a02',1,'mlx::core::simd::operator+(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value+b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#ae690b57b386cbad40565487d6d2393bb',1,'mlx::core::simd::operator+(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a417109cdd61f35954ba2cc37af9b4460',1,'mlx::core::simd::operator+(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#ac33643b5f3cdbd3be0fa7d5784e35007',1,'mlx::core::simd::operator+(T a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#ad6af5c6c5ed4898b49758618e5aee189',1,'operator+(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a09c1a797eb7f43742578680899932f50',1,'operator+(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a551b970f73bb4a3b287653021d000b60',1,'operator+(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a43a225e7e548bb041f3a5d844faaf0da',1,'operator+(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8b6c3fd9d068a2159084359df8b9b449',1,'operator+(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0a5bfe15d95ba540795f4c25ebfa4f07',1,'operator+(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa415ce182fe7582d885fe633fc3527ce',1,'operator+(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a62f891b7dbba0000749cf338f594bedb',1,'operator+(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab43932322f81bf322aa1b0deeee9a987',1,'operator+(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acd15d46ea5827a2a39898ccbb8352eb8',1,'operator+(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a006763fae6e0577fc168ec9446f0f747',1,'operator+(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a12a47e8ac0be788edff57ae0a96d7830',1,'operator+(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af87dfa2122e9c76042dc41fb7f338a87',1,'operator+(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af2737d09c887ee8cd43fdeabceddbe82',1,'operator+(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a12ff4f38aa8474bf76770c7b8e3e18cb',1,'mlx::steel::operator+()'],['../group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c',1,'mlx::core::operator+(const array &a, const array &b)'],['../group__ops.html#ga7d0ec8d01e7cefa6a6b25f11876761b5',1,'mlx::core::operator+(T a, const array &b)'],['../group__ops.html#ga7cc080a4f9d4a667f2099aa0dbfefadd',1,'mlx::core::operator+(const array &a, T b)'],['../namespacemlx_1_1core.html#ac14b984970cafd8fbe24d080949515cc',1,'mlx::core::operator+(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab076069c6f0047c548a8dc29d35dd36a',1,'mlx::core::operator+(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aab9d96b0a168f4d05146000a6212b5d8',1,'mlx::core::operator+(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac4e6f03d7e4ae701b4eefa784f36185b',1,'mlx::core::operator+(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a4cabd600a5271b0d416c91e8d31dd9c1',1,'mlx::core::operator+(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af26df9dc279d71b7cc10892c72162b58',1,'mlx::core::operator+(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#ac3b97eecec9bd8efb313f8f201560343',1,'mlx::core::operator+(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2e3bb121cbde30c2e6d806df0d41ff59',1,'mlx::core::operator+(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ac87ecce4b44b0826e666a169ddc6f878',1,'mlx::core::operator+(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aed3d9cd32698ef0fe65b1280f103b3f5',1,'mlx::core::operator+(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6fa13b9359cf3f575fbda5260e6e035d',1,'mlx::core::operator+(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af240a6471ff827819192808bffeb857a',1,'mlx::core::operator+(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ac25a05679f312b724c406d8b282803c9',1,'mlx::core::operator+(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a54863a54f258acf2b5c734950618e4e1',1,'mlx::core::operator+(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a9f81f5ea8909db9660197217612ee446',1,'mlx::core::operator+(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a13e26c38da0a4e332e0ae4eb0aed9cb8',1,'mlx::core::operator+(const std::complex< float > &x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a59bb13a0bb7f748c8de34415b248bc57',1,'mlx::core::operator+(const complex64_t &x, const std::complex< float > &y)'],['../namespacemlx_1_1core.html#a38a44c412c8be4c8b952d3082cc7db74',1,'mlx::core::operator+(const complex64_t &x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a011dbdbd2413e59e744cf82b05431340',1,'mlx::core::operator+(bool x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a230e3b7c479add1b171fa0aaa3a8b13c',1,'mlx::core::operator+(const complex64_t &x, bool y)'],['../namespacemlx_1_1core.html#a3a6f43c2485f0d42293184f1aecbeaee',1,'mlx::core::operator+(uint32_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a766157c5d5d00fdf3da95eb7cb2981b9',1,'mlx::core::operator+(const complex64_t &x, uint32_t y)'],['../namespacemlx_1_1core.html#a64dceec2bb03eee963a2a1bc1ac69284',1,'mlx::core::operator+(uint64_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#ae36badb78a17cd7d13663a69645fc328',1,'mlx::core::operator+(const complex64_t &x, uint64_t y)'],['../namespacemlx_1_1core.html#ac1afa5d4c856e4b58109eff086e70ffd',1,'mlx::core::operator+(int32_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a8978def3c2cfe2a96314d564613b80db',1,'mlx::core::operator+(const complex64_t &x, int32_t y)'],['../namespacemlx_1_1core.html#a5b8af5ca4c0e37aba0b7530542bd64c2',1,'mlx::core::operator+(int64_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a3eaa72850205c18450c3af9a01cda219',1,'mlx::core::operator+(const complex64_t &x, int64_t y)'],['../namespacemlx_1_1core.html#ad38b38a3faf050735d45eed4438ee27a',1,'mlx::core::operator+(float16_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a358e66ff205bda3e8542427b6d2edadc',1,'mlx::core::operator+(const complex64_t &x, float16_t y)'],['../namespacemlx_1_1core.html#af56d4b85e329e39a825c01a50e3a2522',1,'mlx::core::operator+(bfloat16_t x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a806a495a129ebaab69cc57ca7db831d6',1,'mlx::core::operator+(const complex64_t &x, bfloat16_t y)'],['../namespacemlx_1_1core.html#a09fc6ebda917969383783a112a8547e7',1,'mlx::core::operator+(float x, const complex64_t &y)'],['../namespacemlx_1_1core.html#a7ed0e2cdb65612f54e67166762cb6408',1,'mlx::core::operator+(const complex64_t &x, float y)'],['../namespacemlx_1_1core.html#af7577c91b8c43682f0ebc9eb9758aae4',1,'mlx::core::operator+(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#abe36af9951afd8dd3ffe90ceedeb7f2b',1,'mlx::core::operator+(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#afb9f780dd056a4f975518f71a3b021ee',1,'mlx::core::operator+(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a6a8e093b24c4c789b7cd160f7e7f7de9',1,'mlx::core::operator+(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#af3a603690fd3de9e4f7f2035a4d25621',1,'mlx::core::operator+(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afa2a4bccfeea9688ac922cb638341511',1,'mlx::core::operator+(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a6111e94d51de12391e5d68b765f28fc3',1,'mlx::core::operator+(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7c7dd6d346e0cdf398a896f2c6958258',1,'mlx::core::operator+(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a00872a443f462b0ae0a30c84fb001bc0',1,'mlx::core::operator+(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4f5d80d03bae6d8d90455d3c47a8c116',1,'mlx::core::operator+(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a78f1f388f9d81ed93f60311f4645d8d0',1,'mlx::core::operator+(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aa43e1d6958c5d5a6fa9a625a1660e741',1,'mlx::core::operator+(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ae877e1d5e3cf57734da8b49535fe3fb3',1,'mlx::core::operator+(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a9a5ae769f67f886d59c8e292a8218550',1,'mlx::core::operator+(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a058878237ce50baa4c909d8d15448d7e',1,'mlx::core::operator+(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a95fd207028f125eefbafe9e0522407fe',1,'mlx::core::operator+(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#abc6425a3fbb386f5ea5964b42507e989',1,'mlx::core::operator+(bfloat16_t lhs, float16_t rhs)']]], + ['operator_2b_2b_23',['operator++',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a3efe69356a84d0d4438f033992fcbd9d',1,'mlx::core::array::ArrayIterator']]], + ['operator_2b_3d_24',['operator+=',['../structpocketfft_1_1detail_1_1cmplx.html#ad4e69dcd89bdb7764c9c5807168f911e',1,'pocketfft::detail::cmplx::operator+=(const cmplx &other)'],['../structpocketfft_1_1detail_1_1cmplx.html#affa618d8850a7c232793b7c61db6d184',1,'pocketfft::detail::cmplx::operator+=(const cmplx< T2 > &other)'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab04f480aea9fbba0895068c7558dd400',1,'operator+=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a251780ac4592cc2b1a543e417ff57770',1,'operator+=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24381d991c2d570aa953694f396a69b5',1,'operator+=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7595740d4cc12924905d6bd1b99ee4da',1,'operator+=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac1498acb8c3623b5f412f70ab6a6528b',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abce5ab327110c164f054b43ed47f79a0',1,'operator+=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae0c70198e236ffe1a98f79987c686419',1,'operator+=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a13b3338935440ae51ecc4a356093efc5',1,'operator+=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a0cb8544b4ebd2906ba8e7f2868e8de',1,'operator+=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7b134429ea0c8493800ff8b465410f9c',1,'operator+=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4154f90ab7857ca856f9e15fe1bf5acf',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab9ae6a51e2027b02cac9966e05f3ba68',1,'operator+=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab93ce536eb7998bee00de4af868e31a9',1,'operator+=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad0ae9e2b4874f991a2c853e1c1fe735d',1,'operator+=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a194a6670cc25ade35a24b566f31af785',1,'operator+=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3d0d689516c99003659c5d026847bd2e',1,'operator+=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a007f58508b98bb79e5c323ed0dec89b6',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa7198e580e2a83c1fd01a4b6fdf86a80',1,'operator+=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a15573fefd880adefbba079b1c1bd8082',1,'operator+=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a104cf94cb9e359d1b6ef92ced2ce0c27',1,'operator+=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa04cfcb52191fd23205a1a3572b46ae0',1,'operator+=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad684bc2ae1a2a627cd3e4a4c641e2d77',1,'operator+=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1e28448e35f4934075b397c34ba3d66',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8ad16afd7f1711de83c0cec5af868f76',1,'operator+=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac45e9ca0c7155caebe3d0f7261518077',1,'operator+=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3c62ac679d6aa515144d40ebafe4a188',1,'operator+=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9ff5ab3aef1057fa083b53a65c8aba03',1,'operator+=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae74bb0a3c12cd1a23f3d29ce307d6fb1',1,'operator+=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac188bd19f236b098d603b0d8acd08921',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef9fa600d107b509f2e3df7d6b080e01',1,'operator+=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af5713afb3a62967a02c3c20661951ee4',1,'operator+=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7f1b84352a3ed6171444a43da1fc7e92',1,'operator+=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af1983edd26245e6e51c6e47354095e32',1,'operator+=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8cd55d1a579540eb450e12a8a8a950be',1,'operator+=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a588ef0f7e03f306758524d378278976f',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a74751abec7086f85f4f26ced44f1ca1f',1,'operator+=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4dd3cf0e5aa116ff330352a50c18cde7',1,'operator+=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afb9a0e18c0e40c77e6143fb7d84ebfba',1,'operator+=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adf0cfd9a608a6fb3d57933e32e7d81d2',1,'operator+=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4bd92db6c8b9b5dc96332c7ae3eff8c7',1,'operator+=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5d628a5bc4fa755610392f47a523a1f1',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7c790442f77f2437b482c4a55e224fc3',1,'operator+=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a77bab4481b41be50297b257e95058706',1,'operator+=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7816a97d16b1d2f8a90227bb1da2f6ac',1,'operator+=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac244d140c6149726ea44174d3e836ca3',1,'operator+=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af802541c4c65ee4442acd495de4d27fe',1,'operator+=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac06eb2fea47a09a8a8abdaa1aa9b4603',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5263b2463fecdc97f9521d00bffea059',1,'operator+=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24ca436ab299a710263d65302532dd3b',1,'operator+=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee1bdf0ab2e445293708b476e8cfde3b',1,'operator+=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a190e27077f0fba642a86f5c8f488bcc2',1,'operator+=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a9f2c9d2f21fbf9fbbacd940c6967c9d1',1,'mlx::core::operator+=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a0b1b3c48afc0a785282e43435bba8418',1,'mlx::core::operator+=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7b763db8194e6fcb1b87eab143dfa47a',1,'mlx::core::operator+=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a827167f6a1ae55428fd218ddd51ec3b6',1,'mlx::core::operator+=(float &lhs, _MLX_Float16 rhs)']]], + ['operator_2d_25',['operator-',['../structpocketfft_1_1detail_1_1cmplx.html#a460da5db36d1c72fb1ed3496fd3abde4',1,'pocketfft::detail::cmplx::operator-()'],['../namespacemlx_1_1core_1_1simd.html#af5be79b8dada8f8e91ae7c03c16606ec',1,'mlx::core::simd::operator-(Simd< T, N > v)'],['../namespacemlx_1_1core_1_1simd.html#ad5761065b4a655cd086d88846ae08d97',1,'mlx::core::simd::operator-(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#acc490f7f5195acfa7b7c5df7afb39438',1,'mlx::core::simd::operator-(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a678cddce777549a39474449d56fd1de6',1,'mlx::core::simd::operator-(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a70563bcd6c28802d11199812ffef38c8',1,'mlx::core::simd::operator-(Simd< T, 1 > in)'],['../namespacemlx_1_1core_1_1simd.html#ab1f7f553d3a9176a70404a29cad06619',1,'mlx::core::simd::operator-(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value - b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#aa73282cb05b65b931b97ce35c46bae20',1,'mlx::core::simd::operator-(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a - b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#ab35a129d6e31b86c06b61252c7b26d4e',1,'mlx::core::simd::operator-(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value - b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a727a13b3d26f9e7cae7f091105867904',1,'mlx::core::simd::operator-(Simd< float16_t, N > v)'],['../namespacemlx_1_1core_1_1simd.html#a6e39cc693b30ad8e530392baf4bb5b0e',1,'mlx::core::simd::operator-(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aad9cc064528e4189a5b7dd816a134ae6',1,'mlx::core::simd::operator-(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a7434ba1ab2ad798fe8557a9b45035e81',1,'mlx::core::simd::operator-(T a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#a226cfd54d49f02e35c5aab3139c7596b',1,'operator-(complex64_t x): complex.h'],['../backend_2metal_2kernels_2complex_8h.html#af5608264cf920688607059b4e8cd3117',1,'operator-(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6aedc8d6d0980134ac69b96f22d9a855',1,'operator-(_MLX_BFloat16 x): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a333f67614dbf8027439a7e124052cb85',1,'operator-(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a891aa4bf46c20a26a55061736aba25f1',1,'operator-(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7ad7ff44a3200853711869f7a577d931',1,'operator-(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af69ef8f1d8ecae0e6f755bf1c46cf075',1,'operator-(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5bd875a54b79b2dcedf674807c3e53c5',1,'operator-(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab02f8646b47806e1d2038f248df03f06',1,'operator-(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab27b26182c7c6e08af37e6d511fd9253',1,'operator-(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5868c85c988ec3432cf86d7df40e464d',1,'operator-(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad03ef47e6cc7521bbfb45740dee20f88',1,'operator-(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab789f8a400512ff27e36b3373170f0c5',1,'operator-(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7f601b22ecc480132d82ad782e5363bf',1,'operator-(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a152366ab4e2ccc867e919af6c74ced91',1,'operator-(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a42bead8ef0beb9f3452128d64cd4df9d',1,'operator-(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aca8ef21c16984ccb329b3bd0c1e4be48',1,'mlx::steel::operator-()'],['../group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816',1,'mlx::core::operator-(const array &a)'],['../group__ops.html#ga0c7f3cb36d4ca516c7a33142f88b9181',1,'mlx::core::operator-(const array &a, const array &b)'],['../group__ops.html#gae68d3d0691ba951501218e98439f3465',1,'mlx::core::operator-(T a, const array &b)'],['../group__ops.html#gaf5e5d882c51ad0a0ea315c274d5439b2',1,'mlx::core::operator-(const array &a, T b)'],['../namespacemlx_1_1core.html#a622ce842fe44e4b6a95e03242341b459',1,'mlx::core::operator-(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#af32a99d930d49e9b178472d7a65531ab',1,'mlx::core::operator-(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a3555a2b31fc0925850d3240e85e03ec5',1,'mlx::core::operator-(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a46080889fd9e5c3f9916508e97dff5ad',1,'mlx::core::operator-(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a9ca27fd1e512c8ed126342e565da12ae',1,'mlx::core::operator-(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3803f8d36558d32bb7dd6e580ea683b4',1,'mlx::core::operator-(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#af5d865528989ca66b3d357e5ce4e0300',1,'mlx::core::operator-(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#afb784b960f55aeb4edd7f567fa74d443',1,'mlx::core::operator-(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a29cbacf4b399c24728fb0808fad498f9',1,'mlx::core::operator-(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aececc0e451237aa6c0d1a2c3d828c86e',1,'mlx::core::operator-(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a383a26cc2689c98fd6c4435ade8dc669',1,'mlx::core::operator-(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad6311ef8df59bdfb212b5cf8169246b2',1,'mlx::core::operator-(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a23b7329bc1c93c8ac0a1f576565fefb0',1,'mlx::core::operator-(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad8d650bf63998abd716ee0ca28e1cbb9',1,'mlx::core::operator-(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a7339b33201254e9119d99d3a728ded72',1,'mlx::core::operator-(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a064318b7a16e5cb6d0a6407501b5c7dc',1,'mlx::core::operator-(_MLX_BFloat16 lhs)'],['../namespacemlx_1_1core.html#a7bae3ff296d9a60ff3c7e448f7fbc6bd',1,'mlx::core::operator-(const complex64_t &v)'],['../namespacemlx_1_1core.html#afb5069ecebdfd9d388c26f83df12c93c',1,'mlx::core::operator-(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a8d126e3f3fa9f8c1c1ae1b09f94df487',1,'mlx::core::operator-(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ad04f1ccd2cd7c487a2f2aaa055939f64',1,'mlx::core::operator-(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a15eb2ea76508ff823fa0591e811d0b7d',1,'mlx::core::operator-(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a96d9577db38d6809d022893e32feeda1',1,'mlx::core::operator-(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a5d9c02765c1672930757416411567bf2',1,'mlx::core::operator-(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a6105d3b5266666b7c6bb9469285a9ec3',1,'mlx::core::operator-(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a777aa772dfb205b25d26f3180d98a2f6',1,'mlx::core::operator-(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a085eb092f4ada47f8169de62886cff90',1,'mlx::core::operator-(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab25e5d211e2c8785b45c3a81a6282e2b',1,'mlx::core::operator-(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#abf5d09561a81b0f0b32d59d77e32e16f',1,'mlx::core::operator-(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4ce6867dbb4d1631d1870dac14022dbb',1,'mlx::core::operator-(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a8a049e646e0442064cfe9e202d7047c5',1,'mlx::core::operator-(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a78e2a1cfc65453185bcca13bd4f523cf',1,'mlx::core::operator-(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#af143cf68673e06390d4bb2ec2892bd22',1,'mlx::core::operator-(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a46d502dfe0b027955950d4e716c2eb26',1,'mlx::core::operator-(_MLX_Float16 lhs)'],['../namespacemlx_1_1core.html#a2631e78c6f0a602f6754ac577ec75f83',1,'mlx::core::operator-(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a73d79cbd75d543d0837b8a51bf103f9e',1,'mlx::core::operator-(bfloat16_t lhs, float16_t rhs)']]], + ['operator_2d_3d_26',['operator-=',['../structpocketfft_1_1detail_1_1cmplx.html#a12441ff423274bd1b54245933d69ad7e',1,'pocketfft::detail::cmplx::operator-=()'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab225043bd02bb423930bc98aae9c2bca',1,'operator-=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac2f1e1f2365cfa531b1519aa9ff67695',1,'operator-=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a513501355a5912a1263fd8b10864142b',1,'operator-=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab4f4ecd62c3d8b3363d02019573dc9f1',1,'operator-=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a92d1348f201d78fcd474f75d5b23ef68',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3eefe9a7f5fb226335ea687012f32d5c',1,'operator-=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef62c7e3e494b6a511a7833c0d942a60',1,'operator-=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad30726cc8b69fd300d33c2a46e123c28',1,'operator-=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8859b5b8dc241e4f58243c85d2630cc8',1,'operator-=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7003e1e5881e3d106257f22b6a3e59fe',1,'operator-=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3165e37d393be50c2cfa9ddcba153684',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a76f5bd895b7214cbc3cea3440992718a',1,'operator-=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7167343d90eb70e5a0d5fa9ec5398e94',1,'operator-=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9b31c363ebc93d592b6fa0e27b00335a',1,'operator-=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a967a1d7b5664f616e5b6f2d257367f0c',1,'operator-=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aff19193e1b2cee29a8737318e95cc74a',1,'operator-=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aede0cc4179507b739849948f1a2fed4b',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7e1a6056f9c96f3c89fe204dbf103be5',1,'operator-=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9d06cceea5c179bcc608452188bd7d6a',1,'operator-=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0aa9ffe056f49fda181bbacbd60556ea',1,'operator-=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ada5685d99c2d6708d1c4ef826d68e879',1,'operator-=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a726cecf778b8584b6f7c37db1b064576',1,'operator-=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3816a35f8468156d59c239256c12dcf3',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa332fae098e7c6dc23b98bc0026f1070',1,'operator-=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afb3cd302e0b78902c62111dce4494fe8',1,'operator-=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abb884888f14086cc674657677cb4b8bc',1,'operator-=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a38bb89f925eca4f9c042f6ee7a2c0193',1,'operator-=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac30c580713f354916088a7dc049ae4cd',1,'operator-=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a715c824ee8c87e0256114a85624d9949',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7bc91aaaf476a37063264d1d53d862cc',1,'operator-=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab155f418f15cabd86ff942c6f9472ddb',1,'operator-=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aaa66dc6d7b2c5efbfaa97ca9c7872bd8',1,'operator-=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a696978d9401e09200045b2d8aad045c2',1,'operator-=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae998d8f423a9fb73405cfbd4b836bc72',1,'operator-=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a279d09ab8542f1c1a8dc8173b65946b6',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a491dadfae957cd7cc0c36188d910f6f6',1,'operator-=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9a837c3b9c4e42f53d7cd1ed0d266e2f',1,'operator-=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acf7af2284269544064b68e807064bba4',1,'operator-=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a28d297705e29009197418546ef435393',1,'operator-=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a948579a4d9ba276523190b03b09578fb',1,'operator-=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a4b98a0a11db5b77cf9168df37c8bc7',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31a3d8f2ff8038f7e0d717845c039808',1,'operator-=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1dac193d9f1c8c0eb4473441895f8c58',1,'operator-=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad2817d53fdd4b112babfb6f0b38c8f39',1,'operator-=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa705d87cf4b78e9d7c6b07dd0c66cac6',1,'operator-=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a542affc376726840647a6e93acf2c1a7',1,'operator-=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#add18cfe4c0d38e95c6dff6bab3e7a932',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab1de7e7e7304ff3598925d2e69134764',1,'operator-=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0d3fb52437c677c5d0f1a3642384b15c',1,'operator-=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adda64cae388baac1f138b06dc8595237',1,'operator-=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af20874a61c6c3f4c3fd045a96e806644',1,'operator-=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a8b8a55690df46d97fcfc2a60120783af',1,'mlx::core::operator-=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#ab03949b1f60fa035ce454a894cd73ae9',1,'mlx::core::operator-=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#adaf70bbfb3667df0d08fd3c99896e20a',1,'mlx::core::operator-=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a321c98e5a78621d3c9a3895f707f2f1c',1,'mlx::core::operator-=(float &lhs, _MLX_Float16 rhs)']]], + ['operator_2f_27',['operator/',['../namespacemlx_1_1core_1_1simd.html#ac86a54a5e2ccc79bc92739f143bc0bef',1,'mlx::core::simd::operator/(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#ac5d10f465c21ab259041042ff0159187',1,'mlx::core::simd::operator/(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a18a2689f4ae197c5b204fe9b3370da4c',1,'mlx::core::simd::operator/(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a1d45c3b97cecfff86a2e43ae1f7fa185',1,'mlx::core::simd::operator/(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value/b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a89be64949908f19dd42aa7e38b320b0c',1,'mlx::core::simd::operator/(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a/b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a1c61bd3ac3ec5d8d2da65b45d59f543e',1,'mlx::core::simd::operator/(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value/b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#aab8837750c84794369e630d8ea0b408c',1,'mlx::core::simd::operator/(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a0585ea196b665710115e48b7ebef0fc1',1,'mlx::core::simd::operator/(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a075f637ff3f983ada0fd6288ab8d91d7',1,'mlx::core::simd::operator/(T a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#ae6a708f67d6fd9b0962aa8877cec6d35',1,'operator/(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9f16a44e1c9836ca57edc1d7b93b5d7c',1,'operator/(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aacaedf12f862c76457133336dd6fc446',1,'operator/(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a584a513596de20663dad951a5b81695e',1,'operator/(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad8f7b11669736fbd6ed2e28211d877d4',1,'operator/(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a59515695ebc48844345fa5120511aed1',1,'operator/(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a8c8ac6736440fdca366ebdefe2a12b9f',1,'operator/(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad6859b04680d0d26d75fd6c4dd74ee24',1,'operator/(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4720cc79ab2b8e39952ea9ef20e51250',1,'operator/(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a72d10ec0e62949247da129eb3a83fb9b',1,'operator/(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad6399ba2b8708899739b4cdbb44add8d',1,'operator/(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a998b1ba877a606aedf722ab46b290403',1,'operator/(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa3277ae33976c70f7bd937ddff027b72',1,'operator/(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa708a970a200822c99c0489f389469fa',1,'operator/(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a6bde717aca2051499f73a3eee199bfdd',1,'mlx::steel::operator/()'],['../group__ops.html#gaeedf77f722b394429f1a7f6c367883bf',1,'mlx::core::operator/(const array &a, const array &b)'],['../group__ops.html#ga7366ec7f453be2a4dc449f0faa1bf554',1,'mlx::core::operator/(double a, const array &b)'],['../group__ops.html#gadfb324ae9b4feb2c7ea0ac6ade639f38',1,'mlx::core::operator/(const array &a, double b)'],['../namespacemlx_1_1core.html#a7573ac3b93ddecd69e9c88a26fc84ba9',1,'mlx::core::operator/(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a40e868dad70401d9aa9ee9c32235c315',1,'mlx::core::operator/(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a7587c28fbd2023b134e5fc12bb0dde23',1,'mlx::core::operator/(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a92cdd377c408becf4cf83c1ee9b7085d',1,'mlx::core::operator/(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aef89566301cb133d98c8e7bdd2b7bec6',1,'mlx::core::operator/(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a94e7b51185590492b46916685641276f',1,'mlx::core::operator/(_MLX_BFloat16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a04584788c08180835219d0ea1e2b97b1',1,'mlx::core::operator/(bool lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad5af96e2ff09d207eb1e1980fe3e7c2d',1,'mlx::core::operator/(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ac2217bf760038cd011781158923149ed',1,'mlx::core::operator/(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aea414c04bddc4b9b609262e97398f1b4',1,'mlx::core::operator/(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a27fe23230cd082c0363b9451b731ce6b',1,'mlx::core::operator/(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abdd9bb8fb4411e5924f3eb7ef1bb52f8',1,'mlx::core::operator/(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a50bae338a7353f8b0ed3441071bb0cf6',1,'mlx::core::operator/(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aab26a3284dd3ac7d47c8b5b3a3290ce3',1,'mlx::core::operator/(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a749f48db01de38f259a0c6750a97fa77',1,'mlx::core::operator/(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a32a6a08a2a4652975b0a1bd1fcf3eafd',1,'mlx::core::operator/(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a4b66fb38ddc5cc0c2489583d5c499602',1,'mlx::core::operator/(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a45726f1905b709cf8253e6efa046027b',1,'mlx::core::operator/(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afd4170c1e364384f30e6bae341146fa6',1,'mlx::core::operator/(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aef85739d150b9d5609973da8a3f1086a',1,'mlx::core::operator/(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af52a941f8ed9b25eec91402c7b9e281f',1,'mlx::core::operator/(_MLX_Float16 lhs, bool rhs)'],['../namespacemlx_1_1core.html#a477cade78296bc85894170f62db68870',1,'mlx::core::operator/(bool lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a22f5a2257e11423fc2fe18e2dce91590',1,'mlx::core::operator/(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a640d3574dfe6ad934c720ae8bdd78bfa',1,'mlx::core::operator/(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a6f65d8fd0cdddc96fc01f6af95804873',1,'mlx::core::operator/(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a517019d42d4e426b7b98e1c719bb47ce',1,'mlx::core::operator/(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a0beb7a223c542015a4eff4aed814a9dd',1,'mlx::core::operator/(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#abc9b1bd5018d46514bc19d23db2e5063',1,'mlx::core::operator/(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af22937df654ddbd6e398ef12764d18c0',1,'mlx::core::operator/(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a775aed5f49b530c57e71cbac81404d45',1,'mlx::core::operator/(uint64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a97efcd96d6be666e5608034ae77289ef',1,'mlx::core::operator/(float16_t lhs, bfloat16_t rhs)'],['../namespacemlx_1_1core.html#a899851f85dbddd96f9d36319b82542a0',1,'mlx::core::operator/(bfloat16_t lhs, float16_t rhs)']]], + ['operator_2f_3d_28',['operator/=',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5aa3b8c68a2b58d41ea33eaabbf83095',1,'operator/=(device _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a90a1c5130db515db48624d8587edbb91',1,'operator/=(device float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a65f30a2dc199134e35bc7c5d431b2263',1,'operator/=(thread _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7172d84db640e6c49dff0d08dd64b53e',1,'operator/=(thread float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acf7cb9927bf09022088401923f2e1916',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a86b2a001cbec0d3a8d762a3c7ff47b0b',1,'operator/=(threadgroup float &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a744f72ba83522fe3cc2a49a007b42543',1,'operator/=(device _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a77c678665b34df7652dcde053ca73185',1,'operator/=(device half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae0614b6b199d8a65ae95d4621b118b82',1,'operator/=(thread _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa846fde89c7d2d18b18ef180a8a9c8a3',1,'operator/=(thread half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08e778be18e4a291c108fcc528b981d3',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a6b9e49ad9ea256d2d0220c0d81552602',1,'operator/=(threadgroup half &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab933bc3cdf9adfea10ab9dba5292c812',1,'operator/=(device _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a25e7c5d2ecf3375756d59074f333858f',1,'operator/=(device int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ae4a80fde67eea9a0a37b2803946544',1,'operator/=(thread _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a912393b7208fa45bd1e87f30b218b68b',1,'operator/=(thread int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a18963246f2b640874bef6dca7049f64d',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0e2c2c2cb50b3a55ff213f18978aca35',1,'operator/=(threadgroup int16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a64f1136b17006f168ef837e17240814f',1,'operator/=(device _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae46d75b8046d557452d74513f1106710',1,'operator/=(device int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08d2460e259b9106d90d889481ad60d5',1,'operator/=(thread _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f7fd418408806ef498745c6fdb2c062',1,'operator/=(thread int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac703495cb370b52526a5a2d36ae26038',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4ca11d43174baf0a729f93b35eabcbea',1,'operator/=(threadgroup int32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9f835a0a80c411580c97b65fdc5bdfd3',1,'operator/=(device _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a17f47ec9cff60f8e1b3477a2793b7ac0',1,'operator/=(device int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5be23e296bbed3a885586a6424b1666e',1,'operator/=(thread _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afba39221eb54e272aae79910b3cd7ef5',1,'operator/=(thread int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac057d95a2bf087575584aa6f9a2c6bf5',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab986ae2cec780a1f494b7b4468b7ba11',1,'operator/=(threadgroup int64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a44522c2304c6396bbe6b9d32000f4b6f',1,'operator/=(device _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aef8e7e499ea9d432aa743d83c076f945',1,'operator/=(device uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3a0a3edbf1ba2314551454059c3f422b',1,'operator/=(thread _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acb9f0aef9fbdfde8a4f46e33b0d6c52f',1,'operator/=(thread uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a303dfcc81ffd355f866f863d7d9f0fa5',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint16_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a359edd4bcb8776861ceb26a3005624c0',1,'operator/=(threadgroup uint16_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#adc9f32cc6f40768df4285fba2e4783c7',1,'operator/=(device _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae71f66d814a03f6377c9d86cf0a2b5d7',1,'operator/=(device uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad0125b6baba3065a87a174ec27aa9a61',1,'operator/=(thread _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5cc74ad3e522d7104e6e2117751151ad',1,'operator/=(thread uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab3b594321fb42b0c2da99954d1e0976c',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4a0023e2fd08875156cd6ef747fbb5cd',1,'operator/=(threadgroup uint32_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a4358ee606e66ba2081fcf94f9c3b5915',1,'operator/=(device _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ad1e7ef6f065695d4b1d017547b60ef62',1,'operator/=(device uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a284dfc702f0f67b9c233b87162eeabdd',1,'operator/=(thread _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab8f211ea896fc5190004f3ad6ad8932f',1,'operator/=(thread uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7e1bcf3bc06cbcbc304c0cdf729802bc',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abbe42648a46092137b303ccd08f7df86',1,'operator/=(threadgroup uint64_t &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af1a12a1efb618a57da6dd41ae18cb53c',1,'operator/=(device _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a94686039356dfa9aa45608a8b0562fdc',1,'operator/=(thread _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa251d6483d3b099d1b5311fbe6f0bce2',1,'operator/=(threadgroup _MLX_BFloat16 &lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1core.html#a045ff27257cb6d8ab7a94771ba5a17e6',1,'mlx::core::operator/=(_MLX_BFloat16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#a58112951a56a0f9f8c90b60fe74f9508',1,'mlx::core::operator/=(float &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae736defc89a04fbaf7627ad2695bb838',1,'mlx::core::operator/=(_MLX_Float16 &lhs, const float &rhs)'],['../namespacemlx_1_1core.html#ab1f260710251256ef737dd59be9e143c',1,'mlx::core::operator/=(float &lhs, _MLX_Float16 rhs)']]], + ['operator_3c_29',['operator<',['../namespacemlx_1_1core_1_1simd.html#a6cd6e41660608d17ca8d38658d5e385c',1,'mlx::core::simd::operator<(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#ad9bebf95b37fa0c6517be82af5ccd4eb',1,'mlx::core::simd::operator<(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ac962a14c88c87082fc70a9c0370f35b0',1,'mlx::core::simd::operator<(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a23b59272b0760326844fffe20db9b3e2',1,'mlx::core::simd::operator<(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a01259c9188e6ecd48979cdc2fd766372',1,'mlx::core::simd::operator<(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#acf35d81032bb9043804fd1de43540f60',1,'mlx::core::simd::operator<(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#a3f63139b42029ba8d7b3b8ef10f5ac96',1,'mlx::core::simd::operator<(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#aaf29bfdcfdbb9a0acb9f4a6ed622868f',1,'mlx::core::simd::operator<(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a9e0c9b3e986809be5e87aacc4612bb8e',1,'mlx::core::simd::operator<(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#a67674e32596a9dae2258bb8e0e6a2058',1,'operator<(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9ef6a57b7185e9ca49e255fec1a44e25',1,'operator<(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aab02c65bc38ea66335b2192ead4095a8',1,'operator<(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae91686513e284bcc9635833744bbdda1',1,'operator<(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2486f3b5de85b0d57f458d8f21f82b42',1,'operator<(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a435a2aec4c777b4b184ff5d24992e8a1',1,'operator<(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abdd04257e6a73883b5f56f1186d0e906',1,'operator<(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a69984aaa05ae1d4fccccf7f57e8ecb4a',1,'operator<(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a501cc01d5bf15d9f03aa28545f9624ea',1,'operator<(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1b029e4ca72125a5f9471f582c819705',1,'operator<(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0736a76f56578d26ba1422dc8b744a18',1,'operator<(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a24b1fa8998c892f90f8dde7c34fb10a5',1,'operator<(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af80ff2020ec2c4b406c5fdae3fe55e63',1,'operator<(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac03f6eefb836373d37dc280b0d813d78',1,'operator<(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#adb5f24b57d98214fc215a06475f21412',1,'mlx::steel::operator<()'],['../group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3',1,'mlx::core::operator<(const array &a, const array &b)'],['../group__ops.html#ga1ef8ea11cf15ce628c54201fa42748ef',1,'mlx::core::operator<(T a, const array &b)'],['../group__ops.html#ga95e72226dc7a79c40b3d16f990922050',1,'mlx::core::operator<(const array &a, T b)'],['../namespacemlx_1_1core.html#a987d631e1508e8df55d98ddd57e4d086',1,'mlx::core::operator<(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad3fb46370cd8f0992866fad9e2c64a3c',1,'mlx::core::operator<(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a3026691bf7ee5095243a8611bf3411aa',1,'mlx::core::operator<(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0d42d6c1d5f77a96e2f296b8ebd79ee6',1,'mlx::core::operator<(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ab5ce08a7de0a0ca00d61f7a7f8ea3ab4',1,'mlx::core::operator<(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abce8b7f24b61e5ec0f9a3afe20845caf',1,'mlx::core::operator<(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#aff97612627ae1ed260c43c0a7af0d306',1,'mlx::core::operator<(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a9119e518234df7923cae2b3802d59bf2',1,'mlx::core::operator<(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#aefb9b05ce8864ada99a920ab32017b89',1,'mlx::core::operator<(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abc55f3676c2d112a6e9ab276bd6b1796',1,'mlx::core::operator<(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#afe6581a2c45f24d7fab1e4006c1e3c70',1,'mlx::core::operator<(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aca1d50cdd9506481dcc4cd1ad4a4f734',1,'mlx::core::operator<(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a310720f513b6a2490e9df80c65f1bfb3',1,'mlx::core::operator<(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a29e457a170b6cefb6ba1e394c96c6f7b',1,'mlx::core::operator<(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#afd4519985b6b207ec41ad8530d1036df',1,'mlx::core::operator<(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae1e41ca94022e43a00cdfc5845102daa',1,'mlx::core::operator<(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ac80f4022bffd95b57526685ce8e1cbc1',1,'mlx::core::operator<(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a3a8f6f0af477788c4f0aa98abfc5f1ab',1,'mlx::core::operator<(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a3728ed9b6cbd152bf675251a0501b466',1,'mlx::core::operator<(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a5b9ad811a5e1358100c5423dd70ea387',1,'mlx::core::operator<(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a5c77e1db83995d3e06a8a26265bce5d6',1,'mlx::core::operator<(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab8a0a3f70664049b35ce1887bd8ff5c2',1,'mlx::core::operator<(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6652d93bfb2d426e261a1712a181a4d2',1,'mlx::core::operator<(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a03758b8d13da2de07cc4f4fc45d2854b',1,'mlx::core::operator<(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a325161b81a9ff179fd37d949780a17ba',1,'mlx::core::operator<(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a92eca79fce8233e4299343eee3996511',1,'mlx::core::operator<(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#adb016662b8f7eb680abfe1a421eabe72',1,'mlx::core::operator<(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3c_3c_30',['operator<<',['../namespacemlx_1_1core_1_1simd.html#ae21cbfd232edd7fe0f6f6c9fa11a354e',1,'mlx::core::simd::operator<<(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a56fccba38270fe3ae9fa7b2ecdeb5e87',1,'mlx::core::simd::operator<<(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a4ecd782ffa497ac7dc2482a232b0dd00',1,'mlx::core::simd::operator<<(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a33232e2342d5a3e542c9428924a25830',1,'mlx::core::simd::operator<<(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value<< b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a50044315dc365f026830416f6b615c77',1,'mlx::core::simd::operator<<(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a<< b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a68e7b952915e629d246d1ffac98b54ce',1,'mlx::core::simd::operator<<(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value<< b), 1 >'],['../group__ops.html#gad656c30f9fd7d9467e405657b325aa7e',1,'mlx::core::operator<<(const array &a, const array &b)'],['../namespacemlx_1_1core.html#a1e5c30e316afa30c14bc48b92afdb794',1,'mlx::core::operator<<(std::ostream &os, const Device &d)'],['../namespacemlx_1_1core.html#a4ddd07021b36c848d6fb1dd9ac276822',1,'mlx::core::operator<<(std::ostream &os, const Stream &s)'],['../namespacemlx_1_1core.html#a0023c267cf81345fad65e7a797954cd3',1,'mlx::core::operator<<(std::ostream &os, const Dtype &d)'],['../namespacemlx_1_1core.html#a1fd58658474fb842d648dcf8f7d9f078',1,'mlx::core::operator<<(std::ostream &os, const Dtype::Kind &k)'],['../namespacemlx_1_1core.html#a123331f01188bd76e37623b63b6b4340',1,'mlx::core::operator<<(std::ostream &os, array a)'],['../namespacemlx_1_1core.html#a4e733bba89760abed32393e085812b22',1,'mlx::core::operator<<(std::ostream &os, const std::vector< int > &v)'],['../namespacemlx_1_1core.html#a5e5bd5c57b1cf19776bdb41e732861d9',1,'mlx::core::operator<<(std::ostream &os, const std::vector< int64_t > &v)'],['../namespacemlx_1_1core.html#a42a19c8442b173606e714364227e7d45',1,'mlx::core::operator<<(std::ostream &os, const complex64_t &v)'],['../namespacemlx_1_1core.html#a57eb97a5eba99a846ac429795e407574',1,'mlx::core::operator<<(std::ostream &os, const float16_t &v)'],['../namespacemlx_1_1core.html#a7db909d54cf07375e89424c32c07a29c',1,'mlx::core::operator<<(std::ostream &os, const bfloat16_t &v)']]], + ['operator_3c_3d_31',['operator<=',['../namespacemlx_1_1core_1_1simd.html#a4d5e4c31af23d2871e09b88c1f6e418c',1,'mlx::core::simd::operator<=(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#ae0fcb84973e4762a543ad3843db4f153',1,'mlx::core::simd::operator<=(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aadd49786edc08f867e592d234327a031',1,'mlx::core::simd::operator<=(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aec6783f79ca181d6782a810ffb267482',1,'mlx::core::simd::operator<=(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a05240b8fd6f54632b676d4b66449f799',1,'mlx::core::simd::operator<=(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a914e821c358e05dfe8d0208888646793',1,'mlx::core::simd::operator<=(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#ad1570f6937d194a09e61d0e3a70ef578',1,'mlx::core::simd::operator<=(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#a46c6ea18a9edd2a9cdba2ab62ca4782c',1,'mlx::core::simd::operator<=(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#accd17f741cab18590fdbe388d4783967',1,'mlx::core::simd::operator<=(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#aee04c9a63c6716a99a027418354debb0',1,'operator<=(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#af469c58cffeab488c681f4b33f02cd05',1,'operator<=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a5a81eae168dfafd299c2b94e3e8558cf',1,'operator<=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f486bf02c6ad5b9b6a96d3450f03e47',1,'operator<=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#acba9efe192d22b7781b4622103c7a944',1,'operator<=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aff100489cc40ad276c2d5d67a9df67db',1,'operator<=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7eac96f64ca42991caf819c8e8c8d2bc',1,'operator<=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a88c11cd37600de5480570da3d2ae5732',1,'operator<=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a08c7d12a0d16565fbf052dba2db8b22d',1,'operator<=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2b9de9624c0a507b4ead85f898ad9daf',1,'operator<=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a28f8d21c5eef047c701cf690ce9c2ef0',1,'operator<=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a14b56c687053ee2432398a25663c068f',1,'operator<=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0f360806708b95a3be400af0b8871b57',1,'operator<=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a80d288f22cadfdf5e904410349e616a1',1,'operator<=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a6cc3bab5e7f6e7c719c82afa90ad2827',1,'mlx::steel::operator<=()'],['../group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6',1,'mlx::core::operator<=(const array &a, const array &b)'],['../group__ops.html#ga150a9be467c9f91482a6d6fc13504bc4',1,'mlx::core::operator<=(T a, const array &b)'],['../group__ops.html#ga624eeccef0cc4b130e1325abfea057cb',1,'mlx::core::operator<=(const array &a, T b)'],['../namespacemlx_1_1core.html#a0066a47cb21223ddebc77992ee874fb9',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2593dbace3ce50e7146d9514726a543f',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a88654bcf6c9728517a2933ca2e29a7c1',1,'mlx::core::operator<=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a5d4f449e9c1699b99fcf894dd15e8af3',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a6b678bea8fdcda1f11c6691b56a15211',1,'mlx::core::operator<=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae8aacc606ea16f018a90eae758830a35',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a25668dea4ffb51c7c00eeecb9530d1d8',1,'mlx::core::operator<=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a084558b6a5487549799c49c37c9e9652',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ade2e2a0daa79d5c52f278f85f03dde2e',1,'mlx::core::operator<=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a750a2d2b4976ad94b08994d081f83445',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ade5a175ff45347689ac4c798d04c8ffc',1,'mlx::core::operator<=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae25e0c01b46612f039313a4825ba6428',1,'mlx::core::operator<=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a5c90f16d8f6edf4b75c96b945b9fa591',1,'mlx::core::operator<=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a8cd6583fa0fc9957f993e00b2ec01d91',1,'mlx::core::operator<=(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a012130a0458cbc30b88365e0e0eab232',1,'mlx::core::operator<=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae8c890bdcffadee8c5dab85c907f57eb',1,'mlx::core::operator<=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a43cb070553c1f2fffb32ef6670e30980',1,'mlx::core::operator<=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ac759b7798d668a99535e59e26d6ba192',1,'mlx::core::operator<=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a70e528a789b5660d98e783b045aaa379',1,'mlx::core::operator<=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a40bd8abb8a4d989ddabbb298518bd7f5',1,'mlx::core::operator<=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a4155d4b0c76f37ab5e0b54f9cd683f35',1,'mlx::core::operator<=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad8bb648d0603a206e0392990c911ca0b',1,'mlx::core::operator<=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ace72a5853f2afd6510dcb97d54fa650d',1,'mlx::core::operator<=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ab38f7a0d3c0809071ff5d3af859018d6',1,'mlx::core::operator<=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a7904b886d7b535a6af0a885d00597323',1,'mlx::core::operator<=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a57952168bd0b54c2677204d4ab1cb6e5',1,'mlx::core::operator<=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a6235dc5f4db517618bb3449b08c96e8b',1,'mlx::core::operator<=(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3d_32',['operator=',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a027b84cddc8d476f736ac1f1a9991fe4',1,'mlx::core::allocator::Allocator::operator=(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a2e971b47339b1d0849a334a902a9df3c',1,'mlx::core::allocator::Allocator::operator=(Allocator &&other)=delete'],['../classmlx_1_1core_1_1array.html#a8acf2b4c75f9b7f79da6675dbc36cf36',1,'mlx::core::array::operator=(const array &other) &&=delete'],['../classmlx_1_1core_1_1array.html#a5c89c2406a610b32943955f9a5060fbd',1,'mlx::core::array::operator=(array &&other) &&=delete'],['../classmlx_1_1core_1_1array.html#ad3277ff68f1336aa217f9cbe40181479',1,'mlx::core::array::operator=(array &&other) &=default'],['../classmlx_1_1core_1_1array.html#a5da41aabecf4c8055b7515341bf57147',1,'mlx::core::array::operator=(const array &other) &'],['../structmlx_1_1core_1_1array_1_1_data.html#a68e9417954fe811b5e41e6317a526748',1,'mlx::core::array::Data::operator=()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a3f42a1362b4a513fa89e7b3dcc570a8e',1,'mlx::core::metal::CommandEncoder::operator=()'],['../classmlx_1_1core_1_1metal_1_1_device.html#ad1d6382fd18a46b1906e1b43e0bd2e73',1,'mlx::core::metal::Device::operator=()'],['../classmlx_1_1core_1_1metal_1_1_residency_set.html#aef97dbbc755940789f99a26164591c45',1,'mlx::core::metal::ResidencySet::operator=()'],['../structmlx_1_1core_1_1_function_exporter.html#a7ec0f53eb2783d5b1953be612e36d5c7',1,'mlx::core::FunctionExporter::operator=()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a957211656a13b4c0d126989a9aba3e25',1,'mlx::core::io::FileWriter::operator=()'],['../classmlx_1_1core_1_1_primitive.html#a6b1be7ea92f3a7bb19875c70259dad6b',1,'mlx::core::Primitive::operator=(const Primitive &other)=delete'],['../classmlx_1_1core_1_1_primitive.html#a50bbddd43e1ba0cf5f127cd7aa756a9e',1,'mlx::core::Primitive::operator=(Primitive &&other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a0a859309a4f192f2679e07f2e4ff4d22',1,'mlx::core::UnaryPrimitive::operator=(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#ab90b2ea80f1d914be03cf44def5db5a5',1,'mlx::core::UnaryPrimitive::operator=(UnaryPrimitive &&other)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ab170dbd2ce34c51e2eeebf5d08e7e2db',1,'mlx::core::scheduler::Scheduler::operator=(const Scheduler &)=delete'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a035ea35f4dd8ee985973080f14029379',1,'mlx::core::scheduler::Scheduler::operator=(Scheduler &&)=delete'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#a0f65b0523b8ddd989f338da6cb2860e3',1,'mlx::core::_MLX_BFloat16::operator=(std::vector< bool >::reference x)'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#abb8cd44ee22b17c55333ff2eb4e13a14',1,'mlx::core::_MLX_BFloat16::operator=(const float &x)'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a608a099bf7116ee608dcfd31ea3ade2c',1,'mlx::core::_MLX_Float16::operator=(std::vector< bool >::reference x)'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a35543c3653d477c46350697fb808373d',1,'mlx::core::_MLX_Float16::operator=(const float &x)'],['../structmlx_1_1core_1_1_command_encoder.html#a3f42a1362b4a513fa89e7b3dcc570a8e',1,'mlx::core::CommandEncoder::operator=()']]], + ['operator_3d_3d_33',['operator==',['../namespacemlx_1_1core_1_1simd.html#a273fcc5387c1c9878e658ba6bc32f00c',1,'mlx::core::simd::operator==(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#a46ede415296683771bb22246a813482a',1,'mlx::core::simd::operator==(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a63768090c16e5dcffccadf550d169abc',1,'mlx::core::simd::operator==(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a7928482ed5d25932be80413c7239125c',1,'mlx::core::simd::operator==(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a51de2acf3dcd55c7c52e3ce7ed6ed9d7',1,'mlx::core::simd::operator==(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a4877ae5406d081680b785a86ad656e03',1,'mlx::core::simd::operator==(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#acafae9e62680565cd1f1c50c64d7ce4f',1,'mlx::core::simd::operator==(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#aa837052ddcb02f4d9bc39b07399b4d91',1,'mlx::core::simd::operator==(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aaacbf6671080409e822fbb218e3fdf00',1,'mlx::core::simd::operator==(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#abfc19f03616441245dfc7726b278f190',1,'operator==(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a49a13b06a325ed3cca4004b6a0cde065',1,'operator==(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a0aa3bfcfab53700488e5f386e6de60d5',1,'operator==(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3936148781ab1c4f33f58d12c116f370',1,'operator==(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae753526b669fba27771089dc809abd66',1,'operator==(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a05a4f197a71d0f16879032f44492bb79',1,'operator==(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae86f5917847b1ec9f313996250f2e0be',1,'operator==(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aab74ec4d33a64b92b908717d500f1ecf',1,'operator==(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac30a2c1fa6f172af903fdeb6a8632606',1,'operator==(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab4e9ad547aa23daa351075e0ecc58fa2',1,'operator==(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa5fa1a8f2b39c3508fe38205469756d1',1,'operator==(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aeadc1f36c6bdc219294ce9341d80afa5',1,'operator==(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a3ae2091ada1e39e857fbc53c97bdb79f',1,'operator==(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac7b4d295f3c7b1e09964f24f306422da',1,'operator==(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#abcc797f27e87e857b41c1a8d33ee2c78',1,'mlx::steel::operator==()'],['../namespacemlx_1_1core.html#a937503d72b66c661bf3f5fdcd98ef97c',1,'mlx::core::operator==(const Device &lhs, const Device &rhs)'],['../group__ops.html#gaa30cf69f3d22f65615f5e1696dd5703f',1,'mlx::core::operator==(const array &a, const array &b)'],['../group__ops.html#gaf115782d009ac2a547fcca395c9ec797',1,'mlx::core::operator==(T a, const array &b)'],['../group__ops.html#ga3ad3ed7aece2650943a35082dbe3a0a5',1,'mlx::core::operator==(const array &a, T b)'],['../namespacemlx_1_1core.html#ac470f937a379d6356c8f567c97cd7481',1,'mlx::core::operator==(const Stream &lhs, const Stream &rhs)'],['../namespacemlx_1_1core.html#aec63a0472cb943fe39f31e7678555572',1,'mlx::core::operator==(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad05311ca8e2f19ffe5849e963837cec7',1,'mlx::core::operator==(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aaaf591cb2188381e6cbd857132d04eb7',1,'mlx::core::operator==(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7ef33c33509ccccf1ab217500e8b3c1a',1,'mlx::core::operator==(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#abec4200a718b7c5ed80b7abcc4447260',1,'mlx::core::operator==(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ad853981b1c5ba69b07d54c7b77055d22',1,'mlx::core::operator==(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a752d6cb4172a9cb91e5da19582329c6d',1,'mlx::core::operator==(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0175beb3de139faa08479a88215b35ea',1,'mlx::core::operator==(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a61da2851cb3beeef28049228346c28b5',1,'mlx::core::operator==(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#aa24713cb9e39bacb516c992eb03d2b2b',1,'mlx::core::operator==(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a6d565dd93c46259f9486d9fdf0969589',1,'mlx::core::operator==(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a24e79a82557861de64dad66d36e6ff30',1,'mlx::core::operator==(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#af27d515ac390d62bd852b73ea759a947',1,'mlx::core::operator==(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ae3e1e8b7a5410e0edf35f31f74295e2f',1,'mlx::core::operator==(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aaa22230a66b15c3e774d8ce45783a746',1,'mlx::core::operator==(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ae2a0bcdc171d7e9745d33e1d9aac4f8a',1,'mlx::core::operator==(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a331ec62442a8d3eb8ccba7b4de5168d1',1,'mlx::core::operator==(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#acfcaefe0990eb3533e2b11a6f2657492',1,'mlx::core::operator==(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a8d48dbd49cccff07777affb2a412058c',1,'mlx::core::operator==(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a88eae27edd22fa4418776672023cb276',1,'mlx::core::operator==(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a188b363f633ea360407b3f9cf4e1f1a6',1,'mlx::core::operator==(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ae065fe5c42c1a333d7858d19f6434fa9',1,'mlx::core::operator==(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a2f98db199deb6d7a82551fa4afec655a',1,'mlx::core::operator==(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a85f83add412cb320b5cd1c3da6aadbd5',1,'mlx::core::operator==(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7e2cee66c3ca1b56f4f3d7fd1d6e0be1',1,'mlx::core::operator==(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#ad436557da5c7fea71fc58182a876cfe5',1,'mlx::core::operator==(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3e_34',['operator>',['../namespacemlx_1_1core_1_1simd.html#abd37e62eff936a64677b5aba787b4d18',1,'mlx::core::simd::operator>(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#a71a6902e729e3facdc609e93cd12d485',1,'mlx::core::simd::operator>(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ab7b291b3559792e18208e17432d25342',1,'mlx::core::simd::operator>(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ad8b67f9ced9c7f3cb472b9c3df817f08',1,'mlx::core::simd::operator>(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a4113a94fb8dcd0d88f14ec9d82089508',1,'mlx::core::simd::operator>(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#ac971bfa5c7ec8abc432eab5f3c5646aa',1,'mlx::core::simd::operator>(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#a35d875fa7bce02a6171f37240a346e1d',1,'mlx::core::simd::operator>(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#acf2391cc4d945887d7820501ba14ba89',1,'mlx::core::simd::operator>(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#aa17e031474fa87f6ea7855257dcc9ece',1,'mlx::core::simd::operator>(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#a032a8d3eec2384c9f03066f7fd945995',1,'operator>(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae394c0a10e47d1d047854a888402eb57',1,'operator>(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ab9cd098786d2f4c855c42e4a6f30ab3e',1,'operator>(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a55600f3b9859e2891e0e0b5690867b72',1,'operator>(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#afd7cdb8ed2a9820efe9cf322c06f188c',1,'operator>(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a31bbdbe0b62b90a4d6ea4bb0a7db586b',1,'operator>(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a68125e66f74eaffe5ea9267638ce870d',1,'operator>(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac89eb6b29edad8cca63727ab97171c29',1,'operator>(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a74e477567c9477c2cf0684f81ef4498f',1,'operator>(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2d37130b6fd79b425f5ba92b65e36bed',1,'operator>(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a41d55d167e9dc63bf29d15e0ff004869',1,'operator>(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aa95f9ebfdab3c5f524775651362ce914',1,'operator>(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2826bd301bb5393473ccd363f2052c0d',1,'operator>(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a62a512d0edd894759c69f724b970fbdb',1,'operator>(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#a7512eadda6160e4c9d9e6aa4049fac20',1,'mlx::steel::operator>()'],['../group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb',1,'mlx::core::operator>(const array &a, const array &b)'],['../group__ops.html#ga32e106e794e2c32e4e7decee2df2477f',1,'mlx::core::operator>(T a, const array &b)'],['../group__ops.html#ga96552b90e89923c5d2064cc427775ec5',1,'mlx::core::operator>(const array &a, T b)'],['../namespacemlx_1_1core.html#aedc4e9df4bf71c0ac34fcfae60cdf550',1,'mlx::core::operator>(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a14c188303d09b97867bcfd34519aa4a6',1,'mlx::core::operator>(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#ac97736fadafa7efa201624d0e1128ee8',1,'mlx::core::operator>(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a3c41a304126bc225bdc68062d1eb6e7e',1,'mlx::core::operator>(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ab594f3ae1ee13227fae940fef0d00cb9',1,'mlx::core::operator>(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a01dabc077a872c115a9a9ccd95f1acec',1,'mlx::core::operator>(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#adabbd8768d216873617768249473a5c7',1,'mlx::core::operator>(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#adae1b14669d27ce1fe0c214771c07b77',1,'mlx::core::operator>(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#ab03a22961d99fa12d3e74b3116e94e8f',1,'mlx::core::operator>(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a42011a27a3d23a60be5be44ee7cac87c',1,'mlx::core::operator>(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a50f6a94bb36d89cf28817aff88ab89c8',1,'mlx::core::operator>(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac173de50ee57b1b066d49363ba978c53',1,'mlx::core::operator>(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#ab09f1b4879aa3190c2f66c9bd1224021',1,'mlx::core::operator>(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a91eb6ca854217424129a55ae95a123b5',1,'mlx::core::operator>(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a58d5795d8312599d101ae16f194e4a2a',1,'mlx::core::operator>(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#aafa3bbeda78610c4285f3e57042268f3',1,'mlx::core::operator>(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a8a928d76a6fbf3d336296401e14617a4',1,'mlx::core::operator>(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ade2f9222fd433cd4d673c6182f256235',1,'mlx::core::operator>(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#ae24c337810c841ff23e327efde7045e1',1,'mlx::core::operator>(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acf401ede354fcc998b13ea6442994d7e',1,'mlx::core::operator>(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#a2bb28a9a0894a73ae1b27e7f4da0841a',1,'mlx::core::operator>(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a09d631e8a85fd7ae72e1a868b8f9b9cb',1,'mlx::core::operator>(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a49421ea65b5a98df080d75b1636b2157',1,'mlx::core::operator>(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a692ce931b660415e17f92d18a8e0d446',1,'mlx::core::operator>(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a579bb87b3ede5663d7cd68c7c0f6fb9e',1,'mlx::core::operator>(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#af810587a17e692f4eec256d3c3cd27de',1,'mlx::core::operator>(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a50f4177d3ca03a95fc2614e100c7391d',1,'mlx::core::operator>(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3e_3d_35',['operator>=',['../namespacemlx_1_1core_1_1simd.html#a87e11ab36aae3328fe3d5230bdf31692',1,'mlx::core::simd::operator>=(Simd< T, N > a, U b)'],['../namespacemlx_1_1core_1_1simd.html#a4e65febbfa8b4df2970c1d78801b3c66',1,'mlx::core::simd::operator>=(T a, Simd< U, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a673b4d8d228f35f06cf5b882335f04d5',1,'mlx::core::simd::operator>=(Simd< T1, N > a, Simd< T2, N > b)'],['../namespacemlx_1_1core_1_1simd.html#a530ac8728e4d7e7be2482d5b2467906c',1,'mlx::core::simd::operator>=(Simd< T1, 1 > a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#ac7f3848b48c8e23c71c85fcc9909b933',1,'mlx::core::simd::operator>=(T1 a, Simd< T2, 1 > b)'],['../namespacemlx_1_1core_1_1simd.html#a034d7b57cb3c6ca711c573515327d1a8',1,'mlx::core::simd::operator>=(Simd< T1, 1 > a, T2 b)'],['../namespacemlx_1_1core_1_1simd.html#a8d7dcf1914ce8fe8518d84b0f2a5fe91',1,'mlx::core::simd::operator>=(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#aecdc08fcc70b158749a93a7a0f688aa3',1,'mlx::core::simd::operator>=(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#ab9097573af69cc66d1427d0f52507e7a',1,'mlx::core::simd::operator>=(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../backend_2metal_2kernels_2complex_8h.html#aafbd686c180398c98b33d7643f893a46',1,'operator>=(complex64_t a, complex64_t b): complex.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a430dd11fbf4c6f39bc1506ab43b2341f',1,'operator>=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a64f6787a96386246f83a8981d274150e',1,'operator>=(_MLX_BFloat16 lhs, float rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1a788f82212afad30e4c2ee40f1c313c',1,'operator>=(float lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ae88617c4a012c5dc12781a349a28c886',1,'operator>=(_MLX_BFloat16 lhs, half rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a467a88531150a4d9d30fce07c49c126e',1,'operator>=(half lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a9e21c5ea9dd724dc2ca8c54ad908f09c',1,'operator>=(_MLX_BFloat16 lhs, int32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2f6286d222e2176bcbdc824c5d598100',1,'operator>=(int32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#abec53064aa96265385ecc57de5fbc74c',1,'operator>=(_MLX_BFloat16 lhs, uint32_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#ac766839f8f9e4863e8e18418c342c875',1,'operator>=(uint32_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a2807fa6862b0f9689c81199b1e695ed8',1,'operator>=(_MLX_BFloat16 lhs, int64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#aee3ae0d0d1f941463b06eca0bf041b2b',1,'operator>=(int64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a523eda93c809733368e2b45382d2add6',1,'operator>=(_MLX_BFloat16 lhs, uint64_t rhs): bf16.h'],['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a1f4e90909ac1c7280f4c7d1977c55fb7',1,'operator>=(uint64_t lhs, _MLX_BFloat16 rhs): bf16.h'],['../namespacemlx_1_1steel.html#aa3c95c60cf69603705bb4636de547bcb',1,'mlx::steel::operator>=()'],['../group__ops.html#ga3a41895f25ed083a36994d95fa102546',1,'mlx::core::operator>=(const array &a, const array &b)'],['../group__ops.html#gaf509f2cb3b18963232f20d6c3bd229b2',1,'mlx::core::operator>=(T a, const array &b)'],['../group__ops.html#gafa0eb25d5978674bfc9e59d4145ec590',1,'mlx::core::operator>=(const array &a, T b)'],['../namespacemlx_1_1core.html#a8494764f5c686743ede66dc76d85d955',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a019df48807b506d9995856684bf7797a',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, float rhs)'],['../namespacemlx_1_1core.html#a96ab6405430efb887cdb5c828cb67d6e',1,'mlx::core::operator>=(float lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ac18be72269b1bcfb0249cc00a0600681',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, double rhs)'],['../namespacemlx_1_1core.html#aeb879815228efbd2c8f80986e1c8d41f',1,'mlx::core::operator>=(double lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0051156f6a568f58cd54850f746fb507',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ae93556906e115625ed1b62d36cf21b70',1,'mlx::core::operator>=(int32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#ab81ad16e3be591dfc9e42ac3c19b055f',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a6cfe9b03e7c5f1eb9374208a552c3cc9',1,'mlx::core::operator>=(uint32_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2f5add83812fb137dd9226c6c01e45d5',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#ad1014a836e7ce9301de8588eef1e89ee',1,'mlx::core::operator>=(int64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a17791561434dc995de9f268d145c0ed1',1,'mlx::core::operator>=(_MLX_BFloat16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a3755925b24a903045937464be117de2f',1,'mlx::core::operator>=(uint64_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a6262aeb513d27fc8313293b261e72abb',1,'mlx::core::operator>=(const complex64_t &a, const complex64_t &b)'],['../namespacemlx_1_1core.html#a6feb4b3ea511b0eda4d1ec9725f3fb4c',1,'mlx::core::operator>=(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a03b3f7fcb755ec075985ab26336926f0',1,'mlx::core::operator>=(_MLX_Float16 lhs, float rhs)'],['../namespacemlx_1_1core.html#aecfbf5ef4872ae447eb4a374e4db28e4',1,'mlx::core::operator>=(float lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ae4690f349b2483f5d1a4b75aba67399f',1,'mlx::core::operator>=(_MLX_Float16 lhs, double rhs)'],['../namespacemlx_1_1core.html#a667e95146dd5199e67bcb121b984b1f0',1,'mlx::core::operator>=(double lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a3375f1562f148bdc07451f2b6e54e6df',1,'mlx::core::operator>=(_MLX_Float16 lhs, int32_t rhs)'],['../namespacemlx_1_1core.html#ae83df12368cb07ccb1c10c1117ff3922',1,'mlx::core::operator>=(int32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#ad41251938cf852b5560c1180944ebb49',1,'mlx::core::operator>=(_MLX_Float16 lhs, uint32_t rhs)'],['../namespacemlx_1_1core.html#a4ddb5ef0b88929086f9b09729fda0dde',1,'mlx::core::operator>=(uint32_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a0908a61ab261aff726922b33fa6ed159',1,'mlx::core::operator>=(_MLX_Float16 lhs, int64_t rhs)'],['../namespacemlx_1_1core.html#a0fdadf87edd8a0a57c63953fb0ebe053',1,'mlx::core::operator>=(int64_t lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a47c82778e43032c0bbf5d59407e81dc9',1,'mlx::core::operator>=(_MLX_Float16 lhs, uint64_t rhs)'],['../namespacemlx_1_1core.html#a14e6c43b924eacca1b2dac1d5d00ca2b',1,'mlx::core::operator>=(uint64_t lhs, _MLX_Float16 rhs)']]], + ['operator_3e_3e_36',['operator>>',['../namespacemlx_1_1core_1_1simd.html#a6e45c9c2f0591d9d5dd37a07ebcc3c2a',1,'mlx::core::simd::operator>>(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#aa35a2aab733e4bfc80a9f4e3f508daee',1,'mlx::core::simd::operator>>(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#aebf93b8179621e83bb3f3c4a8816eca8',1,'mlx::core::simd::operator>>(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a1108d186d57c2010c743d3f9297befc7',1,'mlx::core::simd::operator>>(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value > > b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a9ac36abfb7dffc7ad24b4d0c295452e5',1,'mlx::core::simd::operator>>(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a > > b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a4bf8c887eb6943563ceb1e603d1325b1',1,'mlx::core::simd::operator>>(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value > > b), 1 >'],['../group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a',1,'mlx::core::operator>>()']]], + ['operator_5b_5d_37',['operator[]',['../classpocketfft_1_1detail_1_1arr.html#aea0bd899b19e03f54dfd6c188727061a',1,'pocketfft::detail::arr::operator[](size_t idx)'],['../classpocketfft_1_1detail_1_1arr.html#a99c54f96bc79c7cdd8925c1663462842',1,'pocketfft::detail::arr::operator[](size_t idx) const'],['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a71b02f67c47b24adb296eafd2c7a3598',1,'pocketfft::detail::sincos_2pibyn::operator[]()'],['../classpocketfft_1_1detail_1_1cndarr.html#ae4852d1fe936a5d61832b507816c7054',1,'pocketfft::detail::cndarr::operator[]()'],['../classpocketfft_1_1detail_1_1ndarr.html#a2b2c4e205e8b5c32c9fe55dfd7b8c8d8',1,'pocketfft::detail::ndarr::operator[]()'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a235268dc56eb1bb5b86cd3aade67b77c',1,'mlx::core::simd::Simd::operator[](int idx) const'],['../structmlx_1_1core_1_1simd_1_1_simd.html#ae877ce4884241399de8e28090441f557',1,'mlx::core::simd::Simd::operator[](int idx)'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a4b24316469cd9ecc88f8c073ab1a862e',1,'mlx::core::simd::Simd< float16_t, N >::operator[](int idx) const'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a97043111a44318b5eb68977ecacbb638',1,'mlx::core::simd::Simd< float16_t, N >::operator[](int idx)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a235268dc56eb1bb5b86cd3aade67b77c',1,'mlx::core::simd::Simd< T, 1 >::operator[](int idx) const'],['../structmlx_1_1core_1_1simd_1_1_simd.html#ae877ce4884241399de8e28090441f557',1,'mlx::core::simd::Simd< T, 1 >::operator[](int idx)'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a235268dc56eb1bb5b86cd3aade67b77c',1,'mlx::core::simd::Simd< float16_t, N >::operator[](int idx) const'],['../structmlx_1_1core_1_1simd_1_1_simd.html#ae877ce4884241399de8e28090441f557',1,'mlx::core::simd::Simd< float16_t, N >::operator[](int idx)']]], + ['operator_5e_38',['operator^',['../namespacemlx_1_1core_1_1simd.html#a25b3de1947dbab7c4864b41ec226453b',1,'mlx::core::simd::operator^(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a93e69a8170b8fe14f0a3188b4e8ccd49',1,'mlx::core::simd::operator^(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a16c4a2c8fc59a2e2fcc05db243289706',1,'mlx::core::simd::operator^(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a369178519e0e91fa936c0fd4aa9ee109',1,'mlx::core::simd::operator^(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value ^ b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a5b877b5eb7044d9b2a42a9af4af21f01',1,'mlx::core::simd::operator^(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a ^ b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a16fa3c809e46b5cae3e8abfaf98199a4',1,'mlx::core::simd::operator^(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value ^ b), 1 >'],['../group__ops.html#gac3a6fe18694e84b3d63458e9553ac181',1,'mlx::core::operator^(const array &a, const array &b)'],['../namespacemlx_1_1core.html#ae36ea40b8477bfa12d41aae8245225c9',1,'mlx::core::operator^(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a03fc96696f5c6d9411841889d05f4670',1,'mlx::core::operator^(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a55130edf926366db0d6207989e609b7c',1,'mlx::core::operator^(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a0b75198f364d742a1c25dd13e398f2c2',1,'mlx::core::operator^(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a7f205f1b10b23180a23bf2be4bb726b1',1,'mlx::core::operator^(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a9edfe65f3c6da583c7b109290ec94b22',1,'mlx::core::operator^(uint16_t lhs, _MLX_Float16 rhs)']]], + ['operator_5e_3d_39',['operator^=',['../namespacemlx_1_1core.html#a97cb7d3eac404a442e84656cefe7cfb4',1,'mlx::core::operator^=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#abcfd2d9615c96561fd44dfb9c341cf8e',1,'mlx::core::operator^=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ae78083d766b9cf6f87cded341bbcd63e',1,'mlx::core::operator^=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#acf36c10779fbf1efbe1e6a7fd41176cd',1,'mlx::core::operator^=(_MLX_Float16 &lhs, uint16_t rhs)']]], + ['operator_7c_40',['operator|',['../namespacemlx_1_1core_1_1simd.html#ab2b540d7329491000e7722f9b3ef797d',1,'mlx::core::simd::operator|(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#a0cd57bba23daed624df5e2b06b676dca',1,'mlx::core::simd::operator|(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#acd57dc91aa205d9d3f8804df4261a7fb',1,'mlx::core::simd::operator|(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a60805b5f57ddbbf74f700b54cd3fc4f8',1,'mlx::core::simd::operator|(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value|b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a495d15a18ee4a6dda22e37e8dc02e45b',1,'mlx::core::simd::operator|(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a|b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a6449faa1666afe1186d55b61bb3e5b5a',1,'mlx::core::simd::operator|(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value|b), 1 >'],['../group__ops.html#ga52392a2a98f09a80da8d338c4908bd02',1,'mlx::core::operator|(const array &a, const array &b)'],['../namespacemlx_1_1core.html#af84ed854132c1514dca5a524fdb7ed05',1,'mlx::core::operator|(_MLX_BFloat16 lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a7423aac70f9f2e3fb6a5c9a3fc96f703',1,'mlx::core::operator|(_MLX_BFloat16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a19805f505cb7ac72bfab66c339ea7900',1,'mlx::core::operator|(uint16_t lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a2d933573edf4ed305fddd8a0caef1ee8',1,'mlx::core::operator|(_MLX_Float16 lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#afab3d4eb1b36a276922879ce6e44b7f5',1,'mlx::core::operator|(_MLX_Float16 lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#ab132729fa6912d22a8e402057eb4ba12',1,'mlx::core::operator|(uint16_t lhs, _MLX_Float16 rhs)']]], + ['operator_7c_3d_41',['operator|=',['../namespacemlx_1_1core.html#a8e1d21375ae4b89b3cbea3a46d262abd',1,'mlx::core::operator|=(_MLX_BFloat16 &lhs, _MLX_BFloat16 rhs)'],['../namespacemlx_1_1core.html#a28d6c2f89e73b7b874dd1f67f853a96f',1,'mlx::core::operator|=(_MLX_BFloat16 &lhs, uint16_t rhs)'],['../namespacemlx_1_1core.html#a2d8470b69cbbeefece08d3ffd46c0082',1,'mlx::core::operator|=(_MLX_Float16 &lhs, _MLX_Float16 rhs)'],['../namespacemlx_1_1core.html#a359c6257097a304c00d41d64296ef4c9',1,'mlx::core::operator|=(_MLX_Float16 &lhs, uint16_t rhs)']]], + ['operator_7c_7c_42',['operator||',['../namespacemlx_1_1core_1_1simd.html#ab380b8f73672727a38ea0931e731fe4a',1,'mlx::core::simd::operator||(Simd< T, N > x, U y)'],['../namespacemlx_1_1core_1_1simd.html#ac34f6b278627949d2ee68cdbf3d2f50f',1,'mlx::core::simd::operator||(T1 x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#ab2bc61c02b9096163e9db91a3f88788f',1,'mlx::core::simd::operator||(Simd< T1, N > x, Simd< T2, N > y)'],['../namespacemlx_1_1core_1_1simd.html#a7a1c3be1c37d41e450469f2e98cd9dde',1,'mlx::core::simd::operator||(Simd< T1, 1 > a, Simd< T2, 1 > b) -> Simd< decltype(a.value||b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a0c8bd67982681ecd53cd8d739be3a5a9',1,'mlx::core::simd::operator||(T1 a, Simd< T2, 1 > b) -> Simd< decltype(a||b.value), 1 >'],['../namespacemlx_1_1core_1_1simd.html#aad2d440fbb9e5478b5ed24400a859942',1,'mlx::core::simd::operator||(Simd< T1, 1 > a, T2 b) -> Simd< decltype(a.value||b), 1 >'],['../namespacemlx_1_1core_1_1simd.html#a38e83534a648d0743dc4c7deb9a7fd49',1,'mlx::core::simd::operator||(Simd< float16_t, N > a, Simd< float16_t, N > b)'],['../namespacemlx_1_1core_1_1simd.html#acdcdaea84869a0b05c08139c10f13a06',1,'mlx::core::simd::operator||(Simd< float16_t, N > a, T b)'],['../namespacemlx_1_1core_1_1simd.html#aa9ac1951153211b2ff95dd34a3427797',1,'mlx::core::simd::operator||(T a, Simd< float16_t, N > b)'],['../namespacemlx_1_1steel.html#a1bb3ac5061a04e407fc4cdcc9f6ea03f',1,'mlx::steel::operator||()'],['../group__ops.html#ga27af56a98270d4d76d139f0f9171b83a',1,'mlx::core::operator||()']]], + ['out_5fof_5fbounds_43',['out_of_bounds',['../struct_read_writer.html#a08e10626fbc789b6dff9172fd6c36f7c',1,'ReadWriter::out_of_bounds() const'],['../struct_read_writer.html#a6f946aea5452109dca7fc70ed39c6efe',1,'ReadWriter::out_of_bounds() const'],['../struct_read_writer.html#a8f40d7f343d32134fe27a694abfde6bf',1,'ReadWriter::out_of_bounds() const']]], + ['outer_44',['outer',['../group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d',1,'mlx::core']]], + ['output_5fshape_45',['output_shape',['../classmlx_1_1core_1_1_broadcast_axes.html#aaa495110c16fbbc642fbb224ef8dfae6',1,'mlx::core::BroadcastAxes::output_shape()'],['../classmlx_1_1core_1_1_broadcast.html#a00c39c113fe3e698771e2e6b595c32cd',1,'mlx::core::Broadcast::output_shape()'],['../classmlx_1_1core_1_1_expand_dims.html#a3814ad4697eccb75fdb9275017a3fd67',1,'mlx::core::ExpandDims::output_shape()'],['../classmlx_1_1core_1_1_flatten.html#a2f8e1defb9c33af2dec29ff8697132aa',1,'mlx::core::Flatten::output_shape()'],['../classmlx_1_1core_1_1_reshape.html#aa15020d7d844d714d42bc60b44aeefc1',1,'mlx::core::Reshape::output_shape()'],['../classmlx_1_1core_1_1_squeeze.html#aadf1d3b85839390a2ec560603aeed04a',1,'mlx::core::Squeeze::output_shape()'],['../classmlx_1_1core_1_1_unflatten.html#a4c760c8fe981fd2ac17a31ff9faff10a',1,'mlx::core::Unflatten::output_shape()']]], + ['output_5fshapes_46',['output_shapes',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a50934862ccdb16a3dcce6626c5727080',1,'mlx::core::fast::AffineQuantize::output_shapes()'],['../classmlx_1_1core_1_1_primitive.html#aa5b443d71db1c7ed31a5ae6e31b7fe29',1,'mlx::core::Primitive::output_shapes()'],['../classmlx_1_1core_1_1_abs.html#ac9d55481e5490423e4aaf02b95cafc75',1,'mlx::core::Abs::output_shapes()'],['../classmlx_1_1core_1_1_add.html#a50877893083fd78b31aa25152f750418',1,'mlx::core::Add::output_shapes()'],['../classmlx_1_1core_1_1_arange.html#a447083a1403d3d42a7ad9c307a666946',1,'mlx::core::Arange::output_shapes()'],['../classmlx_1_1core_1_1_arc_cos.html#a67a5025f8d7e5bac22888ad4bf813679',1,'mlx::core::ArcCos::output_shapes()'],['../classmlx_1_1core_1_1_arc_cosh.html#a3ab82e9f0452faea735338abccb5f0ac',1,'mlx::core::ArcCosh::output_shapes()'],['../classmlx_1_1core_1_1_arc_sin.html#a0217b9a4e18196ed65ba96b4ad096ecd',1,'mlx::core::ArcSin::output_shapes()'],['../classmlx_1_1core_1_1_arc_sinh.html#a2f668f230d93c7b90e62200a0b7cb6f6',1,'mlx::core::ArcSinh::output_shapes()'],['../classmlx_1_1core_1_1_arc_tan.html#a2ebabfd1c2963199df0d7610b7ddf422',1,'mlx::core::ArcTan::output_shapes()'],['../classmlx_1_1core_1_1_arc_tan2.html#acb8e5cf85c4bc58f909ce2e8b83c3619',1,'mlx::core::ArcTan2::output_shapes()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6806f04142d850f107a18a71900759c6',1,'mlx::core::ArcTanh::output_shapes()'],['../classmlx_1_1core_1_1_arg_partition.html#a587ce69b0639683ba646652f887d0239',1,'mlx::core::ArgPartition::output_shapes()'],['../classmlx_1_1core_1_1_arg_reduce.html#a81a70885480c1d436329025091b2fa4c',1,'mlx::core::ArgReduce::output_shapes()'],['../classmlx_1_1core_1_1_arg_sort.html#a219ce04a811397a900c3235d8e6aef5c',1,'mlx::core::ArgSort::output_shapes()'],['../classmlx_1_1core_1_1_as_type.html#a3975b31cfd86d6eb33dc73554b357b88',1,'mlx::core::AsType::output_shapes()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a49c9d2688d3cca8abf5698a250d57d56',1,'mlx::core::BitwiseBinary::output_shapes()'],['../classmlx_1_1core_1_1_broadcast_axes.html#a42c4385e65851d58e4411a4afe73f58e',1,'mlx::core::BroadcastAxes::output_shapes()'],['../classmlx_1_1core_1_1_broadcast.html#adef65b1ec75efbe43e5574ec81b7c0ac',1,'mlx::core::Broadcast::output_shapes()'],['../classmlx_1_1core_1_1_ceil.html#a3bf7db5178ed26e23d9ba360ba34ab85',1,'mlx::core::Ceil::output_shapes()'],['../classmlx_1_1core_1_1_compiled.html#a15cb081590ee024ba11476494581a4d4',1,'mlx::core::Compiled::output_shapes()'],['../classmlx_1_1core_1_1_concatenate.html#af8415a2fe28804a1437d0876ba15615f',1,'mlx::core::Concatenate::output_shapes()'],['../classmlx_1_1core_1_1_conjugate.html#afd68332463d12e69c47388f6b81ae96c',1,'mlx::core::Conjugate::output_shapes()'],['../classmlx_1_1core_1_1_contiguous.html#afff58fbf61f0c26b3606208dd2fa2072',1,'mlx::core::Contiguous::output_shapes()'],['../classmlx_1_1core_1_1_copy.html#a6bbe5fd9ce3cb5a39853b316106d2674',1,'mlx::core::Copy::output_shapes()'],['../classmlx_1_1core_1_1_cos.html#a923312e71c5a003a38b37ab67ec82580',1,'mlx::core::Cos::output_shapes()'],['../classmlx_1_1core_1_1_cosh.html#adf58c7e24b5059e66007132bc16dfe49',1,'mlx::core::Cosh::output_shapes()'],['../classmlx_1_1core_1_1_divide.html#a9563d9ee243204cfdaac6aca34853cd7',1,'mlx::core::Divide::output_shapes()'],['../classmlx_1_1core_1_1_div_mod.html#a1b7f104346cb5423ac15371b45c7ef86',1,'mlx::core::DivMod::output_shapes()'],['../classmlx_1_1core_1_1_select.html#a10e837a391542b364186288a87e11513',1,'mlx::core::Select::output_shapes()'],['../classmlx_1_1core_1_1_remainder.html#ab4de49818d1fdea8cdfef502f519b255',1,'mlx::core::Remainder::output_shapes()'],['../classmlx_1_1core_1_1_equal.html#ae714c2b0641fc9c339a2f8483bb4e257',1,'mlx::core::Equal::output_shapes()'],['../classmlx_1_1core_1_1_erf.html#ace70b96c48419e29243982ed697f6411',1,'mlx::core::Erf::output_shapes()'],['../classmlx_1_1core_1_1_erf_inv.html#a067cac7a7244b4dae6629c7e4466589f',1,'mlx::core::ErfInv::output_shapes()'],['../classmlx_1_1core_1_1_exp.html#aef2b3c24dba3ca3a63a210d3bd8e39b6',1,'mlx::core::Exp::output_shapes()'],['../classmlx_1_1core_1_1_expm1.html#ae78f03a204687f16164ed702cfc0d5cc',1,'mlx::core::Expm1::output_shapes()'],['../classmlx_1_1core_1_1_expand_dims.html#af64bd4bc2cc5f5c58869f34cd974bb3c',1,'mlx::core::ExpandDims::output_shapes()'],['../classmlx_1_1core_1_1_flatten.html#a5069a73ba1e7b52b7b051f692db6d0d2',1,'mlx::core::Flatten::output_shapes()'],['../classmlx_1_1core_1_1_floor.html#a0a62dee6df6a82fcd955bf7670be2cd5',1,'mlx::core::Floor::output_shapes()'],['../classmlx_1_1core_1_1_gather.html#a53d89a6c4ebb634bc208bd85aa2fcda1',1,'mlx::core::Gather::output_shapes()'],['../classmlx_1_1core_1_1_gather_axis.html#abc483c7da7747263b2f1498f98b4d96d',1,'mlx::core::GatherAxis::output_shapes()'],['../classmlx_1_1core_1_1_greater.html#af798a7cd704a2a9a8b3ecb6ef49583b0',1,'mlx::core::Greater::output_shapes()'],['../classmlx_1_1core_1_1_greater_equal.html#a1a77c18d89ee227171ff38efef6cacf6',1,'mlx::core::GreaterEqual::output_shapes()'],['../classmlx_1_1core_1_1_hadamard.html#aa709166de3c493308689769579d665e8',1,'mlx::core::Hadamard::output_shapes()'],['../classmlx_1_1core_1_1_imag.html#ad4f847483ba07d20aba5b927c2689be8',1,'mlx::core::Imag::output_shapes()'],['../classmlx_1_1core_1_1_less.html#ad7604a75b79260d263ac0c7d959cadd5',1,'mlx::core::Less::output_shapes()'],['../classmlx_1_1core_1_1_less_equal.html#a5598c700e881673098928e47b4da9ff8',1,'mlx::core::LessEqual::output_shapes()'],['../classmlx_1_1core_1_1_log.html#ab2cae6889352ca0674f6463f8f52d77d',1,'mlx::core::Log::output_shapes()'],['../classmlx_1_1core_1_1_log1p.html#a73a02ddf0f125fff83462d97146a0a08',1,'mlx::core::Log1p::output_shapes()'],['../classmlx_1_1core_1_1_logical_not.html#ad3889969521c6a040aa2f26caee219b7',1,'mlx::core::LogicalNot::output_shapes()'],['../classmlx_1_1core_1_1_logical_and.html#a266f1eaced19b8b11e273de9219cf9ed',1,'mlx::core::LogicalAnd::output_shapes()'],['../classmlx_1_1core_1_1_logical_or.html#a931b98fca3e19085af9fa97a43db8ced',1,'mlx::core::LogicalOr::output_shapes()'],['../classmlx_1_1core_1_1_log_add_exp.html#a234f8c8ea5f5bf2fb7e371588fea98b9',1,'mlx::core::LogAddExp::output_shapes()'],['../classmlx_1_1core_1_1_matmul.html#abfabe69f428f7f125bf5665713a0eb5c',1,'mlx::core::Matmul::output_shapes()'],['../classmlx_1_1core_1_1_maximum.html#a888a69fb68726c3c18973f3ea38cfd2b',1,'mlx::core::Maximum::output_shapes()'],['../classmlx_1_1core_1_1_minimum.html#af921b5202ebf9716972bcf0e3056742a',1,'mlx::core::Minimum::output_shapes()'],['../classmlx_1_1core_1_1_multiply.html#adfd4c7f89660b42ab58e088b1ae19435',1,'mlx::core::Multiply::output_shapes()'],['../classmlx_1_1core_1_1_negative.html#a606fb13a48d10c88707f1a2c41bee9e8',1,'mlx::core::Negative::output_shapes()'],['../classmlx_1_1core_1_1_not_equal.html#ad1e8a577dc103d96f1ab65bf3b389d35',1,'mlx::core::NotEqual::output_shapes()'],['../classmlx_1_1core_1_1_number_of_elements.html#a6cdf307348ba22b3dc8f90f1fb1e0757',1,'mlx::core::NumberOfElements::output_shapes()'],['../classmlx_1_1core_1_1_partition.html#a5e62aa0109e53fb4acb861ef39787b4a',1,'mlx::core::Partition::output_shapes()'],['../classmlx_1_1core_1_1_power.html#af23ed795bdcdc4c3f91f0d4c1bb1d928',1,'mlx::core::Power::output_shapes()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a7d57a31d41c58e1bd88ffe9c6b0dbf52',1,'mlx::core::QuantizedMatmul::output_shapes()'],['../classmlx_1_1core_1_1_real.html#a75999bd0b97d97a5675b9cdbab27dcff',1,'mlx::core::Real::output_shapes()'],['../classmlx_1_1core_1_1_reshape.html#aed3a83606d6917b2c344607101a2c43d',1,'mlx::core::Reshape::output_shapes()'],['../classmlx_1_1core_1_1_reduce.html#aaf3da1c98cdf530803118b382c5f58bc',1,'mlx::core::Reduce::output_shapes()'],['../classmlx_1_1core_1_1_round.html#a61821399e177e142723fc986e437d459',1,'mlx::core::Round::output_shapes()'],['../classmlx_1_1core_1_1_scatter_axis.html#af9688c010e1abee9b7b3788f11d91cc5',1,'mlx::core::ScatterAxis::output_shapes()'],['../classmlx_1_1core_1_1_sigmoid.html#aff024a3309584724c9842f172a4e440b',1,'mlx::core::Sigmoid::output_shapes()'],['../classmlx_1_1core_1_1_sign.html#a2260f2e8e081010192eb8a6f90acde6e',1,'mlx::core::Sign::output_shapes()'],['../classmlx_1_1core_1_1_sin.html#abdd433ecbb54898161b43aa9e14ec7f1',1,'mlx::core::Sin::output_shapes()'],['../classmlx_1_1core_1_1_sinh.html#ae04d8f6175c691a8f0d2a9fdd15af0ad',1,'mlx::core::Sinh::output_shapes()'],['../classmlx_1_1core_1_1_slice_update.html#abb6376f13c4269bd9e739e131893da53',1,'mlx::core::SliceUpdate::output_shapes()'],['../classmlx_1_1core_1_1_dynamic_slice.html#a920dc4d1ee4976065e6d91fe3ecfbbf3',1,'mlx::core::DynamicSlice::output_shapes()'],['../classmlx_1_1core_1_1_dynamic_slice_update.html#a804c03c745fc563e209a7bfb3d425a91',1,'mlx::core::DynamicSliceUpdate::output_shapes()'],['../classmlx_1_1core_1_1_softmax.html#a1a798a4dcd62486362d4b58582357490',1,'mlx::core::Softmax::output_shapes()'],['../classmlx_1_1core_1_1_sort.html#acc0a3f078b3f4c83e6e1137cb81ee62c',1,'mlx::core::Sort::output_shapes()'],['../classmlx_1_1core_1_1_square.html#a0513541766bb997ed166643fe95a6d38',1,'mlx::core::Square::output_shapes()'],['../classmlx_1_1core_1_1_sqrt.html#ae45215d61e2e99749d9a0bae291edd45',1,'mlx::core::Sqrt::output_shapes()'],['../classmlx_1_1core_1_1_stop_gradient.html#a8af7641d478505d1dc39c75ba7d5a3cf',1,'mlx::core::StopGradient::output_shapes()'],['../classmlx_1_1core_1_1_subtract.html#aaaff4872bde70ad40cf90e6131ea0489',1,'mlx::core::Subtract::output_shapes()'],['../classmlx_1_1core_1_1_squeeze.html#a839d9d72ac0a19e1146b5b470292a174',1,'mlx::core::Squeeze::output_shapes()'],['../classmlx_1_1core_1_1_tan.html#a9e4bba311bb24617dbb5ca591bc2868e',1,'mlx::core::Tan::output_shapes()'],['../classmlx_1_1core_1_1_tanh.html#a8873286b69b805486fa83c4806843f3d',1,'mlx::core::Tanh::output_shapes()'],['../classmlx_1_1core_1_1_unflatten.html#a068cf053b5b0612fafd4a2d53d42f9fa',1,'mlx::core::Unflatten::output_shapes()'],['../classmlx_1_1core_1_1_transpose.html#ac9328f43900bedec555909d09202ccd7',1,'mlx::core::Transpose::output_shapes()'],['../classmlx_1_1core_1_1_eigh.html#a9892f5b72dec19a5a2f7af5efcf2a952',1,'mlx::core::Eigh::output_shapes()']]], + ['outputs_47',['outputs',['../classmlx_1_1core_1_1array.html#a2c186fd527f984f0589d4183b4976289',1,'mlx::core::array::outputs()'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#aefa48740fdee884f02e2d379bca4e78f',1,'mlx::core::metal::CommandEncoder::outputs()'],['../structmlx_1_1core_1_1_command_encoder.html#aefa48740fdee884f02e2d379bca4e78f',1,'mlx::core::CommandEncoder::outputs()']]], + ['overwrite_5fdescriptor_48',['overwrite_descriptor',['../classmlx_1_1core_1_1array.html#a95e6b156c8e05439f076b85c05079387',1,'mlx::core::array']]] ]; diff --git a/docs/build/html/search/namespaces_0.js b/docs/build/html/search/namespaces_0.js index 1e427aa4f..f9c2e1fc6 100644 --- a/docs/build/html/search/namespaces_0.js +++ b/docs/build/html/search/namespaces_0.js @@ -10,13 +10,15 @@ var searchData= ['mlx_3a_3acore_3a_3adistributed_7',['distributed',['../namespacemlx_1_1core_1_1distributed.html',1,'mlx::core']]], ['mlx_3a_3acore_3a_3adistributed_3a_3adetail_8',['detail',['../namespacemlx_1_1core_1_1distributed_1_1detail.html',1,'mlx::core::distributed']]], ['mlx_3a_3acore_3a_3adistributed_3a_3ampi_9',['mpi',['../namespacemlx_1_1core_1_1distributed_1_1mpi.html',1,'mlx::core::distributed']]], - ['mlx_3a_3acore_3a_3aenv_10',['env',['../namespacemlx_1_1core_1_1env.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3afast_11',['fast',['../namespacemlx_1_1core_1_1fast.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3afft_12',['fft',['../namespacemlx_1_1core_1_1fft.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3aio_13',['io',['../namespacemlx_1_1core_1_1io.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3alinalg_14',['linalg',['../namespacemlx_1_1core_1_1linalg.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3ametal_15',['metal',['../namespacemlx_1_1core_1_1metal.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3arandom_16',['random',['../namespacemlx_1_1core_1_1random.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3ascheduler_17',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html',1,'mlx::core']]], - ['mlx_3a_3asteel_18',['steel',['../namespacemlx_1_1steel.html',1,'mlx']]] + ['mlx_3a_3acore_3a_3adistributed_3a_3aring_10',['ring',['../namespacemlx_1_1core_1_1distributed_1_1ring.html',1,'mlx::core::distributed']]], + ['mlx_3a_3acore_3a_3aenv_11',['env',['../namespacemlx_1_1core_1_1env.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3afast_12',['fast',['../namespacemlx_1_1core_1_1fast.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3afft_13',['fft',['../namespacemlx_1_1core_1_1fft.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3aio_14',['io',['../namespacemlx_1_1core_1_1io.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3alinalg_15',['linalg',['../namespacemlx_1_1core_1_1linalg.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3ametal_16',['metal',['../namespacemlx_1_1core_1_1metal.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3arandom_17',['random',['../namespacemlx_1_1core_1_1random.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3ascheduler_18',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3asimd_19',['simd',['../namespacemlx_1_1core_1_1simd.html',1,'mlx::core']]], + ['mlx_3a_3asteel_20',['steel',['../namespacemlx_1_1steel.html',1,'mlx']]] ]; diff --git a/docs/build/html/search/searchdata.js b/docs/build/html/search/searchdata.js index 5054c9ceb..494441a0f 100644 --- a/docs/build/html/search/searchdata.js +++ b/docs/build/html/search/searchdata.js @@ -3,11 +3,11 @@ var indexSectionsWithContent = 0: "_abcdefghijklmnopqrstuvwxz~", 1: "_abcdefghijklmnopqrstuvw", 2: "mp", - 3: "abcdefghijklmopqrstu", + 3: "abcdefghijklmnopqrstu", 4: "_abcdefghijklmnopqrstuvwz~", 5: "abcdefghijklmnopqrstuvwxz", - 6: "abdefgiklmnprstv", - 7: "bcdkorsv", + 6: "abcdefgiklmnprstv", + 7: "bcdkorstv", 8: "abcdefgilmnoprstuvx", 9: "aeio", 10: "_abcdfhijmoprs", diff --git a/docs/build/html/search/typedefs_1.js b/docs/build/html/search/typedefs_1.js index 5f51db01d..c4feb9399 100644 --- a/docs/build/html/search/typedefs_1.js +++ b/docs/build/html/search/typedefs_1.js @@ -1,6 +1,6 @@ var searchData= [ ['bfloat16_5ft_0',['bfloat16_t',['../backend_2metal_2kernels_2metal__3__0_2bf16_8h.html#a7782de82393104dd4ad754ce3b316e82',1,'bfloat16_t: bf16.h'],['../backend_2metal_2kernels_2metal__3__1_2bf16_8h.html#a58e15a77da988b9104fee00cdf8b280e',1,'bfloat16_t: bf16.h'],['../namespacemlx_1_1core.html#acb5d16c9b83778c7621c38e522e0060b',1,'mlx::core::bfloat16_t']]], - ['block_5fmerge_5fsort_5ft_1',['block_merge_sort_t',['../struct_kernel_merge_sort.html#adae7850e057fc30d5328c7b3dcc998fa',1,'KernelMergeSort::block_merge_sort_t'],['../struct_kernel_multi_block_merge_sort.html#af27e9af4b58640c0aa620bc4efc68dff',1,'KernelMultiBlockMergeSort::block_merge_sort_t']]], + ['block_5fmerge_5fsort_5ft_1',['block_merge_sort_t',['../struct_kernel_merge_sort.html#a37b94702630a4f97254d1da847d420c0',1,'KernelMergeSort::block_merge_sort_t'],['../struct_kernel_multi_block_merge_sort.html#aaf08efb1c540b276b0f1bbdeb514544d',1,'KernelMultiBlockMergeSort::block_merge_sort_t']]], ['bool_5fconstant_2',['bool_constant',['../namespacemlx_1_1steel.html#adbb34bcf0d2dca6b9fb803d591d00da9',1,'mlx::steel']]] ]; diff --git a/docs/build/html/search/typedefs_10.js b/docs/build/html/search/typedefs_10.js new file mode 100644 index 000000000..7dfc187eb --- /dev/null +++ b/docs/build/html/search/typedefs_10.js @@ -0,0 +1,9 @@ +var searchData= +[ + ['v_0',['v',['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html#a3d47d5ad1ff8981bd9876a5fc1870174',1,'mlx::core::simd::ScalarT< bool, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html#af2775b07509324182bd715aac65b7eb0',1,'mlx::core::simd::ScalarT< int8_t, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html#aaf352e77f7ab310c40a31d3dd2bde0eb',1,'mlx::core::simd::ScalarT< uint64_t, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html#aa36db163e4909aea98b7129764184801',1,'mlx::core::simd::ScalarT< int64_t, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT< bool, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT< int64_t, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT< int8_t, N >::v'],['../structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49',1,'mlx::core::simd::ScalarT< uint64_t, N >::v']]], + ['valt_1',['ValT',['../struct_kernel_merge_sort.html#a92910d137ea47703820cde5666d80574',1,'KernelMergeSort']]], + ['value_5ftype_2',['value_type',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#ab86a2740ed9ce3199135372ff1d88c76',1,'pocketfft::detail::threading::aligned_allocator::value_type'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ae24fe304397e961687d0d4c7012b8ae4',1,'mlx::core::array::ArrayIterator::value_type'],['../structmlx_1_1steel_1_1integral__constant.html#a0569cc1334e0bc4f474304b33d365759',1,'mlx::steel::integral_constant::value_type']]], + ['valueandgradfn_3',['ValueAndGradFn',['../namespacemlx_1_1core.html#ab79d66ddf1ec38b2f2c01234892a2230',1,'mlx::core']]], + ['void_5ft_4',['void_t',['../namespacemetal.html#a192322c772aa8b168d59edc55fb806f1',1,'metal']]], + ['vtype_5ft_5',['vtype_t',['../namespacepocketfft_1_1detail.html#a3edfb93aeed2f8258183d463ea291d62',1,'pocketfft::detail']]] +]; diff --git a/docs/build/html/search/typedefs_2.js b/docs/build/html/search/typedefs_2.js index bc4afafdc..f0dd69b60 100644 --- a/docs/build/html/search/typedefs_2.js +++ b/docs/build/html/search/typedefs_2.js @@ -1,5 +1,4 @@ var searchData= [ - ['deleter_0',['Deleter',['../namespacemlx_1_1core.html#af834c1e18d6f11c4f233a2e1ce814a4b',1,'mlx::core']]], - ['difference_5ftype_1',['difference_type',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#adcee44c77980fc2370a2c31e203aead5',1,'mlx::core::array::ArrayIterator']]] + ['col_5ffrag_5ftype_0',['col_frag_type',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#adbb262a3c872e26533b68a39db16459e',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >']]] ]; diff --git a/docs/build/html/search/typedefs_3.js b/docs/build/html/search/typedefs_3.js index 3c5c76d09..bc4afafdc 100644 --- a/docs/build/html/search/typedefs_3.js +++ b/docs/build/html/search/typedefs_3.js @@ -1,5 +1,5 @@ var searchData= [ - ['elem_5ftype_0',['elem_type',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a80078f0dfa4c225e79d9b460202d5e2c',1,'mlx::steel::MMATile::elem_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a80078f0dfa4c225e79d9b460202d5e2c',1,'mlx::steel::MMATile::elem_type']]], - ['enable_5ffor_5farrays_5ft_1',['enable_for_arrays_t',['../namespacemlx_1_1core.html#af89751d79339f3e4d9318ea97d64d114',1,'mlx::core']]] + ['deleter_0',['Deleter',['../namespacemlx_1_1core.html#af834c1e18d6f11c4f233a2e1ce814a4b',1,'mlx::core']]], + ['difference_5ftype_1',['difference_type',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#adcee44c77980fc2370a2c31e203aead5',1,'mlx::core::array::ArrayIterator']]] ]; diff --git a/docs/build/html/search/typedefs_4.js b/docs/build/html/search/typedefs_4.js index bdcdaf9e4..3c5c76d09 100644 --- a/docs/build/html/search/typedefs_4.js +++ b/docs/build/html/search/typedefs_4.js @@ -1,6 +1,5 @@ var searchData= [ - ['false_5ftype_0',['false_type',['../namespacemlx_1_1steel.html#ab0ef721cedc2b5a97f60d76b765aff2e',1,'mlx::steel']]], - ['float16_5ft_1',['float16_t',['../backend_2metal_2kernels_2utils_8h.html#acb8ddf4a29129846b673c50ba7078773',1,'float16_t: utils.h'],['../namespacemlx_1_1core.html#afbd2769c30e721afc85a7b9fb55b8e52',1,'mlx::core::float16_t']]], - ['frag_5ftype_2',['frag_type',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aac25cd0a9bdf24aa2af809c95f0bd171',1,'mlx::steel::MMATile::frag_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aac25cd0a9bdf24aa2af809c95f0bd171',1,'mlx::steel::MMATile::frag_type']]] + ['elem_5ftype_0',['elem_type',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a80078f0dfa4c225e79d9b460202d5e2c',1,'mlx::steel::MMATile::elem_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a80078f0dfa4c225e79d9b460202d5e2c',1,'mlx::steel::MMATile::elem_type']]], + ['enable_5ffor_5farrays_5ft_1',['enable_for_arrays_t',['../namespacemlx_1_1core.html#af89751d79339f3e4d9318ea97d64d114',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/typedefs_5.js b/docs/build/html/search/typedefs_5.js index f6a5488f1..4d30c1fc9 100644 --- a/docs/build/html/search/typedefs_5.js +++ b/docs/build/html/search/typedefs_5.js @@ -1,6 +1,6 @@ var searchData= [ - ['ggufload_0',['GGUFLoad',['../namespacemlx_1_1core.html#aa5b0f7f13a941e1f41c411194e9033c7',1,'mlx::core']]], - ['ggufmetadata_1',['GGUFMetaData',['../namespacemlx_1_1core.html#a8c2c1b9a37aadfb48f4c3a7e806e32e3',1,'mlx::core']]], - ['groupimpl_2',['GroupImpl',['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#ab9a91276b3c84ea63f1d1831ef4079dd',1,'mlx::core::distributed::mpi']]] + ['false_5ftype_0',['false_type',['../namespacemlx_1_1steel.html#ab0ef721cedc2b5a97f60d76b765aff2e',1,'mlx::steel']]], + ['float16_5ft_1',['float16_t',['../backend_2metal_2kernels_2utils_8h.html#acb8ddf4a29129846b673c50ba7078773',1,'float16_t: utils.h'],['../namespacemlx_1_1core.html#afbd2769c30e721afc85a7b9fb55b8e52',1,'mlx::core::float16_t']]], + ['frag_5ftype_2',['frag_type',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8536bfaa108031c2ea3e9ccdc766ee5b',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::frag_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aac25cd0a9bdf24aa2af809c95f0bd171',1,'mlx::steel::MMATile::frag_type'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8536bfaa108031c2ea3e9ccdc766ee5b',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::frag_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aac25cd0a9bdf24aa2af809c95f0bd171',1,'mlx::steel::MMATile::frag_type']]] ]; diff --git a/docs/build/html/search/typedefs_6.js b/docs/build/html/search/typedefs_6.js index 80c07e679..92ea6b60c 100644 --- a/docs/build/html/search/typedefs_6.js +++ b/docs/build/html/search/typedefs_6.js @@ -1,6 +1,6 @@ var searchData= [ - ['idx_5ft_0',['idx_t',['../struct_kernel_merge_sort.html#a0df65b709ae7f153a2bf381179d55e00',1,'KernelMergeSort']]], - ['int_1',['Int',['../namespacemlx_1_1steel.html#afe36ddf6725498d273e5eef4f1579891',1,'mlx::steel']]], - ['iterator_5fcategory_2',['iterator_category',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a2cbf481e39164245668b3be6cbcc614d',1,'mlx::core::array::ArrayIterator']]] + ['ggufload_0',['GGUFLoad',['../namespacemlx_1_1core.html#aa5b0f7f13a941e1f41c411194e9033c7',1,'mlx::core']]], + ['ggufmetadata_1',['GGUFMetaData',['../namespacemlx_1_1core.html#a8c2c1b9a37aadfb48f4c3a7e806e32e3',1,'mlx::core']]], + ['groupimpl_2',['GroupImpl',['../namespacemlx_1_1core_1_1distributed_1_1mpi.html#ab9a91276b3c84ea63f1d1831ef4079dd',1,'mlx::core::distributed::mpi::GroupImpl'],['../namespacemlx_1_1core_1_1distributed_1_1ring.html#a6e4d590e07f0cf3cc2d15f258f9438ed',1,'mlx::core::distributed::ring::GroupImpl']]] ]; diff --git a/docs/build/html/search/typedefs_7.js b/docs/build/html/search/typedefs_7.js index 710dde3d2..43f117bd9 100644 --- a/docs/build/html/search/typedefs_7.js +++ b/docs/build/html/search/typedefs_7.js @@ -1,4 +1,6 @@ var searchData= [ - ['kwargs_0',['Kwargs',['../namespacemlx_1_1core.html#af3efb38b31c0bc08754a4edfda656b83',1,'mlx::core']]] + ['idxt_0',['IdxT',['../struct_kernel_merge_sort.html#a587b32e6748cffd2ace5a70b8c59dee4',1,'KernelMergeSort']]], + ['int_1',['Int',['../namespacemlx_1_1steel.html#afe36ddf6725498d273e5eef4f1579891',1,'mlx::steel']]], + ['iterator_5fcategory_2',['iterator_category',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a2cbf481e39164245668b3be6cbcc614d',1,'mlx::core::array::ArrayIterator']]] ]; diff --git a/docs/build/html/search/typedefs_8.js b/docs/build/html/search/typedefs_8.js index 828e26aa0..710dde3d2 100644 --- a/docs/build/html/search/typedefs_8.js +++ b/docs/build/html/search/typedefs_8.js @@ -1,5 +1,4 @@ var searchData= [ - ['loader_5fa_5ft_0',['loader_a_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#aa8a04ed74d2259f99b337d4662c64d83',1,'mlx::steel::GEMMKernel::loader_a_t'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#aa8a04ed74d2259f99b337d4662c64d83',1,'mlx::steel::GEMMKernel::loader_a_t']]], - ['loader_5fb_5ft_1',['loader_b_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#aa98f32278b5fd98c93ae5483c3596395',1,'mlx::steel::GEMMKernel::loader_b_t'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#aa98f32278b5fd98c93ae5483c3596395',1,'mlx::steel::GEMMKernel::loader_b_t']]] + ['kwargs_0',['Kwargs',['../namespacemlx_1_1core.html#af3efb38b31c0bc08754a4edfda656b83',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/typedefs_9.js b/docs/build/html/search/typedefs_9.js index 457d9102d..828e26aa0 100644 --- a/docs/build/html/search/typedefs_9.js +++ b/docs/build/html/search/typedefs_9.js @@ -1,10 +1,5 @@ var searchData= [ - ['mask_5ft_0',['mask_t',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a270ab3da7c98a12525a59952742cc97d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], - ['mat_5ftype_1',['mat_type',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1eeb197c9bdf4db42892a39cdb9bd73a',1,'mlx::steel::MMATile::mat_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1eeb197c9bdf4db42892a39cdb9bd73a',1,'mlx::steel::MMATile::mat_type']]], - ['metalkernelfunction_2',['MetalKernelFunction',['../namespacemlx_1_1core_1_1fast.html#aa45bf61e7a5c4ad0114b82ed80ae0dbd',1,'mlx::core::fast']]], - ['mma_5ft_3',['mma_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel::mma_t'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel::mma_t']]], - ['mmafrag_5facc_5ft_4',['MMAFrag_acc_t',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae2c42cb6d0dde785859164c195f4d13c',1,'mlx::steel::BlockMMA::MMAFrag_acc_t'],['../structmlx_1_1steel_1_1_block_m_m_a.html#ae2c42cb6d0dde785859164c195f4d13c',1,'mlx::steel::BlockMMA::MMAFrag_acc_t']]], - ['mmafrag_5ft_5',['MMAFrag_t',['../structmlx_1_1steel_1_1_m_m_a_tile.html#abe33de70e34300745bad9aa822fd0382',1,'mlx::steel::MMATile::MMAFrag_t'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#abe33de70e34300745bad9aa822fd0382',1,'mlx::steel::MMATile::MMAFrag_t']]], - ['mtlfclist_6',['MTLFCList',['../namespacemlx_1_1core_1_1metal.html#a616e09a1ef321d527770721cef264c54',1,'mlx::core::metal']]] + ['loader_5fa_5ft_0',['loader_a_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#aa8a04ed74d2259f99b337d4662c64d83',1,'mlx::steel::GEMMKernel::loader_a_t'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#aa8a04ed74d2259f99b337d4662c64d83',1,'mlx::steel::GEMMKernel::loader_a_t']]], + ['loader_5fb_5ft_1',['loader_b_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#aa98f32278b5fd98c93ae5483c3596395',1,'mlx::steel::GEMMKernel::loader_b_t'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#aa98f32278b5fd98c93ae5483c3596395',1,'mlx::steel::GEMMKernel::loader_b_t']]] ]; diff --git a/docs/build/html/search/typedefs_a.js b/docs/build/html/search/typedefs_a.js index 335d411ea..6c2e4e087 100644 --- a/docs/build/html/search/typedefs_a.js +++ b/docs/build/html/search/typedefs_a.js @@ -1,4 +1,10 @@ var searchData= [ - ['nomask_5ft_0',['nomask_t',['../kernels_2gemv__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'gemv_masked.h']]] + ['mask_5ft_0',['mask_t',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a270ab3da7c98a12525a59952742cc97d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], + ['mat_5ftype_1',['mat_type',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a25675ae18947a97c6e04157b540103a9',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mat_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1eeb197c9bdf4db42892a39cdb9bd73a',1,'mlx::steel::MMATile::mat_type'],['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a25675ae18947a97c6e04157b540103a9',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::mat_type'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1eeb197c9bdf4db42892a39cdb9bd73a',1,'mlx::steel::MMATile::mat_type']]], + ['metalkernelfunction_2',['MetalKernelFunction',['../namespacemlx_1_1core_1_1fast.html#aa45bf61e7a5c4ad0114b82ed80ae0dbd',1,'mlx::core::fast']]], + ['mma_5ft_3',['mma_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel::mma_t'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel::mma_t']]], + ['mmafrag_5facc_5ft_4',['MMAFrag_acc_t',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae2c42cb6d0dde785859164c195f4d13c',1,'mlx::steel::BlockMMA::MMAFrag_acc_t'],['../structmlx_1_1steel_1_1_block_m_m_a.html#ae2c42cb6d0dde785859164c195f4d13c',1,'mlx::steel::BlockMMA::MMAFrag_acc_t']]], + ['mmafrag_5ft_5',['MMAFrag_t',['../structmlx_1_1steel_1_1_m_m_a_tile.html#abe33de70e34300745bad9aa822fd0382',1,'mlx::steel::MMATile::MMAFrag_t'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#abe33de70e34300745bad9aa822fd0382',1,'mlx::steel::MMATile::MMAFrag_t']]], + ['mtlfclist_6',['MTLFCList',['../namespacemlx_1_1core_1_1metal.html#a616e09a1ef321d527770721cef264c54',1,'mlx::core::metal']]] ]; diff --git a/docs/build/html/search/typedefs_b.js b/docs/build/html/search/typedefs_b.js index 17b5278b9..79dbda7d7 100644 --- a/docs/build/html/search/typedefs_b.js +++ b/docs/build/html/search/typedefs_b.js @@ -1,5 +1,4 @@ var searchData= [ - ['parentsmap_0',['ParentsMap',['../namespacemlx_1_1core_1_1detail.html#aac5e13ecbb521f3ac0e27d98d15fa985',1,'mlx::core::detail']]], - ['pointer_5felement_5ft_1',['pointer_element_t',['../namespacemetal.html#ac82ee6c3fbe9ec5c78c07329424aaec9',1,'metal']]] + ['nomask_5ft_0',['nomask_t',['../gemv__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'gemv_masked.h']]] ]; diff --git a/docs/build/html/search/typedefs_c.js b/docs/build/html/search/typedefs_c.js index ba03c67d8..17b5278b9 100644 --- a/docs/build/html/search/typedefs_c.js +++ b/docs/build/html/search/typedefs_c.js @@ -1,5 +1,5 @@ var searchData= [ - ['radixfunc_0',['RadixFunc',['../backend_2metal_2kernels_2fft_8h.html#a6ba62eabfd5428644aabf89ddaa0128d',1,'fft.h']]], - ['reference_1',['reference',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a44e2e1f29191c20ec4390de4fa0bd59f',1,'mlx::core::array::ArrayIterator']]] + ['parentsmap_0',['ParentsMap',['../namespacemlx_1_1core_1_1detail.html#aac5e13ecbb521f3ac0e27d98d15fa985',1,'mlx::core::detail']]], + ['pointer_5felement_5ft_1',['pointer_element_t',['../namespacemetal.html#ac82ee6c3fbe9ec5c78c07329424aaec9',1,'metal']]] ]; diff --git a/docs/build/html/search/typedefs_d.js b/docs/build/html/search/typedefs_d.js index 698974fcc..0530e4920 100644 --- a/docs/build/html/search/typedefs_d.js +++ b/docs/build/html/search/typedefs_d.js @@ -1,11 +1,6 @@ var searchData= [ - ['safetensorsload_0',['SafetensorsLoad',['../namespacemlx_1_1core.html#a688cd7917b1365065e8059e9964c3d45',1,'mlx::core']]], - ['shape_1',['Shape',['../namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416',1,'mlx::core']]], - ['shape_5ft_2',['shape_t',['../namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::detail::shape_t'],['../namespacepocketfft.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::shape_t']]], - ['shapeelem_3',['ShapeElem',['../namespacemlx_1_1core.html#a167cdec84c0ae62b5b299c617384346e',1,'mlx::core']]], - ['simplevalueandgradfn_4',['SimpleValueAndGradFn',['../namespacemlx_1_1core.html#a2689b8f1181648cb1685204fea9f3066',1,'mlx::core']]], - ['streamordevice_5',['StreamOrDevice',['../namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58',1,'mlx::core']]], - ['stride_5ft_6',['stride_t',['../namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::detail::stride_t'],['../namespacepocketfft.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::stride_t']]], - ['strides_7',['Strides',['../namespacemlx_1_1core.html#a79939016d0972ded7db37130da2a8b5c',1,'mlx::core']]] + ['radixfunc_0',['RadixFunc',['../backend_2metal_2kernels_2fft_8h.html#a6ba62eabfd5428644aabf89ddaa0128d',1,'fft.h']]], + ['reference_1',['reference',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a44e2e1f29191c20ec4390de4fa0bd59f',1,'mlx::core::array::ArrayIterator']]], + ['row_5ffrag_5ftype_2',['row_frag_type',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a3dcd4301390937f89ed1dde6d28e341f',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >']]] ]; diff --git a/docs/build/html/search/typedefs_e.js b/docs/build/html/search/typedefs_e.js index 6422079f7..5b0e46490 100644 --- a/docs/build/html/search/typedefs_e.js +++ b/docs/build/html/search/typedefs_e.js @@ -1,7 +1,12 @@ var searchData= [ - ['templatearg_0',['TemplateArg',['../namespacemlx_1_1core_1_1fast.html#a9390693ff7be931f3ef3428e2ea4c3f9',1,'mlx::core::fast']]], - ['thread_5fsort_5ft_1',['thread_sort_t',['../struct_block_merge_sort.html#ad2474d16721f4ceb954125728a0e2ea2',1,'BlockMergeSort']]], - ['true_5ftype_2',['true_type',['../namespacemlx_1_1steel.html#a594a6ccb75b38b5ae4ddd0d9ad047b3a',1,'mlx::steel']]], - ['type_3',['type',['../structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e',1,'pocketfft::detail::add_vec::type'],['../structmlx_1_1steel_1_1integral__constant.html#a6492c15b37d160d3a33e1cbe770aa3f1',1,'mlx::steel::integral_constant::type'],['../structmetal_1_1make__void.html#a53b29d640a06df62047fffdec0571392',1,'metal::make_void::type'],['../struct_conditional_type.html#a00bac71c43763817c4422bf0363dc92b',1,'ConditionalType::type']]] + ['safetensorsload_0',['SafetensorsLoad',['../namespacemlx_1_1core.html#a688cd7917b1365065e8059e9964c3d45',1,'mlx::core']]], + ['scalar_5ft_1',['scalar_t',['../structmlx_1_1core_1_1simd_1_1_simd.html#a26040194a37172b6aed7c5d1685362fb',1,'mlx::core::simd::Simd::scalar_t'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a422e15f018cd242dd62617f4213dace0',1,'mlx::core::simd::Simd< float16_t, N >::scalar_t'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a26040194a37172b6aed7c5d1685362fb',1,'mlx::core::simd::Simd< T, 1 >::scalar_t'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a26040194a37172b6aed7c5d1685362fb',1,'mlx::core::simd::Simd< float16_t, N >::scalar_t']]], + ['shape_2',['Shape',['../namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416',1,'mlx::core']]], + ['shape_5ft_3',['shape_t',['../namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::detail::shape_t'],['../namespacepocketfft.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::shape_t']]], + ['shapeelem_4',['ShapeElem',['../namespacemlx_1_1core.html#a167cdec84c0ae62b5b299c617384346e',1,'mlx::core']]], + ['simplevalueandgradfn_5',['SimpleValueAndGradFn',['../namespacemlx_1_1core.html#a2689b8f1181648cb1685204fea9f3066',1,'mlx::core']]], + ['streamordevice_6',['StreamOrDevice',['../namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58',1,'mlx::core']]], + ['stride_5ft_7',['stride_t',['../namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::detail::stride_t'],['../namespacepocketfft.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::stride_t']]], + ['strides_8',['Strides',['../namespacemlx_1_1core.html#a79939016d0972ded7db37130da2a8b5c',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/typedefs_f.js b/docs/build/html/search/typedefs_f.js index cfd1d3816..65dc80fa8 100644 --- a/docs/build/html/search/typedefs_f.js +++ b/docs/build/html/search/typedefs_f.js @@ -1,8 +1,7 @@ var searchData= [ - ['val_5ft_0',['val_t',['../struct_kernel_merge_sort.html#a4e3f09896275956fc4c23e1f157dca3b',1,'KernelMergeSort']]], - ['value_5ftype_1',['value_type',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#ab86a2740ed9ce3199135372ff1d88c76',1,'pocketfft::detail::threading::aligned_allocator::value_type'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ae24fe304397e961687d0d4c7012b8ae4',1,'mlx::core::array::ArrayIterator::value_type'],['../structmlx_1_1steel_1_1integral__constant.html#a0569cc1334e0bc4f474304b33d365759',1,'mlx::steel::integral_constant::value_type']]], - ['valueandgradfn_2',['ValueAndGradFn',['../namespacemlx_1_1core.html#ab79d66ddf1ec38b2f2c01234892a2230',1,'mlx::core']]], - ['void_5ft_3',['void_t',['../namespacemetal.html#a192322c772aa8b168d59edc55fb806f1',1,'metal']]], - ['vtype_5ft_4',['vtype_t',['../namespacepocketfft_1_1detail.html#a3edfb93aeed2f8258183d463ea291d62',1,'pocketfft::detail']]] + ['templatearg_0',['TemplateArg',['../namespacemlx_1_1core_1_1fast.html#a9390693ff7be931f3ef3428e2ea4c3f9',1,'mlx::core::fast']]], + ['thread_5fsort_5ft_1',['thread_sort_t',['../struct_block_merge_sort.html#afdec1074b7de04cbe083e3270b4ece88',1,'BlockMergeSort']]], + ['true_5ftype_2',['true_type',['../namespacemlx_1_1steel.html#a594a6ccb75b38b5ae4ddd0d9ad047b3a',1,'mlx::steel']]], + ['type_3',['type',['../structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e',1,'pocketfft::detail::add_vec::type'],['../structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html#a257b1c81fb9f559c48ee90497013494e',1,'pocketfft::detail::add_vec< cmplx< T > >::type'],['../structmlx_1_1steel_1_1integral__constant.html#a6492c15b37d160d3a33e1cbe770aa3f1',1,'mlx::steel::integral_constant::type'],['../structmetal_1_1make__void.html#a53b29d640a06df62047fffdec0571392',1,'metal::make_void::type'],['../structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.html#a98fbc2aa99dd26bb35aa9cd1826318d8',1,'metal::pointer_element< thread T * >::type'],['../structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.html#ab36a7c5a64c0693dd3d8ccb322c163d4',1,'metal::pointer_element< device T * >::type'],['../structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.html#ad154b55b9e450a6376016488c8e68c53',1,'metal::pointer_element< constant T * >::type'],['../structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.html#a78c718d6da9d393c139a385f42472362',1,'metal::pointer_element< threadgroup T * >::type'],['../struct_conditional_type.html#a00bac71c43763817c4422bf0363dc92b',1,'ConditionalType::type'],['../struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.html#abf6b75b3bfc5e3baf8c37afdb8a7e4fc',1,'ConditionalType< true, T, U >::type'],['../structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e',1,'pocketfft::detail::add_vec< cmplx< T > >::type'],['../struct_conditional_type.html#a00bac71c43763817c4422bf0363dc92b',1,'ConditionalType< true, T, U >::type']]] ]; diff --git a/docs/build/html/search/variables_12.js b/docs/build/html/search/variables_12.js index e501fe30d..0d1d36bf4 100644 --- a/docs/build/html/search/variables_12.js +++ b/docs/build/html/search/variables_12.js @@ -9,28 +9,22 @@ var searchData= ['signedinteger_6',['signedinteger',['../namespacemlx_1_1core.html#a24e1618af591d737d73729665e868001',1,'mlx::core']]], ['simd_5fsize_7',['SIMD_SIZE',['../quantized_8h.html#a62969a218d93680f5e35d0c61b160b99',1,'quantized.h']]], ['simd_5fsize_8',['simd_size',['../backend_2metal_2kernels_2reduction_2ops_8h.html#a515b75d563a93d3c09ee677948dc83e3',1,'ops.h']]], - ['sm_9',['sm',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa85451edf6900fd6af164d4d50889ae3',1,'mlx::steel::BlockMMA']]], - ['sn_10',['sn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ade420e8b811d597345783c324c23a34a',1,'mlx::steel::BlockMMA']]], - ['softmax_5fkernels_11',['softmax_kernels',['../jit_2softmax_8h.html#a1cbfb210a9a765c6620e9f1247ccef12',1,'softmax.h']]], - ['softmax_5fn_5freads_12',['SOFTMAX_N_READS',['../defines_8h.html#a722995df24286b27b7da3d74b73f768d',1,'defines.h']]], - ['split_5fk_5fpartition_5fsize_13',['split_k_partition_size',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a9f5a67b2343645b570e109c3837d4042',1,'mlx::steel::GEMMSpiltKParams']]], - ['split_5fk_5fpartition_5fstride_14',['split_k_partition_stride',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a12144ce89d404812cd862611d770b9fb',1,'mlx::steel::GEMMSpiltKParams']]], - ['split_5fk_5fpartitions_15',['split_k_partitions',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#ae06c27116905d4ff3b9b436e588a93fd',1,'mlx::steel::GEMMSpiltKParams']]], - ['src_16',['src',['../struct_quantized_block_loader.html#abbf8249ca99e3e87b296ddd60a984b76',1,'QuantizedBlockLoader::src'],['../structmlx_1_1steel_1_1_block_loader.html#ad1db14517568ae9eddfb6986ef31c7aa',1,'mlx::steel::BlockLoader::src'],['../structmlx_1_1steel_1_1_block_loader_t.html#a7004a4efaa483cc79b8b79810a17c777',1,'mlx::steel::BlockLoaderT::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a656a46ee27486482b45ff90b3d626255',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a57552e9cfbafad71d47b2f3a8e027bdf',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7bfbcc4a1e3eef7aef5dd8e8c374a95f',1,'mlx::steel::Conv2DWeightBlockLoader::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#afe21e46e08523232830c25eb1b4ade16',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a8b04a69952404a04029dacc424df6e8f',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1ee2922961b5fcb1db577928c4d9d731',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a50f458dbb74d61be2ed24727d8d43614',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src']]], - ['src_5fld_17',['src_ld',['../struct_quantized_block_loader.html#a8050977d473d1a24fae5c833e609839e',1,'QuantizedBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_block_loader.html#aadafc50f7f06af434149d7469df4714d',1,'mlx::steel::BlockLoader::src_ld'],['../structmlx_1_1steel_1_1_block_loader_t.html#aeba87e81185da6b20a092c5d240d3321',1,'mlx::steel::BlockLoaderT::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7464ec687323fa79050702952ed9084f',1,'mlx::steel::Conv2DWeightBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#aa6bedc0cbb447eaf70c03f2e26df2cb2',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6918c1df7712c4e408e2871467ea7987',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src_ld']]], - ['start_5frow_18',['start_row',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a220e033b689c8d6a6f319dae02b38334',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], - ['steel_5fconv_5fgeneral_5fkernels_19',['steel_conv_general_kernels',['../jit_2steel__conv_8h.html#ae4ca1720029316b08ea92b7662347d47',1,'steel_conv.h']]], - ['steel_5fconv_5fkernels_20',['steel_conv_kernels',['../jit_2steel__conv_8h.html#a386d79077465df56659416fd84adb513',1,'steel_conv.h']]], - ['steel_5fgemm_5ffused_5fkernels_21',['steel_gemm_fused_kernels',['../steel__gemm_8h.html#a4c6009fd5357b730805f2fd4ba6e093e',1,'steel_gemm.h']]], - ['steel_5fgemm_5fmasked_5fkernels_22',['steel_gemm_masked_kernels',['../steel__gemm_8h.html#a62a358fd3ec5365081920d07aceb581c',1,'steel_gemm.h']]], - ['steel_5fgemm_5fsplitk_5faccum_5faxbpy_5fkernels_23',['steel_gemm_splitk_accum_axbpy_kernels',['../steel__gemm_8h.html#a40a86e1381c241aba8511e51a981a4bf',1,'steel_gemm.h']]], - ['steel_5fgemm_5fsplitk_5faccum_5fkernels_24',['steel_gemm_splitk_accum_kernels',['../steel__gemm_8h.html#a144a64b8d94f0371fb144e2cc308fcf9',1,'steel_gemm.h']]], - ['steel_5fgemm_5fsplitk_5fkernels_25',['steel_gemm_splitk_kernels',['../steel__gemm_8h.html#a92108ab01d826e38bca83d8569b947d9',1,'steel_gemm.h']]], - ['stop_26',['stop',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a456ad1c0c9e731833a2f8411c4ed51aa',1,'mlx::core::scheduler::StreamThread']]], - ['str_27',['str',['../classpocketfft_1_1detail_1_1arr__info.html#abe1f7b92501b4e0e5a38fd26294ac5a4',1,'pocketfft::detail::arr_info::str'],['../struct_m_l_x_conv_params.html#a862191e8ab1bc8a47aa1396b36d46058',1,'MLXConvParams::str']]], - ['stream_28',['stream',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a8462e4acffcd385c6248bd7102e6bcb1',1,'mlx::core::scheduler::StreamThread']]], - ['strided_5fdevice_5fidx_29',['strided_device_idx',['../struct_read_writer.html#a4c0b12484aac4fd6759d67c190391989',1,'ReadWriter']]], - ['strided_5fshared_5fidx_30',['strided_shared_idx',['../struct_read_writer.html#ace40adb02cfb33d89c98353327c251fc',1,'ReadWriter']]], - ['strides_31',['strides',['../structmlx_1_1core_1_1_reduction_plan.html#a58bc6189e5e7175dae92632a7bcfd53e',1,'mlx::core::ReductionPlan::strides'],['../struct_indices.html#a510b7fe052c5826911dd17d7ccb9e73f',1,'Indices::strides'],['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63954de7da62942ec69afcaaa19d46f2',1,'mlx::core::fast::CustomKernelShapeInfo::strides']]], - ['swizzle_5flog_32',['swizzle_log',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#ad0713159d4f710cd9a066596593d8840',1,'mlx::steel::ImplicitGemmConv2DParams::swizzle_log'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#af9ff2c06dd8994126634531440325be7',1,'mlx::steel::GEMMParams::swizzle_log']]] + ['size_9',['size',['../structmlx_1_1core_1_1simd_1_1_simd.html#aa2b56facc70ba4e8d33a74def204a1fd',1,'mlx::core::simd::Simd::size'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#a14e16c6e2ef5e89135cf8e85dc9f1f1f',1,'mlx::core::simd::Simd< T, 1 >::size'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a2629cb8da72b6f922ed14cc7b6c43ce7',1,'mlx::core::simd::Simd< float16_t, N >::size'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aa2b56facc70ba4e8d33a74def204a1fd',1,'mlx::core::simd::Simd< T, 1 >::size'],['../structmlx_1_1core_1_1simd_1_1_simd.html#aa2b56facc70ba4e8d33a74def204a1fd',1,'mlx::core::simd::Simd< float16_t, N >::size']]], + ['sm_10',['sm',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa85451edf6900fd6af164d4d50889ae3',1,'mlx::steel::BlockMMA']]], + ['sn_11',['sn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ade420e8b811d597345783c324c23a34a',1,'mlx::steel::BlockMMA']]], + ['softmax_5fkernels_12',['softmax_kernels',['../jit_2softmax_8h.html#a1cbfb210a9a765c6620e9f1247ccef12',1,'softmax.h']]], + ['softmax_5fn_5freads_13',['SOFTMAX_N_READS',['../defines_8h.html#a722995df24286b27b7da3d74b73f768d',1,'defines.h']]], + ['split_5fk_5fpartition_5fsize_14',['split_k_partition_size',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a9f5a67b2343645b570e109c3837d4042',1,'mlx::steel::GEMMSpiltKParams']]], + ['split_5fk_5fpartition_5fstride_15',['split_k_partition_stride',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a12144ce89d404812cd862611d770b9fb',1,'mlx::steel::GEMMSpiltKParams']]], + ['split_5fk_5fpartitions_16',['split_k_partitions',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#ae06c27116905d4ff3b9b436e588a93fd',1,'mlx::steel::GEMMSpiltKParams']]], + ['src_17',['src',['../struct_quantized_block_loader.html#abbf8249ca99e3e87b296ddd60a984b76',1,'QuantizedBlockLoader::src'],['../structmlx_1_1steel_1_1_block_loader.html#ad1db14517568ae9eddfb6986ef31c7aa',1,'mlx::steel::BlockLoader::src'],['../structmlx_1_1steel_1_1_block_loader_t.html#a7004a4efaa483cc79b8b79810a17c777',1,'mlx::steel::BlockLoaderT::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a656a46ee27486482b45ff90b3d626255',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a57552e9cfbafad71d47b2f3a8e027bdf',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7bfbcc4a1e3eef7aef5dd8e8c374a95f',1,'mlx::steel::Conv2DWeightBlockLoader::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#afe21e46e08523232830c25eb1b4ade16',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a8b04a69952404a04029dacc424df6e8f',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1ee2922961b5fcb1db577928c4d9d731',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a50f458dbb74d61be2ed24727d8d43614',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src']]], + ['src_5fld_18',['src_ld',['../struct_quantized_block_loader.html#a8050977d473d1a24fae5c833e609839e',1,'QuantizedBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_block_loader.html#aadafc50f7f06af434149d7469df4714d',1,'mlx::steel::BlockLoader::src_ld'],['../structmlx_1_1steel_1_1_block_loader_t.html#aeba87e81185da6b20a092c5d240d3321',1,'mlx::steel::BlockLoaderT::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7464ec687323fa79050702952ed9084f',1,'mlx::steel::Conv2DWeightBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#aa6bedc0cbb447eaf70c03f2e26df2cb2',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6918c1df7712c4e408e2871467ea7987',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src_ld']]], + ['start_5frow_19',['start_row',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a220e033b689c8d6a6f319dae02b38334',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], + ['stop_20',['stop',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a456ad1c0c9e731833a2f8411c4ed51aa',1,'mlx::core::scheduler::StreamThread']]], + ['str_21',['str',['../classpocketfft_1_1detail_1_1arr__info.html#abe1f7b92501b4e0e5a38fd26294ac5a4',1,'pocketfft::detail::arr_info::str'],['../struct_m_l_x_conv_params.html#a862191e8ab1bc8a47aa1396b36d46058',1,'MLXConvParams::str']]], + ['stream_22',['stream',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a8462e4acffcd385c6248bd7102e6bcb1',1,'mlx::core::scheduler::StreamThread']]], + ['strided_5fdevice_5fidx_23',['strided_device_idx',['../struct_read_writer.html#a4c0b12484aac4fd6759d67c190391989',1,'ReadWriter']]], + ['strided_5fshared_5fidx_24',['strided_shared_idx',['../struct_read_writer.html#ace40adb02cfb33d89c98353327c251fc',1,'ReadWriter']]], + ['strides_25',['strides',['../structmlx_1_1core_1_1_reduction_plan.html#a58bc6189e5e7175dae92632a7bcfd53e',1,'mlx::core::ReductionPlan::strides'],['../struct_indices.html#a510b7fe052c5826911dd17d7ccb9e73f',1,'Indices::strides'],['../structmlx_1_1core_1_1fast_1_1_custom_kernel_shape_info.html#a63954de7da62942ec69afcaaa19d46f2',1,'mlx::core::fast::CustomKernelShapeInfo::strides']]], + ['swizzle_5flog_26',['swizzle_log',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#ad0713159d4f710cd9a066596593d8840',1,'mlx::steel::ImplicitGemmConv2DParams::swizzle_log'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#af9ff2c06dd8994126634531440325be7',1,'mlx::steel::GEMMParams::swizzle_log']]] ]; diff --git a/docs/build/html/search/variables_15.js b/docs/build/html/search/variables_15.js index 43059d0b6..65481f281 100644 --- a/docs/build/html/search/variables_15.js +++ b/docs/build/html/search/variables_15.js @@ -2,8 +2,8 @@ var searchData= [ ['v_0',['v',['../structmlx_1_1steel_1_1_block_loader_1_1_read_vector.html#a20963f7191251defca48bf8a843d019d',1,'mlx::steel::BlockLoader::ReadVector']]], ['v_5fstrides_1',['V_strides',['../structmlx_1_1steel_1_1_attn_params.html#ad1495980297901b8ded1fb6dd73979b1',1,'mlx::steel::AttnParams']]], - ['val_2',['val',['../structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534',1,'pocketfft::detail::VLEN::val'],['../structmlx__atomic.html#a6f6651b8dd8149917c50cd99b13c6747',1,'mlx_atomic::val']]], + ['val_2',['val',['../structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534',1,'pocketfft::detail::VLEN::val'],['../structmlx__atomic.html#a6f6651b8dd8149917c50cd99b13c6747',1,'mlx_atomic::val'],['../structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html#a8dbf729fcd8c4a16e41b546c7405543d',1,'mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >::val'],['../structmlx__atomic.html#a6f6651b8dd8149917c50cd99b13c6747',1,'mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >::val']]], ['val_5ffrags_3',['val_frags',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a684e6c6d9f00f583994285b60aaa3b62',1,'mlx::steel::MMATile']]], - ['value_4',['value',['../structmlx_1_1steel_1_1integral__constant.html#a4efa69cb3fd42ac0dcad46578600d637',1,'mlx::steel::integral_constant']]], - ['vec_5fsize_5',['vec_size',['../structmlx_1_1steel_1_1_block_loader.html#a58bdf9b9c81962733e22ecdeae28c092',1,'mlx::steel::BlockLoader::vec_size'],['../structmlx_1_1steel_1_1_block_loader_t.html#a9ac651d9e5097507c57b10dfeb40bfe5',1,'mlx::steel::BlockLoaderT::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#adcc83bf6c02391cc2375e55c06a1c9a4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a71c313e1597a2bb99f7b07d434e119d2',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a10109dc9553207f5a365799e4969c6d2',1,'mlx::steel::Conv2DWeightBlockLoader::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a6b0b18428516d1d6dcae3beb3faee81c',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a006153d274aa13d5fd4448b4607fff3a',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1587047caa339cf5b2c06adc4b332ab8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08dba753ec7c8ea2892775746933b3e7',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::vec_size']]] + ['value_4',['value',['../structmlx_1_1core_1_1simd_1_1_simd.html#a36e2b7db5ce6eb4dd456e99a4cd2c2cf',1,'mlx::core::simd::Simd::value'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#ac348445fd44bce2b6ee77adeac7d82df',1,'mlx::core::simd::Simd< T, 1 >::value'],['../structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#aa6042509bb67de25bedbee1ee1d66094',1,'mlx::core::simd::Simd< float16_t, N >::value'],['../structmlx_1_1steel_1_1integral__constant.html#a4efa69cb3fd42ac0dcad46578600d637',1,'mlx::steel::integral_constant::value'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a36e2b7db5ce6eb4dd456e99a4cd2c2cf',1,'mlx::core::simd::Simd< T, 1 >::value'],['../structmlx_1_1core_1_1simd_1_1_simd.html#a36e2b7db5ce6eb4dd456e99a4cd2c2cf',1,'mlx::core::simd::Simd< float16_t, N >::value']]], + ['vec_5fsize_5',['vec_size',['../structmlx_1_1steel_1_1_block_loader.html#a58bdf9b9c81962733e22ecdeae28c092',1,'mlx::steel::BlockLoader::vec_size'],['../structmlx_1_1steel_1_1_block_loader_t.html#a9ac651d9e5097507c57b10dfeb40bfe5',1,'mlx::steel::BlockLoaderT::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#adcc83bf6c02391cc2375e55c06a1c9a4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a71c313e1597a2bb99f7b07d434e119d2',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a10109dc9553207f5a365799e4969c6d2',1,'mlx::steel::Conv2DWeightBlockLoader::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#a71449551bbfe56058440755dfd50fc75',1,'mlx::steel::ChannelHelper< 1 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#acfb18991a77a9d1d4a79918ac5f387af',1,'mlx::steel::ChannelHelper< 2 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#a5cb83774601c29564a6bbc010fc0bf7f',1,'mlx::steel::ChannelHelper< 3 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#af28cdbe2a3c027d95832de07f60448ca',1,'mlx::steel::ChannelHelper< 4 >::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a6b0b18428516d1d6dcae3beb3faee81c',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a006153d274aa13d5fd4448b4607fff3a',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1587047caa339cf5b2c06adc4b332ab8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08dba753ec7c8ea2892775746933b3e7',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper< 1 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper< 2 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper< 3 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper< 4 >::vec_size']]] ]; diff --git a/docs/build/html/search/variables_3.js b/docs/build/html/search/variables_3.js index 7d3c6ad73..76a434569 100644 --- a/docs/build/html/search/variables_3.js +++ b/docs/build/html/search/variables_3.js @@ -5,7 +5,7 @@ var searchData= ['device_2',['device',['../structmlx_1_1core_1_1_stream.html#a406b1b0162287a4162fab1f70e2ff3bb',1,'mlx::core::Stream']]], ['digits_3',['digits',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#af6a681edff230c8d734a1feefb8d1879',1,'metal::_numeric_limits_impl< bfloat16_t >']]], ['digits10_4',['digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a0f48dd0c8a2d2dfa825067fb212b2e6b',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['dim_5',['dim',['../struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364',1,'LoopedElemToLoc']]], + ['dim_5',['dim',['../struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364',1,'LoopedElemToLoc::dim'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a7be6bf560080472d61e74b522979ef1e',1,'LoopedElemToLoc< 1, OffsetT, true >::dim'],['../struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364',1,'LoopedElemToLoc< 1, OffsetT, false >::dim'],['../struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364',1,'LoopedElemToLoc< 1, OffsetT, true >::dim']]], ['do_5faxpby_6',['do_axpby',['../steel__gemm__fused_8h.html#a703f06c849c89c37af7b1d27b0804a29',1,'steel_gemm_fused.h']]], ['do_5fgather_7',['do_gather',['../steel__gemm__fused_8h.html#a60efac3ac3b7cd64d096bbae38a3ac69',1,'steel_gemm_fused.h']]], ['do_5fread_8',['do_read',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a13eb86acf6abe288c19645935a47d2ad',1,'mlx::steel::Conv2DWeightBlockLoader::do_read'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a640155880483e1042ec5f647b9adaac6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::do_read']]], diff --git a/docs/build/html/search/variables_4.js b/docs/build/html/search/variables_4.js index 0fd7ea268..5dacc414d 100644 --- a/docs/build/html/search/variables_4.js +++ b/docs/build/html/search/variables_4.js @@ -4,5 +4,5 @@ var searchData= ['elems_5fper_5fthread_1',['elems_per_thread',['../struct_read_writer.html#a444230a0182ce6ba1898c04ce6e669a7',1,'ReadWriter']]], ['elems_5fper_5fthread_5f_2',['elems_per_thread_',['../backend_2metal_2kernels_2fft_8h.html#ad395c11e6f2aee72cd1928fba93a35a3',1,'fft.h']]], ['encoder_3',['encoder',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a58e435217b9922f882507ebf48bfbbdd',1,'mlx::core::metal::DeviceStream']]], - ['excess_4',['excess',['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper']]] + ['excess_4',['excess',['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#ada22a8bd8a89078cfa28874055c8e753',1,'mlx::steel::ChannelHelper< 1 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#acc490f3999230aa592c61bbed7eb7cfe',1,'mlx::steel::ChannelHelper< 2 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#aae404674763f3dc73c5ab29169f8b80f',1,'mlx::steel::ChannelHelper< 3 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#aecdd8331fec703d739a6f07b9b901ac8',1,'mlx::steel::ChannelHelper< 4 >::excess'],['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper< 1 >::excess'],['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper< 2 >::excess'],['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper< 3 >::excess'],['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper< 4 >::excess']]] ]; diff --git a/docs/build/html/search/variables_5.js b/docs/build/html/search/variables_5.js index b650631d7..563f10e07 100644 --- a/docs/build/html/search/variables_5.js +++ b/docs/build/html/search/variables_5.js @@ -1,19 +1,18 @@ var searchData= [ - ['f_0',['f',['../unionmlx_1_1core_1_1detail_1_1_int_or_float.html#a90bd738e3b6b96a1a67685e8ab75b7ec',1,'mlx::core::detail::IntOrFloat']]], - ['f_5fout_5fjump_5fh_1',['f_out_jump_h',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a78d48b55cf182f000abece0e5e7fadcb',1,'mlx::steel::Conv2DGeneralJumpParams']]], - ['f_5fout_5fjump_5fw_2',['f_out_jump_w',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a0fd755691482cb03ea4534b4a556c197',1,'mlx::steel::Conv2DGeneralJumpParams']]], - ['f_5fwgt_5fjump_5fh_3',['f_wgt_jump_h',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a198ba0c2740ab4ded99345edf58917a7',1,'mlx::steel::Conv2DGeneralJumpParams']]], - ['f_5fwgt_5fjump_5fw_4',['f_wgt_jump_w',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a568435a612574ab19a051a48055d4cfc',1,'mlx::steel::Conv2DGeneralJumpParams']]], - ['fdc_5',['fdc',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#a42efa2a1fddc11f71987377b9048f953',1,'mlx::steel::GEMMAddMMParams']]], - ['fence_6',['fence',['../structmlx_1_1core_1_1metal_1_1_fence.html#aeccd8f2b81418ae9fc446ae2b6e15b87',1,'mlx::core::metal::Fence::fence'],['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a876199de8da1efa9a362451029638499',1,'mlx::core::metal::DeviceStream::fence']]], - ['fence_5fmtx_7',['fence_mtx',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a6fa08cca881fc3798ae45994a11a4fcd',1,'mlx::core::metal::DeviceStream']]], - ['finite_5fmax_8',['finite_max',['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits']]], - ['finite_5fmin_9',['finite_min',['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits']]], - ['flip_10',['flip',['../struct_m_l_x_conv_params.html#a8b30cda15eda20f84f12db868f21d0ef',1,'MLXConvParams']]], - ['float16_11',['float16',['../namespacemlx_1_1core.html#abf228ee9d8ec48c03bb15adcc4e1f3ec',1,'mlx::core']]], - ['float32_12',['float32',['../namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319',1,'mlx::core']]], - ['floating_13',['floating',['../namespacemlx_1_1core.html#ac9f9ea13cf0661e671569d37d14a128a',1,'mlx::core']]], - ['forward_14',['FORWARD',['../namespacepocketfft_1_1detail.html#aecc5444a333360628be65a6f91ceb824',1,'pocketfft::detail::FORWARD'],['../namespacepocketfft.html#aecc5444a333360628be65a6f91ceb824',1,'pocketfft::FORWARD']]], - ['forward_15',['forward',['../structpocketfft_1_1detail_1_1_exec_c2_c.html#a63e27292b327597674deede9debe1c43',1,'pocketfft::detail::ExecC2C::forward'],['../structpocketfft_1_1detail_1_1_exec_r2_r.html#a5ec66ebb2ccd079f62b068ddd1fc7bdf',1,'pocketfft::detail::ExecR2R::forward']]] + ['f_5fout_5fjump_5fh_0',['f_out_jump_h',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a78d48b55cf182f000abece0e5e7fadcb',1,'mlx::steel::Conv2DGeneralJumpParams']]], + ['f_5fout_5fjump_5fw_1',['f_out_jump_w',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a0fd755691482cb03ea4534b4a556c197',1,'mlx::steel::Conv2DGeneralJumpParams']]], + ['f_5fwgt_5fjump_5fh_2',['f_wgt_jump_h',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a198ba0c2740ab4ded99345edf58917a7',1,'mlx::steel::Conv2DGeneralJumpParams']]], + ['f_5fwgt_5fjump_5fw_3',['f_wgt_jump_w',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#a568435a612574ab19a051a48055d4cfc',1,'mlx::steel::Conv2DGeneralJumpParams']]], + ['fdc_4',['fdc',['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#a42efa2a1fddc11f71987377b9048f953',1,'mlx::steel::GEMMAddMMParams']]], + ['fence_5',['fence',['../structmlx_1_1core_1_1metal_1_1_fence.html#aeccd8f2b81418ae9fc446ae2b6e15b87',1,'mlx::core::metal::Fence::fence'],['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a876199de8da1efa9a362451029638499',1,'mlx::core::metal::DeviceStream::fence']]], + ['fence_5fmtx_6',['fence_mtx',['../structmlx_1_1core_1_1metal_1_1_device_stream.html#a6fa08cca881fc3798ae45994a11a4fcd',1,'mlx::core::metal::DeviceStream']]], + ['finite_5fmax_7',['finite_max',['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits::finite_max'],['../struct_limits_3_01uint8__t_01_4.html#a55f48b89033e8c8683f8540ec6b23f02',1,'Limits< uint8_t >::finite_max'],['../struct_limits_3_01uint16__t_01_4.html#a9d517d8265ea1898b6b16e91b8595146',1,'Limits< uint16_t >::finite_max'],['../struct_limits_3_01uint32__t_01_4.html#a0698139f3fe440d7aa08ac5029d72235',1,'Limits< uint32_t >::finite_max'],['../struct_limits_3_01uint64__t_01_4.html#aff101ff38be5ccdbb9790aecb3069071',1,'Limits< uint64_t >::finite_max'],['../struct_limits_3_01int8__t_01_4.html#a24cdab873e0fb778393c69f1dc9ecf73',1,'Limits< int8_t >::finite_max'],['../struct_limits_3_01int16__t_01_4.html#acb2936d1cdbf347a9a014c8e036a5782',1,'Limits< int16_t >::finite_max'],['../struct_limits_3_01int32__t_01_4.html#aa9ed9f0e8c7400d8fc92e1cba9588794',1,'Limits< int32_t >::finite_max'],['../struct_limits_3_01int64__t_01_4.html#a6c7254b641878fa0fb9538814c45457a',1,'Limits< int64_t >::finite_max'],['../struct_limits_3_01half_01_4.html#aedaf0190aabf23da20510e558e2690b4',1,'Limits< half >::finite_max'],['../struct_limits_3_01float_01_4.html#a291eea590113fc1858b7f83f2e0c977d',1,'Limits< float >::finite_max'],['../struct_limits_3_01bfloat16__t_01_4.html#a6337dc35207b3f6f7185cd73eabac211',1,'Limits< bfloat16_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< bfloat16_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< bool >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< complex64_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< float >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< half >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< int16_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< int32_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< int64_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< int8_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< uint16_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< uint32_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< uint64_t >::finite_max'],['../struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5',1,'Limits< uint8_t >::finite_max']]], + ['finite_5fmin_8',['finite_min',['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits::finite_min'],['../struct_limits_3_01uint8__t_01_4.html#a60cea662971b09f78ef19f1da4760b73',1,'Limits< uint8_t >::finite_min'],['../struct_limits_3_01uint16__t_01_4.html#a1a7c029eccba4ab89743abdfaabfa7b4',1,'Limits< uint16_t >::finite_min'],['../struct_limits_3_01uint32__t_01_4.html#ad5d811fce62f44488190ff01d9e7608b',1,'Limits< uint32_t >::finite_min'],['../struct_limits_3_01uint64__t_01_4.html#a6556e7de6e0670da8f768bbc4479deae',1,'Limits< uint64_t >::finite_min'],['../struct_limits_3_01int8__t_01_4.html#a592797ce82cc2f7e27b0c477165b3452',1,'Limits< int8_t >::finite_min'],['../struct_limits_3_01int16__t_01_4.html#a158c4dbc9333939691b1637478e28e39',1,'Limits< int16_t >::finite_min'],['../struct_limits_3_01int32__t_01_4.html#ad9777dc6a84dcb9c63b598189ff0a4ff',1,'Limits< int32_t >::finite_min'],['../struct_limits_3_01int64__t_01_4.html#af80726162b44a741aae679f1fe85142a',1,'Limits< int64_t >::finite_min'],['../struct_limits_3_01half_01_4.html#a98d153748be68dbb428c50df3c0285ab',1,'Limits< half >::finite_min'],['../struct_limits_3_01float_01_4.html#afaa5162a47083447c5ac758d6dc02a8b',1,'Limits< float >::finite_min'],['../struct_limits_3_01bfloat16__t_01_4.html#ae4132a37154707cc31bbc1734636cf36',1,'Limits< bfloat16_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< bfloat16_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< bool >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< complex64_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< float >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< half >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< int16_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< int32_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< int64_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< int8_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< uint16_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< uint32_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< uint64_t >::finite_min'],['../struct_limits.html#ae7469d21f2688797ca3e388d919ef05e',1,'Limits< uint8_t >::finite_min']]], + ['flip_9',['flip',['../struct_m_l_x_conv_params.html#a8b30cda15eda20f84f12db868f21d0ef',1,'MLXConvParams']]], + ['float16_10',['float16',['../namespacemlx_1_1core.html#abf228ee9d8ec48c03bb15adcc4e1f3ec',1,'mlx::core']]], + ['float32_11',['float32',['../namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319',1,'mlx::core']]], + ['floating_12',['floating',['../namespacemlx_1_1core.html#ac9f9ea13cf0661e671569d37d14a128a',1,'mlx::core']]], + ['forward_13',['FORWARD',['../namespacepocketfft_1_1detail.html#aecc5444a333360628be65a6f91ceb824',1,'pocketfft::detail::FORWARD'],['../namespacepocketfft.html#aecc5444a333360628be65a6f91ceb824',1,'pocketfft::FORWARD']]], + ['forward_14',['forward',['../structpocketfft_1_1detail_1_1_exec_c2_c.html#a63e27292b327597674deede9debe1c43',1,'pocketfft::detail::ExecC2C::forward'],['../structpocketfft_1_1detail_1_1_exec_r2_r.html#a5ec66ebb2ccd079f62b068ddd1fc7bdf',1,'pocketfft::detail::ExecR2R::forward']]] ]; diff --git a/docs/build/html/search/variables_6.js b/docs/build/html/search/variables_6.js index 1249c5b97..b1804a11a 100644 --- a/docs/build/html/search/variables_6.js +++ b/docs/build/html/search/variables_6.js @@ -5,13 +5,12 @@ var searchData= ['gemm_5fk_5fiterations_2',['gemm_k_iterations',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a8b50863e4e2d3481c154be6c3629bf51',1,'mlx::steel::ImplicitGemmConv2DParams']]], ['gemm_5fk_5fiterations_5faligned_3',['gemm_k_iterations_aligned',['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0d7f419ba265805b418e93ce1ca2e0f9',1,'mlx::steel::GEMMParams::gemm_k_iterations_aligned'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#aa37e05a03ac8b34ec7dc31ca42f68998',1,'mlx::steel::GEMMSpiltKParams::gemm_k_iterations_aligned']]], ['gemm_5fparams_4',['gemm_params',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ab0724eb3ef52ee773b6607f6433b9f2c',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::gemm_params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#acc778b3c0b7ec38a43e8ea943df8704c',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::gemm_params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af59f9d356c4c3ec5627dc5a263d239d4',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::gemm_params']]], - ['gemv_5fmasked_5fkernel_5',['gemv_masked_kernel',['../jit_2gemv__masked_8h.html#a933f06c211f86c37673dee329ed6901f',1,'gemv_masked.h']]], - ['generic_6',['generic',['../namespacemlx_1_1core.html#a34d69c4d46aa9b2a4a79dba7aba093d2',1,'mlx::core']]], - ['gpu_7',['gpu',['../structmlx_1_1core_1_1_device.html#a45ed081b56ae5d4ddd39c83a5d8a1616',1,'mlx::core::Device']]], - ['gqa_5ffactor_8',['gqa_factor',['../structmlx_1_1steel_1_1_attn_params.html#a3b3e18cb993ab24819c852bc64288841',1,'mlx::steel::AttnParams']]], - ['grid_9',['grid',['../struct_read_writer.html#ac7a957f99873d3797081f5d620f3b2c8',1,'ReadWriter']]], - ['group_5fstep_5fcnt_10',['group_step_cnt',['../struct_quantized_block_loader.html#a234feacde36a4afc0d740332a3769fb6',1,'QuantizedBlockLoader']]], - ['group_5fsteps_11',['group_steps',['../struct_quantized_block_loader.html#a31e14175f3d4902d9fe5ab5a219f61ba',1,'QuantizedBlockLoader']]], - ['group_5fstride_12',['group_stride',['../struct_quantized_block_loader.html#a0ace7e3762ecfa5a4106e7dee7e1b6ab',1,'QuantizedBlockLoader']]], - ['groups_13',['groups',['../struct_m_l_x_conv_params.html#af7a5590ac0974c7841c7f8b9fda0cbed',1,'MLXConvParams']]] + ['generic_5',['generic',['../namespacemlx_1_1core.html#a34d69c4d46aa9b2a4a79dba7aba093d2',1,'mlx::core']]], + ['gpu_6',['gpu',['../structmlx_1_1core_1_1_device.html#a45ed081b56ae5d4ddd39c83a5d8a1616',1,'mlx::core::Device']]], + ['gqa_5ffactor_7',['gqa_factor',['../structmlx_1_1steel_1_1_attn_params.html#a3b3e18cb993ab24819c852bc64288841',1,'mlx::steel::AttnParams']]], + ['grid_8',['grid',['../struct_read_writer.html#ac7a957f99873d3797081f5d620f3b2c8',1,'ReadWriter']]], + ['group_5fstep_5fcnt_9',['group_step_cnt',['../struct_quantized_block_loader.html#a234feacde36a4afc0d740332a3769fb6',1,'QuantizedBlockLoader']]], + ['group_5fsteps_10',['group_steps',['../struct_quantized_block_loader.html#a31e14175f3d4902d9fe5ab5a219f61ba',1,'QuantizedBlockLoader']]], + ['group_5fstride_11',['group_stride',['../struct_quantized_block_loader.html#a0ace7e3762ecfa5a4106e7dee7e1b6ab',1,'QuantizedBlockLoader']]], + ['groups_12',['groups',['../struct_m_l_x_conv_params.html#af7a5590ac0974c7841c7f8b9fda0cbed',1,'MLXConvParams']]] ]; diff --git a/docs/build/html/search/variables_8.js b/docs/build/html/search/variables_8.js index e957505d2..eb1ee606c 100644 --- a/docs/build/html/search/variables_8.js +++ b/docs/build/html/search/variables_8.js @@ -1,28 +1,31 @@ var searchData= [ - ['i_0',['i',['../structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d',1,'pocketfft::detail::cmplx::i'],['../unionmlx_1_1core_1_1detail_1_1_int_or_float.html#aa6d4b02e99df90a321e0eeeb98e71b49',1,'mlx::core::detail::IntOrFloat::i'],['../unionbool4__or__uint.html#ab24d95aaf4203ddf3e6b1ed19397ced7',1,'bool4_or_uint::i']]], + ['i_0',['i',['../structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d',1,'pocketfft::detail::cmplx::i'],['../unionbool4__or__uint.html#ab24d95aaf4203ddf3e6b1ed19397ced7',1,'bool4_or_uint::i']]], ['idil_1',['idil',['../struct_m_l_x_conv_params.html#af900fdb4c4d4ea35eed02940dee8d4d1',1,'MLXConvParams']]], ['imag_2',['imag',['../structcomplex64__t.html#a94037c0cf8451aaff7cb4d154a8426de',1,'complex64_t']]], ['in_3',['in',['../struct_read_writer.html#ab6057215920138f28fd00f0e7ea8afa4',1,'ReadWriter']]], ['in_5fstrides_4',['in_strides',['../struct_m_l_x_conv_params.html#a98e0b625974a98512b63b8a00bd8df5f',1,'MLXConvParams']]], - ['index_5',['index',['../struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333',1,'LoopedElemToLoc::index'],['../structmlx_1_1core_1_1_device.html#a5e345748fe318a267833ab7398b364ac',1,'mlx::core::Device::index'],['../structmlx_1_1core_1_1_stream.html#a9d0dafc1899333e1176eb2bbc0a8b626',1,'mlx::core::Stream::index']]], + ['index_5',['index',['../struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333',1,'LoopedElemToLoc::index'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a2cd3b616739b3d5b41e5b46ae335957d',1,'LoopedElemToLoc< 1, OffsetT, true >::index'],['../structmlx_1_1core_1_1_device.html#a5e345748fe318a267833ab7398b364ac',1,'mlx::core::Device::index'],['../structmlx_1_1core_1_1_stream.html#a9d0dafc1899333e1176eb2bbc0a8b626',1,'mlx::core::Stream::index'],['../struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333',1,'LoopedElemToLoc< 1, OffsetT, false >::index'],['../struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333',1,'LoopedElemToLoc< 1, OffsetT, true >::index']]], ['inexact_6',['inexact',['../namespacemlx_1_1core.html#a54c6fae21b7f2fea8e6f80011ef38534',1,'mlx::core']]], - ['init_7',['init',['../struct_cum_max.html#a16480052a2eeb4340e546838aab59cc4',1,'CumMax::init'],['../struct_cum_min.html#a8b67f739c620d0cc194b533190990ab9',1,'CumMin::init'],['../struct_less_than.html#abf97a6b0163048e4ba96460939dbd3a3',1,'LessThan::init']]], - ['inner_5flooper_8',['inner_looper',['../struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40',1,'LoopedElemToLoc']]], - ['inp_5fjump_5fc_9',['inp_jump_c',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a78d30e843d65d1829623afb0b607f0a5',1,'mlx::steel::ImplicitGemmConv2DParams']]], - ['inp_5fjump_5fh_10',['inp_jump_h',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a03685a4066cdb11ffb647408e2c5b122',1,'mlx::steel::ImplicitGemmConv2DParams']]], - ['inp_5fjump_5fw_11',['inp_jump_w',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#acf168c72f4a86b72b8f5f386f07c9d8c',1,'mlx::steel::ImplicitGemmConv2DParams']]], - ['int16_12',['int16',['../namespacemlx_1_1core.html#a71ebba4ad1afa730962f0692c4f42f07',1,'mlx::core']]], - ['int32_13',['int32',['../namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3',1,'mlx::core']]], - ['int64_14',['int64',['../namespacemlx_1_1core.html#a9019bdc191054ada0a502c7c34cef5b8',1,'mlx::core']]], - ['int8_15',['int8',['../namespacemlx_1_1core.html#a892e934e146dd938d144cee8813ca672',1,'mlx::core']]], - ['integer_16',['integer',['../namespacemlx_1_1core.html#a074d000f25ae3ed77450e6a5fec4b38b',1,'mlx::core']]], - ['inv_17',['inv',['../struct_read_writer.html#a773fa8524515bdc2ff8b0e2060741813',1,'ReadWriter']]], - ['inv_5f_18',['inv_',['../backend_2metal_2kernels_2fft_8h.html#a7a83318497519ff3ff0141b7d511ed38',1,'fft.h']]], - ['is_19',['iS',['../struct_m_l_x_conv_params.html#a72e1c3b4da0f70622cf18036bbf97fe6',1,'MLXConvParams']]], - ['is_5farray_5fv_20',['is_array_v',['../namespacemlx_1_1core.html#a01b0d64a75dfa2e95d6c7b5c53d708af',1,'mlx::core']]], - ['is_5farrays_5fv_21',['is_arrays_v',['../namespacemlx_1_1core.html#a94c1057929b390e5613304afa16dfbda',1,'mlx::core']]], - ['is_5fintegral_5fv_22',['is_integral_v',['../namespacemlx_1_1steel.html#a92a3465716ea7fd682d22cecc08d45fd',1,'mlx::steel']]], - ['is_5fmetal_5fatomic_23',['is_metal_atomic',['../atomic_8h.html#a91a8bdcae647947a83c6689d7f252d24',1,'atomic.h']]], - ['is_5fpower_5fof_5f2_5f_24',['is_power_of_2_',['../backend_2metal_2kernels_2fft_8h.html#a2a4df90e329b84ee6c1890ba7c265c9c',1,'fft.h']]] + ['inf_7',['inf',['../namespacemlx_1_1core_1_1simd.html#a771b6597803beb800ff5e7560c41e341',1,'mlx::core::simd']]], + ['init_8',['init',['../struct_cum_prod_3_01bool_01_4.html#ae7a8b0ba9e6898356b87b18766e76d2c',1,'CumProd< bool >::init'],['../struct_cum_max.html#a16480052a2eeb4340e546838aab59cc4',1,'CumMax::init'],['../struct_cum_min.html#a8b67f739c620d0cc194b533190990ab9',1,'CumMin::init'],['../struct_less_than.html#abf97a6b0163048e4ba96460939dbd3a3',1,'LessThan::init']]], + ['inner_5flooper_9',['inner_looper',['../struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40',1,'LoopedElemToLoc::inner_looper'],['../struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40',1,'LoopedElemToLoc< 1, OffsetT, false >::inner_looper'],['../struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40',1,'LoopedElemToLoc< 1, OffsetT, true >::inner_looper']]], + ['inp_5fjump_5fc_10',['inp_jump_c',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a78d30e843d65d1829623afb0b607f0a5',1,'mlx::steel::ImplicitGemmConv2DParams']]], + ['inp_5fjump_5fh_11',['inp_jump_h',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a03685a4066cdb11ffb647408e2c5b122',1,'mlx::steel::ImplicitGemmConv2DParams']]], + ['inp_5fjump_5fw_12',['inp_jump_w',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#acf168c72f4a86b72b8f5f386f07c9d8c',1,'mlx::steel::ImplicitGemmConv2DParams']]], + ['int16_13',['int16',['../namespacemlx_1_1core.html#a71ebba4ad1afa730962f0692c4f42f07',1,'mlx::core']]], + ['int32_14',['int32',['../namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3',1,'mlx::core']]], + ['int64_15',['int64',['../namespacemlx_1_1core.html#a9019bdc191054ada0a502c7c34cef5b8',1,'mlx::core']]], + ['int8_16',['int8',['../namespacemlx_1_1core.html#a892e934e146dd938d144cee8813ca672',1,'mlx::core']]], + ['integer_17',['integer',['../namespacemlx_1_1core.html#a074d000f25ae3ed77450e6a5fec4b38b',1,'mlx::core']]], + ['inv_18',['inv',['../struct_read_writer.html#a773fa8524515bdc2ff8b0e2060741813',1,'ReadWriter']]], + ['inv_5f_19',['inv_',['../backend_2metal_2kernels_2fft_8h.html#a7a83318497519ff3ff0141b7d511ed38',1,'fft.h']]], + ['is_20',['iS',['../struct_m_l_x_conv_params.html#a72e1c3b4da0f70622cf18036bbf97fe6',1,'MLXConvParams']]], + ['is_5farray_5fv_21',['is_array_v',['../namespacemlx_1_1core.html#a01b0d64a75dfa2e95d6c7b5c53d708af',1,'mlx::core']]], + ['is_5farrays_5fv_22',['is_arrays_v',['../namespacemlx_1_1core.html#a94c1057929b390e5613304afa16dfbda',1,'mlx::core']]], + ['is_5fcomplex_23',['is_complex',['../namespacemlx_1_1core_1_1simd.html#a64e80f096a8baf99ba8d396414473cc7',1,'mlx::core::simd']]], + ['is_5fcomplex_3c_20t_2c_20std_3a_3avoid_5ft_3c_20decltype_28std_3a_3adeclval_3c_20t_20_3e_28_29_2ereal_28_29_29_3e_20_3e_24',['is_complex< T, std::void_t< decltype(std::declval< T >().real())> >',['../namespacemlx_1_1core_1_1simd.html#a29fe8445e54a61f6bccc8d50f142ca54',1,'mlx::core::simd']]], + ['is_5fintegral_5fv_25',['is_integral_v',['../namespacemlx_1_1steel.html#a92a3465716ea7fd682d22cecc08d45fd',1,'mlx::steel']]], + ['is_5fmetal_5fatomic_26',['is_metal_atomic',['../atomic_8h.html#a91a8bdcae647947a83c6689d7f252d24',1,'atomic.h']]], + ['is_5fpower_5fof_5f2_5f_27',['is_power_of_2_',['../backend_2metal_2kernels_2fft_8h.html#a2a4df90e329b84ee6c1890ba7c265c9c',1,'fft.h']]] ]; diff --git a/docs/build/html/search/variables_a.js b/docs/build/html/search/variables_a.js index 8045ac906..87ce2ae97 100644 --- a/docs/build/html/search/variables_a.js +++ b/docs/build/html/search/variables_a.js @@ -5,15 +5,17 @@ var searchData= ['kcols_2',['kCols',['../structmlx_1_1steel_1_1_c_shape.html#a01b09227356b6a682a0694523a8e6901',1,'mlx::steel::CShape::kCols'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a948784652e93830887ee8ad506ec3257',1,'mlx::steel::MMATile::kCols']]], ['kcolsperthread_3',['kColsPerThread',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1ea49efd92696b15302ee4b52ecd548c',1,'mlx::steel::MMATile']]], ['kdil_4',['kdil',['../struct_m_l_x_conv_params.html#a7611db8f1621c7e09fc685ed44073b14',1,'MLXConvParams']]], - ['kelemsperfrag_5',['kElemsPerFrag',['../structmlx_1_1steel_1_1_m_m_a_tile.html#aef0ea2387e1ff5767bff8563b2d36bd6',1,'mlx::steel::MMATile']]], - ['kelemspertile_6',['kElemsPerTile',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a98357339ec98f804a1b12597937b318f',1,'mlx::steel::MMATile']]], - ['kfragcols_7',['kFragCols',['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad095371db98e7c335ec41ca77c10f906',1,'mlx::steel::MMATile']]], - ['kfragrows_8',['kFragRows',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a594142f957ffb99296a243f7af7b59e7',1,'mlx::steel::MMATile']]], - ['kfragsize_9',['kFragSize',['../structmlx_1_1steel_1_1_block_m_m_a.html#aee8caec45c1f9e4428586effbfe6137d',1,'mlx::steel::BlockMMA']]], - ['kl_10',['kL',['../structmlx_1_1steel_1_1_attn_params.html#a497b7404bcd25b535c3589c61f269f63',1,'mlx::steel::AttnParams']]], - ['knumfrags_11',['kNumFrags',['../structmlx_1_1steel_1_1_m_m_a_tile.html#ae326e7693eb77c22d5a6e3e9219019d3',1,'mlx::steel::MMATile']]], - ['krows_12',['kRows',['../structmlx_1_1steel_1_1_c_shape.html#a5caf36cb9acf9f90ba59a9b0b4197993',1,'mlx::steel::CShape::kRows'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a60ea6b8ff2923b7fe6f598e74ac54323',1,'mlx::steel::MMATile::kRows']]], - ['krowsperthread_13',['kRowsPerThread',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a5b1d1c85a5046108a4e38bdc5a0ea74e',1,'mlx::steel::MMATile']]], - ['ktilecols_14',['kTileCols',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a46324d40f8ad61cade08a1ebad6d9ad4',1,'mlx::steel::MMATile']]], - ['ktilerows_15',['kTileRows',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1d126b14910385ab644e224ac1d0307a',1,'mlx::steel::MMATile']]] + ['kelemcols_5',['kElemCols',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a7c212200d86b4e93f274d99addf668bd',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >']]], + ['kelemrows_6',['kElemRows',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a76aa5aa690dbcc954e957d767fad661f',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >']]], + ['kelemsperfrag_7',['kElemsPerFrag',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a3c34dfdc944db110f4735f1b25307cf0',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::kElemsPerFrag'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#aef0ea2387e1ff5767bff8563b2d36bd6',1,'mlx::steel::MMATile::kElemsPerFrag']]], + ['kelemspertile_8',['kElemsPerTile',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a98357339ec98f804a1b12597937b318f',1,'mlx::steel::MMATile']]], + ['kfragcols_9',['kFragCols',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a211102315e2afbcfcd2e2c201b638e9f',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::kFragCols'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#ad095371db98e7c335ec41ca77c10f906',1,'mlx::steel::MMATile::kFragCols']]], + ['kfragrows_10',['kFragRows',['../structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a2fe53db449c692226f23f6b99fb2c0d4',1,'mlx::steel::BaseMMAFrag< T, 8, 8 >::kFragRows'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a594142f957ffb99296a243f7af7b59e7',1,'mlx::steel::MMATile::kFragRows']]], + ['kfragsize_11',['kFragSize',['../structmlx_1_1steel_1_1_block_m_m_a.html#aee8caec45c1f9e4428586effbfe6137d',1,'mlx::steel::BlockMMA']]], + ['kl_12',['kL',['../structmlx_1_1steel_1_1_attn_params.html#a497b7404bcd25b535c3589c61f269f63',1,'mlx::steel::AttnParams']]], + ['knumfrags_13',['kNumFrags',['../structmlx_1_1steel_1_1_m_m_a_tile.html#ae326e7693eb77c22d5a6e3e9219019d3',1,'mlx::steel::MMATile']]], + ['krows_14',['kRows',['../structmlx_1_1steel_1_1_c_shape.html#a5caf36cb9acf9f90ba59a9b0b4197993',1,'mlx::steel::CShape::kRows'],['../structmlx_1_1steel_1_1_m_m_a_tile.html#a60ea6b8ff2923b7fe6f598e74ac54323',1,'mlx::steel::MMATile::kRows']]], + ['krowsperthread_15',['kRowsPerThread',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a5b1d1c85a5046108a4e38bdc5a0ea74e',1,'mlx::steel::MMATile']]], + ['ktilecols_16',['kTileCols',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a46324d40f8ad61cade08a1ebad6d9ad4',1,'mlx::steel::MMATile']]], + ['ktilerows_17',['kTileRows',['../structmlx_1_1steel_1_1_m_m_a_tile.html#a1d126b14910385ab644e224ac1d0307a',1,'mlx::steel::MMATile']]] ]; diff --git a/docs/build/html/search/variables_c.js b/docs/build/html/search/variables_c.js index 2fb2ac910..00c5eda48 100644 --- a/docs/build/html/search/variables_c.js +++ b/docs/build/html/search/variables_c.js @@ -3,14 +3,26 @@ var searchData= ['m_0',['M',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a2117fc93662d5177c8f3e7c2dbb9e2db',1,'mlx::steel::ImplicitGemmConv2DParams::M'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a85b20a4c4558cc78d76fcbd045a9c694',1,'mlx::steel::GEMMParams::M'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a8bab0cf8a20d2abefe294a7505917e7e',1,'mlx::steel::GEMMSpiltKParams::M']]], ['mask_5fh_1',['mask_h',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0b892c1a7edb9ed20c076d8945855c19',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], ['mask_5fw_2',['mask_w',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a19ddba7259c3c2c02ed90f3f635557be',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], - ['max_3',['max',['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits::max'],['../structmlx_1_1core_1_1finfo.html#a9975908d99d25e9225d0c70bead0abc2',1,'mlx::core::finfo::max']]], + ['max_3',['max',['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits::max'],['../struct_limits_3_01uint8__t_01_4.html#a1570fb640e2e41f96776db5ca08d500c',1,'Limits< uint8_t >::max'],['../struct_limits_3_01uint16__t_01_4.html#a228b33556ba4cb7e6137ab6258628488',1,'Limits< uint16_t >::max'],['../struct_limits_3_01uint32__t_01_4.html#a91fa8f7214ec936976a8324c7431c651',1,'Limits< uint32_t >::max'],['../struct_limits_3_01uint64__t_01_4.html#aa8c2257881a4e1fa8596fa07dba5e107',1,'Limits< uint64_t >::max'],['../struct_limits_3_01int8__t_01_4.html#a96fed01fa9249226be69760652643289',1,'Limits< int8_t >::max'],['../struct_limits_3_01int16__t_01_4.html#a12d64c398ca7609b7c906f3cf1a6f678',1,'Limits< int16_t >::max'],['../struct_limits_3_01int32__t_01_4.html#af756344b31e84222dd73d3445dcd5640',1,'Limits< int32_t >::max'],['../struct_limits_3_01int64__t_01_4.html#ac9c420604c0f3d237ddfb2b8a2439224',1,'Limits< int64_t >::max'],['../struct_limits_3_01half_01_4.html#a4f9515dbf2a622074f121bea39a7b175',1,'Limits< half >::max'],['../struct_limits_3_01float_01_4.html#aba172b22b388190aa3969ef16885d8a6',1,'Limits< float >::max'],['../struct_limits_3_01bfloat16__t_01_4.html#a0ead3618da6718629ea9fa4670b5005f',1,'Limits< bfloat16_t >::max'],['../struct_limits_3_01bool_01_4.html#acbd2132145888d51220558a101ffcff4',1,'Limits< bool >::max'],['../struct_limits_3_01complex64__t_01_4.html#ac01c274b224b90f5210b675a484f4607',1,'Limits< complex64_t >::max'],['../structmlx_1_1core_1_1finfo.html#a9975908d99d25e9225d0c70bead0abc2',1,'mlx::core::finfo::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< bfloat16_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< bool >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< complex64_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< float >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< half >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< int16_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< int32_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< int64_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< int8_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< uint16_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< uint32_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< uint64_t >::max'],['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits< uint8_t >::max']]], ['max_5fdigits10_4',['max_digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a8d3905e6f158379a0c52682266e8d0e2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], ['max_5fexponent_5',['max_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61bb136f819fa392c50bdf3c38f3aad2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], ['max_5fexponent10_6',['max_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a76bfb2deb0e0afc011f77bf5a6d0ed94',1,'metal::_numeric_limits_impl< bfloat16_t >']]], ['max_5freduce_5fspecialized_5fdims_7',['MAX_REDUCE_SPECIALIZED_DIMS',['../defines_8h.html#a15629f1b81a2b6f1cca26d07a2734623',1,'defines.h']]], - ['max_5fthreads_8',['max_threads',['../namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70',1,'pocketfft::detail::threading']]], - ['min_9',['min',['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits::min'],['../structmlx_1_1core_1_1finfo.html#a91332dcbb502799b6cc8e1c6933c042b',1,'mlx::core::finfo::min']]], - ['min_5fexponent_10',['min_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a13829f8c7a7c0efdc8946eff5d3c9470',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['min_5fexponent10_11',['min_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#aeaed172780720e06b8731cef3177e277',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['mtx_12',['mtx',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a70410c9e612f871663929f1e8441a976',1,'mlx::core::scheduler::StreamThread']]] + ['max_5fsize_8',['max_size',['../namespacemlx_1_1core_1_1simd.html#ac91bd36c7caafd3c7ff176e7e2f81887',1,'mlx::core::simd']]], + ['max_5fsize_3c_20double_20_3e_9',['max_size< double >',['../namespacemlx_1_1core_1_1simd.html#a3fa3d1f571027c5cdd1dce5d2cd041e3',1,'mlx::core::simd']]], + ['max_5fsize_3c_20float_20_3e_10',['max_size< float >',['../namespacemlx_1_1core_1_1simd.html#ae745e117cacfe455df39aa4569c34c11',1,'mlx::core::simd']]], + ['max_5fsize_3c_20float16_5ft_20_3e_11',['max_size< float16_t >',['../namespacemlx_1_1core_1_1simd.html#a155df1de3c26e1a3725b63e9e97c0b53',1,'mlx::core::simd']]], + ['max_5fsize_3c_20int_20_3e_12',['max_size< int >',['../namespacemlx_1_1core_1_1simd.html#ab25fc96fa6f00d0a8c335b8da293fbbb',1,'mlx::core::simd']]], + ['max_5fsize_3c_20int16_5ft_20_3e_13',['max_size< int16_t >',['../namespacemlx_1_1core_1_1simd.html#a7e63a5eb08898b84fd4000dadc460fd9',1,'mlx::core::simd']]], + ['max_5fsize_3c_20int64_5ft_20_3e_14',['max_size< int64_t >',['../namespacemlx_1_1core_1_1simd.html#a7913cb2854ffc37efcf26635a097f0a9',1,'mlx::core::simd']]], + ['max_5fsize_3c_20int8_5ft_20_3e_15',['max_size< int8_t >',['../namespacemlx_1_1core_1_1simd.html#ac368e4701363cfece4935e57f3c709b1',1,'mlx::core::simd']]], + ['max_5fsize_3c_20uint16_5ft_20_3e_16',['max_size< uint16_t >',['../namespacemlx_1_1core_1_1simd.html#a0cc9ca2925c25d2eb225af9125bd6bc4',1,'mlx::core::simd']]], + ['max_5fsize_3c_20uint32_5ft_20_3e_17',['max_size< uint32_t >',['../namespacemlx_1_1core_1_1simd.html#a06cb29f91deeaec69471058044abd2aa',1,'mlx::core::simd']]], + ['max_5fsize_3c_20uint64_5ft_20_3e_18',['max_size< uint64_t >',['../namespacemlx_1_1core_1_1simd.html#ab367b9b65be2fda4830a56fc9cc0cd2f',1,'mlx::core::simd']]], + ['max_5fsize_3c_20uint8_5ft_20_3e_19',['max_size< uint8_t >',['../namespacemlx_1_1core_1_1simd.html#a8f731e5a287c714dfc92879fe37503d5',1,'mlx::core::simd']]], + ['max_5fthreads_20',['max_threads',['../namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70',1,'pocketfft::detail::threading']]], + ['min_21',['min',['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits::min'],['../struct_limits_3_01uint8__t_01_4.html#a408bd5a337e7292f06e63da81193629a',1,'Limits< uint8_t >::min'],['../struct_limits_3_01uint16__t_01_4.html#ae173984c3be8b6750f27daed581805fe',1,'Limits< uint16_t >::min'],['../struct_limits_3_01uint32__t_01_4.html#ab0c3975e02053b234c7b606ababa66e1',1,'Limits< uint32_t >::min'],['../struct_limits_3_01uint64__t_01_4.html#a80627f39e951398283942cefa48f4dd0',1,'Limits< uint64_t >::min'],['../struct_limits_3_01int8__t_01_4.html#a7a809307d2bba80382f0645d277eaa4b',1,'Limits< int8_t >::min'],['../struct_limits_3_01int16__t_01_4.html#adca7139647801e223c35b0abc7da5240',1,'Limits< int16_t >::min'],['../struct_limits_3_01int32__t_01_4.html#af336a1b22a8ed6a83a4cfb5bf8869771',1,'Limits< int32_t >::min'],['../struct_limits_3_01int64__t_01_4.html#a1c90fb96af515badaccaa835b08f7428',1,'Limits< int64_t >::min'],['../struct_limits_3_01half_01_4.html#aca7b036c257878bf1b80912fb5d4516d',1,'Limits< half >::min'],['../struct_limits_3_01float_01_4.html#a3225e334d372ee86128c89a440d8648f',1,'Limits< float >::min'],['../struct_limits_3_01bfloat16__t_01_4.html#a2fd1811b9f615b2b897904bc27d1cb49',1,'Limits< bfloat16_t >::min'],['../struct_limits_3_01bool_01_4.html#a139f787b57536d455490b8ef801d37cc',1,'Limits< bool >::min'],['../struct_limits_3_01complex64__t_01_4.html#aa67b04aa7abcd67f7af0808737ab8e14',1,'Limits< complex64_t >::min'],['../structmlx_1_1core_1_1finfo.html#a91332dcbb502799b6cc8e1c6933c042b',1,'mlx::core::finfo::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< bfloat16_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< bool >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< complex64_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< float >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< half >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< int16_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< int32_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< int64_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< int8_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< uint16_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< uint32_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< uint64_t >::min'],['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits< uint8_t >::min']]], + ['min_5fexponent_22',['min_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a13829f8c7a7c0efdc8946eff5d3c9470',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['min_5fexponent10_23',['min_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#aeaed172780720e06b8731cef3177e277',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['mtx_24',['mtx',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a70410c9e612f871663929f1e8441a976',1,'mlx::core::scheduler::StreamThread']]] ]; diff --git a/docs/build/html/search/variables_d.js b/docs/build/html/search/variables_d.js index 7bb10f43c..25eb614ce 100644 --- a/docs/build/html/search/variables_d.js +++ b/docs/build/html/search/variables_d.js @@ -1,9 +1,9 @@ var searchData= [ - ['n_0',['N',['../struct_m_l_x_conv_params.html#ae6b7054dc3cffa8e6aedeb29fa7da932',1,'MLXConvParams::N'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a213f5ea4018120d8b61ab82754aaba83',1,'mlx::steel::ImplicitGemmConv2DParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a174626ab98515d89923b2841a664b9a1',1,'mlx::steel::GEMMParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a1103e79fb8962812b9a3c9d5c902ff86',1,'mlx::steel::GEMMSpiltKParams::N']]], + ['n_0',['N',['../struct_m_l_x_conv_params.html#ae6b7054dc3cffa8e6aedeb29fa7da932',1,'MLXConvParams::N'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a213f5ea4018120d8b61ab82754aaba83',1,'mlx::steel::ImplicitGemmConv2DParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a174626ab98515d89923b2841a664b9a1',1,'mlx::steel::GEMMParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a1103e79fb8962812b9a3c9d5c902ff86',1,'mlx::steel::GEMMSpiltKParams::N'],['../namespacemlx_1_1core_1_1simd.html#a12b1553495a0c99d52472bd2a6626ddb',1,'mlx::core::simd::N']]], ['n_1',['n',['../struct_read_writer.html#a655346c9ebfc33a69da3f1c1d4238dfb',1,'ReadWriter']]], - ['n_5fchannels_2',['n_channels',['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper']]], - ['n_5fper_5fblock_3',['N_PER_BLOCK',['../struct_kernel_merge_sort.html#a959aaf5bfb70796a525fed318f7ae8ab',1,'KernelMergeSort::N_PER_BLOCK'],['../struct_kernel_multi_block_merge_sort.html#ae5113ca5852d11999ae932439af95a5c',1,'KernelMultiBlockMergeSort::N_PER_BLOCK']]], + ['n_5fchannels_2',['n_channels',['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#a06c2fb9c93660e8f6916228cd77f9494',1,'mlx::steel::ChannelHelper< 1 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#ac66ff37bc2cf78d96667192a6cca73b5',1,'mlx::steel::ChannelHelper< 2 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#a071c015713b7bab09930661165517eff',1,'mlx::steel::ChannelHelper< 3 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#a167b00a84adf93b60e3d7a943d5eb977',1,'mlx::steel::ChannelHelper< 4 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper< 1 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper< 2 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper< 3 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper< 4 >::n_channels']]], + ['n_5fper_5fblock_3',['N_PER_BLOCK',['../struct_kernel_merge_sort.html#a959aaf5bfb70796a525fed318f7ae8ab',1,'KernelMergeSort::N_PER_BLOCK'],['../struct_kernel_multi_block_merge_sort.html#a4a0c920980c5e56471558d22a727a514',1,'KernelMultiBlockMergeSort::N_PER_BLOCK']]], ['n_5freads_4',['n_reads',['../struct_quantized_block_loader.html#a6213479f7a6d9314d8879f8856b0b6fb',1,'QuantizedBlockLoader']]], ['n_5frows_5',['n_rows',['../structmlx_1_1steel_1_1_block_loader.html#a973804e5b1d418c98c90861cda1a6fb5',1,'mlx::steel::BlockLoader::n_rows'],['../structmlx_1_1steel_1_1_block_loader_t.html#a0ccc7caa93e6e709981a1a08159d41dc',1,'mlx::steel::BlockLoaderT::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a097c48a23e1bd7d8cf3e9d531397602f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a3ec8a92c9e6643c1d5bf8af278026fe8',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a593ec140370d53f8c968f6240116d38b',1,'mlx::steel::Conv2DWeightBlockLoader::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a8b6c0936c9ad2766242664f034d1115f',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae905e56c1129606e93dbbcd7baed8f0f',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#abff29c5d96645d9113314c9a997dd7a8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aaebb6da2cac9961f5edf52d16c18de7d',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::n_rows']]], ['names_6',['names',['../structmlx_1_1core_1_1_node_namer.html#a57823f9a2cdc60b2f06f857b36019277',1,'mlx::core::NodeNamer']]], diff --git a/docs/build/html/search/variables_e.js b/docs/build/html/search/variables_e.js index 35c03ce09..8ce1661a2 100644 --- a/docs/build/html/search/variables_e.js +++ b/docs/build/html/search/variables_e.js @@ -2,8 +2,8 @@ var searchData= [ ['o_0',['O',['../struct_m_l_x_conv_params.html#ad55ff586d30072d8154865f9dfe92d97',1,'MLXConvParams']]], ['o_5fstrides_1',['O_strides',['../structmlx_1_1steel_1_1_attn_params.html#ab210f29dcc3a732aba34894cd5a42cf7',1,'mlx::steel::AttnParams']]], - ['offset_2',['offset',['../struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791',1,'LoopedElemToLoc']]], - ['op_3',['op',['../structmlx_1_1core_1_1_default_strided_reduce.html#ac871f55a7ddd205574974cb4492a240b',1,'mlx::core::DefaultStridedReduce::op'],['../structmlx_1_1core_1_1_default_contiguous_reduce.html#a1928f07db988715cc177999e386f4830',1,'mlx::core::DefaultContiguousReduce::op']]], + ['offset_2',['offset',['../struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791',1,'LoopedElemToLoc::offset'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a3a18944c158e2747a6ddebb420299a3b',1,'LoopedElemToLoc< 1, OffsetT, true >::offset'],['../struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#af792b1fd4e8286f97b9b863c127a2d9a',1,'LoopedElemToLoc< 1, OffsetT, false >::offset'],['../struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791',1,'LoopedElemToLoc< 1, OffsetT, false >::offset'],['../struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791',1,'LoopedElemToLoc< 1, OffsetT, true >::offset']]], + ['op_3',['op',['../structmlx_1_1core_1_1_vector_scalar.html#a5fe1744adb58aaa845acca1e46725537',1,'mlx::core::VectorScalar::op'],['../structmlx_1_1core_1_1_scalar_vector.html#ac9c2214744bc972150740e169b603b9b',1,'mlx::core::ScalarVector::op'],['../structmlx_1_1core_1_1_vector_vector.html#a6d69d070c75cf0281e11e36e0717ab50',1,'mlx::core::VectorVector::op']]], ['ortho_4',['ortho',['../structpocketfft_1_1detail_1_1_exec_dcst.html#aea17551a49acaca5e7808dc181d38b7f',1,'pocketfft::detail::ExecDcst']]], ['os_5',['oS',['../struct_m_l_x_conv_params.html#a19ccb9fecfccdc18b6a7f0cc43adbc6e',1,'MLXConvParams']]], ['out_6',['out',['../struct_read_writer.html#abea3b913c952c505d0ca4e529c7316ef',1,'ReadWriter']]], diff --git a/docs/build/html/searchindex.js b/docs/build/html/searchindex.js index 4c92bec32..9520257c7 100644 --- a/docs/build/html/searchindex.js +++ b/docs/build/html/searchindex.js @@ -1 +1 @@ -Search.setIndex({"alltitles": {"A Simple Example": [[500, "a-simple-example"]], "Array": [[325, null]], "Attention layer": [[6, "attention-layer"]], "Automatic Differentiation": [[494, "automatic-differentiation"]], "Automatic Vectorization": [[494, "automatic-vectorization"]], "Basics": [[498, "basics"]], "Basics of Compile": [[491, "basics-of-compile"]], "Basics of Exporting": [[493, "basics-of-exporting"]], "Binary Size Minimization": [[9, "binary-size-minimization"]], "Binding to Python": [[2, "binding-to-python"]], "Build Options": [[9, "id4"]], "Build Requirements": [[9, "build-requirements"]], "Build and Install": [[9, null]], "Build from source": [[9, "build-from-source"]], "Building and Binding": [[2, "building-and-binding"]], "Building with CMake": [[2, "building-with-cmake"]], "Building with setuptools": [[2, "building-with-setuptools"]], "C++ API": [[9, "c-api"]], "C++ API Reference": [[8, null]], "Common Optimizers": [[485, null]], "Compilation": [[491, null]], "Compiling Training Graphs": [[491, "compiling-training-graphs"]], "Complex Example": [[1, "complex-example"]], "Conversion to NumPy and Other Frameworks": [[497, null]], "Converting the weights": [[6, "converting-the-weights"]], "Custom Extensions in MLX": [[2, null]], "Custom Metal Kernels": [[1, null]], "Data Types": [[326, null]], "Debugging": [[491, "debugging"]], "Devices and Streams": [[327, null]], "Differences from NumPy": [[495, "differences-from-numpy"]], "Distributed Communication": [[328, null], [492, null]], "Download the code": [[2, null], [6, null]], "Encoder layer": [[6, "encoder-layer"]], "Example Speedup": [[491, "example-speedup"]], "Examples": [[8, null]], "Export Functions": [[329, null]], "Exporting Functions": [[493, null]], "Exporting Modules": [[493, "exporting-modules"]], "Exporting Multiple Traces": [[493, "exporting-multiple-traces"]], "FFT": [[331, null]], "Fast": [[330, null]], "Full model": [[6, "full-model"]], "Function Transforms": [[494, null]], "Function and Graph Transformations": [[498, "function-and-graph-transformations"]], "Functions": [[460, null]], "Further Reading": [[8, null]], "Generation": [[6, "generation"]], "Getting Started": [[492, "getting-started"]], "Grid Sample VJP": [[1, "grid-sample-vjp"]], "Implementing the CPU Back-end": [[2, "implementing-the-cpu-back-end"]], "Implementing the GPU Back-end": [[2, "implementing-the-gpu-back-end"]], "Implementing the Primitive": [[2, "implementing-the-primitive"]], "Implementing the model": [[6, "implementing-the-model"]], "Importing Functions in C++": [[493, "importing-functions-in-c"]], "In Place Updates": [[495, "in-place-updates"]], "Indexing Arrays": [[495, null]], "Initializers": [[461, null]], "Inspecting Modules": [[334, "inspecting-modules"]], "Install": [[8, null]], "Installing MPI": [[492, "installing-mpi"]], "Introducing the Example": [[2, "introducing-the-example"]], "JAX": [[497, "jax"]], "LLM inference": [[6, null]], "Layers": [[462, null]], "Lazy Evaluation": [[496, null]], "Linear Algebra": [[332, null]], "Linear Regression": [[5, null]], "Loss Functions": [[463, null]], "MLX": [[8, null]], "Metal": [[333, null]], "Metal Debugger": [[3, null]], "Metal not found": [[9, "metal-not-found"]], "Module": [[464, null]], "More Examples": [[493, "more-examples"]], "Multi-Layer Perceptron": [[7, null]], "Neural Networks": [[334, null]], "Only Compute What You Use": [[496, "only-compute-what-you-use"]], "Operations": [[0, null], [2, "operations"], [465, null]], "Operations and Primitives": [[2, "operations-and-primitives"]], "Optimizer": [[486, null]], "Optimizers": [[466, null]], "Package Variables": [[4, "id1"]], "Parameters": [[334, "parameters"]], "Primitive Transforms": [[2, "primitive-transforms"]], "Primitives": [[2, "primitives"]], "Pure Functions": [[491, "pure-functions"]], "Putting it all together": [[6, "putting-it-all-together"]], "PyTorch": [[497, "pytorch"]], "Python API": [[9, "python-api"]], "Python API Reference": [[8, null]], "Python Installation": [[9, "python-installation"]], "Quick Start Guide": [[498, null]], "Quick Start with Neural Networks": [[334, "quick-start-with-neural-networks"]], "Random": [[488, null]], "Results": [[2, "results"]], "Saving and Loading": [[466, "saving-and-loading"]], "Saving and Loading Arrays": [[499, null]], "Schedulers": [[487, null]], "Scripts": [[2, "scripts"], [6, "scripts"]], "Serialization Formats": [[499, "id1"]], "Setting up Remote Hosts": [[492, "setting-up-remote-hosts"]], "Shapeless Compilation": [[491, "shapeless-compilation"]], "Shapeless Exports": [[493, "shapeless-exports"]], "Simple Example": [[1, "simple-example"]], "Specifying the Stream": [[501, "specifying-the-stream"]], "Supported Data Types": [[326, "id2"]], "TensorFlow": [[497, "tensorflow"]], "The Module Class": [[334, "the-module-class"]], "Training Example": [[492, "training-example"]], "Transformations with Compile": [[491, "transformations-with-compile"]], "Transformations with Imported Functions": [[493, "transformations-with-imported-functions"]], "Transforming Compute Graphs": [[496, "transforming-compute-graphs"]], "Transforms": [[489, null]], "Tree Utils": [[490, null]], "Troubleshooting": [[9, "troubleshooting"], [9, "id3"]], "Tuning All Reduce": [[492, "tuning-all-reduce"]], "Unified Memory": [[500, null]], "Updating the Parameters": [[334, "updating-the-parameters"]], "Usage": [[2, "usage"], [8, null]], "Using MLX in C++": [[4, null]], "Using Shape/Strides": [[1, "using-shape-strides"]], "Using Streams": [[501, null]], "Using the Primitive": [[2, "using-the-primitive"]], "Value and Grad": [[334, "value-and-grad"]], "Weight loading and benchmarking": [[6, "weight-loading-and-benchmarking"]], "When to Evaluate": [[496, "when-to-evaluate"]], "Why Lazy Evaluation": [[496, "why-lazy-evaluation"]], "Xcode Workflow": [[3, "xcode-workflow"]], "mlx.core.Device": [[10, null]], "mlx.core.Dtype": [[11, null]], "mlx.core.DtypeCategory": [[12, null]], "mlx.core.Stream": [[324, null]], "mlx.core.abs": [[13, null]], "mlx.core.add": [[14, null]], "mlx.core.addmm": [[15, null]], "mlx.core.all": [[16, null]], "mlx.core.allclose": [[17, null]], "mlx.core.any": [[18, null]], "mlx.core.arange": [[19, null]], "mlx.core.arccos": [[20, null]], "mlx.core.arccosh": [[21, null]], "mlx.core.arcsin": [[22, null]], "mlx.core.arcsinh": [[23, null]], "mlx.core.arctan": [[24, null]], "mlx.core.arctan2": [[25, null]], "mlx.core.arctanh": [[26, null]], "mlx.core.argmax": [[27, null]], "mlx.core.argmin": [[28, null]], "mlx.core.argpartition": [[29, null]], "mlx.core.argsort": [[30, null]], "mlx.core.array": [[31, null]], "mlx.core.array.T": [[32, null]], "mlx.core.array.abs": [[33, null]], "mlx.core.array.all": [[34, null]], "mlx.core.array.any": [[35, null]], "mlx.core.array.argmax": [[36, null]], "mlx.core.array.argmin": [[37, null]], "mlx.core.array.astype": [[38, null]], "mlx.core.array.at": [[39, null]], "mlx.core.array.conj": [[40, null]], "mlx.core.array.cos": [[41, null]], "mlx.core.array.cummax": [[42, null]], "mlx.core.array.cummin": [[43, null]], "mlx.core.array.cumprod": [[44, null]], "mlx.core.array.cumsum": [[45, null]], "mlx.core.array.diag": [[46, null]], "mlx.core.array.diagonal": [[47, null]], "mlx.core.array.dtype": [[48, null]], "mlx.core.array.exp": [[49, null]], "mlx.core.array.flatten": [[50, null]], "mlx.core.array.item": [[51, null]], "mlx.core.array.itemsize": [[52, null]], "mlx.core.array.log": [[53, null]], "mlx.core.array.log10": [[54, null]], "mlx.core.array.log1p": [[55, null]], "mlx.core.array.log2": [[56, null]], "mlx.core.array.logsumexp": [[57, null]], "mlx.core.array.max": [[58, null]], "mlx.core.array.mean": [[59, null]], "mlx.core.array.min": [[60, null]], "mlx.core.array.moveaxis": [[61, null]], "mlx.core.array.nbytes": [[62, null]], "mlx.core.array.ndim": [[63, null]], "mlx.core.array.prod": [[64, null]], "mlx.core.array.reciprocal": [[65, null]], "mlx.core.array.reshape": [[66, null]], "mlx.core.array.round": [[67, null]], "mlx.core.array.rsqrt": [[68, null]], "mlx.core.array.shape": [[69, null]], "mlx.core.array.sin": [[70, null]], "mlx.core.array.size": [[71, null]], "mlx.core.array.split": [[72, null]], "mlx.core.array.sqrt": [[73, null]], "mlx.core.array.square": [[74, null]], "mlx.core.array.squeeze": [[75, null]], "mlx.core.array.std": [[76, null]], "mlx.core.array.sum": [[77, null]], "mlx.core.array.swapaxes": [[78, null]], "mlx.core.array.tolist": [[79, null]], "mlx.core.array.transpose": [[80, null]], "mlx.core.array.var": [[81, null]], "mlx.core.array.view": [[82, null]], "mlx.core.array_equal": [[83, null]], "mlx.core.as_strided": [[84, null]], "mlx.core.atleast_1d": [[85, null]], "mlx.core.atleast_2d": [[86, null]], "mlx.core.atleast_3d": [[87, null]], "mlx.core.bitwise_and": [[88, null]], "mlx.core.bitwise_or": [[89, null]], "mlx.core.bitwise_xor": [[90, null]], "mlx.core.block_masked_mm": [[91, null]], "mlx.core.broadcast_to": [[92, null]], "mlx.core.ceil": [[93, null]], "mlx.core.clip": [[94, null]], "mlx.core.compile": [[95, null]], "mlx.core.concatenate": [[96, null]], "mlx.core.conj": [[97, null]], "mlx.core.conjugate": [[98, null]], "mlx.core.conv1d": [[99, null]], "mlx.core.conv2d": [[100, null]], "mlx.core.conv3d": [[101, null]], "mlx.core.conv_general": [[102, null]], "mlx.core.conv_transpose1d": [[103, null]], "mlx.core.conv_transpose2d": [[104, null]], "mlx.core.conv_transpose3d": [[105, null]], "mlx.core.convolve": [[106, null]], "mlx.core.cos": [[107, null]], "mlx.core.cosh": [[108, null]], "mlx.core.cummax": [[109, null]], "mlx.core.cummin": [[110, null]], "mlx.core.cumprod": [[111, null]], "mlx.core.cumsum": [[112, null]], "mlx.core.custom_function": [[113, null]], "mlx.core.default_device": [[114, null]], "mlx.core.default_stream": [[115, null]], "mlx.core.degrees": [[116, null]], "mlx.core.dequantize": [[117, null]], "mlx.core.diag": [[118, null]], "mlx.core.diagonal": [[119, null]], "mlx.core.disable_compile": [[120, null]], "mlx.core.distributed.Group": [[121, null]], "mlx.core.distributed.all_gather": [[122, null]], "mlx.core.distributed.all_sum": [[123, null]], "mlx.core.distributed.init": [[124, null]], "mlx.core.distributed.is_available": [[125, null]], "mlx.core.distributed.recv": [[126, null]], "mlx.core.distributed.recv_like": [[127, null]], "mlx.core.distributed.send": [[128, null]], "mlx.core.divide": [[129, null]], "mlx.core.divmod": [[130, null]], "mlx.core.einsum": [[131, null]], "mlx.core.einsum_path": [[132, null]], "mlx.core.enable_compile": [[133, null]], "mlx.core.equal": [[134, null]], "mlx.core.erf": [[135, null]], "mlx.core.erfinv": [[136, null]], "mlx.core.eval": [[137, null]], "mlx.core.exp": [[138, null]], "mlx.core.expand_dims": [[139, null]], "mlx.core.expm1": [[140, null]], "mlx.core.export_function": [[141, null]], "mlx.core.export_to_dot": [[142, null]], "mlx.core.exporter": [[143, null]], "mlx.core.eye": [[144, null]], "mlx.core.fast.layer_norm": [[145, null]], "mlx.core.fast.metal_kernel": [[146, null]], "mlx.core.fast.rms_norm": [[147, null]], "mlx.core.fast.rope": [[148, null]], "mlx.core.fast.scaled_dot_product_attention": [[149, null]], "mlx.core.fft.fft": [[150, null]], "mlx.core.fft.fft2": [[151, null]], "mlx.core.fft.fftn": [[152, null]], "mlx.core.fft.ifft": [[153, null]], "mlx.core.fft.ifft2": [[154, null]], "mlx.core.fft.ifftn": [[155, null]], "mlx.core.fft.irfft": [[156, null]], "mlx.core.fft.irfft2": [[157, null]], "mlx.core.fft.irfftn": [[158, null]], "mlx.core.fft.rfft": [[159, null]], "mlx.core.fft.rfft2": [[160, null]], "mlx.core.fft.rfftn": [[161, null]], "mlx.core.finfo": [[162, null]], "mlx.core.flatten": [[163, null]], "mlx.core.floor": [[164, null]], "mlx.core.floor_divide": [[165, null]], "mlx.core.full": [[166, null]], "mlx.core.gather_mm": [[167, null]], "mlx.core.gather_qmm": [[168, null]], "mlx.core.grad": [[169, null]], "mlx.core.greater": [[170, null]], "mlx.core.greater_equal": [[171, null]], "mlx.core.hadamard_transform": [[172, null]], "mlx.core.identity": [[173, null]], "mlx.core.imag": [[174, null]], "mlx.core.import_function": [[175, null]], "mlx.core.inner": [[176, null]], "mlx.core.isclose": [[177, null]], "mlx.core.isfinite": [[178, null]], "mlx.core.isinf": [[179, null]], "mlx.core.isnan": [[180, null]], "mlx.core.isneginf": [[181, null]], "mlx.core.isposinf": [[182, null]], "mlx.core.issubdtype": [[183, null]], "mlx.core.jvp": [[184, null]], "mlx.core.kron": [[185, null]], "mlx.core.left_shift": [[186, null]], "mlx.core.less": [[187, null]], "mlx.core.less_equal": [[188, null]], "mlx.core.linalg.cholesky": [[189, null]], "mlx.core.linalg.cholesky_inv": [[190, null]], "mlx.core.linalg.cross": [[191, null]], "mlx.core.linalg.eigh": [[192, null]], "mlx.core.linalg.eigvalsh": [[193, null]], "mlx.core.linalg.inv": [[194, null]], "mlx.core.linalg.norm": [[195, null]], "mlx.core.linalg.qr": [[196, null]], "mlx.core.linalg.svd": [[197, null]], "mlx.core.linalg.tri_inv": [[198, null]], "mlx.core.linspace": [[199, null]], "mlx.core.load": [[200, null]], "mlx.core.log": [[201, null]], "mlx.core.log10": [[202, null]], "mlx.core.log1p": [[203, null]], "mlx.core.log2": [[204, null]], "mlx.core.logaddexp": [[205, null]], "mlx.core.logical_and": [[206, null]], "mlx.core.logical_not": [[207, null]], "mlx.core.logical_or": [[208, null]], "mlx.core.logsumexp": [[209, null]], "mlx.core.matmul": [[210, null]], "mlx.core.max": [[211, null]], "mlx.core.maximum": [[212, null]], "mlx.core.mean": [[213, null]], "mlx.core.meshgrid": [[214, null]], "mlx.core.metal.clear_cache": [[215, null]], "mlx.core.metal.device_info": [[216, null]], "mlx.core.metal.get_active_memory": [[217, null]], "mlx.core.metal.get_cache_memory": [[218, null]], "mlx.core.metal.get_peak_memory": [[219, null]], "mlx.core.metal.is_available": [[220, null]], "mlx.core.metal.reset_peak_memory": [[221, null]], "mlx.core.metal.set_cache_limit": [[222, null]], "mlx.core.metal.set_memory_limit": [[223, null]], "mlx.core.metal.set_wired_limit": [[224, null]], "mlx.core.metal.start_capture": [[225, null]], "mlx.core.metal.stop_capture": [[226, null]], "mlx.core.min": [[227, null]], "mlx.core.minimum": [[228, null]], "mlx.core.moveaxis": [[229, null]], "mlx.core.multiply": [[230, null]], "mlx.core.nan_to_num": [[231, null]], "mlx.core.negative": [[232, null]], "mlx.core.new_stream": [[233, null]], "mlx.core.not_equal": [[234, null]], "mlx.core.ones": [[235, null]], "mlx.core.ones_like": [[236, null]], "mlx.core.outer": [[237, null]], "mlx.core.pad": [[238, null]], "mlx.core.partition": [[239, null]], "mlx.core.power": [[240, null]], "mlx.core.prod": [[241, null]], "mlx.core.put_along_axis": [[242, null]], "mlx.core.quantize": [[243, null]], "mlx.core.quantized_matmul": [[244, null]], "mlx.core.radians": [[245, null]], "mlx.core.random.bernoulli": [[246, null]], "mlx.core.random.categorical": [[247, null]], "mlx.core.random.gumbel": [[248, null]], "mlx.core.random.key": [[249, null]], "mlx.core.random.laplace": [[250, null]], "mlx.core.random.multivariate_normal": [[251, null]], "mlx.core.random.normal": [[252, null]], "mlx.core.random.permutation": [[253, null]], "mlx.core.random.randint": [[254, null]], "mlx.core.random.seed": [[255, null]], "mlx.core.random.split": [[256, null]], "mlx.core.random.truncated_normal": [[257, null]], "mlx.core.random.uniform": [[258, null]], "mlx.core.real": [[259, null]], "mlx.core.reciprocal": [[260, null]], "mlx.core.remainder": [[261, null]], "mlx.core.repeat": [[262, null]], "mlx.core.reshape": [[263, null]], "mlx.core.right_shift": [[264, null]], "mlx.core.roll": [[265, null]], "mlx.core.round": [[266, null]], "mlx.core.rsqrt": [[267, null]], "mlx.core.save": [[268, null]], "mlx.core.save_gguf": [[269, null]], "mlx.core.save_safetensors": [[270, null]], "mlx.core.savez": [[271, null]], "mlx.core.savez_compressed": [[272, null]], "mlx.core.set_default_device": [[273, null]], "mlx.core.set_default_stream": [[274, null]], "mlx.core.sigmoid": [[275, null]], "mlx.core.sign": [[276, null]], "mlx.core.sin": [[277, null]], "mlx.core.sinh": [[278, null]], "mlx.core.slice": [[279, null]], "mlx.core.slice_update": [[280, null]], "mlx.core.softmax": [[281, null]], "mlx.core.sort": [[282, null]], "mlx.core.split": [[283, null]], "mlx.core.sqrt": [[284, null]], "mlx.core.square": [[285, null]], "mlx.core.squeeze": [[286, null]], "mlx.core.stack": [[287, null]], "mlx.core.std": [[288, null]], "mlx.core.stop_gradient": [[289, null]], "mlx.core.stream": [[290, null]], "mlx.core.subtract": [[291, null]], "mlx.core.sum": [[292, null]], "mlx.core.swapaxes": [[293, null]], "mlx.core.synchronize": [[294, null]], "mlx.core.take": [[295, null]], "mlx.core.take_along_axis": [[296, null]], "mlx.core.tan": [[297, null]], "mlx.core.tanh": [[298, null]], "mlx.core.tensordot": [[299, null]], "mlx.core.tile": [[300, null]], "mlx.core.topk": [[301, null]], "mlx.core.trace": [[302, null]], "mlx.core.transpose": [[303, null]], "mlx.core.tri": [[304, null]], "mlx.core.tril": [[305, null]], "mlx.core.triu": [[306, null]], "mlx.core.unflatten": [[307, null]], "mlx.core.value_and_grad": [[308, null]], "mlx.core.var": [[309, null]], "mlx.core.view": [[310, null]], "mlx.core.vjp": [[311, null]], "mlx.core.vmap": [[312, null]], "mlx.core.where": [[313, null]], "mlx.core.zeros": [[314, null]], "mlx.core.zeros_like": [[315, null]], "mlx.nn.ALiBi": [[335, null]], "mlx.nn.AvgPool1d": [[336, null]], "mlx.nn.AvgPool2d": [[337, null]], "mlx.nn.AvgPool3d": [[338, null]], "mlx.nn.BatchNorm": [[339, null]], "mlx.nn.CELU": [[340, null]], "mlx.nn.Conv1d": [[341, null]], "mlx.nn.Conv2d": [[342, null]], "mlx.nn.Conv3d": [[343, null]], "mlx.nn.ConvTranspose1d": [[344, null]], "mlx.nn.ConvTranspose2d": [[345, null]], "mlx.nn.ConvTranspose3d": [[346, null]], "mlx.nn.Dropout": [[347, null]], "mlx.nn.Dropout2d": [[348, null]], "mlx.nn.Dropout3d": [[349, null]], "mlx.nn.ELU": [[350, null]], "mlx.nn.Embedding": [[351, null]], "mlx.nn.GELU": [[352, null]], "mlx.nn.GLU": [[353, null]], "mlx.nn.GRU": [[354, null]], "mlx.nn.GroupNorm": [[355, null]], "mlx.nn.HardShrink": [[356, null]], "mlx.nn.HardTanh": [[357, null]], "mlx.nn.Hardswish": [[358, null]], "mlx.nn.InstanceNorm": [[359, null]], "mlx.nn.LSTM": [[360, null]], "mlx.nn.LayerNorm": [[361, null]], "mlx.nn.LeakyReLU": [[362, null]], "mlx.nn.Linear": [[363, null]], "mlx.nn.LogSigmoid": [[364, null]], "mlx.nn.LogSoftmax": [[365, null]], "mlx.nn.MaxPool1d": [[366, null]], "mlx.nn.MaxPool2d": [[367, null]], "mlx.nn.MaxPool3d": [[368, null]], "mlx.nn.Mish": [[369, null]], "mlx.nn.Module.apply": [[370, null]], "mlx.nn.Module.apply_to_modules": [[371, null]], "mlx.nn.Module.children": [[372, null]], "mlx.nn.Module.eval": [[373, null]], "mlx.nn.Module.filter_and_map": [[374, null]], "mlx.nn.Module.freeze": [[375, null]], "mlx.nn.Module.leaf_modules": [[376, null]], "mlx.nn.Module.load_weights": [[377, null]], "mlx.nn.Module.modules": [[378, null]], "mlx.nn.Module.named_modules": [[379, null]], "mlx.nn.Module.parameters": [[380, null]], "mlx.nn.Module.save_weights": [[381, null]], "mlx.nn.Module.set_dtype": [[382, null]], "mlx.nn.Module.state": [[383, null]], "mlx.nn.Module.train": [[384, null]], "mlx.nn.Module.trainable_parameters": [[385, null]], "mlx.nn.Module.training": [[386, null]], "mlx.nn.Module.unfreeze": [[387, null]], "mlx.nn.Module.update": [[388, null]], "mlx.nn.Module.update_modules": [[389, null]], "mlx.nn.MultiHeadAttention": [[390, null]], "mlx.nn.PReLU": [[391, null]], "mlx.nn.QuantizedEmbedding": [[392, null]], "mlx.nn.QuantizedLinear": [[393, null]], "mlx.nn.RMSNorm": [[394, null]], "mlx.nn.RNN": [[395, null]], "mlx.nn.ReLU": [[396, null]], "mlx.nn.ReLU6": [[397, null]], "mlx.nn.RoPE": [[398, null]], "mlx.nn.SELU": [[399, null]], "mlx.nn.Sequential": [[400, null]], "mlx.nn.SiLU": [[401, null]], "mlx.nn.Sigmoid": [[402, null]], "mlx.nn.SinusoidalPositionalEncoding": [[403, null]], "mlx.nn.Softmax": [[404, null]], "mlx.nn.Softmin": [[405, null]], "mlx.nn.Softplus": [[406, null]], "mlx.nn.Softshrink": [[407, null]], "mlx.nn.Softsign": [[408, null]], "mlx.nn.Step": [[409, null]], "mlx.nn.Tanh": [[410, null]], "mlx.nn.Transformer": [[411, null]], "mlx.nn.Upsample": [[412, null]], "mlx.nn.celu": [[421, null]], "mlx.nn.elu": [[422, null]], "mlx.nn.gelu": [[423, null]], "mlx.nn.gelu_approx": [[424, null]], "mlx.nn.gelu_fast_approx": [[425, null]], "mlx.nn.glu": [[426, null]], "mlx.nn.hard_shrink": [[427, null]], "mlx.nn.hard_tanh": [[428, null]], "mlx.nn.hardswish": [[429, null]], "mlx.nn.init.constant": [[413, null]], "mlx.nn.init.glorot_normal": [[414, null]], "mlx.nn.init.glorot_uniform": [[415, null]], "mlx.nn.init.he_normal": [[416, null]], "mlx.nn.init.he_uniform": [[417, null]], "mlx.nn.init.identity": [[418, null]], "mlx.nn.init.normal": [[419, null]], "mlx.nn.init.uniform": [[420, null]], "mlx.nn.leaky_relu": [[430, null]], "mlx.nn.log_sigmoid": [[431, null]], "mlx.nn.log_softmax": [[432, null]], "mlx.nn.losses.binary_cross_entropy": [[433, null]], "mlx.nn.losses.cosine_similarity_loss": [[434, null]], "mlx.nn.losses.cross_entropy": [[435, null]], "mlx.nn.losses.gaussian_nll_loss": [[436, null]], "mlx.nn.losses.hinge_loss": [[437, null]], "mlx.nn.losses.huber_loss": [[438, null]], "mlx.nn.losses.kl_div_loss": [[439, null]], "mlx.nn.losses.l1_loss": [[440, null]], "mlx.nn.losses.log_cosh_loss": [[441, null]], "mlx.nn.losses.margin_ranking_loss": [[442, null]], "mlx.nn.losses.mse_loss": [[443, null]], "mlx.nn.losses.nll_loss": [[444, null]], "mlx.nn.losses.smooth_l1_loss": [[445, null]], "mlx.nn.losses.triplet_loss": [[446, null]], "mlx.nn.mish": [[447, null]], "mlx.nn.prelu": [[448, null]], "mlx.nn.quantize": [[316, null]], "mlx.nn.relu": [[449, null]], "mlx.nn.relu6": [[450, null]], "mlx.nn.selu": [[451, null]], "mlx.nn.sigmoid": [[452, null]], "mlx.nn.silu": [[453, null]], "mlx.nn.softmax": [[454, null]], "mlx.nn.softmin": [[455, null]], "mlx.nn.softplus": [[456, null]], "mlx.nn.softshrink": [[457, null]], "mlx.nn.step": [[458, null]], "mlx.nn.tanh": [[459, null]], "mlx.nn.value_and_grad": [[317, null]], "mlx.optimizers.AdaDelta": [[467, null]], "mlx.optimizers.Adafactor": [[468, null]], "mlx.optimizers.Adagrad": [[469, null]], "mlx.optimizers.Adam": [[470, null]], "mlx.optimizers.AdamW": [[471, null]], "mlx.optimizers.Adamax": [[472, null]], "mlx.optimizers.Lion": [[473, null]], "mlx.optimizers.Optimizer.apply_gradients": [[474, null]], "mlx.optimizers.Optimizer.init": [[475, null]], "mlx.optimizers.Optimizer.state": [[476, null]], "mlx.optimizers.Optimizer.update": [[477, null]], "mlx.optimizers.RMSprop": [[478, null]], "mlx.optimizers.SGD": [[479, null]], "mlx.optimizers.clip_grad_norm": [[318, null]], "mlx.optimizers.cosine_decay": [[480, null]], "mlx.optimizers.exponential_decay": [[481, null]], "mlx.optimizers.join_schedules": [[482, null]], "mlx.optimizers.linear_schedule": [[483, null]], "mlx.optimizers.step_decay": [[484, null]], "mlx.utils.tree_flatten": [[319, null]], "mlx.utils.tree_map": [[320, null]], "mlx.utils.tree_map_with_path": [[321, null]], "mlx.utils.tree_reduce": [[322, null]], "mlx.utils.tree_unflatten": [[323, null]], "x86 Shell": [[9, "x86-shell"]]}, "docnames": ["cpp/ops", "dev/custom_metal_kernels", "dev/extensions", "dev/metal_debugger", "dev/mlx_in_cpp", "examples/linear_regression", "examples/llama-inference", "examples/mlp", "index", "install", "python/_autosummary/mlx.core.Device", "python/_autosummary/mlx.core.Dtype", "python/_autosummary/mlx.core.DtypeCategory", "python/_autosummary/mlx.core.abs", "python/_autosummary/mlx.core.add", "python/_autosummary/mlx.core.addmm", "python/_autosummary/mlx.core.all", "python/_autosummary/mlx.core.allclose", "python/_autosummary/mlx.core.any", "python/_autosummary/mlx.core.arange", "python/_autosummary/mlx.core.arccos", "python/_autosummary/mlx.core.arccosh", "python/_autosummary/mlx.core.arcsin", "python/_autosummary/mlx.core.arcsinh", "python/_autosummary/mlx.core.arctan", "python/_autosummary/mlx.core.arctan2", "python/_autosummary/mlx.core.arctanh", "python/_autosummary/mlx.core.argmax", "python/_autosummary/mlx.core.argmin", "python/_autosummary/mlx.core.argpartition", "python/_autosummary/mlx.core.argsort", "python/_autosummary/mlx.core.array", "python/_autosummary/mlx.core.array.T", "python/_autosummary/mlx.core.array.abs", "python/_autosummary/mlx.core.array.all", "python/_autosummary/mlx.core.array.any", "python/_autosummary/mlx.core.array.argmax", "python/_autosummary/mlx.core.array.argmin", "python/_autosummary/mlx.core.array.astype", "python/_autosummary/mlx.core.array.at", "python/_autosummary/mlx.core.array.conj", "python/_autosummary/mlx.core.array.cos", "python/_autosummary/mlx.core.array.cummax", "python/_autosummary/mlx.core.array.cummin", "python/_autosummary/mlx.core.array.cumprod", "python/_autosummary/mlx.core.array.cumsum", "python/_autosummary/mlx.core.array.diag", "python/_autosummary/mlx.core.array.diagonal", "python/_autosummary/mlx.core.array.dtype", "python/_autosummary/mlx.core.array.exp", "python/_autosummary/mlx.core.array.flatten", "python/_autosummary/mlx.core.array.item", "python/_autosummary/mlx.core.array.itemsize", "python/_autosummary/mlx.core.array.log", "python/_autosummary/mlx.core.array.log10", "python/_autosummary/mlx.core.array.log1p", "python/_autosummary/mlx.core.array.log2", "python/_autosummary/mlx.core.array.logsumexp", "python/_autosummary/mlx.core.array.max", "python/_autosummary/mlx.core.array.mean", "python/_autosummary/mlx.core.array.min", "python/_autosummary/mlx.core.array.moveaxis", "python/_autosummary/mlx.core.array.nbytes", "python/_autosummary/mlx.core.array.ndim", "python/_autosummary/mlx.core.array.prod", "python/_autosummary/mlx.core.array.reciprocal", "python/_autosummary/mlx.core.array.reshape", "python/_autosummary/mlx.core.array.round", "python/_autosummary/mlx.core.array.rsqrt", "python/_autosummary/mlx.core.array.shape", "python/_autosummary/mlx.core.array.sin", "python/_autosummary/mlx.core.array.size", "python/_autosummary/mlx.core.array.split", "python/_autosummary/mlx.core.array.sqrt", "python/_autosummary/mlx.core.array.square", "python/_autosummary/mlx.core.array.squeeze", "python/_autosummary/mlx.core.array.std", "python/_autosummary/mlx.core.array.sum", "python/_autosummary/mlx.core.array.swapaxes", "python/_autosummary/mlx.core.array.tolist", "python/_autosummary/mlx.core.array.transpose", "python/_autosummary/mlx.core.array.var", "python/_autosummary/mlx.core.array.view", "python/_autosummary/mlx.core.array_equal", "python/_autosummary/mlx.core.as_strided", "python/_autosummary/mlx.core.atleast_1d", "python/_autosummary/mlx.core.atleast_2d", "python/_autosummary/mlx.core.atleast_3d", "python/_autosummary/mlx.core.bitwise_and", "python/_autosummary/mlx.core.bitwise_or", "python/_autosummary/mlx.core.bitwise_xor", "python/_autosummary/mlx.core.block_masked_mm", "python/_autosummary/mlx.core.broadcast_to", "python/_autosummary/mlx.core.ceil", "python/_autosummary/mlx.core.clip", "python/_autosummary/mlx.core.compile", "python/_autosummary/mlx.core.concatenate", "python/_autosummary/mlx.core.conj", "python/_autosummary/mlx.core.conjugate", "python/_autosummary/mlx.core.conv1d", "python/_autosummary/mlx.core.conv2d", "python/_autosummary/mlx.core.conv3d", "python/_autosummary/mlx.core.conv_general", "python/_autosummary/mlx.core.conv_transpose1d", "python/_autosummary/mlx.core.conv_transpose2d", "python/_autosummary/mlx.core.conv_transpose3d", "python/_autosummary/mlx.core.convolve", "python/_autosummary/mlx.core.cos", "python/_autosummary/mlx.core.cosh", "python/_autosummary/mlx.core.cummax", "python/_autosummary/mlx.core.cummin", "python/_autosummary/mlx.core.cumprod", "python/_autosummary/mlx.core.cumsum", "python/_autosummary/mlx.core.custom_function", "python/_autosummary/mlx.core.default_device", "python/_autosummary/mlx.core.default_stream", "python/_autosummary/mlx.core.degrees", "python/_autosummary/mlx.core.dequantize", "python/_autosummary/mlx.core.diag", "python/_autosummary/mlx.core.diagonal", "python/_autosummary/mlx.core.disable_compile", "python/_autosummary/mlx.core.distributed.Group", "python/_autosummary/mlx.core.distributed.all_gather", "python/_autosummary/mlx.core.distributed.all_sum", "python/_autosummary/mlx.core.distributed.init", "python/_autosummary/mlx.core.distributed.is_available", "python/_autosummary/mlx.core.distributed.recv", "python/_autosummary/mlx.core.distributed.recv_like", "python/_autosummary/mlx.core.distributed.send", "python/_autosummary/mlx.core.divide", "python/_autosummary/mlx.core.divmod", "python/_autosummary/mlx.core.einsum", "python/_autosummary/mlx.core.einsum_path", "python/_autosummary/mlx.core.enable_compile", "python/_autosummary/mlx.core.equal", "python/_autosummary/mlx.core.erf", "python/_autosummary/mlx.core.erfinv", "python/_autosummary/mlx.core.eval", "python/_autosummary/mlx.core.exp", "python/_autosummary/mlx.core.expand_dims", "python/_autosummary/mlx.core.expm1", "python/_autosummary/mlx.core.export_function", "python/_autosummary/mlx.core.export_to_dot", "python/_autosummary/mlx.core.exporter", "python/_autosummary/mlx.core.eye", "python/_autosummary/mlx.core.fast.layer_norm", "python/_autosummary/mlx.core.fast.metal_kernel", "python/_autosummary/mlx.core.fast.rms_norm", "python/_autosummary/mlx.core.fast.rope", "python/_autosummary/mlx.core.fast.scaled_dot_product_attention", "python/_autosummary/mlx.core.fft.fft", "python/_autosummary/mlx.core.fft.fft2", "python/_autosummary/mlx.core.fft.fftn", "python/_autosummary/mlx.core.fft.ifft", "python/_autosummary/mlx.core.fft.ifft2", "python/_autosummary/mlx.core.fft.ifftn", "python/_autosummary/mlx.core.fft.irfft", "python/_autosummary/mlx.core.fft.irfft2", "python/_autosummary/mlx.core.fft.irfftn", "python/_autosummary/mlx.core.fft.rfft", "python/_autosummary/mlx.core.fft.rfft2", "python/_autosummary/mlx.core.fft.rfftn", "python/_autosummary/mlx.core.finfo", "python/_autosummary/mlx.core.flatten", "python/_autosummary/mlx.core.floor", "python/_autosummary/mlx.core.floor_divide", "python/_autosummary/mlx.core.full", "python/_autosummary/mlx.core.gather_mm", "python/_autosummary/mlx.core.gather_qmm", "python/_autosummary/mlx.core.grad", "python/_autosummary/mlx.core.greater", "python/_autosummary/mlx.core.greater_equal", "python/_autosummary/mlx.core.hadamard_transform", "python/_autosummary/mlx.core.identity", "python/_autosummary/mlx.core.imag", "python/_autosummary/mlx.core.import_function", "python/_autosummary/mlx.core.inner", "python/_autosummary/mlx.core.isclose", "python/_autosummary/mlx.core.isfinite", "python/_autosummary/mlx.core.isinf", "python/_autosummary/mlx.core.isnan", "python/_autosummary/mlx.core.isneginf", "python/_autosummary/mlx.core.isposinf", "python/_autosummary/mlx.core.issubdtype", "python/_autosummary/mlx.core.jvp", "python/_autosummary/mlx.core.kron", "python/_autosummary/mlx.core.left_shift", "python/_autosummary/mlx.core.less", "python/_autosummary/mlx.core.less_equal", "python/_autosummary/mlx.core.linalg.cholesky", "python/_autosummary/mlx.core.linalg.cholesky_inv", "python/_autosummary/mlx.core.linalg.cross", "python/_autosummary/mlx.core.linalg.eigh", "python/_autosummary/mlx.core.linalg.eigvalsh", "python/_autosummary/mlx.core.linalg.inv", "python/_autosummary/mlx.core.linalg.norm", "python/_autosummary/mlx.core.linalg.qr", "python/_autosummary/mlx.core.linalg.svd", "python/_autosummary/mlx.core.linalg.tri_inv", "python/_autosummary/mlx.core.linspace", "python/_autosummary/mlx.core.load", "python/_autosummary/mlx.core.log", "python/_autosummary/mlx.core.log10", "python/_autosummary/mlx.core.log1p", "python/_autosummary/mlx.core.log2", "python/_autosummary/mlx.core.logaddexp", "python/_autosummary/mlx.core.logical_and", "python/_autosummary/mlx.core.logical_not", "python/_autosummary/mlx.core.logical_or", "python/_autosummary/mlx.core.logsumexp", "python/_autosummary/mlx.core.matmul", "python/_autosummary/mlx.core.max", "python/_autosummary/mlx.core.maximum", "python/_autosummary/mlx.core.mean", "python/_autosummary/mlx.core.meshgrid", "python/_autosummary/mlx.core.metal.clear_cache", "python/_autosummary/mlx.core.metal.device_info", "python/_autosummary/mlx.core.metal.get_active_memory", "python/_autosummary/mlx.core.metal.get_cache_memory", "python/_autosummary/mlx.core.metal.get_peak_memory", "python/_autosummary/mlx.core.metal.is_available", "python/_autosummary/mlx.core.metal.reset_peak_memory", "python/_autosummary/mlx.core.metal.set_cache_limit", "python/_autosummary/mlx.core.metal.set_memory_limit", "python/_autosummary/mlx.core.metal.set_wired_limit", "python/_autosummary/mlx.core.metal.start_capture", "python/_autosummary/mlx.core.metal.stop_capture", "python/_autosummary/mlx.core.min", "python/_autosummary/mlx.core.minimum", "python/_autosummary/mlx.core.moveaxis", "python/_autosummary/mlx.core.multiply", "python/_autosummary/mlx.core.nan_to_num", "python/_autosummary/mlx.core.negative", "python/_autosummary/mlx.core.new_stream", "python/_autosummary/mlx.core.not_equal", "python/_autosummary/mlx.core.ones", "python/_autosummary/mlx.core.ones_like", "python/_autosummary/mlx.core.outer", "python/_autosummary/mlx.core.pad", "python/_autosummary/mlx.core.partition", "python/_autosummary/mlx.core.power", "python/_autosummary/mlx.core.prod", "python/_autosummary/mlx.core.put_along_axis", "python/_autosummary/mlx.core.quantize", "python/_autosummary/mlx.core.quantized_matmul", "python/_autosummary/mlx.core.radians", "python/_autosummary/mlx.core.random.bernoulli", "python/_autosummary/mlx.core.random.categorical", "python/_autosummary/mlx.core.random.gumbel", "python/_autosummary/mlx.core.random.key", "python/_autosummary/mlx.core.random.laplace", "python/_autosummary/mlx.core.random.multivariate_normal", "python/_autosummary/mlx.core.random.normal", "python/_autosummary/mlx.core.random.permutation", "python/_autosummary/mlx.core.random.randint", "python/_autosummary/mlx.core.random.seed", "python/_autosummary/mlx.core.random.split", "python/_autosummary/mlx.core.random.truncated_normal", "python/_autosummary/mlx.core.random.uniform", "python/_autosummary/mlx.core.real", "python/_autosummary/mlx.core.reciprocal", "python/_autosummary/mlx.core.remainder", "python/_autosummary/mlx.core.repeat", "python/_autosummary/mlx.core.reshape", "python/_autosummary/mlx.core.right_shift", "python/_autosummary/mlx.core.roll", "python/_autosummary/mlx.core.round", "python/_autosummary/mlx.core.rsqrt", "python/_autosummary/mlx.core.save", "python/_autosummary/mlx.core.save_gguf", "python/_autosummary/mlx.core.save_safetensors", "python/_autosummary/mlx.core.savez", "python/_autosummary/mlx.core.savez_compressed", "python/_autosummary/mlx.core.set_default_device", "python/_autosummary/mlx.core.set_default_stream", "python/_autosummary/mlx.core.sigmoid", "python/_autosummary/mlx.core.sign", "python/_autosummary/mlx.core.sin", "python/_autosummary/mlx.core.sinh", "python/_autosummary/mlx.core.slice", "python/_autosummary/mlx.core.slice_update", "python/_autosummary/mlx.core.softmax", "python/_autosummary/mlx.core.sort", "python/_autosummary/mlx.core.split", "python/_autosummary/mlx.core.sqrt", "python/_autosummary/mlx.core.square", "python/_autosummary/mlx.core.squeeze", "python/_autosummary/mlx.core.stack", "python/_autosummary/mlx.core.std", "python/_autosummary/mlx.core.stop_gradient", "python/_autosummary/mlx.core.stream", "python/_autosummary/mlx.core.subtract", "python/_autosummary/mlx.core.sum", "python/_autosummary/mlx.core.swapaxes", "python/_autosummary/mlx.core.synchronize", "python/_autosummary/mlx.core.take", "python/_autosummary/mlx.core.take_along_axis", "python/_autosummary/mlx.core.tan", "python/_autosummary/mlx.core.tanh", "python/_autosummary/mlx.core.tensordot", "python/_autosummary/mlx.core.tile", "python/_autosummary/mlx.core.topk", "python/_autosummary/mlx.core.trace", "python/_autosummary/mlx.core.transpose", "python/_autosummary/mlx.core.tri", "python/_autosummary/mlx.core.tril", "python/_autosummary/mlx.core.triu", "python/_autosummary/mlx.core.unflatten", "python/_autosummary/mlx.core.value_and_grad", "python/_autosummary/mlx.core.var", "python/_autosummary/mlx.core.view", "python/_autosummary/mlx.core.vjp", "python/_autosummary/mlx.core.vmap", "python/_autosummary/mlx.core.where", "python/_autosummary/mlx.core.zeros", "python/_autosummary/mlx.core.zeros_like", "python/_autosummary/mlx.nn.quantize", "python/_autosummary/mlx.nn.value_and_grad", "python/_autosummary/mlx.optimizers.clip_grad_norm", "python/_autosummary/mlx.utils.tree_flatten", "python/_autosummary/mlx.utils.tree_map", "python/_autosummary/mlx.utils.tree_map_with_path", "python/_autosummary/mlx.utils.tree_reduce", "python/_autosummary/mlx.utils.tree_unflatten", "python/_autosummary/stream_class", "python/array", "python/data_types", "python/devices_and_streams", "python/distributed", "python/export", "python/fast", "python/fft", "python/linalg", "python/metal", "python/nn", "python/nn/_autosummary/mlx.nn.ALiBi", "python/nn/_autosummary/mlx.nn.AvgPool1d", "python/nn/_autosummary/mlx.nn.AvgPool2d", "python/nn/_autosummary/mlx.nn.AvgPool3d", "python/nn/_autosummary/mlx.nn.BatchNorm", "python/nn/_autosummary/mlx.nn.CELU", "python/nn/_autosummary/mlx.nn.Conv1d", "python/nn/_autosummary/mlx.nn.Conv2d", "python/nn/_autosummary/mlx.nn.Conv3d", "python/nn/_autosummary/mlx.nn.ConvTranspose1d", "python/nn/_autosummary/mlx.nn.ConvTranspose2d", "python/nn/_autosummary/mlx.nn.ConvTranspose3d", "python/nn/_autosummary/mlx.nn.Dropout", "python/nn/_autosummary/mlx.nn.Dropout2d", "python/nn/_autosummary/mlx.nn.Dropout3d", "python/nn/_autosummary/mlx.nn.ELU", "python/nn/_autosummary/mlx.nn.Embedding", "python/nn/_autosummary/mlx.nn.GELU", "python/nn/_autosummary/mlx.nn.GLU", "python/nn/_autosummary/mlx.nn.GRU", "python/nn/_autosummary/mlx.nn.GroupNorm", "python/nn/_autosummary/mlx.nn.HardShrink", "python/nn/_autosummary/mlx.nn.HardTanh", "python/nn/_autosummary/mlx.nn.Hardswish", "python/nn/_autosummary/mlx.nn.InstanceNorm", "python/nn/_autosummary/mlx.nn.LSTM", "python/nn/_autosummary/mlx.nn.LayerNorm", "python/nn/_autosummary/mlx.nn.LeakyReLU", "python/nn/_autosummary/mlx.nn.Linear", "python/nn/_autosummary/mlx.nn.LogSigmoid", "python/nn/_autosummary/mlx.nn.LogSoftmax", "python/nn/_autosummary/mlx.nn.MaxPool1d", "python/nn/_autosummary/mlx.nn.MaxPool2d", "python/nn/_autosummary/mlx.nn.MaxPool3d", "python/nn/_autosummary/mlx.nn.Mish", "python/nn/_autosummary/mlx.nn.Module.apply", "python/nn/_autosummary/mlx.nn.Module.apply_to_modules", "python/nn/_autosummary/mlx.nn.Module.children", "python/nn/_autosummary/mlx.nn.Module.eval", "python/nn/_autosummary/mlx.nn.Module.filter_and_map", "python/nn/_autosummary/mlx.nn.Module.freeze", "python/nn/_autosummary/mlx.nn.Module.leaf_modules", "python/nn/_autosummary/mlx.nn.Module.load_weights", "python/nn/_autosummary/mlx.nn.Module.modules", "python/nn/_autosummary/mlx.nn.Module.named_modules", "python/nn/_autosummary/mlx.nn.Module.parameters", "python/nn/_autosummary/mlx.nn.Module.save_weights", "python/nn/_autosummary/mlx.nn.Module.set_dtype", "python/nn/_autosummary/mlx.nn.Module.state", "python/nn/_autosummary/mlx.nn.Module.train", "python/nn/_autosummary/mlx.nn.Module.trainable_parameters", "python/nn/_autosummary/mlx.nn.Module.training", "python/nn/_autosummary/mlx.nn.Module.unfreeze", "python/nn/_autosummary/mlx.nn.Module.update", "python/nn/_autosummary/mlx.nn.Module.update_modules", "python/nn/_autosummary/mlx.nn.MultiHeadAttention", "python/nn/_autosummary/mlx.nn.PReLU", "python/nn/_autosummary/mlx.nn.QuantizedEmbedding", "python/nn/_autosummary/mlx.nn.QuantizedLinear", "python/nn/_autosummary/mlx.nn.RMSNorm", "python/nn/_autosummary/mlx.nn.RNN", "python/nn/_autosummary/mlx.nn.ReLU", "python/nn/_autosummary/mlx.nn.ReLU6", "python/nn/_autosummary/mlx.nn.RoPE", "python/nn/_autosummary/mlx.nn.SELU", "python/nn/_autosummary/mlx.nn.Sequential", "python/nn/_autosummary/mlx.nn.SiLU", "python/nn/_autosummary/mlx.nn.Sigmoid", "python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding", "python/nn/_autosummary/mlx.nn.Softmax", "python/nn/_autosummary/mlx.nn.Softmin", "python/nn/_autosummary/mlx.nn.Softplus", "python/nn/_autosummary/mlx.nn.Softshrink", "python/nn/_autosummary/mlx.nn.Softsign", "python/nn/_autosummary/mlx.nn.Step", "python/nn/_autosummary/mlx.nn.Tanh", "python/nn/_autosummary/mlx.nn.Transformer", "python/nn/_autosummary/mlx.nn.Upsample", "python/nn/_autosummary/mlx.nn.init.constant", "python/nn/_autosummary/mlx.nn.init.glorot_normal", "python/nn/_autosummary/mlx.nn.init.glorot_uniform", "python/nn/_autosummary/mlx.nn.init.he_normal", "python/nn/_autosummary/mlx.nn.init.he_uniform", "python/nn/_autosummary/mlx.nn.init.identity", "python/nn/_autosummary/mlx.nn.init.normal", "python/nn/_autosummary/mlx.nn.init.uniform", "python/nn/_autosummary_functions/mlx.nn.celu", "python/nn/_autosummary_functions/mlx.nn.elu", "python/nn/_autosummary_functions/mlx.nn.gelu", "python/nn/_autosummary_functions/mlx.nn.gelu_approx", "python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx", "python/nn/_autosummary_functions/mlx.nn.glu", "python/nn/_autosummary_functions/mlx.nn.hard_shrink", "python/nn/_autosummary_functions/mlx.nn.hard_tanh", "python/nn/_autosummary_functions/mlx.nn.hardswish", "python/nn/_autosummary_functions/mlx.nn.leaky_relu", "python/nn/_autosummary_functions/mlx.nn.log_sigmoid", "python/nn/_autosummary_functions/mlx.nn.log_softmax", "python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy", "python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss", "python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy", "python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss", "python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss", "python/nn/_autosummary_functions/mlx.nn.losses.huber_loss", "python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss", "python/nn/_autosummary_functions/mlx.nn.losses.l1_loss", "python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss", "python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss", "python/nn/_autosummary_functions/mlx.nn.losses.mse_loss", "python/nn/_autosummary_functions/mlx.nn.losses.nll_loss", "python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss", "python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss", "python/nn/_autosummary_functions/mlx.nn.mish", "python/nn/_autosummary_functions/mlx.nn.prelu", "python/nn/_autosummary_functions/mlx.nn.relu", "python/nn/_autosummary_functions/mlx.nn.relu6", "python/nn/_autosummary_functions/mlx.nn.selu", "python/nn/_autosummary_functions/mlx.nn.sigmoid", "python/nn/_autosummary_functions/mlx.nn.silu", "python/nn/_autosummary_functions/mlx.nn.softmax", "python/nn/_autosummary_functions/mlx.nn.softmin", "python/nn/_autosummary_functions/mlx.nn.softplus", "python/nn/_autosummary_functions/mlx.nn.softshrink", "python/nn/_autosummary_functions/mlx.nn.step", "python/nn/_autosummary_functions/mlx.nn.tanh", "python/nn/functions", "python/nn/init", "python/nn/layers", "python/nn/losses", "python/nn/module", "python/ops", "python/optimizers", "python/optimizers/_autosummary/mlx.optimizers.AdaDelta", "python/optimizers/_autosummary/mlx.optimizers.Adafactor", "python/optimizers/_autosummary/mlx.optimizers.Adagrad", "python/optimizers/_autosummary/mlx.optimizers.Adam", "python/optimizers/_autosummary/mlx.optimizers.AdamW", "python/optimizers/_autosummary/mlx.optimizers.Adamax", "python/optimizers/_autosummary/mlx.optimizers.Lion", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.init", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.state", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.update", "python/optimizers/_autosummary/mlx.optimizers.RMSprop", "python/optimizers/_autosummary/mlx.optimizers.SGD", "python/optimizers/_autosummary/mlx.optimizers.cosine_decay", "python/optimizers/_autosummary/mlx.optimizers.exponential_decay", "python/optimizers/_autosummary/mlx.optimizers.join_schedules", "python/optimizers/_autosummary/mlx.optimizers.linear_schedule", "python/optimizers/_autosummary/mlx.optimizers.step_decay", "python/optimizers/common_optimizers", "python/optimizers/optimizer", "python/optimizers/schedulers", "python/random", "python/transforms", "python/tree_utils", "usage/compile", "usage/distributed", "usage/export", "usage/function_transforms", "usage/indexing", "usage/lazy_evaluation", "usage/numpy", "usage/quick_start", "usage/saving_and_loading", "usage/unified_memory", "usage/using_streams"], "envversion": {"sphinx": 62, "sphinx.domains.c": 3, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 9, "sphinx.domains.index": 1, "sphinx.domains.javascript": 3, "sphinx.domains.math": 2, "sphinx.domains.python": 4, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.intersphinx": 1}, "filenames": ["cpp/ops.rst", "dev/custom_metal_kernels.rst", "dev/extensions.rst", "dev/metal_debugger.rst", "dev/mlx_in_cpp.rst", "examples/linear_regression.rst", "examples/llama-inference.rst", "examples/mlp.rst", "index.rst", "install.rst", "python/_autosummary/mlx.core.Device.rst", "python/_autosummary/mlx.core.Dtype.rst", "python/_autosummary/mlx.core.DtypeCategory.rst", "python/_autosummary/mlx.core.abs.rst", "python/_autosummary/mlx.core.add.rst", "python/_autosummary/mlx.core.addmm.rst", "python/_autosummary/mlx.core.all.rst", "python/_autosummary/mlx.core.allclose.rst", "python/_autosummary/mlx.core.any.rst", "python/_autosummary/mlx.core.arange.rst", "python/_autosummary/mlx.core.arccos.rst", "python/_autosummary/mlx.core.arccosh.rst", "python/_autosummary/mlx.core.arcsin.rst", "python/_autosummary/mlx.core.arcsinh.rst", "python/_autosummary/mlx.core.arctan.rst", "python/_autosummary/mlx.core.arctan2.rst", "python/_autosummary/mlx.core.arctanh.rst", "python/_autosummary/mlx.core.argmax.rst", "python/_autosummary/mlx.core.argmin.rst", "python/_autosummary/mlx.core.argpartition.rst", "python/_autosummary/mlx.core.argsort.rst", "python/_autosummary/mlx.core.array.rst", "python/_autosummary/mlx.core.array.T.rst", "python/_autosummary/mlx.core.array.abs.rst", "python/_autosummary/mlx.core.array.all.rst", "python/_autosummary/mlx.core.array.any.rst", "python/_autosummary/mlx.core.array.argmax.rst", "python/_autosummary/mlx.core.array.argmin.rst", "python/_autosummary/mlx.core.array.astype.rst", "python/_autosummary/mlx.core.array.at.rst", "python/_autosummary/mlx.core.array.conj.rst", "python/_autosummary/mlx.core.array.cos.rst", "python/_autosummary/mlx.core.array.cummax.rst", "python/_autosummary/mlx.core.array.cummin.rst", "python/_autosummary/mlx.core.array.cumprod.rst", "python/_autosummary/mlx.core.array.cumsum.rst", "python/_autosummary/mlx.core.array.diag.rst", "python/_autosummary/mlx.core.array.diagonal.rst", "python/_autosummary/mlx.core.array.dtype.rst", "python/_autosummary/mlx.core.array.exp.rst", "python/_autosummary/mlx.core.array.flatten.rst", "python/_autosummary/mlx.core.array.item.rst", "python/_autosummary/mlx.core.array.itemsize.rst", "python/_autosummary/mlx.core.array.log.rst", "python/_autosummary/mlx.core.array.log10.rst", "python/_autosummary/mlx.core.array.log1p.rst", "python/_autosummary/mlx.core.array.log2.rst", "python/_autosummary/mlx.core.array.logsumexp.rst", "python/_autosummary/mlx.core.array.max.rst", "python/_autosummary/mlx.core.array.mean.rst", "python/_autosummary/mlx.core.array.min.rst", "python/_autosummary/mlx.core.array.moveaxis.rst", "python/_autosummary/mlx.core.array.nbytes.rst", "python/_autosummary/mlx.core.array.ndim.rst", "python/_autosummary/mlx.core.array.prod.rst", "python/_autosummary/mlx.core.array.reciprocal.rst", "python/_autosummary/mlx.core.array.reshape.rst", "python/_autosummary/mlx.core.array.round.rst", "python/_autosummary/mlx.core.array.rsqrt.rst", "python/_autosummary/mlx.core.array.shape.rst", "python/_autosummary/mlx.core.array.sin.rst", "python/_autosummary/mlx.core.array.size.rst", "python/_autosummary/mlx.core.array.split.rst", "python/_autosummary/mlx.core.array.sqrt.rst", "python/_autosummary/mlx.core.array.square.rst", "python/_autosummary/mlx.core.array.squeeze.rst", "python/_autosummary/mlx.core.array.std.rst", "python/_autosummary/mlx.core.array.sum.rst", "python/_autosummary/mlx.core.array.swapaxes.rst", "python/_autosummary/mlx.core.array.tolist.rst", "python/_autosummary/mlx.core.array.transpose.rst", "python/_autosummary/mlx.core.array.var.rst", "python/_autosummary/mlx.core.array.view.rst", "python/_autosummary/mlx.core.array_equal.rst", "python/_autosummary/mlx.core.as_strided.rst", "python/_autosummary/mlx.core.atleast_1d.rst", "python/_autosummary/mlx.core.atleast_2d.rst", "python/_autosummary/mlx.core.atleast_3d.rst", "python/_autosummary/mlx.core.bitwise_and.rst", "python/_autosummary/mlx.core.bitwise_or.rst", "python/_autosummary/mlx.core.bitwise_xor.rst", "python/_autosummary/mlx.core.block_masked_mm.rst", "python/_autosummary/mlx.core.broadcast_to.rst", "python/_autosummary/mlx.core.ceil.rst", "python/_autosummary/mlx.core.clip.rst", "python/_autosummary/mlx.core.compile.rst", "python/_autosummary/mlx.core.concatenate.rst", "python/_autosummary/mlx.core.conj.rst", "python/_autosummary/mlx.core.conjugate.rst", "python/_autosummary/mlx.core.conv1d.rst", "python/_autosummary/mlx.core.conv2d.rst", "python/_autosummary/mlx.core.conv3d.rst", "python/_autosummary/mlx.core.conv_general.rst", "python/_autosummary/mlx.core.conv_transpose1d.rst", "python/_autosummary/mlx.core.conv_transpose2d.rst", "python/_autosummary/mlx.core.conv_transpose3d.rst", "python/_autosummary/mlx.core.convolve.rst", "python/_autosummary/mlx.core.cos.rst", "python/_autosummary/mlx.core.cosh.rst", "python/_autosummary/mlx.core.cummax.rst", "python/_autosummary/mlx.core.cummin.rst", "python/_autosummary/mlx.core.cumprod.rst", "python/_autosummary/mlx.core.cumsum.rst", "python/_autosummary/mlx.core.custom_function.rst", "python/_autosummary/mlx.core.default_device.rst", "python/_autosummary/mlx.core.default_stream.rst", "python/_autosummary/mlx.core.degrees.rst", "python/_autosummary/mlx.core.dequantize.rst", "python/_autosummary/mlx.core.diag.rst", "python/_autosummary/mlx.core.diagonal.rst", "python/_autosummary/mlx.core.disable_compile.rst", "python/_autosummary/mlx.core.distributed.Group.rst", "python/_autosummary/mlx.core.distributed.all_gather.rst", "python/_autosummary/mlx.core.distributed.all_sum.rst", "python/_autosummary/mlx.core.distributed.init.rst", "python/_autosummary/mlx.core.distributed.is_available.rst", "python/_autosummary/mlx.core.distributed.recv.rst", "python/_autosummary/mlx.core.distributed.recv_like.rst", "python/_autosummary/mlx.core.distributed.send.rst", "python/_autosummary/mlx.core.divide.rst", "python/_autosummary/mlx.core.divmod.rst", "python/_autosummary/mlx.core.einsum.rst", "python/_autosummary/mlx.core.einsum_path.rst", "python/_autosummary/mlx.core.enable_compile.rst", "python/_autosummary/mlx.core.equal.rst", "python/_autosummary/mlx.core.erf.rst", "python/_autosummary/mlx.core.erfinv.rst", "python/_autosummary/mlx.core.eval.rst", "python/_autosummary/mlx.core.exp.rst", "python/_autosummary/mlx.core.expand_dims.rst", "python/_autosummary/mlx.core.expm1.rst", "python/_autosummary/mlx.core.export_function.rst", "python/_autosummary/mlx.core.export_to_dot.rst", "python/_autosummary/mlx.core.exporter.rst", "python/_autosummary/mlx.core.eye.rst", "python/_autosummary/mlx.core.fast.layer_norm.rst", "python/_autosummary/mlx.core.fast.metal_kernel.rst", "python/_autosummary/mlx.core.fast.rms_norm.rst", "python/_autosummary/mlx.core.fast.rope.rst", "python/_autosummary/mlx.core.fast.scaled_dot_product_attention.rst", "python/_autosummary/mlx.core.fft.fft.rst", "python/_autosummary/mlx.core.fft.fft2.rst", "python/_autosummary/mlx.core.fft.fftn.rst", "python/_autosummary/mlx.core.fft.ifft.rst", "python/_autosummary/mlx.core.fft.ifft2.rst", "python/_autosummary/mlx.core.fft.ifftn.rst", "python/_autosummary/mlx.core.fft.irfft.rst", "python/_autosummary/mlx.core.fft.irfft2.rst", "python/_autosummary/mlx.core.fft.irfftn.rst", "python/_autosummary/mlx.core.fft.rfft.rst", "python/_autosummary/mlx.core.fft.rfft2.rst", "python/_autosummary/mlx.core.fft.rfftn.rst", "python/_autosummary/mlx.core.finfo.rst", "python/_autosummary/mlx.core.flatten.rst", "python/_autosummary/mlx.core.floor.rst", "python/_autosummary/mlx.core.floor_divide.rst", "python/_autosummary/mlx.core.full.rst", "python/_autosummary/mlx.core.gather_mm.rst", "python/_autosummary/mlx.core.gather_qmm.rst", "python/_autosummary/mlx.core.grad.rst", "python/_autosummary/mlx.core.greater.rst", "python/_autosummary/mlx.core.greater_equal.rst", "python/_autosummary/mlx.core.hadamard_transform.rst", "python/_autosummary/mlx.core.identity.rst", "python/_autosummary/mlx.core.imag.rst", "python/_autosummary/mlx.core.import_function.rst", "python/_autosummary/mlx.core.inner.rst", "python/_autosummary/mlx.core.isclose.rst", "python/_autosummary/mlx.core.isfinite.rst", "python/_autosummary/mlx.core.isinf.rst", "python/_autosummary/mlx.core.isnan.rst", "python/_autosummary/mlx.core.isneginf.rst", "python/_autosummary/mlx.core.isposinf.rst", "python/_autosummary/mlx.core.issubdtype.rst", "python/_autosummary/mlx.core.jvp.rst", "python/_autosummary/mlx.core.kron.rst", "python/_autosummary/mlx.core.left_shift.rst", "python/_autosummary/mlx.core.less.rst", "python/_autosummary/mlx.core.less_equal.rst", "python/_autosummary/mlx.core.linalg.cholesky.rst", "python/_autosummary/mlx.core.linalg.cholesky_inv.rst", "python/_autosummary/mlx.core.linalg.cross.rst", "python/_autosummary/mlx.core.linalg.eigh.rst", "python/_autosummary/mlx.core.linalg.eigvalsh.rst", "python/_autosummary/mlx.core.linalg.inv.rst", "python/_autosummary/mlx.core.linalg.norm.rst", "python/_autosummary/mlx.core.linalg.qr.rst", "python/_autosummary/mlx.core.linalg.svd.rst", "python/_autosummary/mlx.core.linalg.tri_inv.rst", "python/_autosummary/mlx.core.linspace.rst", "python/_autosummary/mlx.core.load.rst", "python/_autosummary/mlx.core.log.rst", "python/_autosummary/mlx.core.log10.rst", "python/_autosummary/mlx.core.log1p.rst", "python/_autosummary/mlx.core.log2.rst", "python/_autosummary/mlx.core.logaddexp.rst", "python/_autosummary/mlx.core.logical_and.rst", "python/_autosummary/mlx.core.logical_not.rst", "python/_autosummary/mlx.core.logical_or.rst", "python/_autosummary/mlx.core.logsumexp.rst", "python/_autosummary/mlx.core.matmul.rst", "python/_autosummary/mlx.core.max.rst", "python/_autosummary/mlx.core.maximum.rst", "python/_autosummary/mlx.core.mean.rst", "python/_autosummary/mlx.core.meshgrid.rst", "python/_autosummary/mlx.core.metal.clear_cache.rst", "python/_autosummary/mlx.core.metal.device_info.rst", "python/_autosummary/mlx.core.metal.get_active_memory.rst", "python/_autosummary/mlx.core.metal.get_cache_memory.rst", "python/_autosummary/mlx.core.metal.get_peak_memory.rst", "python/_autosummary/mlx.core.metal.is_available.rst", "python/_autosummary/mlx.core.metal.reset_peak_memory.rst", "python/_autosummary/mlx.core.metal.set_cache_limit.rst", "python/_autosummary/mlx.core.metal.set_memory_limit.rst", "python/_autosummary/mlx.core.metal.set_wired_limit.rst", "python/_autosummary/mlx.core.metal.start_capture.rst", "python/_autosummary/mlx.core.metal.stop_capture.rst", "python/_autosummary/mlx.core.min.rst", "python/_autosummary/mlx.core.minimum.rst", "python/_autosummary/mlx.core.moveaxis.rst", "python/_autosummary/mlx.core.multiply.rst", "python/_autosummary/mlx.core.nan_to_num.rst", "python/_autosummary/mlx.core.negative.rst", "python/_autosummary/mlx.core.new_stream.rst", "python/_autosummary/mlx.core.not_equal.rst", "python/_autosummary/mlx.core.ones.rst", "python/_autosummary/mlx.core.ones_like.rst", "python/_autosummary/mlx.core.outer.rst", "python/_autosummary/mlx.core.pad.rst", "python/_autosummary/mlx.core.partition.rst", "python/_autosummary/mlx.core.power.rst", "python/_autosummary/mlx.core.prod.rst", "python/_autosummary/mlx.core.put_along_axis.rst", "python/_autosummary/mlx.core.quantize.rst", "python/_autosummary/mlx.core.quantized_matmul.rst", "python/_autosummary/mlx.core.radians.rst", "python/_autosummary/mlx.core.random.bernoulli.rst", "python/_autosummary/mlx.core.random.categorical.rst", "python/_autosummary/mlx.core.random.gumbel.rst", "python/_autosummary/mlx.core.random.key.rst", "python/_autosummary/mlx.core.random.laplace.rst", "python/_autosummary/mlx.core.random.multivariate_normal.rst", "python/_autosummary/mlx.core.random.normal.rst", "python/_autosummary/mlx.core.random.permutation.rst", "python/_autosummary/mlx.core.random.randint.rst", "python/_autosummary/mlx.core.random.seed.rst", "python/_autosummary/mlx.core.random.split.rst", "python/_autosummary/mlx.core.random.truncated_normal.rst", "python/_autosummary/mlx.core.random.uniform.rst", "python/_autosummary/mlx.core.real.rst", "python/_autosummary/mlx.core.reciprocal.rst", "python/_autosummary/mlx.core.remainder.rst", "python/_autosummary/mlx.core.repeat.rst", "python/_autosummary/mlx.core.reshape.rst", "python/_autosummary/mlx.core.right_shift.rst", "python/_autosummary/mlx.core.roll.rst", "python/_autosummary/mlx.core.round.rst", "python/_autosummary/mlx.core.rsqrt.rst", "python/_autosummary/mlx.core.save.rst", "python/_autosummary/mlx.core.save_gguf.rst", "python/_autosummary/mlx.core.save_safetensors.rst", "python/_autosummary/mlx.core.savez.rst", "python/_autosummary/mlx.core.savez_compressed.rst", "python/_autosummary/mlx.core.set_default_device.rst", "python/_autosummary/mlx.core.set_default_stream.rst", "python/_autosummary/mlx.core.sigmoid.rst", "python/_autosummary/mlx.core.sign.rst", "python/_autosummary/mlx.core.sin.rst", "python/_autosummary/mlx.core.sinh.rst", "python/_autosummary/mlx.core.slice.rst", "python/_autosummary/mlx.core.slice_update.rst", "python/_autosummary/mlx.core.softmax.rst", "python/_autosummary/mlx.core.sort.rst", "python/_autosummary/mlx.core.split.rst", "python/_autosummary/mlx.core.sqrt.rst", "python/_autosummary/mlx.core.square.rst", "python/_autosummary/mlx.core.squeeze.rst", "python/_autosummary/mlx.core.stack.rst", "python/_autosummary/mlx.core.std.rst", "python/_autosummary/mlx.core.stop_gradient.rst", "python/_autosummary/mlx.core.stream.rst", "python/_autosummary/mlx.core.subtract.rst", "python/_autosummary/mlx.core.sum.rst", "python/_autosummary/mlx.core.swapaxes.rst", "python/_autosummary/mlx.core.synchronize.rst", "python/_autosummary/mlx.core.take.rst", "python/_autosummary/mlx.core.take_along_axis.rst", "python/_autosummary/mlx.core.tan.rst", "python/_autosummary/mlx.core.tanh.rst", "python/_autosummary/mlx.core.tensordot.rst", "python/_autosummary/mlx.core.tile.rst", "python/_autosummary/mlx.core.topk.rst", "python/_autosummary/mlx.core.trace.rst", "python/_autosummary/mlx.core.transpose.rst", "python/_autosummary/mlx.core.tri.rst", "python/_autosummary/mlx.core.tril.rst", "python/_autosummary/mlx.core.triu.rst", "python/_autosummary/mlx.core.unflatten.rst", "python/_autosummary/mlx.core.value_and_grad.rst", "python/_autosummary/mlx.core.var.rst", "python/_autosummary/mlx.core.view.rst", "python/_autosummary/mlx.core.vjp.rst", "python/_autosummary/mlx.core.vmap.rst", "python/_autosummary/mlx.core.where.rst", "python/_autosummary/mlx.core.zeros.rst", "python/_autosummary/mlx.core.zeros_like.rst", "python/_autosummary/mlx.nn.quantize.rst", "python/_autosummary/mlx.nn.value_and_grad.rst", "python/_autosummary/mlx.optimizers.clip_grad_norm.rst", "python/_autosummary/mlx.utils.tree_flatten.rst", "python/_autosummary/mlx.utils.tree_map.rst", "python/_autosummary/mlx.utils.tree_map_with_path.rst", "python/_autosummary/mlx.utils.tree_reduce.rst", "python/_autosummary/mlx.utils.tree_unflatten.rst", "python/_autosummary/stream_class.rst", "python/array.rst", "python/data_types.rst", "python/devices_and_streams.rst", "python/distributed.rst", "python/export.rst", "python/fast.rst", "python/fft.rst", "python/linalg.rst", "python/metal.rst", "python/nn.rst", "python/nn/_autosummary/mlx.nn.ALiBi.rst", "python/nn/_autosummary/mlx.nn.AvgPool1d.rst", "python/nn/_autosummary/mlx.nn.AvgPool2d.rst", "python/nn/_autosummary/mlx.nn.AvgPool3d.rst", "python/nn/_autosummary/mlx.nn.BatchNorm.rst", "python/nn/_autosummary/mlx.nn.CELU.rst", "python/nn/_autosummary/mlx.nn.Conv1d.rst", "python/nn/_autosummary/mlx.nn.Conv2d.rst", "python/nn/_autosummary/mlx.nn.Conv3d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose1d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose2d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose3d.rst", "python/nn/_autosummary/mlx.nn.Dropout.rst", "python/nn/_autosummary/mlx.nn.Dropout2d.rst", "python/nn/_autosummary/mlx.nn.Dropout3d.rst", "python/nn/_autosummary/mlx.nn.ELU.rst", "python/nn/_autosummary/mlx.nn.Embedding.rst", "python/nn/_autosummary/mlx.nn.GELU.rst", "python/nn/_autosummary/mlx.nn.GLU.rst", "python/nn/_autosummary/mlx.nn.GRU.rst", "python/nn/_autosummary/mlx.nn.GroupNorm.rst", "python/nn/_autosummary/mlx.nn.HardShrink.rst", "python/nn/_autosummary/mlx.nn.HardTanh.rst", "python/nn/_autosummary/mlx.nn.Hardswish.rst", "python/nn/_autosummary/mlx.nn.InstanceNorm.rst", "python/nn/_autosummary/mlx.nn.LSTM.rst", "python/nn/_autosummary/mlx.nn.LayerNorm.rst", "python/nn/_autosummary/mlx.nn.LeakyReLU.rst", "python/nn/_autosummary/mlx.nn.Linear.rst", "python/nn/_autosummary/mlx.nn.LogSigmoid.rst", "python/nn/_autosummary/mlx.nn.LogSoftmax.rst", "python/nn/_autosummary/mlx.nn.MaxPool1d.rst", "python/nn/_autosummary/mlx.nn.MaxPool2d.rst", "python/nn/_autosummary/mlx.nn.MaxPool3d.rst", "python/nn/_autosummary/mlx.nn.Mish.rst", "python/nn/_autosummary/mlx.nn.Module.apply.rst", "python/nn/_autosummary/mlx.nn.Module.apply_to_modules.rst", "python/nn/_autosummary/mlx.nn.Module.children.rst", "python/nn/_autosummary/mlx.nn.Module.eval.rst", "python/nn/_autosummary/mlx.nn.Module.filter_and_map.rst", "python/nn/_autosummary/mlx.nn.Module.freeze.rst", "python/nn/_autosummary/mlx.nn.Module.leaf_modules.rst", "python/nn/_autosummary/mlx.nn.Module.load_weights.rst", "python/nn/_autosummary/mlx.nn.Module.modules.rst", "python/nn/_autosummary/mlx.nn.Module.named_modules.rst", "python/nn/_autosummary/mlx.nn.Module.parameters.rst", "python/nn/_autosummary/mlx.nn.Module.save_weights.rst", "python/nn/_autosummary/mlx.nn.Module.set_dtype.rst", "python/nn/_autosummary/mlx.nn.Module.state.rst", "python/nn/_autosummary/mlx.nn.Module.train.rst", "python/nn/_autosummary/mlx.nn.Module.trainable_parameters.rst", "python/nn/_autosummary/mlx.nn.Module.training.rst", "python/nn/_autosummary/mlx.nn.Module.unfreeze.rst", "python/nn/_autosummary/mlx.nn.Module.update.rst", "python/nn/_autosummary/mlx.nn.Module.update_modules.rst", "python/nn/_autosummary/mlx.nn.MultiHeadAttention.rst", "python/nn/_autosummary/mlx.nn.PReLU.rst", "python/nn/_autosummary/mlx.nn.QuantizedEmbedding.rst", "python/nn/_autosummary/mlx.nn.QuantizedLinear.rst", "python/nn/_autosummary/mlx.nn.RMSNorm.rst", "python/nn/_autosummary/mlx.nn.RNN.rst", "python/nn/_autosummary/mlx.nn.ReLU.rst", "python/nn/_autosummary/mlx.nn.ReLU6.rst", "python/nn/_autosummary/mlx.nn.RoPE.rst", "python/nn/_autosummary/mlx.nn.SELU.rst", "python/nn/_autosummary/mlx.nn.Sequential.rst", "python/nn/_autosummary/mlx.nn.SiLU.rst", "python/nn/_autosummary/mlx.nn.Sigmoid.rst", "python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.rst", "python/nn/_autosummary/mlx.nn.Softmax.rst", "python/nn/_autosummary/mlx.nn.Softmin.rst", "python/nn/_autosummary/mlx.nn.Softplus.rst", "python/nn/_autosummary/mlx.nn.Softshrink.rst", "python/nn/_autosummary/mlx.nn.Softsign.rst", "python/nn/_autosummary/mlx.nn.Step.rst", "python/nn/_autosummary/mlx.nn.Tanh.rst", "python/nn/_autosummary/mlx.nn.Transformer.rst", "python/nn/_autosummary/mlx.nn.Upsample.rst", "python/nn/_autosummary/mlx.nn.init.constant.rst", "python/nn/_autosummary/mlx.nn.init.glorot_normal.rst", "python/nn/_autosummary/mlx.nn.init.glorot_uniform.rst", "python/nn/_autosummary/mlx.nn.init.he_normal.rst", "python/nn/_autosummary/mlx.nn.init.he_uniform.rst", "python/nn/_autosummary/mlx.nn.init.identity.rst", "python/nn/_autosummary/mlx.nn.init.normal.rst", "python/nn/_autosummary/mlx.nn.init.uniform.rst", "python/nn/_autosummary_functions/mlx.nn.celu.rst", "python/nn/_autosummary_functions/mlx.nn.elu.rst", "python/nn/_autosummary_functions/mlx.nn.gelu.rst", "python/nn/_autosummary_functions/mlx.nn.gelu_approx.rst", "python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.rst", "python/nn/_autosummary_functions/mlx.nn.glu.rst", "python/nn/_autosummary_functions/mlx.nn.hard_shrink.rst", "python/nn/_autosummary_functions/mlx.nn.hard_tanh.rst", "python/nn/_autosummary_functions/mlx.nn.hardswish.rst", "python/nn/_autosummary_functions/mlx.nn.leaky_relu.rst", "python/nn/_autosummary_functions/mlx.nn.log_sigmoid.rst", "python/nn/_autosummary_functions/mlx.nn.log_softmax.rst", "python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.rst", "python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.rst", "python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.rst", "python/nn/_autosummary_functions/mlx.nn.mish.rst", "python/nn/_autosummary_functions/mlx.nn.prelu.rst", "python/nn/_autosummary_functions/mlx.nn.relu.rst", "python/nn/_autosummary_functions/mlx.nn.relu6.rst", "python/nn/_autosummary_functions/mlx.nn.selu.rst", "python/nn/_autosummary_functions/mlx.nn.sigmoid.rst", "python/nn/_autosummary_functions/mlx.nn.silu.rst", "python/nn/_autosummary_functions/mlx.nn.softmax.rst", "python/nn/_autosummary_functions/mlx.nn.softmin.rst", "python/nn/_autosummary_functions/mlx.nn.softplus.rst", "python/nn/_autosummary_functions/mlx.nn.softshrink.rst", "python/nn/_autosummary_functions/mlx.nn.step.rst", "python/nn/_autosummary_functions/mlx.nn.tanh.rst", "python/nn/functions.rst", "python/nn/init.rst", "python/nn/layers.rst", "python/nn/losses.rst", "python/nn/module.rst", "python/ops.rst", "python/optimizers.rst", "python/optimizers/_autosummary/mlx.optimizers.AdaDelta.rst", "python/optimizers/_autosummary/mlx.optimizers.Adafactor.rst", "python/optimizers/_autosummary/mlx.optimizers.Adagrad.rst", "python/optimizers/_autosummary/mlx.optimizers.Adam.rst", "python/optimizers/_autosummary/mlx.optimizers.AdamW.rst", "python/optimizers/_autosummary/mlx.optimizers.Adamax.rst", "python/optimizers/_autosummary/mlx.optimizers.Lion.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.rst", "python/optimizers/_autosummary/mlx.optimizers.RMSprop.rst", "python/optimizers/_autosummary/mlx.optimizers.SGD.rst", "python/optimizers/_autosummary/mlx.optimizers.cosine_decay.rst", "python/optimizers/_autosummary/mlx.optimizers.exponential_decay.rst", "python/optimizers/_autosummary/mlx.optimizers.join_schedules.rst", "python/optimizers/_autosummary/mlx.optimizers.linear_schedule.rst", "python/optimizers/_autosummary/mlx.optimizers.step_decay.rst", "python/optimizers/common_optimizers.rst", "python/optimizers/optimizer.rst", "python/optimizers/schedulers.rst", "python/random.rst", "python/transforms.rst", "python/tree_utils.rst", "usage/compile.rst", "usage/distributed.rst", "usage/export.rst", "usage/function_transforms.rst", "usage/indexing.rst", "usage/lazy_evaluation.rst", "usage/numpy.rst", "usage/quick_start.rst", "usage/saving_and_loading.rst", "usage/unified_memory.rst", "usage/using_streams.rst"], "indexentries": {"__init__() (array method)": [[31, "mlx.core.array.__init__", false]], "__init__() (custom_function method)": [[113, "mlx.core.custom_function.__init__", false]], "__init__() (device method)": [[10, "mlx.core.Device.__init__", false]], "__init__() (dtype method)": [[11, "mlx.core.Dtype.__init__", false]], "__init__() (dtypecategory method)": [[12, "mlx.core.DtypeCategory.__init__", false]], "__init__() (finfo method)": [[162, "mlx.core.finfo.__init__", false]], "__init__() (group method)": [[121, "mlx.core.distributed.Group.__init__", false]], "__init__() (stream method)": [[324, "mlx.core.Stream.__init__", false]], "abs (c++ function)": [[0, "_CPPv43absRK5array14StreamOrDevice", false]], "abs() (array method)": [[33, "mlx.core.array.abs", false]], "abs() (in module mlx.core)": [[13, "mlx.core.abs", false]], "adadelta (class in mlx.optimizers)": [[467, "mlx.optimizers.AdaDelta", false]], "adafactor (class in mlx.optimizers)": [[468, "mlx.optimizers.Adafactor", false]], "adagrad (class in mlx.optimizers)": [[469, "mlx.optimizers.Adagrad", false]], "adam (class in mlx.optimizers)": [[470, "mlx.optimizers.Adam", false]], "adamax (class in mlx.optimizers)": [[472, "mlx.optimizers.Adamax", false]], "adamw (class in mlx.optimizers)": [[471, "mlx.optimizers.AdamW", false]], "add (c++ function)": [[0, "_CPPv43addRK5arrayRK5array14StreamOrDevice", false]], "add() (in module mlx.core)": [[14, "mlx.core.add", false]], "addmm (c++ function)": [[0, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", false]], "addmm() (in module mlx.core)": [[15, "mlx.core.addmm", false]], "alibi (class in mlx.nn)": [[335, "mlx.nn.ALiBi", false]], "all (c++ function)": [[0, "_CPPv43allRK5array14StreamOrDevice", false], [0, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43allRK5arrayb14StreamOrDevice", false], [0, "_CPPv43allRK5arrayib14StreamOrDevice", false]], "all() (array method)": [[34, "mlx.core.array.all", false]], "all() (in module mlx.core)": [[16, "mlx.core.all", false]], "all_gather() (in module mlx.core.distributed)": [[122, "mlx.core.distributed.all_gather", false]], "all_sum() (in module mlx.core.distributed)": [[123, "mlx.core.distributed.all_sum", false]], "allclose (c++ function)": [[0, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "allclose() (in module mlx.core)": [[17, "mlx.core.allclose", false]], "any (c++ function)": [[0, "_CPPv43anyRK5array14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayb14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayib14StreamOrDevice", false]], "any() (array method)": [[35, "mlx.core.array.any", false]], "any() (in module mlx.core)": [[18, "mlx.core.any", false]], "apply() (module method)": [[370, "mlx.nn.Module.apply", false]], "apply_gradients() (optimizer method)": [[474, "mlx.optimizers.Optimizer.apply_gradients", false]], "apply_to_modules() (module method)": [[371, "mlx.nn.Module.apply_to_modules", false]], "arange (c++ function)": [[0, "_CPPv46aranged14StreamOrDevice", false], [0, "_CPPv46aranged5Dtype14StreamOrDevice", false], [0, "_CPPv46arangedd14StreamOrDevice", false], [0, "_CPPv46arangedd5Dtype14StreamOrDevice", false], [0, "_CPPv46arangeddd14StreamOrDevice", false], [0, "_CPPv46arangeddd5Dtype14StreamOrDevice", false], [0, "_CPPv46arangei14StreamOrDevice", false], [0, "_CPPv46arangeii14StreamOrDevice", false], [0, "_CPPv46arangeiii14StreamOrDevice", false]], "arange() (in module mlx.core)": [[19, "mlx.core.arange", false]], "arccos (c++ function)": [[0, "_CPPv46arccosRK5array14StreamOrDevice", false]], "arccos() (in module mlx.core)": [[20, "mlx.core.arccos", false]], "arccosh (c++ function)": [[0, "_CPPv47arccoshRK5array14StreamOrDevice", false]], "arccosh() (in module mlx.core)": [[21, "mlx.core.arccosh", false]], "arcsin (c++ function)": [[0, "_CPPv46arcsinRK5array14StreamOrDevice", false]], "arcsin() (in module mlx.core)": [[22, "mlx.core.arcsin", false]], "arcsinh (c++ function)": [[0, "_CPPv47arcsinhRK5array14StreamOrDevice", false]], "arcsinh() (in module mlx.core)": [[23, "mlx.core.arcsinh", false]], "arctan (c++ function)": [[0, "_CPPv46arctanRK5array14StreamOrDevice", false]], "arctan() (in module mlx.core)": [[24, "mlx.core.arctan", false]], "arctan2 (c++ function)": [[0, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", false]], "arctan2() (in module mlx.core)": [[25, "mlx.core.arctan2", false]], "arctanh (c++ function)": [[0, "_CPPv47arctanhRK5array14StreamOrDevice", false]], "arctanh() (in module mlx.core)": [[26, "mlx.core.arctanh", false]], "argmax (c++ function)": [[0, "_CPPv46argmaxRK5array14StreamOrDevice", false], [0, "_CPPv46argmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv46argmaxRK5arrayib14StreamOrDevice", false]], "argmax() (array method)": [[36, "mlx.core.array.argmax", false]], "argmax() (in module mlx.core)": [[27, "mlx.core.argmax", false]], "argmin (c++ function)": [[0, "_CPPv46argminRK5array14StreamOrDevice", false], [0, "_CPPv46argminRK5arrayb14StreamOrDevice", false], [0, "_CPPv46argminRK5arrayib14StreamOrDevice", false]], "argmin() (array method)": [[37, "mlx.core.array.argmin", false]], "argmin() (in module mlx.core)": [[28, "mlx.core.argmin", false]], "argpartition (c++ function)": [[0, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", false]], "argpartition() (in module mlx.core)": [[29, "mlx.core.argpartition", false]], "argsort (c++ function)": [[0, "_CPPv47argsortRK5array14StreamOrDevice", false], [0, "_CPPv47argsortRK5arrayi14StreamOrDevice", false]], "argsort() (in module mlx.core)": [[30, "mlx.core.argsort", false]], "array (class in mlx.core)": [[31, "mlx.core.array", false]], "array_equal (c++ function)": [[0, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", false], [0, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", false]], "array_equal() (in module mlx.core)": [[83, "mlx.core.array_equal", false]], "as_strided (c++ function)": [[0, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", false]], "as_strided() (in module mlx.core)": [[84, "mlx.core.as_strided", false]], "astype (c++ function)": [[0, "_CPPv46astype5array5Dtype14StreamOrDevice", false]], "astype() (array method)": [[38, "mlx.core.array.astype", false]], "at (array property)": [[39, "mlx.core.array.at", false]], "atleast_1d (c++ function)": [[0, "_CPPv410atleast_1dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_1d() (in module mlx.core)": [[85, "mlx.core.atleast_1d", false]], "atleast_2d (c++ function)": [[0, "_CPPv410atleast_2dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_2d() (in module mlx.core)": [[86, "mlx.core.atleast_2d", false]], "atleast_3d (c++ function)": [[0, "_CPPv410atleast_3dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_3d() (in module mlx.core)": [[87, "mlx.core.atleast_3d", false]], "avgpool1d (class in mlx.nn)": [[336, "mlx.nn.AvgPool1d", false]], "avgpool2d (class in mlx.nn)": [[337, "mlx.nn.AvgPool2d", false]], "avgpool3d (class in mlx.nn)": [[338, "mlx.nn.AvgPool3d", false]], "batchnorm (class in mlx.nn)": [[339, "mlx.nn.BatchNorm", false]], "bernoulli() (in module mlx.core.random)": [[246, "mlx.core.random.bernoulli", false]], "binary_cross_entropy (class in mlx.nn.losses)": [[433, "mlx.nn.losses.binary_cross_entropy", false]], "bitwise_and (c++ function)": [[0, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", false]], "bitwise_and() (in module mlx.core)": [[88, "mlx.core.bitwise_and", false]], "bitwise_or (c++ function)": [[0, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", false]], "bitwise_or() (in module mlx.core)": [[89, "mlx.core.bitwise_or", false]], "bitwise_xor (c++ function)": [[0, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", false]], "bitwise_xor() (in module mlx.core)": [[90, "mlx.core.bitwise_xor", false]], "block_masked_mm (c++ function)": [[0, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "block_masked_mm() (in module mlx.core)": [[91, "mlx.core.block_masked_mm", false]], "broadcast_arrays (c++ function)": [[0, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "broadcast_to (c++ function)": [[0, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", false]], "broadcast_to() (in module mlx.core)": [[92, "mlx.core.broadcast_to", false]], "categorical() (in module mlx.core.random)": [[247, "mlx.core.random.categorical", false]], "ceil (c++ function)": [[0, "_CPPv44ceilRK5array14StreamOrDevice", false]], "ceil() (in module mlx.core)": [[93, "mlx.core.ceil", false]], "celu (class in mlx.nn)": [[340, "mlx.nn.CELU", false], [421, "mlx.nn.celu", false]], "children() (module method)": [[372, "mlx.nn.Module.children", false]], "cholesky() (in module mlx.core.linalg)": [[189, "mlx.core.linalg.cholesky", false]], "cholesky_inv() (in module mlx.core.linalg)": [[190, "mlx.core.linalg.cholesky_inv", false]], "clear_cache() (in module mlx.core.metal)": [[215, "mlx.core.metal.clear_cache", false]], "clip (c++ function)": [[0, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", false]], "clip() (in module mlx.core)": [[94, "mlx.core.clip", false]], "clip_grad_norm() (in module mlx.optimizers)": [[318, "mlx.optimizers.clip_grad_norm", false]], "compile() (in module mlx.core)": [[95, "mlx.core.compile", false]], "concatenate (c++ function)": [[0, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", false]], "concatenate() (in module mlx.core)": [[96, "mlx.core.concatenate", false]], "conj() (array method)": [[40, "mlx.core.array.conj", false]], "conj() (in module mlx.core)": [[97, "mlx.core.conj", false]], "conjugate (c++ function)": [[0, "_CPPv49conjugateRK5array14StreamOrDevice", false]], "conjugate() (in module mlx.core)": [[98, "mlx.core.conjugate", false]], "constant() (in module mlx.nn.init)": [[413, "mlx.nn.init.constant", false]], "contiguous (c++ function)": [[0, "_CPPv410contiguousRK5arrayb14StreamOrDevice", false]], "conv1d (c++ function)": [[0, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv1d (class in mlx.nn)": [[341, "mlx.nn.Conv1d", false]], "conv1d() (in module mlx.core)": [[99, "mlx.core.conv1d", false]], "conv2d (c++ function)": [[0, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv2d (class in mlx.nn)": [[342, "mlx.nn.Conv2d", false]], "conv2d() (in module mlx.core)": [[100, "mlx.core.conv2d", false]], "conv3d (c++ function)": [[0, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv3d (class in mlx.nn)": [[343, "mlx.nn.Conv3d", false]], "conv3d() (in module mlx.core)": [[101, "mlx.core.conv3d", false]], "conv_general (c++ function)": [[0, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false], [0, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false]], "conv_general() (in module mlx.core)": [[102, "mlx.core.conv_general", false]], "conv_transpose1d (c++ function)": [[0, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv_transpose1d() (in module mlx.core)": [[103, "mlx.core.conv_transpose1d", false]], "conv_transpose2d (c++ function)": [[0, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv_transpose2d() (in module mlx.core)": [[104, "mlx.core.conv_transpose2d", false]], "conv_transpose3d (c++ function)": [[0, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv_transpose3d() (in module mlx.core)": [[105, "mlx.core.conv_transpose3d", false]], "convolve() (in module mlx.core)": [[106, "mlx.core.convolve", false]], "convtranspose1d (class in mlx.nn)": [[344, "mlx.nn.ConvTranspose1d", false]], "convtranspose2d (class in mlx.nn)": [[345, "mlx.nn.ConvTranspose2d", false]], "convtranspose3d (class in mlx.nn)": [[346, "mlx.nn.ConvTranspose3d", false]], "copy (c++ function)": [[0, "_CPPv44copy5array14StreamOrDevice", false]], "cos (c++ function)": [[0, "_CPPv43cosRK5array14StreamOrDevice", false]], "cos() (array method)": [[41, "mlx.core.array.cos", false]], "cos() (in module mlx.core)": [[107, "mlx.core.cos", false]], "cosh (c++ function)": [[0, "_CPPv44coshRK5array14StreamOrDevice", false]], "cosh() (in module mlx.core)": [[108, "mlx.core.cosh", false]], "cosine_decay() (in module mlx.optimizers)": [[480, "mlx.optimizers.cosine_decay", false]], "cosine_similarity_loss (class in mlx.nn.losses)": [[434, "mlx.nn.losses.cosine_similarity_loss", false]], "cross() (in module mlx.core.linalg)": [[191, "mlx.core.linalg.cross", false]], "cross_entropy (class in mlx.nn.losses)": [[435, "mlx.nn.losses.cross_entropy", false]], "cummax (c++ function)": [[0, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", false]], "cummax() (array method)": [[42, "mlx.core.array.cummax", false]], "cummax() (in module mlx.core)": [[109, "mlx.core.cummax", false]], "cummin (c++ function)": [[0, "_CPPv46cumminRK5arrayibb14StreamOrDevice", false]], "cummin() (array method)": [[43, "mlx.core.array.cummin", false]], "cummin() (in module mlx.core)": [[110, "mlx.core.cummin", false]], "cumprod (c++ function)": [[0, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", false]], "cumprod() (array method)": [[44, "mlx.core.array.cumprod", false]], "cumprod() (in module mlx.core)": [[111, "mlx.core.cumprod", false]], "cumsum (c++ function)": [[0, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", false]], "cumsum() (array method)": [[45, "mlx.core.array.cumsum", false]], "cumsum() (in module mlx.core)": [[112, "mlx.core.cumsum", false]], "custom_function (class in mlx.core)": [[113, "mlx.core.custom_function", false]], "default_device() (in module mlx.core)": [[114, "mlx.core.default_device", false]], "default_stream() (in module mlx.core)": [[115, "mlx.core.default_stream", false]], "degrees (c++ function)": [[0, "_CPPv47degreesRK5array14StreamOrDevice", false]], "degrees() (in module mlx.core)": [[116, "mlx.core.degrees", false]], "depends (c++ function)": [[0, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", false]], "dequantize (c++ function)": [[0, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", false]], "dequantize() (in module mlx.core)": [[117, "mlx.core.dequantize", false]], "device (class in mlx.core)": [[10, "mlx.core.Device", false]], "device_info() (in module mlx.core.metal)": [[216, "mlx.core.metal.device_info", false]], "diag (c++ function)": [[0, "_CPPv44diagRK5arrayi14StreamOrDevice", false]], "diag() (array method)": [[46, "mlx.core.array.diag", false]], "diag() (in module mlx.core)": [[118, "mlx.core.diag", false]], "diagonal (c++ function)": [[0, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", false]], "diagonal() (array method)": [[47, "mlx.core.array.diagonal", false]], "diagonal() (in module mlx.core)": [[119, "mlx.core.diagonal", false]], "disable_compile() (in module mlx.core)": [[120, "mlx.core.disable_compile", false]], "divide (c++ function)": [[0, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", false]], "divide() (in module mlx.core)": [[129, "mlx.core.divide", false]], "divmod (c++ function)": [[0, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", false]], "divmod() (in module mlx.core)": [[130, "mlx.core.divmod", false]], "dropout (class in mlx.nn)": [[347, "mlx.nn.Dropout", false]], "dropout2d (class in mlx.nn)": [[348, "mlx.nn.Dropout2d", false]], "dropout3d (class in mlx.nn)": [[349, "mlx.nn.Dropout3d", false]], "dtype (array property)": [[48, "mlx.core.array.dtype", false]], "dtype (class in mlx.core)": [[11, "mlx.core.Dtype", false]], "dtypecategory (class in mlx.core)": [[12, "mlx.core.DtypeCategory", false]], "eigh() (in module mlx.core.linalg)": [[192, "mlx.core.linalg.eigh", false]], "eigvalsh() (in module mlx.core.linalg)": [[193, "mlx.core.linalg.eigvalsh", false]], "einsum() (in module mlx.core)": [[131, "mlx.core.einsum", false]], "einsum_path() (in module mlx.core)": [[132, "mlx.core.einsum_path", false]], "elu (class in mlx.nn)": [[350, "mlx.nn.ELU", false], [422, "mlx.nn.elu", false]], "embedding (class in mlx.nn)": [[351, "mlx.nn.Embedding", false]], "enable_compile() (in module mlx.core)": [[133, "mlx.core.enable_compile", false]], "equal (c++ function)": [[0, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", false]], "equal() (in module mlx.core)": [[134, "mlx.core.equal", false]], "erf (c++ function)": [[0, "_CPPv43erfRK5array14StreamOrDevice", false]], "erf() (in module mlx.core)": [[135, "mlx.core.erf", false]], "erfinv (c++ function)": [[0, "_CPPv46erfinvRK5array14StreamOrDevice", false]], "erfinv() (in module mlx.core)": [[136, "mlx.core.erfinv", false]], "eval() (in module mlx.core)": [[137, "mlx.core.eval", false]], "eval() (module method)": [[373, "mlx.nn.Module.eval", false]], "exp (c++ function)": [[0, "_CPPv43expRK5array14StreamOrDevice", false]], "exp() (array method)": [[49, "mlx.core.array.exp", false]], "exp() (in module mlx.core)": [[138, "mlx.core.exp", false]], "expand_dims (c++ function)": [[0, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", false]], "expand_dims() (in module mlx.core)": [[139, "mlx.core.expand_dims", false]], "expm1 (c++ function)": [[0, "_CPPv45expm1RK5array14StreamOrDevice", false]], "expm1() (in module mlx.core)": [[140, "mlx.core.expm1", false]], "exponential_decay() (in module mlx.optimizers)": [[481, "mlx.optimizers.exponential_decay", false]], "export_function() (in module mlx.core)": [[141, "mlx.core.export_function", false]], "export_to_dot() (in module mlx.core)": [[142, "mlx.core.export_to_dot", false]], "exporter() (in module mlx.core)": [[143, "mlx.core.exporter", false]], "eye (c++ function)": [[0, "_CPPv43eyei14StreamOrDevice", false], [0, "_CPPv43eyei5Dtype14StreamOrDevice", false], [0, "_CPPv43eyeii14StreamOrDevice", false], [0, "_CPPv43eyeiii14StreamOrDevice", false], [0, "_CPPv43eyeiii5Dtype14StreamOrDevice", false]], "eye() (in module mlx.core)": [[144, "mlx.core.eye", false]], "fft() (in module mlx.core.fft)": [[150, "mlx.core.fft.fft", false]], "fft2() (in module mlx.core.fft)": [[151, "mlx.core.fft.fft2", false]], "fftn() (in module mlx.core.fft)": [[152, "mlx.core.fft.fftn", false]], "filter_and_map() (module method)": [[374, "mlx.nn.Module.filter_and_map", false]], "finfo (class in mlx.core)": [[162, "mlx.core.finfo", false]], "flatten (c++ function)": [[0, "_CPPv47flattenRK5array14StreamOrDevice", false], [0, "_CPPv47flattenRK5arrayii14StreamOrDevice", false]], "flatten() (array method)": [[50, "mlx.core.array.flatten", false]], "flatten() (in module mlx.core)": [[163, "mlx.core.flatten", false]], "floor (c++ function)": [[0, "_CPPv45floorRK5array14StreamOrDevice", false]], "floor() (in module mlx.core)": [[164, "mlx.core.floor", false]], "floor_divide (c++ function)": [[0, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", false]], "floor_divide() (in module mlx.core)": [[165, "mlx.core.floor_divide", false]], "freeze() (module method)": [[375, "mlx.nn.Module.freeze", false]], "full (c++ function)": [[0, "_CPPv44full5Shape5array14StreamOrDevice", false], [0, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", false], [0, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", false], [0, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", false]], "full() (in module mlx.core)": [[166, "mlx.core.full", false]], "gather (c++ function)": [[0, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", false], [0, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", false]], "gather_mm (c++ function)": [[0, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "gather_mm() (in module mlx.core)": [[167, "mlx.core.gather_mm", false]], "gather_qmm (c++ function)": [[0, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", false]], "gather_qmm() (in module mlx.core)": [[168, "mlx.core.gather_qmm", false]], "gaussian_nll_loss (class in mlx.nn.losses)": [[436, "mlx.nn.losses.gaussian_nll_loss", false]], "gelu (class in mlx.nn)": [[352, "mlx.nn.GELU", false], [423, "mlx.nn.gelu", false]], "gelu_approx (class in mlx.nn)": [[424, "mlx.nn.gelu_approx", false]], "gelu_fast_approx (class in mlx.nn)": [[425, "mlx.nn.gelu_fast_approx", false]], "get_active_memory() (in module mlx.core.metal)": [[217, "mlx.core.metal.get_active_memory", false]], "get_cache_memory() (in module mlx.core.metal)": [[218, "mlx.core.metal.get_cache_memory", false]], "get_peak_memory() (in module mlx.core.metal)": [[219, "mlx.core.metal.get_peak_memory", false]], "glorot_normal() (in module mlx.nn.init)": [[414, "mlx.nn.init.glorot_normal", false]], "glorot_uniform() (in module mlx.nn.init)": [[415, "mlx.nn.init.glorot_uniform", false]], "glu (class in mlx.nn)": [[353, "mlx.nn.GLU", false], [426, "mlx.nn.glu", false]], "grad() (in module mlx.core)": [[169, "mlx.core.grad", false]], "greater (c++ function)": [[0, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", false]], "greater() (in module mlx.core)": [[170, "mlx.core.greater", false]], "greater_equal (c++ function)": [[0, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", false]], "greater_equal() (in module mlx.core)": [[171, "mlx.core.greater_equal", false]], "group (class in mlx.core.distributed)": [[121, "mlx.core.distributed.Group", false]], "groupnorm (class in mlx.nn)": [[355, "mlx.nn.GroupNorm", false]], "gru (class in mlx.nn)": [[354, "mlx.nn.GRU", false]], "gumbel() (in module mlx.core.random)": [[248, "mlx.core.random.gumbel", false]], "hadamard_transform (c++ function)": [[0, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", false]], "hadamard_transform() (in module mlx.core)": [[172, "mlx.core.hadamard_transform", false]], "hard_shrink (class in mlx.nn)": [[427, "mlx.nn.hard_shrink", false]], "hard_tanh (class in mlx.nn)": [[428, "mlx.nn.hard_tanh", false]], "hardshrink (class in mlx.nn)": [[356, "mlx.nn.HardShrink", false]], "hardswish (class in mlx.nn)": [[358, "mlx.nn.Hardswish", false], [429, "mlx.nn.hardswish", false]], "hardtanh (class in mlx.nn)": [[357, "mlx.nn.HardTanh", false]], "he_normal() (in module mlx.nn.init)": [[416, "mlx.nn.init.he_normal", false]], "he_uniform() (in module mlx.nn.init)": [[417, "mlx.nn.init.he_uniform", false]], "hinge_loss (class in mlx.nn.losses)": [[437, "mlx.nn.losses.hinge_loss", false]], "huber_loss (class in mlx.nn.losses)": [[438, "mlx.nn.losses.huber_loss", false]], "identity (c++ function)": [[0, "_CPPv48identityi14StreamOrDevice", false], [0, "_CPPv48identityi5Dtype14StreamOrDevice", false]], "identity() (in module mlx.core)": [[173, "mlx.core.identity", false]], "identity() (in module mlx.nn.init)": [[418, "mlx.nn.init.identity", false]], "ifft() (in module mlx.core.fft)": [[153, "mlx.core.fft.ifft", false]], "ifft2() (in module mlx.core.fft)": [[154, "mlx.core.fft.ifft2", false]], "ifftn() (in module mlx.core.fft)": [[155, "mlx.core.fft.ifftn", false]], "imag (c++ function)": [[0, "_CPPv44imagRK5array14StreamOrDevice", false]], "imag() (in module mlx.core)": [[174, "mlx.core.imag", false]], "import_function() (in module mlx.core)": [[175, "mlx.core.import_function", false]], "init() (in module mlx.core.distributed)": [[124, "mlx.core.distributed.init", false]], "init() (optimizer method)": [[475, "mlx.optimizers.Optimizer.init", false]], "inner (c++ function)": [[0, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", false]], "inner() (in module mlx.core)": [[176, "mlx.core.inner", false]], "instancenorm (class in mlx.nn)": [[359, "mlx.nn.InstanceNorm", false]], "inv() (in module mlx.core.linalg)": [[194, "mlx.core.linalg.inv", false]], "irfft() (in module mlx.core.fft)": [[156, "mlx.core.fft.irfft", false]], "irfft2() (in module mlx.core.fft)": [[157, "mlx.core.fft.irfft2", false]], "irfftn() (in module mlx.core.fft)": [[158, "mlx.core.fft.irfftn", false]], "is_available() (in module mlx.core.distributed)": [[125, "mlx.core.distributed.is_available", false]], "is_available() (in module mlx.core.metal)": [[220, "mlx.core.metal.is_available", false]], "isclose (c++ function)": [[0, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "isclose() (in module mlx.core)": [[177, "mlx.core.isclose", false]], "isfinite (c++ function)": [[0, "_CPPv48isfiniteRK5array14StreamOrDevice", false]], "isfinite() (in module mlx.core)": [[178, "mlx.core.isfinite", false]], "isinf (c++ function)": [[0, "_CPPv45isinfRK5array14StreamOrDevice", false]], "isinf() (in module mlx.core)": [[179, "mlx.core.isinf", false]], "isnan (c++ function)": [[0, "_CPPv45isnanRK5array14StreamOrDevice", false]], "isnan() (in module mlx.core)": [[180, "mlx.core.isnan", false]], "isneginf (c++ function)": [[0, "_CPPv48isneginfRK5array14StreamOrDevice", false]], "isneginf() (in module mlx.core)": [[181, "mlx.core.isneginf", false]], "isposinf (c++ function)": [[0, "_CPPv48isposinfRK5array14StreamOrDevice", false]], "isposinf() (in module mlx.core)": [[182, "mlx.core.isposinf", false]], "issubdtype() (in module mlx.core)": [[183, "mlx.core.issubdtype", false]], "item() (array method)": [[51, "mlx.core.array.item", false]], "itemsize (array property)": [[52, "mlx.core.array.itemsize", false]], "join_schedules() (in module mlx.optimizers)": [[482, "mlx.optimizers.join_schedules", false]], "jvp() (in module mlx.core)": [[184, "mlx.core.jvp", false]], "key() (in module mlx.core.random)": [[249, "mlx.core.random.key", false]], "kl_div_loss (class in mlx.nn.losses)": [[439, "mlx.nn.losses.kl_div_loss", false]], "kron (c++ function)": [[0, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", false]], "kron() (in module mlx.core)": [[185, "mlx.core.kron", false]], "l1_loss (class in mlx.nn.losses)": [[440, "mlx.nn.losses.l1_loss", false]], "laplace() (in module mlx.core.random)": [[250, "mlx.core.random.laplace", false]], "layer_norm() (in module mlx.core.fast)": [[145, "mlx.core.fast.layer_norm", false]], "layernorm (class in mlx.nn)": [[361, "mlx.nn.LayerNorm", false]], "leaf_modules() (module method)": [[376, "mlx.nn.Module.leaf_modules", false]], "leaky_relu (class in mlx.nn)": [[430, "mlx.nn.leaky_relu", false]], "leakyrelu (class in mlx.nn)": [[362, "mlx.nn.LeakyReLU", false]], "left_shift (c++ function)": [[0, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", false]], "left_shift() (in module mlx.core)": [[186, "mlx.core.left_shift", false]], "less (c++ function)": [[0, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", false]], "less() (in module mlx.core)": [[187, "mlx.core.less", false]], "less_equal (c++ function)": [[0, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", false]], "less_equal() (in module mlx.core)": [[188, "mlx.core.less_equal", false]], "linear (class in mlx.nn)": [[363, "mlx.nn.Linear", false]], "linear_schedule() (in module mlx.optimizers)": [[483, "mlx.optimizers.linear_schedule", false]], "linspace (c++ function)": [[0, "_CPPv48linspaceddi5Dtype14StreamOrDevice", false]], "linspace() (in module mlx.core)": [[199, "mlx.core.linspace", false]], "lion (class in mlx.optimizers)": [[473, "mlx.optimizers.Lion", false]], "load() (in module mlx.core)": [[200, "mlx.core.load", false]], "load_weights() (module method)": [[377, "mlx.nn.Module.load_weights", false]], "log (c++ function)": [[0, "_CPPv43logRK5array14StreamOrDevice", false]], "log() (array method)": [[53, "mlx.core.array.log", false]], "log() (in module mlx.core)": [[201, "mlx.core.log", false]], "log10 (c++ function)": [[0, "_CPPv45log10RK5array14StreamOrDevice", false]], "log10() (array method)": [[54, "mlx.core.array.log10", false]], "log10() (in module mlx.core)": [[202, "mlx.core.log10", false]], "log1p (c++ function)": [[0, "_CPPv45log1pRK5array14StreamOrDevice", false]], "log1p() (array method)": [[55, "mlx.core.array.log1p", false]], "log1p() (in module mlx.core)": [[203, "mlx.core.log1p", false]], "log2 (c++ function)": [[0, "_CPPv44log2RK5array14StreamOrDevice", false]], "log2() (array method)": [[56, "mlx.core.array.log2", false]], "log2() (in module mlx.core)": [[204, "mlx.core.log2", false]], "log_cosh_loss (class in mlx.nn.losses)": [[441, "mlx.nn.losses.log_cosh_loss", false]], "log_sigmoid (class in mlx.nn)": [[431, "mlx.nn.log_sigmoid", false]], "log_softmax (class in mlx.nn)": [[432, "mlx.nn.log_softmax", false]], "logaddexp (c++ function)": [[0, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", false]], "logaddexp() (in module mlx.core)": [[205, "mlx.core.logaddexp", false]], "logical_and (c++ function)": [[0, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", false]], "logical_and() (in module mlx.core)": [[206, "mlx.core.logical_and", false]], "logical_not (c++ function)": [[0, "_CPPv411logical_notRK5array14StreamOrDevice", false]], "logical_not() (in module mlx.core)": [[207, "mlx.core.logical_not", false]], "logical_or (c++ function)": [[0, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", false]], "logical_or() (in module mlx.core)": [[208, "mlx.core.logical_or", false]], "logsigmoid (class in mlx.nn)": [[364, "mlx.nn.LogSigmoid", false]], "logsoftmax (class in mlx.nn)": [[365, "mlx.nn.LogSoftmax", false]], "logsumexp (c++ function)": [[0, "_CPPv49logsumexpRK5array14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", false]], "logsumexp() (array method)": [[57, "mlx.core.array.logsumexp", false]], "logsumexp() (in module mlx.core)": [[209, "mlx.core.logsumexp", false]], "lstm (class in mlx.nn)": [[360, "mlx.nn.LSTM", false]], "margin_ranking_loss (class in mlx.nn.losses)": [[442, "mlx.nn.losses.margin_ranking_loss", false]], "matmul (c++ function)": [[0, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", false]], "matmul() (in module mlx.core)": [[210, "mlx.core.matmul", false]], "max (c++ function)": [[0, "_CPPv43maxRK5array14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayib14StreamOrDevice", false]], "max() (array method)": [[58, "mlx.core.array.max", false]], "max() (in module mlx.core)": [[211, "mlx.core.max", false]], "maximum (c++ function)": [[0, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", false]], "maximum() (in module mlx.core)": [[212, "mlx.core.maximum", false]], "maxpool1d (class in mlx.nn)": [[366, "mlx.nn.MaxPool1d", false]], "maxpool2d (class in mlx.nn)": [[367, "mlx.nn.MaxPool2d", false]], "maxpool3d (class in mlx.nn)": [[368, "mlx.nn.MaxPool3d", false]], "mean (c++ function)": [[0, "_CPPv44meanRK5array14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayib14StreamOrDevice", false]], "mean() (array method)": [[59, "mlx.core.array.mean", false]], "mean() (in module mlx.core)": [[213, "mlx.core.mean", false]], "meshgrid (c++ function)": [[0, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", false]], "meshgrid() (in module mlx.core)": [[214, "mlx.core.meshgrid", false]], "metal_kernel() (in module mlx.core.fast)": [[146, "mlx.core.fast.metal_kernel", false]], "min (c++ function)": [[0, "_CPPv43minRK5array14StreamOrDevice", false], [0, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayib14StreamOrDevice", false]], "min() (array method)": [[60, "mlx.core.array.min", false]], "min() (in module mlx.core)": [[227, "mlx.core.min", false]], "minimum (c++ function)": [[0, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", false]], "minimum() (in module mlx.core)": [[228, "mlx.core.minimum", false]], "mish (class in mlx.nn)": [[369, "mlx.nn.Mish", false], [447, "mlx.nn.mish", false]], "module (class in mlx.nn)": [[464, "mlx.nn.Module", false]], "modules() (module method)": [[378, "mlx.nn.Module.modules", false]], "moveaxis (c++ function)": [[0, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", false]], "moveaxis() (array method)": [[61, "mlx.core.array.moveaxis", false]], "moveaxis() (in module mlx.core)": [[229, "mlx.core.moveaxis", false]], "mse_loss (class in mlx.nn.losses)": [[443, "mlx.nn.losses.mse_loss", false]], "multiheadattention (class in mlx.nn)": [[390, "mlx.nn.MultiHeadAttention", false]], "multiply (c++ function)": [[0, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", false]], "multiply() (in module mlx.core)": [[230, "mlx.core.multiply", false]], "multivariate_normal() (in module mlx.core.random)": [[251, "mlx.core.random.multivariate_normal", false]], "named_modules() (module method)": [[379, "mlx.nn.Module.named_modules", false]], "nan_to_num (c++ function)": [[0, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", false]], "nan_to_num() (in module mlx.core)": [[231, "mlx.core.nan_to_num", false]], "nbytes (array property)": [[62, "mlx.core.array.nbytes", false]], "ndim (array property)": [[63, "mlx.core.array.ndim", false]], "negative (c++ function)": [[0, "_CPPv48negativeRK5array14StreamOrDevice", false]], "negative() (in module mlx.core)": [[232, "mlx.core.negative", false]], "new_stream() (in module mlx.core)": [[233, "mlx.core.new_stream", false]], "nll_loss (class in mlx.nn.losses)": [[444, "mlx.nn.losses.nll_loss", false]], "norm() (in module mlx.core.linalg)": [[195, "mlx.core.linalg.norm", false]], "normal() (in module mlx.core.random)": [[252, "mlx.core.random.normal", false]], "normal() (in module mlx.nn.init)": [[419, "mlx.nn.init.normal", false]], "not_equal (c++ function)": [[0, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", false]], "not_equal() (in module mlx.core)": [[234, "mlx.core.not_equal", false]], "number_of_elements (c++ function)": [[0, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", false]], "ones (c++ function)": [[0, "_CPPv44onesRK5Shape14StreamOrDevice", false], [0, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", false]], "ones() (in module mlx.core)": [[235, "mlx.core.ones", false]], "ones_like (c++ function)": [[0, "_CPPv49ones_likeRK5array14StreamOrDevice", false]], "ones_like() (in module mlx.core)": [[236, "mlx.core.ones_like", false]], "operator!= (c++ function)": [[0, "_CPPv4I0Ene5array1TRK5array", false], [0, "_CPPv4I0Ene5arrayRK5array1T", false], [0, "_CPPv4neRK5arrayRK5array", false]], "operator% (c++ function)": [[0, "_CPPv4I0Erm5array1TRK5array", false], [0, "_CPPv4I0Erm5arrayRK5array1T", false], [0, "_CPPv4rmRK5arrayRK5array", false]], "operator& (c++ function)": [[0, "_CPPv4anRK5arrayRK5array", false]], "operator&& (c++ function)": [[0, "_CPPv4aaRK5arrayRK5array", false]], "operator* (c++ function)": [[0, "_CPPv4I0Eml5array1TRK5array", false], [0, "_CPPv4I0Eml5arrayRK5array1T", false], [0, "_CPPv4mlRK5arrayRK5array", false]], "operator+ (c++ function)": [[0, "_CPPv4I0Epl5array1TRK5array", false], [0, "_CPPv4I0Epl5arrayRK5array1T", false], [0, "_CPPv4plRK5arrayRK5array", false]], "operator- (c++ function)": [[0, "_CPPv4I0Emi5array1TRK5array", false], [0, "_CPPv4I0Emi5arrayRK5array1T", false], [0, "_CPPv4miRK5array", false], [0, "_CPPv4miRK5arrayRK5array", false]], "operator/ (c++ function)": [[0, "_CPPv4dvRK5arrayRK5array", false], [0, "_CPPv4dvRK5arrayd", false], [0, "_CPPv4dvdRK5array", false]], "operator< (c++ function)": [[0, "_CPPv4I0Elt5array1TRK5array", false], [0, "_CPPv4I0Elt5arrayRK5array1T", false], [0, "_CPPv4ltRK5arrayRK5array", false]], "operator<< (c++ function)": [[0, "_CPPv4lsRK5arrayRK5array", false]], "operator<= (c++ function)": [[0, "_CPPv4I0Ele5array1TRK5array", false], [0, "_CPPv4I0Ele5arrayRK5array1T", false], [0, "_CPPv4leRK5arrayRK5array", false]], "operator== (c++ function)": [[0, "_CPPv4I0Eeq5array1TRK5array", false], [0, "_CPPv4I0Eeq5arrayRK5array1T", false], [0, "_CPPv4eqRK5arrayRK5array", false]], "operator> (c++ function)": [[0, "_CPPv4I0Egt5array1TRK5array", false], [0, "_CPPv4I0Egt5arrayRK5array1T", false], [0, "_CPPv4gtRK5arrayRK5array", false]], "operator>= (c++ function)": [[0, "_CPPv4I0Ege5array1TRK5array", false], [0, "_CPPv4I0Ege5arrayRK5array1T", false], [0, "_CPPv4geRK5arrayRK5array", false]], "operator>> (c++ function)": [[0, "_CPPv4rsRK5arrayRK5array", false]], "operator^ (c++ function)": [[0, "_CPPv4eoRK5arrayRK5array", false]], "operator| (c++ function)": [[0, "_CPPv4orRK5arrayRK5array", false]], "operator|| (c++ function)": [[0, "_CPPv4ooRK5arrayRK5array", false]], "optimizer (class in mlx.optimizers)": [[486, "mlx.optimizers.Optimizer", false]], "outer (c++ function)": [[0, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", false]], "outer() (in module mlx.core)": [[237, "mlx.core.outer", false]], "pad (c++ function)": [[0, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", false]], "pad() (in module mlx.core)": [[238, "mlx.core.pad", false]], "parameters() (module method)": [[380, "mlx.nn.Module.parameters", false]], "partition (c++ function)": [[0, "_CPPv49partitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv49partitionRK5arrayii14StreamOrDevice", false]], "partition() (in module mlx.core)": [[239, "mlx.core.partition", false]], "permutation() (in module mlx.core.random)": [[253, "mlx.core.random.permutation", false]], "power (c++ function)": [[0, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", false]], "power() (in module mlx.core)": [[240, "mlx.core.power", false]], "prelu (class in mlx.nn)": [[391, "mlx.nn.PReLU", false], [448, "mlx.nn.prelu", false]], "prod (c++ function)": [[0, "_CPPv44prodRK5array14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayib14StreamOrDevice", false]], "prod() (array method)": [[64, "mlx.core.array.prod", false]], "prod() (in module mlx.core)": [[241, "mlx.core.prod", false]], "put_along_axis (c++ function)": [[0, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false]], "put_along_axis() (in module mlx.core)": [[242, "mlx.core.put_along_axis", false]], "qr() (in module mlx.core.linalg)": [[196, "mlx.core.linalg.qr", false]], "quantize (c++ function)": [[0, "_CPPv48quantizeRK5arrayii14StreamOrDevice", false]], "quantize() (in module mlx.core)": [[243, "mlx.core.quantize", false]], "quantize() (in module mlx.nn)": [[316, "mlx.nn.quantize", false]], "quantized_matmul (c++ function)": [[0, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", false]], "quantized_matmul() (in module mlx.core)": [[244, "mlx.core.quantized_matmul", false]], "quantizedembedding (class in mlx.nn)": [[392, "mlx.nn.QuantizedEmbedding", false]], "quantizedlinear (class in mlx.nn)": [[393, "mlx.nn.QuantizedLinear", false]], "radians (c++ function)": [[0, "_CPPv47radiansRK5array14StreamOrDevice", false]], "radians() (in module mlx.core)": [[245, "mlx.core.radians", false]], "randint() (in module mlx.core.random)": [[254, "mlx.core.random.randint", false]], "real (c++ function)": [[0, "_CPPv44realRK5array14StreamOrDevice", false]], "real() (in module mlx.core)": [[259, "mlx.core.real", false]], "reciprocal (c++ function)": [[0, "_CPPv410reciprocalRK5array14StreamOrDevice", false]], "reciprocal() (array method)": [[65, "mlx.core.array.reciprocal", false]], "reciprocal() (in module mlx.core)": [[260, "mlx.core.reciprocal", false]], "recv() (in module mlx.core.distributed)": [[126, "mlx.core.distributed.recv", false]], "recv_like() (in module mlx.core.distributed)": [[127, "mlx.core.distributed.recv_like", false]], "relu (class in mlx.nn)": [[396, "mlx.nn.ReLU", false], [449, "mlx.nn.relu", false]], "relu6 (class in mlx.nn)": [[397, "mlx.nn.ReLU6", false], [450, "mlx.nn.relu6", false]], "remainder (c++ function)": [[0, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", false]], "remainder() (in module mlx.core)": [[261, "mlx.core.remainder", false]], "repeat (c++ function)": [[0, "_CPPv46repeatRK5arrayi14StreamOrDevice", false], [0, "_CPPv46repeatRK5arrayii14StreamOrDevice", false]], "repeat() (in module mlx.core)": [[262, "mlx.core.repeat", false]], "reset_peak_memory() (in module mlx.core.metal)": [[221, "mlx.core.metal.reset_peak_memory", false]], "reshape (c++ function)": [[0, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", false]], "reshape() (array method)": [[66, "mlx.core.array.reshape", false]], "reshape() (in module mlx.core)": [[263, "mlx.core.reshape", false]], "rfft() (in module mlx.core.fft)": [[159, "mlx.core.fft.rfft", false]], "rfft2() (in module mlx.core.fft)": [[160, "mlx.core.fft.rfft2", false]], "rfftn() (in module mlx.core.fft)": [[161, "mlx.core.fft.rfftn", false]], "right_shift (c++ function)": [[0, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", false]], "right_shift() (in module mlx.core)": [[264, "mlx.core.right_shift", false]], "rms_norm() (in module mlx.core.fast)": [[147, "mlx.core.fast.rms_norm", false]], "rmsnorm (class in mlx.nn)": [[394, "mlx.nn.RMSNorm", false]], "rmsprop (class in mlx.optimizers)": [[478, "mlx.optimizers.RMSprop", false]], "rnn (class in mlx.nn)": [[395, "mlx.nn.RNN", false]], "roll (c++ function)": [[0, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayi14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayii14StreamOrDevice", false]], "roll() (in module mlx.core)": [[265, "mlx.core.roll", false]], "rope (class in mlx.nn)": [[398, "mlx.nn.RoPE", false]], "rope() (in module mlx.core.fast)": [[148, "mlx.core.fast.rope", false]], "round (c++ function)": [[0, "_CPPv45roundRK5array14StreamOrDevice", false], [0, "_CPPv45roundRK5arrayi14StreamOrDevice", false]], "round() (array method)": [[67, "mlx.core.array.round", false]], "round() (in module mlx.core)": [[266, "mlx.core.round", false]], "rsqrt (c++ function)": [[0, "_CPPv45rsqrtRK5array14StreamOrDevice", false]], "rsqrt() (array method)": [[68, "mlx.core.array.rsqrt", false]], "rsqrt() (in module mlx.core)": [[267, "mlx.core.rsqrt", false]], "save() (in module mlx.core)": [[268, "mlx.core.save", false]], "save_gguf() (in module mlx.core)": [[269, "mlx.core.save_gguf", false]], "save_safetensors() (in module mlx.core)": [[270, "mlx.core.save_safetensors", false]], "save_weights() (module method)": [[381, "mlx.nn.Module.save_weights", false]], "savez() (in module mlx.core)": [[271, "mlx.core.savez", false]], "savez_compressed() (in module mlx.core)": [[272, "mlx.core.savez_compressed", false]], "scaled_dot_product_attention() (in module mlx.core.fast)": [[149, "mlx.core.fast.scaled_dot_product_attention", false]], "scatter (c++ function)": [[0, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_add (c++ function)": [[0, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_max (c++ function)": [[0, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_min (c++ function)": [[0, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_prod (c++ function)": [[0, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "seed() (in module mlx.core.random)": [[255, "mlx.core.random.seed", false]], "selu (class in mlx.nn)": [[399, "mlx.nn.SELU", false], [451, "mlx.nn.selu", false]], "send() (in module mlx.core.distributed)": [[128, "mlx.core.distributed.send", false]], "sequential (class in mlx.nn)": [[400, "mlx.nn.Sequential", false]], "set_cache_limit() (in module mlx.core.metal)": [[222, "mlx.core.metal.set_cache_limit", false]], "set_default_device() (in module mlx.core)": [[273, "mlx.core.set_default_device", false]], "set_default_stream() (in module mlx.core)": [[274, "mlx.core.set_default_stream", false]], "set_dtype() (module method)": [[382, "mlx.nn.Module.set_dtype", false]], "set_memory_limit() (in module mlx.core.metal)": [[223, "mlx.core.metal.set_memory_limit", false]], "set_wired_limit() (in module mlx.core.metal)": [[224, "mlx.core.metal.set_wired_limit", false]], "sgd (class in mlx.optimizers)": [[479, "mlx.optimizers.SGD", false]], "shape (array property)": [[69, "mlx.core.array.shape", false]], "sigmoid (c++ function)": [[0, "_CPPv47sigmoidRK5array14StreamOrDevice", false]], "sigmoid (class in mlx.nn)": [[402, "mlx.nn.Sigmoid", false], [452, "mlx.nn.sigmoid", false]], "sigmoid() (in module mlx.core)": [[275, "mlx.core.sigmoid", false]], "sign (c++ function)": [[0, "_CPPv44signRK5array14StreamOrDevice", false]], "sign() (in module mlx.core)": [[276, "mlx.core.sign", false]], "silu (class in mlx.nn)": [[401, "mlx.nn.SiLU", false], [453, "mlx.nn.silu", false]], "sin (c++ function)": [[0, "_CPPv43sinRK5array14StreamOrDevice", false]], "sin() (array method)": [[70, "mlx.core.array.sin", false]], "sin() (in module mlx.core)": [[277, "mlx.core.sin", false]], "sinh (c++ function)": [[0, "_CPPv44sinhRK5array14StreamOrDevice", false]], "sinh() (in module mlx.core)": [[278, "mlx.core.sinh", false]], "sinusoidalpositionalencoding (class in mlx.nn)": [[403, "mlx.nn.SinusoidalPositionalEncoding", false]], "size (array property)": [[71, "mlx.core.array.size", false]], "slice (c++ function)": [[0, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", false]], "slice() (in module mlx.core)": [[279, "mlx.core.slice", false]], "slice_update (c++ function)": [[0, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", false], [0, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", false], [0, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "slice_update() (in module mlx.core)": [[280, "mlx.core.slice_update", false]], "smooth_l1_loss (class in mlx.nn.losses)": [[445, "mlx.nn.losses.smooth_l1_loss", false]], "softmax (c++ function)": [[0, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayib14StreamOrDevice", false]], "softmax (class in mlx.nn)": [[404, "mlx.nn.Softmax", false], [454, "mlx.nn.softmax", false]], "softmax() (in module mlx.core)": [[281, "mlx.core.softmax", false]], "softmin (class in mlx.nn)": [[405, "mlx.nn.Softmin", false], [455, "mlx.nn.softmin", false]], "softplus (class in mlx.nn)": [[406, "mlx.nn.Softplus", false], [456, "mlx.nn.softplus", false]], "softshrink (class in mlx.nn)": [[407, "mlx.nn.Softshrink", false], [457, "mlx.nn.softshrink", false]], "softsign (class in mlx.nn)": [[408, "mlx.nn.Softsign", false]], "sort (c++ function)": [[0, "_CPPv44sortRK5array14StreamOrDevice", false], [0, "_CPPv44sortRK5arrayi14StreamOrDevice", false]], "sort() (in module mlx.core)": [[282, "mlx.core.sort", false]], "split (c++ function)": [[0, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayi14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayii14StreamOrDevice", false]], "split() (array method)": [[72, "mlx.core.array.split", false]], "split() (in module mlx.core)": [[283, "mlx.core.split", false]], "split() (in module mlx.core.random)": [[256, "mlx.core.random.split", false]], "sqrt (c++ function)": [[0, "_CPPv44sqrtRK5array14StreamOrDevice", false]], "sqrt() (array method)": [[73, "mlx.core.array.sqrt", false]], "sqrt() (in module mlx.core)": [[284, "mlx.core.sqrt", false]], "square (c++ function)": [[0, "_CPPv46squareRK5array14StreamOrDevice", false]], "square() (array method)": [[74, "mlx.core.array.square", false]], "square() (in module mlx.core)": [[285, "mlx.core.square", false]], "squeeze (c++ function)": [[0, "_CPPv47squeezeRK5array14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayi14StreamOrDevice", false]], "squeeze() (array method)": [[75, "mlx.core.array.squeeze", false]], "squeeze() (in module mlx.core)": [[286, "mlx.core.squeeze", false]], "stack (c++ function)": [[0, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", false]], "stack() (in module mlx.core)": [[287, "mlx.core.stack", false]], "start_capture() (in module mlx.core.metal)": [[225, "mlx.core.metal.start_capture", false]], "state (module property)": [[383, "mlx.nn.Module.state", false]], "state (optimizer property)": [[476, "mlx.optimizers.Optimizer.state", false]], "std (c++ function)": [[0, "_CPPv4StRK5array14StreamOrDevice", false], [0, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv4StRK5arraybi14StreamOrDevice", false], [0, "_CPPv4StRK5arrayibi14StreamOrDevice", false]], "std() (array method)": [[76, "mlx.core.array.std", false]], "std() (in module mlx.core)": [[288, "mlx.core.std", false]], "step (class in mlx.nn)": [[409, "mlx.nn.Step", false], [458, "mlx.nn.step", false]], "step_decay() (in module mlx.optimizers)": [[484, "mlx.optimizers.step_decay", false]], "stop_capture() (in module mlx.core.metal)": [[226, "mlx.core.metal.stop_capture", false]], "stop_gradient (c++ function)": [[0, "_CPPv413stop_gradientRK5array14StreamOrDevice", false]], "stop_gradient() (in module mlx.core)": [[289, "mlx.core.stop_gradient", false]], "stream (class in mlx.core)": [[324, "mlx.core.Stream", false]], "stream() (in module mlx.core)": [[290, "mlx.core.stream", false]], "subtract (c++ function)": [[0, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", false]], "subtract() (in module mlx.core)": [[291, "mlx.core.subtract", false]], "sum (c++ function)": [[0, "_CPPv43sumRK5array14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayib14StreamOrDevice", false]], "sum() (array method)": [[77, "mlx.core.array.sum", false]], "sum() (in module mlx.core)": [[292, "mlx.core.sum", false]], "svd() (in module mlx.core.linalg)": [[197, "mlx.core.linalg.svd", false]], "swapaxes (c++ function)": [[0, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", false]], "swapaxes() (array method)": [[78, "mlx.core.array.swapaxes", false]], "swapaxes() (in module mlx.core)": [[293, "mlx.core.swapaxes", false]], "synchronize() (in module mlx.core)": [[294, "mlx.core.synchronize", false]], "t (array property)": [[32, "mlx.core.array.T", false]], "take (c++ function)": [[0, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayi14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayii14StreamOrDevice", false]], "take() (in module mlx.core)": [[295, "mlx.core.take", false]], "take_along_axis (c++ function)": [[0, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", false]], "take_along_axis() (in module mlx.core)": [[296, "mlx.core.take_along_axis", false]], "tan (c++ function)": [[0, "_CPPv43tanRK5array14StreamOrDevice", false]], "tan() (in module mlx.core)": [[297, "mlx.core.tan", false]], "tanh (c++ function)": [[0, "_CPPv44tanhRK5array14StreamOrDevice", false]], "tanh (class in mlx.nn)": [[410, "mlx.nn.Tanh", false], [459, "mlx.nn.tanh", false]], "tanh() (in module mlx.core)": [[298, "mlx.core.tanh", false]], "tensordot (c++ function)": [[0, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", false], [0, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", false]], "tensordot() (in module mlx.core)": [[299, "mlx.core.tensordot", false]], "tile (c++ function)": [[0, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "tile() (in module mlx.core)": [[300, "mlx.core.tile", false]], "tolist() (array method)": [[79, "mlx.core.array.tolist", false]], "topk (c++ function)": [[0, "_CPPv44topkRK5arrayi14StreamOrDevice", false], [0, "_CPPv44topkRK5arrayii14StreamOrDevice", false]], "topk() (in module mlx.core)": [[301, "mlx.core.topk", false]], "trace (c++ function)": [[0, "_CPPv45traceRK5array14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", false]], "trace() (in module mlx.core)": [[302, "mlx.core.trace", false]], "train() (module method)": [[384, "mlx.nn.Module.train", false]], "trainable_parameters() (module method)": [[385, "mlx.nn.Module.trainable_parameters", false]], "training (module property)": [[386, "mlx.nn.Module.training", false]], "transformer (class in mlx.nn)": [[411, "mlx.nn.Transformer", false]], "transpose (c++ function)": [[0, "_CPPv49transposeRK5array14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "transpose() (array method)": [[80, "mlx.core.array.transpose", false]], "transpose() (in module mlx.core)": [[303, "mlx.core.transpose", false]], "tree_flatten() (in module mlx.utils)": [[319, "mlx.utils.tree_flatten", false]], "tree_map() (in module mlx.utils)": [[320, "mlx.utils.tree_map", false]], "tree_map_with_path() (in module mlx.utils)": [[321, "mlx.utils.tree_map_with_path", false]], "tree_reduce() (in module mlx.utils)": [[322, "mlx.utils.tree_reduce", false]], "tree_unflatten() (in module mlx.utils)": [[323, "mlx.utils.tree_unflatten", false]], "tri (c++ function)": [[0, "_CPPv43trii5Dtype14StreamOrDevice", false], [0, "_CPPv43triiii5Dtype14StreamOrDevice", false]], "tri() (in module mlx.core)": [[304, "mlx.core.tri", false]], "tri_inv() (in module mlx.core.linalg)": [[198, "mlx.core.linalg.tri_inv", false]], "tril (c++ function)": [[0, "_CPPv44tril5arrayi14StreamOrDevice", false]], "tril() (in module mlx.core)": [[305, "mlx.core.tril", false]], "triplet_loss (class in mlx.nn.losses)": [[446, "mlx.nn.losses.triplet_loss", false]], "triu (c++ function)": [[0, "_CPPv44triu5arrayi14StreamOrDevice", false]], "triu() (in module mlx.core)": [[306, "mlx.core.triu", false]], "truncated_normal() (in module mlx.core.random)": [[257, "mlx.core.random.truncated_normal", false]], "unflatten (c++ function)": [[0, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", false]], "unflatten() (in module mlx.core)": [[307, "mlx.core.unflatten", false]], "unfreeze() (module method)": [[387, "mlx.nn.Module.unfreeze", false]], "uniform() (in module mlx.core.random)": [[258, "mlx.core.random.uniform", false]], "uniform() (in module mlx.nn.init)": [[420, "mlx.nn.init.uniform", false]], "update() (module method)": [[388, "mlx.nn.Module.update", false]], "update() (optimizer method)": [[477, "mlx.optimizers.Optimizer.update", false]], "update_modules() (module method)": [[389, "mlx.nn.Module.update_modules", false]], "upsample (class in mlx.nn)": [[412, "mlx.nn.Upsample", false]], "value_and_grad() (in module mlx.core)": [[308, "mlx.core.value_and_grad", false]], "value_and_grad() (in module mlx.nn)": [[317, "mlx.nn.value_and_grad", false]], "var (c++ function)": [[0, "_CPPv43varRK5array14StreamOrDevice", false], [0, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv43varRK5arraybi14StreamOrDevice", false], [0, "_CPPv43varRK5arrayibi14StreamOrDevice", false]], "var() (array method)": [[81, "mlx.core.array.var", false]], "var() (in module mlx.core)": [[309, "mlx.core.var", false]], "view (c++ function)": [[0, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", false]], "view() (array method)": [[82, "mlx.core.array.view", false]], "view() (in module mlx.core)": [[310, "mlx.core.view", false]], "vjp() (in module mlx.core)": [[311, "mlx.core.vjp", false]], "vmap() (in module mlx.core)": [[312, "mlx.core.vmap", false]], "where (c++ function)": [[0, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", false]], "where() (in module mlx.core)": [[313, "mlx.core.where", false]], "zeros (c++ function)": [[0, "_CPPv45zerosRK5Shape14StreamOrDevice", false], [0, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", false]], "zeros() (in module mlx.core)": [[314, "mlx.core.zeros", false]], "zeros_like (c++ function)": [[0, "_CPPv410zeros_likeRK5array14StreamOrDevice", false]], "zeros_like() (in module mlx.core)": [[315, "mlx.core.zeros_like", false]]}, "objects": {"": [[0, 0, 1, "_CPPv43absRK5array14StreamOrDevice", "abs"], [0, 1, 1, "_CPPv43absRK5array14StreamOrDevice", "abs::a"], [0, 1, 1, "_CPPv43absRK5array14StreamOrDevice", "abs::s"], [0, 0, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::a"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::b"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::s"], [0, 0, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::a"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::alpha"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::b"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::beta"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::c"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::s"], [0, 0, 1, "_CPPv43allRK5array14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all"], [0, 1, 1, "_CPPv43allRK5array14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::axes"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::axis"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5array14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::s"], [0, 0, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::a"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::atol"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::b"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::equal_nan"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::rtol"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::s"], [0, 0, 1, "_CPPv43anyRK5array14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any"], [0, 1, 1, "_CPPv43anyRK5array14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::axes"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::axis"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5array14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::s"], [0, 0, 1, "_CPPv46aranged14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangedd14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeddd14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangei14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeii14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeiii14StreamOrDevice", "arange"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46aranged14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangei14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46aranged14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangei14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::stop"], [0, 0, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos"], [0, 1, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos::a"], [0, 1, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos::s"], [0, 0, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh"], [0, 1, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh::a"], [0, 1, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh::s"], [0, 0, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin"], [0, 1, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin::a"], [0, 1, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin::s"], [0, 0, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh"], [0, 1, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh::a"], [0, 1, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh::s"], [0, 0, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan"], [0, 0, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::a"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::b"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::s"], [0, 1, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan::a"], [0, 1, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan::s"], [0, 0, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh"], [0, 1, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh::a"], [0, 1, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh::s"], [0, 0, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax"], [0, 0, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax"], [0, 0, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax"], [0, 1, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::axis"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::keepdims"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::keepdims"], [0, 1, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax::s"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::s"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::s"], [0, 0, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin"], [0, 0, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin"], [0, 0, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin"], [0, 1, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::axis"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::keepdims"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::keepdims"], [0, 1, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin::s"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::s"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::s"], [0, 0, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition"], [0, 0, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::a"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::a"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::axis"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::kth"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::kth"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::s"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::s"], [0, 0, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort"], [0, 0, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort"], [0, 1, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort::a"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::a"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::axis"], [0, 1, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort::s"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::s"], [0, 0, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal"], [0, 0, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::a"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::a"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::b"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::b"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::equal_nan"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::s"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::s"], [0, 0, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::a"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::offset"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::s"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::shape"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::strides"], [0, 0, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::a"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::dtype"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::s"], [0, 0, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d"], [0, 0, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d"], [0, 1, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d::a"], [0, 1, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d::a"], [0, 1, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d::s"], [0, 1, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d::s"], [0, 0, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d"], [0, 0, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d"], [0, 1, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d::a"], [0, 1, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d::a"], [0, 1, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d::s"], [0, 1, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d::s"], [0, 0, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d"], [0, 0, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d"], [0, 1, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d::a"], [0, 1, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d::a"], [0, 1, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d::s"], [0, 1, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d::s"], [0, 0, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::a"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::b"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::s"], [0, 0, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::a"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::b"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::s"], [0, 0, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::a"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::b"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::s"], [0, 0, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::a"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::b"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::block_size"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_lhs"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_out"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_rhs"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::s"], [0, 0, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays"], [0, 1, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays::inputs"], [0, 1, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays::s"], [0, 0, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::a"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::s"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::shape"], [0, 0, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil"], [0, 1, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil::a"], [0, 1, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil::s"], [0, 0, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a_max"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a_min"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::s"], [0, 0, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate"], [0, 0, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::axis"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::s"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::s"], [0, 0, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::a"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::s"], [0, 0, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::a"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::allow_col_major"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::s"], [0, 0, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::dilation"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::groups"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::input"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::padding"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::s"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::stride"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::weight"], [0, 0, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::dilation"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::groups"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::input"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::padding"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::s"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::stride"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::weight"], [0, 0, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::dilation"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::groups"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::input"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::padding"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::s"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::stride"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::weight"], [0, 0, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general"], [0, 0, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::flip"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::flip"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::groups"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::groups"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input_dilation"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::kernel_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::kernel_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding_hi"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding_lo"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::s"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::s"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::stride"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::stride"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::weight"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::weight"], [0, 0, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::dilation"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::groups"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::input"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::padding"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::s"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::stride"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::weight"], [0, 0, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::dilation"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::groups"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::input"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::padding"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::s"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::stride"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::weight"], [0, 0, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::dilation"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::groups"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::input"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::padding"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::s"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::stride"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::weight"], [0, 0, 1, "_CPPv44copy5array14StreamOrDevice", "copy"], [0, 1, 1, "_CPPv44copy5array14StreamOrDevice", "copy::a"], [0, 1, 1, "_CPPv44copy5array14StreamOrDevice", "copy::s"], [0, 0, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos"], [0, 1, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos::a"], [0, 1, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos::s"], [0, 0, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh"], [0, 1, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh::a"], [0, 1, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh::s"], [0, 0, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::a"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::axis"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::inclusive"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::reverse"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::s"], [0, 0, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::a"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::axis"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::inclusive"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::reverse"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::s"], [0, 0, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::a"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::axis"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::inclusive"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::reverse"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::s"], [0, 0, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::a"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::axis"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::inclusive"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::reverse"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::s"], [0, 0, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees"], [0, 1, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees::a"], [0, 1, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees::s"], [0, 0, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends"], [0, 1, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends::dependencies"], [0, 1, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends::inputs"], [0, 0, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::biases"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::bits"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::group_size"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::s"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::scales"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::w"], [0, 0, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::a"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::k"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::s"], [0, 0, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::a"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::axis1"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::axis2"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::offset"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::s"], [0, 0, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::a"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::b"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::s"], [0, 0, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::a"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::b"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::s"], [0, 0, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::a"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::b"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::s"], [0, 0, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf"], [0, 1, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf::a"], [0, 1, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf::s"], [0, 0, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv"], [0, 1, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv::a"], [0, 1, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv::s"], [0, 0, 1, "_CPPv43expRK5array14StreamOrDevice", "exp"], [0, 1, 1, "_CPPv43expRK5array14StreamOrDevice", "exp::a"], [0, 1, 1, "_CPPv43expRK5array14StreamOrDevice", "exp::s"], [0, 0, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims"], [0, 0, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::a"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::a"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::axes"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::axis"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::s"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::s"], [0, 0, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1"], [0, 1, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1::a"], [0, 1, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1::s"], [0, 0, 1, "_CPPv43eyei14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeii14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeiii14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::dtype"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::dtype"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::k"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::k"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyei14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyei14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::s"], [0, 0, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten"], [0, 0, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten"], [0, 1, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten::a"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::a"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::end_axis"], [0, 1, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten::s"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::s"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::start_axis"], [0, 0, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor"], [0, 1, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor::a"], [0, 1, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor::s"], [0, 0, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::a"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::b"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::s"], [0, 0, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full"], [0, 0, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full"], [0, 2, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::T"], [0, 2, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::T"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::vals"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::vals"], [0, 0, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather"], [0, 0, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::axes"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::axis"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::slice_sizes"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::slice_sizes"], [0, 0, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::a"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::b"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::lhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::rhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::s"], [0, 0, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::biases"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::bits"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::group_size"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::lhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::rhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::s"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::scales"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::transpose"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::w"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::x"], [0, 0, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::a"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::b"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::s"], [0, 0, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::a"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::b"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::s"], [0, 0, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::a"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::s"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::scale"], [0, 0, 1, "_CPPv48identityi14StreamOrDevice", "identity"], [0, 0, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::dtype"], [0, 1, 1, "_CPPv48identityi14StreamOrDevice", "identity::n"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::n"], [0, 1, 1, "_CPPv48identityi14StreamOrDevice", "identity::s"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::s"], [0, 0, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag"], [0, 1, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag::a"], [0, 1, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag::s"], [0, 0, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::a"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::b"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::s"], [0, 0, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::a"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::atol"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::b"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::equal_nan"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::rtol"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::s"], [0, 0, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite"], [0, 1, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite::a"], [0, 1, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite::s"], [0, 0, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf"], [0, 1, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf::a"], [0, 1, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf::s"], [0, 0, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan"], [0, 1, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan::a"], [0, 1, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan::s"], [0, 0, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf"], [0, 1, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf::a"], [0, 1, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf::s"], [0, 0, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf"], [0, 1, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf::a"], [0, 1, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf::s"], [0, 0, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::a"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::b"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::s"], [0, 0, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::a"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::b"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::s"], [0, 0, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::a"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::b"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::s"], [0, 0, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::a"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::b"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::s"], [0, 0, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::dtype"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::num"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::s"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::start"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::stop"], [0, 0, 1, "_CPPv43logRK5array14StreamOrDevice", "log"], [0, 0, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10"], [0, 1, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10::a"], [0, 1, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10::s"], [0, 0, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p"], [0, 1, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p::a"], [0, 1, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p::s"], [0, 0, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2"], [0, 1, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2::a"], [0, 1, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2::s"], [0, 1, 1, "_CPPv43logRK5array14StreamOrDevice", "log::a"], [0, 1, 1, "_CPPv43logRK5array14StreamOrDevice", "log::s"], [0, 0, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::a"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::b"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::s"], [0, 0, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::a"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::b"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::s"], [0, 0, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not"], [0, 1, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not::a"], [0, 1, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not::s"], [0, 0, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::a"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::b"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::s"], [0, 0, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp"], [0, 1, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::axes"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::axis"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::s"], [0, 0, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::a"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::b"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::s"], [0, 0, 1, "_CPPv43maxRK5array14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max"], [0, 1, 1, "_CPPv43maxRK5array14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::axes"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::axis"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5array14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::s"], [0, 0, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::a"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::b"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::s"], [0, 0, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean"], [0, 1, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::axes"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::axis"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::s"], [0, 0, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::arrays"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::indexing"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::s"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::sparse"], [0, 0, 1, "_CPPv43minRK5array14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min"], [0, 1, 1, "_CPPv43minRK5array14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::axes"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::axis"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5array14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::s"], [0, 0, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::a"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::b"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::s"], [0, 0, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::a"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::destination"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::s"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::source"], [0, 0, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::a"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::b"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::s"], [0, 0, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::a"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::nan"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::neginf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::posinf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::s"], [0, 0, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative"], [0, 1, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative::a"], [0, 1, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative::s"], [0, 0, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::a"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::b"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::s"], [0, 0, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::a"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::axes"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::dtype"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::inverted"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::s"], [0, 0, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones"], [0, 0, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::dtype"], [0, 1, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones::shape"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::shape"], [0, 0, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like"], [0, 1, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like::a"], [0, 1, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like::s"], [0, 0, 1, "_CPPv4I0Ene5array1TRK5array", "operator!="], [0, 0, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!="], [0, 0, 1, "_CPPv4neRK5arrayRK5array", "operator!="], [0, 2, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::T"], [0, 2, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::T"], [0, 1, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::a"], [0, 1, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::a"], [0, 1, 1, "_CPPv4neRK5arrayRK5array", "operator!=::a"], [0, 1, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::b"], [0, 1, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::b"], [0, 1, 1, "_CPPv4neRK5arrayRK5array", "operator!=::b"], [0, 0, 1, "_CPPv4I0Erm5array1TRK5array", "operator%"], [0, 0, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%"], [0, 0, 1, "_CPPv4rmRK5arrayRK5array", "operator%"], [0, 2, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::T"], [0, 2, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::T"], [0, 1, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::a"], [0, 1, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::a"], [0, 1, 1, "_CPPv4rmRK5arrayRK5array", "operator%::a"], [0, 1, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::b"], [0, 1, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::b"], [0, 1, 1, "_CPPv4rmRK5arrayRK5array", "operator%::b"], [0, 0, 1, "_CPPv4anRK5arrayRK5array", "operator&"], [0, 0, 1, "_CPPv4aaRK5arrayRK5array", "operator&&"], [0, 1, 1, "_CPPv4aaRK5arrayRK5array", "operator&&::a"], [0, 1, 1, "_CPPv4aaRK5arrayRK5array", "operator&&::b"], [0, 1, 1, "_CPPv4anRK5arrayRK5array", "operator&::a"], [0, 1, 1, "_CPPv4anRK5arrayRK5array", "operator&::b"], [0, 0, 1, "_CPPv4I0Eml5array1TRK5array", "operator*"], [0, 0, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*"], [0, 0, 1, "_CPPv4mlRK5arrayRK5array", "operator*"], [0, 2, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::T"], [0, 2, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::T"], [0, 1, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::a"], [0, 1, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::a"], [0, 1, 1, "_CPPv4mlRK5arrayRK5array", "operator*::a"], [0, 1, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::b"], [0, 1, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::b"], [0, 1, 1, "_CPPv4mlRK5arrayRK5array", "operator*::b"], [0, 0, 1, "_CPPv4I0Epl5array1TRK5array", "operator+"], [0, 0, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+"], [0, 0, 1, "_CPPv4plRK5arrayRK5array", "operator+"], [0, 2, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::T"], [0, 2, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::T"], [0, 1, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::a"], [0, 1, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::a"], [0, 1, 1, "_CPPv4plRK5arrayRK5array", "operator+::a"], [0, 1, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::b"], [0, 1, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::b"], [0, 1, 1, "_CPPv4plRK5arrayRK5array", "operator+::b"], [0, 0, 1, "_CPPv4I0Emi5array1TRK5array", "operator-"], [0, 0, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-"], [0, 0, 1, "_CPPv4miRK5array", "operator-"], [0, 0, 1, "_CPPv4miRK5arrayRK5array", "operator-"], [0, 2, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::T"], [0, 2, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::T"], [0, 1, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::a"], [0, 1, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::a"], [0, 1, 1, "_CPPv4miRK5array", "operator-::a"], [0, 1, 1, "_CPPv4miRK5arrayRK5array", "operator-::a"], [0, 1, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::b"], [0, 1, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::b"], [0, 1, 1, "_CPPv4miRK5arrayRK5array", "operator-::b"], [0, 0, 1, "_CPPv4dvRK5arrayRK5array", "operator/"], [0, 0, 1, "_CPPv4dvRK5arrayd", "operator/"], [0, 0, 1, "_CPPv4dvdRK5array", "operator/"], [0, 1, 1, "_CPPv4dvRK5arrayRK5array", "operator/::a"], [0, 1, 1, "_CPPv4dvRK5arrayd", "operator/::a"], [0, 1, 1, "_CPPv4dvdRK5array", "operator/::a"], [0, 1, 1, "_CPPv4dvRK5arrayRK5array", "operator/::b"], [0, 1, 1, "_CPPv4dvRK5arrayd", "operator/::b"], [0, 1, 1, "_CPPv4dvdRK5array", "operator/::b"], [0, 0, 1, "_CPPv4I0Elt5array1TRK5array", "operator<"], [0, 0, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<"], [0, 0, 1, "_CPPv4ltRK5arrayRK5array", "operator<"], [0, 2, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::T"], [0, 2, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::T"], [0, 1, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::a"], [0, 1, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::a"], [0, 1, 1, "_CPPv4ltRK5arrayRK5array", "operator<::a"], [0, 1, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::b"], [0, 1, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::b"], [0, 1, 1, "_CPPv4ltRK5arrayRK5array", "operator<::b"], [0, 0, 1, "_CPPv4lsRK5arrayRK5array", "operator<<"], [0, 1, 1, "_CPPv4lsRK5arrayRK5array", "operator<<::a"], [0, 1, 1, "_CPPv4lsRK5arrayRK5array", "operator<<::b"], [0, 0, 1, "_CPPv4I0Ele5array1TRK5array", "operator<="], [0, 0, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<="], [0, 0, 1, "_CPPv4leRK5arrayRK5array", "operator<="], [0, 2, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::T"], [0, 2, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::T"], [0, 1, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::a"], [0, 1, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::a"], [0, 1, 1, "_CPPv4leRK5arrayRK5array", "operator<=::a"], [0, 1, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::b"], [0, 1, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::b"], [0, 1, 1, "_CPPv4leRK5arrayRK5array", "operator<=::b"], [0, 0, 1, "_CPPv4I0Eeq5array1TRK5array", "operator=="], [0, 0, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator=="], [0, 0, 1, "_CPPv4eqRK5arrayRK5array", "operator=="], [0, 2, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::T"], [0, 2, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::T"], [0, 1, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::a"], [0, 1, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::a"], [0, 1, 1, "_CPPv4eqRK5arrayRK5array", "operator==::a"], [0, 1, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::b"], [0, 1, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::b"], [0, 1, 1, "_CPPv4eqRK5arrayRK5array", "operator==::b"], [0, 0, 1, "_CPPv4I0Egt5array1TRK5array", "operator>"], [0, 0, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>"], [0, 0, 1, "_CPPv4gtRK5arrayRK5array", "operator>"], [0, 2, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::T"], [0, 2, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::T"], [0, 1, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::a"], [0, 1, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::a"], [0, 1, 1, "_CPPv4gtRK5arrayRK5array", "operator>::a"], [0, 1, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::b"], [0, 1, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::b"], [0, 1, 1, "_CPPv4gtRK5arrayRK5array", "operator>::b"], [0, 0, 1, "_CPPv4I0Ege5array1TRK5array", "operator>="], [0, 0, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>="], [0, 0, 1, "_CPPv4geRK5arrayRK5array", "operator>="], [0, 2, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::T"], [0, 2, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::T"], [0, 1, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::a"], [0, 1, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::a"], [0, 1, 1, "_CPPv4geRK5arrayRK5array", "operator>=::a"], [0, 1, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::b"], [0, 1, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::b"], [0, 1, 1, "_CPPv4geRK5arrayRK5array", "operator>=::b"], [0, 0, 1, "_CPPv4rsRK5arrayRK5array", "operator>>"], [0, 1, 1, "_CPPv4rsRK5arrayRK5array", "operator>>::a"], [0, 1, 1, "_CPPv4rsRK5arrayRK5array", "operator>>::b"], [0, 0, 1, "_CPPv4eoRK5arrayRK5array", "operator^"], [0, 1, 1, "_CPPv4eoRK5arrayRK5array", "operator^::a"], [0, 1, 1, "_CPPv4eoRK5arrayRK5array", "operator^::b"], [0, 0, 1, "_CPPv4orRK5arrayRK5array", "operator|"], [0, 1, 1, "_CPPv4orRK5arrayRK5array", "operator|::a"], [0, 1, 1, "_CPPv4orRK5arrayRK5array", "operator|::b"], [0, 0, 1, "_CPPv4ooRK5arrayRK5array", "operator||"], [0, 1, 1, "_CPPv4ooRK5arrayRK5array", "operator||::a"], [0, 1, 1, "_CPPv4ooRK5arrayRK5array", "operator||::b"], [0, 0, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::a"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::b"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::s"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice", "pad::axes"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice", "pad::high_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice", "pad::low_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::s"], [0, 0, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition"], [0, 0, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::a"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::a"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::axis"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::kth"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::kth"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::s"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::s"], [0, 0, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::a"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::b"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::s"], [0, 0, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod"], [0, 1, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::axes"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::axis"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::s"], [0, 0, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::a"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::axis"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::indices"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::s"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::values"], [0, 0, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::bits"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::group_size"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::s"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::w"], [0, 0, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::biases"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::bits"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::group_size"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::s"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::scales"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::transpose"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::w"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::x"], [0, 0, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::a"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::s"], [0, 0, 1, "_CPPv44realRK5array14StreamOrDevice", "real"], [0, 1, 1, "_CPPv44realRK5array14StreamOrDevice", "real::a"], [0, 1, 1, "_CPPv44realRK5array14StreamOrDevice", "real::s"], [0, 0, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal"], [0, 1, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal::a"], [0, 1, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal::s"], [0, 0, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::a"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::b"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::s"], [0, 0, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat"], [0, 0, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::arr"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::arr"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::axis"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::repeats"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::repeats"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::s"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::s"], [0, 0, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::a"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::s"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::shape"], [0, 0, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::a"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::b"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::s"], [0, 0, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::axes"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::axes"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::axis"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::axis"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::shift"], [0, 0, 1, "_CPPv45roundRK5array14StreamOrDevice", "round"], [0, 0, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round"], [0, 1, 1, "_CPPv45roundRK5array14StreamOrDevice", "round::a"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::a"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::decimals"], [0, 1, 1, "_CPPv45roundRK5array14StreamOrDevice", "round::s"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::s"], [0, 0, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt"], [0, 1, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt::a"], [0, 1, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt::s"], [0, 0, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter"], [0, 0, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::a"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::a"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::axes"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::axis"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::indices"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::indices"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::s"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::s"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::updates"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::updates"], [0, 0, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add"], [0, 0, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::a"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::a"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::axes"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::axis"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::indices"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::indices"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::s"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::s"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::updates"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::updates"], [0, 0, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max"], [0, 0, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::a"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::a"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::axes"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::axis"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::indices"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::indices"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::s"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::s"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::updates"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::updates"], [0, 0, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min"], [0, 0, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::a"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::a"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::axes"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::axis"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::indices"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::indices"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::s"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::s"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::updates"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::updates"], [0, 0, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod"], [0, 0, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::a"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::a"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::axes"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::axis"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::indices"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::indices"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::s"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::s"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::updates"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::updates"], [0, 0, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid"], [0, 1, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid::a"], [0, 1, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid::s"], [0, 0, 1, "_CPPv44signRK5array14StreamOrDevice", "sign"], [0, 1, 1, "_CPPv44signRK5array14StreamOrDevice", "sign::a"], [0, 1, 1, "_CPPv44signRK5array14StreamOrDevice", "sign::s"], [0, 0, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin"], [0, 1, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin::a"], [0, 1, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin::s"], [0, 0, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh"], [0, 1, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh::a"], [0, 1, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh::s"], [0, 0, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::axes"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::slice_size"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::strides"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::strides"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::axes"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::strides"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::update"], [0, 0, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax"], [0, 0, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax"], [0, 0, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::axes"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::axis"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::s"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::s"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::s"], [0, 0, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort"], [0, 0, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort"], [0, 1, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort::a"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::a"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::axis"], [0, 1, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort::s"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::s"], [0, 0, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::s"], [0, 0, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt"], [0, 1, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt::a"], [0, 1, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt::s"], [0, 0, 1, "_CPPv46squareRK5array14StreamOrDevice", "square"], [0, 1, 1, "_CPPv46squareRK5array14StreamOrDevice", "square::a"], [0, 1, 1, "_CPPv46squareRK5array14StreamOrDevice", "square::s"], [0, 0, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze"], [0, 0, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze"], [0, 0, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze"], [0, 1, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::axes"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::axis"], [0, 1, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze::s"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::s"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::s"], [0, 0, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack"], [0, 0, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack::arrays"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::arrays"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::axis"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack::s"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::s"], [0, 0, 1, "_CPPv4StRK5array14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std"], [0, 1, 1, "_CPPv4StRK5array14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::axes"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::axis"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5array14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::s"], [0, 0, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient"], [0, 1, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient::a"], [0, 1, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient::s"], [0, 0, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::a"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::b"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::s"], [0, 0, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum"], [0, 1, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::axes"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::axis"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::s"], [0, 0, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::a"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::axis1"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::axis2"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::s"], [0, 0, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::index"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::index"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::indices"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::indices"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::s"], [0, 0, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::a"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::axis"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::indices"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::s"], [0, 0, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan"], [0, 1, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan::a"], [0, 1, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan::s"], [0, 0, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh"], [0, 1, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh::a"], [0, 1, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh::s"], [0, 0, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot"], [0, 0, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::axes_a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::axes_b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::axis"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::s"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::s"], [0, 0, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::arr"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::reps"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::s"], [0, 0, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk"], [0, 0, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::a"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::a"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::axis"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::k"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::k"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::s"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::s"], [0, 0, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace"], [0, 0, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace"], [0, 0, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace"], [0, 1, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::axis1"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::axis1"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::axis2"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::axis2"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::dtype"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::offset"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::offset"], [0, 1, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace::s"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::s"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::s"], [0, 0, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose"], [0, 0, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose"], [0, 0, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose"], [0, 1, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::axes"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::axes"], [0, 1, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose::s"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::s"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::s"], [0, 0, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri"], [0, 0, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::k"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::m"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::n"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::n"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::s"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::s"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::type"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::type"], [0, 0, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::k"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::s"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::x"], [0, 0, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::k"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::s"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::x"], [0, 0, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::a"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::axis"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::s"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::shape"], [0, 0, 1, "_CPPv43varRK5array14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var"], [0, 1, 1, "_CPPv43varRK5array14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::axes"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::axis"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5array14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::s"], [0, 0, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::a"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::dtype"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::s"], [0, 0, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::condition"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::s"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::x"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::y"], [0, 0, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros"], [0, 0, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::dtype"], [0, 1, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros::shape"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::shape"], [0, 0, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like"], [0, 1, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like::a"], [0, 1, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like::s"]], "mlx.core": [[10, 3, 1, "", "Device"], [11, 3, 1, "", "Dtype"], [12, 3, 1, "", "DtypeCategory"], [324, 3, 1, "", "Stream"], [13, 5, 1, "", "abs"], [14, 5, 1, "", "add"], [15, 5, 1, "", "addmm"], [16, 5, 1, "", "all"], [17, 5, 1, "", "allclose"], [18, 5, 1, "", "any"], [19, 5, 1, "", "arange"], [20, 5, 1, "", "arccos"], [21, 5, 1, "", "arccosh"], [22, 5, 1, "", "arcsin"], [23, 5, 1, "", "arcsinh"], [24, 5, 1, "", "arctan"], [25, 5, 1, "", "arctan2"], [26, 5, 1, "", "arctanh"], [27, 5, 1, "", "argmax"], [28, 5, 1, "", "argmin"], [29, 5, 1, "", "argpartition"], [30, 5, 1, "", "argsort"], [31, 3, 1, "", "array"], [83, 5, 1, "", "array_equal"], [84, 5, 1, "", "as_strided"], [85, 5, 1, "", "atleast_1d"], [86, 5, 1, "", "atleast_2d"], [87, 5, 1, "", "atleast_3d"], [88, 5, 1, "", "bitwise_and"], [89, 5, 1, "", "bitwise_or"], [90, 5, 1, "", "bitwise_xor"], [91, 5, 1, "", "block_masked_mm"], [92, 5, 1, "", "broadcast_to"], [93, 5, 1, "", "ceil"], [94, 5, 1, "", "clip"], [95, 5, 1, "", "compile"], [96, 5, 1, "", "concatenate"], [97, 5, 1, "", "conj"], [98, 5, 1, "", "conjugate"], [99, 5, 1, "", "conv1d"], [100, 5, 1, "", "conv2d"], [101, 5, 1, "", "conv3d"], [102, 5, 1, "", "conv_general"], [103, 5, 1, "", "conv_transpose1d"], [104, 5, 1, "", "conv_transpose2d"], [105, 5, 1, "", "conv_transpose3d"], [106, 5, 1, "", "convolve"], [107, 5, 1, "", "cos"], [108, 5, 1, "", "cosh"], [109, 5, 1, "", "cummax"], [110, 5, 1, "", "cummin"], [111, 5, 1, "", "cumprod"], [112, 5, 1, "", "cumsum"], [113, 3, 1, "", "custom_function"], [114, 5, 1, "", "default_device"], [115, 5, 1, "", "default_stream"], [116, 5, 1, "", "degrees"], [117, 5, 1, "", "dequantize"], [118, 5, 1, "", "diag"], [119, 5, 1, "", "diagonal"], [120, 5, 1, "", "disable_compile"], [129, 5, 1, "", "divide"], [130, 5, 1, "", "divmod"], [131, 5, 1, "", "einsum"], [132, 5, 1, "", "einsum_path"], [133, 5, 1, "", "enable_compile"], [134, 5, 1, "", "equal"], [135, 5, 1, "", "erf"], [136, 5, 1, "", "erfinv"], [137, 5, 1, "", "eval"], [138, 5, 1, "", "exp"], [139, 5, 1, "", "expand_dims"], [140, 5, 1, "", "expm1"], [141, 5, 1, "", "export_function"], [142, 5, 1, "", "export_to_dot"], [143, 5, 1, "", "exporter"], [144, 5, 1, "", "eye"], [162, 3, 1, "", "finfo"], [163, 5, 1, "", "flatten"], [164, 5, 1, "", "floor"], [165, 5, 1, "", "floor_divide"], [166, 5, 1, "", "full"], [167, 5, 1, "", "gather_mm"], [168, 5, 1, "", "gather_qmm"], [169, 5, 1, "", "grad"], [170, 5, 1, "", "greater"], [171, 5, 1, "", "greater_equal"], [172, 5, 1, "", "hadamard_transform"], [173, 5, 1, "", "identity"], [174, 5, 1, "", "imag"], [175, 5, 1, "", "import_function"], [176, 5, 1, "", "inner"], [177, 5, 1, "", "isclose"], [178, 5, 1, "", "isfinite"], [179, 5, 1, "", "isinf"], [180, 5, 1, "", "isnan"], [181, 5, 1, "", "isneginf"], [182, 5, 1, "", "isposinf"], [183, 5, 1, "", "issubdtype"], [184, 5, 1, "", "jvp"], [185, 5, 1, "", "kron"], [186, 5, 1, "", "left_shift"], [187, 5, 1, "", "less"], [188, 5, 1, "", "less_equal"], [199, 5, 1, "", "linspace"], [200, 5, 1, "", "load"], [201, 5, 1, "", "log"], [202, 5, 1, "", "log10"], [203, 5, 1, "", "log1p"], [204, 5, 1, "", "log2"], [205, 5, 1, "", "logaddexp"], [206, 5, 1, "", "logical_and"], [207, 5, 1, "", "logical_not"], [208, 5, 1, "", "logical_or"], [209, 5, 1, "", "logsumexp"], [210, 5, 1, "", "matmul"], [211, 5, 1, "", "max"], [212, 5, 1, "", "maximum"], [213, 5, 1, "", "mean"], [214, 5, 1, "", "meshgrid"], [227, 5, 1, "", "min"], [228, 5, 1, "", "minimum"], [229, 5, 1, "", "moveaxis"], [230, 5, 1, "", "multiply"], [231, 5, 1, "", "nan_to_num"], [232, 5, 1, "", "negative"], [233, 5, 1, "", "new_stream"], [234, 5, 1, "", "not_equal"], [235, 5, 1, "", "ones"], [236, 5, 1, "", "ones_like"], [237, 5, 1, "", "outer"], [238, 5, 1, "", "pad"], [239, 5, 1, "", "partition"], [240, 5, 1, "", "power"], [241, 5, 1, "", "prod"], [242, 5, 1, "", "put_along_axis"], [243, 5, 1, "", "quantize"], [244, 5, 1, "", "quantized_matmul"], [245, 5, 1, "", "radians"], [259, 5, 1, "", "real"], [260, 5, 1, "", "reciprocal"], [261, 5, 1, "", "remainder"], [262, 5, 1, "", "repeat"], [263, 5, 1, "", "reshape"], [264, 5, 1, "", "right_shift"], [265, 5, 1, "", "roll"], [266, 5, 1, "", "round"], [267, 5, 1, "", "rsqrt"], [268, 5, 1, "", "save"], [269, 5, 1, "", "save_gguf"], [270, 5, 1, "", "save_safetensors"], [271, 5, 1, "", "savez"], [272, 5, 1, "", "savez_compressed"], [273, 5, 1, "", "set_default_device"], [274, 5, 1, "", "set_default_stream"], [275, 5, 1, "", "sigmoid"], [276, 5, 1, "", "sign"], [277, 5, 1, "", "sin"], [278, 5, 1, "", "sinh"], [279, 5, 1, "", "slice"], [280, 5, 1, "", "slice_update"], [281, 5, 1, "", "softmax"], [282, 5, 1, "", "sort"], [283, 5, 1, "", "split"], [284, 5, 1, "", "sqrt"], [285, 5, 1, "", "square"], [286, 5, 1, "", "squeeze"], [287, 5, 1, "", "stack"], [288, 5, 1, "", "std"], [289, 5, 1, "", "stop_gradient"], [290, 5, 1, "", "stream"], [291, 5, 1, "", "subtract"], [292, 5, 1, "", "sum"], [293, 5, 1, "", "swapaxes"], [294, 5, 1, "", "synchronize"], [295, 5, 1, "", "take"], [296, 5, 1, "", "take_along_axis"], [297, 5, 1, "", "tan"], [298, 5, 1, "", "tanh"], [299, 5, 1, "", "tensordot"], [300, 5, 1, "", "tile"], [301, 5, 1, "", "topk"], [302, 5, 1, "", "trace"], [303, 5, 1, "", "transpose"], [304, 5, 1, "", "tri"], [305, 5, 1, "", "tril"], [306, 5, 1, "", "triu"], [307, 5, 1, "", "unflatten"], [308, 5, 1, "", "value_and_grad"], [309, 5, 1, "", "var"], [310, 5, 1, "", "view"], [311, 5, 1, "", "vjp"], [312, 5, 1, "", "vmap"], [313, 5, 1, "", "where"], [314, 5, 1, "", "zeros"], [315, 5, 1, "", "zeros_like"]], "mlx.core.Device": [[10, 4, 1, "", "__init__"]], "mlx.core.Dtype": [[11, 4, 1, "", "__init__"]], "mlx.core.DtypeCategory": [[12, 4, 1, "", "__init__"]], "mlx.core.Stream": [[324, 4, 1, "", "__init__"]], "mlx.core.array": [[32, 6, 1, "", "T"], [31, 4, 1, "", "__init__"], [33, 4, 1, "", "abs"], [34, 4, 1, "", "all"], [35, 4, 1, "", "any"], [36, 4, 1, "", "argmax"], [37, 4, 1, "", "argmin"], [38, 4, 1, "", "astype"], [39, 6, 1, "", "at"], [40, 4, 1, "", "conj"], [41, 4, 1, "", "cos"], [42, 4, 1, "", "cummax"], [43, 4, 1, "", "cummin"], [44, 4, 1, "", "cumprod"], [45, 4, 1, "", "cumsum"], [46, 4, 1, "", "diag"], [47, 4, 1, "", "diagonal"], [48, 6, 1, "", "dtype"], [49, 4, 1, "", "exp"], [50, 4, 1, "", "flatten"], [51, 4, 1, "", "item"], [52, 6, 1, "", "itemsize"], [53, 4, 1, "", "log"], [54, 4, 1, "", "log10"], [55, 4, 1, "", "log1p"], [56, 4, 1, "", "log2"], [57, 4, 1, "", "logsumexp"], [58, 4, 1, "", "max"], [59, 4, 1, "", "mean"], [60, 4, 1, "", "min"], [61, 4, 1, "", "moveaxis"], [62, 6, 1, "", "nbytes"], [63, 6, 1, "", "ndim"], [64, 4, 1, "", "prod"], [65, 4, 1, "", "reciprocal"], [66, 4, 1, "", "reshape"], [67, 4, 1, "", "round"], [68, 4, 1, "", "rsqrt"], [69, 6, 1, "", "shape"], [70, 4, 1, "", "sin"], [71, 6, 1, "", "size"], [72, 4, 1, "", "split"], [73, 4, 1, "", "sqrt"], [74, 4, 1, "", "square"], [75, 4, 1, "", "squeeze"], [76, 4, 1, "", "std"], [77, 4, 1, "", "sum"], [78, 4, 1, "", "swapaxes"], [79, 4, 1, "", "tolist"], [80, 4, 1, "", "transpose"], [81, 4, 1, "", "var"], [82, 4, 1, "", "view"]], "mlx.core.custom_function": [[113, 4, 1, "", "__init__"]], "mlx.core.distributed": [[121, 3, 1, "", "Group"], [122, 5, 1, "", "all_gather"], [123, 5, 1, "", "all_sum"], [124, 5, 1, "", "init"], [125, 5, 1, "", "is_available"], [126, 5, 1, "", "recv"], [127, 5, 1, "", "recv_like"], [128, 5, 1, "", "send"]], "mlx.core.distributed.Group": [[121, 4, 1, "", "__init__"]], "mlx.core.fast": [[145, 5, 1, "", "layer_norm"], [146, 5, 1, "", "metal_kernel"], [147, 5, 1, "", "rms_norm"], [148, 5, 1, "", "rope"], [149, 5, 1, "", "scaled_dot_product_attention"]], "mlx.core.fft": [[150, 5, 1, "", "fft"], [151, 5, 1, "", "fft2"], [152, 5, 1, "", "fftn"], [153, 5, 1, "", "ifft"], [154, 5, 1, "", "ifft2"], [155, 5, 1, "", "ifftn"], [156, 5, 1, "", "irfft"], [157, 5, 1, "", "irfft2"], [158, 5, 1, "", "irfftn"], [159, 5, 1, "", "rfft"], [160, 5, 1, "", "rfft2"], [161, 5, 1, "", "rfftn"]], "mlx.core.finfo": [[162, 4, 1, "", "__init__"]], "mlx.core.linalg": [[189, 5, 1, "", "cholesky"], [190, 5, 1, "", "cholesky_inv"], [191, 5, 1, "", "cross"], [192, 5, 1, "", "eigh"], [193, 5, 1, "", "eigvalsh"], [194, 5, 1, "", "inv"], [195, 5, 1, "", "norm"], [196, 5, 1, "", "qr"], [197, 5, 1, "", "svd"], [198, 5, 1, "", "tri_inv"]], "mlx.core.metal": [[215, 5, 1, "", "clear_cache"], [216, 5, 1, "", "device_info"], [217, 5, 1, "", "get_active_memory"], [218, 5, 1, "", "get_cache_memory"], [219, 5, 1, "", "get_peak_memory"], [220, 5, 1, "", "is_available"], [221, 5, 1, "", "reset_peak_memory"], [222, 5, 1, "", "set_cache_limit"], [223, 5, 1, "", "set_memory_limit"], [224, 5, 1, "", "set_wired_limit"], [225, 5, 1, "", "start_capture"], [226, 5, 1, "", "stop_capture"]], "mlx.core.random": [[246, 5, 1, "", "bernoulli"], [247, 5, 1, "", "categorical"], [248, 5, 1, "", "gumbel"], [249, 5, 1, "", "key"], [250, 5, 1, "", "laplace"], [251, 5, 1, "", "multivariate_normal"], [252, 5, 1, "", "normal"], [253, 5, 1, "", "permutation"], [254, 5, 1, "", "randint"], [255, 5, 1, "", "seed"], [256, 5, 1, "", "split"], [257, 5, 1, "", "truncated_normal"], [258, 5, 1, "", "uniform"]], "mlx.nn": [[335, 3, 1, "", "ALiBi"], [336, 3, 1, "", "AvgPool1d"], [337, 3, 1, "", "AvgPool2d"], [338, 3, 1, "", "AvgPool3d"], [339, 3, 1, "", "BatchNorm"], [340, 3, 1, "", "CELU"], [341, 3, 1, "", "Conv1d"], [342, 3, 1, "", "Conv2d"], [343, 3, 1, "", "Conv3d"], [344, 3, 1, "", "ConvTranspose1d"], [345, 3, 1, "", "ConvTranspose2d"], [346, 3, 1, "", "ConvTranspose3d"], [347, 3, 1, "", "Dropout"], [348, 3, 1, "", "Dropout2d"], [349, 3, 1, "", "Dropout3d"], [350, 3, 1, "", "ELU"], [351, 3, 1, "", "Embedding"], [352, 3, 1, "", "GELU"], [353, 3, 1, "", "GLU"], [354, 3, 1, "", "GRU"], [355, 3, 1, "", "GroupNorm"], [356, 3, 1, "", "HardShrink"], [357, 3, 1, "", "HardTanh"], [358, 3, 1, "", "Hardswish"], [359, 3, 1, "", "InstanceNorm"], [360, 3, 1, "", "LSTM"], [361, 3, 1, "", "LayerNorm"], [362, 3, 1, "", "LeakyReLU"], [363, 3, 1, "", "Linear"], [364, 3, 1, "", "LogSigmoid"], [365, 3, 1, "", "LogSoftmax"], [366, 3, 1, "", "MaxPool1d"], [367, 3, 1, "", "MaxPool2d"], [368, 3, 1, "", "MaxPool3d"], [369, 3, 1, "", "Mish"], [464, 3, 1, "", "Module"], [390, 3, 1, "", "MultiHeadAttention"], [391, 3, 1, "", "PReLU"], [392, 3, 1, "", "QuantizedEmbedding"], [393, 3, 1, "", "QuantizedLinear"], [394, 3, 1, "", "RMSNorm"], [395, 3, 1, "", "RNN"], [396, 3, 1, "", "ReLU"], [397, 3, 1, "", "ReLU6"], [398, 3, 1, "", "RoPE"], [399, 3, 1, "", "SELU"], [400, 3, 1, "", "Sequential"], [401, 3, 1, "", "SiLU"], [402, 3, 1, "", "Sigmoid"], [403, 3, 1, "", "SinusoidalPositionalEncoding"], [404, 3, 1, "", "Softmax"], [405, 3, 1, "", "Softmin"], [406, 3, 1, "", "Softplus"], [407, 3, 1, "", "Softshrink"], [408, 3, 1, "", "Softsign"], [409, 3, 1, "", "Step"], [410, 3, 1, "", "Tanh"], [411, 3, 1, "", "Transformer"], [412, 3, 1, "", "Upsample"], [421, 3, 1, "", "celu"], [422, 3, 1, "", "elu"], [423, 3, 1, "", "gelu"], [424, 3, 1, "", "gelu_approx"], [425, 3, 1, "", "gelu_fast_approx"], [426, 3, 1, "", "glu"], [427, 3, 1, "", "hard_shrink"], [428, 3, 1, "", "hard_tanh"], [429, 3, 1, "", "hardswish"], [430, 3, 1, "", "leaky_relu"], [431, 3, 1, "", "log_sigmoid"], [432, 3, 1, "", "log_softmax"], [447, 3, 1, "", "mish"], [448, 3, 1, "", "prelu"], [316, 5, 1, "", "quantize"], [449, 3, 1, "", "relu"], [450, 3, 1, "", "relu6"], [451, 3, 1, "", "selu"], [452, 3, 1, "", "sigmoid"], [453, 3, 1, "", "silu"], [454, 3, 1, "", "softmax"], [455, 3, 1, "", "softmin"], [456, 3, 1, "", "softplus"], [457, 3, 1, "", "softshrink"], [458, 3, 1, "", "step"], [459, 3, 1, "", "tanh"], [317, 5, 1, "", "value_and_grad"]], "mlx.nn.Module": [[370, 4, 1, "", "apply"], [371, 4, 1, "", "apply_to_modules"], [372, 4, 1, "", "children"], [373, 4, 1, "", "eval"], [374, 4, 1, "", "filter_and_map"], [375, 4, 1, "", "freeze"], [376, 4, 1, "", "leaf_modules"], [377, 4, 1, "", "load_weights"], [378, 4, 1, "", "modules"], [379, 4, 1, "", "named_modules"], [380, 4, 1, "", "parameters"], [381, 4, 1, "", "save_weights"], [382, 4, 1, "", "set_dtype"], [383, 6, 1, "", "state"], [384, 4, 1, "", "train"], [385, 4, 1, "", "trainable_parameters"], [386, 6, 1, "", "training"], [387, 4, 1, "", "unfreeze"], [388, 4, 1, "", "update"], [389, 4, 1, "", "update_modules"]], "mlx.nn.init": [[413, 5, 1, "", "constant"], [414, 5, 1, "", "glorot_normal"], [415, 5, 1, "", "glorot_uniform"], [416, 5, 1, "", "he_normal"], [417, 5, 1, "", "he_uniform"], [418, 5, 1, "", "identity"], [419, 5, 1, "", "normal"], [420, 5, 1, "", "uniform"]], "mlx.nn.losses": [[433, 3, 1, "", "binary_cross_entropy"], [434, 3, 1, "", "cosine_similarity_loss"], [435, 3, 1, "", "cross_entropy"], [436, 3, 1, "", "gaussian_nll_loss"], [437, 3, 1, "", "hinge_loss"], [438, 3, 1, "", "huber_loss"], [439, 3, 1, "", "kl_div_loss"], [440, 3, 1, "", "l1_loss"], [441, 3, 1, "", "log_cosh_loss"], [442, 3, 1, "", "margin_ranking_loss"], [443, 3, 1, "", "mse_loss"], [444, 3, 1, "", "nll_loss"], [445, 3, 1, "", "smooth_l1_loss"], [446, 3, 1, "", "triplet_loss"]], "mlx.optimizers": [[467, 3, 1, "", "AdaDelta"], [468, 3, 1, "", "Adafactor"], [469, 3, 1, "", "Adagrad"], [470, 3, 1, "", "Adam"], [471, 3, 1, "", "AdamW"], [472, 3, 1, "", "Adamax"], [473, 3, 1, "", "Lion"], [486, 3, 1, "", "Optimizer"], [478, 3, 1, "", "RMSprop"], [479, 3, 1, "", "SGD"], [318, 5, 1, "", "clip_grad_norm"], [480, 5, 1, "", "cosine_decay"], [481, 5, 1, "", "exponential_decay"], [482, 5, 1, "", "join_schedules"], [483, 5, 1, "", "linear_schedule"], [484, 5, 1, "", "step_decay"]], "mlx.optimizers.Optimizer": [[474, 4, 1, "", "apply_gradients"], [475, 4, 1, "", "init"], [476, 6, 1, "", "state"], [477, 4, 1, "", "update"]], "mlx.utils": [[319, 5, 1, "", "tree_flatten"], [320, 5, 1, "", "tree_map"], [321, 5, 1, "", "tree_map_with_path"], [322, 5, 1, "", "tree_reduce"], [323, 5, 1, "", "tree_unflatten"]]}, "objnames": {"0": ["cpp", "function", "C++ function"], "1": ["cpp", "functionParam", "C++ function parameter"], "2": ["cpp", "templateParam", "C++ template parameter"], "3": ["py", "class", "Python class"], "4": ["py", "method", "Python method"], "5": ["py", "function", "Python function"], "6": ["py", "property", "Python property"]}, "objtypes": {"0": "cpp:function", "1": "cpp:functionParam", "2": "cpp:templateParam", "3": "py:class", "4": "py:method", "5": "py:function", "6": "py:property"}, "terms": {"": [0, 1, 2, 5, 6, 7, 48, 52, 63, 95, 115, 117, 151, 152, 154, 155, 157, 158, 160, 161, 169, 190, 195, 197, 200, 213, 237, 243, 247, 266, 269, 270, 288, 290, 308, 309, 310, 312, 317, 334, 337, 338, 354, 360, 367, 368, 374, 375, 377, 381, 382, 383, 387, 395, 466, 475, 476, 488, 491, 493, 494, 497, 498, 499, 500], "0": [0, 1, 2, 4, 5, 6, 7, 9, 10, 15, 19, 39, 46, 47, 50, 67, 72, 76, 81, 84, 96, 99, 100, 101, 102, 103, 104, 105, 113, 118, 119, 144, 146, 149, 163, 167, 169, 175, 185, 192, 194, 195, 196, 198, 215, 222, 224, 231, 238, 246, 250, 252, 253, 258, 262, 266, 279, 280, 283, 287, 288, 302, 304, 305, 306, 307, 308, 309, 312, 318, 319, 321, 322, 334, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 352, 355, 356, 359, 361, 362, 366, 367, 368, 391, 396, 398, 403, 407, 409, 411, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 424, 425, 427, 428, 429, 430, 433, 435, 437, 438, 442, 445, 446, 448, 449, 450, 451, 457, 458, 461, 464, 467, 468, 470, 471, 472, 473, 475, 478, 479, 480, 481, 482, 483, 484, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499], "00005": 5, "0001": 403, "0005": 424, "001": 468, "00364": 5, "01": [5, 362, 430, 471], "0137595": 416, "015": 425, "0184009": 417, "02264": 415, "024": 494, "02765": 416, "0300242": 417, "044715": [352, 424], "0485873": 435, "05": [17, 177, 339, 355, 359, 361, 394], "0507": 451, "05202": 6, "06": [436, 446, 467], "0638": 442, "06450": 361, "0645099": 419, "06561": 481, "06675": 473, "07467": 394, "08": [17, 177, 434, 469, 470, 471, 472, 478], "08022": 359, "081": 484, "08415": 425, "08494": 355, "08619": 417, "08681": [369, 447], "09864": 6, "0999938": 482, "0999961": 480, "0f": 0, "1": [0, 1, 2, 3, 4, 6, 7, 15, 19, 29, 30, 39, 47, 50, 99, 100, 101, 102, 103, 104, 105, 113, 118, 119, 140, 141, 142, 143, 146, 149, 150, 151, 153, 154, 156, 157, 158, 159, 160, 161, 163, 172, 176, 183, 185, 190, 191, 192, 193, 195, 196, 210, 214, 223, 237, 239, 243, 247, 250, 251, 252, 258, 275, 279, 280, 282, 295, 301, 302, 307, 308, 318, 321, 322, 326, 334, 336, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 352, 353, 354, 355, 359, 360, 361, 363, 366, 391, 394, 395, 398, 402, 403, 409, 412, 414, 415, 416, 417, 418, 419, 420, 421, 422, 424, 425, 426, 428, 431, 432, 433, 434, 435, 436, 437, 438, 439, 441, 442, 444, 445, 446, 451, 452, 454, 455, 456, 458, 461, 464, 466, 467, 468, 469, 470, 471, 472, 473, 475, 478, 479, 480, 481, 482, 483, 484, 491, 492, 493, 494, 495, 497, 498, 499, 500], "10": [0, 3, 6, 7, 185, 202, 266, 271, 320, 334, 377, 461, 482, 484, 491, 492, 495], "100": [2, 5, 6, 433, 483, 491, 494, 496, 500], "1000": [480, 491], "10000": 398, "101": 483, "1024": [1, 6], "105361": 433, "109": 2, "10_000": 5, "10x": 473, "11": 195, "114": 2, "12": [6, 172, 185, 482], "1212": 467, "12451": 415, "128": [271, 334], "13": 9, "14": [9, 185], "15": [1, 9, 185, 195, 224, 322, 491], "150594": 414, "15268": 416, "16": [1, 146, 326, 336, 338, 359, 366, 368, 370, 464], "1606": 425, "1607": [359, 361], "16384": 172, "16506": 417, "17": [4, 9], "177208": 416, "18": 185, "1803": 355, "1908": [369, 447], "1910": 394, "191107": 414, "1985": 195, "1_000": 5, "1d": [0, 99, 103, 106, 269, 296], "1e": [0, 5, 7, 17, 177, 339, 355, 359, 361, 362, 394, 434, 436, 446, 466, 467, 468, 469, 470, 471, 472, 475, 478, 480, 481, 482, 483, 484], "1e3": 491, "1st": 243, "2": [0, 1, 2, 4, 5, 6, 7, 39, 100, 104, 113, 118, 119, 135, 141, 142, 143, 151, 154, 156, 157, 158, 159, 160, 161, 163, 172, 183, 185, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 204, 210, 243, 251, 256, 279, 280, 299, 302, 304, 305, 306, 307, 318, 322, 326, 334, 336, 337, 338, 342, 345, 352, 362, 366, 367, 368, 394, 403, 412, 413, 414, 415, 416, 417, 418, 419, 420, 424, 435, 436, 438, 445, 446, 461, 464, 466, 467, 469, 470, 471, 475, 478, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "20": [172, 185, 195], "200": [6, 482, 494], "2002": 6, "2011": 469, "2012": [467, 478], "2015": [348, 470, 472], "2019": [6, 471], "2020": 6, "2021": 6, "20397": 433, "20_000": 6, "21": [6, 185, 484], "2104": 6, "223144": 433, "223404": 415, "225": 195, "225763": 442, "2302": 473, "23607": [195, 196], "24": 185, "24264": 195, "247": 6, "25": [9, 391, 412], "25211": 416, "256": [1, 2, 7, 146], "256995": 442, "27": 4, "28": [172, 185], "2d": [0, 100, 104, 119, 243, 339, 348], "2nd": 243, "2x": 497, "3": [0, 1, 2, 4, 6, 9, 101, 105, 113, 141, 143, 163, 183, 185, 191, 192, 193, 195, 196, 279, 280, 307, 318, 322, 338, 343, 346, 352, 368, 412, 415, 417, 424, 429, 468, 473, 488, 491, 493, 495, 497, 498], "30": 468, "3118": 497, "32": [1, 6, 7, 91, 243, 244, 326, 337, 338, 367, 368, 394, 491], "330": 6, "33333": 412, "348587": 435, "363207": 414, "36788": 491, "379159": 415, "380709": 419, "39": 6, "3d": [0, 2, 101, 105, 339, 349, 412], "3f": [2, 7, 491], "3x": 2, "4": [0, 1, 2, 6, 117, 146, 149, 163, 168, 185, 195, 243, 244, 271, 279, 307, 316, 322, 326, 336, 337, 338, 339, 359, 366, 367, 368, 392, 393, 411, 412, 414, 415, 416, 433, 491, 492, 493, 495, 498, 500], "4096": [491, 494, 500], "40x": 1, "41421": 195, "417497": 420, "42": 323, "437": 6, "44": 6, "447214": 196, "458835": 416, "475": 6, "48095": 414, "4d": [1, 412], "4m": 1, "5": [0, 1, 2, 5, 6, 9, 185, 195, 223, 246, 279, 322, 336, 339, 347, 348, 349, 352, 356, 359, 366, 407, 412, 413, 416, 417, 424, 427, 445, 457, 461, 466, 478, 480, 481, 491, 494, 495], "50": [0, 199], "500": [6, 500], "5000": 2, "510826": 433, "512": [2, 3, 6, 411, 500], "534422": 419, "539245": 433, "53947": 414, "55": 1, "5701": 467, "573409": 442, "57771": 196, "579": 6, "5f": 5, "6": [1, 2, 6, 113, 185, 195, 271, 279, 397, 411, 415, 424, 425, 429, 436, 446, 450, 478, 491, 495, 498], "61278": 414, "617261": 420, "628": 6, "633": 6, "639": 494, "64": [0, 1, 91, 117, 168, 243, 244, 316, 326, 392, 393], "64331": 417, "666329": 417, "66667": 412, "67326": 451, "676": 1, "690": 6, "6967": 416, "7": [2, 6, 185, 195, 243, 495], "702": [352, 425], "707107": 192, "71828": 491, "74166": 195, "74597": 195, "75": 412, "75596": 442, "75787": 416, "765166": 442, "773433": 442, "776856": 415, "793615": 417, "79854": 417, "7b": 6, "7m": 1, "8": [0, 1, 2, 6, 9, 195, 243, 326, 337, 338, 359, 367, 368, 411, 434, 467, 468, 469, 470, 471, 472, 478, 491, 495, 498, 500], "8192": [6, 172], "84804": 195, "863726": 420, "883935": 420, "890597": 415, "894427": 196, "89613": 414, "8gb": 6, "8x": 1, "9": [4, 9, 195, 435, 467, 470, 471, 472, 473, 475, 481, 484, 497], "90041": 415, "912766": 415, "916291": 433, "95": 7, "982273": 419, "99": [473, 478], "995016": 414, "999": [470, 471, 472], "A": [0, 2, 6, 8, 9, 10, 69, 83, 95, 141, 142, 145, 146, 147, 149, 169, 183, 184, 190, 192, 193, 195, 196, 197, 200, 209, 210, 211, 216, 227, 243, 246, 247, 248, 250, 251, 252, 253, 254, 257, 258, 283, 287, 290, 308, 311, 312, 316, 317, 318, 319, 320, 321, 322, 323, 324, 334, 339, 348, 354, 355, 359, 361, 374, 378, 379, 382, 388, 389, 394, 400, 403, 411, 414, 415, 417, 425, 446, 447, 464, 466, 470, 472, 474, 475, 477, 482, 491, 492, 493, 494, 496, 497], "AS": 167, "And": [4, 6, 412], "As": [7, 39, 295, 334], "At": [94, 307], "But": [493, 500], "By": [6, 316, 382, 433, 494, 497], "For": [0, 1, 2, 4, 6, 9, 39, 113, 149, 167, 183, 195, 243, 323, 334, 339, 348, 352, 370, 375, 384, 387, 393, 398, 403, 412, 414, 415, 416, 417, 433, 461, 466, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "If": [0, 1, 2, 4, 6, 9, 16, 17, 18, 19, 27, 28, 29, 30, 79, 83, 84, 94, 96, 106, 109, 110, 111, 112, 118, 119, 122, 123, 124, 126, 127, 128, 137, 145, 148, 149, 159, 160, 161, 165, 166, 169, 177, 189, 190, 191, 195, 200, 209, 210, 211, 213, 214, 222, 223, 227, 231, 235, 238, 239, 241, 242, 247, 251, 253, 262, 265, 281, 282, 283, 288, 292, 294, 295, 296, 299, 301, 302, 308, 309, 312, 314, 316, 320, 322, 339, 341, 342, 343, 344, 345, 346, 355, 361, 363, 375, 377, 387, 393, 395, 398, 400, 403, 412, 433, 435, 446, 468, 470, 471, 491, 492, 493, 494, 496, 499, 500, 501], "In": [0, 1, 2, 6, 7, 39, 149, 210, 243, 320, 334, 348, 355, 464, 467, 469, 470, 472, 473, 474, 490, 491, 492, 493, 494, 496, 499, 500], "It": [2, 6, 9, 127, 169, 274, 308, 318, 322, 334, 389, 393, 474, 486, 497, 499], "Its": [334, 493], "No": [2, 6, 192, 193], "Not": [95, 234, 491], "ON": [3, 4, 9], "Of": 494, "On": [1, 491, 494, 496], "One": [150, 153, 159, 238, 267, 491, 493, 494], "THE": 9, "That": 6, "The": [0, 1, 2, 3, 4, 6, 7, 8, 9, 13, 14, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 48, 52, 62, 63, 69, 79, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 116, 117, 118, 119, 122, 123, 124, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 138, 139, 140, 141, 142, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 184, 185, 186, 187, 188, 191, 192, 193, 195, 196, 197, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 218, 219, 222, 223, 224, 225, 227, 228, 229, 230, 232, 234, 235, 236, 237, 238, 239, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 269, 270, 275, 276, 277, 278, 279, 280, 281, 282, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 326, 328, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 347, 348, 349, 351, 353, 354, 355, 359, 360, 361, 363, 366, 367, 368, 370, 371, 375, 377, 381, 382, 383, 384, 387, 388, 389, 390, 392, 393, 394, 395, 398, 400, 403, 409, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 458, 461, 464, 466, 467, 468, 469, 470, 471, 472, 473, 476, 478, 479, 480, 483, 486, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "Then": [5, 9], "There": [1, 2, 334, 412, 491], "These": [1, 2, 95, 242, 296, 435, 500], "To": [0, 2, 3, 5, 6, 7, 9, 222, 334, 461, 466, 491, 492, 493, 494, 498], "With": [2, 493], "_": [1, 3, 5, 6, 321, 334, 480, 481, 482, 483, 484, 488, 491, 496, 500], "__call__": [1, 6, 7, 334, 464, 493], "__init__": [2, 6, 7, 10, 11, 12, 31, 113, 121, 162, 324, 334, 464], "__main__": [2, 6], "__name__": [2, 6], "_a": 2, "_ext": 2, "_f": 195, "_in": [414, 415], "_out": [414, 415], "_p": 446, "_val": 428, "a1": 167, "a2": 167, "a_": 195, "a_max": [0, 94], "a_min": [0, 94], "a_ndim": 1, "a_shap": 1, "a_strid": 1, "a_view": 497, "ab": [0, 17, 177, 195, 308, 355, 359, 361, 369, 394, 425, 447, 491, 493], "abil": 492, "abl": [2, 4, 243], "abort": 113, "about": [1, 2, 6, 7, 132, 216, 496, 500], "abov": [1, 2, 6, 243, 305, 334, 412, 492, 493, 494, 495, 496, 500], "absolut": [0, 13, 17, 177, 424, 425, 445], "acc": 322, "acceler": [2, 4, 339], "access": [0, 6, 51, 334, 464, 475, 492, 496, 500], "accord": [0, 248, 313, 316, 390, 414, 415, 416, 417], "accordingli": 2, "accumul": [322, 394], "accuraci": 7, "accustom": 6, "achiev": [334, 492], "across": [1, 2, 9, 355, 492], "act": [2, 441], "action": 334, "activ": [2, 9, 217, 348, 409, 411, 427, 447, 457, 458, 460, 491], "actual": [6, 19, 377, 464, 496], "ad": [0, 1, 2, 5, 9, 145, 359, 464, 467, 468, 469, 470, 471, 472, 478, 492, 496, 499], "adadelta": 466, "adafactor": 466, "adagrad": 466, "adam": [466, 472, 473, 482, 483], "adamax": 466, "adamw": [466, 473], "adapt": [467, 468, 469, 492], "add": [0, 1, 2, 3, 4, 6, 15, 39, 139, 205, 238, 243, 341, 342, 343, 344, 345, 346, 493, 494, 500], "add_argu": 6, "add_depend": 2, "add_execut": 4, "add_fun": 493, "add_librari": 2, "addit": [0, 2, 4, 6, 9, 14, 15, 141, 145, 147, 149, 200, 339, 355, 361, 390, 394, 464, 494], "addmm": 0, "address": 2, "adjac": 348, "advanc": [6, 491], "advantag": 500, "advis": 497, "affin": [339, 355, 359, 361, 363, 393], "after": [2, 6, 7, 29, 163, 165, 168, 215, 239, 243, 339, 355, 361, 370, 371, 375, 377, 384, 387, 388, 389, 390, 411, 445, 491, 500], "after_1": 238, "after_2": 238, "after_i": 238, "after_n": 238, "afternoon": 6, "again": [6, 9, 334, 491], "against": [0, 4], "aggreg": 390, "ago": 6, "ai": 113, "ainv": [194, 198], "albeit": 500, "algebra": 8, "algorithm": [412, 473], "alia": [97, 98, 352], "alibi": 334, "align": [190, 243, 354, 360], "align_corn": 412, "all": [0, 1, 2, 3, 7, 9, 17, 29, 39, 85, 86, 87, 95, 100, 101, 102, 104, 105, 113, 122, 123, 124, 142, 144, 152, 155, 158, 161, 167, 168, 197, 210, 238, 239, 265, 286, 316, 334, 370, 371, 375, 378, 379, 380, 385, 387, 390, 403, 411, 412, 461, 464, 486, 488, 491, 495, 496, 498, 501], "all_avg": 492, "all_reduce_grad": 492, "all_sum": 492, "allclos": [0, 1, 146], "alloc": [2, 218, 222, 223, 464], "allow": [0, 1, 2, 141, 143, 183, 318, 334, 389, 464, 486, 492, 495, 498], "allow_col_major": 0, "almost": 6, "alon": [2, 497], "along": [0, 2, 27, 28, 95, 96, 109, 110, 111, 112, 122, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 167, 168, 172, 191, 195, 242, 253, 262, 265, 281, 283, 287, 295, 296, 299, 300, 301, 302, 310, 334, 353, 395, 426], "alpha": [0, 2, 15, 243, 340, 350, 421, 422, 446, 448, 451, 471, 478], "alpha_": 2, "alreadi": [2, 3, 6, 492], "also": [0, 1, 2, 4, 6, 7, 8, 9, 12, 14, 88, 89, 90, 120, 129, 130, 134, 152, 155, 158, 161, 170, 171, 186, 187, 188, 205, 212, 228, 230, 234, 240, 243, 261, 264, 291, 316, 317, 328, 334, 374, 388, 390, 392, 393, 401, 423, 451, 453, 460, 466, 491, 492, 493, 494, 495, 496, 497, 498, 501], "altern": 488, "alwai": [1, 84, 175, 217, 319, 493, 494], "am": 6, "among": 2, "amount": [6, 219, 336, 366, 493], "amus": 6, "an": [0, 1, 2, 3, 4, 6, 7, 9, 11, 16, 18, 31, 85, 86, 87, 92, 99, 100, 101, 102, 103, 104, 105, 121, 126, 127, 128, 137, 141, 143, 144, 145, 149, 163, 166, 173, 175, 178, 189, 195, 200, 223, 224, 229, 235, 236, 238, 241, 242, 243, 244, 253, 262, 263, 265, 266, 283, 286, 293, 295, 296, 299, 300, 304, 307, 312, 314, 315, 319, 320, 321, 322, 334, 347, 352, 355, 360, 361, 363, 370, 390, 391, 393, 395, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 424, 448, 461, 466, 467, 477, 481, 486, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "anaconda": 492, "anchor": 446, "angl": [116, 245, 362], "angular": [148, 398], "ani": [0, 1, 2, 6, 8, 19, 95, 113, 319, 320, 321, 322, 323, 334, 352, 370, 371, 374, 383, 393, 411, 412, 461, 483, 490, 491, 494, 496, 498, 499, 500], "anonym": 491, "anoth": [0, 94, 183, 210, 291, 313, 326, 334, 370, 491, 493, 494, 495, 500], "anwywher": 9, "anyhow": 6, "anymor": 6, "anyth": [6, 308, 496], "anytim": 496, "api": [1, 2, 141, 143, 175, 352, 492, 493, 494], "app": 9, "append": [6, 210, 491, 496], "appl": [2, 6, 8, 9, 500], "appli": [0, 39, 148, 149, 167, 197, 320, 321, 322, 334, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 348, 349, 350, 352, 353, 355, 356, 357, 358, 359, 361, 362, 363, 364, 365, 366, 367, 368, 369, 371, 384, 391, 393, 394, 395, 396, 397, 399, 401, 402, 404, 405, 406, 407, 408, 409, 410, 412, 421, 422, 423, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 461, 470, 471, 474, 477, 483, 486, 491, 492], "applic": [3, 9], "apply_fn": 371, "apply_gradi": 466, "apply_to_modul": [334, 375], "approach": [441, 494], "appropri": [2, 491], "approx": 352, "approxim": [17, 352, 423, 424, 425], "ar": [0, 1, 2, 5, 6, 7, 8, 9, 17, 19, 83, 91, 92, 94, 95, 102, 106, 113, 119, 126, 127, 137, 144, 146, 149, 151, 152, 154, 155, 157, 158, 160, 161, 163, 168, 169, 177, 178, 179, 180, 181, 182, 183, 184, 192, 193, 195, 196, 200, 210, 223, 237, 238, 239, 243, 244, 246, 247, 248, 253, 254, 257, 258, 265, 271, 272, 286, 287, 295, 308, 311, 312, 316, 319, 320, 326, 339, 341, 342, 343, 344, 345, 346, 347, 348, 349, 355, 359, 361, 363, 377, 390, 393, 412, 433, 435, 436, 460, 464, 466, 473, 475, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "arang": [0, 1, 195, 253, 326, 412, 495, 497], "arbitrari": [319, 464], "arbitrarili": [1, 95, 334, 490, 494, 498], "arc": 0, "arcco": 0, "arccosh": 0, "architectur": [6, 9, 216, 334, 389, 500], "archiv": 499, "arcsin": 0, "arcsinh": 0, "arctan": 0, "arctan2": 0, "arctanh": 0, "arg": [2, 6, 11, 19, 121, 137, 141, 142, 143, 162, 175, 271, 272, 324], "arg1": 183, "arg2": 183, "argmax": [0, 7], "argmin": 0, "argnam": [169, 308], "argnum": [2, 113, 169, 308, 494], "argpars": 6, "argpartit": 0, "argsort": 0, "argument": [1, 32, 66, 80, 95, 137, 169, 308, 320, 321, 322, 334, 412, 488, 492, 493, 494, 499, 500, 501], "argumentpars": 6, "ari": [85, 86, 87], "aris": 497, "arm": 9, "arm64": 9, "around": 6, "arr": [0, 268, 495], "arr_0": 499, "arrai": [0, 1, 2, 4, 6, 7, 8, 11, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 116, 117, 118, 119, 122, 123, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 318, 334, 339, 360, 370, 377, 380, 385, 391, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 458, 461, 464, 467, 468, 469, 470, 471, 472, 473, 478, 479, 480, 481, 482, 483, 484, 491, 492, 493, 494, 496, 497, 498, 500], "array_equ": [0, 17, 177], "arrayfir": 8, "arxiv": [6, 355, 359, 361, 369, 394, 425, 447, 467, 473], "as_strid": 0, "ascend": [192, 193], "ask": [6, 492], "assert": [1, 2, 146], "assign": [0, 2, 39, 464], "associ": [2, 271, 272, 496], "assum": [0, 2, 6, 91, 191, 192, 193, 196, 320, 334, 355], "astyp": [0, 1, 2, 6, 146, 370, 497], "atleast": 0, "atleast_1d": 0, "atleast_2d": 0, "atleast_3d": 0, "atol": [0, 17, 177], "atom": [1, 146], "atomic_fetch_add_explicit": 1, "atomic_output": [1, 146], "attach": 2, "attempt": 95, "attend": 390, "attent": [149, 375, 390, 403, 411], "attention_norm": 6, "attribut": [1, 10, 11, 12, 31, 162, 324, 383, 464, 486], "audio": 412, "auto": [0, 2, 4, 9, 493], "autom": 494, "automat": [1, 2, 8, 146, 200, 492, 498, 499, 500], "autoregress": 6, "avail": [2, 5, 6, 7, 9, 11, 125, 220, 328, 500], "averag": [336, 337, 338, 467, 468, 470, 471, 472, 492], "avgpool1d": 334, "avgpool2d": 334, "avgpool3d": 334, "avoid": [1, 2, 382, 491], "awai": [2, 6], "awar": [491, 496], "ax": [0, 2, 16, 18, 27, 28, 80, 113, 139, 151, 152, 154, 155, 157, 158, 160, 161, 163, 176, 195, 209, 211, 213, 227, 238, 241, 265, 279, 280, 281, 286, 288, 292, 293, 299, 303, 309, 494], "axes_a": 0, "axes_b": 0, "axi": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 72, 75, 76, 77, 81, 96, 109, 110, 111, 112, 119, 122, 139, 145, 147, 150, 153, 156, 157, 158, 159, 160, 161, 163, 172, 191, 195, 209, 211, 213, 227, 229, 238, 239, 241, 242, 247, 253, 262, 265, 281, 282, 283, 286, 287, 288, 292, 293, 295, 296, 300, 301, 302, 303, 307, 309, 310, 312, 336, 337, 338, 353, 366, 367, 368, 395, 426, 432, 434, 435, 439, 444, 446, 454, 455, 495], "axis1": [0, 47, 78, 119, 293, 302], "axis2": [0, 47, 78, 119, 293, 302], "axpbi": 2, "axpby_": 2, "axpby_gener": 2, "axpby_general_bfloat16": 2, "axpby_general_complex64": 2, "axpby_general_float16": 2, "axpby_general_float32": 2, "axpby_impl": 2, "axpby_impl_acceler": 2, "b": [0, 1, 2, 3, 4, 6, 14, 15, 17, 25, 83, 88, 89, 90, 91, 129, 130, 134, 146, 149, 165, 167, 170, 171, 175, 176, 177, 185, 186, 187, 188, 191, 195, 205, 206, 208, 210, 212, 228, 230, 234, 237, 240, 243, 250, 261, 264, 291, 299, 308, 321, 322, 353, 363, 395, 412, 426, 494, 495, 496, 497, 498, 499, 500], "b1": 167, "b2": 167, "b_": [354, 360], "b_stride": 1, "ba": [470, 472], "back": [6, 113, 220, 497], "backend": [1, 9, 124, 125], "backward": [1, 491, 494], "bad": 496, "balanc": 441, "baltimor": 195, "bandwidth": [491, 492], "bar": 492, "base": [0, 2, 4, 148, 195, 202, 204, 240, 398, 411, 464, 466, 472, 486, 488, 491, 495], "base_idx": 1, "basi": 486, "basic": [5, 266, 494], "batch": [6, 15, 91, 149, 167, 168, 210, 251, 339, 341, 342, 343, 344, 345, 346, 348, 349, 354, 360, 390, 395, 412, 496], "batch_idx": 1, "batch_iter": [7, 466], "batch_siz": [7, 466], "batchnorm": 334, "becaus": [6, 217, 334, 491, 493, 496], "been": [0, 2, 6, 218, 496], "befor": [1, 2, 6, 9, 29, 146, 239, 374, 411, 475, 492, 495, 496], "before_1": 238, "before_2": 238, "before_i": 238, "before_n": 238, "beforehand": 237, "beggin": 265, "begin": [84, 190, 219, 243, 354, 360, 409, 427, 438, 445, 451, 457, 458], "behav": 113, "behavior": [251, 441, 495, 496], "behaviour": [113, 189, 190], "behind": 494, "being": [289, 334], "bell": 2, "below": [2, 9, 195, 304, 306, 326, 412, 496], "bench": 2, "benchmark": [2, 491], "benefici": [348, 349, 496], "best": 492, "beta": [0, 2, 15, 117, 243, 339, 355, 359, 361, 445, 466, 470, 471, 472, 473], "beta_": 2, "beta_1": [468, 470, 471, 472, 473], "beta_2": [470, 471, 472, 473], "better": [494, 500], "between": [0, 2, 8, 94, 163, 411, 434, 437, 438, 441, 482, 492, 496, 497, 500], "beyond": [265, 480, 483], "bfloat16": [2, 12, 172, 326, 497], "bfloat16_t": 2, "bia": [6, 117, 145, 168, 243, 244, 320, 334, 341, 342, 343, 344, 345, 346, 354, 360, 361, 363, 375, 377, 387, 390, 393, 395, 470, 471, 472, 475, 494], "bias": [0, 117, 168, 243, 244, 354, 360, 375, 387, 390], "bias_correct": [470, 471], "bicub": 412, "big": [1, 491], "bigger": [6, 468], "bilinear": [1, 412], "binari": [200, 268, 269, 270, 271, 272, 310, 409, 433, 458, 491], "binary_cross_entropi": [334, 491], "bit": [0, 117, 168, 186, 243, 244, 264, 316, 326, 370, 392, 393, 394], "bitwis": [0, 88, 89, 90, 186, 264], "bitwise_and": 0, "bitwise_or": 0, "bitwise_xor": 0, "block": [0, 2, 6, 91, 411], "block_masked_mm": 0, "block_siz": [0, 91], "bn": 339, "bodi": [1, 146], "bool": [0, 1, 2, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 76, 77, 79, 81, 83, 95, 102, 109, 110, 111, 112, 124, 125, 141, 143, 146, 148, 168, 177, 183, 189, 190, 195, 198, 200, 209, 211, 213, 214, 220, 223, 227, 241, 244, 288, 292, 309, 316, 339, 341, 342, 343, 344, 345, 346, 354, 355, 359, 360, 361, 363, 370, 374, 375, 377, 382, 384, 387, 390, 393, 395, 398, 403, 411, 412, 433, 436, 468, 470, 471, 479], "bool_": [12, 326], "boolean": [0, 17, 83, 149, 177, 178, 179, 180, 181, 182, 183, 206, 207, 208, 326, 386, 495], "both": [1, 2, 14, 88, 89, 90, 129, 130, 134, 170, 171, 183, 186, 187, 188, 195, 205, 212, 228, 230, 234, 240, 247, 261, 264, 291, 316, 336, 337, 338, 359, 360, 366, 367, 368, 466, 491, 492, 493, 494, 498, 500], "bottom": 412, "bound": [0, 254, 257, 258, 352, 420, 491, 495, 500], "boundari": 482, "bracket": 6, "brain": 326, "break": 497, "bregler": 348, "broadcast": [0, 2, 14, 17, 88, 89, 90, 92, 94, 129, 130, 134, 149, 166, 170, 171, 177, 186, 187, 188, 205, 210, 212, 228, 230, 234, 240, 242, 246, 247, 251, 257, 258, 261, 264, 291, 296, 313, 390], "broadcast_arrai": [0, 2], "broadcast_to": 0, "broadcasted_input": 2, "brought": 8, "btl_tcp_link": 492, "buffer": [1, 2, 217, 497], "bui": 6, "build": [3, 4, 6, 8, 416, 464, 491, 493], "build_ext": [2, 9], "build_shared_lib": [2, 9], "built": [1, 2, 4, 9, 496], "bundl": 6, "byte": [52, 62, 217, 218, 219, 222, 223, 224, 326], "c": [0, 1, 2, 6, 15, 195, 339, 341, 342, 343, 344, 345, 346, 348, 349, 359, 360, 497, 498, 500], "c_": [360, 473], "c_in": [99, 100, 101, 102, 103, 104, 105], "c_out": [99, 100, 101, 102, 103, 104, 105], "c_pad": 1, "c_t": [360, 473], "cach": [6, 9, 215, 217, 218, 222, 491], "calcul": [195, 433, 436, 442, 468], "call": [2, 3, 6, 7, 32, 127, 165, 175, 215, 219, 334, 351, 375, 387, 392, 400, 464, 466, 475, 491, 492, 493, 494, 496], "callabl": [95, 113, 141, 143, 146, 169, 175, 184, 308, 311, 312, 316, 317, 319, 320, 321, 322, 370, 371, 374, 382, 395, 400, 411, 413, 414, 415, 416, 417, 418, 419, 420, 467, 468, 469, 470, 471, 472, 473, 478, 479, 480, 481, 482, 483, 484], "can": [1, 2, 3, 4, 6, 8, 9, 14, 19, 66, 80, 84, 88, 89, 90, 95, 119, 120, 121, 129, 130, 134, 137, 141, 142, 149, 170, 171, 175, 186, 187, 188, 195, 205, 212, 224, 228, 230, 234, 240, 246, 247, 254, 257, 258, 261, 264, 269, 291, 302, 307, 308, 322, 334, 337, 338, 351, 352, 367, 368, 374, 387, 392, 400, 412, 435, 461, 464, 466, 474, 475, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "cannot": [6, 94, 495, 497], "captur": [2, 3, 95, 113, 225, 226, 334, 491], "care": [6, 493, 496], "carefulli": [491, 493], "carri": 2, "cartesian": 214, "case": [2, 6, 122, 123, 124, 126, 127, 128, 152, 155, 156, 158, 159, 160, 161, 163, 189, 190, 191, 192, 193, 194, 196, 197, 198, 210, 263, 286, 307, 337, 338, 348, 367, 368, 409, 427, 445, 451, 457, 458, 474, 475, 491, 493, 494, 498, 499, 500, 501], "cast": [2, 38, 159, 160, 161, 200, 370, 382, 497], "caster": 2, "categor": 6, "categori": [12, 183, 326], "catlas_saxpbi": 2, "caus": [334, 491, 496], "causal": 6, "caution": 84, "cd": [3, 9], "cdf": [248, 352, 423], "cdot": [425, 434, 437, 453], "ceil": 0, "ceildiv": 1, "cell": 360, "celu": 334, "certain": [2, 384, 491], "chang": [84, 95, 141, 143, 175, 274, 310, 388, 393, 412, 438, 445, 491, 497], "channel": [1, 99, 100, 101, 102, 103, 104, 105, 339, 341, 342, 343, 344, 345, 346, 348, 349], "channel_idx": 1, "charact": 319, "check": [0, 2, 9, 83, 125, 183, 192, 193, 220, 377, 493, 494, 495], "checklist": 492, "checkout": [3, 491], "checkpoint": [411, 466], "chen": 473, "child": 389, "children": 334, "chip": 9, "choleski": 190, "choos": [6, 148, 398], "chosen": 132, "clamp": 163, "clang": 9, "clariti": 494, "class": [2, 6, 7, 10, 11, 12, 31, 113, 121, 162, 324, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 464, 467, 468, 469, 470, 471, 472, 473, 478, 479, 486], "class_pred": 316, "classif": [416, 417], "classifi": 7, "classmethod": [392, 393], "clear": 215, "click": 9, "clip": [0, 318, 433, 468], "clip_threshold": 468, "clipped_grad": 318, "clone": 9, "close": [5, 8, 9, 17, 177], "closer": 320, "cmake": [3, 4, 9], "cmake_arg": 3, "cmake_build_parallel_level": 9, "cmake_build_typ": 9, "cmake_current_list_dir": 2, "cmake_cxx_standard": 4, "cmake_cxx_standard_requir": 4, "cmake_host_system_processor": 9, "cmake_library_output_directori": 2, "cmake_minimum_requir": 4, "cmakebuild": 2, "cmakeextens": 2, "cmakelist": [2, 4], "cmdclass": 2, "co": [0, 2, 113, 403, 494], "code": [1, 146, 491, 492, 493, 496], "coeffici": [2, 467, 468, 470, 471, 472, 473], "col": 304, "col_contigu": 2, "cold": 9, "collect": [2, 320, 321, 490], "column": [2, 144, 173, 192, 243], "com": [9, 492], "combin": [6, 197, 322], "come": [2, 6, 492, 494], "command": [2, 3, 4, 9, 492], "command_buff": 2, "common": [2, 466, 491, 496], "commonli": [7, 388, 461, 491], "commun": [8, 121, 124, 125], "compar": [2, 83, 491], "comparison": [17, 134, 170, 171, 187, 188, 234], "compat": [6, 141, 143, 149, 175, 247, 251, 352, 499], "compil": [0, 3, 4, 8, 9, 120, 133, 146, 492, 493, 494, 496], "compiled_fun": [491, 493], "compiled_grad_fn": 491, "complet": [5, 6, 9, 223, 388, 389, 493, 494, 500], "complex": [2, 97, 98, 157, 158, 159, 160, 161, 174, 192, 193, 259, 319, 326, 334, 389, 491, 493, 494], "complex64": [2, 12, 326], "complex64_t": 2, "complexflo": 12, "compon": [2, 4, 6], "compos": [8, 334, 491, 494, 498], "composit": 498, "compress": 272, "compromis": 6, "comput": [0, 1, 2, 5, 6, 7, 8, 9, 109, 110, 111, 112, 113, 117, 132, 140, 148, 169, 184, 185, 189, 190, 191, 192, 193, 194, 195, 198, 205, 213, 237, 243, 261, 281, 288, 289, 299, 308, 309, 311, 317, 334, 339, 354, 355, 359, 360, 361, 375, 388, 393, 394, 398, 411, 414, 415, 416, 417, 424, 425, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 466, 467, 468, 470, 471, 472, 473, 477, 491, 492, 493, 494, 498, 500], "computation": 496, "compute_encod": 2, "concaten": [0, 6, 122], "concept": 464, "concis": 6, "concret": [2, 354, 360, 363, 395, 496, 500], "conda": [9, 492], "condit": [0, 2, 313, 491, 500], "config": [2, 4, 492], "configu": 466, "configur": [117, 492], "confirm": 492, "confus": 7, "conj": 98, "conjug": [0, 97], "connect": 492, "consecut": [148, 243, 398], "consequ": 6, "consid": [6, 17, 83, 177, 319, 320, 321, 355, 490], "consider": 491, "const": [0, 1, 2, 436], "constant": [0, 2, 6, 9, 113, 145, 147, 238, 334, 339, 355, 361, 394, 436, 446, 478, 480, 491, 493, 497], "constant_valu": 238, "constitut": 320, "construct": [0, 2, 7, 46, 118, 166, 235, 300, 314], "consum": 496, "contain": [2, 6, 9, 29, 30, 69, 95, 119, 132, 156, 157, 158, 167, 168, 192, 195, 206, 207, 208, 243, 283, 313, 318, 334, 374, 376, 377, 383, 411, 442, 461, 464, 491, 492, 494], "content": [9, 374, 491], "context": [290, 493], "contigu": [0, 1, 2, 84, 146], "continu": [340, 421, 494], "contract": [0, 132], "contribut": 2, "contriv": [494, 500], "control": [0, 362, 488, 496], "conv": 106, "conv1d": [0, 334], "conv2d": [0, 334], "conv3d": [0, 334], "conv_gener": 0, "conv_transpose1d": 0, "conv_transpose2d": 0, "conv_transpose3d": 0, "conveni": [1, 2, 7, 183], "convent": [19, 106, 131, 132, 412], "convers": 8, "convert": [0, 1, 2, 79, 85, 86, 87, 116, 163, 245, 392, 393, 496, 497, 498], "convolut": [0, 99, 100, 101, 102, 103, 104, 105, 106, 341, 342, 343, 344, 345, 346, 348, 349], "convolv": [99, 100, 101, 102, 103, 104, 105], "convtranspose1d": 334, "convtranspose2d": 334, "convtranspose3d": 334, "coordin": [0, 214], "copi": [0, 1, 2, 6, 8, 239, 282, 497], "copy_inplac": 2, "copytyp": 2, "core": [1, 2, 3, 4, 5, 6, 7, 316, 334, 336, 337, 338, 339, 359, 366, 367, 368, 377, 380, 382, 385, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 435, 442, 461, 464, 466, 491, 492, 497, 498], "corner": 412, "correct": [2, 9, 470, 471, 472, 495, 496], "correctli": 39, "correl": [102, 348], "correspond": [0, 1, 2, 16, 18, 79, 94, 117, 119, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 192, 209, 211, 227, 241, 279, 280, 292, 299, 307, 312, 320, 494], "cos_first": 403, "cosh": [0, 441], "cosin": [0, 20, 21, 107, 108, 434, 480, 482, 494], "cosine_decai": [466, 482], "cosine_similarity_loss": 334, "cost": [9, 468, 492, 496], "costli": 496, "cot": 1, "cot_index": 1, "cotan": [2, 113], "cotang": [1, 2, 113, 311], "could": [6, 334], "count": [334, 482], "counter": 488, "cours": 494, "coursera": 478, "cout": [4, 493], "cov": 251, "covari": [251, 339], "cover": 2, "cpp": [2, 4], "cpu": [8, 9, 192, 193, 196, 500], "cpython": 2, "crash": [84, 491], "creat": [0, 2, 6, 9, 84, 124, 144, 173, 290, 334, 464, 466, 482, 491, 493, 495, 497], "create_additive_causal_mask": 6, "criteria": 2, "cross": [7, 102, 433, 435], "cross_entropi": [7, 334], "crowd": 6, "cry": 6, "cubic": 412, "cummax": 0, "cummin": 0, "cumprod": 0, "cumsum": 0, "cumul": [0, 84, 109, 110, 111, 112], "current": [6, 8, 9, 84, 91, 101, 104, 105, 128, 216, 218, 243, 322, 334, 468, 492, 496], "custom": [8, 113, 146, 411], "custom_decod": 411, "custom_encod": 411, "custom_funct": 1, "custom_kernel_myexp_float": 1, "custom_tim": 2, "cvpr": 348, "cxx": 4, "cycl": 490, "d": [0, 1, 2, 6, 101, 105, 118, 119, 149, 176, 195, 210, 214, 237, 295, 302, 304, 305, 306, 323, 343, 346, 349, 354, 360, 395, 467, 470, 472, 500], "d1": 500, "d2": 500, "d2fdx2": 494, "d_i": 363, "dampen": 479, "darwin": 2, "data": [0, 2, 7, 8, 11, 19, 126, 144, 159, 160, 166, 173, 199, 231, 235, 248, 257, 302, 304, 310, 314, 349, 413, 414, 415, 416, 417, 418, 419, 420, 491, 492, 493, 495, 497], "dataset": [5, 492, 496], "datatyp": 52, "dbuild_shared_lib": 9, "dcmake_build_typ": [4, 9], "ddof": [0, 76, 81, 288, 309], "deal": 491, "debug": [1, 3, 492], "debugg": 8, "decai": [468, 471, 473, 479, 480, 481, 484], "decay_r": [468, 481, 484], "decay_step": 480, "decent": 7, "decid": [320, 374], "decim": [0, 67, 266], "declar": 2, "decltyp": 1, "decod": 411, "decomposit": [189, 190, 197], "decor": [1, 113], "decoupl": 471, "deep": [339, 414, 415, 416, 417], "def": [1, 2, 5, 6, 7, 113, 141, 143, 146, 308, 334, 464, 491, 492, 493, 494, 495, 496, 497, 500], "default": [1, 2, 9, 15, 16, 17, 18, 19, 27, 28, 29, 30, 83, 84, 91, 95, 96, 99, 100, 101, 102, 103, 104, 105, 113, 114, 115, 117, 118, 119, 122, 123, 124, 126, 127, 128, 141, 143, 144, 146, 148, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 167, 168, 169, 172, 173, 177, 185, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 209, 211, 213, 214, 222, 223, 224, 227, 231, 235, 238, 239, 241, 243, 244, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 262, 263, 266, 273, 274, 282, 283, 286, 287, 288, 290, 292, 294, 299, 301, 302, 303, 304, 305, 306, 307, 308, 309, 312, 314, 316, 326, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 350, 353, 354, 356, 359, 360, 362, 363, 366, 367, 368, 370, 375, 377, 382, 384, 387, 390, 391, 392, 393, 395, 398, 403, 407, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 464, 467, 468, 469, 470, 471, 472, 473, 478, 479, 480, 488, 490, 491, 493, 494, 497, 499, 501], "default_devic": 501, "default_stream": 501, "defin": [1, 2, 5, 6, 7, 9, 113, 127, 146, 168, 191, 195, 244, 316, 319, 497], "definit": [113, 189, 190, 251], "degre": [0, 245, 446], "delta": [438, 467], "delv": [416, 417], "demonstr": 497, "denomin": [359, 434, 467, 469, 470, 471, 472, 478], "dens": [214, 500], "depend": [0, 2, 3, 4, 5, 9, 79, 195, 354, 360, 395, 491, 492, 495, 499, 500], "depth": [319, 338, 343, 346, 349, 368, 494], "dequant": [0, 243], "deriv": [2, 493, 494, 496], "descend": 372, "descent": [479, 491, 496], "describ": [2, 496], "descript": [2, 4, 6, 326], "design": [1, 5, 8, 488, 500], "destin": [0, 2, 61, 128, 229, 242], "destroi": 491, "detach": 494, "detail": [1, 2, 11, 222, 334, 348, 398, 403, 412, 414, 415, 416, 417, 467, 469, 470, 472, 473, 495, 498], "detect": 491, "determin": [0, 2, 119, 251, 322, 326, 381, 499], "dev": [2, 9], "develop": [2, 4, 9], "developer_dir": 9, "deviat": [0, 252, 288, 414, 416, 419], "deviatoin": 0, "devic": [1, 2, 8, 9, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 114, 115, 116, 117, 118, 119, 122, 123, 126, 127, 128, 129, 130, 131, 134, 135, 136, 138, 139, 140, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 170, 171, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 216, 223, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 309, 310, 313, 314, 315, 324, 500, 501], "device_info": 224, "devicetyp": 10, "df": 497, "dfdx": [493, 494, 495], "dft": [150, 151, 152, 153, 154, 155, 159, 160, 161], "dhwc": 349, "diag": [0, 197], "diagon": [0, 46, 118, 144, 302, 304, 305, 306], "dict": [95, 137, 142, 200, 216, 269, 270, 271, 316, 318, 380, 385, 388, 389, 464, 466, 474, 475, 477, 490, 493, 494, 499], "dict_kei": [320, 475], "dictionari": [6, 95, 141, 175, 200, 216, 269, 270, 318, 319, 322, 334, 374, 383, 388, 389, 476, 490, 499], "did": 6, "diff": 2, "differ": [8, 183, 291, 310, 445, 491, 493, 494], "differenti": [1, 2, 8, 340, 421], "difficult": 494, "difficulti": [414, 415], "dilat": [0, 99, 100, 101, 102, 103, 104, 105, 341, 342, 344, 345], "dim": [1, 6, 148, 149, 351, 355, 359, 361, 390, 392, 394, 398, 403, 411], "dimens": [0, 1, 2, 6, 16, 18, 27, 28, 63, 69, 79, 85, 86, 87, 95, 100, 101, 102, 104, 105, 119, 139, 148, 149, 157, 158, 160, 161, 163, 167, 168, 176, 189, 190, 192, 193, 194, 195, 196, 197, 198, 209, 210, 211, 213, 227, 241, 242, 243, 247, 256, 288, 292, 296, 299, 303, 309, 339, 341, 342, 343, 344, 345, 346, 348, 349, 353, 354, 355, 359, 360, 361, 390, 394, 395, 398, 411, 412, 426, 435, 491, 494], "dimension": [31, 145, 147, 150, 151, 152, 153, 154, 155, 159, 160, 161, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 351, 363, 366, 367, 368, 392, 393, 403, 495, 497], "dir": 4, "direct": [2, 6, 372, 473, 500], "directli": [2, 6, 84], "directori": [2, 4, 6, 9], "disabl": [120, 222, 491], "disable_compil": 491, "disappoint": 6, "discard": [6, 319], "discov": 9, "discoveri": 473, "discret": [106, 150, 151, 152, 153, 154, 155, 159, 160, 161, 351, 392], "discuss": 2, "disk": 6, "dispatch": 2, "dispatch_thread": 2, "dispatchthread": 1, "displai": 334, "distanc": [6, 446], "distribut": [8, 9, 246, 247, 248, 250, 251, 252, 257, 258, 363, 414, 415, 416, 417, 419, 420, 436, 439, 444, 446, 461], "diverg": 439, "divid": [0, 2, 39, 165, 243, 261, 492], "divis": [0, 129, 165, 243, 261], "divisor": [288, 309], "divmod": 0, "dloss_dw": 494, "dloss_dx": 494, "dlpack": 497, "dlvalu": 308, "dmlx_build_cpu": 9, "dmlx_build_gguf": 9, "dmlx_build_safetensor": 9, "dmlx_metal_debug": 3, "dmlx_metal_jit": 9, "do": [0, 2, 6, 9, 310, 334, 376, 387, 461, 464, 491, 492, 493, 494, 496], "doc": [2, 7, 492], "document": [2, 3, 4, 66, 80, 146, 269, 270, 326, 491, 493, 494, 495], "doe": [0, 2, 3, 6, 9, 217, 310, 318, 334, 491, 495, 496, 497], "doesn": [2, 334, 493], "domain": [257, 492], "don": [1, 9, 491, 500], "done": [334, 347, 394, 491, 492, 496, 497], "dot": [142, 189, 194, 198, 299, 319, 379, 390], "doubl": [0, 6], "doubt": 6, "down": [6, 318], "downsampl": [336, 337, 338, 366, 367, 368], "dparam": 308, "draw": 247, "drop": 374, "dropout": [334, 348, 349, 384, 411, 491], "dropout2d": 334, "dropout3d": 334, "dst": 128, "dt": 135, "dtype": [0, 1, 2, 6, 12, 19, 31, 38, 39, 79, 82, 126, 127, 144, 146, 162, 163, 166, 173, 183, 185, 192, 193, 195, 196, 199, 235, 248, 250, 251, 252, 254, 257, 258, 279, 280, 302, 304, 307, 310, 314, 326, 382, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 435, 442, 480, 481, 482, 483, 484, 491, 492, 493, 494, 495, 497, 498, 499], "dtypecategori": [183, 326], "dual": 441, "duchi": 469, "duplic": 493, "dure": [3, 95, 347, 348, 349, 412, 497], "dx": 113, "dy": 113, "dyld": 492, "dyld_library_path": 492, "dylib": 2, "dynam": [0, 493, 496], "e": [2, 7, 9, 113, 135, 146, 167, 168, 184, 275, 339, 341, 342, 343, 344, 345, 346, 348, 349, 355, 359, 361, 375, 394, 431, 432, 454, 455, 460, 466, 469, 491, 493, 496, 501], "e5": 326, "e8": 326, "each": [0, 1, 2, 69, 117, 137, 148, 168, 183, 189, 190, 192, 193, 194, 197, 198, 210, 214, 238, 243, 244, 247, 262, 271, 272, 283, 300, 303, 310, 312, 313, 348, 349, 351, 354, 355, 360, 395, 398, 411, 433, 435, 488, 491, 492, 493, 496], "eager": 496, "earli": 348, "earlier": 2, "eas": 6, "easi": [2, 334, 492], "easier": [1, 142, 496], "edg": [94, 238, 412, 491], "edit": [9, 389], "effect": [348, 491, 496], "effici": [6, 8, 167, 348, 398, 496, 498], "eigenvalu": [192, 193], "eigenvector": 192, "einstein": [131, 132], "einsum": 132, "either": [9, 14, 66, 79, 80, 88, 89, 90, 94, 129, 130, 134, 165, 170, 171, 175, 186, 187, 188, 195, 205, 210, 212, 228, 230, 234, 240, 261, 264, 291, 308, 337, 338, 367, 368, 400, 412, 416, 417, 497], "elem": [1, 146], "elem_to_loc": [1, 2], "element": [0, 1, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 29, 71, 84, 88, 89, 90, 93, 107, 108, 109, 110, 111, 112, 117, 129, 130, 134, 135, 136, 138, 140, 144, 164, 165, 168, 170, 171, 177, 178, 179, 180, 181, 182, 186, 187, 188, 201, 202, 203, 204, 205, 206, 207, 208, 212, 214, 228, 230, 232, 234, 239, 240, 243, 244, 260, 261, 262, 264, 265, 267, 275, 276, 277, 278, 284, 285, 291, 295, 297, 298, 301, 308, 310, 313, 340, 347, 348, 349, 354, 358, 360, 369, 391, 395, 398, 402, 421, 428, 429, 431, 432, 447, 448, 450, 453, 454, 455, 456, 491, 494], "elementwis": [1, 97, 98], "elif": 6, "ellipsi": 495, "elman": 395, "els": [0, 2, 6, 334, 375, 492, 496], "elsewher": [304, 495], "elu": [334, 451], "emb": [6, 351, 392, 403], "embed": [6, 316, 334, 392, 398, 403, 434], "empti": 251, "enabl": [3, 6, 9, 95, 133, 479], "enclos": 493, "encod": [2, 148, 398, 403, 411, 435], "encount": [2, 494], "end": [119, 190, 220, 243, 265, 354, 360, 409, 427, 438, 445, 451, 457, 458, 480, 483, 493], "end_axi": [0, 50, 163], "end_encod": 2, "endif": 2, "endl": [4, 493], "endswith": 375, "enhanc": [6, 398, 496], "enjoi": 2, "enough": [2, 496], "ensur": [0, 1, 2, 9, 146, 318, 441, 492, 493], "ensure_row_contigu": [1, 146], "enter": 6, "entir": [16, 18, 27, 28, 209, 211, 213, 227, 241, 288, 292, 309, 348, 349], "entri": [0, 253, 307, 348, 349], "entropi": [7, 433, 435], "enumer": 334, "environ": [9, 120, 133, 492], "ep": [5, 145, 147, 339, 355, 359, 361, 394, 434, 436, 446, 466, 467, 468, 469, 470, 471, 472, 478], "epoch": 7, "epsilon": [339, 355, 359, 361, 394, 434, 436, 467, 469, 470, 471, 472, 478], "epsilon_1": 468, "epsilon_2": 468, "equal": [0, 1, 17, 29, 83, 144, 171, 177, 188, 234, 239, 254, 283, 359, 363], "equal_nan": [0, 17, 83, 177], "equat": [131, 132], "equival": [0, 2, 32, 66, 80, 127, 130, 165, 168, 172, 295, 340, 350, 352, 356, 357, 358, 364, 365, 389, 391, 393, 396, 397, 399, 401, 404, 405, 406, 407, 408, 410], "erf": [0, 136, 491], "erfinv": 0, "error": [0, 2, 9, 124, 135, 136, 223, 224, 283, 352, 423, 424, 425, 441, 443, 491, 494, 497], "error_norm": 5, "estim": 472, "eta": 473, "etc": [2, 243, 334, 412, 492], "eval": [2, 3, 5, 6, 7, 334, 464, 466, 491, 492, 493, 494, 496, 498], "eval_cpu": 2, "eval_fn": 7, "eval_gpu": 2, "evalu": [2, 6, 7, 8, 128, 137, 184, 311, 334, 373, 384, 464, 466, 491, 493, 498], "even": [1, 2, 6, 95, 491, 493, 496, 497], "evenli": [0, 199], "everi": [243, 320, 466, 484, 494], "everyth": [6, 492], "everywher": 0, "exact": [424, 425], "exactli": [2, 6, 148, 377, 494], "exampl": [0, 3, 4, 5, 6, 7, 9, 19, 39, 113, 141, 142, 143, 146, 149, 163, 175, 183, 185, 192, 193, 195, 196, 279, 280, 290, 295, 307, 318, 321, 322, 334, 336, 337, 338, 339, 359, 366, 367, 368, 375, 377, 384, 387, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 435, 442, 461, 466, 475, 480, 481, 482, 483, 484, 488, 494, 495, 496, 497, 498, 499], "exce": 318, "exceed": 223, "except": [8, 113, 144, 156, 157, 159, 160, 161, 355, 377, 493, 495, 497], "exclud": [242, 296], "exclus": [0, 84, 90], "execut": [2, 4, 9, 85, 86, 87, 185, 219, 497, 500], "execute_process": 4, "exist": [2, 3, 6, 375, 387], "exp": [0, 1, 140, 146, 205, 209, 248, 281, 340, 350, 402, 421, 422, 439, 451, 452, 456, 491, 493, 500], "exp_elementwis": [1, 146], "expand_dim": 0, "expect": [2, 6, 341, 342, 343, 344, 345, 346, 347, 348, 349, 403, 411, 436, 491, 495], "expens": 411, "expensive_fun": 496, "experiment": [141, 143, 175, 497], "explain": 2, "explicit": [2, 475, 488, 497], "explicitli": [167, 334, 488], "explor": 9, "expm1": 0, "exponenti": [0, 138, 140, 340, 350, 399, 421, 422, 451, 481], "exponential_decai": 466, "export": [8, 9, 141, 142, 175], "export_funct": 493, "ext_modul": 2, "extend": [2, 238], "extens": [8, 200, 225, 381, 499], "extern": 497, "extra": [1, 320, 321, 493], "extract": [0, 6, 46, 118, 119, 279, 334, 374, 464, 492], "extras_requir": 2, "extrem": [495, 496], "ey": [0, 6, 194, 198], "f": [0, 2, 5, 7, 113, 195, 334, 360, 471, 491, 497], "f_jvp": 113, "f_t": 360, "f_vjp": 113, "f_vmap": 113, "face": 6, "factor": [2, 15, 172, 189, 190, 196, 412, 435, 481, 484], "fail": 491, "fall": [2, 113], "fallback": 2, "fals": [0, 1, 2, 6, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 76, 77, 81, 83, 95, 102, 109, 110, 111, 112, 124, 141, 143, 146, 177, 183, 189, 190, 195, 198, 200, 209, 211, 213, 214, 223, 227, 241, 288, 292, 309, 313, 316, 319, 320, 321, 322, 326, 355, 359, 361, 363, 375, 377, 387, 390, 393, 398, 403, 411, 412, 433, 436, 468, 470, 471, 479, 493, 497], "famili": 6, "fan": [414, 415, 416, 417], "fan_in": [414, 415, 416, 417], "fan_out": [414, 415, 416, 417], "far": 466, "fast": [1, 8, 352, 425, 492, 500], "faster": [1, 2, 9, 130, 423, 433, 491, 494], "featur": [1, 8, 99, 100, 101, 102, 103, 104, 105, 148, 339, 354, 355, 359, 360, 361, 363, 393, 394, 395, 398, 411, 412, 491, 496], "feed": 6, "feed_forward": 6, "feedforward": [414, 415], "feel": 6, "fetch": 1, "few": [1, 2, 6, 7, 8, 9, 492, 493, 496, 498], "ffn": 6, "ffn_norm": 6, "fft": 8, "figur": 492, "file": [4, 6, 9, 141, 142, 143, 175, 200, 268, 269, 270, 271, 272, 377, 381, 492, 493, 494, 499], "file_or_weight": 377, "fill": [0, 2, 166, 236, 304, 315, 413, 414, 415, 416, 417, 419, 420], "filter": [0, 106, 341, 342, 343, 344, 345, 346, 370, 374], "filter_and_map": 334, "filter_fn": [370, 374], "final": [2, 4, 5, 6, 7, 172, 480, 483], "find": [2, 4, 5, 9, 492], "find_packag": [2, 4], "finder": 9, "fine": [488, 493, 496], "finetun": 334, "finish": 2, "finit": [0, 178, 231], "first": [0, 1, 2, 3, 4, 5, 6, 7, 9, 119, 122, 163, 169, 183, 185, 186, 197, 206, 208, 210, 239, 256, 264, 293, 299, 302, 308, 319, 321, 322, 334, 337, 338, 355, 367, 368, 412, 434, 442, 468, 472, 475, 491, 493, 494, 497, 500], "first_lay": 496, "firt": 491, "fit": [2, 243, 500], "five": 491, "fix": [2, 6, 9, 491, 496], "flag": [2, 4, 9, 491, 497], "flat": [167, 168, 319, 323], "flat_param": 271, "flatten": [0, 29, 30, 109, 110, 111, 112, 195, 237, 239, 242, 262, 265, 282, 295, 296, 301, 319, 491], "flexibl": 8, "flexibli": 389, "flip": [0, 102, 106], "float": [0, 1, 2, 12, 15, 17, 19, 79, 145, 146, 147, 148, 149, 162, 165, 166, 172, 177, 183, 195, 231, 244, 246, 250, 252, 318, 326, 339, 347, 348, 349, 355, 359, 361, 370, 382, 394, 398, 403, 409, 411, 412, 413, 414, 415, 416, 417, 419, 420, 434, 435, 436, 438, 442, 445, 446, 457, 458, 467, 468, 469, 470, 471, 472, 473, 478, 479, 480, 481, 483, 484], "float16": [1, 2, 12, 146, 172, 200, 326, 370, 496, 497], "float16_t": [1, 2], "float32": [0, 1, 2, 12, 19, 144, 146, 149, 172, 173, 183, 192, 193, 195, 196, 199, 235, 248, 250, 251, 252, 257, 258, 280, 304, 314, 326, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 435, 442, 480, 481, 482, 483, 484, 491, 492, 493, 494, 495, 496, 497, 498, 499], "float64": 183, "floor": [0, 1, 165], "floor_divid": 0, "flow": [0, 289, 496], "flush": 2, "fn": [175, 317, 320, 321, 322, 498], "follow": [1, 2, 4, 6, 7, 8, 9, 19, 106, 117, 149, 167, 195, 238, 243, 321, 334, 424, 425, 439, 467, 468, 469, 472, 473, 479, 488, 491, 492, 493, 494, 500], "foo": 492, "food": 6, "forc": [6, 7, 334, 492, 498], "forg": 9, "formal": [117, 243], "format": [6, 142, 200, 268, 269, 270, 271, 272, 497], "formul": [340, 350], "formula": 445, "forth": 412, "forward": [1, 2, 308, 491, 496], "found": [4, 374], "four": 339, "fourier": [150, 151, 152, 153, 154, 155, 159, 160, 161], "fourth": 493, "frac": [135, 243, 275, 339, 347, 348, 349, 355, 359, 361, 363, 394, 402, 414, 415, 416, 417, 434, 436, 438, 441, 452, 454, 455, 467, 469, 470, 471, 472, 478], "fraction": 19, "framework": [2, 8], "free": 222, "freez": [334, 387, 464], "freq": 148, "frequenc": [148, 398, 403], "frequent": [491, 496], "friend": 6, "fro": 195, "frobeniu": 195, "from": [0, 1, 2, 4, 6, 7, 8, 84, 116, 117, 119, 122, 123, 126, 127, 128, 146, 157, 158, 160, 161, 166, 167, 172, 175, 195, 200, 210, 214, 219, 222, 236, 243, 245, 246, 247, 248, 249, 250, 254, 257, 271, 279, 286, 289, 291, 295, 296, 301, 302, 313, 315, 319, 320, 321, 322, 323, 334, 363, 375, 377, 390, 414, 415, 416, 417, 419, 420, 436, 445, 461, 466, 490, 491, 492, 493, 494, 496, 497, 498, 499, 500], "from_embed": 392, "from_linear": 393, "front": [2, 493], "frozen": [334, 375, 385, 387, 393, 464], "fuction": 130, "full": [0, 1, 2, 7, 66, 80, 106, 146, 281, 388, 389, 436, 491, 492, 493, 496], "full_turn": 403, "fulli": [2, 8, 492, 497, 500], "fun": [95, 141, 143, 169, 184, 308, 311, 312, 491, 493, 495, 496, 500], "fun1": 496, "func": 395, "function": [0, 1, 2, 3, 5, 6, 7, 8, 17, 19, 84, 95, 113, 130, 135, 136, 141, 143, 146, 169, 175, 177, 184, 189, 190, 192, 193, 194, 195, 196, 197, 198, 210, 224, 275, 308, 311, 312, 317, 318, 320, 321, 322, 334, 340, 350, 352, 353, 356, 357, 358, 364, 365, 369, 371, 375, 382, 387, 391, 395, 396, 397, 399, 400, 401, 402, 404, 405, 406, 407, 408, 409, 410, 411, 423, 424, 425, 426, 427, 428, 429, 431, 432, 433, 447, 452, 454, 455, 456, 457, 458, 459, 461, 466, 475, 488, 490, 492, 495, 496, 497, 499], "functionexport": 143, "functool": 491, "further": [2, 9, 494], "fuse": [1, 491], "fusibl": 491, "futur": [6, 141, 143, 175, 393, 495, 496], "fx": 113, "g": [3, 9, 113, 146, 195, 243, 360, 460, 478, 479, 493, 496, 501], "g_t": [360, 467, 469, 470, 471, 472, 473, 478, 479], "gain": [414, 415, 416, 417], "gamma": [339, 355, 359, 361, 394, 414, 415, 416, 417], "gap": 1, "gate": [353, 354, 426], "gather": [0, 122, 167, 168], "gather_mm": [0, 168], "gather_qmm": 0, "gaurante": 310, "gaussian": [5, 352, 423, 424, 425, 436], "gaussian_nll_loss": 334, "gelu": [334, 424, 425, 491], "gelu_approx": [334, 352, 423], "gelu_fast_approx": [334, 352, 423], "geluapprox": 352, "gelufast": 352, "gener": [0, 1, 2, 3, 5, 12, 19, 102, 144, 146, 157, 158, 199, 214, 246, 251, 252, 253, 254, 257, 258, 411, 488, 491, 495, 496, 501], "general_": 2, "generate_stub": 9, "geq": [409, 458], "get": [2, 5, 7, 9, 100, 101, 102, 104, 105, 114, 115, 162, 216, 217, 218, 219, 249, 334, 491, 493, 494, 496, 500], "get_cache_memori": 215, "get_command_encod": 2, "get_kernel": 2, "gguf": [9, 200, 269, 499], "gh": 1, "gii": 1, "git": 9, "github": [5, 7, 9, 491], "give": [2, 6, 7, 29, 491], "given": [0, 2, 9, 16, 18, 29, 39, 84, 92, 94, 96, 109, 110, 111, 112, 117, 119, 132, 137, 139, 149, 150, 151, 152, 153, 154, 155, 159, 160, 161, 166, 167, 195, 209, 211, 213, 222, 227, 231, 233, 241, 251, 253, 254, 265, 266, 274, 281, 283, 288, 292, 294, 300, 301, 302, 304, 305, 306, 309, 324, 347, 374, 390, 434, 436, 442], "gix": 1, "gix_mult": 1, "giy_mult": 1, "global": [120, 122, 123, 124, 126, 127, 128, 133, 255, 318, 488, 491], "glorot": [414, 415], "glorot_norm": 334, "glorot_uniform": 334, "glu": [6, 334], "gm": 1, "gn": 1, "go": [2, 6, 494], "golub": 195, "good": [2, 9, 466, 491, 492, 500], "goroshin": 348, "gower": 6, "gpu": [1, 3, 8, 9, 216, 495, 500], "gputrac": [3, 225], "grad": [2, 5, 7, 113, 308, 318, 466, 474, 491, 492, 493, 494, 495, 496, 498], "grad_fn": [5, 491, 494], "gradient": [0, 5, 7, 113, 169, 289, 308, 317, 318, 334, 375, 388, 393, 411, 441, 464, 466, 467, 468, 470, 471, 472, 473, 474, 477, 479, 491, 492, 494, 495, 496, 497, 498], "grain": 488, "graph": [2, 6, 7, 8, 142, 493, 494], "great": 3, "greater": [0, 6, 29, 140, 171, 239, 318, 409, 458], "greater_equ": 0, "grep": 9, "grid": [2, 146, 214], "grid_dim": 2, "grid_grad": 1, "grid_idx": 1, "grid_sampl": 1, "grid_sample_grad": 1, "grid_sample_ref": 1, "grid_sample_vjp": 1, "grid_shap": 1, "grid_siz": 1, "ground": [5, 6, 435, 445], "group": [0, 1, 99, 100, 101, 102, 103, 104, 105, 117, 122, 123, 124, 126, 127, 128, 149, 168, 243, 244, 310, 316, 341, 342, 355, 392, 393, 492], "group_dim": 2, "group_siz": [0, 117, 168, 243, 244, 316, 392, 393], "groupnorm": 334, "grow": 496, "gru": 334, "guid": [2, 4, 8, 493], "gw": 1, "h": [1, 2, 4, 99, 100, 101, 103, 104, 105, 195, 339, 342, 343, 345, 346, 348, 349, 354, 360, 395, 494, 496], "h_": [354, 360, 395], "h_in": 1, "h_stride": 1, "h_t": [354, 360, 395], "ha": [2, 3, 6, 7, 8, 9, 79, 95, 119, 128, 156, 157, 159, 160, 161, 169, 189, 190, 192, 193, 194, 197, 198, 214, 218, 247, 339, 354, 360, 363, 395, 464, 466, 491, 493, 495, 496, 498, 500], "had": 6, "hadamard": [0, 172], "hadamard_transform": 0, "half": [2, 19, 254, 258, 398, 496], "halv": [353, 426], "hand": [6, 494, 496], "handi": 494, "handl": [2, 334, 491], "happen": [2, 6, 145, 411, 466, 491, 496], "happi": 6, "hard": 6, "hard_shrink": [334, 356], "hard_tanh": [334, 357], "hardcod": 491, "hardshrink": [334, 427], "hardswish": 334, "hardtanh": [334, 428], "hat": [117, 243], "have": [0, 1, 2, 6, 9, 17, 83, 85, 86, 87, 91, 113, 122, 149, 157, 158, 160, 161, 168, 177, 210, 225, 247, 310, 319, 360, 390, 400, 473, 475, 490, 491, 492, 493, 495, 496, 500], "haven": 6, "hazan": 469, "he": [6, 416, 417], "he_norm": 334, "he_uniform": 334, "head": [149, 390, 411], "header": [2, 146], "heart": 6, "heavi": 6, "height": [337, 338, 339, 342, 343, 345, 346, 348, 349, 367, 368], "hello": [319, 323], "help": [2, 6, 491, 500], "helper": [6, 146, 491], "henc": [0, 2, 243, 491], "hendryck": 425, "here": [2, 6, 466, 491, 493, 494, 496, 499, 500], "hermitian": [192, 193], "hf": 360, "hg": 360, "hh": 395, "hi": [6, 360], "hidden": [354, 360, 395, 411], "hidden_dim": [7, 464, 466], "hidden_s": [354, 360, 395], "hierarchi": 326, "high": [254, 258, 334, 351, 420, 461], "high_pad_s": 0, "higher": [2, 176, 224, 442, 494], "highli": 9, "him": 6, "hing": 437, "hinge_loss": 334, "hinton": 478, "hit": 2, "hn": 354, "ho": 360, "hold": [2, 6, 11, 12, 195, 491], "homebrew": 492, "hopkin": 195, "host": 2, "host1": 492, "host2": 492, "host_nam": 1, "hostfil": 492, "hostnam": 492, "hot": 435, "hour": 6, "how": [2, 4, 6, 7, 334, 336, 337, 338, 341, 342, 343, 344, 345, 346, 351, 366, 367, 368, 392, 412, 474, 491, 495, 500], "howev": [2, 113, 334, 352, 355, 475, 488, 491, 492, 496, 497], "hr": 354, "http": [355, 359, 361, 369, 394, 425, 447], "huber": 438, "huber_loss": 334, "human": [416, 417], "hundr": 9, "hurri": 6, "hutter": 471, "hyperbol": [0, 21, 23, 26, 108, 278, 298, 410, 459], "hz": 354, "i": [0, 1, 2, 3, 4, 6, 7, 8, 9, 17, 19, 29, 38, 79, 84, 94, 100, 101, 102, 104, 105, 106, 109, 110, 111, 112, 113, 118, 119, 122, 123, 125, 126, 127, 128, 130, 137, 141, 143, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 165, 166, 167, 168, 172, 175, 177, 178, 183, 184, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 200, 205, 209, 210, 214, 220, 223, 224, 238, 239, 242, 243, 244, 251, 252, 253, 263, 265, 268, 269, 270, 275, 281, 283, 288, 289, 294, 295, 296, 299, 302, 303, 307, 308, 309, 310, 311, 312, 313, 316, 318, 319, 320, 321, 322, 326, 328, 334, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 347, 348, 349, 352, 354, 355, 359, 360, 361, 363, 366, 367, 368, 374, 375, 381, 383, 384, 386, 387, 389, 390, 391, 393, 394, 395, 398, 403, 409, 411, 412, 416, 417, 423, 425, 433, 434, 436, 441, 442, 445, 446, 448, 453, 458, 464, 466, 468, 470, 471, 473, 474, 475, 480, 482, 483, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "i386": 9, "i_n": 1, "i_nw": 1, "i_s": 1, "i_sw": 1, "i_t": 360, "iclr": [470, 471, 472], "id": [7, 9], "idea": [494, 496], "idempot": [375, 387], "ident": [0, 113, 128, 144, 289, 334, 384], "identifi": [2, 319, 490], "idim": 7, "idiom": [7, 491], "idx": [39, 495], "ie": [387, 492], "ieee": 326, "ignor": [6, 39, 94, 95, 137, 468], "ih": 395, "ii": 1, "ij": 214, "imag": [0, 342, 343, 345, 346, 348, 349, 412], "imagenet": [416, 417], "imaginari": 174, "immedi": [6, 370], "implement": [0, 1, 5, 7, 148, 149, 195, 351, 374, 390, 398, 400, 403, 409, 411, 412, 458, 467, 468, 469, 472, 473, 474, 486, 491, 494], "impli": 310, "implicit": [488, 491, 494], "implicitli": 496, "import": [2, 3, 5, 6, 7, 9, 113, 172, 175, 195, 271, 308, 319, 320, 321, 322, 323, 334, 336, 337, 338, 339, 359, 366, 367, 368, 377, 412, 433, 435, 442, 461, 464, 466, 491, 492, 494, 495, 496, 497, 498], "import_funct": 493, "imported_ab": 493, "imported_fun": 493, "imported_funct": 493, "improv": [1, 2, 3, 6, 433, 467, 468, 469, 470, 471, 472, 478, 491, 492], "in_ax": [312, 494], "in_channel": [341, 342, 343, 344, 345, 346], "in_dim": [334, 464], "in_proj": 464, "inci": 2, "includ": [1, 2, 4, 109, 110, 111, 112, 142, 146, 217, 218, 223, 361, 371, 383, 393, 436, 466, 491, 493, 494, 495, 498, 499, 501], "include_dir": 2, "inclus": [0, 42, 43, 44, 45, 109, 110, 111, 112, 163], "incom": 2, "inconveni": 491, "incorpor": 497, "incorrect": 497, "increas": 224, "increment": 19, "incur": [6, 9], "incx": 2, "independ": [121, 348, 349], "index": [0, 1, 2, 8, 10, 29, 39, 139, 144, 169, 214, 239, 279, 280, 295, 296, 308], "indic": [0, 2, 17, 27, 28, 29, 30, 39, 167, 168, 169, 177, 178, 179, 180, 181, 182, 183, 197, 242, 279, 280, 283, 295, 296, 308, 384, 386, 435, 442, 482, 495], "indices_or_sect": [72, 283], "indirectli": 497, "individu": [334, 348, 349], "ineffici": [495, 496], "inexact": [12, 183], "inf": [195, 231, 390], "infer": [8, 166, 200, 302, 307, 492, 493], "infin": [0, 179, 181, 182, 231, 366, 367, 368, 472], "infinit": [17, 177, 178], "info": [6, 9], "inform": [3, 4, 6, 7, 9, 132, 162, 216, 269, 270, 326, 334, 339, 352, 390, 493, 494, 500], "inherit": [7, 490], "inifn": 179, "init": [334, 391, 461, 466, 480, 481, 483, 484, 492], "init_fn": [413, 414, 415, 416, 417, 418, 419, 420, 461], "init_valu": 1, "initi": [1, 3, 5, 6, 124, 322, 334, 339, 355, 359, 361, 363, 391, 394, 413, 414, 415, 416, 417, 418, 419, 420, 464, 475, 480, 481, 483, 484, 491, 492, 493, 496], "initializer_list": 0, "inject": 0, "inlin": 0, "inner": [0, 491], "inorm": 359, "inp": [1, 146], "inp_ndim": 1, "inp_shap": 1, "inp_strid": 1, "inplac": [2, 9], "input": [0, 1, 2, 5, 6, 13, 14, 15, 16, 17, 18, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 116, 118, 119, 122, 123, 128, 129, 130, 131, 132, 134, 135, 136, 138, 139, 140, 141, 142, 143, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 167, 168, 169, 170, 171, 172, 174, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 227, 228, 229, 230, 231, 232, 234, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 253, 256, 259, 260, 261, 262, 263, 264, 265, 266, 267, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 295, 296, 297, 298, 299, 300, 301, 302, 303, 305, 306, 307, 308, 309, 310, 312, 313, 315, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 348, 349, 351, 353, 354, 355, 359, 360, 361, 363, 366, 367, 368, 390, 393, 394, 395, 398, 409, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 436, 437, 438, 439, 441, 442, 444, 446, 458, 461, 491, 493, 494, 495, 498, 499], "input_dil": [0, 102], "input_dim": [7, 334, 363, 393], "input_nam": [1, 146], "input_s": [354, 360, 395], "inputs1": 442, "inputs2": 442, "insert": [119, 139, 500], "insid": [491, 493], "inspect": [3, 491, 498], "inspir": 8, "instabl": 446, "instal": [2, 4], "instanc": [6, 39, 113, 243, 323, 334, 359, 370, 371, 372, 375, 377, 378, 379, 384, 387, 388, 389, 400, 464, 497], "instancenorm": 334, "instanti": [1, 2, 7, 496], "instantiate_kernel": 2, "instead": [2, 9, 113, 334, 389, 403, 492, 494, 496], "instruct": [4, 493], "int": [0, 1, 2, 4, 6, 7, 10, 16, 18, 19, 27, 28, 29, 30, 34, 35, 36, 37, 42, 43, 44, 45, 46, 47, 50, 57, 58, 59, 60, 61, 64, 67, 69, 72, 75, 76, 77, 78, 79, 81, 84, 91, 92, 96, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 117, 118, 119, 126, 127, 128, 132, 139, 144, 148, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 166, 168, 169, 173, 183, 191, 195, 199, 209, 211, 213, 216, 217, 218, 219, 222, 223, 224, 227, 229, 235, 238, 239, 241, 242, 243, 244, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 262, 263, 265, 266, 279, 280, 281, 282, 283, 286, 287, 288, 292, 293, 295, 296, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 312, 314, 316, 334, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 351, 353, 354, 355, 359, 360, 361, 363, 366, 367, 368, 390, 392, 393, 394, 395, 398, 403, 411, 426, 434, 435, 439, 444, 446, 464, 480, 482, 483, 484], "int16": 326, "int32": [0, 1, 12, 19, 39, 163, 183, 185, 195, 254, 279, 307, 326, 412, 495, 498], "int64": [12, 326], "int64_t": 2, "int8": [12, 326], "int_0": 135, "integ": [0, 12, 165, 167, 168, 183, 195, 216, 238, 243, 244, 246, 253, 254, 283, 295, 299, 312, 326, 351, 382, 482, 495], "integr": [19, 295, 496], "intend": [0, 491], "interact": 411, "interest": 500, "interfac": 2, "intermedi": 497, "intern": 339, "interpol": 412, "interpret": 4, "interv": [19, 199, 254, 258], "introduc": [0, 265], "intuit": 334, "invalid": [0, 84], "invers": [0, 20, 21, 22, 23, 24, 25, 26, 136, 153, 154, 155, 156, 157, 158, 190, 194, 198], "invert": 0, "involv": [466, 491], "iogpu": 224, "iostream": 4, "ip": 492, "is_avail": 124, "is_equival": 2, "is_floating_point": 2, "is_leaf": [319, 320, 321, 322], "is_leaf_fn": 374, "isclos": 0, "isfinit": 0, "ish": 6, "ishmael": 6, "isinf": 0, "isnan": 0, "isneginf": 0, "isposinf": 0, "issu": [492, 494, 497], "issubdtyp": [12, 326], "item": [0, 2, 5, 6, 7, 320, 466, 493, 496, 497, 498], "iter": [5, 7, 197, 320, 321, 488, 491, 496], "iterm": 9, "itertool": [6, 320], "its": [0, 1, 2, 9, 149, 190, 210, 239, 256, 304, 317, 323, 334, 393, 466, 470, 471, 472, 492, 496, 497, 500], "itself": [2, 316, 475], "ix": 1, "ix_n": 1, "ix_nw": 1, "ix_s": 1, "ix_sw": 1, "iy_n": 1, "iy_nw": 1, "iy_s": 1, "iy_sw": 1, "j": [6, 9, 195, 348, 469, 470, 472], "j8": 2, "jacobian": [2, 184, 311, 498], "jain": 348, "jax": [8, 488], "jit": 146, "jmlr": 469, "jnp": 497, "john": 195, "join": 482, "join_schedul": 466, "jointli": 251, "just": [2, 4, 7, 361, 491, 493, 495], "jvp": [2, 113, 498], "k": [0, 6, 46, 91, 118, 144, 149, 167, 172, 301, 304, 305, 306, 363, 375], "kaim": 417, "keep": [2, 16, 18, 27, 28, 209, 211, 213, 227, 241, 288, 292, 309, 334, 374, 494, 496], "keepdim": [0, 16, 18, 27, 28, 34, 35, 36, 37, 57, 58, 59, 60, 64, 76, 77, 81, 195, 209, 211, 213, 227, 241, 281, 288, 292, 309], "kei": [1, 3, 6, 141, 149, 175, 216, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 319, 320, 374, 375, 387, 390, 475, 488, 490, 493, 494], "kept": 224, "kernel": [2, 8, 9, 99, 100, 101, 102, 103, 104, 105, 146, 336, 366, 491, 495], "kernel_dil": [0, 102], "kernel_s": [336, 337, 338, 341, 342, 343, 344, 345, 346, 366, 367, 368], "key_cach": 6, "key_input_dim": 390, "key_proj": 6, "keyword": [141, 169, 271, 272, 308, 320, 334, 488, 493, 499, 501], "kind": 6, "kingma": [470, 472], "kl_div_loss": 334, "kname": 2, "know": [2, 6], "known": [401, 453], "kron": 0, "kroneck": [0, 185], "kth": [0, 29, 239], "kullback": 439, "kw_onli": 2, "kwarg": [11, 121, 141, 142, 175, 271, 272, 324, 493, 501], "l": [6, 7, 189, 190, 192, 193, 334, 339, 341, 344, 354, 360, 395, 445], "l1": [308, 438, 440, 441, 445], "l1_loss": 334, "l2": [438, 441, 479], "l2_loss": 334, "l_": 438, "la": 195, "label": [3, 5, 435, 442], "label_smooth": 435, "lack": 495, "lambd": [356, 407, 427, 457], "lambda": [320, 321, 322, 334, 356, 370, 375, 382, 407, 427, 451, 457, 467, 468, 469, 470, 471, 472, 473, 478, 479, 491, 492, 493, 494], "languag": [1, 2, 4], "larg": [6, 334, 390, 441, 491, 492, 493, 496], "larger": [1, 148, 224, 398, 473], "largest": [195, 231, 301], "lasso": 308, "last": [0, 1, 6, 30, 79, 145, 147, 152, 155, 157, 158, 160, 161, 163, 167, 168, 176, 189, 190, 192, 193, 194, 196, 197, 198, 210, 219, 247, 282, 299, 310, 341, 342, 343, 344, 345, 346, 348, 349, 355, 412, 497], "latenc": 492, "later": [3, 9, 466], "launch": [1, 2, 124, 492, 495], "layer": [8, 145, 316, 334, 336, 337, 338, 348, 349, 354, 355, 360, 361, 363, 366, 367, 368, 384, 389, 392, 393, 395, 400, 411, 460, 464, 493], "layer_s": 7, "layernorm": 334, "layout": 1, "lazi": [8, 464, 498], "lazili": [6, 334], "lceil": 91, "ld": [354, 360, 395], "lead": [0, 19, 84, 491], "leaf": [95, 316, 319, 320, 321, 322, 374], "leaf_modul": 334, "leaki": [362, 430], "leaky_relu": 334, "leakyrelu": 334, "learn": [5, 7, 8, 339, 355, 359, 361, 391, 394, 466, 467, 468, 469, 470, 471, 472, 473, 478, 479], "learnabl": [341, 342, 343, 344, 345, 346, 400], "learning_r": [7, 466, 467, 468, 469, 470, 471, 472, 473, 475, 478, 479, 480, 481, 482, 483, 484, 491], "least": [6, 85, 86, 87, 94, 189, 190, 192, 193, 194, 196, 197, 198, 243], "leav": [2, 137, 320, 321, 322], "lectur": 478, "lecun": 348, "left": [0, 6, 148, 186, 195, 243, 265, 352, 398, 412, 424, 425, 436, 438, 446], "left_shift": 0, "leibler": 439, "len": [6, 152, 155, 158, 161, 172, 482], "length": [6, 286, 339, 341, 344, 354, 360, 395, 482], "leq": [438, 451], "less": [0, 1, 6, 29, 188, 224, 239, 398, 445], "less_equ": 0, "let": [1, 2, 5, 6, 190, 491, 493, 494, 496, 497], "level": [0, 167, 168, 416, 417], "lh": [354, 360, 395], "lhs_indic": [0, 167, 168], "lhs_mask": 91, "lib": 492, "libmlx": 9, "libmlx_ext": 2, "libmpi": 492, "librari": [2, 4, 9, 328, 334, 493], "like": [2, 6, 8, 127, 141, 143, 175, 183, 236, 315, 349, 441, 475, 477, 491, 492, 493, 494, 496, 497, 498, 500], "likelihood": [436, 444], "limit": [0, 2, 94, 222, 223, 224, 495], "linalg": 172, "line": [6, 492, 493, 496, 497], "linear": [0, 2, 6, 7, 8, 316, 320, 334, 340, 350, 352, 353, 362, 377, 393, 395, 396, 397, 399, 401, 412, 421, 422, 423, 424, 425, 426, 430, 449, 450, 451, 453, 461, 464, 475, 483, 491, 493], "linear1": 6, "linear2": 6, "linear3": 6, "linear_schedul": [466, 482], "linearli": 390, "link": [2, 4, 9], "linspac": 0, "lion": 466, "list": [1, 6, 11, 16, 18, 31, 72, 79, 84, 85, 86, 87, 92, 95, 96, 102, 132, 137, 146, 151, 152, 154, 155, 157, 158, 160, 161, 166, 169, 184, 195, 209, 211, 213, 214, 227, 235, 238, 241, 246, 247, 248, 250, 251, 252, 254, 257, 258, 269, 281, 283, 287, 288, 292, 299, 300, 303, 308, 309, 311, 314, 319, 322, 323, 334, 375, 377, 378, 379, 380, 385, 387, 388, 389, 464, 466, 470, 471, 472, 473, 482, 490, 491, 492, 493, 494, 496], "liter": [2, 238, 412, 416, 417, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446], "littl": 6, "liu": 6, "live": [8, 146, 500], "ll": [1, 5, 7, 438, 491, 494], "llama": 6, "llamaattent": 6, "llamaencoderlay": 6, "llm": 8, "load": [7, 8, 328, 377, 492], "load_weight": [334, 496], "loader": 7, "loader_path": 2, "loan": 195, "loc": [1, 250, 252], "local": [334, 348, 492], "locat": [0, 2, 4, 84, 279, 280, 388, 389, 492, 500], "log": [0, 203, 205, 209, 364, 365, 431, 432, 433, 436, 439, 441, 444, 456], "log10": 0, "log1p": 0, "log2": 0, "log_cosh_loss": 334, "log_sigmoid": [334, 364], "log_softmax": [334, 365], "logaddexp": 0, "logarithm": [0, 201, 202, 203, 204], "logcosh": 441, "logic": [0, 2, 206, 207, 208], "logical_and": 0, "logical_not": 0, "logical_or": 0, "logist": [0, 5, 275, 425, 453], "logit": [6, 247, 433, 435, 491], "logsigmoid": 334, "logsoftmax": 334, "logsumexp": 0, "long": 6, "longer": [6, 106, 494], "look": [2, 6, 492], "lookup": 351, "loop": [6, 7, 491, 492, 494, 496], "loshchilov": 471, "loss": [5, 7, 308, 334, 466, 491, 492, 494, 496], "loss_and_grad": 334, "loss_and_grad_fn": [7, 466, 491, 494], "loss_fn": [5, 7, 466, 491, 494], "loss_grad_fn": 492, "lot": [492, 494], "low": [254, 258, 420, 461], "low_pad_s": 0, "lower": [189, 190, 192, 193, 198, 243, 254, 257, 258, 304, 420], "lr": [5, 473], "lr_schedul": [480, 481, 482, 484], "lstm": 334, "lto": 2, "lu": 6, "luckili": 496, "lvalu": 308, "m": [0, 2, 4, 6, 9, 91, 144, 167, 172, 195, 304, 467, 491], "m1": [1, 6, 491, 494, 500], "m10": 326, "m7": 326, "m_": [470, 471, 472, 473], "m_t": [470, 471, 472, 473], "mac": 492, "machin": [6, 8, 9, 478, 492], "maco": [9, 224], "macosx": 9, "made": [6, 328], "mai": [2, 4, 141, 143, 175, 195, 316, 348, 492, 494, 495], "main": [4, 8, 119, 144, 146, 302, 320, 321, 334, 492], "maintain": [348, 349, 473], "major": [0, 2], "make": [1, 2, 3, 4, 6, 7, 9, 142, 143, 210, 233, 274, 334, 480, 481, 483, 484, 491, 496, 498, 500], "make_shar": 2, "malloc_or_wait": 2, "man": 6, "manag": [290, 488, 492, 493, 500], "mani": [2, 84, 283, 341, 342, 343, 344, 345, 346, 351, 392, 491, 492, 493, 496], "manual": 334, "map": [2, 7, 39, 200, 320, 351, 370, 493], "map_fn": [370, 374], "map_torch_to_mlx": 6, "margin": [442, 446], "margin_ranking_loss": 334, "mask": [0, 6, 91, 149, 384, 390, 495], "mask_lh": [0, 91], "mask_n": 1, "mask_nw": 1, "mask_out": [0, 91], "mask_rh": [0, 91], "mask_s": 1, "mask_sw": 1, "matadata": 200, "match": [9, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 217, 377, 412, 435, 492, 495, 497], "materi": [6, 8], "math": [6, 446, 491], "mathbf": 190, "mathcal": 363, "mathemat": 195, "mathrm": [135, 275, 359], "matmul": [0, 167, 500], "matric": [195, 196, 197], "matrix": [0, 5, 15, 46, 91, 117, 118, 144, 167, 168, 172, 173, 189, 190, 192, 193, 194, 195, 196, 197, 198, 210, 214, 243, 244, 251, 392, 393, 418, 461], "matter": [6, 334, 493], "max": [0, 1, 2, 195, 212, 340, 366, 367, 368, 391, 421, 428, 429, 434, 436, 437, 442, 446, 448, 450, 468, 472, 491, 494, 500], "max_buffer_s": 216, "max_freq": 403, "max_i": 243, "max_norm": 318, "max_recommended_working_set_s": [216, 224], "max_val": 428, "maximum": [0, 7, 27, 39, 94, 109, 219, 223, 318, 334, 362, 366, 367, 368, 396, 403, 424, 425, 430, 449, 464, 496], "maxpool1d": 334, "maxpool2d": 334, "maxpool3d": 334, "maxtotalthreadsperthreadgroup": 2, "mca": 492, "md": 195, "me": 6, "mean": [0, 1, 5, 6, 7, 147, 250, 251, 252, 308, 334, 339, 355, 375, 394, 419, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 491, 494, 497], "meant": 113, "measur": 500, "mechan": 411, "medic": 349, "meet": 9, "member": [2, 334, 380, 385], "memori": [0, 1, 2, 8, 84, 215, 217, 218, 219, 221, 222, 223, 224, 411, 464, 468, 491, 496, 497], "memory_order_relax": 1, "memory_s": [216, 224], "memoryview": [496, 497], "merg": 491, "meshgrid": 0, "metadata": [5, 200, 269, 270], "metal": [2, 4, 8, 146], "metal_captur": 3, "metal_kernel": 1, "metal_path": 9, "metallib": [2, 9], "method": [2, 6, 10, 11, 31, 113, 121, 162, 316, 324, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 381, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 464, 467, 468, 469, 470, 471, 472, 473, 475, 478, 479, 486], "millisecond": [9, 491, 500], "min": [0, 2, 195, 228, 340, 391, 421, 428, 429, 448, 450], "min_freq": 403, "min_i": 243, "min_val": 428, "mind": [2, 6], "mine": 6, "minibatch": 7, "minim": 492, "minimum": [0, 28, 39, 94, 110, 403, 433, 434], "minsizerel": 9, "minu": 140, "minut": 6, "mish": 334, "mismatch": 493, "miss": [377, 493, 499], "mix": 495, "mkdir": [3, 9], "ml": 9, "mlp": [7, 334, 411, 466], "mlp_dim": [6, 411], "mlx": [1, 3, 5, 6, 7, 9, 328, 334, 461, 464, 466, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "mlx_build_acceler": 4, "mlx_build_benchmark": 9, "mlx_build_cpu": 9, "mlx_build_exampl": 9, "mlx_build_gguf": 9, "mlx_build_met": [2, 4, 9], "mlx_build_metallib": 2, "mlx_build_python_bind": 9, "mlx_build_safetensor": 9, "mlx_build_test": 9, "mlx_cxx_flag": 4, "mlx_disable_compil": [120, 133, 491], "mlx_ext": 2, "mlx_ext_metallib": 2, "mlx_found": 4, "mlx_include_dir": [2, 4], "mlx_librari": 4, "mlx_metal_debug": [3, 9], "mlx_metal_jit": 9, "mlx_root": 4, "mlx_sample_extens": 2, "mlx_trace": 3, "mlxfn": [141, 143, 175, 493], "mnist": 7, "mode": [0, 1, 2, 106, 238, 373, 384, 386, 412, 416, 417, 492], "model": [5, 7, 8, 271, 316, 317, 320, 321, 334, 370, 373, 375, 377, 381, 384, 386, 387, 388, 390, 411, 461, 464, 466, 474, 475, 477, 491, 492, 493, 496], "modest": 2, "modif": 497, "modifi": 497, "modul": [2, 4, 6, 7, 316, 317, 400, 411, 461, 477, 490, 491, 496], "moment": [6, 468, 472], "momentum": [339, 473, 475, 479, 491], "monei": 6, "monoton": 447, "more": [1, 2, 3, 4, 7, 11, 79, 119, 141, 167, 189, 190, 192, 193, 194, 197, 198, 210, 222, 223, 269, 270, 326, 334, 339, 348, 398, 403, 411, 412, 414, 415, 416, 417, 433, 488, 491, 492, 494, 495, 498, 500], "most": [2, 149, 247, 307, 334, 477, 491, 492, 494, 495, 496], "move": [0, 2, 229, 500], "moveaxi": 0, "mpi": 328, "mpiexec": 492, "mpirun": 492, "mse": 308, "mse_loss": 334, "mtl": 2, "mtl_capture_en": 3, "mtlcommandbuff": 2, "mu": 479, "much": [1, 2, 6, 336, 337, 338, 366, 367, 368, 491, 496], "multi": [8, 149, 341, 342, 343, 344, 345, 346, 493, 495, 497], "multidimension": 214, "multiheadattent": [6, 334], "multipl": [0, 1, 9, 15, 91, 143, 145, 147, 167, 168, 210, 230, 243, 244, 390, 403, 481, 482, 484, 491, 496, 499], "multipli": [0, 2, 39, 168, 243, 244, 347, 403, 412], "murtadha": 6, "must": [0, 1, 2, 3, 9, 91, 94, 141, 148, 149, 166, 168, 192, 193, 195, 246, 247, 251, 254, 257, 258, 313, 412, 497], "mx": [1, 2, 3, 4, 5, 6, 7, 39, 97, 98, 113, 124, 127, 141, 142, 143, 146, 163, 175, 183, 185, 192, 193, 195, 196, 200, 253, 271, 279, 280, 307, 308, 318, 334, 336, 337, 338, 339, 350, 359, 362, 366, 367, 368, 370, 377, 381, 396, 412, 413, 414, 415, 416, 417, 418, 419, 420, 422, 430, 433, 434, 435, 439, 442, 449, 459, 461, 464, 466, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "my": [6, 9], "my_devic": 501, "my_path": 271, "myexp": [1, 146], "myexp_strid": 1, "mymlp": 464, "n": [0, 1, 2, 6, 31, 91, 99, 100, 101, 102, 103, 104, 105, 144, 149, 150, 152, 153, 155, 156, 159, 161, 172, 173, 251, 288, 304, 309, 339, 341, 342, 343, 344, 345, 346, 348, 349, 354, 360, 395, 412, 441, 446, 492], "n_kv": 149, "n_q": 149, "n_t": 354, "naiv": [2, 494], "naive_add": 494, "name": [1, 2, 113, 142, 146, 168, 200, 243, 244, 269, 270, 271, 272, 334, 355, 374, 377, 379, 492, 495, 499], "named_modul": 334, "namespac": 4, "nan": [0, 17, 83, 177, 178, 180, 231], "nan_to_num": 0, "nanobind": [2, 411], "nanobind_add_modul": 2, "nativ": 9, "natur": [0, 201, 203, 496], "nb": 2, "nb_domain": 2, "nb_func": 411, "nb_modul": 2, "nb_static": 2, "nbyte": 2, "nc": 339, "ndarrai": [31, 495, 496, 498], "ndhwc": [343, 346, 349], "ndim": [0, 1, 2, 163, 195, 197, 412], "ne": 1, "nearest": [1, 412], "necessari": 334, "necessarili": 301, "need": [1, 2, 4, 6, 7, 8, 9, 83, 243, 334, 388, 389, 403, 411, 488, 492, 494, 496, 497, 498, 500], "neg": [0, 119, 163, 181, 231, 265, 302, 362, 366, 367, 368, 390, 436, 444, 446, 495], "negat": [0, 232], "negative_slop": [362, 430], "neginf": [0, 231], "neighbor": 412, "neither": [169, 308], "nelem": 2, "nervou": 6, "nest": [79, 95, 322, 334, 464, 490, 494], "nesterov": 479, "network": [6, 8, 339, 348, 351, 414, 415, 461, 464, 478, 492], "neural": [6, 8, 351, 414, 415, 447, 461, 464, 478], "never": [6, 496], "new": [0, 2, 7, 92, 119, 229, 233, 263, 287, 303, 310, 320, 321, 382, 390, 464, 466, 477, 482, 491, 493, 495, 496, 497], "new_tre": 321, "next": [2, 4, 6, 7, 222, 493], "nh": [354, 360, 395], "nhwc": [339, 342, 345], "nice": [494, 496], "nlc": [339, 341, 344], "nld": [354, 360, 395], "nlh": [354, 360, 395], "nll": [436, 444], "nll_loss": 334, "nn": [2, 6, 7, 271, 320, 334, 461, 464, 466, 475, 477, 491, 493, 496], "nobodi": 6, "node": [95, 137, 312, 321, 322], "nois": 5, "noisi": 5, "nomins": 2, "non": [0, 1, 2, 4, 9, 214, 385, 395, 447, 464], "none": [1, 2, 6, 10, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 116, 117, 118, 119, 120, 122, 123, 126, 127, 128, 129, 130, 131, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 221, 225, 226, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 270, 271, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 312, 313, 314, 315, 316, 319, 320, 321, 322, 336, 337, 338, 352, 366, 367, 368, 370, 374, 375, 382, 387, 390, 395, 403, 411, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 468, 486, 493, 495], "nonlinear": [395, 491], "nonzero": 495, "noop": 387, "nor": [2, 169, 308], "norm": [6, 147, 318, 355, 446, 472, 473], "norm1": 6, "norm2": 6, "norm_first": 411, "normal": [1, 2, 5, 6, 145, 146, 147, 192, 251, 257, 334, 336, 337, 338, 339, 355, 359, 361, 366, 367, 368, 394, 411, 414, 416, 497, 500], "not_equ": 0, "notabl": [6, 8], "notat": [117, 319, 379], "note": [0, 1, 2, 4, 6, 9, 17, 19, 84, 91, 95, 101, 104, 105, 113, 149, 157, 158, 168, 177, 195, 217, 243, 247, 310, 316, 334, 394, 412, 466, 497, 499], "noth": [6, 113, 334, 496], "notic": [6, 493, 494, 499], "now": [1, 2, 6, 9, 393, 491, 492, 497], "np": [1, 6, 7, 492, 497, 498], "npy": [200, 268, 499], "npz": [6, 200, 271, 272, 377, 381, 499], "nuclear": 195, "nullopt": 0, "num": [0, 6, 199, 256], "num_class": [7, 466], "num_decoder_lay": 411, "num_embed": [351, 392], "num_encoder_lay": 411, "num_epoch": [7, 466], "num_exampl": 5, "num_featur": [5, 339], "num_group": 355, "num_head": [6, 390, 411], "num_it": 5, "num_lay": [6, 7, 466], "num_param": 334, "num_paramet": 391, "num_sampl": 247, "num_split": 0, "number": [0, 2, 12, 19, 62, 71, 95, 100, 101, 102, 104, 105, 117, 142, 144, 149, 168, 169, 173, 184, 199, 231, 238, 243, 244, 247, 250, 252, 256, 258, 262, 265, 266, 299, 300, 304, 308, 311, 312, 316, 334, 339, 341, 342, 343, 344, 345, 346, 348, 349, 355, 359, 390, 391, 411, 412, 414, 415, 416, 417, 480, 482, 483, 488, 491, 494, 501], "number_of_el": 0, "numer": [6, 145, 147, 195, 205, 209, 281, 339, 355, 359, 361, 394, 433, 434, 436, 446, 467, 468, 469, 470, 471, 472, 478, 491, 496], "numpi": [2, 6, 7, 8, 14, 17, 19, 88, 89, 90, 92, 129, 130, 134, 170, 171, 177, 186, 187, 188, 205, 210, 212, 228, 230, 234, 240, 261, 264, 291, 496, 498, 499], "nw": 1, "nwhc": 348, "o": [2, 9, 149, 360], "o_t": 360, "obj": 269, "object": [3, 11, 31, 51, 79, 95, 142, 143, 146, 183, 271, 312, 319, 320, 321, 322, 326, 348, 411, 490], "observ": 6, "occupi": [117, 168, 243, 244], "occur": 497, "odim": 7, "odot": [354, 360], "off": [6, 9, 496], "offer": 441, "offset": [0, 1, 2, 6, 47, 84, 119, 145, 148, 302], "often": 349, "ok": [377, 491, 493, 494], "okai": [491, 496], "old": 6, "older": [141, 143, 175], "omit": [472, 492], "onc": [2, 9, 491, 493], "one": [0, 2, 4, 6, 9, 39, 79, 85, 94, 100, 101, 102, 104, 105, 139, 141, 144, 145, 147, 148, 195, 203, 210, 244, 247, 286, 291, 307, 326, 387, 412, 435, 492, 493, 500], "ones": [0, 2, 6, 236, 271, 280, 304, 388, 389, 466, 492, 495], "ones_lik": 0, "onli": [1, 2, 6, 8, 9, 83, 91, 100, 101, 102, 104, 105, 113, 192, 193, 195, 224, 243, 251, 310, 334, 374, 375, 377, 382, 384, 387, 388, 389, 464, 491, 492, 493, 494, 499, 500], "onlin": 469, "op": [1, 2, 237, 310, 375, 496], "open": [3, 9, 19, 254, 258], "openmpi": 492, "oper": [3, 6, 8, 10, 38, 85, 86, 87, 102, 149, 167, 168, 240, 242, 281, 289, 296, 324, 334, 411, 473, 491, 492, 494, 495, 496, 497, 498, 500, 501], "operand": [131, 132, 167], "opportun": 491, "opt": [474, 492], "optim": [1, 3, 5, 7, 8, 388, 491, 492, 494, 496], "option": [0, 3, 6, 15, 16, 18, 19, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 84, 85, 86, 87, 91, 95, 96, 99, 100, 101, 102, 103, 104, 105, 106, 109, 110, 111, 112, 113, 117, 118, 119, 122, 123, 124, 126, 127, 128, 141, 143, 144, 145, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 166, 167, 168, 169, 173, 181, 182, 185, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 209, 211, 213, 214, 223, 227, 231, 235, 238, 239, 241, 243, 244, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 262, 263, 265, 281, 282, 283, 286, 287, 288, 292, 294, 295, 299, 301, 302, 303, 304, 305, 306, 307, 308, 309, 312, 314, 316, 319, 320, 321, 322, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 354, 360, 363, 366, 367, 368, 370, 374, 375, 377, 382, 387, 390, 392, 393, 395, 398, 403, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 467, 468, 469, 470, 471, 472, 473, 475, 478, 479, 480, 488, 491, 493, 499, 501], "ord": 195, "order": [0, 1, 29, 84, 102, 132, 192, 193, 195, 239, 243, 301, 334, 355, 388, 400, 475, 491, 494], "ordinari": 176, "org": [355, 359, 361, 369, 394, 425, 447], "origin": [6, 119, 318, 339, 383, 414, 415, 416, 417, 467, 468, 469, 472, 473, 493, 497], "orthonorm": 172, "ostream": 2, "ostringstream": 2, "other": [0, 2, 6, 8, 183, 195, 334, 376, 464, 473, 491, 492, 493, 495, 496, 498], "other_input": 334, "otherwis": [19, 102, 124, 223, 253, 316, 319, 320, 321, 322, 375, 377, 387, 409, 411, 412, 427, 433, 438, 445, 457, 458, 496, 497], "our": [1, 2, 6, 7, 400, 467, 468, 469, 472, 473, 492], "out": [0, 1, 2, 9, 91, 146, 175, 348, 349, 384, 491, 492, 493, 494, 495], "out_ax": [312, 494], "out_channel": [341, 342, 343, 344, 345, 346], "out_dim": [334, 464], "out_dtyp": 2, "out_idx": 2, "out_mask": 91, "out_proj": [6, 464], "out_ptr": 2, "out_shap": [1, 2], "outer": [0, 491, 496], "outlier": 441, "output": [0, 1, 2, 6, 9, 16, 17, 18, 19, 29, 84, 91, 92, 95, 97, 98, 109, 110, 111, 112, 113, 131, 142, 144, 145, 146, 147, 148, 149, 156, 159, 160, 161, 166, 167, 169, 172, 173, 177, 195, 199, 209, 211, 213, 214, 227, 231, 235, 236, 239, 241, 242, 246, 247, 248, 250, 251, 252, 254, 257, 258, 271, 272, 279, 280, 281, 286, 288, 292, 296, 302, 304, 308, 309, 310, 311, 312, 313, 314, 315, 339, 341, 342, 343, 344, 345, 346, 359, 363, 390, 393, 409, 411, 412, 414, 415, 416, 417, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 458, 461, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "output_dim": [7, 334, 363, 393], "output_directori": 2, "output_dtyp": [1, 146], "output_fil": 6, "output_nam": [1, 146], "output_shap": [1, 146], "output_strip_trailing_whitespac": 4, "output_vari": 4, "outsid": [146, 163], "over": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 152, 155, 158, 161, 176, 195, 197, 199, 209, 211, 213, 227, 239, 241, 267, 281, 282, 288, 292, 299, 301, 309, 339, 341, 342, 343, 344, 345, 346, 355, 361, 394, 435, 480, 483, 492, 494], "overal": 2, "overhead": [491, 496, 500], "overlap": 1, "overload": 19, "overrid": [2, 133], "overview": 3, "overwrit": 6, "own": [9, 497], "owndata": 497, "p": [9, 246, 334, 347, 348, 349, 446, 470, 472], "pack": [168, 243, 244], "packag": [2, 5, 7, 9, 328, 461, 492], "package_data": 2, "pad": [0, 1, 99, 100, 101, 102, 103, 104, 105, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 336, 337, 338, 341, 342, 343, 344, 345, 346, 366, 367, 368], "pad_valu": 0, "pad_width": [0, 238], "padding_hi": 0, "padding_lo": 0, "page": 498, "pain": 6, "pair": [0, 2, 238, 377, 398], "pairwis": 446, "pan": 6, "paper": [339, 403, 467, 468, 469, 472, 473], "parallel": [492, 500], "param": [308, 316, 334, 461, 493, 494], "paramet": [0, 1, 2, 5, 6, 7, 13, 14, 15, 16, 17, 18, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 116, 117, 118, 119, 122, 123, 124, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 222, 223, 224, 225, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 359, 360, 361, 362, 363, 366, 367, 368, 370, 371, 374, 375, 377, 382, 383, 384, 387, 388, 389, 390, 391, 392, 393, 394, 395, 398, 400, 403, 407, 409, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 458, 460, 461, 464, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 477, 478, 479, 480, 481, 482, 483, 484, 486, 491, 492, 493, 494, 496], "parameter_scal": 468, "parametr": [391, 448], "pars": [6, 142], "parse_arg": 6, "parser": 6, "part": [1, 2, 141, 143, 174, 175, 259, 494, 495], "partial": [388, 389, 491, 496], "particip": [122, 123, 126, 127, 128], "particular": [243, 355], "particularli": 491, "partit": [0, 29], "pass": [1, 2, 6, 7, 66, 80, 237, 238, 308, 317, 319, 320, 321, 334, 375, 387, 388, 389, 400, 491, 492, 493, 496], "password": 492, "path": [3, 4, 9, 132, 141, 142, 143, 175, 225, 271, 272, 316, 321, 377, 492], "pattern": [334, 496], "peak": [219, 221], "penalti": 479, "pep": 497, "per": [6, 7, 117, 149, 168, 243, 244, 316, 339, 355, 359, 361, 394, 486, 491, 492, 496], "perceptron": [8, 493], "perf_count": 491, "perfectli": 496, "perform": [0, 1, 2, 3, 6, 8, 15, 91, 102, 109, 110, 111, 112, 128, 131, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 167, 168, 172, 192, 193, 210, 244, 266, 281, 295, 334, 355, 411, 416, 417, 466, 491, 492, 495, 496, 500], "perhap": [2, 6], "perm": 7, "permtuat": 253, "permut": [0, 7], "persist": 9, "pg": 195, "phi": [352, 423], "physic": 492, "pi": [135, 352, 403, 424, 494], "pick": 2, "pip": [2, 4, 9], "pipelin": 2, "pixel": 348, "place": [2, 6, 39, 265, 266, 316, 492, 496, 497], "placehold": 491, "plai": [2, 6], "plain": 400, "plan": [2, 491], "platform": 9, "plu": [0, 203], "point": [0, 2, 5, 6, 9, 84, 162, 165, 244, 326], "pointer": 2, "pool": [336, 337, 338, 366, 367, 368, 500], "popul": 2, "portion": 347, "posinf": [0, 231], "posit": [0, 6, 29, 119, 148, 163, 169, 182, 189, 190, 229, 231, 239, 251, 265, 302, 308, 320, 334, 341, 342, 343, 344, 345, 346, 390, 398, 403, 436, 446, 493], "possibl": [283, 351, 392, 491, 492, 495, 500], "possibli": [6, 15, 91, 167, 210, 318], "postur": 6, "potenti": 223, "power": [0, 494, 497], "practic": [2, 491], "pre": [9, 149, 433], "preced": 355, "precis": [0, 2, 6, 140, 149, 334, 352, 394, 433, 474, 491], "preclud": 334, "pred": [437, 441], "predic": [316, 382], "predict": [433, 436, 437, 438, 439, 440, 441, 443, 444, 445], "prefix": [312, 319], "prelu": 334, "prepar": [2, 6], "prepend": [3, 210], "preprint": [6, 467, 473], "preprocessor": 9, "present": 1, "preserv": [263, 494], "press": [6, 195], "pressur": 2, "pretti": [491, 496], "prevent": [289, 446, 497], "previou": [222, 223, 224], "primal": [1, 2, 113, 184, 311], "primit": 494, "print": [1, 2, 5, 6, 7, 9, 113, 185, 318, 319, 320, 321, 323, 334, 488, 491, 492, 493, 494, 495, 496, 497, 498], "prior": [242, 295, 296], "priorit": 494, "privat": [2, 4], "prng": [246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 488], "prob": 433, "probabl": [9, 254, 347, 348, 349, 393, 433, 435, 439, 500], "problem": [5, 7, 334], "process": [6, 102, 106, 121, 122, 123, 124, 126, 127, 128, 320, 321, 349, 351, 411, 490, 492], "processor": 9, "prod": [0, 1], "produc": [0, 2, 9, 95, 390, 461, 493], "product": [0, 2, 15, 84, 111, 176, 184, 185, 191, 210, 237, 241, 299, 311, 390, 498], "profil": 3, "program": [4, 219, 492], "programmat": 389, "project": [3, 4, 6, 390, 493], "project_source_dir": 2, "promot": [2, 149], "promote_typ": 2, "promoted_dtyp": 2, "prompt": 6, "propag": [494, 495], "properti": [32, 39, 48, 52, 62, 63, 69, 71, 383, 386, 476, 494], "proportion": 318, "protocol": 497, "provid": [0, 2, 6, 84, 117, 141, 142, 169, 253, 265, 299, 308, 320, 322, 328, 334, 370, 375, 377, 387, 388, 389, 392, 393, 411, 412, 460, 464, 492, 493, 499, 501], "pseudo": 488, "pth": 6, "public": [2, 334], "pun": 0, "pure": [1, 113, 334, 466], "purpos": [1, 195], "purs": 6, "push": 2, "push_back": 2, "put": [0, 1, 7, 242, 491, 492], "put_along_axi": 0, "py": [2, 6, 9, 492], "pypi": 9, "python": [1, 3, 4, 6, 51, 69, 79, 137, 319, 320, 321, 322, 323, 464, 474, 475, 477, 490, 492, 493, 494, 497], "python_execut": 4, "python_requir": 2, "pytorch": [6, 8, 352, 355, 494], "pytorch_compat": 355, "q": [149, 196], "qualifi": 492, "quantiz": [0, 117, 168, 200, 244, 392, 393], "quantized_matmul": 0, "quantizedembed": 334, "quantizedlinear": 334, "quarter": 6, "queri": [6, 149, 224, 390], "query_input_dim": 390, "query_proj": 6, "question": [6, 496], "queue": 3, "quick": [2, 8], "quit": [494, 497], "quotient": [0, 129, 130, 165], "r": [2, 6, 196, 308, 348, 354], "r_t": 354, "race": 500, "radian": [0, 116], "rag": 6, "rain": 6, "rais": [0, 6, 113, 195, 223, 240, 283, 377, 493], "ram": 6, "random": [1, 2, 3, 5, 6, 7, 8, 146, 336, 337, 338, 339, 359, 366, 367, 368, 377, 384, 491, 493, 494, 500, 501], "randomli": [5, 6, 253, 347, 348, 349], "rang": [0, 2, 3, 5, 6, 7, 9, 19, 163, 167, 199, 415, 417, 424, 425, 466, 480, 481, 482, 483, 484, 488, 491, 494, 496, 500], "rank": [0, 126, 127, 128, 442, 492], "rate": [5, 466, 467, 468, 469, 470, 471, 472, 473, 478, 479], "rather": [2, 494, 500], "ratio": [0, 25], "rceil": 91, "re": [7, 9, 461], "readabl": 3, "readi": 2, "real": [0, 156, 157, 158, 159, 160, 161, 189, 190, 192, 193], "realli": 361, "reason": [1, 6, 495], "reboot": 9, "receiv": [126, 127, 316, 482, 497], "reciproc": [0, 267], "reclaim": 222, "recommend": [9, 223, 473], "recompil": [95, 491], "record": [3, 219, 496], "recreat": [323, 466], "rectifi": [362, 396, 397, 416, 417, 430, 449, 450], "recurr": [354, 360, 395], "recurs": [142, 334, 374, 375, 380, 385, 387, 464], "recv": 127, "redirect": 2, "reduc": [0, 1, 9, 16, 18, 27, 28, 123, 209, 211, 213, 227, 241, 288, 292, 309, 322, 339, 411, 441], "reduct": [16, 18, 123, 209, 211, 227, 241, 322, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 492], "redund": 494, "refer": [195, 359, 369, 383, 414, 415, 416, 417, 425, 447, 495], "reflect": [383, 491, 495, 497], "regard": 352, "regardless": [84, 149], "regist": [2, 7], "register_librari": 2, "regress": [8, 441], "regular": [39, 348, 447, 471, 491, 493, 495], "regularli": 2, "reimplement": 2, "rel": [17, 177, 468, 491], "relative_step": 468, "relax": 223, "releas": 4, "relev": 2, "reli": [1, 2], "relu": [334, 391, 411, 448, 461], "relu6": 334, "remain": [0, 6, 224, 308, 321, 347, 348, 349, 492], "remaind": [0, 130], "remov": [0, 119, 210, 247, 286, 435], "rep": [0, 300], "repeat": [0, 300], "repeatedli": 5, "repetit": 262, "replac": [0, 6, 231, 388, 389, 411, 445], "replai": 3, "repli": 6, "repo": [5, 7, 9, 491], "report": [217, 223], "repres": [2, 6, 121, 124, 168, 442, 446, 497], "represent": [6, 243, 310, 319, 323], "request": 2, "requir": [1, 2, 4, 6, 334, 492, 496, 497], "requires_grad": 494, "rerun": [491, 496], "rescal": 318, "research": 8, "reset": 221, "reset_peak_memori": 219, "reshap": [0, 6, 195, 412, 491, 495], "resid": 224, "resolv": 2, "resourc": 2, "resource_limit": 216, "respect": [2, 5, 7, 113, 145, 147, 167, 168, 169, 243, 308, 320, 334, 339, 352, 355, 359, 361, 464, 494, 498], "respons": 2, "rest": [6, 148, 320, 321, 398], "restart": 9, "restor": 265, "result": [0, 6, 15, 19, 39, 79, 84, 95, 142, 145, 147, 168, 185, 195, 210, 244, 251, 262, 287, 320, 321, 322, 403, 433, 491, 494, 497], "resum": 6, "return": [0, 1, 2, 4, 5, 6, 7, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 51, 69, 79, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 116, 117, 118, 119, 122, 123, 124, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 138, 139, 140, 141, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 218, 222, 223, 224, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 266, 267, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 334, 354, 360, 370, 371, 372, 374, 375, 376, 377, 378, 379, 380, 384, 385, 387, 388, 389, 395, 413, 414, 415, 416, 417, 418, 419, 420, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 461, 464, 474, 490, 491, 492, 493, 494, 495, 496, 497, 499, 500], "return_metadata": 200, "revers": [0, 2, 42, 43, 44, 45, 84, 109, 110, 111, 112, 303, 403], "rf": 9, "rfft": 156, "rfft2": 157, "rfftn": 158, "rho": 467, "rhs_indic": [0, 167, 168], "rhs_mask": 91, "right": [0, 1, 2, 9, 243, 264, 265, 352, 412, 424, 425, 436, 438, 446], "right_shift": 0, "rm": [6, 9, 147, 468], "rmsnorm": [6, 334], "rmsprop": 466, "rnn": [334, 354], "roadcast": 254, "robust": 441, "roform": [6, 398], "roll": 0, "root": [0, 6, 147, 267, 284, 394], "rope": [6, 334], "rosetta": 9, "rotari": [6, 148, 398], "rotat": [148, 398], "round": [0, 243], "routin": 2, "row": [0, 1, 2, 84, 144, 146, 173, 243, 304], "row_contigu": 2, "rpath": 2, "rsqrt": 0, "rtol": [0, 17, 177], "rule": [2, 466], "run": [1, 2, 3, 4, 6, 7, 8, 9, 10, 146, 237, 324, 339, 370, 467, 468, 470, 471, 472, 491, 492, 493, 496, 500, 501], "runtim": [6, 124, 328, 491, 492], "runtime_error": 2, "safetensor": [9, 200, 270, 377, 381, 466, 496, 499], "sai": [2, 6, 461, 496], "said": 6, "sake": 494, "same": [0, 2, 6, 9, 17, 39, 83, 92, 95, 100, 101, 102, 104, 105, 106, 122, 145, 147, 156, 159, 160, 161, 168, 169, 177, 184, 238, 247, 265, 266, 280, 310, 311, 313, 321, 334, 337, 338, 339, 347, 355, 359, 367, 368, 392, 413, 414, 415, 416, 417, 418, 419, 420, 435, 446, 464, 474, 488, 491, 492, 493, 495, 500], "sampl": [2, 5, 6, 199, 246, 247, 248, 250, 251, 254, 257, 258, 414, 415, 416, 417, 419, 420, 436, 442, 446, 488, 491, 493], "sat": 6, "save": [3, 6, 8, 200, 225, 243, 269, 270, 271, 272, 381, 493, 496], "save_gguf": 499, "save_safetensor": [381, 466, 499], "save_weight": 334, "savez": [6, 381, 499], "savez_compress": 499, "saw": [6, 494], "scalar": [0, 2, 14, 15, 17, 31, 51, 79, 83, 88, 89, 90, 91, 92, 94, 129, 130, 134, 165, 166, 169, 170, 171, 172, 177, 186, 187, 188, 199, 205, 206, 207, 208, 210, 212, 228, 230, 231, 234, 238, 240, 246, 254, 257, 258, 261, 264, 269, 291, 308, 310, 313, 317, 446, 493, 494, 496, 498], "scale": [0, 2, 6, 15, 117, 145, 147, 148, 149, 168, 172, 243, 244, 250, 252, 318, 348, 349, 361, 390, 398, 399, 403, 412, 451, 468], "scale_arr": 2, "scale_factor": 412, "scale_paramet": 468, "scatter": 0, "scatter_add": 0, "scatter_max": 0, "scatter_min": 0, "scatter_prod": 0, "schedul": [2, 223, 466, 480, 481, 482, 483, 484, 486, 500], "schema": 3, "scipi": 172, "scope": 334, "score": [6, 149, 442], "sdk": 9, "se": 1, "second": [6, 9, 119, 183, 185, 186, 206, 208, 210, 264, 293, 302, 308, 337, 338, 367, 368, 434, 442, 468, 472, 491, 493, 494, 500], "second_layer_a": 496, "second_layer_b": 496, "secret": 6, "section": [1, 6, 9, 283, 446, 491, 492, 494], "see": [1, 2, 4, 6, 7, 9, 11, 12, 33, 34, 35, 36, 37, 40, 41, 42, 43, 44, 45, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 195, 222, 269, 270, 316, 326, 334, 339, 340, 348, 350, 352, 356, 357, 358, 364, 365, 373, 391, 392, 393, 396, 397, 398, 399, 401, 403, 404, 405, 406, 407, 408, 410, 412, 414, 415, 416, 417, 423, 424, 425, 451, 491, 492, 493, 494, 495, 498, 500], "seed": 249, "seen": 497, "select": [0, 3, 9, 192, 193, 301, 313, 370, 374, 382], "self": [6, 7, 10, 31, 32, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 51, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 113, 162, 334, 447, 464], "selu": 334, "semant": [14, 88, 89, 90, 92, 129, 130, 134, 170, 171, 186, 187, 188, 205, 210, 212, 228, 230, 234, 240, 261, 264, 291, 500], "semi": [189, 190, 251], "send": 492, "sennrich": 6, "sensit": 441, "sentencepiec": 6, "separ": [6, 66, 80, 355, 442], "sequenc": [6, 16, 18, 34, 35, 57, 58, 59, 60, 64, 72, 75, 76, 77, 81, 84, 92, 102, 126, 139, 146, 151, 152, 154, 155, 157, 158, 160, 161, 166, 209, 211, 213, 227, 235, 241, 246, 247, 248, 250, 251, 252, 254, 257, 258, 263, 279, 280, 281, 283, 286, 288, 292, 299, 300, 303, 307, 309, 314, 339, 341, 344, 354, 360, 395, 411, 488, 500], "sequenti": [334, 461], "seri": 9, "serial": 466, "set": [2, 4, 6, 7, 9, 95, 113, 120, 122, 123, 124, 126, 127, 128, 133, 145, 148, 216, 222, 223, 224, 273, 274, 290, 352, 361, 363, 373, 375, 382, 383, 384, 387, 388, 393, 398, 409, 434, 446, 458, 464, 466, 468, 470, 471, 475, 488, 493, 494, 496], "set_byt": 2, "set_compute_pipeline_st": 2, "set_data": 2, "set_default_devic": 2, "set_dtyp": 334, "set_input_arrai": 2, "set_memory_limit": 222, "set_output_arrai": 2, "set_vector_byt": 2, "setup": [2, 4, 5, 7, 9, 491, 493], "sever": [6, 9, 99, 100, 101, 102, 103, 104, 105, 271, 272, 491, 499], "sgd": [5, 7, 466, 473, 475, 480, 481, 484, 491], "shade": [1, 2], "shall": 6, "shape": [0, 2, 3, 6, 7, 66, 83, 84, 91, 92, 95, 99, 100, 101, 102, 103, 104, 105, 119, 122, 126, 127, 141, 143, 146, 149, 150, 153, 156, 159, 160, 161, 166, 167, 172, 184, 194, 198, 210, 235, 236, 246, 247, 248, 250, 251, 252, 254, 257, 258, 263, 265, 280, 307, 310, 311, 313, 314, 315, 334, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 348, 349, 354, 359, 360, 363, 366, 367, 368, 377, 395, 413, 414, 415, 416, 417, 418, 419, 420, 435, 446, 466, 491, 493, 494, 495, 498, 500], "shapeless": [0, 95, 141, 143], "share": [8, 117, 168, 243, 244, 310, 492], "shazeer": 6, "shift": [0, 186, 264, 265, 339], "shop": 6, "should": [1, 2, 4, 5, 6, 7, 9, 84, 119, 122, 145, 146, 147, 149, 184, 215, 224, 225, 242, 243, 296, 302, 308, 311, 316, 319, 334, 341, 342, 343, 344, 345, 346, 348, 349, 384, 390, 400, 435, 437, 442, 464, 490, 491, 492, 493, 494, 496, 497, 501], "show": [9, 326, 491], "shown": 2, "shuffl": 7, "side": [0, 238, 336, 337, 338, 366, 367, 368, 491], "sigma": [352, 353, 354, 360, 402, 414, 415, 416, 417, 425, 426, 431, 452, 453], "sigmoid": [0, 6, 334, 364, 401, 425, 431, 433, 453], "sign": [0, 17, 177, 326, 473], "signal": [106, 412], "signatur": [1, 146], "signedinteg": [12, 183], "signific": 243, "silent": [159, 160, 161], "silicon": [2, 6, 8, 9, 500], "silu": 334, "simd": 1, "simd_sum": 1, "simdgroup": 1, "simdgroup_s": 1, "similar": [6, 168, 183, 320, 388, 389, 390, 434, 497, 499], "similarli": [2, 9, 210, 494, 496], "simpl": [2, 6, 7, 334, 351, 460, 466, 491, 492, 493, 494, 496], "simple_axpbi": 2, "simple_tim": 2, "simplest": [2, 334, 492], "simpli": [2, 6, 9, 350, 362, 396, 422, 430, 449, 459, 464, 491, 492, 494], "simplic": 0, "simultan": 1, "sin": [0, 113, 403, 493, 494, 498], "sinc": [1, 2, 6, 7, 168, 219, 464, 473, 482, 491, 493, 497, 500], "sine": [0, 22, 23, 277, 278, 493, 494], "sing": 195, "singer": 469, "singl": [2, 7, 137, 184, 200, 214, 238, 311, 337, 338, 367, 368, 491, 493, 495, 499], "singleton": [0, 16, 18, 27, 28, 124, 209, 210, 211, 213, 227, 241, 288, 292, 309], "singular": [195, 197], "sinh": 0, "sinusoid": 403, "sinusoidalpositionalencod": 334, "size": [0, 1, 2, 6, 7, 52, 69, 91, 100, 101, 104, 105, 117, 139, 145, 146, 147, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 166, 168, 172, 173, 183, 191, 195, 218, 223, 224, 243, 244, 247, 263, 279, 283, 286, 307, 310, 316, 334, 336, 337, 338, 341, 342, 343, 344, 345, 346, 351, 359, 366, 367, 368, 392, 393, 412, 468, 492, 496, 497], "size_in_megabyt": 224, "size_t": [0, 2], "skip": [3, 84], "slice": [0, 280, 495], "slice_s": [0, 279], "slice_upd": 0, "slide": [336, 337, 338, 366, 367, 368], "slight": [6, 496], "slightli": [398, 500], "slope": 362, "slot": 492, "slow": 491, "slowli": 6, "small": [6, 140, 145, 147, 339, 355, 361, 394, 436, 441, 446, 491, 492, 500], "smaller": [0, 9, 239, 473, 491], "smallest": 195, "smile": 6, "smooth": [435, 445, 478], "smooth_l1_loss": 334, "sned": 128, "snippet": 492, "so": [1, 2, 6, 9, 169, 172, 308, 347, 412, 466, 491, 492, 496, 500], "softmax": [0, 6, 149, 334, 365, 432, 435], "softmin": 334, "softplu": [334, 369, 447], "softshrink": 334, "softsign": 334, "solv": 334, "some": [0, 2, 5, 6, 7, 142, 375, 387, 466, 475, 491, 493, 494, 496], "someon": 6, "someth": [5, 6, 495], "sometim": 491, "sonoma": 9, "soon": 6, "sort": [0, 29, 30, 239, 301], "sourc": [0, 1, 2, 3, 4, 61, 126, 127, 146, 229, 303, 492], "space": [0, 2, 199, 433, 444], "spars": [0, 214], "spatial": [100, 101, 102, 104, 105, 336, 337, 338, 355, 366, 367, 368, 412], "speak": [6, 195], "special": 2, "specif": [1, 2, 9, 492, 494], "specifi": [0, 2, 19, 38, 100, 101, 102, 104, 105, 119, 157, 158, 166, 169, 191, 195, 199, 229, 235, 242, 247, 262, 293, 295, 296, 299, 302, 303, 308, 312, 314, 339, 409, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 458, 491, 492, 493, 494, 500], "speed": [1, 2], "spent": 6, "split": [0, 353, 355, 426], "splittabl": 488, "sqrt": [0, 6, 135, 149, 172, 339, 352, 355, 359, 361, 363, 394, 403, 414, 415, 416, 417, 424, 467, 469, 470, 471, 478, 491], "squar": [0, 5, 6, 147, 173, 194, 198, 267, 284, 308, 320, 334, 394, 443, 445, 467, 468, 470, 471, 472, 494, 497], "squeez": [0, 412, 491], "src": [0, 126, 127], "ssh": 492, "stabil": [145, 147, 339, 355, 359, 361, 394, 433, 434, 436, 467, 468, 469, 470, 471, 472, 478], "stabl": [205, 209, 281, 441], "stable_abi": 2, "stack": [0, 491], "standard": [0, 1, 4, 7, 51, 79, 210, 248, 252, 288, 411, 414, 416, 419, 492, 498], "starmap": [6, 320], "start": [0, 1, 2, 5, 6, 8, 9, 19, 148, 199, 225, 279, 280, 283, 322, 491, 493, 495, 500], "start_axi": [0, 50, 163], "start_captur": 3, "start_indic": [279, 280], "state": [6, 7, 334, 354, 360, 395, 466, 475, 488, 491], "static": [9, 491], "static_cast": 2, "std": [0, 2, 4, 419, 493], "step": [0, 3, 4, 6, 7, 19, 334, 354, 360, 395, 468, 475, 480, 482, 483, 484, 491, 492], "step_decai": 466, "step_siz": 484, "still": [6, 9, 195, 491, 496], "stochast": [469, 470, 472, 479, 496], "stood": 6, "stop": [0, 2, 6, 19, 199, 226, 289, 494, 495], "stop_captur": 3, "stop_gradi": [0, 494], "storag": 84, "store": 6, "str": [2, 106, 131, 132, 141, 142, 143, 146, 169, 175, 192, 193, 195, 200, 214, 216, 225, 268, 269, 270, 271, 272, 308, 316, 319, 323, 370, 371, 374, 375, 377, 379, 381, 387, 412, 416, 417, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446], "straight": 6, "strang": 6, "stream": [2, 8, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 115, 116, 117, 118, 119, 122, 123, 126, 127, 128, 129, 130, 131, 134, 135, 136, 138, 139, 140, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 170, 171, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 309, 310, 313, 314, 315, 492, 500], "streamcontext": 290, "streamordevic": [0, 2], "street": 6, "strength": [473, 479], "strict": [124, 170, 187, 375, 377, 387], "strictli": [195, 224], "stride": [0, 2, 84, 99, 100, 101, 102, 103, 104, 105, 336, 337, 338, 341, 342, 343, 344, 345, 346, 366, 367, 368, 398, 495], "string": [0, 2, 132, 141, 146, 175, 216, 238, 493, 497, 499], "structur": [2, 474, 494], "stub": 9, "style": [2, 14, 17, 88, 89, 90, 129, 130, 134, 170, 171, 177, 186, 187, 188, 205, 210, 212, 228, 230, 234, 240, 261, 264, 291], "su": 6, "sub": [0, 7, 119, 256, 279, 280, 302, 316], "subarrai": [119, 283], "subclass": 464, "subdivid": 1, "subdtyp": 183, "subgradi": 469, "sublinear": 468, "submodul": [6, 7, 334, 371, 375, 376, 387, 389], "suboptim": 493, "subscript": [131, 132], "subsect": 6, "subsequ": 466, "subset": [334, 374], "substanti": 9, "subtl": 491, "subtract": [0, 39], "subtyp": [183, 326], "sudo": [9, 224], "sum": [0, 2, 5, 14, 112, 123, 143, 176, 195, 209, 281, 299, 302, 334, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 492, 495, 497], "sum_": [195, 441], "sum_i": 432, "sum_j": [454, 455], "summat": [131, 132], "super": [6, 7, 334, 464], "superset": [320, 474], "support": [1, 2, 6, 8, 9, 17, 91, 101, 104, 105, 149, 163, 172, 177, 189, 190, 192, 193, 194, 196, 197, 198, 200, 210, 243, 251, 492, 494, 495, 497, 499], "suppos": [494, 500], "sure": [2, 3, 6, 9, 334, 491], "surpass": [416, 417], "surpris": 6, "sw": 1, "swap": [0, 106, 223, 293, 389], "swapax": [0, 113], "swiglu": 6, "swish": [401, 453], "switch": 9, "symbol": 473, "symmetr": [100, 101, 104, 105, 189, 190, 192, 193], "symmetri": [192, 193], "synchron": [2, 491], "syntax": [39, 495], "synthet": 5, "sysctl": 224, "system": [4, 6, 9, 216, 217, 218, 224], "t": [0, 1, 2, 4, 6, 9, 135, 146, 149, 168, 189, 190, 244, 308, 334, 354, 360, 395, 467, 468, 469, 470, 471, 472, 473, 478, 479, 491, 493, 494, 500], "t_kv": 149, "t_q": 149, "tabl": [1, 195, 326, 351], "take": [0, 2, 6, 7, 88, 89, 90, 95, 141, 167, 169, 184, 212, 228, 236, 244, 296, 308, 311, 312, 315, 321, 322, 336, 337, 338, 366, 367, 368, 390, 433, 488, 492, 493, 494, 495, 499, 500, 501], "take_along_axi": [0, 495], "taken": [119, 295, 302], "talk": 492, "tan": 0, "tangent": [0, 2, 24, 25, 26, 113, 184, 297, 298, 410, 459], "tangent_i": 2, "tangent_x": 2, "tanh": [0, 334, 352, 354, 360, 369, 395, 424, 447], "target": [2, 308, 433, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 491], "target_include_directori": 2, "target_link_librari": [2, 4], "target_link_opt": 2, "target_sourc": 2, "task": [223, 441], "tau": 479, "tcp": 492, "tell": [4, 6, 491, 497], "temp": 6, "templat": [0, 1, 2, 146], "ten": 496, "tend": 473, "tensor": [200, 299, 446, 497], "tensordot": 0, "term": [2, 436, 467, 468, 469, 470, 471, 472, 478], "termin": 9, "test": [7, 9, 492], "test_imag": 7, "test_label": 7, "text": [6, 352, 354, 360, 369, 395, 402, 409, 414, 415, 416, 417, 424, 427, 428, 429, 436, 437, 438, 441, 442, 445, 447, 448, 451, 452, 457, 458, 468, 473], "textrm": [243, 352, 353, 423, 426], "tf": 497, "tgp_size": 2, "th": [109, 110, 111, 112, 118, 144, 192, 482], "than": [1, 2, 6, 79, 106, 119, 130, 148, 167, 170, 171, 187, 188, 189, 190, 192, 193, 194, 197, 198, 210, 222, 224, 318, 320, 398, 409, 412, 442, 445, 458, 468, 473, 491, 493, 494, 500], "thank": 496, "thei": [1, 2, 5, 6, 9, 17, 106, 168, 177, 400, 437, 464, 473, 490, 491, 492, 493, 496, 498, 499, 500], "them": [0, 2, 6, 122, 334, 375, 387, 492, 493, 500], "themselv": [2, 491], "thi": [0, 1, 2, 4, 6, 7, 9, 16, 17, 18, 19, 27, 28, 29, 30, 84, 113, 133, 141, 143, 146, 167, 168, 172, 175, 177, 184, 189, 190, 192, 193, 194, 195, 196, 197, 198, 205, 209, 210, 211, 213, 215, 217, 224, 227, 239, 241, 247, 274, 281, 282, 283, 288, 292, 295, 301, 309, 318, 321, 322, 334, 347, 348, 349, 353, 354, 360, 371, 372, 374, 375, 378, 379, 380, 385, 387, 388, 389, 390, 393, 395, 409, 414, 415, 416, 417, 424, 425, 426, 433, 441, 458, 464, 475, 490, 491, 492, 493, 494, 496, 497, 499], "thing": [2, 6, 492], "third": [191, 338, 368, 493], "thompson": 348, "those": [2, 6, 334], "though": [2, 6, 491, 493, 496, 497], "thousand": 496, "thread": [1, 2], "thread_index_in_simdgroup": 1, "thread_position_in_grid": [1, 2, 146], "threadgroup": [1, 2, 146], "threads_per_simdgroup": 1, "three": [6, 87, 338, 368, 412], "threefri": 488, "threshold": [409, 438, 445, 458], "through": [1, 2, 289, 411, 473, 491, 493, 494, 497], "throw": [2, 95, 124], "thu": [6, 334], "thumb": 466, "tic": 491, "tieleman": 478, "tile": [0, 149], "time": [2, 6, 9, 223, 300, 334, 354, 360, 395, 491, 494, 496, 500], "timeit": [491, 494], "titl": 2, "tmp": [1, 146], "to_quant": 316, "to_stream": 2, "toc": 491, "togeth": [0, 1, 2, 7, 243, 320, 321, 492], "tok_embed": 6, "token": [6, 351, 392], "told": 6, "toler": [0, 17, 177], "too": [183, 491, 496], "took": 6, "tool": 9, "top": [2, 301, 363, 412], "topk": 0, "torch": [6, 497], "torch_weight": 6, "total": [224, 494], "total_norm": 318, "tpi": 491, "trace": [0, 3, 143, 491], "trace_fil": 3, "tracer": 388, "track": [2, 334, 339], "track_running_stat": 339, "trade": 496, "tradit": [6, 148, 348, 349, 398], "train": [6, 7, 334, 339, 347, 348, 349, 373, 375, 387, 414, 415, 493], "train_imag": [7, 466], "train_label": [7, 466], "trainabl": [7, 317, 334, 464], "trainable_paramet": [334, 374, 475], "transform": [1, 6, 8, 113, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 172, 317, 334, 339, 355, 361, 363, 374, 375, 387, 393, 398, 495], "transformerencod": 271, "transit": 482, "translat": [145, 361], "transpos": [0, 6, 32, 103, 104, 105, 168, 244, 344, 345, 346], "treat": [0, 2, 113, 157, 158, 160, 161, 295, 412, 491], "tree": [8, 95, 137, 169, 308, 312, 319, 320, 321, 322, 323, 474, 475, 477, 486, 494], "tree_flatten": [271, 320, 323, 334, 466, 493], "tree_map": [321, 334, 492], "tree_unflatten": [6, 466, 493], "trembl": 6, "tri": 0, "triangl": [192, 193, 304], "triangular": [189, 190, 198], "trigger": 491, "tril": 0, "trilinear": 412, "triplet": 446, "triplet_loss": 334, "triu": 0, "true": [0, 1, 2, 4, 5, 6, 17, 42, 43, 44, 45, 83, 95, 109, 110, 111, 112, 146, 148, 168, 177, 183, 189, 190, 195, 200, 214, 223, 244, 281, 313, 316, 319, 320, 321, 322, 326, 334, 339, 341, 342, 343, 344, 345, 346, 354, 355, 359, 360, 361, 363, 374, 375, 377, 384, 387, 393, 395, 398, 403, 411, 412, 433, 441, 468, 470, 471, 491, 493], "truncat": [150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 257], "truth": [5, 435, 445], "try": [2, 9], "tupl": [0, 31, 66, 69, 80, 96, 100, 101, 102, 104, 105, 126, 130, 132, 137, 139, 141, 175, 184, 192, 195, 196, 197, 238, 243, 263, 265, 279, 280, 286, 307, 308, 311, 319, 320, 321, 322, 323, 336, 337, 338, 342, 343, 345, 346, 366, 367, 368, 377, 379, 400, 412, 468, 470, 471, 472, 473, 490, 493, 494], "tutori": 2, "twice": 500, "two": [0, 2, 14, 15, 17, 25, 83, 86, 88, 89, 90, 91, 119, 129, 134, 151, 154, 160, 167, 168, 170, 171, 177, 185, 187, 188, 189, 190, 191, 192, 193, 194, 196, 197, 198, 205, 210, 212, 228, 230, 234, 237, 293, 322, 337, 353, 360, 367, 426, 434, 491, 492, 493, 494, 495, 500], "txt": [2, 4], "type": [0, 1, 2, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 69, 79, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 116, 117, 118, 119, 122, 123, 124, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 138, 139, 140, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 216, 222, 223, 224, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 266, 267, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 318, 319, 322, 334, 382, 411, 413, 414, 415, 416, 417, 418, 419, 420, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 491, 493, 495], "type_to_nam": 2, "typenam": [0, 1, 2], "typic": [0, 149, 351, 466, 491, 496], "u": [1, 2, 4, 189, 192, 193, 197, 363, 389, 486, 496], "u_": 467, "u_t": 467, "uint": [1, 2, 146], "uint16": [12, 326], "uint3": 1, "uint32": [12, 27, 28, 29, 30, 247, 326], "uint64": [12, 326], "uint8": [12, 326], "ultra": 6, "unabl": 9, "unam": 9, "unari": 491, "unchang": [148, 289, 398], "uncheck": 9, "uncompress": 271, "undefin": [0, 29, 113, 189, 190, 239, 251, 495], "under": [2, 195], "underli": [2, 310], "understand": [6, 414, 415], "unevalu": 142, "unexpect": [2, 19], "unflatten": 0, "unfreez": [334, 375], "unfrozen": 387, "unifi": 8, "uniform": [3, 334, 363, 377, 415, 417, 461, 488, 491, 494, 500], "uniformli": 258, "unintend": 0, "union": [19, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 85, 86, 87, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 181, 182, 183, 185, 192, 193, 216, 269, 290], "uniqu": [2, 488], "unique_ptr": 2, "unit": [340, 350, 352, 353, 354, 362, 396, 397, 399, 401, 414, 415, 416, 417, 421, 422, 423, 424, 425, 426, 430, 449, 450, 451, 453], "unittest": 9, "univers": 195, "unless": [6, 17, 177, 195, 464], "unlik": [6, 17, 177, 348, 349, 383], "unnecessari": [2, 6], "unnorm": [247, 433, 435], "unscal": 468, "unsign": [168, 243, 244, 326], "unsignedinteg": 12, "unspecifi": [16, 18, 19, 27, 28, 29, 30, 96, 109, 110, 111, 112, 166, 209, 211, 213, 227, 235, 239, 241, 262, 281, 282, 288, 292, 295, 301, 302, 309, 314, 501], "unsqueez": 6, "unsupport": 200, "until": [2, 496, 498], "unus": 2, "up": [1, 2, 6, 113, 491], "upcast": 2, "updat": [0, 1, 2, 5, 6, 7, 39, 95, 280, 316, 320, 322, 339, 370, 371, 377, 382, 383, 384, 389, 466, 468, 471, 473, 474, 475, 479, 480, 481, 482, 483, 484, 491, 492, 493, 496], "update_modul": 334, "uplo": [192, 193], "upon": [6, 320, 321], "upper": [189, 190, 192, 193, 198, 243, 254, 257, 258, 420], "upsampl": 334, "us": [0, 3, 5, 6, 7, 8, 9, 19, 39, 84, 113, 117, 120, 122, 123, 126, 127, 128, 130, 146, 148, 163, 168, 186, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 210, 217, 218, 219, 222, 224, 243, 244, 262, 263, 264, 265, 294, 307, 319, 322, 326, 328, 334, 337, 338, 348, 351, 352, 354, 360, 363, 367, 368, 370, 374, 381, 388, 390, 392, 393, 395, 398, 403, 411, 412, 416, 417, 424, 425, 434, 461, 464, 466, 467, 468, 470, 471, 472, 473, 474, 475, 488, 490, 491, 492, 493, 494, 495, 498, 500], "usag": [411, 491], "user": [2, 6, 334], "usual": [351, 392, 490, 496], "util": [1, 2, 6, 8, 9, 271, 334, 466, 492], "v": [6, 106, 149, 192, 334, 375, 497], "v_": [467, 469, 470, 471, 472, 478, 479], "v_t": [467, 469, 470, 471, 472, 478, 479], "val": [0, 31, 166], "valid": [7, 106, 163, 312, 319, 375, 387, 490], "valid_parameter_filt": 370, "valu": [0, 1, 5, 6, 12, 13, 17, 19, 27, 28, 51, 79, 83, 94, 141, 144, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 166, 175, 177, 191, 195, 197, 199, 216, 224, 231, 238, 242, 246, 247, 248, 250, 251, 252, 254, 257, 258, 265, 269, 295, 296, 308, 312, 317, 319, 320, 321, 322, 326, 337, 338, 340, 347, 348, 349, 350, 356, 359, 363, 367, 368, 374, 390, 391, 407, 409, 411, 413, 433, 434, 435, 436, 437, 438, 440, 441, 442, 443, 444, 445, 458, 464, 468, 471, 480, 481, 483, 484, 494], "value_and_grad": [7, 113, 334, 388, 464, 466, 477, 491, 494, 497, 498], "value_and_grad_fn": 496, "value_cach": 6, "value_dim": 390, "value_input_dim": 390, "value_output_dim": 390, "value_proj": 6, "valueerror": [113, 195, 377, 494], "values_hat": 6, "van": 195, "var": [0, 339, 355, 359, 361, 436], "variabl": [9, 95, 113, 120, 133, 141, 142, 143, 169, 184, 308, 311, 312, 491, 492, 493], "varianc": [0, 288, 309, 339, 355, 436], "variant": [6, 445, 472], "variou": 195, "vector": [0, 2, 5, 8, 176, 184, 195, 295, 311, 312, 351, 435, 493, 498], "verbos": [1, 146], "veri": [6, 390, 492, 496, 500], "verifi": [5, 9], "versa": 265, "version": [2, 4, 9, 117, 141, 143, 175, 205, 209, 243, 281, 312, 488, 494, 495], "versu": 491, "via": [9, 113, 474, 477, 492, 496, 497], "vice": 265, "video": 349, "view": [0, 3, 84, 497], "virtual": 2, "visual": 142, "vjp": [2, 113, 498], "vmap": [2, 113, 493, 494, 496, 498], "vmap_add": 494, "vocab_s": 6, "vocabulari": [351, 392], "void": [1, 2], "vt": 197, "w": [0, 1, 5, 100, 101, 104, 105, 117, 168, 192, 243, 244, 308, 321, 339, 342, 343, 345, 346, 348, 349, 363, 466, 479, 494], "w1": [6, 318], "w2": [6, 318], "w3": 6, "w_": [354, 360, 395, 467, 468, 469, 470, 471, 472, 473, 478, 479], "w_1": 243, "w_g": 243, "w_i": [117, 243], "w_in": 1, "w_q": 243, "w_star": 5, "w_stride": 1, "w_t": [467, 469, 470, 471, 472, 473, 478, 479], "wa": [4, 6, 84, 126, 127, 492, 493, 496], "wai": [2, 6, 9, 334, 412, 491, 492, 493, 494, 495], "wait": [2, 6, 223], "walk": [6, 493], "walkthrough": 2, "walsh": 172, "want": [1, 6, 492, 493, 494, 500], "warm": [2, 491], "warmup": [482, 483], "warmup_init": 468, "watch": [6, 491], "wd": 473, "we": [0, 1, 2, 5, 6, 7, 113, 117, 126, 127, 168, 243, 244, 334, 351, 392, 400, 471, 473, 488, 490, 491, 492, 493, 494, 496, 500], "weight": [0, 5, 99, 100, 101, 102, 103, 104, 105, 145, 147, 320, 334, 377, 381, 392, 393, 433, 435, 464, 468, 471, 473, 475, 479, 494, 496], "weight_decai": [468, 471, 473, 479], "weight_fil": 6, "weights_fp16": 496, "well": [6, 334, 375, 387, 390, 496], "wen": 6, "went": 6, "were": [6, 500], "wet": 6, "what": [2, 6, 320], "whatsoev": 6, "whc": 348, "when": [0, 1, 2, 6, 8, 9, 95, 102, 113, 128, 189, 190, 192, 193, 194, 195, 197, 198, 200, 341, 342, 343, 344, 345, 346, 412, 416, 417, 433, 439, 445, 464, 466, 482, 488, 491, 492, 493, 500], "where": [0, 4, 7, 144, 177, 190, 243, 308, 312, 339, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 352, 354, 355, 359, 360, 361, 363, 374, 391, 394, 395, 409, 416, 417, 422, 423, 425, 436, 442, 448, 451, 453, 458, 475, 492, 494, 495], "wherea": 494, "whether": [141, 143, 146, 168, 192, 193, 198, 244, 354, 360, 374, 390, 395, 433, 436, 442], "which": [0, 1, 2, 6, 7, 8, 9, 19, 38, 84, 95, 102, 119, 122, 123, 126, 127, 128, 137, 141, 143, 148, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 169, 175, 178, 179, 180, 181, 182, 184, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 200, 214, 225, 243, 247, 248, 262, 263, 265, 268, 269, 270, 271, 272, 286, 287, 295, 302, 307, 308, 311, 312, 316, 337, 338, 348, 349, 352, 367, 368, 370, 374, 398, 433, 435, 438, 442, 445, 461, 474, 475, 488, 491, 492, 493, 494, 495, 496, 500, 501], "while": [2, 3, 6, 9, 263, 398, 496, 497], "whistl": 2, "who": 6, "whose": [144, 316, 317], "why": 6, "wide": 496, "width": [337, 338, 339, 342, 343, 345, 346, 348, 349, 367, 368, 392, 393], "window": [9, 336, 337, 338, 366, 367, 368], "wipe": 9, "wire": 224, "wired_limit_mb": 224, "wise": [0, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 88, 89, 90, 93, 107, 108, 129, 130, 134, 135, 136, 138, 140, 164, 165, 170, 171, 177, 186, 187, 188, 201, 202, 203, 204, 205, 206, 207, 208, 212, 228, 230, 232, 234, 240, 260, 261, 264, 267, 275, 276, 277, 278, 284, 285, 291, 297, 298, 340, 348, 349, 358, 369, 391, 402, 421, 428, 429, 431, 432, 447, 448, 450, 453, 454, 455, 456, 491], "wish": 9, "with_logit": 433, "within": [0, 3, 29, 177], "without": [1, 6, 8, 289, 390, 460, 490, 491, 492, 493, 496, 497, 500], "wk": 6, "wl": 2, "wo": 6, "word": 0, "work": [2, 3, 6, 223, 491, 492, 493, 494, 495, 496], "workhors": 334, "world": [323, 492], "worri": [1, 496], "would": [2, 6, 412, 492, 493, 495, 496, 497, 500], "wq": 6, "wrap": [113, 334], "wrapper": 493, "write": [0, 1, 2, 6, 334, 497], "written": [2, 493], "wrong": 493, "wrt": 317, "wv": 6, "x": [0, 1, 2, 4, 5, 6, 7, 39, 91, 113, 122, 123, 127, 128, 135, 140, 141, 142, 145, 146, 147, 168, 172, 173, 175, 195, 244, 248, 253, 266, 271, 275, 305, 306, 313, 320, 322, 334, 336, 337, 338, 339, 340, 350, 352, 353, 355, 359, 361, 362, 363, 366, 367, 368, 369, 370, 391, 394, 396, 402, 403, 409, 412, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 445, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 464, 466, 473, 491, 492, 493, 494, 495, 496, 497, 498, 500], "x1": 434, "x2": 434, "x86_64": 9, "x_1": [434, 442], "x_2": [434, 442], "x_cast": 2, "x_grad": 1, "x_i": [432, 454, 455], "x_j": [454, 455], "x_offset": 2, "x_ptr": 2, "x_shape": 1, "x_stride": 2, "x_t": [354, 360, 395], "x_view": 497, "xcode": 9, "xcodeproj": 3, "xcrun": 9, "xf": 360, "xg": 360, "xi": 360, "xn": 354, "xo": 360, "xor": 90, "xr": 354, "xy": [0, 214], "xz": 354, "x\u00b2": 497, "y": [0, 2, 4, 5, 6, 7, 39, 113, 141, 142, 172, 175, 313, 334, 339, 348, 355, 359, 361, 363, 394, 437, 442, 445, 466, 469, 491, 492, 493, 494, 496, 497], "y_": [437, 441], "y_cast": 2, "y_hat": 334, "y_offset": 2, "y_ptr": 2, "y_stride": 2, "ye": 6, "year": 6, "yet": [6, 195, 334, 464, 475, 494, 495, 496, 498], "yield": [6, 7, 488], "you": [2, 3, 4, 6, 7, 8, 9, 224, 334, 403, 411, 461, 488, 491, 492, 493, 494, 495, 497, 499, 500], "your": [2, 6, 9, 464, 494, 496], "z": [2, 354, 491, 493, 496], "z_t": 354, "zeiler": 467, "zero": [0, 141, 144, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 191, 214, 221, 280, 304, 305, 306, 315, 334, 336, 337, 338, 347, 348, 349, 377, 413, 414, 415, 416, 417, 418, 419, 420, 461, 466, 468, 493, 495], "zero_grad": 494, "zeros_lik": 0, "zhang": 6, "zip": [6, 7], "zip_saf": 2}, "titles": ["Operations", "Custom Metal Kernels", "Custom Extensions in MLX", "Metal Debugger", "Using MLX in C++", "Linear Regression", "LLM inference", "Multi-Layer Perceptron", "MLX", "Build and Install", "mlx.core.Device", "mlx.core.Dtype", "mlx.core.DtypeCategory", "mlx.core.abs", "mlx.core.add", "mlx.core.addmm", "mlx.core.all", "mlx.core.allclose", "mlx.core.any", "mlx.core.arange", "mlx.core.arccos", "mlx.core.arccosh", "mlx.core.arcsin", "mlx.core.arcsinh", "mlx.core.arctan", "mlx.core.arctan2", "mlx.core.arctanh", "mlx.core.argmax", "mlx.core.argmin", "mlx.core.argpartition", "mlx.core.argsort", "mlx.core.array", "mlx.core.array.T", "mlx.core.array.abs", "mlx.core.array.all", "mlx.core.array.any", "mlx.core.array.argmax", "mlx.core.array.argmin", "mlx.core.array.astype", "mlx.core.array.at", "mlx.core.array.conj", "mlx.core.array.cos", "mlx.core.array.cummax", "mlx.core.array.cummin", "mlx.core.array.cumprod", "mlx.core.array.cumsum", "mlx.core.array.diag", "mlx.core.array.diagonal", "mlx.core.array.dtype", "mlx.core.array.exp", "mlx.core.array.flatten", "mlx.core.array.item", "mlx.core.array.itemsize", "mlx.core.array.log", "mlx.core.array.log10", "mlx.core.array.log1p", "mlx.core.array.log2", "mlx.core.array.logsumexp", "mlx.core.array.max", "mlx.core.array.mean", "mlx.core.array.min", "mlx.core.array.moveaxis", "mlx.core.array.nbytes", "mlx.core.array.ndim", "mlx.core.array.prod", "mlx.core.array.reciprocal", "mlx.core.array.reshape", "mlx.core.array.round", "mlx.core.array.rsqrt", "mlx.core.array.shape", "mlx.core.array.sin", "mlx.core.array.size", "mlx.core.array.split", "mlx.core.array.sqrt", "mlx.core.array.square", "mlx.core.array.squeeze", "mlx.core.array.std", "mlx.core.array.sum", "mlx.core.array.swapaxes", "mlx.core.array.tolist", "mlx.core.array.transpose", "mlx.core.array.var", "mlx.core.array.view", "mlx.core.array_equal", "mlx.core.as_strided", "mlx.core.atleast_1d", "mlx.core.atleast_2d", "mlx.core.atleast_3d", "mlx.core.bitwise_and", "mlx.core.bitwise_or", "mlx.core.bitwise_xor", "mlx.core.block_masked_mm", "mlx.core.broadcast_to", "mlx.core.ceil", "mlx.core.clip", "mlx.core.compile", "mlx.core.concatenate", "mlx.core.conj", "mlx.core.conjugate", "mlx.core.conv1d", "mlx.core.conv2d", "mlx.core.conv3d", "mlx.core.conv_general", "mlx.core.conv_transpose1d", "mlx.core.conv_transpose2d", "mlx.core.conv_transpose3d", "mlx.core.convolve", "mlx.core.cos", "mlx.core.cosh", "mlx.core.cummax", "mlx.core.cummin", "mlx.core.cumprod", "mlx.core.cumsum", "mlx.core.custom_function", "mlx.core.default_device", "mlx.core.default_stream", "mlx.core.degrees", "mlx.core.dequantize", "mlx.core.diag", "mlx.core.diagonal", "mlx.core.disable_compile", "mlx.core.distributed.Group", "mlx.core.distributed.all_gather", "mlx.core.distributed.all_sum", "mlx.core.distributed.init", "mlx.core.distributed.is_available", "mlx.core.distributed.recv", "mlx.core.distributed.recv_like", "mlx.core.distributed.send", "mlx.core.divide", "mlx.core.divmod", "mlx.core.einsum", "mlx.core.einsum_path", "mlx.core.enable_compile", "mlx.core.equal", "mlx.core.erf", "mlx.core.erfinv", "mlx.core.eval", "mlx.core.exp", "mlx.core.expand_dims", "mlx.core.expm1", "mlx.core.export_function", "mlx.core.export_to_dot", "mlx.core.exporter", "mlx.core.eye", "mlx.core.fast.layer_norm", "mlx.core.fast.metal_kernel", "mlx.core.fast.rms_norm", "mlx.core.fast.rope", "mlx.core.fast.scaled_dot_product_attention", "mlx.core.fft.fft", "mlx.core.fft.fft2", "mlx.core.fft.fftn", "mlx.core.fft.ifft", "mlx.core.fft.ifft2", "mlx.core.fft.ifftn", "mlx.core.fft.irfft", "mlx.core.fft.irfft2", "mlx.core.fft.irfftn", "mlx.core.fft.rfft", "mlx.core.fft.rfft2", "mlx.core.fft.rfftn", "mlx.core.finfo", "mlx.core.flatten", "mlx.core.floor", "mlx.core.floor_divide", "mlx.core.full", "mlx.core.gather_mm", "mlx.core.gather_qmm", "mlx.core.grad", "mlx.core.greater", "mlx.core.greater_equal", "mlx.core.hadamard_transform", "mlx.core.identity", "mlx.core.imag", "mlx.core.import_function", "mlx.core.inner", "mlx.core.isclose", "mlx.core.isfinite", "mlx.core.isinf", "mlx.core.isnan", "mlx.core.isneginf", "mlx.core.isposinf", "mlx.core.issubdtype", "mlx.core.jvp", "mlx.core.kron", "mlx.core.left_shift", "mlx.core.less", "mlx.core.less_equal", "mlx.core.linalg.cholesky", "mlx.core.linalg.cholesky_inv", "mlx.core.linalg.cross", "mlx.core.linalg.eigh", "mlx.core.linalg.eigvalsh", "mlx.core.linalg.inv", "mlx.core.linalg.norm", "mlx.core.linalg.qr", "mlx.core.linalg.svd", "mlx.core.linalg.tri_inv", "mlx.core.linspace", "mlx.core.load", "mlx.core.log", "mlx.core.log10", "mlx.core.log1p", "mlx.core.log2", "mlx.core.logaddexp", "mlx.core.logical_and", "mlx.core.logical_not", "mlx.core.logical_or", "mlx.core.logsumexp", "mlx.core.matmul", "mlx.core.max", "mlx.core.maximum", "mlx.core.mean", "mlx.core.meshgrid", "mlx.core.metal.clear_cache", "mlx.core.metal.device_info", "mlx.core.metal.get_active_memory", "mlx.core.metal.get_cache_memory", "mlx.core.metal.get_peak_memory", "mlx.core.metal.is_available", "mlx.core.metal.reset_peak_memory", "mlx.core.metal.set_cache_limit", "mlx.core.metal.set_memory_limit", "mlx.core.metal.set_wired_limit", "mlx.core.metal.start_capture", "mlx.core.metal.stop_capture", "mlx.core.min", "mlx.core.minimum", "mlx.core.moveaxis", "mlx.core.multiply", "mlx.core.nan_to_num", "mlx.core.negative", "mlx.core.new_stream", "mlx.core.not_equal", "mlx.core.ones", "mlx.core.ones_like", "mlx.core.outer", "mlx.core.pad", "mlx.core.partition", "mlx.core.power", "mlx.core.prod", "mlx.core.put_along_axis", "mlx.core.quantize", "mlx.core.quantized_matmul", "mlx.core.radians", "mlx.core.random.bernoulli", "mlx.core.random.categorical", "mlx.core.random.gumbel", "mlx.core.random.key", "mlx.core.random.laplace", "mlx.core.random.multivariate_normal", "mlx.core.random.normal", "mlx.core.random.permutation", "mlx.core.random.randint", "mlx.core.random.seed", "mlx.core.random.split", "mlx.core.random.truncated_normal", "mlx.core.random.uniform", "mlx.core.real", "mlx.core.reciprocal", "mlx.core.remainder", "mlx.core.repeat", "mlx.core.reshape", "mlx.core.right_shift", "mlx.core.roll", "mlx.core.round", "mlx.core.rsqrt", "mlx.core.save", "mlx.core.save_gguf", "mlx.core.save_safetensors", "mlx.core.savez", "mlx.core.savez_compressed", "mlx.core.set_default_device", "mlx.core.set_default_stream", "mlx.core.sigmoid", "mlx.core.sign", "mlx.core.sin", "mlx.core.sinh", "mlx.core.slice", "mlx.core.slice_update", "mlx.core.softmax", "mlx.core.sort", "mlx.core.split", "mlx.core.sqrt", "mlx.core.square", "mlx.core.squeeze", "mlx.core.stack", "mlx.core.std", "mlx.core.stop_gradient", "mlx.core.stream", "mlx.core.subtract", "mlx.core.sum", "mlx.core.swapaxes", "mlx.core.synchronize", "mlx.core.take", "mlx.core.take_along_axis", "mlx.core.tan", "mlx.core.tanh", "mlx.core.tensordot", "mlx.core.tile", "mlx.core.topk", "mlx.core.trace", "mlx.core.transpose", "mlx.core.tri", "mlx.core.tril", "mlx.core.triu", "mlx.core.unflatten", "mlx.core.value_and_grad", "mlx.core.var", "mlx.core.view", "mlx.core.vjp", "mlx.core.vmap", "mlx.core.where", "mlx.core.zeros", "mlx.core.zeros_like", "mlx.nn.quantize", "mlx.nn.value_and_grad", "mlx.optimizers.clip_grad_norm", "mlx.utils.tree_flatten", "mlx.utils.tree_map", "mlx.utils.tree_map_with_path", "mlx.utils.tree_reduce", "mlx.utils.tree_unflatten", "mlx.core.Stream", "Array", "Data Types", "Devices and Streams", "Distributed Communication", "Export Functions", "Fast", "FFT", "Linear Algebra", "Metal", "Neural Networks", "mlx.nn.ALiBi", "mlx.nn.AvgPool1d", "mlx.nn.AvgPool2d", "mlx.nn.AvgPool3d", "mlx.nn.BatchNorm", "mlx.nn.CELU", "mlx.nn.Conv1d", "mlx.nn.Conv2d", "mlx.nn.Conv3d", "mlx.nn.ConvTranspose1d", "mlx.nn.ConvTranspose2d", "mlx.nn.ConvTranspose3d", "mlx.nn.Dropout", "mlx.nn.Dropout2d", "mlx.nn.Dropout3d", "mlx.nn.ELU", "mlx.nn.Embedding", "mlx.nn.GELU", "mlx.nn.GLU", "mlx.nn.GRU", "mlx.nn.GroupNorm", "mlx.nn.HardShrink", "mlx.nn.HardTanh", "mlx.nn.Hardswish", "mlx.nn.InstanceNorm", "mlx.nn.LSTM", "mlx.nn.LayerNorm", "mlx.nn.LeakyReLU", "mlx.nn.Linear", "mlx.nn.LogSigmoid", "mlx.nn.LogSoftmax", "mlx.nn.MaxPool1d", "mlx.nn.MaxPool2d", "mlx.nn.MaxPool3d", "mlx.nn.Mish", "mlx.nn.Module.apply", "mlx.nn.Module.apply_to_modules", "mlx.nn.Module.children", "mlx.nn.Module.eval", "mlx.nn.Module.filter_and_map", "mlx.nn.Module.freeze", "mlx.nn.Module.leaf_modules", "mlx.nn.Module.load_weights", "mlx.nn.Module.modules", "mlx.nn.Module.named_modules", "mlx.nn.Module.parameters", "mlx.nn.Module.save_weights", "mlx.nn.Module.set_dtype", "mlx.nn.Module.state", "mlx.nn.Module.train", "mlx.nn.Module.trainable_parameters", "mlx.nn.Module.training", "mlx.nn.Module.unfreeze", "mlx.nn.Module.update", "mlx.nn.Module.update_modules", "mlx.nn.MultiHeadAttention", "mlx.nn.PReLU", "mlx.nn.QuantizedEmbedding", "mlx.nn.QuantizedLinear", "mlx.nn.RMSNorm", "mlx.nn.RNN", "mlx.nn.ReLU", "mlx.nn.ReLU6", "mlx.nn.RoPE", "mlx.nn.SELU", "mlx.nn.Sequential", "mlx.nn.SiLU", "mlx.nn.Sigmoid", "mlx.nn.SinusoidalPositionalEncoding", "mlx.nn.Softmax", "mlx.nn.Softmin", "mlx.nn.Softplus", "mlx.nn.Softshrink", "mlx.nn.Softsign", "mlx.nn.Step", "mlx.nn.Tanh", "mlx.nn.Transformer", "mlx.nn.Upsample", "mlx.nn.init.constant", "mlx.nn.init.glorot_normal", "mlx.nn.init.glorot_uniform", "mlx.nn.init.he_normal", "mlx.nn.init.he_uniform", "mlx.nn.init.identity", "mlx.nn.init.normal", "mlx.nn.init.uniform", "mlx.nn.celu", "mlx.nn.elu", "mlx.nn.gelu", "mlx.nn.gelu_approx", "mlx.nn.gelu_fast_approx", "mlx.nn.glu", "mlx.nn.hard_shrink", "mlx.nn.hard_tanh", "mlx.nn.hardswish", "mlx.nn.leaky_relu", "mlx.nn.log_sigmoid", "mlx.nn.log_softmax", "mlx.nn.losses.binary_cross_entropy", "mlx.nn.losses.cosine_similarity_loss", "mlx.nn.losses.cross_entropy", "mlx.nn.losses.gaussian_nll_loss", "mlx.nn.losses.hinge_loss", "mlx.nn.losses.huber_loss", "mlx.nn.losses.kl_div_loss", "mlx.nn.losses.l1_loss", "mlx.nn.losses.log_cosh_loss", "mlx.nn.losses.margin_ranking_loss", "mlx.nn.losses.mse_loss", "mlx.nn.losses.nll_loss", "mlx.nn.losses.smooth_l1_loss", "mlx.nn.losses.triplet_loss", "mlx.nn.mish", "mlx.nn.prelu", "mlx.nn.relu", "mlx.nn.relu6", "mlx.nn.selu", "mlx.nn.sigmoid", "mlx.nn.silu", "mlx.nn.softmax", "mlx.nn.softmin", "mlx.nn.softplus", "mlx.nn.softshrink", "mlx.nn.step", "mlx.nn.tanh", "Functions", "Initializers", "Layers", "Loss Functions", "Module", "Operations", "Optimizers", "mlx.optimizers.AdaDelta", "mlx.optimizers.Adafactor", "mlx.optimizers.Adagrad", "mlx.optimizers.Adam", "mlx.optimizers.AdamW", "mlx.optimizers.Adamax", "mlx.optimizers.Lion", "mlx.optimizers.Optimizer.apply_gradients", "mlx.optimizers.Optimizer.init", "mlx.optimizers.Optimizer.state", "mlx.optimizers.Optimizer.update", "mlx.optimizers.RMSprop", "mlx.optimizers.SGD", "mlx.optimizers.cosine_decay", "mlx.optimizers.exponential_decay", "mlx.optimizers.join_schedules", "mlx.optimizers.linear_schedule", "mlx.optimizers.step_decay", "Common Optimizers", "Optimizer", "Schedulers", "Random", "Transforms", "Tree Utils", "Compilation", "Distributed Communication", "Exporting Functions", "Function Transforms", "Indexing Arrays", "Lazy Evaluation", "Conversion to NumPy and Other Frameworks", "Quick Start Guide", "Saving and Loading Arrays", "Unified Memory", "Using Streams"], "titleterms": {"A": 500, "In": 495, "The": 334, "ab": [13, 33], "adadelta": 467, "adafactor": 468, "adagrad": 469, "adam": 470, "adamax": 472, "adamw": 471, "add": 14, "addmm": 15, "algebra": 332, "alibi": 335, "all": [6, 16, 34, 492], "all_gath": 122, "all_sum": 123, "allclos": 17, "ani": [18, 35], "api": [8, 9], "appli": 370, "apply_gradi": 474, "apply_to_modul": 371, "arang": 19, "arcco": 20, "arccosh": 21, "arcsin": 22, "arcsinh": 23, "arctan": 24, "arctan2": 25, "arctanh": 26, "argmax": [27, 36], "argmin": [28, 37], "argpartit": 29, "argsort": 30, "arrai": [31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 325, 495, 499], "array_equ": 83, "as_strid": 84, "astyp": 38, "atleast_1d": 85, "atleast_2d": 86, "atleast_3d": 87, "attent": 6, "automat": 494, "avgpool1d": 336, "avgpool2d": 337, "avgpool3d": 338, "back": 2, "basic": [491, 493, 498], "batchnorm": 339, "benchmark": 6, "bernoulli": 246, "binari": 9, "binary_cross_entropi": 433, "bind": 2, "bitwise_and": 88, "bitwise_or": 89, "bitwise_xor": 90, "block_masked_mm": 91, "broadcast_to": 92, "build": [2, 9], "c": [4, 8, 9, 493], "categor": 247, "ceil": 93, "celu": [340, 421], "children": 372, "choleski": 189, "cholesky_inv": 190, "class": 334, "clear_cach": 215, "clip": 94, "clip_grad_norm": 318, "cmake": 2, "co": [41, 107], "code": [2, 6], "common": 485, "commun": [328, 492], "compil": [95, 491], "complex": 1, "comput": 496, "concaten": 96, "conj": [40, 97], "conjug": 98, "constant": 413, "conv1d": [99, 341], "conv2d": [100, 342], "conv3d": [101, 343], "conv_gener": 102, "conv_transpose1d": 103, "conv_transpose2d": 104, "conv_transpose3d": 105, "convers": 497, "convert": 6, "convolv": 106, "convtranspose1d": 344, "convtranspose2d": 345, "convtranspose3d": 346, "core": [10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 324], "cosh": 108, "cosine_decai": 480, "cosine_similarity_loss": 434, "cpu": 2, "cross": 191, "cross_entropi": 435, "cummax": [42, 109], "cummin": [43, 110], "cumprod": [44, 111], "cumsum": [45, 112], "custom": [1, 2], "custom_funct": 113, "data": 326, "debug": 491, "debugg": 3, "default_devic": 114, "default_stream": 115, "degre": 116, "dequant": 117, "devic": [10, 327], "device_info": 216, "diag": [46, 118], "diagon": [47, 119], "differ": 495, "differenti": 494, "disable_compil": 120, "distribut": [121, 122, 123, 124, 125, 126, 127, 128, 328, 492], "divid": 129, "divmod": 130, "download": [2, 6], "dropout": 347, "dropout2d": 348, "dropout3d": 349, "dtype": [11, 48], "dtypecategori": 12, "eigh": 192, "eigvalsh": 193, "einsum": 131, "einsum_path": 132, "elu": [350, 422], "embed": 351, "enable_compil": 133, "encod": 6, "end": 2, "equal": 134, "erf": 135, "erfinv": 136, "eval": [137, 373], "evalu": 496, "exampl": [1, 2, 8, 491, 492, 493, 500], "exp": [49, 138], "expand_dim": 139, "expm1": 140, "exponential_decai": 481, "export": [143, 329, 493], "export_funct": 141, "export_to_dot": 142, "extens": 2, "ey": 144, "fast": [145, 146, 147, 148, 149, 330], "fft": [150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 331], "fft2": 151, "fftn": 152, "filter_and_map": 374, "finfo": 162, "flatten": [50, 163], "floor": 164, "floor_divid": 165, "format": 499, "found": 9, "framework": 497, "freez": 375, "from": [9, 495], "full": [6, 166], "function": [329, 460, 463, 491, 493, 494, 498], "further": 8, "gather_mm": 167, "gather_qmm": 168, "gaussian_nll_loss": 436, "gelu": [352, 423], "gelu_approx": 424, "gelu_fast_approx": 425, "gener": 6, "get": 492, "get_active_memori": 217, "get_cache_memori": 218, "get_peak_memori": 219, "glorot_norm": 414, "glorot_uniform": 415, "glu": [353, 426], "gpu": 2, "grad": [169, 334], "graph": [491, 496, 498], "greater": 170, "greater_equ": 171, "grid": 1, "group": 121, "groupnorm": 355, "gru": 354, "guid": 498, "gumbel": 248, "hadamard_transform": 172, "hard_shrink": 427, "hard_tanh": 428, "hardshrink": 356, "hardswish": [358, 429], "hardtanh": 357, "he_norm": 416, "he_uniform": 417, "hinge_loss": 437, "host": 492, "huber_loss": 438, "ident": [173, 418], "ifft": 153, "ifft2": 154, "ifftn": 155, "imag": 174, "implement": [2, 6], "import": 493, "import_funct": 175, "index": 495, "infer": 6, "init": [124, 413, 414, 415, 416, 417, 418, 419, 420, 475], "initi": 461, "inner": 176, "inspect": 334, "instal": [8, 9, 492], "instancenorm": 359, "introduc": 2, "inv": 194, "irfft": 156, "irfft2": 157, "irfftn": 158, "is_avail": [125, 220], "isclos": 177, "isfinit": 178, "isinf": 179, "isnan": 180, "isneginf": 181, "isposinf": 182, "issubdtyp": 183, "item": 51, "items": 52, "jax": 497, "join_schedul": 482, "jvp": 184, "kei": 249, "kernel": 1, "kl_div_loss": 439, "kron": 185, "l1_loss": 440, "laplac": 250, "layer": [6, 7, 462], "layer_norm": 145, "layernorm": 361, "lazi": 496, "leaf_modul": 376, "leaky_relu": 430, "leakyrelu": 362, "left_shift": 186, "less": 187, "less_equ": 188, "linalg": [189, 190, 191, 192, 193, 194, 195, 196, 197, 198], "linear": [5, 332, 363], "linear_schedul": 483, "linspac": 199, "lion": 473, "llm": 6, "load": [6, 200, 466, 499], "load_weight": 377, "log": [53, 201], "log10": [54, 202], "log1p": [55, 203], "log2": [56, 204], "log_cosh_loss": 441, "log_sigmoid": 431, "log_softmax": 432, "logaddexp": 205, "logical_and": 206, "logical_not": 207, "logical_or": 208, "logsigmoid": 364, "logsoftmax": 365, "logsumexp": [57, 209], "loss": [433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 463], "lstm": 360, "margin_ranking_loss": 442, "matmul": 210, "max": [58, 211], "maximum": 212, "maxpool1d": 366, "maxpool2d": 367, "maxpool3d": 368, "mean": [59, 213], "memori": 500, "meshgrid": 214, "metal": [1, 3, 9, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 333], "metal_kernel": 146, "min": [60, 227], "minim": 9, "minimum": 228, "mish": [369, 447], "mlx": [2, 4, 8, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484], "model": 6, "modul": [334, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 464, 493], "more": 493, "moveaxi": [61, 229], "mpi": 492, "mse_loss": 443, "multi": 7, "multiheadattent": 390, "multipl": 493, "multipli": 230, "multivariate_norm": 251, "named_modul": 379, "nan_to_num": 231, "nbyte": 62, "ndim": 63, "neg": 232, "network": 334, "neural": 334, "new_stream": 233, "nll_loss": 444, "nn": [316, 317, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459], "norm": 195, "normal": [252, 419], "not_equ": 234, "numpi": [495, 497], "ones": 235, "ones_lik": 236, "onli": 496, "oper": [0, 2, 465], "optim": [318, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486], "option": 9, "other": 497, "outer": 237, "packag": 4, "pad": 238, "paramet": [334, 380], "partit": 239, "perceptron": 7, "permut": 253, "place": 495, "power": 240, "prelu": [391, 448], "primit": 2, "prod": [64, 241], "pure": 491, "put": 6, "put_along_axi": 242, "python": [2, 8, 9], "pytorch": 497, "qr": 196, "quantiz": [243, 316], "quantized_matmul": 244, "quantizedembed": 392, "quantizedlinear": 393, "quick": [334, 498], "radian": 245, "randint": 254, "random": [246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 488], "read": 8, "real": 259, "reciproc": [65, 260], "recv": 126, "recv_lik": 127, "reduc": 492, "refer": 8, "regress": 5, "relu": [396, 449], "relu6": [397, 450], "remaind": 261, "remot": 492, "repeat": 262, "requir": 9, "reset_peak_memori": 221, "reshap": [66, 263], "result": 2, "rfft": 159, "rfft2": 160, "rfftn": 161, "right_shift": 264, "rms_norm": 147, "rmsnorm": 394, "rmsprop": 478, "rnn": 395, "roll": 265, "rope": [148, 398], "round": [67, 266], "rsqrt": [68, 267], "sampl": 1, "save": [268, 466, 499], "save_gguf": 269, "save_safetensor": 270, "save_weight": 381, "savez": 271, "savez_compress": 272, "scaled_dot_product_attent": 149, "schedul": 487, "script": [2, 6], "seed": 255, "selu": [399, 451], "send": 128, "sequenti": 400, "serial": 499, "set": 492, "set_cache_limit": 222, "set_default_devic": 273, "set_default_stream": 274, "set_dtyp": 382, "set_memory_limit": 223, "set_wired_limit": 224, "setuptool": 2, "sgd": 479, "shape": [1, 69], "shapeless": [491, 493], "shell": 9, "sigmoid": [275, 402, 452], "sign": 276, "silu": [401, 453], "simpl": [1, 500], "sin": [70, 277], "sinh": 278, "sinusoidalpositionalencod": 403, "size": [9, 71], "slice": 279, "slice_upd": 280, "smooth_l1_loss": 445, "softmax": [281, 404, 454], "softmin": [405, 455], "softplu": [406, 456], "softshrink": [407, 457], "softsign": 408, "sort": 282, "sourc": 9, "specifi": 501, "speedup": 491, "split": [72, 256, 283], "sqrt": [73, 284], "squar": [74, 285], "squeez": [75, 286], "stack": 287, "start": [334, 492, 498], "start_captur": 225, "state": [383, 476], "std": [76, 288], "step": [409, 458], "step_decai": 484, "stop_captur": 226, "stop_gradi": 289, "stream": [290, 324, 327, 501], "stride": 1, "subtract": 291, "sum": [77, 292], "support": 326, "svd": 197, "swapax": [78, 293], "synchron": 294, "t": 32, "take": 295, "take_along_axi": 296, "tan": 297, "tanh": [298, 410, 459], "tensordot": 299, "tensorflow": 497, "tile": 300, "togeth": 6, "tolist": 79, "topk": 301, "trace": [302, 493], "train": [384, 386, 491, 492], "trainable_paramet": 385, "transform": [2, 411, 489, 491, 493, 494, 496, 498], "transpos": [80, 303], "tree": 490, "tree_flatten": 319, "tree_map": 320, "tree_map_with_path": 321, "tree_reduc": 322, "tree_unflatten": 323, "tri": 304, "tri_inv": 198, "tril": 305, "triplet_loss": 446, "triu": 306, "troubleshoot": 9, "truncated_norm": 257, "tune": 492, "type": 326, "unflatten": 307, "unfreez": 387, "unifi": 500, "uniform": [258, 420], "up": 492, "updat": [334, 388, 477, 495], "update_modul": 389, "upsampl": 412, "us": [1, 2, 4, 496, 501], "usag": [2, 8], "util": [319, 320, 321, 322, 323, 490], "valu": 334, "value_and_grad": [308, 317], "var": [81, 309], "variabl": 4, "vector": 494, "view": [82, 310], "vjp": [1, 311], "vmap": 312, "weight": 6, "what": 496, "when": 496, "where": 313, "why": 496, "workflow": 3, "x86": 9, "xcode": 3, "you": 496, "zero": 314, "zeros_lik": 315}}) \ No newline at end of file +Search.setIndex({"alltitles": {"A Simple Example": [[500, "a-simple-example"]], "Array": [[325, null]], "Attention layer": [[6, "attention-layer"]], "Automatic Differentiation": [[494, "automatic-differentiation"]], "Automatic Vectorization": [[494, "automatic-vectorization"]], "Basics": [[498, "basics"]], "Basics of Compile": [[491, "basics-of-compile"]], "Basics of Exporting": [[493, "basics-of-exporting"]], "Binary Size Minimization": [[9, "binary-size-minimization"]], "Binding to Python": [[2, "binding-to-python"]], "Build Options": [[9, "id4"]], "Build Requirements": [[9, "build-requirements"]], "Build and Install": [[9, null]], "Build from source": [[9, "build-from-source"]], "Building and Binding": [[2, "building-and-binding"]], "Building with CMake": [[2, "building-with-cmake"]], "Building with setuptools": [[2, "building-with-setuptools"]], "C++ API": [[9, "c-api"]], "C++ API Reference": [[8, null]], "Common Optimizers": [[485, null]], "Compilation": [[491, null]], "Compiling Training Graphs": [[491, "compiling-training-graphs"]], "Complex Example": [[1, "complex-example"]], "Conversion to NumPy and Other Frameworks": [[497, null]], "Converting the weights": [[6, "converting-the-weights"]], "Custom Extensions in MLX": [[2, null]], "Custom Metal Kernels": [[1, null]], "Data Types": [[326, null]], "Debugging": [[491, "debugging"]], "Devices and Streams": [[327, null]], "Differences from NumPy": [[495, "differences-from-numpy"]], "Distributed Communication": [[328, null], [492, null]], "Download the code": [[2, null], [6, null]], "Encoder layer": [[6, "encoder-layer"]], "Example Speedup": [[491, "example-speedup"]], "Examples": [[8, null]], "Export Functions": [[329, null]], "Exporting Functions": [[493, null]], "Exporting Modules": [[493, "exporting-modules"]], "Exporting Multiple Traces": [[493, "exporting-multiple-traces"]], "FFT": [[331, null]], "Fast": [[330, null]], "Full model": [[6, "full-model"]], "Function Transforms": [[494, null]], "Function and Graph Transformations": [[498, "function-and-graph-transformations"]], "Functions": [[460, null]], "Further Reading": [[8, null]], "Generation": [[6, "generation"]], "Getting Started": [[492, "getting-started"]], "Grid Sample VJP": [[1, "grid-sample-vjp"]], "Implementing the CPU Back-end": [[2, "implementing-the-cpu-back-end"]], "Implementing the GPU Back-end": [[2, "implementing-the-gpu-back-end"]], "Implementing the Primitive": [[2, "implementing-the-primitive"]], "Implementing the model": [[6, "implementing-the-model"]], "Importing Functions in C++": [[493, "importing-functions-in-c"]], "In Place Updates": [[495, "in-place-updates"]], "Indexing Arrays": [[495, null]], "Initializers": [[461, null]], "Inspecting Modules": [[334, "inspecting-modules"]], "Install": [[8, null]], "Installing MPI": [[492, "installing-mpi"]], "Introducing the Example": [[2, "introducing-the-example"]], "JAX": [[497, "jax"]], "LLM inference": [[6, null]], "Layers": [[462, null]], "Lazy Evaluation": [[496, null]], "Linear Algebra": [[332, null]], "Linear Regression": [[5, null]], "Loss Functions": [[463, null]], "MLX": [[8, null]], "Metal": [[333, null]], "Metal Debugger": [[3, null]], "Metal not found": [[9, "metal-not-found"]], "Module": [[464, null]], "More Examples": [[493, "more-examples"]], "Multi-Layer Perceptron": [[7, null]], "Neural Networks": [[334, null]], "Only Compute What You Use": [[496, "only-compute-what-you-use"]], "Operations": [[0, null], [2, "operations"], [465, null]], "Operations and Primitives": [[2, "operations-and-primitives"]], "Optimizer": [[486, null]], "Optimizers": [[466, null]], "Package Variables": [[4, "id1"]], "Parameters": [[334, "parameters"]], "Primitive Transforms": [[2, "primitive-transforms"]], "Primitives": [[2, "primitives"]], "Pure Functions": [[491, "pure-functions"]], "Putting it all together": [[6, "putting-it-all-together"]], "PyTorch": [[497, "pytorch"]], "Python API": [[9, "python-api"]], "Python API Reference": [[8, null]], "Python Installation": [[9, "python-installation"]], "Quick Start Guide": [[498, null]], "Quick Start with Neural Networks": [[334, "quick-start-with-neural-networks"]], "Random": [[488, null]], "Results": [[2, "results"]], "Saving and Loading": [[466, "saving-and-loading"]], "Saving and Loading Arrays": [[499, null]], "Schedulers": [[487, null]], "Scripts": [[2, "scripts"], [6, "scripts"]], "Serialization Formats": [[499, "id1"]], "Setting up Remote Hosts": [[492, "setting-up-remote-hosts"]], "Shapeless Compilation": [[491, "shapeless-compilation"]], "Shapeless Exports": [[493, "shapeless-exports"]], "Simple Example": [[1, "simple-example"]], "Specifying the Stream": [[501, "specifying-the-stream"]], "Supported Data Types": [[326, "id2"]], "TensorFlow": [[497, "tensorflow"]], "The Module Class": [[334, "the-module-class"]], "Training Example": [[492, "training-example"]], "Transformations with Compile": [[491, "transformations-with-compile"]], "Transformations with Imported Functions": [[493, "transformations-with-imported-functions"]], "Transforming Compute Graphs": [[496, "transforming-compute-graphs"]], "Transforms": [[489, null]], "Tree Utils": [[490, null]], "Troubleshooting": [[9, "troubleshooting"], [9, "id3"]], "Tuning All Reduce": [[492, "tuning-all-reduce"]], "Unified Memory": [[500, null]], "Updating the Parameters": [[334, "updating-the-parameters"]], "Usage": [[2, "usage"], [8, null]], "Using MLX in C++": [[4, null]], "Using Shape/Strides": [[1, "using-shape-strides"]], "Using Streams": [[501, null]], "Using the Primitive": [[2, "using-the-primitive"]], "Value and Grad": [[334, "value-and-grad"]], "Weight loading and benchmarking": [[6, "weight-loading-and-benchmarking"]], "When to Evaluate": [[496, "when-to-evaluate"]], "Why Lazy Evaluation": [[496, "why-lazy-evaluation"]], "Xcode Workflow": [[3, "xcode-workflow"]], "mlx.core.Device": [[10, null]], "mlx.core.Dtype": [[11, null]], "mlx.core.DtypeCategory": [[12, null]], "mlx.core.Stream": [[324, null]], "mlx.core.abs": [[13, null]], "mlx.core.add": [[14, null]], "mlx.core.addmm": [[15, null]], "mlx.core.all": [[16, null]], "mlx.core.allclose": [[17, null]], "mlx.core.any": [[18, null]], "mlx.core.arange": [[19, null]], "mlx.core.arccos": [[20, null]], "mlx.core.arccosh": [[21, null]], "mlx.core.arcsin": [[22, null]], "mlx.core.arcsinh": [[23, null]], "mlx.core.arctan": [[24, null]], "mlx.core.arctan2": [[25, null]], "mlx.core.arctanh": [[26, null]], "mlx.core.argmax": [[27, null]], "mlx.core.argmin": [[28, null]], "mlx.core.argpartition": [[29, null]], "mlx.core.argsort": [[30, null]], "mlx.core.array": [[31, null]], "mlx.core.array.T": [[32, null]], "mlx.core.array.abs": [[33, null]], "mlx.core.array.all": [[34, null]], "mlx.core.array.any": [[35, null]], "mlx.core.array.argmax": [[36, null]], "mlx.core.array.argmin": [[37, null]], "mlx.core.array.astype": [[38, null]], "mlx.core.array.at": [[39, null]], "mlx.core.array.conj": [[40, null]], "mlx.core.array.cos": [[41, null]], "mlx.core.array.cummax": [[42, null]], "mlx.core.array.cummin": [[43, null]], "mlx.core.array.cumprod": [[44, null]], "mlx.core.array.cumsum": [[45, null]], "mlx.core.array.diag": [[46, null]], "mlx.core.array.diagonal": [[47, null]], "mlx.core.array.dtype": [[48, null]], "mlx.core.array.exp": [[49, null]], "mlx.core.array.flatten": [[50, null]], "mlx.core.array.item": [[51, null]], "mlx.core.array.itemsize": [[52, null]], "mlx.core.array.log": [[53, null]], "mlx.core.array.log10": [[54, null]], "mlx.core.array.log1p": [[55, null]], "mlx.core.array.log2": [[56, null]], "mlx.core.array.logsumexp": [[57, null]], "mlx.core.array.max": [[58, null]], "mlx.core.array.mean": [[59, null]], "mlx.core.array.min": [[60, null]], "mlx.core.array.moveaxis": [[61, null]], "mlx.core.array.nbytes": [[62, null]], "mlx.core.array.ndim": [[63, null]], "mlx.core.array.prod": [[64, null]], "mlx.core.array.reciprocal": [[65, null]], "mlx.core.array.reshape": [[66, null]], "mlx.core.array.round": [[67, null]], "mlx.core.array.rsqrt": [[68, null]], "mlx.core.array.shape": [[69, null]], "mlx.core.array.sin": [[70, null]], "mlx.core.array.size": [[71, null]], "mlx.core.array.split": [[72, null]], "mlx.core.array.sqrt": [[73, null]], "mlx.core.array.square": [[74, null]], "mlx.core.array.squeeze": [[75, null]], "mlx.core.array.std": [[76, null]], "mlx.core.array.sum": [[77, null]], "mlx.core.array.swapaxes": [[78, null]], "mlx.core.array.tolist": [[79, null]], "mlx.core.array.transpose": [[80, null]], "mlx.core.array.var": [[81, null]], "mlx.core.array.view": [[82, null]], "mlx.core.array_equal": [[83, null]], "mlx.core.as_strided": [[84, null]], "mlx.core.atleast_1d": [[85, null]], "mlx.core.atleast_2d": [[86, null]], "mlx.core.atleast_3d": [[87, null]], "mlx.core.bitwise_and": [[88, null]], "mlx.core.bitwise_or": [[89, null]], "mlx.core.bitwise_xor": [[90, null]], "mlx.core.block_masked_mm": [[91, null]], "mlx.core.broadcast_to": [[92, null]], "mlx.core.ceil": [[93, null]], "mlx.core.clip": [[94, null]], "mlx.core.compile": [[95, null]], "mlx.core.concatenate": [[96, null]], "mlx.core.conj": [[97, null]], "mlx.core.conjugate": [[98, null]], "mlx.core.conv1d": [[99, null]], "mlx.core.conv2d": [[100, null]], "mlx.core.conv3d": [[101, null]], "mlx.core.conv_general": [[102, null]], "mlx.core.conv_transpose1d": [[103, null]], "mlx.core.conv_transpose2d": [[104, null]], "mlx.core.conv_transpose3d": [[105, null]], "mlx.core.convolve": [[106, null]], "mlx.core.cos": [[107, null]], "mlx.core.cosh": [[108, null]], "mlx.core.cummax": [[109, null]], "mlx.core.cummin": [[110, null]], "mlx.core.cumprod": [[111, null]], "mlx.core.cumsum": [[112, null]], "mlx.core.custom_function": [[113, null]], "mlx.core.default_device": [[114, null]], "mlx.core.default_stream": [[115, null]], "mlx.core.degrees": [[116, null]], "mlx.core.dequantize": [[117, null]], "mlx.core.diag": [[118, null]], "mlx.core.diagonal": [[119, null]], "mlx.core.disable_compile": [[120, null]], "mlx.core.distributed.Group": [[121, null]], "mlx.core.distributed.all_gather": [[122, null]], "mlx.core.distributed.all_sum": [[123, null]], "mlx.core.distributed.init": [[124, null]], "mlx.core.distributed.is_available": [[125, null]], "mlx.core.distributed.recv": [[126, null]], "mlx.core.distributed.recv_like": [[127, null]], "mlx.core.distributed.send": [[128, null]], "mlx.core.divide": [[129, null]], "mlx.core.divmod": [[130, null]], "mlx.core.einsum": [[131, null]], "mlx.core.einsum_path": [[132, null]], "mlx.core.enable_compile": [[133, null]], "mlx.core.equal": [[134, null]], "mlx.core.erf": [[135, null]], "mlx.core.erfinv": [[136, null]], "mlx.core.eval": [[137, null]], "mlx.core.exp": [[138, null]], "mlx.core.expand_dims": [[139, null]], "mlx.core.expm1": [[140, null]], "mlx.core.export_function": [[141, null]], "mlx.core.export_to_dot": [[142, null]], "mlx.core.exporter": [[143, null]], "mlx.core.eye": [[144, null]], "mlx.core.fast.layer_norm": [[145, null]], "mlx.core.fast.metal_kernel": [[146, null]], "mlx.core.fast.rms_norm": [[147, null]], "mlx.core.fast.rope": [[148, null]], "mlx.core.fast.scaled_dot_product_attention": [[149, null]], "mlx.core.fft.fft": [[150, null]], "mlx.core.fft.fft2": [[151, null]], "mlx.core.fft.fftn": [[152, null]], "mlx.core.fft.ifft": [[153, null]], "mlx.core.fft.ifft2": [[154, null]], "mlx.core.fft.ifftn": [[155, null]], "mlx.core.fft.irfft": [[156, null]], "mlx.core.fft.irfft2": [[157, null]], "mlx.core.fft.irfftn": [[158, null]], "mlx.core.fft.rfft": [[159, null]], "mlx.core.fft.rfft2": [[160, null]], "mlx.core.fft.rfftn": [[161, null]], "mlx.core.finfo": [[162, null]], "mlx.core.flatten": [[163, null]], "mlx.core.floor": [[164, null]], "mlx.core.floor_divide": [[165, null]], "mlx.core.full": [[166, null]], "mlx.core.gather_mm": [[167, null]], "mlx.core.gather_qmm": [[168, null]], "mlx.core.grad": [[169, null]], "mlx.core.greater": [[170, null]], "mlx.core.greater_equal": [[171, null]], "mlx.core.hadamard_transform": [[172, null]], "mlx.core.identity": [[173, null]], "mlx.core.imag": [[174, null]], "mlx.core.import_function": [[175, null]], "mlx.core.inner": [[176, null]], "mlx.core.isclose": [[177, null]], "mlx.core.isfinite": [[178, null]], "mlx.core.isinf": [[179, null]], "mlx.core.isnan": [[180, null]], "mlx.core.isneginf": [[181, null]], "mlx.core.isposinf": [[182, null]], "mlx.core.issubdtype": [[183, null]], "mlx.core.jvp": [[184, null]], "mlx.core.kron": [[185, null]], "mlx.core.left_shift": [[186, null]], "mlx.core.less": [[187, null]], "mlx.core.less_equal": [[188, null]], "mlx.core.linalg.cholesky": [[189, null]], "mlx.core.linalg.cholesky_inv": [[190, null]], "mlx.core.linalg.cross": [[191, null]], "mlx.core.linalg.eigh": [[192, null]], "mlx.core.linalg.eigvalsh": [[193, null]], "mlx.core.linalg.inv": [[194, null]], "mlx.core.linalg.norm": [[195, null]], "mlx.core.linalg.qr": [[196, null]], "mlx.core.linalg.svd": [[197, null]], "mlx.core.linalg.tri_inv": [[198, null]], "mlx.core.linspace": [[199, null]], "mlx.core.load": [[200, null]], "mlx.core.log": [[201, null]], "mlx.core.log10": [[202, null]], "mlx.core.log1p": [[203, null]], "mlx.core.log2": [[204, null]], "mlx.core.logaddexp": [[205, null]], "mlx.core.logical_and": [[206, null]], "mlx.core.logical_not": [[207, null]], "mlx.core.logical_or": [[208, null]], "mlx.core.logsumexp": [[209, null]], "mlx.core.matmul": [[210, null]], "mlx.core.max": [[211, null]], "mlx.core.maximum": [[212, null]], "mlx.core.mean": [[213, null]], "mlx.core.meshgrid": [[214, null]], "mlx.core.metal.clear_cache": [[215, null]], "mlx.core.metal.device_info": [[216, null]], "mlx.core.metal.get_active_memory": [[217, null]], "mlx.core.metal.get_cache_memory": [[218, null]], "mlx.core.metal.get_peak_memory": [[219, null]], "mlx.core.metal.is_available": [[220, null]], "mlx.core.metal.reset_peak_memory": [[221, null]], "mlx.core.metal.set_cache_limit": [[222, null]], "mlx.core.metal.set_memory_limit": [[223, null]], "mlx.core.metal.set_wired_limit": [[224, null]], "mlx.core.metal.start_capture": [[225, null]], "mlx.core.metal.stop_capture": [[226, null]], "mlx.core.min": [[227, null]], "mlx.core.minimum": [[228, null]], "mlx.core.moveaxis": [[229, null]], "mlx.core.multiply": [[230, null]], "mlx.core.nan_to_num": [[231, null]], "mlx.core.negative": [[232, null]], "mlx.core.new_stream": [[233, null]], "mlx.core.not_equal": [[234, null]], "mlx.core.ones": [[235, null]], "mlx.core.ones_like": [[236, null]], "mlx.core.outer": [[237, null]], "mlx.core.pad": [[238, null]], "mlx.core.partition": [[239, null]], "mlx.core.power": [[240, null]], "mlx.core.prod": [[241, null]], "mlx.core.put_along_axis": [[242, null]], "mlx.core.quantize": [[243, null]], "mlx.core.quantized_matmul": [[244, null]], "mlx.core.radians": [[245, null]], "mlx.core.random.bernoulli": [[246, null]], "mlx.core.random.categorical": [[247, null]], "mlx.core.random.gumbel": [[248, null]], "mlx.core.random.key": [[249, null]], "mlx.core.random.laplace": [[250, null]], "mlx.core.random.multivariate_normal": [[251, null]], "mlx.core.random.normal": [[252, null]], "mlx.core.random.permutation": [[253, null]], "mlx.core.random.randint": [[254, null]], "mlx.core.random.seed": [[255, null]], "mlx.core.random.split": [[256, null]], "mlx.core.random.truncated_normal": [[257, null]], "mlx.core.random.uniform": [[258, null]], "mlx.core.real": [[259, null]], "mlx.core.reciprocal": [[260, null]], "mlx.core.remainder": [[261, null]], "mlx.core.repeat": [[262, null]], "mlx.core.reshape": [[263, null]], "mlx.core.right_shift": [[264, null]], "mlx.core.roll": [[265, null]], "mlx.core.round": [[266, null]], "mlx.core.rsqrt": [[267, null]], "mlx.core.save": [[268, null]], "mlx.core.save_gguf": [[269, null]], "mlx.core.save_safetensors": [[270, null]], "mlx.core.savez": [[271, null]], "mlx.core.savez_compressed": [[272, null]], "mlx.core.set_default_device": [[273, null]], "mlx.core.set_default_stream": [[274, null]], "mlx.core.sigmoid": [[275, null]], "mlx.core.sign": [[276, null]], "mlx.core.sin": [[277, null]], "mlx.core.sinh": [[278, null]], "mlx.core.slice": [[279, null]], "mlx.core.slice_update": [[280, null]], "mlx.core.softmax": [[281, null]], "mlx.core.sort": [[282, null]], "mlx.core.split": [[283, null]], "mlx.core.sqrt": [[284, null]], "mlx.core.square": [[285, null]], "mlx.core.squeeze": [[286, null]], "mlx.core.stack": [[287, null]], "mlx.core.std": [[288, null]], "mlx.core.stop_gradient": [[289, null]], "mlx.core.stream": [[290, null]], "mlx.core.subtract": [[291, null]], "mlx.core.sum": [[292, null]], "mlx.core.swapaxes": [[293, null]], "mlx.core.synchronize": [[294, null]], "mlx.core.take": [[295, null]], "mlx.core.take_along_axis": [[296, null]], "mlx.core.tan": [[297, null]], "mlx.core.tanh": [[298, null]], "mlx.core.tensordot": [[299, null]], "mlx.core.tile": [[300, null]], "mlx.core.topk": [[301, null]], "mlx.core.trace": [[302, null]], "mlx.core.transpose": [[303, null]], "mlx.core.tri": [[304, null]], "mlx.core.tril": [[305, null]], "mlx.core.triu": [[306, null]], "mlx.core.unflatten": [[307, null]], "mlx.core.value_and_grad": [[308, null]], "mlx.core.var": [[309, null]], "mlx.core.view": [[310, null]], "mlx.core.vjp": [[311, null]], "mlx.core.vmap": [[312, null]], "mlx.core.where": [[313, null]], "mlx.core.zeros": [[314, null]], "mlx.core.zeros_like": [[315, null]], "mlx.nn.ALiBi": [[335, null]], "mlx.nn.AvgPool1d": [[336, null]], "mlx.nn.AvgPool2d": [[337, null]], "mlx.nn.AvgPool3d": [[338, null]], "mlx.nn.BatchNorm": [[339, null]], "mlx.nn.CELU": [[340, null]], "mlx.nn.Conv1d": [[341, null]], "mlx.nn.Conv2d": [[342, null]], "mlx.nn.Conv3d": [[343, null]], "mlx.nn.ConvTranspose1d": [[344, null]], "mlx.nn.ConvTranspose2d": [[345, null]], "mlx.nn.ConvTranspose3d": [[346, null]], "mlx.nn.Dropout": [[347, null]], "mlx.nn.Dropout2d": [[348, null]], "mlx.nn.Dropout3d": [[349, null]], "mlx.nn.ELU": [[350, null]], "mlx.nn.Embedding": [[351, null]], "mlx.nn.GELU": [[352, null]], "mlx.nn.GLU": [[353, null]], "mlx.nn.GRU": [[354, null]], "mlx.nn.GroupNorm": [[355, null]], "mlx.nn.HardShrink": [[356, null]], "mlx.nn.HardTanh": [[357, null]], "mlx.nn.Hardswish": [[358, null]], "mlx.nn.InstanceNorm": [[359, null]], "mlx.nn.LSTM": [[360, null]], "mlx.nn.LayerNorm": [[361, null]], "mlx.nn.LeakyReLU": [[362, null]], "mlx.nn.Linear": [[363, null]], "mlx.nn.LogSigmoid": [[364, null]], "mlx.nn.LogSoftmax": [[365, null]], "mlx.nn.MaxPool1d": [[366, null]], "mlx.nn.MaxPool2d": [[367, null]], "mlx.nn.MaxPool3d": [[368, null]], "mlx.nn.Mish": [[369, null]], "mlx.nn.Module.apply": [[370, null]], "mlx.nn.Module.apply_to_modules": [[371, null]], "mlx.nn.Module.children": [[372, null]], "mlx.nn.Module.eval": [[373, null]], "mlx.nn.Module.filter_and_map": [[374, null]], "mlx.nn.Module.freeze": [[375, null]], "mlx.nn.Module.leaf_modules": [[376, null]], "mlx.nn.Module.load_weights": [[377, null]], "mlx.nn.Module.modules": [[378, null]], "mlx.nn.Module.named_modules": [[379, null]], "mlx.nn.Module.parameters": [[380, null]], "mlx.nn.Module.save_weights": [[381, null]], "mlx.nn.Module.set_dtype": [[382, null]], "mlx.nn.Module.state": [[383, null]], "mlx.nn.Module.train": [[384, null]], "mlx.nn.Module.trainable_parameters": [[385, null]], "mlx.nn.Module.training": [[386, null]], "mlx.nn.Module.unfreeze": [[387, null]], "mlx.nn.Module.update": [[388, null]], "mlx.nn.Module.update_modules": [[389, null]], "mlx.nn.MultiHeadAttention": [[390, null]], "mlx.nn.PReLU": [[391, null]], "mlx.nn.QuantizedEmbedding": [[392, null]], "mlx.nn.QuantizedLinear": [[393, null]], "mlx.nn.RMSNorm": [[394, null]], "mlx.nn.RNN": [[395, null]], "mlx.nn.ReLU": [[396, null]], "mlx.nn.ReLU6": [[397, null]], "mlx.nn.RoPE": [[398, null]], "mlx.nn.SELU": [[399, null]], "mlx.nn.Sequential": [[400, null]], "mlx.nn.SiLU": [[401, null]], "mlx.nn.Sigmoid": [[402, null]], "mlx.nn.SinusoidalPositionalEncoding": [[403, null]], "mlx.nn.Softmax": [[404, null]], "mlx.nn.Softmin": [[405, null]], "mlx.nn.Softplus": [[406, null]], "mlx.nn.Softshrink": [[407, null]], "mlx.nn.Softsign": [[408, null]], "mlx.nn.Step": [[409, null]], "mlx.nn.Tanh": [[410, null]], "mlx.nn.Transformer": [[411, null]], "mlx.nn.Upsample": [[412, null]], "mlx.nn.celu": [[421, null]], "mlx.nn.elu": [[422, null]], "mlx.nn.gelu": [[423, null]], "mlx.nn.gelu_approx": [[424, null]], "mlx.nn.gelu_fast_approx": [[425, null]], "mlx.nn.glu": [[426, null]], "mlx.nn.hard_shrink": [[427, null]], "mlx.nn.hard_tanh": [[428, null]], "mlx.nn.hardswish": [[429, null]], "mlx.nn.init.constant": [[413, null]], "mlx.nn.init.glorot_normal": [[414, null]], "mlx.nn.init.glorot_uniform": [[415, null]], "mlx.nn.init.he_normal": [[416, null]], "mlx.nn.init.he_uniform": [[417, null]], "mlx.nn.init.identity": [[418, null]], "mlx.nn.init.normal": [[419, null]], "mlx.nn.init.uniform": [[420, null]], "mlx.nn.leaky_relu": [[430, null]], "mlx.nn.log_sigmoid": [[431, null]], "mlx.nn.log_softmax": [[432, null]], "mlx.nn.losses.binary_cross_entropy": [[433, null]], "mlx.nn.losses.cosine_similarity_loss": [[434, null]], "mlx.nn.losses.cross_entropy": [[435, null]], "mlx.nn.losses.gaussian_nll_loss": [[436, null]], "mlx.nn.losses.hinge_loss": [[437, null]], "mlx.nn.losses.huber_loss": [[438, null]], "mlx.nn.losses.kl_div_loss": [[439, null]], "mlx.nn.losses.l1_loss": [[440, null]], "mlx.nn.losses.log_cosh_loss": [[441, null]], "mlx.nn.losses.margin_ranking_loss": [[442, null]], "mlx.nn.losses.mse_loss": [[443, null]], "mlx.nn.losses.nll_loss": [[444, null]], "mlx.nn.losses.smooth_l1_loss": [[445, null]], "mlx.nn.losses.triplet_loss": [[446, null]], "mlx.nn.mish": [[447, null]], "mlx.nn.prelu": [[448, null]], "mlx.nn.quantize": [[316, null]], "mlx.nn.relu": [[449, null]], "mlx.nn.relu6": [[450, null]], "mlx.nn.selu": [[451, null]], "mlx.nn.sigmoid": [[452, null]], "mlx.nn.silu": [[453, null]], "mlx.nn.softmax": [[454, null]], "mlx.nn.softmin": [[455, null]], "mlx.nn.softplus": [[456, null]], "mlx.nn.softshrink": [[457, null]], "mlx.nn.step": [[458, null]], "mlx.nn.tanh": [[459, null]], "mlx.nn.value_and_grad": [[317, null]], "mlx.optimizers.AdaDelta": [[467, null]], "mlx.optimizers.Adafactor": [[468, null]], "mlx.optimizers.Adagrad": [[469, null]], "mlx.optimizers.Adam": [[470, null]], "mlx.optimizers.AdamW": [[471, null]], "mlx.optimizers.Adamax": [[472, null]], "mlx.optimizers.Lion": [[473, null]], "mlx.optimizers.Optimizer.apply_gradients": [[474, null]], "mlx.optimizers.Optimizer.init": [[475, null]], "mlx.optimizers.Optimizer.state": [[476, null]], "mlx.optimizers.Optimizer.update": [[477, null]], "mlx.optimizers.RMSprop": [[478, null]], "mlx.optimizers.SGD": [[479, null]], "mlx.optimizers.clip_grad_norm": [[318, null]], "mlx.optimizers.cosine_decay": [[480, null]], "mlx.optimizers.exponential_decay": [[481, null]], "mlx.optimizers.join_schedules": [[482, null]], "mlx.optimizers.linear_schedule": [[483, null]], "mlx.optimizers.step_decay": [[484, null]], "mlx.utils.tree_flatten": [[319, null]], "mlx.utils.tree_map": [[320, null]], "mlx.utils.tree_map_with_path": [[321, null]], "mlx.utils.tree_reduce": [[322, null]], "mlx.utils.tree_unflatten": [[323, null]], "x86 Shell": [[9, "x86-shell"]]}, "docnames": ["cpp/ops", "dev/custom_metal_kernels", "dev/extensions", "dev/metal_debugger", "dev/mlx_in_cpp", "examples/linear_regression", "examples/llama-inference", "examples/mlp", "index", "install", "python/_autosummary/mlx.core.Device", "python/_autosummary/mlx.core.Dtype", "python/_autosummary/mlx.core.DtypeCategory", "python/_autosummary/mlx.core.abs", "python/_autosummary/mlx.core.add", "python/_autosummary/mlx.core.addmm", "python/_autosummary/mlx.core.all", "python/_autosummary/mlx.core.allclose", "python/_autosummary/mlx.core.any", "python/_autosummary/mlx.core.arange", "python/_autosummary/mlx.core.arccos", "python/_autosummary/mlx.core.arccosh", "python/_autosummary/mlx.core.arcsin", "python/_autosummary/mlx.core.arcsinh", "python/_autosummary/mlx.core.arctan", "python/_autosummary/mlx.core.arctan2", "python/_autosummary/mlx.core.arctanh", "python/_autosummary/mlx.core.argmax", "python/_autosummary/mlx.core.argmin", "python/_autosummary/mlx.core.argpartition", "python/_autosummary/mlx.core.argsort", "python/_autosummary/mlx.core.array", "python/_autosummary/mlx.core.array.T", "python/_autosummary/mlx.core.array.abs", "python/_autosummary/mlx.core.array.all", "python/_autosummary/mlx.core.array.any", "python/_autosummary/mlx.core.array.argmax", "python/_autosummary/mlx.core.array.argmin", "python/_autosummary/mlx.core.array.astype", "python/_autosummary/mlx.core.array.at", "python/_autosummary/mlx.core.array.conj", "python/_autosummary/mlx.core.array.cos", "python/_autosummary/mlx.core.array.cummax", "python/_autosummary/mlx.core.array.cummin", "python/_autosummary/mlx.core.array.cumprod", "python/_autosummary/mlx.core.array.cumsum", "python/_autosummary/mlx.core.array.diag", "python/_autosummary/mlx.core.array.diagonal", "python/_autosummary/mlx.core.array.dtype", "python/_autosummary/mlx.core.array.exp", "python/_autosummary/mlx.core.array.flatten", "python/_autosummary/mlx.core.array.item", "python/_autosummary/mlx.core.array.itemsize", "python/_autosummary/mlx.core.array.log", "python/_autosummary/mlx.core.array.log10", "python/_autosummary/mlx.core.array.log1p", "python/_autosummary/mlx.core.array.log2", "python/_autosummary/mlx.core.array.logsumexp", "python/_autosummary/mlx.core.array.max", "python/_autosummary/mlx.core.array.mean", "python/_autosummary/mlx.core.array.min", "python/_autosummary/mlx.core.array.moveaxis", "python/_autosummary/mlx.core.array.nbytes", "python/_autosummary/mlx.core.array.ndim", "python/_autosummary/mlx.core.array.prod", "python/_autosummary/mlx.core.array.reciprocal", "python/_autosummary/mlx.core.array.reshape", "python/_autosummary/mlx.core.array.round", "python/_autosummary/mlx.core.array.rsqrt", "python/_autosummary/mlx.core.array.shape", "python/_autosummary/mlx.core.array.sin", "python/_autosummary/mlx.core.array.size", "python/_autosummary/mlx.core.array.split", "python/_autosummary/mlx.core.array.sqrt", "python/_autosummary/mlx.core.array.square", "python/_autosummary/mlx.core.array.squeeze", "python/_autosummary/mlx.core.array.std", "python/_autosummary/mlx.core.array.sum", "python/_autosummary/mlx.core.array.swapaxes", "python/_autosummary/mlx.core.array.tolist", "python/_autosummary/mlx.core.array.transpose", "python/_autosummary/mlx.core.array.var", "python/_autosummary/mlx.core.array.view", "python/_autosummary/mlx.core.array_equal", "python/_autosummary/mlx.core.as_strided", "python/_autosummary/mlx.core.atleast_1d", "python/_autosummary/mlx.core.atleast_2d", "python/_autosummary/mlx.core.atleast_3d", "python/_autosummary/mlx.core.bitwise_and", "python/_autosummary/mlx.core.bitwise_or", "python/_autosummary/mlx.core.bitwise_xor", "python/_autosummary/mlx.core.block_masked_mm", "python/_autosummary/mlx.core.broadcast_to", "python/_autosummary/mlx.core.ceil", "python/_autosummary/mlx.core.clip", "python/_autosummary/mlx.core.compile", "python/_autosummary/mlx.core.concatenate", "python/_autosummary/mlx.core.conj", "python/_autosummary/mlx.core.conjugate", "python/_autosummary/mlx.core.conv1d", "python/_autosummary/mlx.core.conv2d", "python/_autosummary/mlx.core.conv3d", "python/_autosummary/mlx.core.conv_general", "python/_autosummary/mlx.core.conv_transpose1d", "python/_autosummary/mlx.core.conv_transpose2d", "python/_autosummary/mlx.core.conv_transpose3d", "python/_autosummary/mlx.core.convolve", "python/_autosummary/mlx.core.cos", "python/_autosummary/mlx.core.cosh", "python/_autosummary/mlx.core.cummax", "python/_autosummary/mlx.core.cummin", "python/_autosummary/mlx.core.cumprod", "python/_autosummary/mlx.core.cumsum", "python/_autosummary/mlx.core.custom_function", "python/_autosummary/mlx.core.default_device", "python/_autosummary/mlx.core.default_stream", "python/_autosummary/mlx.core.degrees", "python/_autosummary/mlx.core.dequantize", "python/_autosummary/mlx.core.diag", "python/_autosummary/mlx.core.diagonal", "python/_autosummary/mlx.core.disable_compile", "python/_autosummary/mlx.core.distributed.Group", "python/_autosummary/mlx.core.distributed.all_gather", "python/_autosummary/mlx.core.distributed.all_sum", "python/_autosummary/mlx.core.distributed.init", "python/_autosummary/mlx.core.distributed.is_available", "python/_autosummary/mlx.core.distributed.recv", "python/_autosummary/mlx.core.distributed.recv_like", "python/_autosummary/mlx.core.distributed.send", "python/_autosummary/mlx.core.divide", "python/_autosummary/mlx.core.divmod", "python/_autosummary/mlx.core.einsum", "python/_autosummary/mlx.core.einsum_path", "python/_autosummary/mlx.core.enable_compile", "python/_autosummary/mlx.core.equal", "python/_autosummary/mlx.core.erf", "python/_autosummary/mlx.core.erfinv", "python/_autosummary/mlx.core.eval", "python/_autosummary/mlx.core.exp", "python/_autosummary/mlx.core.expand_dims", "python/_autosummary/mlx.core.expm1", "python/_autosummary/mlx.core.export_function", "python/_autosummary/mlx.core.export_to_dot", "python/_autosummary/mlx.core.exporter", "python/_autosummary/mlx.core.eye", "python/_autosummary/mlx.core.fast.layer_norm", "python/_autosummary/mlx.core.fast.metal_kernel", "python/_autosummary/mlx.core.fast.rms_norm", "python/_autosummary/mlx.core.fast.rope", "python/_autosummary/mlx.core.fast.scaled_dot_product_attention", "python/_autosummary/mlx.core.fft.fft", "python/_autosummary/mlx.core.fft.fft2", "python/_autosummary/mlx.core.fft.fftn", "python/_autosummary/mlx.core.fft.ifft", "python/_autosummary/mlx.core.fft.ifft2", "python/_autosummary/mlx.core.fft.ifftn", "python/_autosummary/mlx.core.fft.irfft", "python/_autosummary/mlx.core.fft.irfft2", "python/_autosummary/mlx.core.fft.irfftn", "python/_autosummary/mlx.core.fft.rfft", "python/_autosummary/mlx.core.fft.rfft2", "python/_autosummary/mlx.core.fft.rfftn", "python/_autosummary/mlx.core.finfo", "python/_autosummary/mlx.core.flatten", "python/_autosummary/mlx.core.floor", "python/_autosummary/mlx.core.floor_divide", "python/_autosummary/mlx.core.full", "python/_autosummary/mlx.core.gather_mm", "python/_autosummary/mlx.core.gather_qmm", "python/_autosummary/mlx.core.grad", "python/_autosummary/mlx.core.greater", "python/_autosummary/mlx.core.greater_equal", "python/_autosummary/mlx.core.hadamard_transform", "python/_autosummary/mlx.core.identity", "python/_autosummary/mlx.core.imag", "python/_autosummary/mlx.core.import_function", "python/_autosummary/mlx.core.inner", "python/_autosummary/mlx.core.isclose", "python/_autosummary/mlx.core.isfinite", "python/_autosummary/mlx.core.isinf", "python/_autosummary/mlx.core.isnan", "python/_autosummary/mlx.core.isneginf", "python/_autosummary/mlx.core.isposinf", "python/_autosummary/mlx.core.issubdtype", "python/_autosummary/mlx.core.jvp", "python/_autosummary/mlx.core.kron", "python/_autosummary/mlx.core.left_shift", "python/_autosummary/mlx.core.less", "python/_autosummary/mlx.core.less_equal", "python/_autosummary/mlx.core.linalg.cholesky", "python/_autosummary/mlx.core.linalg.cholesky_inv", "python/_autosummary/mlx.core.linalg.cross", "python/_autosummary/mlx.core.linalg.eigh", "python/_autosummary/mlx.core.linalg.eigvalsh", "python/_autosummary/mlx.core.linalg.inv", "python/_autosummary/mlx.core.linalg.norm", "python/_autosummary/mlx.core.linalg.qr", "python/_autosummary/mlx.core.linalg.svd", "python/_autosummary/mlx.core.linalg.tri_inv", "python/_autosummary/mlx.core.linspace", "python/_autosummary/mlx.core.load", "python/_autosummary/mlx.core.log", "python/_autosummary/mlx.core.log10", "python/_autosummary/mlx.core.log1p", "python/_autosummary/mlx.core.log2", "python/_autosummary/mlx.core.logaddexp", "python/_autosummary/mlx.core.logical_and", "python/_autosummary/mlx.core.logical_not", "python/_autosummary/mlx.core.logical_or", "python/_autosummary/mlx.core.logsumexp", "python/_autosummary/mlx.core.matmul", "python/_autosummary/mlx.core.max", "python/_autosummary/mlx.core.maximum", "python/_autosummary/mlx.core.mean", "python/_autosummary/mlx.core.meshgrid", "python/_autosummary/mlx.core.metal.clear_cache", "python/_autosummary/mlx.core.metal.device_info", "python/_autosummary/mlx.core.metal.get_active_memory", "python/_autosummary/mlx.core.metal.get_cache_memory", "python/_autosummary/mlx.core.metal.get_peak_memory", "python/_autosummary/mlx.core.metal.is_available", "python/_autosummary/mlx.core.metal.reset_peak_memory", "python/_autosummary/mlx.core.metal.set_cache_limit", "python/_autosummary/mlx.core.metal.set_memory_limit", "python/_autosummary/mlx.core.metal.set_wired_limit", "python/_autosummary/mlx.core.metal.start_capture", "python/_autosummary/mlx.core.metal.stop_capture", "python/_autosummary/mlx.core.min", "python/_autosummary/mlx.core.minimum", "python/_autosummary/mlx.core.moveaxis", "python/_autosummary/mlx.core.multiply", "python/_autosummary/mlx.core.nan_to_num", "python/_autosummary/mlx.core.negative", "python/_autosummary/mlx.core.new_stream", "python/_autosummary/mlx.core.not_equal", "python/_autosummary/mlx.core.ones", "python/_autosummary/mlx.core.ones_like", "python/_autosummary/mlx.core.outer", "python/_autosummary/mlx.core.pad", "python/_autosummary/mlx.core.partition", "python/_autosummary/mlx.core.power", "python/_autosummary/mlx.core.prod", "python/_autosummary/mlx.core.put_along_axis", "python/_autosummary/mlx.core.quantize", "python/_autosummary/mlx.core.quantized_matmul", "python/_autosummary/mlx.core.radians", "python/_autosummary/mlx.core.random.bernoulli", "python/_autosummary/mlx.core.random.categorical", "python/_autosummary/mlx.core.random.gumbel", "python/_autosummary/mlx.core.random.key", "python/_autosummary/mlx.core.random.laplace", "python/_autosummary/mlx.core.random.multivariate_normal", "python/_autosummary/mlx.core.random.normal", "python/_autosummary/mlx.core.random.permutation", "python/_autosummary/mlx.core.random.randint", "python/_autosummary/mlx.core.random.seed", "python/_autosummary/mlx.core.random.split", "python/_autosummary/mlx.core.random.truncated_normal", "python/_autosummary/mlx.core.random.uniform", "python/_autosummary/mlx.core.real", "python/_autosummary/mlx.core.reciprocal", "python/_autosummary/mlx.core.remainder", "python/_autosummary/mlx.core.repeat", "python/_autosummary/mlx.core.reshape", "python/_autosummary/mlx.core.right_shift", "python/_autosummary/mlx.core.roll", "python/_autosummary/mlx.core.round", "python/_autosummary/mlx.core.rsqrt", "python/_autosummary/mlx.core.save", "python/_autosummary/mlx.core.save_gguf", "python/_autosummary/mlx.core.save_safetensors", "python/_autosummary/mlx.core.savez", "python/_autosummary/mlx.core.savez_compressed", "python/_autosummary/mlx.core.set_default_device", "python/_autosummary/mlx.core.set_default_stream", "python/_autosummary/mlx.core.sigmoid", "python/_autosummary/mlx.core.sign", "python/_autosummary/mlx.core.sin", "python/_autosummary/mlx.core.sinh", "python/_autosummary/mlx.core.slice", "python/_autosummary/mlx.core.slice_update", "python/_autosummary/mlx.core.softmax", "python/_autosummary/mlx.core.sort", "python/_autosummary/mlx.core.split", "python/_autosummary/mlx.core.sqrt", "python/_autosummary/mlx.core.square", "python/_autosummary/mlx.core.squeeze", "python/_autosummary/mlx.core.stack", "python/_autosummary/mlx.core.std", "python/_autosummary/mlx.core.stop_gradient", "python/_autosummary/mlx.core.stream", "python/_autosummary/mlx.core.subtract", "python/_autosummary/mlx.core.sum", "python/_autosummary/mlx.core.swapaxes", "python/_autosummary/mlx.core.synchronize", "python/_autosummary/mlx.core.take", "python/_autosummary/mlx.core.take_along_axis", "python/_autosummary/mlx.core.tan", "python/_autosummary/mlx.core.tanh", "python/_autosummary/mlx.core.tensordot", "python/_autosummary/mlx.core.tile", "python/_autosummary/mlx.core.topk", "python/_autosummary/mlx.core.trace", "python/_autosummary/mlx.core.transpose", "python/_autosummary/mlx.core.tri", "python/_autosummary/mlx.core.tril", "python/_autosummary/mlx.core.triu", "python/_autosummary/mlx.core.unflatten", "python/_autosummary/mlx.core.value_and_grad", "python/_autosummary/mlx.core.var", "python/_autosummary/mlx.core.view", "python/_autosummary/mlx.core.vjp", "python/_autosummary/mlx.core.vmap", "python/_autosummary/mlx.core.where", "python/_autosummary/mlx.core.zeros", "python/_autosummary/mlx.core.zeros_like", "python/_autosummary/mlx.nn.quantize", "python/_autosummary/mlx.nn.value_and_grad", "python/_autosummary/mlx.optimizers.clip_grad_norm", "python/_autosummary/mlx.utils.tree_flatten", "python/_autosummary/mlx.utils.tree_map", "python/_autosummary/mlx.utils.tree_map_with_path", "python/_autosummary/mlx.utils.tree_reduce", "python/_autosummary/mlx.utils.tree_unflatten", "python/_autosummary/stream_class", "python/array", "python/data_types", "python/devices_and_streams", "python/distributed", "python/export", "python/fast", "python/fft", "python/linalg", "python/metal", "python/nn", "python/nn/_autosummary/mlx.nn.ALiBi", "python/nn/_autosummary/mlx.nn.AvgPool1d", "python/nn/_autosummary/mlx.nn.AvgPool2d", "python/nn/_autosummary/mlx.nn.AvgPool3d", "python/nn/_autosummary/mlx.nn.BatchNorm", "python/nn/_autosummary/mlx.nn.CELU", "python/nn/_autosummary/mlx.nn.Conv1d", "python/nn/_autosummary/mlx.nn.Conv2d", "python/nn/_autosummary/mlx.nn.Conv3d", "python/nn/_autosummary/mlx.nn.ConvTranspose1d", "python/nn/_autosummary/mlx.nn.ConvTranspose2d", "python/nn/_autosummary/mlx.nn.ConvTranspose3d", "python/nn/_autosummary/mlx.nn.Dropout", "python/nn/_autosummary/mlx.nn.Dropout2d", "python/nn/_autosummary/mlx.nn.Dropout3d", "python/nn/_autosummary/mlx.nn.ELU", "python/nn/_autosummary/mlx.nn.Embedding", "python/nn/_autosummary/mlx.nn.GELU", "python/nn/_autosummary/mlx.nn.GLU", "python/nn/_autosummary/mlx.nn.GRU", "python/nn/_autosummary/mlx.nn.GroupNorm", "python/nn/_autosummary/mlx.nn.HardShrink", "python/nn/_autosummary/mlx.nn.HardTanh", "python/nn/_autosummary/mlx.nn.Hardswish", "python/nn/_autosummary/mlx.nn.InstanceNorm", "python/nn/_autosummary/mlx.nn.LSTM", "python/nn/_autosummary/mlx.nn.LayerNorm", "python/nn/_autosummary/mlx.nn.LeakyReLU", "python/nn/_autosummary/mlx.nn.Linear", "python/nn/_autosummary/mlx.nn.LogSigmoid", "python/nn/_autosummary/mlx.nn.LogSoftmax", "python/nn/_autosummary/mlx.nn.MaxPool1d", "python/nn/_autosummary/mlx.nn.MaxPool2d", "python/nn/_autosummary/mlx.nn.MaxPool3d", "python/nn/_autosummary/mlx.nn.Mish", "python/nn/_autosummary/mlx.nn.Module.apply", "python/nn/_autosummary/mlx.nn.Module.apply_to_modules", "python/nn/_autosummary/mlx.nn.Module.children", "python/nn/_autosummary/mlx.nn.Module.eval", "python/nn/_autosummary/mlx.nn.Module.filter_and_map", "python/nn/_autosummary/mlx.nn.Module.freeze", "python/nn/_autosummary/mlx.nn.Module.leaf_modules", "python/nn/_autosummary/mlx.nn.Module.load_weights", "python/nn/_autosummary/mlx.nn.Module.modules", "python/nn/_autosummary/mlx.nn.Module.named_modules", "python/nn/_autosummary/mlx.nn.Module.parameters", "python/nn/_autosummary/mlx.nn.Module.save_weights", "python/nn/_autosummary/mlx.nn.Module.set_dtype", "python/nn/_autosummary/mlx.nn.Module.state", "python/nn/_autosummary/mlx.nn.Module.train", "python/nn/_autosummary/mlx.nn.Module.trainable_parameters", "python/nn/_autosummary/mlx.nn.Module.training", "python/nn/_autosummary/mlx.nn.Module.unfreeze", "python/nn/_autosummary/mlx.nn.Module.update", "python/nn/_autosummary/mlx.nn.Module.update_modules", "python/nn/_autosummary/mlx.nn.MultiHeadAttention", "python/nn/_autosummary/mlx.nn.PReLU", "python/nn/_autosummary/mlx.nn.QuantizedEmbedding", "python/nn/_autosummary/mlx.nn.QuantizedLinear", "python/nn/_autosummary/mlx.nn.RMSNorm", "python/nn/_autosummary/mlx.nn.RNN", "python/nn/_autosummary/mlx.nn.ReLU", "python/nn/_autosummary/mlx.nn.ReLU6", "python/nn/_autosummary/mlx.nn.RoPE", "python/nn/_autosummary/mlx.nn.SELU", "python/nn/_autosummary/mlx.nn.Sequential", "python/nn/_autosummary/mlx.nn.SiLU", "python/nn/_autosummary/mlx.nn.Sigmoid", "python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding", "python/nn/_autosummary/mlx.nn.Softmax", "python/nn/_autosummary/mlx.nn.Softmin", "python/nn/_autosummary/mlx.nn.Softplus", "python/nn/_autosummary/mlx.nn.Softshrink", "python/nn/_autosummary/mlx.nn.Softsign", "python/nn/_autosummary/mlx.nn.Step", "python/nn/_autosummary/mlx.nn.Tanh", "python/nn/_autosummary/mlx.nn.Transformer", "python/nn/_autosummary/mlx.nn.Upsample", "python/nn/_autosummary/mlx.nn.init.constant", "python/nn/_autosummary/mlx.nn.init.glorot_normal", "python/nn/_autosummary/mlx.nn.init.glorot_uniform", "python/nn/_autosummary/mlx.nn.init.he_normal", "python/nn/_autosummary/mlx.nn.init.he_uniform", "python/nn/_autosummary/mlx.nn.init.identity", "python/nn/_autosummary/mlx.nn.init.normal", "python/nn/_autosummary/mlx.nn.init.uniform", "python/nn/_autosummary_functions/mlx.nn.celu", "python/nn/_autosummary_functions/mlx.nn.elu", "python/nn/_autosummary_functions/mlx.nn.gelu", "python/nn/_autosummary_functions/mlx.nn.gelu_approx", "python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx", "python/nn/_autosummary_functions/mlx.nn.glu", "python/nn/_autosummary_functions/mlx.nn.hard_shrink", "python/nn/_autosummary_functions/mlx.nn.hard_tanh", "python/nn/_autosummary_functions/mlx.nn.hardswish", "python/nn/_autosummary_functions/mlx.nn.leaky_relu", "python/nn/_autosummary_functions/mlx.nn.log_sigmoid", "python/nn/_autosummary_functions/mlx.nn.log_softmax", "python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy", "python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss", "python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy", "python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss", "python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss", "python/nn/_autosummary_functions/mlx.nn.losses.huber_loss", "python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss", "python/nn/_autosummary_functions/mlx.nn.losses.l1_loss", "python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss", "python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss", "python/nn/_autosummary_functions/mlx.nn.losses.mse_loss", "python/nn/_autosummary_functions/mlx.nn.losses.nll_loss", "python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss", "python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss", "python/nn/_autosummary_functions/mlx.nn.mish", "python/nn/_autosummary_functions/mlx.nn.prelu", "python/nn/_autosummary_functions/mlx.nn.relu", "python/nn/_autosummary_functions/mlx.nn.relu6", "python/nn/_autosummary_functions/mlx.nn.selu", "python/nn/_autosummary_functions/mlx.nn.sigmoid", "python/nn/_autosummary_functions/mlx.nn.silu", "python/nn/_autosummary_functions/mlx.nn.softmax", "python/nn/_autosummary_functions/mlx.nn.softmin", "python/nn/_autosummary_functions/mlx.nn.softplus", "python/nn/_autosummary_functions/mlx.nn.softshrink", "python/nn/_autosummary_functions/mlx.nn.step", "python/nn/_autosummary_functions/mlx.nn.tanh", "python/nn/functions", "python/nn/init", "python/nn/layers", "python/nn/losses", "python/nn/module", "python/ops", "python/optimizers", "python/optimizers/_autosummary/mlx.optimizers.AdaDelta", "python/optimizers/_autosummary/mlx.optimizers.Adafactor", "python/optimizers/_autosummary/mlx.optimizers.Adagrad", "python/optimizers/_autosummary/mlx.optimizers.Adam", "python/optimizers/_autosummary/mlx.optimizers.AdamW", "python/optimizers/_autosummary/mlx.optimizers.Adamax", "python/optimizers/_autosummary/mlx.optimizers.Lion", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.init", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.state", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.update", "python/optimizers/_autosummary/mlx.optimizers.RMSprop", "python/optimizers/_autosummary/mlx.optimizers.SGD", "python/optimizers/_autosummary/mlx.optimizers.cosine_decay", "python/optimizers/_autosummary/mlx.optimizers.exponential_decay", "python/optimizers/_autosummary/mlx.optimizers.join_schedules", "python/optimizers/_autosummary/mlx.optimizers.linear_schedule", "python/optimizers/_autosummary/mlx.optimizers.step_decay", "python/optimizers/common_optimizers", "python/optimizers/optimizer", "python/optimizers/schedulers", "python/random", "python/transforms", "python/tree_utils", "usage/compile", "usage/distributed", "usage/export", "usage/function_transforms", "usage/indexing", "usage/lazy_evaluation", "usage/numpy", "usage/quick_start", "usage/saving_and_loading", "usage/unified_memory", "usage/using_streams"], "envversion": {"sphinx": 62, "sphinx.domains.c": 3, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 9, "sphinx.domains.index": 1, "sphinx.domains.javascript": 3, "sphinx.domains.math": 2, "sphinx.domains.python": 4, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.intersphinx": 1}, "filenames": ["cpp/ops.rst", "dev/custom_metal_kernels.rst", "dev/extensions.rst", "dev/metal_debugger.rst", "dev/mlx_in_cpp.rst", "examples/linear_regression.rst", "examples/llama-inference.rst", "examples/mlp.rst", "index.rst", "install.rst", "python/_autosummary/mlx.core.Device.rst", "python/_autosummary/mlx.core.Dtype.rst", "python/_autosummary/mlx.core.DtypeCategory.rst", "python/_autosummary/mlx.core.abs.rst", "python/_autosummary/mlx.core.add.rst", "python/_autosummary/mlx.core.addmm.rst", "python/_autosummary/mlx.core.all.rst", "python/_autosummary/mlx.core.allclose.rst", "python/_autosummary/mlx.core.any.rst", "python/_autosummary/mlx.core.arange.rst", "python/_autosummary/mlx.core.arccos.rst", "python/_autosummary/mlx.core.arccosh.rst", "python/_autosummary/mlx.core.arcsin.rst", "python/_autosummary/mlx.core.arcsinh.rst", "python/_autosummary/mlx.core.arctan.rst", "python/_autosummary/mlx.core.arctan2.rst", "python/_autosummary/mlx.core.arctanh.rst", "python/_autosummary/mlx.core.argmax.rst", "python/_autosummary/mlx.core.argmin.rst", "python/_autosummary/mlx.core.argpartition.rst", "python/_autosummary/mlx.core.argsort.rst", "python/_autosummary/mlx.core.array.rst", "python/_autosummary/mlx.core.array.T.rst", "python/_autosummary/mlx.core.array.abs.rst", "python/_autosummary/mlx.core.array.all.rst", "python/_autosummary/mlx.core.array.any.rst", "python/_autosummary/mlx.core.array.argmax.rst", "python/_autosummary/mlx.core.array.argmin.rst", "python/_autosummary/mlx.core.array.astype.rst", "python/_autosummary/mlx.core.array.at.rst", "python/_autosummary/mlx.core.array.conj.rst", "python/_autosummary/mlx.core.array.cos.rst", "python/_autosummary/mlx.core.array.cummax.rst", "python/_autosummary/mlx.core.array.cummin.rst", "python/_autosummary/mlx.core.array.cumprod.rst", "python/_autosummary/mlx.core.array.cumsum.rst", "python/_autosummary/mlx.core.array.diag.rst", "python/_autosummary/mlx.core.array.diagonal.rst", "python/_autosummary/mlx.core.array.dtype.rst", "python/_autosummary/mlx.core.array.exp.rst", "python/_autosummary/mlx.core.array.flatten.rst", "python/_autosummary/mlx.core.array.item.rst", "python/_autosummary/mlx.core.array.itemsize.rst", "python/_autosummary/mlx.core.array.log.rst", "python/_autosummary/mlx.core.array.log10.rst", "python/_autosummary/mlx.core.array.log1p.rst", "python/_autosummary/mlx.core.array.log2.rst", "python/_autosummary/mlx.core.array.logsumexp.rst", "python/_autosummary/mlx.core.array.max.rst", "python/_autosummary/mlx.core.array.mean.rst", "python/_autosummary/mlx.core.array.min.rst", "python/_autosummary/mlx.core.array.moveaxis.rst", "python/_autosummary/mlx.core.array.nbytes.rst", "python/_autosummary/mlx.core.array.ndim.rst", "python/_autosummary/mlx.core.array.prod.rst", "python/_autosummary/mlx.core.array.reciprocal.rst", "python/_autosummary/mlx.core.array.reshape.rst", "python/_autosummary/mlx.core.array.round.rst", "python/_autosummary/mlx.core.array.rsqrt.rst", "python/_autosummary/mlx.core.array.shape.rst", "python/_autosummary/mlx.core.array.sin.rst", "python/_autosummary/mlx.core.array.size.rst", "python/_autosummary/mlx.core.array.split.rst", "python/_autosummary/mlx.core.array.sqrt.rst", "python/_autosummary/mlx.core.array.square.rst", "python/_autosummary/mlx.core.array.squeeze.rst", "python/_autosummary/mlx.core.array.std.rst", "python/_autosummary/mlx.core.array.sum.rst", "python/_autosummary/mlx.core.array.swapaxes.rst", "python/_autosummary/mlx.core.array.tolist.rst", "python/_autosummary/mlx.core.array.transpose.rst", "python/_autosummary/mlx.core.array.var.rst", "python/_autosummary/mlx.core.array.view.rst", "python/_autosummary/mlx.core.array_equal.rst", "python/_autosummary/mlx.core.as_strided.rst", "python/_autosummary/mlx.core.atleast_1d.rst", "python/_autosummary/mlx.core.atleast_2d.rst", "python/_autosummary/mlx.core.atleast_3d.rst", "python/_autosummary/mlx.core.bitwise_and.rst", "python/_autosummary/mlx.core.bitwise_or.rst", "python/_autosummary/mlx.core.bitwise_xor.rst", "python/_autosummary/mlx.core.block_masked_mm.rst", "python/_autosummary/mlx.core.broadcast_to.rst", "python/_autosummary/mlx.core.ceil.rst", "python/_autosummary/mlx.core.clip.rst", "python/_autosummary/mlx.core.compile.rst", "python/_autosummary/mlx.core.concatenate.rst", "python/_autosummary/mlx.core.conj.rst", "python/_autosummary/mlx.core.conjugate.rst", "python/_autosummary/mlx.core.conv1d.rst", "python/_autosummary/mlx.core.conv2d.rst", "python/_autosummary/mlx.core.conv3d.rst", "python/_autosummary/mlx.core.conv_general.rst", "python/_autosummary/mlx.core.conv_transpose1d.rst", "python/_autosummary/mlx.core.conv_transpose2d.rst", "python/_autosummary/mlx.core.conv_transpose3d.rst", "python/_autosummary/mlx.core.convolve.rst", "python/_autosummary/mlx.core.cos.rst", "python/_autosummary/mlx.core.cosh.rst", "python/_autosummary/mlx.core.cummax.rst", "python/_autosummary/mlx.core.cummin.rst", "python/_autosummary/mlx.core.cumprod.rst", "python/_autosummary/mlx.core.cumsum.rst", "python/_autosummary/mlx.core.custom_function.rst", "python/_autosummary/mlx.core.default_device.rst", "python/_autosummary/mlx.core.default_stream.rst", "python/_autosummary/mlx.core.degrees.rst", "python/_autosummary/mlx.core.dequantize.rst", "python/_autosummary/mlx.core.diag.rst", "python/_autosummary/mlx.core.diagonal.rst", "python/_autosummary/mlx.core.disable_compile.rst", "python/_autosummary/mlx.core.distributed.Group.rst", "python/_autosummary/mlx.core.distributed.all_gather.rst", "python/_autosummary/mlx.core.distributed.all_sum.rst", "python/_autosummary/mlx.core.distributed.init.rst", "python/_autosummary/mlx.core.distributed.is_available.rst", "python/_autosummary/mlx.core.distributed.recv.rst", "python/_autosummary/mlx.core.distributed.recv_like.rst", "python/_autosummary/mlx.core.distributed.send.rst", "python/_autosummary/mlx.core.divide.rst", "python/_autosummary/mlx.core.divmod.rst", "python/_autosummary/mlx.core.einsum.rst", "python/_autosummary/mlx.core.einsum_path.rst", "python/_autosummary/mlx.core.enable_compile.rst", "python/_autosummary/mlx.core.equal.rst", "python/_autosummary/mlx.core.erf.rst", "python/_autosummary/mlx.core.erfinv.rst", "python/_autosummary/mlx.core.eval.rst", "python/_autosummary/mlx.core.exp.rst", "python/_autosummary/mlx.core.expand_dims.rst", "python/_autosummary/mlx.core.expm1.rst", "python/_autosummary/mlx.core.export_function.rst", "python/_autosummary/mlx.core.export_to_dot.rst", "python/_autosummary/mlx.core.exporter.rst", "python/_autosummary/mlx.core.eye.rst", "python/_autosummary/mlx.core.fast.layer_norm.rst", "python/_autosummary/mlx.core.fast.metal_kernel.rst", "python/_autosummary/mlx.core.fast.rms_norm.rst", "python/_autosummary/mlx.core.fast.rope.rst", "python/_autosummary/mlx.core.fast.scaled_dot_product_attention.rst", "python/_autosummary/mlx.core.fft.fft.rst", "python/_autosummary/mlx.core.fft.fft2.rst", "python/_autosummary/mlx.core.fft.fftn.rst", "python/_autosummary/mlx.core.fft.ifft.rst", "python/_autosummary/mlx.core.fft.ifft2.rst", "python/_autosummary/mlx.core.fft.ifftn.rst", "python/_autosummary/mlx.core.fft.irfft.rst", "python/_autosummary/mlx.core.fft.irfft2.rst", "python/_autosummary/mlx.core.fft.irfftn.rst", "python/_autosummary/mlx.core.fft.rfft.rst", "python/_autosummary/mlx.core.fft.rfft2.rst", "python/_autosummary/mlx.core.fft.rfftn.rst", "python/_autosummary/mlx.core.finfo.rst", "python/_autosummary/mlx.core.flatten.rst", "python/_autosummary/mlx.core.floor.rst", "python/_autosummary/mlx.core.floor_divide.rst", "python/_autosummary/mlx.core.full.rst", "python/_autosummary/mlx.core.gather_mm.rst", "python/_autosummary/mlx.core.gather_qmm.rst", "python/_autosummary/mlx.core.grad.rst", "python/_autosummary/mlx.core.greater.rst", "python/_autosummary/mlx.core.greater_equal.rst", "python/_autosummary/mlx.core.hadamard_transform.rst", "python/_autosummary/mlx.core.identity.rst", "python/_autosummary/mlx.core.imag.rst", "python/_autosummary/mlx.core.import_function.rst", "python/_autosummary/mlx.core.inner.rst", "python/_autosummary/mlx.core.isclose.rst", "python/_autosummary/mlx.core.isfinite.rst", "python/_autosummary/mlx.core.isinf.rst", "python/_autosummary/mlx.core.isnan.rst", "python/_autosummary/mlx.core.isneginf.rst", "python/_autosummary/mlx.core.isposinf.rst", "python/_autosummary/mlx.core.issubdtype.rst", "python/_autosummary/mlx.core.jvp.rst", "python/_autosummary/mlx.core.kron.rst", "python/_autosummary/mlx.core.left_shift.rst", "python/_autosummary/mlx.core.less.rst", "python/_autosummary/mlx.core.less_equal.rst", "python/_autosummary/mlx.core.linalg.cholesky.rst", "python/_autosummary/mlx.core.linalg.cholesky_inv.rst", "python/_autosummary/mlx.core.linalg.cross.rst", "python/_autosummary/mlx.core.linalg.eigh.rst", "python/_autosummary/mlx.core.linalg.eigvalsh.rst", "python/_autosummary/mlx.core.linalg.inv.rst", "python/_autosummary/mlx.core.linalg.norm.rst", "python/_autosummary/mlx.core.linalg.qr.rst", "python/_autosummary/mlx.core.linalg.svd.rst", "python/_autosummary/mlx.core.linalg.tri_inv.rst", "python/_autosummary/mlx.core.linspace.rst", "python/_autosummary/mlx.core.load.rst", "python/_autosummary/mlx.core.log.rst", "python/_autosummary/mlx.core.log10.rst", "python/_autosummary/mlx.core.log1p.rst", "python/_autosummary/mlx.core.log2.rst", "python/_autosummary/mlx.core.logaddexp.rst", "python/_autosummary/mlx.core.logical_and.rst", "python/_autosummary/mlx.core.logical_not.rst", "python/_autosummary/mlx.core.logical_or.rst", "python/_autosummary/mlx.core.logsumexp.rst", "python/_autosummary/mlx.core.matmul.rst", "python/_autosummary/mlx.core.max.rst", "python/_autosummary/mlx.core.maximum.rst", "python/_autosummary/mlx.core.mean.rst", "python/_autosummary/mlx.core.meshgrid.rst", "python/_autosummary/mlx.core.metal.clear_cache.rst", "python/_autosummary/mlx.core.metal.device_info.rst", "python/_autosummary/mlx.core.metal.get_active_memory.rst", "python/_autosummary/mlx.core.metal.get_cache_memory.rst", "python/_autosummary/mlx.core.metal.get_peak_memory.rst", "python/_autosummary/mlx.core.metal.is_available.rst", "python/_autosummary/mlx.core.metal.reset_peak_memory.rst", "python/_autosummary/mlx.core.metal.set_cache_limit.rst", "python/_autosummary/mlx.core.metal.set_memory_limit.rst", "python/_autosummary/mlx.core.metal.set_wired_limit.rst", "python/_autosummary/mlx.core.metal.start_capture.rst", "python/_autosummary/mlx.core.metal.stop_capture.rst", "python/_autosummary/mlx.core.min.rst", "python/_autosummary/mlx.core.minimum.rst", "python/_autosummary/mlx.core.moveaxis.rst", "python/_autosummary/mlx.core.multiply.rst", "python/_autosummary/mlx.core.nan_to_num.rst", "python/_autosummary/mlx.core.negative.rst", "python/_autosummary/mlx.core.new_stream.rst", "python/_autosummary/mlx.core.not_equal.rst", "python/_autosummary/mlx.core.ones.rst", "python/_autosummary/mlx.core.ones_like.rst", "python/_autosummary/mlx.core.outer.rst", "python/_autosummary/mlx.core.pad.rst", "python/_autosummary/mlx.core.partition.rst", "python/_autosummary/mlx.core.power.rst", "python/_autosummary/mlx.core.prod.rst", "python/_autosummary/mlx.core.put_along_axis.rst", "python/_autosummary/mlx.core.quantize.rst", "python/_autosummary/mlx.core.quantized_matmul.rst", "python/_autosummary/mlx.core.radians.rst", "python/_autosummary/mlx.core.random.bernoulli.rst", "python/_autosummary/mlx.core.random.categorical.rst", "python/_autosummary/mlx.core.random.gumbel.rst", "python/_autosummary/mlx.core.random.key.rst", "python/_autosummary/mlx.core.random.laplace.rst", "python/_autosummary/mlx.core.random.multivariate_normal.rst", "python/_autosummary/mlx.core.random.normal.rst", "python/_autosummary/mlx.core.random.permutation.rst", "python/_autosummary/mlx.core.random.randint.rst", "python/_autosummary/mlx.core.random.seed.rst", "python/_autosummary/mlx.core.random.split.rst", "python/_autosummary/mlx.core.random.truncated_normal.rst", "python/_autosummary/mlx.core.random.uniform.rst", "python/_autosummary/mlx.core.real.rst", "python/_autosummary/mlx.core.reciprocal.rst", "python/_autosummary/mlx.core.remainder.rst", "python/_autosummary/mlx.core.repeat.rst", "python/_autosummary/mlx.core.reshape.rst", "python/_autosummary/mlx.core.right_shift.rst", "python/_autosummary/mlx.core.roll.rst", "python/_autosummary/mlx.core.round.rst", "python/_autosummary/mlx.core.rsqrt.rst", "python/_autosummary/mlx.core.save.rst", "python/_autosummary/mlx.core.save_gguf.rst", "python/_autosummary/mlx.core.save_safetensors.rst", "python/_autosummary/mlx.core.savez.rst", "python/_autosummary/mlx.core.savez_compressed.rst", "python/_autosummary/mlx.core.set_default_device.rst", "python/_autosummary/mlx.core.set_default_stream.rst", "python/_autosummary/mlx.core.sigmoid.rst", "python/_autosummary/mlx.core.sign.rst", "python/_autosummary/mlx.core.sin.rst", "python/_autosummary/mlx.core.sinh.rst", "python/_autosummary/mlx.core.slice.rst", "python/_autosummary/mlx.core.slice_update.rst", "python/_autosummary/mlx.core.softmax.rst", "python/_autosummary/mlx.core.sort.rst", "python/_autosummary/mlx.core.split.rst", "python/_autosummary/mlx.core.sqrt.rst", "python/_autosummary/mlx.core.square.rst", "python/_autosummary/mlx.core.squeeze.rst", "python/_autosummary/mlx.core.stack.rst", "python/_autosummary/mlx.core.std.rst", "python/_autosummary/mlx.core.stop_gradient.rst", "python/_autosummary/mlx.core.stream.rst", "python/_autosummary/mlx.core.subtract.rst", "python/_autosummary/mlx.core.sum.rst", "python/_autosummary/mlx.core.swapaxes.rst", "python/_autosummary/mlx.core.synchronize.rst", "python/_autosummary/mlx.core.take.rst", "python/_autosummary/mlx.core.take_along_axis.rst", "python/_autosummary/mlx.core.tan.rst", "python/_autosummary/mlx.core.tanh.rst", "python/_autosummary/mlx.core.tensordot.rst", "python/_autosummary/mlx.core.tile.rst", "python/_autosummary/mlx.core.topk.rst", "python/_autosummary/mlx.core.trace.rst", "python/_autosummary/mlx.core.transpose.rst", "python/_autosummary/mlx.core.tri.rst", "python/_autosummary/mlx.core.tril.rst", "python/_autosummary/mlx.core.triu.rst", "python/_autosummary/mlx.core.unflatten.rst", "python/_autosummary/mlx.core.value_and_grad.rst", "python/_autosummary/mlx.core.var.rst", "python/_autosummary/mlx.core.view.rst", "python/_autosummary/mlx.core.vjp.rst", "python/_autosummary/mlx.core.vmap.rst", "python/_autosummary/mlx.core.where.rst", "python/_autosummary/mlx.core.zeros.rst", "python/_autosummary/mlx.core.zeros_like.rst", "python/_autosummary/mlx.nn.quantize.rst", "python/_autosummary/mlx.nn.value_and_grad.rst", "python/_autosummary/mlx.optimizers.clip_grad_norm.rst", "python/_autosummary/mlx.utils.tree_flatten.rst", "python/_autosummary/mlx.utils.tree_map.rst", "python/_autosummary/mlx.utils.tree_map_with_path.rst", "python/_autosummary/mlx.utils.tree_reduce.rst", "python/_autosummary/mlx.utils.tree_unflatten.rst", "python/_autosummary/stream_class.rst", "python/array.rst", "python/data_types.rst", "python/devices_and_streams.rst", "python/distributed.rst", "python/export.rst", "python/fast.rst", "python/fft.rst", "python/linalg.rst", "python/metal.rst", "python/nn.rst", "python/nn/_autosummary/mlx.nn.ALiBi.rst", "python/nn/_autosummary/mlx.nn.AvgPool1d.rst", "python/nn/_autosummary/mlx.nn.AvgPool2d.rst", "python/nn/_autosummary/mlx.nn.AvgPool3d.rst", "python/nn/_autosummary/mlx.nn.BatchNorm.rst", "python/nn/_autosummary/mlx.nn.CELU.rst", "python/nn/_autosummary/mlx.nn.Conv1d.rst", "python/nn/_autosummary/mlx.nn.Conv2d.rst", "python/nn/_autosummary/mlx.nn.Conv3d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose1d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose2d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose3d.rst", "python/nn/_autosummary/mlx.nn.Dropout.rst", "python/nn/_autosummary/mlx.nn.Dropout2d.rst", "python/nn/_autosummary/mlx.nn.Dropout3d.rst", "python/nn/_autosummary/mlx.nn.ELU.rst", "python/nn/_autosummary/mlx.nn.Embedding.rst", "python/nn/_autosummary/mlx.nn.GELU.rst", "python/nn/_autosummary/mlx.nn.GLU.rst", "python/nn/_autosummary/mlx.nn.GRU.rst", "python/nn/_autosummary/mlx.nn.GroupNorm.rst", "python/nn/_autosummary/mlx.nn.HardShrink.rst", "python/nn/_autosummary/mlx.nn.HardTanh.rst", "python/nn/_autosummary/mlx.nn.Hardswish.rst", "python/nn/_autosummary/mlx.nn.InstanceNorm.rst", "python/nn/_autosummary/mlx.nn.LSTM.rst", "python/nn/_autosummary/mlx.nn.LayerNorm.rst", "python/nn/_autosummary/mlx.nn.LeakyReLU.rst", "python/nn/_autosummary/mlx.nn.Linear.rst", "python/nn/_autosummary/mlx.nn.LogSigmoid.rst", "python/nn/_autosummary/mlx.nn.LogSoftmax.rst", "python/nn/_autosummary/mlx.nn.MaxPool1d.rst", "python/nn/_autosummary/mlx.nn.MaxPool2d.rst", "python/nn/_autosummary/mlx.nn.MaxPool3d.rst", "python/nn/_autosummary/mlx.nn.Mish.rst", "python/nn/_autosummary/mlx.nn.Module.apply.rst", "python/nn/_autosummary/mlx.nn.Module.apply_to_modules.rst", "python/nn/_autosummary/mlx.nn.Module.children.rst", "python/nn/_autosummary/mlx.nn.Module.eval.rst", "python/nn/_autosummary/mlx.nn.Module.filter_and_map.rst", "python/nn/_autosummary/mlx.nn.Module.freeze.rst", "python/nn/_autosummary/mlx.nn.Module.leaf_modules.rst", "python/nn/_autosummary/mlx.nn.Module.load_weights.rst", "python/nn/_autosummary/mlx.nn.Module.modules.rst", "python/nn/_autosummary/mlx.nn.Module.named_modules.rst", "python/nn/_autosummary/mlx.nn.Module.parameters.rst", "python/nn/_autosummary/mlx.nn.Module.save_weights.rst", "python/nn/_autosummary/mlx.nn.Module.set_dtype.rst", "python/nn/_autosummary/mlx.nn.Module.state.rst", "python/nn/_autosummary/mlx.nn.Module.train.rst", "python/nn/_autosummary/mlx.nn.Module.trainable_parameters.rst", "python/nn/_autosummary/mlx.nn.Module.training.rst", "python/nn/_autosummary/mlx.nn.Module.unfreeze.rst", "python/nn/_autosummary/mlx.nn.Module.update.rst", "python/nn/_autosummary/mlx.nn.Module.update_modules.rst", "python/nn/_autosummary/mlx.nn.MultiHeadAttention.rst", "python/nn/_autosummary/mlx.nn.PReLU.rst", "python/nn/_autosummary/mlx.nn.QuantizedEmbedding.rst", "python/nn/_autosummary/mlx.nn.QuantizedLinear.rst", "python/nn/_autosummary/mlx.nn.RMSNorm.rst", "python/nn/_autosummary/mlx.nn.RNN.rst", "python/nn/_autosummary/mlx.nn.ReLU.rst", "python/nn/_autosummary/mlx.nn.ReLU6.rst", "python/nn/_autosummary/mlx.nn.RoPE.rst", "python/nn/_autosummary/mlx.nn.SELU.rst", "python/nn/_autosummary/mlx.nn.Sequential.rst", "python/nn/_autosummary/mlx.nn.SiLU.rst", "python/nn/_autosummary/mlx.nn.Sigmoid.rst", "python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.rst", "python/nn/_autosummary/mlx.nn.Softmax.rst", "python/nn/_autosummary/mlx.nn.Softmin.rst", "python/nn/_autosummary/mlx.nn.Softplus.rst", "python/nn/_autosummary/mlx.nn.Softshrink.rst", "python/nn/_autosummary/mlx.nn.Softsign.rst", "python/nn/_autosummary/mlx.nn.Step.rst", "python/nn/_autosummary/mlx.nn.Tanh.rst", "python/nn/_autosummary/mlx.nn.Transformer.rst", "python/nn/_autosummary/mlx.nn.Upsample.rst", "python/nn/_autosummary/mlx.nn.init.constant.rst", "python/nn/_autosummary/mlx.nn.init.glorot_normal.rst", "python/nn/_autosummary/mlx.nn.init.glorot_uniform.rst", "python/nn/_autosummary/mlx.nn.init.he_normal.rst", "python/nn/_autosummary/mlx.nn.init.he_uniform.rst", "python/nn/_autosummary/mlx.nn.init.identity.rst", "python/nn/_autosummary/mlx.nn.init.normal.rst", "python/nn/_autosummary/mlx.nn.init.uniform.rst", "python/nn/_autosummary_functions/mlx.nn.celu.rst", "python/nn/_autosummary_functions/mlx.nn.elu.rst", "python/nn/_autosummary_functions/mlx.nn.gelu.rst", "python/nn/_autosummary_functions/mlx.nn.gelu_approx.rst", "python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.rst", "python/nn/_autosummary_functions/mlx.nn.glu.rst", "python/nn/_autosummary_functions/mlx.nn.hard_shrink.rst", "python/nn/_autosummary_functions/mlx.nn.hard_tanh.rst", "python/nn/_autosummary_functions/mlx.nn.hardswish.rst", "python/nn/_autosummary_functions/mlx.nn.leaky_relu.rst", "python/nn/_autosummary_functions/mlx.nn.log_sigmoid.rst", "python/nn/_autosummary_functions/mlx.nn.log_softmax.rst", "python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.rst", "python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.rst", "python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.rst", "python/nn/_autosummary_functions/mlx.nn.mish.rst", "python/nn/_autosummary_functions/mlx.nn.prelu.rst", "python/nn/_autosummary_functions/mlx.nn.relu.rst", "python/nn/_autosummary_functions/mlx.nn.relu6.rst", "python/nn/_autosummary_functions/mlx.nn.selu.rst", "python/nn/_autosummary_functions/mlx.nn.sigmoid.rst", "python/nn/_autosummary_functions/mlx.nn.silu.rst", "python/nn/_autosummary_functions/mlx.nn.softmax.rst", "python/nn/_autosummary_functions/mlx.nn.softmin.rst", "python/nn/_autosummary_functions/mlx.nn.softplus.rst", "python/nn/_autosummary_functions/mlx.nn.softshrink.rst", "python/nn/_autosummary_functions/mlx.nn.step.rst", "python/nn/_autosummary_functions/mlx.nn.tanh.rst", "python/nn/functions.rst", "python/nn/init.rst", "python/nn/layers.rst", "python/nn/losses.rst", "python/nn/module.rst", "python/ops.rst", "python/optimizers.rst", "python/optimizers/_autosummary/mlx.optimizers.AdaDelta.rst", "python/optimizers/_autosummary/mlx.optimizers.Adafactor.rst", "python/optimizers/_autosummary/mlx.optimizers.Adagrad.rst", "python/optimizers/_autosummary/mlx.optimizers.Adam.rst", "python/optimizers/_autosummary/mlx.optimizers.AdamW.rst", "python/optimizers/_autosummary/mlx.optimizers.Adamax.rst", "python/optimizers/_autosummary/mlx.optimizers.Lion.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.rst", "python/optimizers/_autosummary/mlx.optimizers.RMSprop.rst", "python/optimizers/_autosummary/mlx.optimizers.SGD.rst", "python/optimizers/_autosummary/mlx.optimizers.cosine_decay.rst", "python/optimizers/_autosummary/mlx.optimizers.exponential_decay.rst", "python/optimizers/_autosummary/mlx.optimizers.join_schedules.rst", "python/optimizers/_autosummary/mlx.optimizers.linear_schedule.rst", "python/optimizers/_autosummary/mlx.optimizers.step_decay.rst", "python/optimizers/common_optimizers.rst", "python/optimizers/optimizer.rst", "python/optimizers/schedulers.rst", "python/random.rst", "python/transforms.rst", "python/tree_utils.rst", "usage/compile.rst", "usage/distributed.rst", "usage/export.rst", "usage/function_transforms.rst", "usage/indexing.rst", "usage/lazy_evaluation.rst", "usage/numpy.rst", "usage/quick_start.rst", "usage/saving_and_loading.rst", "usage/unified_memory.rst", "usage/using_streams.rst"], "indexentries": {"__init__() (array method)": [[31, "mlx.core.array.__init__", false]], "__init__() (custom_function method)": [[113, "mlx.core.custom_function.__init__", false]], "__init__() (device method)": [[10, "mlx.core.Device.__init__", false]], "__init__() (dtype method)": [[11, "mlx.core.Dtype.__init__", false]], "__init__() (dtypecategory method)": [[12, "mlx.core.DtypeCategory.__init__", false]], "__init__() (finfo method)": [[162, "mlx.core.finfo.__init__", false]], "__init__() (group method)": [[121, "mlx.core.distributed.Group.__init__", false]], "__init__() (stream method)": [[324, "mlx.core.Stream.__init__", false]], "abs (c++ function)": [[0, "_CPPv43absRK5array14StreamOrDevice", false]], "abs() (array method)": [[33, "mlx.core.array.abs", false]], "abs() (in module mlx.core)": [[13, "mlx.core.abs", false]], "adadelta (class in mlx.optimizers)": [[467, "mlx.optimizers.AdaDelta", false]], "adafactor (class in mlx.optimizers)": [[468, "mlx.optimizers.Adafactor", false]], "adagrad (class in mlx.optimizers)": [[469, "mlx.optimizers.Adagrad", false]], "adam (class in mlx.optimizers)": [[470, "mlx.optimizers.Adam", false]], "adamax (class in mlx.optimizers)": [[472, "mlx.optimizers.Adamax", false]], "adamw (class in mlx.optimizers)": [[471, "mlx.optimizers.AdamW", false]], "add (c++ function)": [[0, "_CPPv43addRK5arrayRK5array14StreamOrDevice", false]], "add() (in module mlx.core)": [[14, "mlx.core.add", false]], "addmm (c++ function)": [[0, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", false]], "addmm() (in module mlx.core)": [[15, "mlx.core.addmm", false]], "alibi (class in mlx.nn)": [[335, "mlx.nn.ALiBi", false]], "all (c++ function)": [[0, "_CPPv43allRK5array14StreamOrDevice", false], [0, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43allRK5arrayb14StreamOrDevice", false], [0, "_CPPv43allRK5arrayib14StreamOrDevice", false]], "all() (array method)": [[34, "mlx.core.array.all", false]], "all() (in module mlx.core)": [[16, "mlx.core.all", false]], "all_gather() (in module mlx.core.distributed)": [[122, "mlx.core.distributed.all_gather", false]], "all_sum() (in module mlx.core.distributed)": [[123, "mlx.core.distributed.all_sum", false]], "allclose (c++ function)": [[0, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "allclose() (in module mlx.core)": [[17, "mlx.core.allclose", false]], "any (c++ function)": [[0, "_CPPv43anyRK5array14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayb14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayib14StreamOrDevice", false]], "any() (array method)": [[35, "mlx.core.array.any", false]], "any() (in module mlx.core)": [[18, "mlx.core.any", false]], "apply() (module method)": [[370, "mlx.nn.Module.apply", false]], "apply_gradients() (optimizer method)": [[474, "mlx.optimizers.Optimizer.apply_gradients", false]], "apply_to_modules() (module method)": [[371, "mlx.nn.Module.apply_to_modules", false]], "arange (c++ function)": [[0, "_CPPv46aranged14StreamOrDevice", false], [0, "_CPPv46aranged5Dtype14StreamOrDevice", false], [0, "_CPPv46arangedd14StreamOrDevice", false], [0, "_CPPv46arangedd5Dtype14StreamOrDevice", false], [0, "_CPPv46arangeddd14StreamOrDevice", false], [0, "_CPPv46arangeddd5Dtype14StreamOrDevice", false], [0, "_CPPv46arangei14StreamOrDevice", false], [0, "_CPPv46arangeii14StreamOrDevice", false], [0, "_CPPv46arangeiii14StreamOrDevice", false]], "arange() (in module mlx.core)": [[19, "mlx.core.arange", false]], "arccos (c++ function)": [[0, "_CPPv46arccosRK5array14StreamOrDevice", false]], "arccos() (in module mlx.core)": [[20, "mlx.core.arccos", false]], "arccosh (c++ function)": [[0, "_CPPv47arccoshRK5array14StreamOrDevice", false]], "arccosh() (in module mlx.core)": [[21, "mlx.core.arccosh", false]], "arcsin (c++ function)": [[0, "_CPPv46arcsinRK5array14StreamOrDevice", false]], "arcsin() (in module mlx.core)": [[22, "mlx.core.arcsin", false]], "arcsinh (c++ function)": [[0, "_CPPv47arcsinhRK5array14StreamOrDevice", false]], "arcsinh() (in module mlx.core)": [[23, "mlx.core.arcsinh", false]], "arctan (c++ function)": [[0, "_CPPv46arctanRK5array14StreamOrDevice", false]], "arctan() (in module mlx.core)": [[24, "mlx.core.arctan", false]], "arctan2 (c++ function)": [[0, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", false]], "arctan2() (in module mlx.core)": [[25, "mlx.core.arctan2", false]], "arctanh (c++ function)": [[0, "_CPPv47arctanhRK5array14StreamOrDevice", false]], "arctanh() (in module mlx.core)": [[26, "mlx.core.arctanh", false]], "argmax (c++ function)": [[0, "_CPPv46argmaxRK5array14StreamOrDevice", false], [0, "_CPPv46argmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv46argmaxRK5arrayib14StreamOrDevice", false]], "argmax() (array method)": [[36, "mlx.core.array.argmax", false]], "argmax() (in module mlx.core)": [[27, "mlx.core.argmax", false]], "argmin (c++ function)": [[0, "_CPPv46argminRK5array14StreamOrDevice", false], [0, "_CPPv46argminRK5arrayb14StreamOrDevice", false], [0, "_CPPv46argminRK5arrayib14StreamOrDevice", false]], "argmin() (array method)": [[37, "mlx.core.array.argmin", false]], "argmin() (in module mlx.core)": [[28, "mlx.core.argmin", false]], "argpartition (c++ function)": [[0, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", false]], "argpartition() (in module mlx.core)": [[29, "mlx.core.argpartition", false]], "argsort (c++ function)": [[0, "_CPPv47argsortRK5array14StreamOrDevice", false], [0, "_CPPv47argsortRK5arrayi14StreamOrDevice", false]], "argsort() (in module mlx.core)": [[30, "mlx.core.argsort", false]], "array (class in mlx.core)": [[31, "mlx.core.array", false]], "array_equal (c++ function)": [[0, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", false], [0, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", false]], "array_equal() (in module mlx.core)": [[83, "mlx.core.array_equal", false]], "as_strided (c++ function)": [[0, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", false]], "as_strided() (in module mlx.core)": [[84, "mlx.core.as_strided", false]], "astype (c++ function)": [[0, "_CPPv46astype5array5Dtype14StreamOrDevice", false]], "astype() (array method)": [[38, "mlx.core.array.astype", false]], "at (array property)": [[39, "mlx.core.array.at", false]], "atleast_1d (c++ function)": [[0, "_CPPv410atleast_1dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_1d() (in module mlx.core)": [[85, "mlx.core.atleast_1d", false]], "atleast_2d (c++ function)": [[0, "_CPPv410atleast_2dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_2d() (in module mlx.core)": [[86, "mlx.core.atleast_2d", false]], "atleast_3d (c++ function)": [[0, "_CPPv410atleast_3dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_3d() (in module mlx.core)": [[87, "mlx.core.atleast_3d", false]], "avgpool1d (class in mlx.nn)": [[336, "mlx.nn.AvgPool1d", false]], "avgpool2d (class in mlx.nn)": [[337, "mlx.nn.AvgPool2d", false]], "avgpool3d (class in mlx.nn)": [[338, "mlx.nn.AvgPool3d", false]], "batchnorm (class in mlx.nn)": [[339, "mlx.nn.BatchNorm", false]], "bernoulli() (in module mlx.core.random)": [[246, "mlx.core.random.bernoulli", false]], "binary_cross_entropy (class in mlx.nn.losses)": [[433, "mlx.nn.losses.binary_cross_entropy", false]], "bitwise_and (c++ function)": [[0, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", false]], "bitwise_and() (in module mlx.core)": [[88, "mlx.core.bitwise_and", false]], "bitwise_or (c++ function)": [[0, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", false]], "bitwise_or() (in module mlx.core)": [[89, "mlx.core.bitwise_or", false]], "bitwise_xor (c++ function)": [[0, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", false]], "bitwise_xor() (in module mlx.core)": [[90, "mlx.core.bitwise_xor", false]], "block_masked_mm (c++ function)": [[0, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "block_masked_mm() (in module mlx.core)": [[91, "mlx.core.block_masked_mm", false]], "broadcast_arrays (c++ function)": [[0, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "broadcast_to (c++ function)": [[0, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", false]], "broadcast_to() (in module mlx.core)": [[92, "mlx.core.broadcast_to", false]], "categorical() (in module mlx.core.random)": [[247, "mlx.core.random.categorical", false]], "ceil (c++ function)": [[0, "_CPPv44ceilRK5array14StreamOrDevice", false]], "ceil() (in module mlx.core)": [[93, "mlx.core.ceil", false]], "celu (class in mlx.nn)": [[340, "mlx.nn.CELU", false], [421, "mlx.nn.celu", false]], "children() (module method)": [[372, "mlx.nn.Module.children", false]], "cholesky() (in module mlx.core.linalg)": [[189, "mlx.core.linalg.cholesky", false]], "cholesky_inv() (in module mlx.core.linalg)": [[190, "mlx.core.linalg.cholesky_inv", false]], "clear_cache() (in module mlx.core.metal)": [[215, "mlx.core.metal.clear_cache", false]], "clip (c++ function)": [[0, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", false]], "clip() (in module mlx.core)": [[94, "mlx.core.clip", false]], "clip_grad_norm() (in module mlx.optimizers)": [[318, "mlx.optimizers.clip_grad_norm", false]], "compile() (in module mlx.core)": [[95, "mlx.core.compile", false]], "concatenate (c++ function)": [[0, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", false]], "concatenate() (in module mlx.core)": [[96, "mlx.core.concatenate", false]], "conj() (array method)": [[40, "mlx.core.array.conj", false]], "conj() (in module mlx.core)": [[97, "mlx.core.conj", false]], "conjugate (c++ function)": [[0, "_CPPv49conjugateRK5array14StreamOrDevice", false]], "conjugate() (in module mlx.core)": [[98, "mlx.core.conjugate", false]], "constant() (in module mlx.nn.init)": [[413, "mlx.nn.init.constant", false]], "contiguous (c++ function)": [[0, "_CPPv410contiguousRK5arrayb14StreamOrDevice", false]], "conv1d (c++ function)": [[0, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv1d (class in mlx.nn)": [[341, "mlx.nn.Conv1d", false]], "conv1d() (in module mlx.core)": [[99, "mlx.core.conv1d", false]], "conv2d (c++ function)": [[0, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv2d (class in mlx.nn)": [[342, "mlx.nn.Conv2d", false]], "conv2d() (in module mlx.core)": [[100, "mlx.core.conv2d", false]], "conv3d (c++ function)": [[0, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv3d (class in mlx.nn)": [[343, "mlx.nn.Conv3d", false]], "conv3d() (in module mlx.core)": [[101, "mlx.core.conv3d", false]], "conv_general (c++ function)": [[0, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false], [0, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false]], "conv_general() (in module mlx.core)": [[102, "mlx.core.conv_general", false]], "conv_transpose1d (c++ function)": [[0, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv_transpose1d() (in module mlx.core)": [[103, "mlx.core.conv_transpose1d", false]], "conv_transpose2d (c++ function)": [[0, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv_transpose2d() (in module mlx.core)": [[104, "mlx.core.conv_transpose2d", false]], "conv_transpose3d (c++ function)": [[0, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv_transpose3d() (in module mlx.core)": [[105, "mlx.core.conv_transpose3d", false]], "convolve() (in module mlx.core)": [[106, "mlx.core.convolve", false]], "convtranspose1d (class in mlx.nn)": [[344, "mlx.nn.ConvTranspose1d", false]], "convtranspose2d (class in mlx.nn)": [[345, "mlx.nn.ConvTranspose2d", false]], "convtranspose3d (class in mlx.nn)": [[346, "mlx.nn.ConvTranspose3d", false]], "copy (c++ function)": [[0, "_CPPv44copy5array14StreamOrDevice", false]], "cos (c++ function)": [[0, "_CPPv43cosRK5array14StreamOrDevice", false]], "cos() (array method)": [[41, "mlx.core.array.cos", false]], "cos() (in module mlx.core)": [[107, "mlx.core.cos", false]], "cosh (c++ function)": [[0, "_CPPv44coshRK5array14StreamOrDevice", false]], "cosh() (in module mlx.core)": [[108, "mlx.core.cosh", false]], "cosine_decay() (in module mlx.optimizers)": [[480, "mlx.optimizers.cosine_decay", false]], "cosine_similarity_loss (class in mlx.nn.losses)": [[434, "mlx.nn.losses.cosine_similarity_loss", false]], "cross() (in module mlx.core.linalg)": [[191, "mlx.core.linalg.cross", false]], "cross_entropy (class in mlx.nn.losses)": [[435, "mlx.nn.losses.cross_entropy", false]], "cummax (c++ function)": [[0, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", false]], "cummax() (array method)": [[42, "mlx.core.array.cummax", false]], "cummax() (in module mlx.core)": [[109, "mlx.core.cummax", false]], "cummin (c++ function)": [[0, "_CPPv46cumminRK5arrayibb14StreamOrDevice", false]], "cummin() (array method)": [[43, "mlx.core.array.cummin", false]], "cummin() (in module mlx.core)": [[110, "mlx.core.cummin", false]], "cumprod (c++ function)": [[0, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", false]], "cumprod() (array method)": [[44, "mlx.core.array.cumprod", false]], "cumprod() (in module mlx.core)": [[111, "mlx.core.cumprod", false]], "cumsum (c++ function)": [[0, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", false]], "cumsum() (array method)": [[45, "mlx.core.array.cumsum", false]], "cumsum() (in module mlx.core)": [[112, "mlx.core.cumsum", false]], "custom_function (class in mlx.core)": [[113, "mlx.core.custom_function", false]], "default_device() (in module mlx.core)": [[114, "mlx.core.default_device", false]], "default_stream() (in module mlx.core)": [[115, "mlx.core.default_stream", false]], "degrees (c++ function)": [[0, "_CPPv47degreesRK5array14StreamOrDevice", false]], "degrees() (in module mlx.core)": [[116, "mlx.core.degrees", false]], "depends (c++ function)": [[0, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", false]], "dequantize (c++ function)": [[0, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", false]], "dequantize() (in module mlx.core)": [[117, "mlx.core.dequantize", false]], "device (class in mlx.core)": [[10, "mlx.core.Device", false]], "device_info() (in module mlx.core.metal)": [[216, "mlx.core.metal.device_info", false]], "diag (c++ function)": [[0, "_CPPv44diagRK5arrayi14StreamOrDevice", false]], "diag() (array method)": [[46, "mlx.core.array.diag", false]], "diag() (in module mlx.core)": [[118, "mlx.core.diag", false]], "diagonal (c++ function)": [[0, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", false]], "diagonal() (array method)": [[47, "mlx.core.array.diagonal", false]], "diagonal() (in module mlx.core)": [[119, "mlx.core.diagonal", false]], "disable_compile() (in module mlx.core)": [[120, "mlx.core.disable_compile", false]], "divide (c++ function)": [[0, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", false]], "divide() (in module mlx.core)": [[129, "mlx.core.divide", false]], "divmod (c++ function)": [[0, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", false]], "divmod() (in module mlx.core)": [[130, "mlx.core.divmod", false]], "dropout (class in mlx.nn)": [[347, "mlx.nn.Dropout", false]], "dropout2d (class in mlx.nn)": [[348, "mlx.nn.Dropout2d", false]], "dropout3d (class in mlx.nn)": [[349, "mlx.nn.Dropout3d", false]], "dtype (array property)": [[48, "mlx.core.array.dtype", false]], "dtype (class in mlx.core)": [[11, "mlx.core.Dtype", false]], "dtypecategory (class in mlx.core)": [[12, "mlx.core.DtypeCategory", false]], "eigh() (in module mlx.core.linalg)": [[192, "mlx.core.linalg.eigh", false]], "eigvalsh() (in module mlx.core.linalg)": [[193, "mlx.core.linalg.eigvalsh", false]], "einsum() (in module mlx.core)": [[131, "mlx.core.einsum", false]], "einsum_path() (in module mlx.core)": [[132, "mlx.core.einsum_path", false]], "elu (class in mlx.nn)": [[350, "mlx.nn.ELU", false], [422, "mlx.nn.elu", false]], "embedding (class in mlx.nn)": [[351, "mlx.nn.Embedding", false]], "enable_compile() (in module mlx.core)": [[133, "mlx.core.enable_compile", false]], "equal (c++ function)": [[0, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", false]], "equal() (in module mlx.core)": [[134, "mlx.core.equal", false]], "erf (c++ function)": [[0, "_CPPv43erfRK5array14StreamOrDevice", false]], "erf() (in module mlx.core)": [[135, "mlx.core.erf", false]], "erfinv (c++ function)": [[0, "_CPPv46erfinvRK5array14StreamOrDevice", false]], "erfinv() (in module mlx.core)": [[136, "mlx.core.erfinv", false]], "eval() (in module mlx.core)": [[137, "mlx.core.eval", false]], "eval() (module method)": [[373, "mlx.nn.Module.eval", false]], "exp (c++ function)": [[0, "_CPPv43expRK5array14StreamOrDevice", false]], "exp() (array method)": [[49, "mlx.core.array.exp", false]], "exp() (in module mlx.core)": [[138, "mlx.core.exp", false]], "expand_dims (c++ function)": [[0, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", false]], "expand_dims() (in module mlx.core)": [[139, "mlx.core.expand_dims", false]], "expm1 (c++ function)": [[0, "_CPPv45expm1RK5array14StreamOrDevice", false]], "expm1() (in module mlx.core)": [[140, "mlx.core.expm1", false]], "exponential_decay() (in module mlx.optimizers)": [[481, "mlx.optimizers.exponential_decay", false]], "export_function() (in module mlx.core)": [[141, "mlx.core.export_function", false]], "export_to_dot() (in module mlx.core)": [[142, "mlx.core.export_to_dot", false]], "exporter() (in module mlx.core)": [[143, "mlx.core.exporter", false]], "eye (c++ function)": [[0, "_CPPv43eyei14StreamOrDevice", false], [0, "_CPPv43eyei5Dtype14StreamOrDevice", false], [0, "_CPPv43eyeii14StreamOrDevice", false], [0, "_CPPv43eyeiii14StreamOrDevice", false], [0, "_CPPv43eyeiii5Dtype14StreamOrDevice", false]], "eye() (in module mlx.core)": [[144, "mlx.core.eye", false]], "fft() (in module mlx.core.fft)": [[150, "mlx.core.fft.fft", false]], "fft2() (in module mlx.core.fft)": [[151, "mlx.core.fft.fft2", false]], "fftn() (in module mlx.core.fft)": [[152, "mlx.core.fft.fftn", false]], "filter_and_map() (module method)": [[374, "mlx.nn.Module.filter_and_map", false]], "finfo (class in mlx.core)": [[162, "mlx.core.finfo", false]], "flatten (c++ function)": [[0, "_CPPv47flattenRK5array14StreamOrDevice", false], [0, "_CPPv47flattenRK5arrayii14StreamOrDevice", false]], "flatten() (array method)": [[50, "mlx.core.array.flatten", false]], "flatten() (in module mlx.core)": [[163, "mlx.core.flatten", false]], "floor (c++ function)": [[0, "_CPPv45floorRK5array14StreamOrDevice", false]], "floor() (in module mlx.core)": [[164, "mlx.core.floor", false]], "floor_divide (c++ function)": [[0, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", false]], "floor_divide() (in module mlx.core)": [[165, "mlx.core.floor_divide", false]], "freeze() (module method)": [[375, "mlx.nn.Module.freeze", false]], "full (c++ function)": [[0, "_CPPv44full5Shape5array14StreamOrDevice", false], [0, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", false], [0, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", false], [0, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", false]], "full() (in module mlx.core)": [[166, "mlx.core.full", false]], "gather (c++ function)": [[0, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", false], [0, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", false]], "gather_mm (c++ function)": [[0, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "gather_mm() (in module mlx.core)": [[167, "mlx.core.gather_mm", false]], "gather_qmm (c++ function)": [[0, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", false]], "gather_qmm() (in module mlx.core)": [[168, "mlx.core.gather_qmm", false]], "gaussian_nll_loss (class in mlx.nn.losses)": [[436, "mlx.nn.losses.gaussian_nll_loss", false]], "gelu (class in mlx.nn)": [[352, "mlx.nn.GELU", false], [423, "mlx.nn.gelu", false]], "gelu_approx (class in mlx.nn)": [[424, "mlx.nn.gelu_approx", false]], "gelu_fast_approx (class in mlx.nn)": [[425, "mlx.nn.gelu_fast_approx", false]], "get_active_memory() (in module mlx.core.metal)": [[217, "mlx.core.metal.get_active_memory", false]], "get_cache_memory() (in module mlx.core.metal)": [[218, "mlx.core.metal.get_cache_memory", false]], "get_peak_memory() (in module mlx.core.metal)": [[219, "mlx.core.metal.get_peak_memory", false]], "glorot_normal() (in module mlx.nn.init)": [[414, "mlx.nn.init.glorot_normal", false]], "glorot_uniform() (in module mlx.nn.init)": [[415, "mlx.nn.init.glorot_uniform", false]], "glu (class in mlx.nn)": [[353, "mlx.nn.GLU", false], [426, "mlx.nn.glu", false]], "grad() (in module mlx.core)": [[169, "mlx.core.grad", false]], "greater (c++ function)": [[0, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", false]], "greater() (in module mlx.core)": [[170, "mlx.core.greater", false]], "greater_equal (c++ function)": [[0, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", false]], "greater_equal() (in module mlx.core)": [[171, "mlx.core.greater_equal", false]], "group (class in mlx.core.distributed)": [[121, "mlx.core.distributed.Group", false]], "groupnorm (class in mlx.nn)": [[355, "mlx.nn.GroupNorm", false]], "gru (class in mlx.nn)": [[354, "mlx.nn.GRU", false]], "gumbel() (in module mlx.core.random)": [[248, "mlx.core.random.gumbel", false]], "hadamard_transform (c++ function)": [[0, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", false]], "hadamard_transform() (in module mlx.core)": [[172, "mlx.core.hadamard_transform", false]], "hard_shrink (class in mlx.nn)": [[427, "mlx.nn.hard_shrink", false]], "hard_tanh (class in mlx.nn)": [[428, "mlx.nn.hard_tanh", false]], "hardshrink (class in mlx.nn)": [[356, "mlx.nn.HardShrink", false]], "hardswish (class in mlx.nn)": [[358, "mlx.nn.Hardswish", false], [429, "mlx.nn.hardswish", false]], "hardtanh (class in mlx.nn)": [[357, "mlx.nn.HardTanh", false]], "he_normal() (in module mlx.nn.init)": [[416, "mlx.nn.init.he_normal", false]], "he_uniform() (in module mlx.nn.init)": [[417, "mlx.nn.init.he_uniform", false]], "hinge_loss (class in mlx.nn.losses)": [[437, "mlx.nn.losses.hinge_loss", false]], "huber_loss (class in mlx.nn.losses)": [[438, "mlx.nn.losses.huber_loss", false]], "identity (c++ function)": [[0, "_CPPv48identityi14StreamOrDevice", false], [0, "_CPPv48identityi5Dtype14StreamOrDevice", false]], "identity() (in module mlx.core)": [[173, "mlx.core.identity", false]], "identity() (in module mlx.nn.init)": [[418, "mlx.nn.init.identity", false]], "ifft() (in module mlx.core.fft)": [[153, "mlx.core.fft.ifft", false]], "ifft2() (in module mlx.core.fft)": [[154, "mlx.core.fft.ifft2", false]], "ifftn() (in module mlx.core.fft)": [[155, "mlx.core.fft.ifftn", false]], "imag (c++ function)": [[0, "_CPPv44imagRK5array14StreamOrDevice", false]], "imag() (in module mlx.core)": [[174, "mlx.core.imag", false]], "import_function() (in module mlx.core)": [[175, "mlx.core.import_function", false]], "init() (in module mlx.core.distributed)": [[124, "mlx.core.distributed.init", false]], "init() (optimizer method)": [[475, "mlx.optimizers.Optimizer.init", false]], "inner (c++ function)": [[0, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", false]], "inner() (in module mlx.core)": [[176, "mlx.core.inner", false]], "instancenorm (class in mlx.nn)": [[359, "mlx.nn.InstanceNorm", false]], "inv() (in module mlx.core.linalg)": [[194, "mlx.core.linalg.inv", false]], "irfft() (in module mlx.core.fft)": [[156, "mlx.core.fft.irfft", false]], "irfft2() (in module mlx.core.fft)": [[157, "mlx.core.fft.irfft2", false]], "irfftn() (in module mlx.core.fft)": [[158, "mlx.core.fft.irfftn", false]], "is_available() (in module mlx.core.distributed)": [[125, "mlx.core.distributed.is_available", false]], "is_available() (in module mlx.core.metal)": [[220, "mlx.core.metal.is_available", false]], "isclose (c++ function)": [[0, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "isclose() (in module mlx.core)": [[177, "mlx.core.isclose", false]], "isfinite (c++ function)": [[0, "_CPPv48isfiniteRK5array14StreamOrDevice", false]], "isfinite() (in module mlx.core)": [[178, "mlx.core.isfinite", false]], "isinf (c++ function)": [[0, "_CPPv45isinfRK5array14StreamOrDevice", false]], "isinf() (in module mlx.core)": [[179, "mlx.core.isinf", false]], "isnan (c++ function)": [[0, "_CPPv45isnanRK5array14StreamOrDevice", false]], "isnan() (in module mlx.core)": [[180, "mlx.core.isnan", false]], "isneginf (c++ function)": [[0, "_CPPv48isneginfRK5array14StreamOrDevice", false]], "isneginf() (in module mlx.core)": [[181, "mlx.core.isneginf", false]], "isposinf (c++ function)": [[0, "_CPPv48isposinfRK5array14StreamOrDevice", false]], "isposinf() (in module mlx.core)": [[182, "mlx.core.isposinf", false]], "issubdtype() (in module mlx.core)": [[183, "mlx.core.issubdtype", false]], "item() (array method)": [[51, "mlx.core.array.item", false]], "itemsize (array property)": [[52, "mlx.core.array.itemsize", false]], "join_schedules() (in module mlx.optimizers)": [[482, "mlx.optimizers.join_schedules", false]], "jvp() (in module mlx.core)": [[184, "mlx.core.jvp", false]], "key() (in module mlx.core.random)": [[249, "mlx.core.random.key", false]], "kl_div_loss (class in mlx.nn.losses)": [[439, "mlx.nn.losses.kl_div_loss", false]], "kron (c++ function)": [[0, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", false]], "kron() (in module mlx.core)": [[185, "mlx.core.kron", false]], "l1_loss (class in mlx.nn.losses)": [[440, "mlx.nn.losses.l1_loss", false]], "laplace() (in module mlx.core.random)": [[250, "mlx.core.random.laplace", false]], "layer_norm() (in module mlx.core.fast)": [[145, "mlx.core.fast.layer_norm", false]], "layernorm (class in mlx.nn)": [[361, "mlx.nn.LayerNorm", false]], "leaf_modules() (module method)": [[376, "mlx.nn.Module.leaf_modules", false]], "leaky_relu (class in mlx.nn)": [[430, "mlx.nn.leaky_relu", false]], "leakyrelu (class in mlx.nn)": [[362, "mlx.nn.LeakyReLU", false]], "left_shift (c++ function)": [[0, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", false]], "left_shift() (in module mlx.core)": [[186, "mlx.core.left_shift", false]], "less (c++ function)": [[0, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", false]], "less() (in module mlx.core)": [[187, "mlx.core.less", false]], "less_equal (c++ function)": [[0, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", false]], "less_equal() (in module mlx.core)": [[188, "mlx.core.less_equal", false]], "linear (class in mlx.nn)": [[363, "mlx.nn.Linear", false]], "linear_schedule() (in module mlx.optimizers)": [[483, "mlx.optimizers.linear_schedule", false]], "linspace (c++ function)": [[0, "_CPPv48linspaceddi5Dtype14StreamOrDevice", false]], "linspace() (in module mlx.core)": [[199, "mlx.core.linspace", false]], "lion (class in mlx.optimizers)": [[473, "mlx.optimizers.Lion", false]], "load() (in module mlx.core)": [[200, "mlx.core.load", false]], "load_weights() (module method)": [[377, "mlx.nn.Module.load_weights", false]], "log (c++ function)": [[0, "_CPPv43logRK5array14StreamOrDevice", false]], "log() (array method)": [[53, "mlx.core.array.log", false]], "log() (in module mlx.core)": [[201, "mlx.core.log", false]], "log10 (c++ function)": [[0, "_CPPv45log10RK5array14StreamOrDevice", false]], "log10() (array method)": [[54, "mlx.core.array.log10", false]], "log10() (in module mlx.core)": [[202, "mlx.core.log10", false]], "log1p (c++ function)": [[0, "_CPPv45log1pRK5array14StreamOrDevice", false]], "log1p() (array method)": [[55, "mlx.core.array.log1p", false]], "log1p() (in module mlx.core)": [[203, "mlx.core.log1p", false]], "log2 (c++ function)": [[0, "_CPPv44log2RK5array14StreamOrDevice", false]], "log2() (array method)": [[56, "mlx.core.array.log2", false]], "log2() (in module mlx.core)": [[204, "mlx.core.log2", false]], "log_cosh_loss (class in mlx.nn.losses)": [[441, "mlx.nn.losses.log_cosh_loss", false]], "log_sigmoid (class in mlx.nn)": [[431, "mlx.nn.log_sigmoid", false]], "log_softmax (class in mlx.nn)": [[432, "mlx.nn.log_softmax", false]], "logaddexp (c++ function)": [[0, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", false]], "logaddexp() (in module mlx.core)": [[205, "mlx.core.logaddexp", false]], "logical_and (c++ function)": [[0, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", false]], "logical_and() (in module mlx.core)": [[206, "mlx.core.logical_and", false]], "logical_not (c++ function)": [[0, "_CPPv411logical_notRK5array14StreamOrDevice", false]], "logical_not() (in module mlx.core)": [[207, "mlx.core.logical_not", false]], "logical_or (c++ function)": [[0, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", false]], "logical_or() (in module mlx.core)": [[208, "mlx.core.logical_or", false]], "logsigmoid (class in mlx.nn)": [[364, "mlx.nn.LogSigmoid", false]], "logsoftmax (class in mlx.nn)": [[365, "mlx.nn.LogSoftmax", false]], "logsumexp (c++ function)": [[0, "_CPPv49logsumexpRK5array14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", false]], "logsumexp() (array method)": [[57, "mlx.core.array.logsumexp", false]], "logsumexp() (in module mlx.core)": [[209, "mlx.core.logsumexp", false]], "lstm (class in mlx.nn)": [[360, "mlx.nn.LSTM", false]], "margin_ranking_loss (class in mlx.nn.losses)": [[442, "mlx.nn.losses.margin_ranking_loss", false]], "matmul (c++ function)": [[0, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", false]], "matmul() (in module mlx.core)": [[210, "mlx.core.matmul", false]], "max (c++ function)": [[0, "_CPPv43maxRK5array14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayib14StreamOrDevice", false]], "max() (array method)": [[58, "mlx.core.array.max", false]], "max() (in module mlx.core)": [[211, "mlx.core.max", false]], "maximum (c++ function)": [[0, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", false]], "maximum() (in module mlx.core)": [[212, "mlx.core.maximum", false]], "maxpool1d (class in mlx.nn)": [[366, "mlx.nn.MaxPool1d", false]], "maxpool2d (class in mlx.nn)": [[367, "mlx.nn.MaxPool2d", false]], "maxpool3d (class in mlx.nn)": [[368, "mlx.nn.MaxPool3d", false]], "mean (c++ function)": [[0, "_CPPv44meanRK5array14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayib14StreamOrDevice", false]], "mean() (array method)": [[59, "mlx.core.array.mean", false]], "mean() (in module mlx.core)": [[213, "mlx.core.mean", false]], "meshgrid (c++ function)": [[0, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", false]], "meshgrid() (in module mlx.core)": [[214, "mlx.core.meshgrid", false]], "metal_kernel() (in module mlx.core.fast)": [[146, "mlx.core.fast.metal_kernel", false]], "min (c++ function)": [[0, "_CPPv43minRK5array14StreamOrDevice", false], [0, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayib14StreamOrDevice", false]], "min() (array method)": [[60, "mlx.core.array.min", false]], "min() (in module mlx.core)": [[227, "mlx.core.min", false]], "minimum (c++ function)": [[0, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", false]], "minimum() (in module mlx.core)": [[228, "mlx.core.minimum", false]], "mish (class in mlx.nn)": [[369, "mlx.nn.Mish", false], [447, "mlx.nn.mish", false]], "module (class in mlx.nn)": [[464, "mlx.nn.Module", false]], "modules() (module method)": [[378, "mlx.nn.Module.modules", false]], "moveaxis (c++ function)": [[0, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", false]], "moveaxis() (array method)": [[61, "mlx.core.array.moveaxis", false]], "moveaxis() (in module mlx.core)": [[229, "mlx.core.moveaxis", false]], "mse_loss (class in mlx.nn.losses)": [[443, "mlx.nn.losses.mse_loss", false]], "multiheadattention (class in mlx.nn)": [[390, "mlx.nn.MultiHeadAttention", false]], "multiply (c++ function)": [[0, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", false]], "multiply() (in module mlx.core)": [[230, "mlx.core.multiply", false]], "multivariate_normal() (in module mlx.core.random)": [[251, "mlx.core.random.multivariate_normal", false]], "named_modules() (module method)": [[379, "mlx.nn.Module.named_modules", false]], "nan_to_num (c++ function)": [[0, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", false]], "nan_to_num() (in module mlx.core)": [[231, "mlx.core.nan_to_num", false]], "nbytes (array property)": [[62, "mlx.core.array.nbytes", false]], "ndim (array property)": [[63, "mlx.core.array.ndim", false]], "negative (c++ function)": [[0, "_CPPv48negativeRK5array14StreamOrDevice", false]], "negative() (in module mlx.core)": [[232, "mlx.core.negative", false]], "new_stream() (in module mlx.core)": [[233, "mlx.core.new_stream", false]], "nll_loss (class in mlx.nn.losses)": [[444, "mlx.nn.losses.nll_loss", false]], "norm() (in module mlx.core.linalg)": [[195, "mlx.core.linalg.norm", false]], "normal() (in module mlx.core.random)": [[252, "mlx.core.random.normal", false]], "normal() (in module mlx.nn.init)": [[419, "mlx.nn.init.normal", false]], "not_equal (c++ function)": [[0, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", false]], "not_equal() (in module mlx.core)": [[234, "mlx.core.not_equal", false]], "number_of_elements (c++ function)": [[0, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", false]], "ones (c++ function)": [[0, "_CPPv44onesRK5Shape14StreamOrDevice", false], [0, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", false]], "ones() (in module mlx.core)": [[235, "mlx.core.ones", false]], "ones_like (c++ function)": [[0, "_CPPv49ones_likeRK5array14StreamOrDevice", false]], "ones_like() (in module mlx.core)": [[236, "mlx.core.ones_like", false]], "operator!= (c++ function)": [[0, "_CPPv4I0Ene5array1TRK5array", false], [0, "_CPPv4I0Ene5arrayRK5array1T", false], [0, "_CPPv4neRK5arrayRK5array", false]], "operator% (c++ function)": [[0, "_CPPv4I0Erm5array1TRK5array", false], [0, "_CPPv4I0Erm5arrayRK5array1T", false], [0, "_CPPv4rmRK5arrayRK5array", false]], "operator& (c++ function)": [[0, "_CPPv4anRK5arrayRK5array", false]], "operator&& (c++ function)": [[0, "_CPPv4aaRK5arrayRK5array", false]], "operator* (c++ function)": [[0, "_CPPv4I0Eml5array1TRK5array", false], [0, "_CPPv4I0Eml5arrayRK5array1T", false], [0, "_CPPv4mlRK5arrayRK5array", false]], "operator+ (c++ function)": [[0, "_CPPv4I0Epl5array1TRK5array", false], [0, "_CPPv4I0Epl5arrayRK5array1T", false], [0, "_CPPv4plRK5arrayRK5array", false]], "operator- (c++ function)": [[0, "_CPPv4I0Emi5array1TRK5array", false], [0, "_CPPv4I0Emi5arrayRK5array1T", false], [0, "_CPPv4miRK5array", false], [0, "_CPPv4miRK5arrayRK5array", false]], "operator/ (c++ function)": [[0, "_CPPv4dvRK5arrayRK5array", false], [0, "_CPPv4dvRK5arrayd", false], [0, "_CPPv4dvdRK5array", false]], "operator< (c++ function)": [[0, "_CPPv4I0Elt5array1TRK5array", false], [0, "_CPPv4I0Elt5arrayRK5array1T", false], [0, "_CPPv4ltRK5arrayRK5array", false]], "operator<< (c++ function)": [[0, "_CPPv4lsRK5arrayRK5array", false]], "operator<= (c++ function)": [[0, "_CPPv4I0Ele5array1TRK5array", false], [0, "_CPPv4I0Ele5arrayRK5array1T", false], [0, "_CPPv4leRK5arrayRK5array", false]], "operator== (c++ function)": [[0, "_CPPv4I0Eeq5array1TRK5array", false], [0, "_CPPv4I0Eeq5arrayRK5array1T", false], [0, "_CPPv4eqRK5arrayRK5array", false]], "operator> (c++ function)": [[0, "_CPPv4I0Egt5array1TRK5array", false], [0, "_CPPv4I0Egt5arrayRK5array1T", false], [0, "_CPPv4gtRK5arrayRK5array", false]], "operator>= (c++ function)": [[0, "_CPPv4I0Ege5array1TRK5array", false], [0, "_CPPv4I0Ege5arrayRK5array1T", false], [0, "_CPPv4geRK5arrayRK5array", false]], "operator>> (c++ function)": [[0, "_CPPv4rsRK5arrayRK5array", false]], "operator^ (c++ function)": [[0, "_CPPv4eoRK5arrayRK5array", false]], "operator| (c++ function)": [[0, "_CPPv4orRK5arrayRK5array", false]], "operator|| (c++ function)": [[0, "_CPPv4ooRK5arrayRK5array", false]], "optimizer (class in mlx.optimizers)": [[486, "mlx.optimizers.Optimizer", false]], "outer (c++ function)": [[0, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", false]], "outer() (in module mlx.core)": [[237, "mlx.core.outer", false]], "pad (c++ function)": [[0, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", false]], "pad() (in module mlx.core)": [[238, "mlx.core.pad", false]], "parameters() (module method)": [[380, "mlx.nn.Module.parameters", false]], "partition (c++ function)": [[0, "_CPPv49partitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv49partitionRK5arrayii14StreamOrDevice", false]], "partition() (in module mlx.core)": [[239, "mlx.core.partition", false]], "permutation() (in module mlx.core.random)": [[253, "mlx.core.random.permutation", false]], "power (c++ function)": [[0, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", false]], "power() (in module mlx.core)": [[240, "mlx.core.power", false]], "prelu (class in mlx.nn)": [[391, "mlx.nn.PReLU", false], [448, "mlx.nn.prelu", false]], "prod (c++ function)": [[0, "_CPPv44prodRK5array14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayib14StreamOrDevice", false]], "prod() (array method)": [[64, "mlx.core.array.prod", false]], "prod() (in module mlx.core)": [[241, "mlx.core.prod", false]], "put_along_axis (c++ function)": [[0, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false]], "put_along_axis() (in module mlx.core)": [[242, "mlx.core.put_along_axis", false]], "qr() (in module mlx.core.linalg)": [[196, "mlx.core.linalg.qr", false]], "quantize (c++ function)": [[0, "_CPPv48quantizeRK5arrayii14StreamOrDevice", false]], "quantize() (in module mlx.core)": [[243, "mlx.core.quantize", false]], "quantize() (in module mlx.nn)": [[316, "mlx.nn.quantize", false]], "quantized_matmul (c++ function)": [[0, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", false]], "quantized_matmul() (in module mlx.core)": [[244, "mlx.core.quantized_matmul", false]], "quantizedembedding (class in mlx.nn)": [[392, "mlx.nn.QuantizedEmbedding", false]], "quantizedlinear (class in mlx.nn)": [[393, "mlx.nn.QuantizedLinear", false]], "radians (c++ function)": [[0, "_CPPv47radiansRK5array14StreamOrDevice", false]], "radians() (in module mlx.core)": [[245, "mlx.core.radians", false]], "randint() (in module mlx.core.random)": [[254, "mlx.core.random.randint", false]], "real (c++ function)": [[0, "_CPPv44realRK5array14StreamOrDevice", false]], "real() (in module mlx.core)": [[259, "mlx.core.real", false]], "reciprocal (c++ function)": [[0, "_CPPv410reciprocalRK5array14StreamOrDevice", false]], "reciprocal() (array method)": [[65, "mlx.core.array.reciprocal", false]], "reciprocal() (in module mlx.core)": [[260, "mlx.core.reciprocal", false]], "recv() (in module mlx.core.distributed)": [[126, "mlx.core.distributed.recv", false]], "recv_like() (in module mlx.core.distributed)": [[127, "mlx.core.distributed.recv_like", false]], "relu (class in mlx.nn)": [[396, "mlx.nn.ReLU", false], [449, "mlx.nn.relu", false]], "relu6 (class in mlx.nn)": [[397, "mlx.nn.ReLU6", false], [450, "mlx.nn.relu6", false]], "remainder (c++ function)": [[0, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", false]], "remainder() (in module mlx.core)": [[261, "mlx.core.remainder", false]], "repeat (c++ function)": [[0, "_CPPv46repeatRK5arrayi14StreamOrDevice", false], [0, "_CPPv46repeatRK5arrayii14StreamOrDevice", false]], "repeat() (in module mlx.core)": [[262, "mlx.core.repeat", false]], "reset_peak_memory() (in module mlx.core.metal)": [[221, "mlx.core.metal.reset_peak_memory", false]], "reshape (c++ function)": [[0, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", false]], "reshape() (array method)": [[66, "mlx.core.array.reshape", false]], "reshape() (in module mlx.core)": [[263, "mlx.core.reshape", false]], "rfft() (in module mlx.core.fft)": [[159, "mlx.core.fft.rfft", false]], "rfft2() (in module mlx.core.fft)": [[160, "mlx.core.fft.rfft2", false]], "rfftn() (in module mlx.core.fft)": [[161, "mlx.core.fft.rfftn", false]], "right_shift (c++ function)": [[0, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", false]], "right_shift() (in module mlx.core)": [[264, "mlx.core.right_shift", false]], "rms_norm() (in module mlx.core.fast)": [[147, "mlx.core.fast.rms_norm", false]], "rmsnorm (class in mlx.nn)": [[394, "mlx.nn.RMSNorm", false]], "rmsprop (class in mlx.optimizers)": [[478, "mlx.optimizers.RMSprop", false]], "rnn (class in mlx.nn)": [[395, "mlx.nn.RNN", false]], "roll (c++ function)": [[0, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayi14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayii14StreamOrDevice", false]], "roll() (in module mlx.core)": [[265, "mlx.core.roll", false]], "rope (class in mlx.nn)": [[398, "mlx.nn.RoPE", false]], "rope() (in module mlx.core.fast)": [[148, "mlx.core.fast.rope", false]], "round (c++ function)": [[0, "_CPPv45roundRK5array14StreamOrDevice", false], [0, "_CPPv45roundRK5arrayi14StreamOrDevice", false]], "round() (array method)": [[67, "mlx.core.array.round", false]], "round() (in module mlx.core)": [[266, "mlx.core.round", false]], "rsqrt (c++ function)": [[0, "_CPPv45rsqrtRK5array14StreamOrDevice", false]], "rsqrt() (array method)": [[68, "mlx.core.array.rsqrt", false]], "rsqrt() (in module mlx.core)": [[267, "mlx.core.rsqrt", false]], "save() (in module mlx.core)": [[268, "mlx.core.save", false]], "save_gguf() (in module mlx.core)": [[269, "mlx.core.save_gguf", false]], "save_safetensors() (in module mlx.core)": [[270, "mlx.core.save_safetensors", false]], "save_weights() (module method)": [[381, "mlx.nn.Module.save_weights", false]], "savez() (in module mlx.core)": [[271, "mlx.core.savez", false]], "savez_compressed() (in module mlx.core)": [[272, "mlx.core.savez_compressed", false]], "scaled_dot_product_attention() (in module mlx.core.fast)": [[149, "mlx.core.fast.scaled_dot_product_attention", false]], "scatter (c++ function)": [[0, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_add (c++ function)": [[0, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_add_axis (c++ function)": [[0, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false]], "scatter_max (c++ function)": [[0, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_min (c++ function)": [[0, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_prod (c++ function)": [[0, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "seed() (in module mlx.core.random)": [[255, "mlx.core.random.seed", false]], "selu (class in mlx.nn)": [[399, "mlx.nn.SELU", false], [451, "mlx.nn.selu", false]], "send() (in module mlx.core.distributed)": [[128, "mlx.core.distributed.send", false]], "sequential (class in mlx.nn)": [[400, "mlx.nn.Sequential", false]], "set_cache_limit() (in module mlx.core.metal)": [[222, "mlx.core.metal.set_cache_limit", false]], "set_default_device() (in module mlx.core)": [[273, "mlx.core.set_default_device", false]], "set_default_stream() (in module mlx.core)": [[274, "mlx.core.set_default_stream", false]], "set_dtype() (module method)": [[382, "mlx.nn.Module.set_dtype", false]], "set_memory_limit() (in module mlx.core.metal)": [[223, "mlx.core.metal.set_memory_limit", false]], "set_wired_limit() (in module mlx.core.metal)": [[224, "mlx.core.metal.set_wired_limit", false]], "sgd (class in mlx.optimizers)": [[479, "mlx.optimizers.SGD", false]], "shape (array property)": [[69, "mlx.core.array.shape", false]], "sigmoid (c++ function)": [[0, "_CPPv47sigmoidRK5array14StreamOrDevice", false]], "sigmoid (class in mlx.nn)": [[402, "mlx.nn.Sigmoid", false], [452, "mlx.nn.sigmoid", false]], "sigmoid() (in module mlx.core)": [[275, "mlx.core.sigmoid", false]], "sign (c++ function)": [[0, "_CPPv44signRK5array14StreamOrDevice", false]], "sign() (in module mlx.core)": [[276, "mlx.core.sign", false]], "silu (class in mlx.nn)": [[401, "mlx.nn.SiLU", false], [453, "mlx.nn.silu", false]], "sin (c++ function)": [[0, "_CPPv43sinRK5array14StreamOrDevice", false]], "sin() (array method)": [[70, "mlx.core.array.sin", false]], "sin() (in module mlx.core)": [[277, "mlx.core.sin", false]], "sinh (c++ function)": [[0, "_CPPv44sinhRK5array14StreamOrDevice", false]], "sinh() (in module mlx.core)": [[278, "mlx.core.sinh", false]], "sinusoidalpositionalencoding (class in mlx.nn)": [[403, "mlx.nn.SinusoidalPositionalEncoding", false]], "size (array property)": [[71, "mlx.core.array.size", false]], "slice (c++ function)": [[0, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", false]], "slice() (in module mlx.core)": [[279, "mlx.core.slice", false]], "slice_update (c++ function)": [[0, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", false], [0, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", false], [0, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "slice_update() (in module mlx.core)": [[280, "mlx.core.slice_update", false]], "smooth_l1_loss (class in mlx.nn.losses)": [[445, "mlx.nn.losses.smooth_l1_loss", false]], "softmax (c++ function)": [[0, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayib14StreamOrDevice", false]], "softmax (class in mlx.nn)": [[404, "mlx.nn.Softmax", false], [454, "mlx.nn.softmax", false]], "softmax() (in module mlx.core)": [[281, "mlx.core.softmax", false]], "softmin (class in mlx.nn)": [[405, "mlx.nn.Softmin", false], [455, "mlx.nn.softmin", false]], "softplus (class in mlx.nn)": [[406, "mlx.nn.Softplus", false], [456, "mlx.nn.softplus", false]], "softshrink (class in mlx.nn)": [[407, "mlx.nn.Softshrink", false], [457, "mlx.nn.softshrink", false]], "softsign (class in mlx.nn)": [[408, "mlx.nn.Softsign", false]], "sort (c++ function)": [[0, "_CPPv44sortRK5array14StreamOrDevice", false], [0, "_CPPv44sortRK5arrayi14StreamOrDevice", false]], "sort() (in module mlx.core)": [[282, "mlx.core.sort", false]], "split (c++ function)": [[0, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayi14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayii14StreamOrDevice", false]], "split() (array method)": [[72, "mlx.core.array.split", false]], "split() (in module mlx.core)": [[283, "mlx.core.split", false]], "split() (in module mlx.core.random)": [[256, "mlx.core.random.split", false]], "sqrt (c++ function)": [[0, "_CPPv44sqrtRK5array14StreamOrDevice", false]], "sqrt() (array method)": [[73, "mlx.core.array.sqrt", false]], "sqrt() (in module mlx.core)": [[284, "mlx.core.sqrt", false]], "square (c++ function)": [[0, "_CPPv46squareRK5array14StreamOrDevice", false]], "square() (array method)": [[74, "mlx.core.array.square", false]], "square() (in module mlx.core)": [[285, "mlx.core.square", false]], "squeeze (c++ function)": [[0, "_CPPv47squeezeRK5array14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayi14StreamOrDevice", false]], "squeeze() (array method)": [[75, "mlx.core.array.squeeze", false]], "squeeze() (in module mlx.core)": [[286, "mlx.core.squeeze", false]], "stack (c++ function)": [[0, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", false]], "stack() (in module mlx.core)": [[287, "mlx.core.stack", false]], "start_capture() (in module mlx.core.metal)": [[225, "mlx.core.metal.start_capture", false]], "state (module property)": [[383, "mlx.nn.Module.state", false]], "state (optimizer property)": [[476, "mlx.optimizers.Optimizer.state", false]], "std (c++ function)": [[0, "_CPPv4StRK5array14StreamOrDevice", false], [0, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv4StRK5arraybi14StreamOrDevice", false], [0, "_CPPv4StRK5arrayibi14StreamOrDevice", false]], "std() (array method)": [[76, "mlx.core.array.std", false]], "std() (in module mlx.core)": [[288, "mlx.core.std", false]], "step (class in mlx.nn)": [[409, "mlx.nn.Step", false], [458, "mlx.nn.step", false]], "step_decay() (in module mlx.optimizers)": [[484, "mlx.optimizers.step_decay", false]], "stop_capture() (in module mlx.core.metal)": [[226, "mlx.core.metal.stop_capture", false]], "stop_gradient (c++ function)": [[0, "_CPPv413stop_gradientRK5array14StreamOrDevice", false]], "stop_gradient() (in module mlx.core)": [[289, "mlx.core.stop_gradient", false]], "stream (class in mlx.core)": [[324, "mlx.core.Stream", false]], "stream() (in module mlx.core)": [[290, "mlx.core.stream", false]], "subtract (c++ function)": [[0, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", false]], "subtract() (in module mlx.core)": [[291, "mlx.core.subtract", false]], "sum (c++ function)": [[0, "_CPPv43sumRK5array14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayib14StreamOrDevice", false]], "sum() (array method)": [[77, "mlx.core.array.sum", false]], "sum() (in module mlx.core)": [[292, "mlx.core.sum", false]], "svd() (in module mlx.core.linalg)": [[197, "mlx.core.linalg.svd", false]], "swapaxes (c++ function)": [[0, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", false]], "swapaxes() (array method)": [[78, "mlx.core.array.swapaxes", false]], "swapaxes() (in module mlx.core)": [[293, "mlx.core.swapaxes", false]], "synchronize() (in module mlx.core)": [[294, "mlx.core.synchronize", false]], "t (array property)": [[32, "mlx.core.array.T", false]], "take (c++ function)": [[0, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayi14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayii14StreamOrDevice", false]], "take() (in module mlx.core)": [[295, "mlx.core.take", false]], "take_along_axis (c++ function)": [[0, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", false]], "take_along_axis() (in module mlx.core)": [[296, "mlx.core.take_along_axis", false]], "tan (c++ function)": [[0, "_CPPv43tanRK5array14StreamOrDevice", false]], "tan() (in module mlx.core)": [[297, "mlx.core.tan", false]], "tanh (c++ function)": [[0, "_CPPv44tanhRK5array14StreamOrDevice", false]], "tanh (class in mlx.nn)": [[410, "mlx.nn.Tanh", false], [459, "mlx.nn.tanh", false]], "tanh() (in module mlx.core)": [[298, "mlx.core.tanh", false]], "tensordot (c++ function)": [[0, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", false], [0, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", false]], "tensordot() (in module mlx.core)": [[299, "mlx.core.tensordot", false]], "tile (c++ function)": [[0, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "tile() (in module mlx.core)": [[300, "mlx.core.tile", false]], "tolist() (array method)": [[79, "mlx.core.array.tolist", false]], "topk (c++ function)": [[0, "_CPPv44topkRK5arrayi14StreamOrDevice", false], [0, "_CPPv44topkRK5arrayii14StreamOrDevice", false]], "topk() (in module mlx.core)": [[301, "mlx.core.topk", false]], "trace (c++ function)": [[0, "_CPPv45traceRK5array14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", false]], "trace() (in module mlx.core)": [[302, "mlx.core.trace", false]], "train() (module method)": [[384, "mlx.nn.Module.train", false]], "trainable_parameters() (module method)": [[385, "mlx.nn.Module.trainable_parameters", false]], "training (module property)": [[386, "mlx.nn.Module.training", false]], "transformer (class in mlx.nn)": [[411, "mlx.nn.Transformer", false]], "transpose (c++ function)": [[0, "_CPPv49transposeRK5array14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "transpose() (array method)": [[80, "mlx.core.array.transpose", false]], "transpose() (in module mlx.core)": [[303, "mlx.core.transpose", false]], "tree_flatten() (in module mlx.utils)": [[319, "mlx.utils.tree_flatten", false]], "tree_map() (in module mlx.utils)": [[320, "mlx.utils.tree_map", false]], "tree_map_with_path() (in module mlx.utils)": [[321, "mlx.utils.tree_map_with_path", false]], "tree_reduce() (in module mlx.utils)": [[322, "mlx.utils.tree_reduce", false]], "tree_unflatten() (in module mlx.utils)": [[323, "mlx.utils.tree_unflatten", false]], "tri (c++ function)": [[0, "_CPPv43trii5Dtype14StreamOrDevice", false], [0, "_CPPv43triiii5Dtype14StreamOrDevice", false]], "tri() (in module mlx.core)": [[304, "mlx.core.tri", false]], "tri_inv() (in module mlx.core.linalg)": [[198, "mlx.core.linalg.tri_inv", false]], "tril (c++ function)": [[0, "_CPPv44tril5arrayi14StreamOrDevice", false]], "tril() (in module mlx.core)": [[305, "mlx.core.tril", false]], "triplet_loss (class in mlx.nn.losses)": [[446, "mlx.nn.losses.triplet_loss", false]], "triu (c++ function)": [[0, "_CPPv44triu5arrayi14StreamOrDevice", false]], "triu() (in module mlx.core)": [[306, "mlx.core.triu", false]], "truncated_normal() (in module mlx.core.random)": [[257, "mlx.core.random.truncated_normal", false]], "unflatten (c++ function)": [[0, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", false]], "unflatten() (in module mlx.core)": [[307, "mlx.core.unflatten", false]], "unfreeze() (module method)": [[387, "mlx.nn.Module.unfreeze", false]], "uniform() (in module mlx.core.random)": [[258, "mlx.core.random.uniform", false]], "uniform() (in module mlx.nn.init)": [[420, "mlx.nn.init.uniform", false]], "update() (module method)": [[388, "mlx.nn.Module.update", false]], "update() (optimizer method)": [[477, "mlx.optimizers.Optimizer.update", false]], "update_modules() (module method)": [[389, "mlx.nn.Module.update_modules", false]], "upsample (class in mlx.nn)": [[412, "mlx.nn.Upsample", false]], "value_and_grad() (in module mlx.core)": [[308, "mlx.core.value_and_grad", false]], "value_and_grad() (in module mlx.nn)": [[317, "mlx.nn.value_and_grad", false]], "var (c++ function)": [[0, "_CPPv43varRK5array14StreamOrDevice", false], [0, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv43varRK5arraybi14StreamOrDevice", false], [0, "_CPPv43varRK5arrayibi14StreamOrDevice", false]], "var() (array method)": [[81, "mlx.core.array.var", false]], "var() (in module mlx.core)": [[309, "mlx.core.var", false]], "view (c++ function)": [[0, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", false]], "view() (array method)": [[82, "mlx.core.array.view", false]], "view() (in module mlx.core)": [[310, "mlx.core.view", false]], "vjp() (in module mlx.core)": [[311, "mlx.core.vjp", false]], "vmap() (in module mlx.core)": [[312, "mlx.core.vmap", false]], "where (c++ function)": [[0, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", false]], "where() (in module mlx.core)": [[313, "mlx.core.where", false]], "zeros (c++ function)": [[0, "_CPPv45zerosRK5Shape14StreamOrDevice", false], [0, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", false]], "zeros() (in module mlx.core)": [[314, "mlx.core.zeros", false]], "zeros_like (c++ function)": [[0, "_CPPv410zeros_likeRK5array14StreamOrDevice", false]], "zeros_like() (in module mlx.core)": [[315, "mlx.core.zeros_like", false]]}, "objects": {"": [[0, 0, 1, "_CPPv43absRK5array14StreamOrDevice", "abs"], [0, 1, 1, "_CPPv43absRK5array14StreamOrDevice", "abs::a"], [0, 1, 1, "_CPPv43absRK5array14StreamOrDevice", "abs::s"], [0, 0, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::a"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::b"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::s"], [0, 0, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::a"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::alpha"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::b"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::beta"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::c"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::s"], [0, 0, 1, "_CPPv43allRK5array14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all"], [0, 1, 1, "_CPPv43allRK5array14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::axes"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::axis"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5array14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::s"], [0, 0, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::a"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::atol"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::b"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::equal_nan"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::rtol"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::s"], [0, 0, 1, "_CPPv43anyRK5array14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any"], [0, 1, 1, "_CPPv43anyRK5array14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::axes"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::axis"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5array14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::s"], [0, 0, 1, "_CPPv46aranged14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangedd14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeddd14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangei14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeii14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeiii14StreamOrDevice", "arange"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46aranged14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangei14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46aranged14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangei14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::stop"], [0, 0, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos"], [0, 1, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos::a"], [0, 1, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos::s"], [0, 0, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh"], [0, 1, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh::a"], [0, 1, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh::s"], [0, 0, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin"], [0, 1, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin::a"], [0, 1, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin::s"], [0, 0, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh"], [0, 1, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh::a"], [0, 1, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh::s"], [0, 0, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan"], [0, 0, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::a"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::b"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::s"], [0, 1, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan::a"], [0, 1, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan::s"], [0, 0, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh"], [0, 1, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh::a"], [0, 1, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh::s"], [0, 0, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax"], [0, 0, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax"], [0, 0, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax"], [0, 1, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::axis"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::keepdims"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::keepdims"], [0, 1, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax::s"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::s"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::s"], [0, 0, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin"], [0, 0, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin"], [0, 0, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin"], [0, 1, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::axis"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::keepdims"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::keepdims"], [0, 1, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin::s"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::s"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::s"], [0, 0, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition"], [0, 0, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::a"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::a"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::axis"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::kth"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::kth"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::s"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::s"], [0, 0, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort"], [0, 0, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort"], [0, 1, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort::a"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::a"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::axis"], [0, 1, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort::s"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::s"], [0, 0, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal"], [0, 0, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::a"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::a"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::b"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::b"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::equal_nan"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::s"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::s"], [0, 0, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::a"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::offset"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::s"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::shape"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::strides"], [0, 0, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::a"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::dtype"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::s"], [0, 0, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d"], [0, 0, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d"], [0, 1, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d::a"], [0, 1, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d::a"], [0, 1, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d::s"], [0, 1, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d::s"], [0, 0, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d"], [0, 0, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d"], [0, 1, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d::a"], [0, 1, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d::a"], [0, 1, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d::s"], [0, 1, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d::s"], [0, 0, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d"], [0, 0, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d"], [0, 1, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d::a"], [0, 1, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d::a"], [0, 1, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d::s"], [0, 1, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d::s"], [0, 0, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::a"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::b"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::s"], [0, 0, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::a"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::b"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::s"], [0, 0, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::a"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::b"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::s"], [0, 0, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::a"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::b"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::block_size"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_lhs"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_out"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_rhs"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::s"], [0, 0, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays"], [0, 1, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays::inputs"], [0, 1, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays::s"], [0, 0, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::a"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::s"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::shape"], [0, 0, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil"], [0, 1, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil::a"], [0, 1, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil::s"], [0, 0, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a_max"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a_min"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::s"], [0, 0, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate"], [0, 0, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::axis"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::s"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::s"], [0, 0, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::a"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::s"], [0, 0, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::a"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::allow_col_major"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::s"], [0, 0, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::dilation"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::groups"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::input"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::padding"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::s"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::stride"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::weight"], [0, 0, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::dilation"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::groups"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::input"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::padding"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::s"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::stride"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::weight"], [0, 0, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::dilation"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::groups"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::input"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::padding"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::s"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::stride"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::weight"], [0, 0, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general"], [0, 0, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::flip"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::flip"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::groups"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::groups"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input_dilation"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::kernel_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::kernel_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding_hi"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding_lo"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::s"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::s"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::stride"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::stride"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::weight"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::weight"], [0, 0, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::dilation"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::groups"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::input"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::padding"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::s"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::stride"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::weight"], [0, 0, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::dilation"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::groups"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::input"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::padding"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::s"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::stride"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::weight"], [0, 0, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::dilation"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::groups"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::input"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::padding"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::s"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::stride"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::weight"], [0, 0, 1, "_CPPv44copy5array14StreamOrDevice", "copy"], [0, 1, 1, "_CPPv44copy5array14StreamOrDevice", "copy::a"], [0, 1, 1, "_CPPv44copy5array14StreamOrDevice", "copy::s"], [0, 0, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos"], [0, 1, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos::a"], [0, 1, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos::s"], [0, 0, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh"], [0, 1, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh::a"], [0, 1, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh::s"], [0, 0, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::a"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::axis"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::inclusive"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::reverse"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::s"], [0, 0, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::a"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::axis"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::inclusive"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::reverse"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::s"], [0, 0, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::a"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::axis"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::inclusive"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::reverse"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::s"], [0, 0, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::a"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::axis"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::inclusive"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::reverse"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::s"], [0, 0, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees"], [0, 1, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees::a"], [0, 1, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees::s"], [0, 0, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends"], [0, 1, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends::dependencies"], [0, 1, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends::inputs"], [0, 0, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::biases"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::bits"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::group_size"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::s"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::scales"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::w"], [0, 0, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::a"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::k"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::s"], [0, 0, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::a"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::axis1"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::axis2"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::offset"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::s"], [0, 0, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::a"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::b"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::s"], [0, 0, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::a"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::b"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::s"], [0, 0, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::a"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::b"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::s"], [0, 0, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf"], [0, 1, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf::a"], [0, 1, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf::s"], [0, 0, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv"], [0, 1, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv::a"], [0, 1, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv::s"], [0, 0, 1, "_CPPv43expRK5array14StreamOrDevice", "exp"], [0, 1, 1, "_CPPv43expRK5array14StreamOrDevice", "exp::a"], [0, 1, 1, "_CPPv43expRK5array14StreamOrDevice", "exp::s"], [0, 0, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims"], [0, 0, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::a"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::a"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::axes"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::axis"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::s"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::s"], [0, 0, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1"], [0, 1, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1::a"], [0, 1, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1::s"], [0, 0, 1, "_CPPv43eyei14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeii14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeiii14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::dtype"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::dtype"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::k"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::k"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyei14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyei14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::s"], [0, 0, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten"], [0, 0, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten"], [0, 1, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten::a"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::a"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::end_axis"], [0, 1, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten::s"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::s"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::start_axis"], [0, 0, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor"], [0, 1, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor::a"], [0, 1, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor::s"], [0, 0, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::a"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::b"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::s"], [0, 0, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full"], [0, 0, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full"], [0, 2, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::T"], [0, 2, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::T"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::vals"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::vals"], [0, 0, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather"], [0, 0, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::axes"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::axis"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::slice_sizes"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::slice_sizes"], [0, 0, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::a"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::b"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::lhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::rhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::s"], [0, 0, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::biases"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::bits"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::group_size"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::lhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::rhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::s"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::scales"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::transpose"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::w"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::x"], [0, 0, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::a"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::b"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::s"], [0, 0, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::a"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::b"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::s"], [0, 0, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::a"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::s"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::scale"], [0, 0, 1, "_CPPv48identityi14StreamOrDevice", "identity"], [0, 0, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::dtype"], [0, 1, 1, "_CPPv48identityi14StreamOrDevice", "identity::n"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::n"], [0, 1, 1, "_CPPv48identityi14StreamOrDevice", "identity::s"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::s"], [0, 0, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag"], [0, 1, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag::a"], [0, 1, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag::s"], [0, 0, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::a"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::b"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::s"], [0, 0, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::a"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::atol"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::b"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::equal_nan"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::rtol"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::s"], [0, 0, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite"], [0, 1, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite::a"], [0, 1, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite::s"], [0, 0, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf"], [0, 1, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf::a"], [0, 1, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf::s"], [0, 0, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan"], [0, 1, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan::a"], [0, 1, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan::s"], [0, 0, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf"], [0, 1, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf::a"], [0, 1, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf::s"], [0, 0, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf"], [0, 1, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf::a"], [0, 1, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf::s"], [0, 0, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::a"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::b"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::s"], [0, 0, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::a"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::b"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::s"], [0, 0, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::a"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::b"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::s"], [0, 0, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::a"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::b"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::s"], [0, 0, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::dtype"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::num"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::s"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::start"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::stop"], [0, 0, 1, "_CPPv43logRK5array14StreamOrDevice", "log"], [0, 0, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10"], [0, 1, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10::a"], [0, 1, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10::s"], [0, 0, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p"], [0, 1, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p::a"], [0, 1, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p::s"], [0, 0, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2"], [0, 1, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2::a"], [0, 1, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2::s"], [0, 1, 1, "_CPPv43logRK5array14StreamOrDevice", "log::a"], [0, 1, 1, "_CPPv43logRK5array14StreamOrDevice", "log::s"], [0, 0, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::a"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::b"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::s"], [0, 0, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::a"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::b"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::s"], [0, 0, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not"], [0, 1, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not::a"], [0, 1, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not::s"], [0, 0, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::a"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::b"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::s"], [0, 0, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp"], [0, 1, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::axes"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::axis"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::s"], [0, 0, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::a"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::b"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::s"], [0, 0, 1, "_CPPv43maxRK5array14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max"], [0, 1, 1, "_CPPv43maxRK5array14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::axes"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::axis"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5array14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::s"], [0, 0, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::a"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::b"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::s"], [0, 0, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean"], [0, 1, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::axes"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::axis"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::s"], [0, 0, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::arrays"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::indexing"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::s"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::sparse"], [0, 0, 1, "_CPPv43minRK5array14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min"], [0, 1, 1, "_CPPv43minRK5array14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::axes"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::axis"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5array14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::s"], [0, 0, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::a"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::b"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::s"], [0, 0, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::a"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::destination"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::s"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::source"], [0, 0, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::a"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::b"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::s"], [0, 0, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::a"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::nan"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::neginf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::posinf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::s"], [0, 0, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative"], [0, 1, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative::a"], [0, 1, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative::s"], [0, 0, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::a"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::b"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::s"], [0, 0, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::a"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::axes"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::dtype"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::inverted"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::s"], [0, 0, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones"], [0, 0, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::dtype"], [0, 1, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones::shape"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::shape"], [0, 0, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like"], [0, 1, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like::a"], [0, 1, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like::s"], [0, 0, 1, "_CPPv4I0Ene5array1TRK5array", "operator!="], [0, 0, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!="], [0, 0, 1, "_CPPv4neRK5arrayRK5array", "operator!="], [0, 2, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::T"], [0, 2, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::T"], [0, 1, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::a"], [0, 1, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::a"], [0, 1, 1, "_CPPv4neRK5arrayRK5array", "operator!=::a"], [0, 1, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::b"], [0, 1, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::b"], [0, 1, 1, "_CPPv4neRK5arrayRK5array", "operator!=::b"], [0, 0, 1, "_CPPv4I0Erm5array1TRK5array", "operator%"], [0, 0, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%"], [0, 0, 1, "_CPPv4rmRK5arrayRK5array", "operator%"], [0, 2, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::T"], [0, 2, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::T"], [0, 1, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::a"], [0, 1, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::a"], [0, 1, 1, "_CPPv4rmRK5arrayRK5array", "operator%::a"], [0, 1, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::b"], [0, 1, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::b"], [0, 1, 1, "_CPPv4rmRK5arrayRK5array", "operator%::b"], [0, 0, 1, "_CPPv4anRK5arrayRK5array", "operator&"], [0, 0, 1, "_CPPv4aaRK5arrayRK5array", "operator&&"], [0, 1, 1, "_CPPv4aaRK5arrayRK5array", "operator&&::a"], [0, 1, 1, "_CPPv4aaRK5arrayRK5array", "operator&&::b"], [0, 1, 1, "_CPPv4anRK5arrayRK5array", "operator&::a"], [0, 1, 1, "_CPPv4anRK5arrayRK5array", "operator&::b"], [0, 0, 1, "_CPPv4I0Eml5array1TRK5array", "operator*"], [0, 0, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*"], [0, 0, 1, "_CPPv4mlRK5arrayRK5array", "operator*"], [0, 2, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::T"], [0, 2, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::T"], [0, 1, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::a"], [0, 1, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::a"], [0, 1, 1, "_CPPv4mlRK5arrayRK5array", "operator*::a"], [0, 1, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::b"], [0, 1, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::b"], [0, 1, 1, "_CPPv4mlRK5arrayRK5array", "operator*::b"], [0, 0, 1, "_CPPv4I0Epl5array1TRK5array", "operator+"], [0, 0, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+"], [0, 0, 1, "_CPPv4plRK5arrayRK5array", "operator+"], [0, 2, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::T"], [0, 2, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::T"], [0, 1, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::a"], [0, 1, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::a"], [0, 1, 1, "_CPPv4plRK5arrayRK5array", "operator+::a"], [0, 1, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::b"], [0, 1, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::b"], [0, 1, 1, "_CPPv4plRK5arrayRK5array", "operator+::b"], [0, 0, 1, "_CPPv4I0Emi5array1TRK5array", "operator-"], [0, 0, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-"], [0, 0, 1, "_CPPv4miRK5array", "operator-"], [0, 0, 1, "_CPPv4miRK5arrayRK5array", "operator-"], [0, 2, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::T"], [0, 2, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::T"], [0, 1, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::a"], [0, 1, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::a"], [0, 1, 1, "_CPPv4miRK5array", "operator-::a"], [0, 1, 1, "_CPPv4miRK5arrayRK5array", "operator-::a"], [0, 1, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::b"], [0, 1, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::b"], [0, 1, 1, "_CPPv4miRK5arrayRK5array", "operator-::b"], [0, 0, 1, "_CPPv4dvRK5arrayRK5array", "operator/"], [0, 0, 1, "_CPPv4dvRK5arrayd", "operator/"], [0, 0, 1, "_CPPv4dvdRK5array", "operator/"], [0, 1, 1, "_CPPv4dvRK5arrayRK5array", "operator/::a"], [0, 1, 1, "_CPPv4dvRK5arrayd", "operator/::a"], [0, 1, 1, "_CPPv4dvdRK5array", "operator/::a"], [0, 1, 1, "_CPPv4dvRK5arrayRK5array", "operator/::b"], [0, 1, 1, "_CPPv4dvRK5arrayd", "operator/::b"], [0, 1, 1, "_CPPv4dvdRK5array", "operator/::b"], [0, 0, 1, "_CPPv4I0Elt5array1TRK5array", "operator<"], [0, 0, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<"], [0, 0, 1, "_CPPv4ltRK5arrayRK5array", "operator<"], [0, 2, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::T"], [0, 2, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::T"], [0, 1, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::a"], [0, 1, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::a"], [0, 1, 1, "_CPPv4ltRK5arrayRK5array", "operator<::a"], [0, 1, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::b"], [0, 1, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::b"], [0, 1, 1, "_CPPv4ltRK5arrayRK5array", "operator<::b"], [0, 0, 1, "_CPPv4lsRK5arrayRK5array", "operator<<"], [0, 1, 1, "_CPPv4lsRK5arrayRK5array", "operator<<::a"], [0, 1, 1, "_CPPv4lsRK5arrayRK5array", "operator<<::b"], [0, 0, 1, "_CPPv4I0Ele5array1TRK5array", "operator<="], [0, 0, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<="], [0, 0, 1, "_CPPv4leRK5arrayRK5array", "operator<="], [0, 2, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::T"], [0, 2, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::T"], [0, 1, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::a"], [0, 1, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::a"], [0, 1, 1, "_CPPv4leRK5arrayRK5array", "operator<=::a"], [0, 1, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::b"], [0, 1, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::b"], [0, 1, 1, "_CPPv4leRK5arrayRK5array", "operator<=::b"], [0, 0, 1, "_CPPv4I0Eeq5array1TRK5array", "operator=="], [0, 0, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator=="], [0, 0, 1, "_CPPv4eqRK5arrayRK5array", "operator=="], [0, 2, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::T"], [0, 2, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::T"], [0, 1, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::a"], [0, 1, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::a"], [0, 1, 1, "_CPPv4eqRK5arrayRK5array", "operator==::a"], [0, 1, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::b"], [0, 1, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::b"], [0, 1, 1, "_CPPv4eqRK5arrayRK5array", "operator==::b"], [0, 0, 1, "_CPPv4I0Egt5array1TRK5array", "operator>"], [0, 0, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>"], [0, 0, 1, "_CPPv4gtRK5arrayRK5array", "operator>"], [0, 2, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::T"], [0, 2, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::T"], [0, 1, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::a"], [0, 1, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::a"], [0, 1, 1, "_CPPv4gtRK5arrayRK5array", "operator>::a"], [0, 1, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::b"], [0, 1, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::b"], [0, 1, 1, "_CPPv4gtRK5arrayRK5array", "operator>::b"], [0, 0, 1, "_CPPv4I0Ege5array1TRK5array", "operator>="], [0, 0, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>="], [0, 0, 1, "_CPPv4geRK5arrayRK5array", "operator>="], [0, 2, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::T"], [0, 2, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::T"], [0, 1, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::a"], [0, 1, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::a"], [0, 1, 1, "_CPPv4geRK5arrayRK5array", "operator>=::a"], [0, 1, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::b"], [0, 1, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::b"], [0, 1, 1, "_CPPv4geRK5arrayRK5array", "operator>=::b"], [0, 0, 1, "_CPPv4rsRK5arrayRK5array", "operator>>"], [0, 1, 1, "_CPPv4rsRK5arrayRK5array", "operator>>::a"], [0, 1, 1, "_CPPv4rsRK5arrayRK5array", "operator>>::b"], [0, 0, 1, "_CPPv4eoRK5arrayRK5array", "operator^"], [0, 1, 1, "_CPPv4eoRK5arrayRK5array", "operator^::a"], [0, 1, 1, "_CPPv4eoRK5arrayRK5array", "operator^::b"], [0, 0, 1, "_CPPv4orRK5arrayRK5array", "operator|"], [0, 1, 1, "_CPPv4orRK5arrayRK5array", "operator|::a"], [0, 1, 1, "_CPPv4orRK5arrayRK5array", "operator|::b"], [0, 0, 1, "_CPPv4ooRK5arrayRK5array", "operator||"], [0, 1, 1, "_CPPv4ooRK5arrayRK5array", "operator||::a"], [0, 1, 1, "_CPPv4ooRK5arrayRK5array", "operator||::b"], [0, 0, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::a"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::b"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::s"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::axes"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::high_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::low_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 0, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition"], [0, 0, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::a"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::a"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::axis"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::kth"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::kth"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::s"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::s"], [0, 0, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::a"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::b"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::s"], [0, 0, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod"], [0, 1, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::axes"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::axis"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::s"], [0, 0, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::a"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::axis"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::indices"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::s"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::values"], [0, 0, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::bits"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::group_size"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::s"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::w"], [0, 0, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::biases"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::bits"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::group_size"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::s"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::scales"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::transpose"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::w"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::x"], [0, 0, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::a"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::s"], [0, 0, 1, "_CPPv44realRK5array14StreamOrDevice", "real"], [0, 1, 1, "_CPPv44realRK5array14StreamOrDevice", "real::a"], [0, 1, 1, "_CPPv44realRK5array14StreamOrDevice", "real::s"], [0, 0, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal"], [0, 1, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal::a"], [0, 1, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal::s"], [0, 0, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::a"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::b"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::s"], [0, 0, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat"], [0, 0, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::arr"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::arr"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::axis"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::repeats"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::repeats"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::s"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::s"], [0, 0, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::a"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::s"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::shape"], [0, 0, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::a"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::b"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::s"], [0, 0, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::axes"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::axes"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::axis"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::axis"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::shift"], [0, 0, 1, "_CPPv45roundRK5array14StreamOrDevice", "round"], [0, 0, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round"], [0, 1, 1, "_CPPv45roundRK5array14StreamOrDevice", "round::a"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::a"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::decimals"], [0, 1, 1, "_CPPv45roundRK5array14StreamOrDevice", "round::s"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::s"], [0, 0, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt"], [0, 1, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt::a"], [0, 1, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt::s"], [0, 0, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter"], [0, 0, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::a"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::a"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::axes"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::axis"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::indices"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::indices"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::s"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::s"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::updates"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::updates"], [0, 0, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add"], [0, 0, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::a"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::a"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::axes"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::axis"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::indices"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::indices"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::s"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::s"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::updates"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::updates"], [0, 0, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::a"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::axis"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::indices"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::s"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::values"], [0, 0, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max"], [0, 0, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::a"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::a"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::axes"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::axis"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::indices"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::indices"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::s"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::s"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::updates"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::updates"], [0, 0, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min"], [0, 0, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::a"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::a"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::axes"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::axis"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::indices"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::indices"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::s"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::s"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::updates"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::updates"], [0, 0, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod"], [0, 0, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::a"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::a"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::axes"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::axis"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::indices"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::indices"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::s"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::s"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::updates"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::updates"], [0, 0, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid"], [0, 1, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid::a"], [0, 1, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid::s"], [0, 0, 1, "_CPPv44signRK5array14StreamOrDevice", "sign"], [0, 1, 1, "_CPPv44signRK5array14StreamOrDevice", "sign::a"], [0, 1, 1, "_CPPv44signRK5array14StreamOrDevice", "sign::s"], [0, 0, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin"], [0, 1, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin::a"], [0, 1, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin::s"], [0, 0, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh"], [0, 1, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh::a"], [0, 1, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh::s"], [0, 0, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::axes"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::slice_size"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::strides"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::strides"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::axes"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::strides"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::update"], [0, 0, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax"], [0, 0, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax"], [0, 0, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::axes"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::axis"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::s"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::s"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::s"], [0, 0, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort"], [0, 0, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort"], [0, 1, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort::a"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::a"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::axis"], [0, 1, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort::s"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::s"], [0, 0, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::s"], [0, 0, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt"], [0, 1, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt::a"], [0, 1, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt::s"], [0, 0, 1, "_CPPv46squareRK5array14StreamOrDevice", "square"], [0, 1, 1, "_CPPv46squareRK5array14StreamOrDevice", "square::a"], [0, 1, 1, "_CPPv46squareRK5array14StreamOrDevice", "square::s"], [0, 0, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze"], [0, 0, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze"], [0, 0, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze"], [0, 1, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::axes"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::axis"], [0, 1, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze::s"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::s"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::s"], [0, 0, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack"], [0, 0, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack::arrays"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::arrays"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::axis"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack::s"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::s"], [0, 0, 1, "_CPPv4StRK5array14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std"], [0, 1, 1, "_CPPv4StRK5array14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::axes"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::axis"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5array14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::s"], [0, 0, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient"], [0, 1, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient::a"], [0, 1, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient::s"], [0, 0, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::a"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::b"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::s"], [0, 0, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum"], [0, 1, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::axes"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::axis"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::s"], [0, 0, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::a"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::axis1"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::axis2"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::s"], [0, 0, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::index"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::index"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::indices"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::indices"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::s"], [0, 0, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::a"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::axis"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::indices"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::s"], [0, 0, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan"], [0, 1, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan::a"], [0, 1, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan::s"], [0, 0, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh"], [0, 1, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh::a"], [0, 1, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh::s"], [0, 0, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot"], [0, 0, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::axes_a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::axes_b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::axis"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::s"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::s"], [0, 0, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::arr"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::reps"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::s"], [0, 0, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk"], [0, 0, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::a"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::a"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::axis"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::k"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::k"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::s"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::s"], [0, 0, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace"], [0, 0, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace"], [0, 0, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace"], [0, 1, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::axis1"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::axis1"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::axis2"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::axis2"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::dtype"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::offset"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::offset"], [0, 1, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace::s"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::s"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::s"], [0, 0, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose"], [0, 0, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose"], [0, 0, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose"], [0, 1, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::axes"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::axes"], [0, 1, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose::s"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::s"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::s"], [0, 0, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri"], [0, 0, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::k"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::m"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::n"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::n"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::s"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::s"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::type"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::type"], [0, 0, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::k"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::s"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::x"], [0, 0, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::k"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::s"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::x"], [0, 0, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::a"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::axis"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::s"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::shape"], [0, 0, 1, "_CPPv43varRK5array14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var"], [0, 1, 1, "_CPPv43varRK5array14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::axes"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::axis"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5array14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::s"], [0, 0, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::a"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::dtype"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::s"], [0, 0, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::condition"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::s"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::x"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::y"], [0, 0, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros"], [0, 0, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::dtype"], [0, 1, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros::shape"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::shape"], [0, 0, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like"], [0, 1, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like::a"], [0, 1, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like::s"]], "mlx.core": [[10, 3, 1, "", "Device"], [11, 3, 1, "", "Dtype"], [12, 3, 1, "", "DtypeCategory"], [324, 3, 1, "", "Stream"], [13, 5, 1, "", "abs"], [14, 5, 1, "", "add"], [15, 5, 1, "", "addmm"], [16, 5, 1, "", "all"], [17, 5, 1, "", "allclose"], [18, 5, 1, "", "any"], [19, 5, 1, "", "arange"], [20, 5, 1, "", "arccos"], [21, 5, 1, "", "arccosh"], [22, 5, 1, "", "arcsin"], [23, 5, 1, "", "arcsinh"], [24, 5, 1, "", "arctan"], [25, 5, 1, "", "arctan2"], [26, 5, 1, "", "arctanh"], [27, 5, 1, "", "argmax"], [28, 5, 1, "", "argmin"], [29, 5, 1, "", "argpartition"], [30, 5, 1, "", "argsort"], [31, 3, 1, "", "array"], [83, 5, 1, "", "array_equal"], [84, 5, 1, "", "as_strided"], [85, 5, 1, "", "atleast_1d"], [86, 5, 1, "", "atleast_2d"], [87, 5, 1, "", "atleast_3d"], [88, 5, 1, "", "bitwise_and"], [89, 5, 1, "", "bitwise_or"], [90, 5, 1, "", "bitwise_xor"], [91, 5, 1, "", "block_masked_mm"], [92, 5, 1, "", "broadcast_to"], [93, 5, 1, "", "ceil"], [94, 5, 1, "", "clip"], [95, 5, 1, "", "compile"], [96, 5, 1, "", "concatenate"], [97, 5, 1, "", "conj"], [98, 5, 1, "", "conjugate"], [99, 5, 1, "", "conv1d"], [100, 5, 1, "", "conv2d"], [101, 5, 1, "", "conv3d"], [102, 5, 1, "", "conv_general"], [103, 5, 1, "", "conv_transpose1d"], [104, 5, 1, "", "conv_transpose2d"], [105, 5, 1, "", "conv_transpose3d"], [106, 5, 1, "", "convolve"], [107, 5, 1, "", "cos"], [108, 5, 1, "", "cosh"], [109, 5, 1, "", "cummax"], [110, 5, 1, "", "cummin"], [111, 5, 1, "", "cumprod"], [112, 5, 1, "", "cumsum"], [113, 3, 1, "", "custom_function"], [114, 5, 1, "", "default_device"], [115, 5, 1, "", "default_stream"], [116, 5, 1, "", "degrees"], [117, 5, 1, "", "dequantize"], [118, 5, 1, "", "diag"], [119, 5, 1, "", "diagonal"], [120, 5, 1, "", "disable_compile"], [129, 5, 1, "", "divide"], [130, 5, 1, "", "divmod"], [131, 5, 1, "", "einsum"], [132, 5, 1, "", "einsum_path"], [133, 5, 1, "", "enable_compile"], [134, 5, 1, "", "equal"], [135, 5, 1, "", "erf"], [136, 5, 1, "", "erfinv"], [137, 5, 1, "", "eval"], [138, 5, 1, "", "exp"], [139, 5, 1, "", "expand_dims"], [140, 5, 1, "", "expm1"], [141, 5, 1, "", "export_function"], [142, 5, 1, "", "export_to_dot"], [143, 5, 1, "", "exporter"], [144, 5, 1, "", "eye"], [162, 3, 1, "", "finfo"], [163, 5, 1, "", "flatten"], [164, 5, 1, "", "floor"], [165, 5, 1, "", "floor_divide"], [166, 5, 1, "", "full"], [167, 5, 1, "", "gather_mm"], [168, 5, 1, "", "gather_qmm"], [169, 5, 1, "", "grad"], [170, 5, 1, "", "greater"], [171, 5, 1, "", "greater_equal"], [172, 5, 1, "", "hadamard_transform"], [173, 5, 1, "", "identity"], [174, 5, 1, "", "imag"], [175, 5, 1, "", "import_function"], [176, 5, 1, "", "inner"], [177, 5, 1, "", "isclose"], [178, 5, 1, "", "isfinite"], [179, 5, 1, "", "isinf"], [180, 5, 1, "", "isnan"], [181, 5, 1, "", "isneginf"], [182, 5, 1, "", "isposinf"], [183, 5, 1, "", "issubdtype"], [184, 5, 1, "", "jvp"], [185, 5, 1, "", "kron"], [186, 5, 1, "", "left_shift"], [187, 5, 1, "", "less"], [188, 5, 1, "", "less_equal"], [199, 5, 1, "", "linspace"], [200, 5, 1, "", "load"], [201, 5, 1, "", "log"], [202, 5, 1, "", "log10"], [203, 5, 1, "", "log1p"], [204, 5, 1, "", "log2"], [205, 5, 1, "", "logaddexp"], [206, 5, 1, "", "logical_and"], [207, 5, 1, "", "logical_not"], [208, 5, 1, "", "logical_or"], [209, 5, 1, "", "logsumexp"], [210, 5, 1, "", "matmul"], [211, 5, 1, "", "max"], [212, 5, 1, "", "maximum"], [213, 5, 1, "", "mean"], [214, 5, 1, "", "meshgrid"], [227, 5, 1, "", "min"], [228, 5, 1, "", "minimum"], [229, 5, 1, "", "moveaxis"], [230, 5, 1, "", "multiply"], [231, 5, 1, "", "nan_to_num"], [232, 5, 1, "", "negative"], [233, 5, 1, "", "new_stream"], [234, 5, 1, "", "not_equal"], [235, 5, 1, "", "ones"], [236, 5, 1, "", "ones_like"], [237, 5, 1, "", "outer"], [238, 5, 1, "", "pad"], [239, 5, 1, "", "partition"], [240, 5, 1, "", "power"], [241, 5, 1, "", "prod"], [242, 5, 1, "", "put_along_axis"], [243, 5, 1, "", "quantize"], [244, 5, 1, "", "quantized_matmul"], [245, 5, 1, "", "radians"], [259, 5, 1, "", "real"], [260, 5, 1, "", "reciprocal"], [261, 5, 1, "", "remainder"], [262, 5, 1, "", "repeat"], [263, 5, 1, "", "reshape"], [264, 5, 1, "", "right_shift"], [265, 5, 1, "", "roll"], [266, 5, 1, "", "round"], [267, 5, 1, "", "rsqrt"], [268, 5, 1, "", "save"], [269, 5, 1, "", "save_gguf"], [270, 5, 1, "", "save_safetensors"], [271, 5, 1, "", "savez"], [272, 5, 1, "", "savez_compressed"], [273, 5, 1, "", "set_default_device"], [274, 5, 1, "", "set_default_stream"], [275, 5, 1, "", "sigmoid"], [276, 5, 1, "", "sign"], [277, 5, 1, "", "sin"], [278, 5, 1, "", "sinh"], [279, 5, 1, "", "slice"], [280, 5, 1, "", "slice_update"], [281, 5, 1, "", "softmax"], [282, 5, 1, "", "sort"], [283, 5, 1, "", "split"], [284, 5, 1, "", "sqrt"], [285, 5, 1, "", "square"], [286, 5, 1, "", "squeeze"], [287, 5, 1, "", "stack"], [288, 5, 1, "", "std"], [289, 5, 1, "", "stop_gradient"], [290, 5, 1, "", "stream"], [291, 5, 1, "", "subtract"], [292, 5, 1, "", "sum"], [293, 5, 1, "", "swapaxes"], [294, 5, 1, "", "synchronize"], [295, 5, 1, "", "take"], [296, 5, 1, "", "take_along_axis"], [297, 5, 1, "", "tan"], [298, 5, 1, "", "tanh"], [299, 5, 1, "", "tensordot"], [300, 5, 1, "", "tile"], [301, 5, 1, "", "topk"], [302, 5, 1, "", "trace"], [303, 5, 1, "", "transpose"], [304, 5, 1, "", "tri"], [305, 5, 1, "", "tril"], [306, 5, 1, "", "triu"], [307, 5, 1, "", "unflatten"], [308, 5, 1, "", "value_and_grad"], [309, 5, 1, "", "var"], [310, 5, 1, "", "view"], [311, 5, 1, "", "vjp"], [312, 5, 1, "", "vmap"], [313, 5, 1, "", "where"], [314, 5, 1, "", "zeros"], [315, 5, 1, "", "zeros_like"]], "mlx.core.Device": [[10, 4, 1, "", "__init__"]], "mlx.core.Dtype": [[11, 4, 1, "", "__init__"]], "mlx.core.DtypeCategory": [[12, 4, 1, "", "__init__"]], "mlx.core.Stream": [[324, 4, 1, "", "__init__"]], "mlx.core.array": [[32, 6, 1, "", "T"], [31, 4, 1, "", "__init__"], [33, 4, 1, "", "abs"], [34, 4, 1, "", "all"], [35, 4, 1, "", "any"], [36, 4, 1, "", "argmax"], [37, 4, 1, "", "argmin"], [38, 4, 1, "", "astype"], [39, 6, 1, "", "at"], [40, 4, 1, "", "conj"], [41, 4, 1, "", "cos"], [42, 4, 1, "", "cummax"], [43, 4, 1, "", "cummin"], [44, 4, 1, "", "cumprod"], [45, 4, 1, "", "cumsum"], [46, 4, 1, "", "diag"], [47, 4, 1, "", "diagonal"], [48, 6, 1, "", "dtype"], [49, 4, 1, "", "exp"], [50, 4, 1, "", "flatten"], [51, 4, 1, "", "item"], [52, 6, 1, "", "itemsize"], [53, 4, 1, "", "log"], [54, 4, 1, "", "log10"], [55, 4, 1, "", "log1p"], [56, 4, 1, "", "log2"], [57, 4, 1, "", "logsumexp"], [58, 4, 1, "", "max"], [59, 4, 1, "", "mean"], [60, 4, 1, "", "min"], [61, 4, 1, "", "moveaxis"], [62, 6, 1, "", "nbytes"], [63, 6, 1, "", "ndim"], [64, 4, 1, "", "prod"], [65, 4, 1, "", "reciprocal"], [66, 4, 1, "", "reshape"], [67, 4, 1, "", "round"], [68, 4, 1, "", "rsqrt"], [69, 6, 1, "", "shape"], [70, 4, 1, "", "sin"], [71, 6, 1, "", "size"], [72, 4, 1, "", "split"], [73, 4, 1, "", "sqrt"], [74, 4, 1, "", "square"], [75, 4, 1, "", "squeeze"], [76, 4, 1, "", "std"], [77, 4, 1, "", "sum"], [78, 4, 1, "", "swapaxes"], [79, 4, 1, "", "tolist"], [80, 4, 1, "", "transpose"], [81, 4, 1, "", "var"], [82, 4, 1, "", "view"]], "mlx.core.custom_function": [[113, 4, 1, "", "__init__"]], "mlx.core.distributed": [[121, 3, 1, "", "Group"], [122, 5, 1, "", "all_gather"], [123, 5, 1, "", "all_sum"], [124, 5, 1, "", "init"], [125, 5, 1, "", "is_available"], [126, 5, 1, "", "recv"], [127, 5, 1, "", "recv_like"], [128, 5, 1, "", "send"]], "mlx.core.distributed.Group": [[121, 4, 1, "", "__init__"]], "mlx.core.fast": [[145, 5, 1, "", "layer_norm"], [146, 5, 1, "", "metal_kernel"], [147, 5, 1, "", "rms_norm"], [148, 5, 1, "", "rope"], [149, 5, 1, "", "scaled_dot_product_attention"]], "mlx.core.fft": [[150, 5, 1, "", "fft"], [151, 5, 1, "", "fft2"], [152, 5, 1, "", "fftn"], [153, 5, 1, "", "ifft"], [154, 5, 1, "", "ifft2"], [155, 5, 1, "", "ifftn"], [156, 5, 1, "", "irfft"], [157, 5, 1, "", "irfft2"], [158, 5, 1, "", "irfftn"], [159, 5, 1, "", "rfft"], [160, 5, 1, "", "rfft2"], [161, 5, 1, "", "rfftn"]], "mlx.core.finfo": [[162, 4, 1, "", "__init__"]], "mlx.core.linalg": [[189, 5, 1, "", "cholesky"], [190, 5, 1, "", "cholesky_inv"], [191, 5, 1, "", "cross"], [192, 5, 1, "", "eigh"], [193, 5, 1, "", "eigvalsh"], [194, 5, 1, "", "inv"], [195, 5, 1, "", "norm"], [196, 5, 1, "", "qr"], [197, 5, 1, "", "svd"], [198, 5, 1, "", "tri_inv"]], "mlx.core.metal": [[215, 5, 1, "", "clear_cache"], [216, 5, 1, "", "device_info"], [217, 5, 1, "", "get_active_memory"], [218, 5, 1, "", "get_cache_memory"], [219, 5, 1, "", "get_peak_memory"], [220, 5, 1, "", "is_available"], [221, 5, 1, "", "reset_peak_memory"], [222, 5, 1, "", "set_cache_limit"], [223, 5, 1, "", "set_memory_limit"], [224, 5, 1, "", "set_wired_limit"], [225, 5, 1, "", "start_capture"], [226, 5, 1, "", "stop_capture"]], "mlx.core.random": [[246, 5, 1, "", "bernoulli"], [247, 5, 1, "", "categorical"], [248, 5, 1, "", "gumbel"], [249, 5, 1, "", "key"], [250, 5, 1, "", "laplace"], [251, 5, 1, "", "multivariate_normal"], [252, 5, 1, "", "normal"], [253, 5, 1, "", "permutation"], [254, 5, 1, "", "randint"], [255, 5, 1, "", "seed"], [256, 5, 1, "", "split"], [257, 5, 1, "", "truncated_normal"], [258, 5, 1, "", "uniform"]], "mlx.nn": [[335, 3, 1, "", "ALiBi"], [336, 3, 1, "", "AvgPool1d"], [337, 3, 1, "", "AvgPool2d"], [338, 3, 1, "", "AvgPool3d"], [339, 3, 1, "", "BatchNorm"], [340, 3, 1, "", "CELU"], [341, 3, 1, "", "Conv1d"], [342, 3, 1, "", "Conv2d"], [343, 3, 1, "", "Conv3d"], [344, 3, 1, "", "ConvTranspose1d"], [345, 3, 1, "", "ConvTranspose2d"], [346, 3, 1, "", "ConvTranspose3d"], [347, 3, 1, "", "Dropout"], [348, 3, 1, "", "Dropout2d"], [349, 3, 1, "", "Dropout3d"], [350, 3, 1, "", "ELU"], [351, 3, 1, "", "Embedding"], [352, 3, 1, "", "GELU"], [353, 3, 1, "", "GLU"], [354, 3, 1, "", "GRU"], [355, 3, 1, "", "GroupNorm"], [356, 3, 1, "", "HardShrink"], [357, 3, 1, "", "HardTanh"], [358, 3, 1, "", "Hardswish"], [359, 3, 1, "", "InstanceNorm"], [360, 3, 1, "", "LSTM"], [361, 3, 1, "", "LayerNorm"], [362, 3, 1, "", "LeakyReLU"], [363, 3, 1, "", "Linear"], [364, 3, 1, "", "LogSigmoid"], [365, 3, 1, "", "LogSoftmax"], [366, 3, 1, "", "MaxPool1d"], [367, 3, 1, "", "MaxPool2d"], [368, 3, 1, "", "MaxPool3d"], [369, 3, 1, "", "Mish"], [464, 3, 1, "", "Module"], [390, 3, 1, "", "MultiHeadAttention"], [391, 3, 1, "", "PReLU"], [392, 3, 1, "", "QuantizedEmbedding"], [393, 3, 1, "", "QuantizedLinear"], [394, 3, 1, "", "RMSNorm"], [395, 3, 1, "", "RNN"], [396, 3, 1, "", "ReLU"], [397, 3, 1, "", "ReLU6"], [398, 3, 1, "", "RoPE"], [399, 3, 1, "", "SELU"], [400, 3, 1, "", "Sequential"], [401, 3, 1, "", "SiLU"], [402, 3, 1, "", "Sigmoid"], [403, 3, 1, "", "SinusoidalPositionalEncoding"], [404, 3, 1, "", "Softmax"], [405, 3, 1, "", "Softmin"], [406, 3, 1, "", "Softplus"], [407, 3, 1, "", "Softshrink"], [408, 3, 1, "", "Softsign"], [409, 3, 1, "", "Step"], [410, 3, 1, "", "Tanh"], [411, 3, 1, "", "Transformer"], [412, 3, 1, "", "Upsample"], [421, 3, 1, "", "celu"], [422, 3, 1, "", "elu"], [423, 3, 1, "", "gelu"], [424, 3, 1, "", "gelu_approx"], [425, 3, 1, "", "gelu_fast_approx"], [426, 3, 1, "", "glu"], [427, 3, 1, "", "hard_shrink"], [428, 3, 1, "", "hard_tanh"], [429, 3, 1, "", "hardswish"], [430, 3, 1, "", "leaky_relu"], [431, 3, 1, "", "log_sigmoid"], [432, 3, 1, "", "log_softmax"], [447, 3, 1, "", "mish"], [448, 3, 1, "", "prelu"], [316, 5, 1, "", "quantize"], [449, 3, 1, "", "relu"], [450, 3, 1, "", "relu6"], [451, 3, 1, "", "selu"], [452, 3, 1, "", "sigmoid"], [453, 3, 1, "", "silu"], [454, 3, 1, "", "softmax"], [455, 3, 1, "", "softmin"], [456, 3, 1, "", "softplus"], [457, 3, 1, "", "softshrink"], [458, 3, 1, "", "step"], [459, 3, 1, "", "tanh"], [317, 5, 1, "", "value_and_grad"]], "mlx.nn.Module": [[370, 4, 1, "", "apply"], [371, 4, 1, "", "apply_to_modules"], [372, 4, 1, "", "children"], [373, 4, 1, "", "eval"], [374, 4, 1, "", "filter_and_map"], [375, 4, 1, "", "freeze"], [376, 4, 1, "", "leaf_modules"], [377, 4, 1, "", "load_weights"], [378, 4, 1, "", "modules"], [379, 4, 1, "", "named_modules"], [380, 4, 1, "", "parameters"], [381, 4, 1, "", "save_weights"], [382, 4, 1, "", "set_dtype"], [383, 6, 1, "", "state"], [384, 4, 1, "", "train"], [385, 4, 1, "", "trainable_parameters"], [386, 6, 1, "", "training"], [387, 4, 1, "", "unfreeze"], [388, 4, 1, "", "update"], [389, 4, 1, "", "update_modules"]], "mlx.nn.init": [[413, 5, 1, "", "constant"], [414, 5, 1, "", "glorot_normal"], [415, 5, 1, "", "glorot_uniform"], [416, 5, 1, "", "he_normal"], [417, 5, 1, "", "he_uniform"], [418, 5, 1, "", "identity"], [419, 5, 1, "", "normal"], [420, 5, 1, "", "uniform"]], "mlx.nn.losses": [[433, 3, 1, "", "binary_cross_entropy"], [434, 3, 1, "", "cosine_similarity_loss"], [435, 3, 1, "", "cross_entropy"], [436, 3, 1, "", "gaussian_nll_loss"], [437, 3, 1, "", "hinge_loss"], [438, 3, 1, "", "huber_loss"], [439, 3, 1, "", "kl_div_loss"], [440, 3, 1, "", "l1_loss"], [441, 3, 1, "", "log_cosh_loss"], [442, 3, 1, "", "margin_ranking_loss"], [443, 3, 1, "", "mse_loss"], [444, 3, 1, "", "nll_loss"], [445, 3, 1, "", "smooth_l1_loss"], [446, 3, 1, "", "triplet_loss"]], "mlx.optimizers": [[467, 3, 1, "", "AdaDelta"], [468, 3, 1, "", "Adafactor"], [469, 3, 1, "", "Adagrad"], [470, 3, 1, "", "Adam"], [471, 3, 1, "", "AdamW"], [472, 3, 1, "", "Adamax"], [473, 3, 1, "", "Lion"], [486, 3, 1, "", "Optimizer"], [478, 3, 1, "", "RMSprop"], [479, 3, 1, "", "SGD"], [318, 5, 1, "", "clip_grad_norm"], [480, 5, 1, "", "cosine_decay"], [481, 5, 1, "", "exponential_decay"], [482, 5, 1, "", "join_schedules"], [483, 5, 1, "", "linear_schedule"], [484, 5, 1, "", "step_decay"]], "mlx.optimizers.Optimizer": [[474, 4, 1, "", "apply_gradients"], [475, 4, 1, "", "init"], [476, 6, 1, "", "state"], [477, 4, 1, "", "update"]], "mlx.utils": [[319, 5, 1, "", "tree_flatten"], [320, 5, 1, "", "tree_map"], [321, 5, 1, "", "tree_map_with_path"], [322, 5, 1, "", "tree_reduce"], [323, 5, 1, "", "tree_unflatten"]]}, "objnames": {"0": ["cpp", "function", "C++ function"], "1": ["cpp", "functionParam", "C++ function parameter"], "2": ["cpp", "templateParam", "C++ template parameter"], "3": ["py", "class", "Python class"], "4": ["py", "method", "Python method"], "5": ["py", "function", "Python function"], "6": ["py", "property", "Python property"]}, "objtypes": {"0": "cpp:function", "1": "cpp:functionParam", "2": "cpp:templateParam", "3": "py:class", "4": "py:method", "5": "py:function", "6": "py:property"}, "terms": {"": [0, 1, 2, 5, 6, 7, 48, 52, 63, 95, 115, 117, 151, 152, 154, 155, 157, 158, 160, 161, 169, 190, 195, 197, 200, 213, 237, 243, 247, 266, 269, 270, 288, 290, 308, 309, 310, 312, 317, 334, 337, 338, 354, 360, 367, 368, 374, 375, 377, 381, 382, 383, 387, 395, 466, 475, 476, 488, 491, 493, 494, 497, 498, 499, 500], "0": [0, 1, 2, 4, 5, 6, 7, 9, 10, 15, 19, 39, 46, 47, 50, 67, 72, 76, 81, 84, 96, 99, 100, 101, 102, 103, 104, 105, 113, 118, 119, 144, 146, 149, 163, 167, 169, 175, 185, 192, 194, 195, 196, 198, 215, 222, 224, 231, 238, 246, 250, 252, 253, 258, 262, 266, 279, 280, 283, 287, 288, 302, 304, 305, 306, 307, 308, 309, 312, 318, 319, 321, 322, 334, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 352, 355, 356, 359, 361, 362, 366, 367, 368, 391, 396, 398, 403, 407, 409, 411, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 424, 425, 427, 428, 429, 430, 433, 435, 437, 438, 442, 445, 446, 448, 449, 450, 451, 457, 458, 461, 464, 467, 468, 470, 471, 472, 473, 475, 478, 479, 480, 481, 482, 483, 484, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499], "00005": 5, "0001": 403, "0005": 424, "001": 468, "00364": 5, "01": [5, 362, 430, 471], "0137595": 416, "015": 425, "0184009": 417, "02264": 415, "024": 494, "02765": 416, "0300242": 417, "044715": [352, 424], "0485873": 435, "05": [17, 177, 339, 355, 359, 361, 394], "0507": 451, "05202": 6, "06": [436, 446, 467], "0638": 442, "06450": 361, "0645099": 419, "06561": 481, "06675": 473, "07467": 394, "08": [17, 177, 434, 469, 470, 471, 472, 478], "08022": 359, "081": 484, "08415": 425, "08494": 355, "08619": 417, "08681": [369, 447], "09864": 6, "0999938": 482, "0999961": 480, "0f": 0, "1": [0, 1, 2, 3, 4, 6, 7, 15, 19, 29, 30, 39, 47, 50, 99, 100, 101, 102, 103, 104, 105, 113, 118, 119, 140, 141, 142, 143, 146, 149, 150, 151, 153, 154, 156, 157, 158, 159, 160, 161, 163, 172, 176, 183, 185, 190, 191, 192, 193, 195, 196, 210, 214, 223, 237, 239, 243, 247, 250, 251, 252, 258, 275, 279, 280, 282, 295, 301, 302, 307, 308, 318, 321, 322, 326, 334, 336, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 352, 353, 354, 355, 359, 360, 361, 363, 366, 391, 394, 395, 398, 402, 403, 409, 412, 414, 415, 416, 417, 418, 419, 420, 421, 422, 424, 425, 426, 428, 431, 432, 433, 434, 435, 436, 437, 438, 439, 441, 442, 444, 445, 446, 451, 452, 454, 455, 456, 458, 461, 464, 466, 467, 468, 469, 470, 471, 472, 473, 475, 478, 479, 480, 481, 482, 483, 484, 491, 492, 493, 494, 495, 497, 498, 499, 500], "10": [0, 3, 6, 7, 185, 202, 266, 271, 320, 334, 377, 461, 482, 484, 491, 492, 495], "100": [2, 5, 6, 433, 483, 491, 494, 496, 500], "1000": [480, 491], "10000": 398, "101": 483, "1024": [1, 6], "105361": 433, "109": 2, "10_000": 5, "10x": 473, "11": 195, "114": 2, "12": [6, 172, 185, 482], "1212": 467, "12451": 415, "128": [271, 334], "13": 9, "14": [9, 185], "15": [1, 9, 185, 195, 224, 322, 491], "150594": 414, "15268": 416, "16": [1, 146, 326, 336, 338, 359, 366, 368, 370, 464], "1606": 425, "1607": [359, 361], "16384": 172, "16506": 417, "17": [4, 9], "177208": 416, "18": 185, "1803": 355, "1908": [369, 447], "1910": 394, "191107": 414, "1985": 195, "1_000": 5, "1d": [0, 99, 103, 106, 269, 296], "1e": [0, 5, 7, 17, 177, 339, 355, 359, 361, 362, 394, 434, 436, 446, 466, 467, 468, 469, 470, 471, 472, 475, 478, 480, 481, 482, 483, 484], "1e3": 491, "1st": 243, "2": [0, 1, 2, 4, 5, 6, 7, 39, 100, 104, 113, 118, 119, 135, 141, 142, 143, 151, 154, 156, 157, 158, 159, 160, 161, 163, 172, 183, 185, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 204, 210, 243, 251, 256, 279, 280, 299, 302, 304, 305, 306, 307, 318, 322, 326, 334, 336, 337, 338, 342, 345, 352, 362, 366, 367, 368, 394, 403, 412, 413, 414, 415, 416, 417, 418, 419, 420, 424, 435, 436, 438, 445, 446, 461, 464, 466, 467, 469, 470, 471, 475, 478, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "20": [172, 185, 195], "200": [6, 482, 494], "2002": 6, "2011": 469, "2012": [467, 478], "2015": [348, 470, 472], "2019": [6, 471], "2020": 6, "2021": 6, "20397": 433, "20_000": 6, "21": [6, 185, 484], "2104": 6, "223144": 433, "223404": 415, "225": 195, "225763": 442, "2302": 473, "23607": [195, 196], "24": 185, "24264": 195, "247": 6, "25": [9, 391, 412], "25211": 416, "256": [1, 2, 7, 146], "256995": 442, "27": 4, "28": [172, 185], "2d": [0, 100, 104, 119, 243, 339, 348], "2nd": 243, "2x": 497, "3": [0, 1, 2, 4, 6, 9, 101, 105, 113, 141, 143, 163, 183, 185, 191, 192, 193, 195, 196, 279, 280, 307, 318, 322, 338, 343, 346, 352, 368, 412, 415, 417, 424, 429, 468, 473, 488, 491, 493, 495, 497, 498], "30": 468, "3118": 497, "32": [1, 6, 7, 91, 243, 244, 326, 337, 338, 367, 368, 394, 491], "330": 6, "33333": 412, "348587": 435, "363207": 414, "36788": 491, "379159": 415, "380709": 419, "39": 6, "3d": [0, 2, 101, 105, 339, 349, 412], "3f": [2, 7, 491], "3x": 2, "4": [0, 1, 2, 6, 117, 146, 149, 163, 168, 185, 195, 243, 244, 271, 279, 307, 316, 322, 326, 336, 337, 338, 339, 359, 366, 367, 368, 392, 393, 411, 412, 414, 415, 416, 433, 491, 492, 493, 495, 498, 500], "4096": [491, 494, 500], "40x": 1, "41421": 195, "417497": 420, "42": 323, "437": 6, "44": 6, "447214": 196, "458835": 416, "475": 6, "48095": 414, "4d": [1, 412], "4m": 1, "5": [0, 1, 2, 5, 6, 9, 185, 195, 223, 246, 279, 322, 336, 339, 347, 348, 349, 352, 356, 359, 366, 407, 412, 413, 416, 417, 424, 427, 445, 457, 461, 466, 478, 480, 481, 491, 494, 495], "50": [0, 199], "500": [6, 500], "5000": 2, "510826": 433, "512": [2, 3, 6, 411, 500], "534422": 419, "539245": 433, "53947": 414, "55": 1, "5701": 467, "573409": 442, "57771": 196, "579": 6, "5f": 5, "6": [1, 2, 6, 113, 185, 195, 271, 279, 397, 411, 415, 424, 425, 429, 436, 446, 450, 478, 491, 495, 498], "61278": 414, "617261": 420, "628": 6, "633": 6, "639": 494, "64": [0, 1, 91, 117, 168, 243, 244, 316, 326, 392, 393], "64331": 417, "666329": 417, "66667": 412, "67326": 451, "676": 1, "690": 6, "6967": 416, "7": [2, 6, 185, 195, 243, 495], "702": [352, 425], "707107": 192, "71828": 491, "74166": 195, "74597": 195, "75": 412, "75596": 442, "75787": 416, "765166": 442, "773433": 442, "776856": 415, "793615": 417, "79854": 417, "7b": 6, "7m": 1, "8": [0, 1, 2, 6, 9, 195, 243, 326, 337, 338, 359, 367, 368, 411, 434, 467, 468, 469, 470, 471, 472, 478, 491, 495, 498, 500], "8192": [6, 172], "84804": 195, "863726": 420, "883935": 420, "890597": 415, "894427": 196, "89613": 414, "8gb": 6, "8x": 1, "9": [4, 9, 195, 435, 467, 470, 471, 472, 473, 475, 481, 484, 497], "90041": 415, "912766": 415, "916291": 433, "95": 7, "982273": 419, "99": [473, 478], "995016": 414, "999": [470, 471, 472], "A": [0, 2, 6, 8, 9, 10, 69, 83, 95, 141, 142, 145, 146, 147, 149, 169, 183, 184, 190, 192, 193, 195, 196, 197, 200, 209, 210, 211, 216, 227, 243, 246, 247, 248, 250, 251, 252, 253, 254, 257, 258, 283, 287, 290, 308, 311, 312, 316, 317, 318, 319, 320, 321, 322, 323, 324, 334, 339, 348, 354, 355, 359, 361, 374, 378, 379, 382, 388, 389, 394, 400, 403, 411, 414, 415, 417, 425, 446, 447, 464, 466, 470, 472, 474, 475, 477, 482, 491, 492, 493, 494, 496, 497], "AS": 167, "And": [4, 6, 412], "As": [7, 39, 295, 334], "At": [94, 307], "But": [493, 500], "By": [6, 316, 382, 433, 494, 497], "For": [0, 1, 2, 4, 6, 9, 39, 113, 149, 167, 183, 195, 243, 323, 334, 339, 348, 352, 370, 375, 384, 387, 393, 398, 403, 412, 414, 415, 416, 417, 433, 461, 466, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "If": [0, 1, 2, 4, 6, 9, 16, 17, 18, 19, 27, 28, 29, 30, 79, 83, 84, 94, 96, 106, 109, 110, 111, 112, 118, 119, 122, 123, 124, 126, 127, 128, 137, 145, 148, 149, 159, 160, 161, 165, 166, 169, 177, 189, 190, 191, 195, 200, 209, 210, 211, 213, 214, 222, 223, 227, 231, 235, 238, 239, 241, 242, 247, 251, 253, 262, 265, 281, 282, 283, 288, 292, 294, 295, 296, 299, 301, 302, 308, 309, 312, 314, 316, 320, 322, 339, 341, 342, 343, 344, 345, 346, 355, 361, 363, 375, 377, 387, 393, 395, 398, 400, 403, 412, 433, 435, 446, 468, 470, 471, 491, 492, 493, 494, 496, 499, 500, 501], "In": [0, 1, 2, 6, 7, 39, 149, 210, 243, 320, 334, 348, 355, 464, 467, 469, 470, 472, 473, 474, 490, 491, 492, 493, 494, 496, 499, 500], "It": [2, 6, 9, 127, 169, 274, 308, 318, 322, 334, 389, 393, 474, 486, 497, 499], "Its": [334, 493], "No": [2, 6, 192, 193], "Not": [95, 234, 491], "ON": [3, 4, 9], "Of": 494, "On": [1, 491, 494, 496], "One": [150, 153, 159, 238, 267, 491, 493, 494], "THE": 9, "That": 6, "The": [0, 1, 2, 3, 4, 6, 7, 8, 9, 13, 14, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 48, 52, 62, 63, 69, 79, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 116, 117, 118, 119, 122, 123, 124, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 138, 139, 140, 141, 142, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 184, 185, 186, 187, 188, 191, 192, 193, 195, 196, 197, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 218, 219, 222, 223, 224, 225, 227, 228, 229, 230, 232, 234, 235, 236, 237, 238, 239, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 269, 270, 275, 276, 277, 278, 279, 280, 281, 282, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 326, 328, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 347, 348, 349, 351, 353, 354, 355, 359, 360, 361, 363, 366, 367, 368, 370, 371, 375, 377, 381, 382, 383, 384, 387, 388, 389, 390, 392, 393, 394, 395, 398, 400, 403, 409, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 458, 461, 464, 466, 467, 468, 469, 470, 471, 472, 473, 476, 478, 479, 480, 483, 486, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "Then": [5, 9], "There": [1, 2, 334, 412, 491], "These": [1, 2, 95, 242, 296, 435, 500], "To": [0, 2, 3, 5, 6, 7, 9, 222, 334, 461, 466, 491, 492, 493, 494, 498], "With": [2, 493], "_": [1, 3, 5, 6, 321, 334, 480, 481, 482, 483, 484, 488, 491, 496, 500], "__call__": [1, 6, 7, 334, 464, 493], "__init__": [2, 6, 7, 10, 11, 12, 31, 113, 121, 162, 324, 334, 464], "__main__": [2, 6], "__name__": [2, 6], "_a": 2, "_ext": 2, "_f": 195, "_in": [414, 415], "_out": [414, 415], "_p": 446, "_val": 428, "a1": 167, "a2": 167, "a_": 195, "a_max": [0, 94], "a_min": [0, 94], "a_ndim": 1, "a_shap": 1, "a_strid": 1, "a_view": 497, "ab": [0, 17, 177, 195, 308, 355, 359, 361, 369, 394, 425, 447, 491, 493], "abil": 492, "abl": [2, 4, 243], "abort": 113, "about": [1, 2, 6, 7, 132, 216, 496, 500], "abov": [1, 2, 6, 243, 305, 334, 412, 492, 493, 494, 495, 496, 500], "absolut": [0, 13, 17, 177, 424, 425, 445], "acc": 322, "acceler": [2, 4, 339], "access": [0, 6, 51, 334, 464, 475, 492, 496, 500], "accord": [0, 248, 313, 316, 390, 414, 415, 416, 417], "accordingli": 2, "accumul": [322, 394], "accuraci": 7, "accustom": 6, "achiev": [334, 492], "across": [1, 2, 9, 355, 492], "act": [2, 441], "action": 334, "activ": [2, 9, 217, 348, 409, 411, 427, 447, 457, 458, 460, 491], "actual": [6, 19, 377, 464, 496], "ad": [0, 1, 2, 5, 9, 145, 359, 464, 467, 468, 469, 470, 471, 472, 478, 492, 496, 499], "adadelta": 466, "adafactor": 466, "adagrad": 466, "adam": [466, 472, 473, 482, 483], "adamax": 466, "adamw": [466, 473], "adapt": [467, 468, 469, 492], "add": [0, 1, 2, 3, 4, 6, 15, 39, 139, 205, 238, 243, 341, 342, 343, 344, 345, 346, 493, 494, 500], "add_argu": 6, "add_depend": 2, "add_execut": 4, "add_fun": 493, "add_librari": 2, "addit": [0, 2, 4, 6, 9, 14, 15, 141, 145, 147, 149, 200, 339, 355, 361, 390, 394, 464, 494], "addmm": 0, "address": 2, "adjac": 348, "advanc": [6, 491], "advantag": 500, "advis": 497, "affin": [339, 355, 359, 361, 363, 393], "after": [2, 6, 7, 29, 163, 165, 168, 215, 239, 243, 339, 355, 361, 370, 371, 375, 377, 384, 387, 388, 389, 390, 411, 445, 491, 500], "after_1": 238, "after_2": 238, "after_i": 238, "after_n": 238, "afternoon": 6, "again": [6, 9, 334, 491], "against": [0, 4], "aggreg": 390, "ago": 6, "ai": 113, "ainv": [194, 198], "albeit": 500, "algebra": 8, "algorithm": [412, 473], "alia": [97, 98, 352], "alibi": 334, "align": [190, 243, 354, 360], "align_corn": 412, "all": [0, 1, 2, 3, 7, 9, 17, 29, 39, 85, 86, 87, 95, 100, 101, 102, 104, 105, 113, 122, 123, 124, 142, 144, 152, 155, 158, 161, 167, 168, 197, 210, 238, 239, 265, 286, 316, 334, 370, 371, 375, 378, 379, 380, 385, 387, 390, 403, 411, 412, 461, 464, 486, 488, 491, 495, 496, 498, 501], "all_avg": 492, "all_reduce_grad": 492, "all_sum": 492, "allclos": [0, 1, 146], "alloc": [2, 218, 222, 223, 464], "allow": [0, 1, 2, 141, 143, 183, 318, 334, 389, 464, 486, 492, 495, 498], "allow_col_major": 0, "almost": 6, "alon": [2, 497], "along": [0, 2, 27, 28, 95, 96, 109, 110, 111, 112, 122, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 167, 168, 172, 191, 195, 242, 253, 262, 265, 281, 283, 287, 295, 296, 299, 300, 301, 302, 310, 334, 353, 395, 426], "alpha": [0, 2, 15, 243, 340, 350, 421, 422, 446, 448, 451, 471, 478], "alpha_": 2, "alreadi": [2, 3, 6, 492], "also": [0, 1, 2, 4, 6, 7, 8, 9, 12, 14, 88, 89, 90, 120, 129, 130, 134, 152, 155, 158, 161, 170, 171, 186, 187, 188, 205, 212, 228, 230, 234, 240, 243, 261, 264, 291, 316, 317, 328, 334, 374, 388, 390, 392, 393, 401, 423, 451, 453, 460, 466, 491, 492, 493, 494, 495, 496, 497, 498, 501], "altern": 488, "alwai": [1, 84, 175, 217, 319, 493, 494], "am": 6, "among": 2, "amount": [6, 219, 336, 366, 493], "amus": 6, "an": [0, 1, 2, 3, 4, 6, 7, 9, 11, 16, 18, 31, 85, 86, 87, 92, 99, 100, 101, 102, 103, 104, 105, 121, 126, 127, 128, 137, 141, 143, 144, 145, 149, 163, 166, 173, 175, 178, 189, 195, 200, 223, 224, 229, 235, 236, 238, 241, 242, 243, 244, 253, 262, 263, 265, 266, 283, 286, 293, 295, 296, 299, 300, 304, 307, 312, 314, 315, 319, 320, 321, 322, 334, 347, 352, 355, 360, 361, 363, 370, 390, 391, 393, 395, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 424, 448, 461, 466, 467, 477, 481, 486, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "anaconda": 492, "anchor": 446, "angl": [116, 245, 362], "angular": [148, 398], "ani": [0, 1, 2, 6, 8, 19, 95, 113, 124, 319, 320, 321, 322, 323, 334, 352, 370, 371, 374, 383, 393, 411, 412, 461, 483, 490, 491, 494, 496, 498, 499, 500], "anonym": 491, "anoth": [0, 94, 183, 210, 291, 313, 326, 334, 370, 491, 493, 494, 495, 500], "anwywher": 9, "anyhow": 6, "anymor": 6, "anyth": [6, 308, 496], "anytim": 496, "api": [1, 2, 141, 143, 175, 352, 492, 493, 494], "app": 9, "append": [6, 210, 491, 496], "appl": [2, 6, 8, 9, 500], "appli": [0, 39, 148, 149, 167, 197, 320, 321, 322, 334, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 348, 349, 350, 352, 353, 355, 356, 357, 358, 359, 361, 362, 363, 364, 365, 366, 367, 368, 369, 371, 384, 391, 393, 394, 395, 396, 397, 399, 401, 402, 404, 405, 406, 407, 408, 409, 410, 412, 421, 422, 423, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 461, 470, 471, 474, 477, 483, 486, 491, 492], "applic": [3, 9], "apply_fn": 371, "apply_gradi": 466, "apply_to_modul": [334, 375], "approach": [441, 494], "appropri": [2, 491], "approx": 352, "approxim": [17, 352, 423, 424, 425], "ar": [0, 1, 2, 5, 6, 7, 8, 9, 17, 19, 83, 91, 92, 94, 95, 102, 106, 113, 119, 126, 127, 137, 144, 146, 149, 151, 152, 154, 155, 157, 158, 160, 161, 163, 168, 169, 177, 178, 179, 180, 181, 182, 183, 184, 192, 193, 195, 196, 200, 210, 223, 237, 238, 239, 243, 244, 246, 247, 248, 253, 254, 257, 258, 265, 271, 272, 286, 287, 295, 308, 311, 312, 316, 319, 320, 326, 339, 341, 342, 343, 344, 345, 346, 347, 348, 349, 355, 359, 361, 363, 377, 390, 393, 412, 433, 435, 436, 460, 464, 466, 473, 475, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "arang": [0, 1, 195, 253, 326, 412, 495, 497], "arbitrari": [319, 464], "arbitrarili": [1, 95, 334, 490, 494, 498], "arc": 0, "arcco": 0, "arccosh": 0, "architectur": [6, 9, 216, 334, 389, 500], "archiv": 499, "arcsin": 0, "arcsinh": 0, "arctan": 0, "arctan2": 0, "arctanh": 0, "arg": [2, 6, 11, 19, 121, 137, 141, 142, 143, 162, 175, 271, 272, 324], "arg1": 183, "arg2": 183, "argmax": [0, 7], "argmin": 0, "argnam": [169, 308], "argnum": [2, 113, 169, 308, 494], "argpars": 6, "argpartit": 0, "argsort": 0, "argument": [1, 32, 66, 80, 95, 137, 169, 308, 320, 321, 322, 334, 412, 488, 492, 493, 494, 499, 500, 501], "argumentpars": 6, "ari": [85, 86, 87], "aris": 497, "arm": 9, "arm64": 9, "around": 6, "arr": [0, 268, 495], "arr_0": 499, "arrai": [0, 1, 2, 4, 6, 7, 8, 11, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 116, 117, 118, 119, 122, 123, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 318, 334, 339, 360, 370, 377, 380, 385, 391, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 458, 461, 464, 467, 468, 469, 470, 471, 472, 473, 478, 479, 480, 481, 482, 483, 484, 491, 492, 493, 494, 496, 497, 498, 500], "array_equ": [0, 17, 177], "arrayfir": 8, "arxiv": [6, 355, 359, 361, 369, 394, 425, 447, 467, 473], "as_strid": 0, "ascend": [192, 193], "ask": [6, 492], "assert": [1, 2, 146], "assign": [0, 2, 39, 464], "associ": [2, 271, 272, 496], "assum": [0, 2, 6, 91, 191, 192, 193, 196, 320, 334, 355], "astyp": [0, 1, 2, 6, 146, 370, 497], "atleast": 0, "atleast_1d": 0, "atleast_2d": 0, "atleast_3d": 0, "atol": [0, 17, 177], "atom": [1, 146], "atomic_fetch_add_explicit": 1, "atomic_output": [1, 146], "attach": 2, "attempt": 95, "attend": 390, "attent": [149, 375, 390, 403, 411], "attention_norm": 6, "attribut": [1, 10, 11, 12, 31, 162, 324, 383, 464, 486], "audio": 412, "auto": [0, 2, 4, 9, 493], "autom": 494, "automat": [1, 2, 8, 146, 200, 492, 498, 499, 500], "autoregress": 6, "avail": [2, 5, 6, 7, 9, 11, 124, 125, 220, 328, 500], "averag": [336, 337, 338, 467, 468, 470, 471, 472, 492], "avgpool1d": 334, "avgpool2d": 334, "avgpool3d": 334, "avoid": [1, 2, 382, 491], "awai": [2, 6], "awar": [491, 496], "ax": [0, 2, 16, 18, 27, 28, 80, 113, 139, 151, 152, 154, 155, 157, 158, 160, 161, 163, 176, 195, 209, 211, 213, 227, 238, 241, 265, 279, 280, 281, 286, 288, 292, 293, 299, 303, 309, 494], "axes_a": 0, "axes_b": 0, "axi": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 72, 75, 76, 77, 81, 96, 109, 110, 111, 112, 119, 122, 139, 145, 147, 150, 153, 156, 157, 158, 159, 160, 161, 163, 172, 191, 195, 209, 211, 213, 227, 229, 238, 239, 241, 242, 247, 253, 262, 265, 281, 282, 283, 286, 287, 288, 292, 293, 295, 296, 300, 301, 302, 303, 307, 309, 310, 312, 336, 337, 338, 353, 366, 367, 368, 395, 426, 432, 434, 435, 439, 444, 446, 454, 455, 495], "axis1": [0, 47, 78, 119, 293, 302], "axis2": [0, 47, 78, 119, 293, 302], "axpbi": 2, "axpby_": 2, "axpby_gener": 2, "axpby_general_bfloat16": 2, "axpby_general_complex64": 2, "axpby_general_float16": 2, "axpby_general_float32": 2, "axpby_impl": 2, "axpby_impl_acceler": 2, "b": [0, 1, 2, 3, 4, 6, 14, 15, 17, 25, 83, 88, 89, 90, 91, 129, 130, 134, 146, 149, 165, 167, 170, 171, 175, 176, 177, 185, 186, 187, 188, 191, 195, 205, 206, 208, 210, 212, 228, 230, 234, 237, 240, 243, 250, 261, 264, 291, 299, 308, 321, 322, 353, 363, 395, 412, 426, 494, 495, 496, 497, 498, 499, 500], "b1": 167, "b2": 167, "b_": [354, 360], "b_stride": 1, "ba": [470, 472], "back": [6, 113, 220, 497], "backend": [1, 9, 124, 125], "backward": [1, 491, 494], "bad": 496, "balanc": 441, "baltimor": 195, "bandwidth": [491, 492], "bar": 492, "base": [0, 2, 4, 148, 195, 202, 204, 240, 398, 411, 464, 466, 472, 486, 488, 491, 495], "base_idx": 1, "basi": 486, "basic": [5, 266, 494], "batch": [6, 15, 91, 149, 167, 168, 210, 251, 339, 341, 342, 343, 344, 345, 346, 348, 349, 354, 360, 390, 395, 412, 496], "batch_idx": 1, "batch_iter": [7, 466], "batch_siz": [7, 466], "batchnorm": 334, "becaus": [6, 217, 334, 491, 493, 496], "been": [0, 2, 6, 218, 496], "befor": [1, 2, 6, 9, 29, 146, 239, 374, 411, 475, 492, 495, 496], "before_1": 238, "before_2": 238, "before_i": 238, "before_n": 238, "beforehand": 237, "beggin": 265, "begin": [84, 190, 219, 243, 354, 360, 409, 427, 438, 445, 451, 457, 458], "behav": 113, "behavior": [251, 441, 495, 496], "behaviour": [113, 189, 190], "behind": 494, "being": [289, 334], "bell": 2, "below": [2, 9, 195, 304, 306, 326, 412, 496], "bench": 2, "benchmark": [2, 491], "benefici": [348, 349, 496], "best": 492, "beta": [0, 2, 15, 117, 243, 339, 355, 359, 361, 445, 466, 470, 471, 472, 473], "beta_": 2, "beta_1": [468, 470, 471, 472, 473], "beta_2": [470, 471, 472, 473], "better": [494, 500], "between": [0, 2, 8, 94, 163, 411, 434, 437, 438, 441, 482, 492, 496, 497, 500], "beyond": [265, 480, 483], "bfloat16": [2, 12, 172, 326, 497], "bfloat16_t": 2, "bia": [6, 117, 145, 168, 243, 244, 320, 334, 341, 342, 343, 344, 345, 346, 354, 360, 361, 363, 375, 377, 387, 390, 393, 395, 470, 471, 472, 475, 494], "bias": [0, 117, 168, 243, 244, 354, 360, 375, 387, 390], "bias_correct": [470, 471], "bicub": 412, "big": [1, 491], "bigger": [6, 468], "bilinear": [1, 412], "binari": [200, 268, 269, 270, 271, 272, 310, 409, 433, 458, 491], "binary_cross_entropi": [334, 491], "bit": [0, 117, 168, 186, 243, 244, 264, 316, 326, 370, 392, 393, 394], "bitwis": [0, 88, 89, 90, 186, 264], "bitwise_and": 0, "bitwise_or": 0, "bitwise_xor": 0, "block": [0, 2, 6, 91, 411], "block_masked_mm": 0, "block_siz": [0, 91], "bn": 339, "bodi": [1, 146], "bool": [0, 1, 2, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 76, 77, 79, 81, 83, 95, 102, 109, 110, 111, 112, 124, 125, 141, 143, 146, 148, 168, 177, 183, 189, 190, 195, 198, 200, 209, 211, 213, 214, 220, 223, 227, 241, 244, 288, 292, 309, 316, 339, 341, 342, 343, 344, 345, 346, 354, 355, 359, 360, 361, 363, 370, 374, 375, 377, 382, 384, 387, 390, 393, 395, 398, 403, 411, 412, 433, 436, 468, 470, 471, 479], "bool_": [12, 326], "boolean": [0, 17, 83, 149, 177, 178, 179, 180, 181, 182, 183, 206, 207, 208, 326, 386, 495], "both": [1, 2, 14, 88, 89, 90, 129, 130, 134, 170, 171, 183, 186, 187, 188, 195, 205, 212, 228, 230, 234, 240, 247, 261, 264, 291, 316, 336, 337, 338, 359, 360, 366, 367, 368, 466, 491, 492, 493, 494, 498, 500], "bottom": 412, "bound": [0, 254, 257, 258, 352, 420, 491, 495, 500], "boundari": 482, "bracket": 6, "brain": 326, "break": 497, "bregler": 348, "broadcast": [0, 2, 14, 17, 88, 89, 90, 92, 94, 129, 130, 134, 149, 166, 170, 171, 177, 186, 187, 188, 205, 210, 212, 228, 230, 234, 240, 242, 246, 247, 251, 257, 258, 261, 264, 291, 296, 313, 390], "broadcast_arrai": [0, 2], "broadcast_to": 0, "broadcasted_input": 2, "brought": 8, "btl_tcp_link": 492, "buffer": [1, 2, 217, 497], "bui": 6, "build": [3, 4, 6, 8, 416, 464, 491, 493], "build_ext": [2, 9], "build_shared_lib": [2, 9], "built": [1, 2, 4, 9, 496], "bundl": 6, "byte": [52, 62, 217, 218, 219, 222, 223, 224, 326], "c": [0, 1, 2, 6, 15, 195, 339, 341, 342, 343, 344, 345, 346, 348, 349, 359, 360, 497, 498, 500], "c_": [360, 473], "c_in": [99, 100, 101, 102, 103, 104, 105], "c_out": [99, 100, 101, 102, 103, 104, 105], "c_pad": 1, "c_t": [360, 473], "cach": [6, 9, 215, 217, 218, 222, 491], "calcul": [195, 433, 436, 442, 468], "call": [2, 3, 6, 7, 32, 124, 127, 165, 175, 215, 219, 334, 351, 375, 387, 392, 400, 464, 466, 475, 491, 492, 493, 494, 496], "callabl": [95, 113, 141, 143, 146, 169, 175, 184, 308, 311, 312, 316, 317, 319, 320, 321, 322, 370, 371, 374, 382, 395, 400, 411, 413, 414, 415, 416, 417, 418, 419, 420, 467, 468, 469, 470, 471, 472, 473, 478, 479, 480, 481, 482, 483, 484], "can": [1, 2, 3, 4, 6, 8, 9, 14, 19, 66, 80, 84, 88, 89, 90, 95, 119, 120, 121, 129, 130, 134, 137, 141, 142, 149, 170, 171, 175, 186, 187, 188, 195, 205, 212, 224, 228, 230, 234, 240, 246, 247, 254, 257, 258, 261, 264, 269, 291, 302, 307, 308, 322, 334, 337, 338, 351, 352, 367, 368, 374, 387, 392, 400, 412, 435, 461, 464, 466, 474, 475, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "cannot": [6, 94, 495, 497], "captur": [2, 3, 95, 113, 225, 226, 334, 491], "care": [6, 493, 496], "carefulli": [491, 493], "carri": 2, "cartesian": 214, "case": [2, 6, 122, 123, 124, 126, 127, 128, 152, 155, 156, 158, 159, 160, 161, 163, 189, 190, 191, 192, 193, 194, 196, 197, 198, 210, 263, 286, 307, 337, 338, 348, 367, 368, 409, 427, 445, 451, 457, 458, 474, 475, 491, 493, 494, 498, 499, 500, 501], "cast": [2, 38, 159, 160, 161, 200, 370, 382, 497], "caster": 2, "categor": 6, "categori": [12, 183, 326], "catlas_saxpbi": 2, "caus": [334, 491, 496], "causal": 6, "caution": 84, "cd": [3, 9], "cdf": [248, 352, 423], "cdot": [425, 434, 437, 453], "ceil": 0, "ceildiv": 1, "cell": 360, "celu": 334, "certain": [2, 384, 491], "chang": [84, 95, 141, 143, 175, 274, 310, 388, 393, 412, 438, 445, 491, 497], "channel": [1, 99, 100, 101, 102, 103, 104, 105, 339, 341, 342, 343, 344, 345, 346, 348, 349], "channel_idx": 1, "charact": 319, "check": [0, 2, 9, 83, 125, 183, 192, 193, 220, 377, 493, 494, 495], "checklist": 492, "checkout": [3, 491], "checkpoint": [411, 466], "chen": 473, "child": 389, "children": 334, "chip": 9, "choleski": 190, "choos": [6, 148, 398], "chosen": 132, "clamp": 163, "clang": 9, "clariti": 494, "class": [2, 6, 7, 10, 11, 12, 31, 113, 121, 162, 324, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 464, 467, 468, 469, 470, 471, 472, 473, 478, 479, 486], "class_pred": 316, "classif": [416, 417], "classifi": 7, "classmethod": [392, 393], "clear": 215, "click": 9, "clip": [0, 318, 433, 468], "clip_threshold": 468, "clipped_grad": 318, "clone": 9, "close": [5, 8, 9, 17, 177], "closer": 320, "cmake": [3, 4, 9], "cmake_arg": 3, "cmake_build_parallel_level": 9, "cmake_build_typ": 9, "cmake_current_list_dir": 2, "cmake_cxx_standard": 4, "cmake_cxx_standard_requir": 4, "cmake_host_system_processor": 9, "cmake_library_output_directori": 2, "cmake_minimum_requir": 4, "cmakebuild": 2, "cmakeextens": 2, "cmakelist": [2, 4], "cmdclass": 2, "co": [0, 2, 113, 403, 494], "code": [1, 146, 491, 492, 493, 496], "coeffici": [2, 467, 468, 470, 471, 472, 473], "col": 304, "col_contigu": 2, "cold": 9, "collect": [2, 320, 321, 490], "column": [2, 144, 173, 192, 243], "com": [9, 492], "combin": [6, 197, 322], "come": [2, 6, 492, 494], "command": [2, 3, 4, 9, 492], "command_buff": 2, "common": [2, 466, 491, 496], "commonli": [7, 388, 461, 491], "commun": [8, 121, 124, 125], "compar": [2, 83, 491], "comparison": [17, 134, 170, 171, 187, 188, 234], "compat": [6, 141, 143, 149, 175, 247, 251, 352, 499], "compil": [0, 3, 4, 8, 9, 120, 133, 146, 492, 493, 494, 496], "compiled_fun": [491, 493], "compiled_grad_fn": 491, "complet": [5, 6, 9, 223, 388, 389, 493, 494, 500], "complex": [2, 97, 98, 157, 158, 159, 160, 161, 174, 192, 193, 259, 319, 326, 334, 389, 491, 493, 494], "complex64": [2, 12, 326], "complex64_t": 2, "complexflo": 12, "compon": [2, 4, 6], "compos": [8, 334, 491, 494, 498], "composit": 498, "compress": 272, "compromis": 6, "comput": [0, 1, 2, 5, 6, 7, 8, 9, 109, 110, 111, 112, 113, 117, 132, 140, 148, 169, 184, 185, 189, 190, 191, 192, 193, 194, 195, 198, 205, 213, 237, 243, 261, 281, 288, 289, 299, 308, 309, 311, 317, 334, 339, 354, 355, 359, 360, 361, 375, 388, 393, 394, 398, 411, 414, 415, 416, 417, 424, 425, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 466, 467, 468, 470, 471, 472, 473, 477, 491, 492, 493, 494, 498, 500], "computation": 496, "compute_encod": 2, "concaten": [0, 6, 122], "concept": 464, "concis": 6, "concret": [2, 354, 360, 363, 395, 496, 500], "conda": [9, 492], "condit": [0, 2, 313, 491, 500], "config": [2, 4, 492], "configu": 466, "configur": [117, 492], "confirm": 492, "confus": 7, "conj": 98, "conjug": [0, 97], "connect": 492, "consecut": [148, 243, 398], "consequ": 6, "consid": [6, 17, 83, 177, 319, 320, 321, 355, 490], "consider": 491, "const": [0, 1, 2, 436], "constant": [0, 2, 6, 9, 113, 145, 147, 238, 334, 339, 355, 361, 394, 436, 446, 478, 480, 491, 493, 497], "constant_valu": 238, "constitut": 320, "construct": [0, 2, 7, 46, 118, 166, 235, 300, 314], "consum": 496, "contain": [2, 6, 9, 29, 30, 69, 95, 119, 132, 156, 157, 158, 167, 168, 192, 195, 206, 207, 208, 243, 283, 313, 318, 334, 374, 376, 377, 383, 411, 442, 461, 464, 491, 492, 494], "content": [9, 374, 491], "context": [290, 493], "contigu": [0, 1, 2, 84, 146], "continu": [340, 421, 494], "contract": [0, 132], "contribut": 2, "contriv": [494, 500], "control": [0, 362, 488, 496], "conv": 106, "conv1d": [0, 334], "conv2d": [0, 334], "conv3d": [0, 334], "conv_gener": 0, "conv_transpose1d": 0, "conv_transpose2d": 0, "conv_transpose3d": 0, "conveni": [1, 2, 7, 183], "convent": [19, 106, 131, 132, 412], "convers": 8, "convert": [0, 1, 2, 79, 85, 86, 87, 116, 163, 245, 392, 393, 496, 497, 498], "convolut": [0, 99, 100, 101, 102, 103, 104, 105, 106, 341, 342, 343, 344, 345, 346, 348, 349], "convolv": [99, 100, 101, 102, 103, 104, 105], "convtranspose1d": 334, "convtranspose2d": 334, "convtranspose3d": 334, "coordin": [0, 214], "copi": [0, 1, 2, 6, 8, 239, 282, 497], "copy_inplac": 2, "copytyp": 2, "core": [1, 2, 3, 4, 5, 6, 7, 316, 334, 336, 337, 338, 339, 359, 366, 367, 368, 377, 380, 382, 385, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 435, 442, 461, 464, 466, 491, 492, 497, 498], "corner": 412, "correct": [2, 9, 470, 471, 472, 495, 496], "correctli": 39, "correl": [102, 348], "correspond": [0, 1, 2, 16, 18, 79, 94, 117, 119, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 192, 209, 211, 227, 241, 279, 280, 292, 299, 307, 312, 320, 494], "cos_first": 403, "cosh": [0, 441], "cosin": [0, 20, 21, 107, 108, 434, 480, 482, 494], "cosine_decai": [466, 482], "cosine_similarity_loss": 334, "cost": [9, 468, 492, 496], "costli": 496, "cot": 1, "cot_index": 1, "cotan": [2, 113], "cotang": [1, 2, 113, 311], "could": [6, 334], "count": [334, 482], "counter": 488, "cours": 494, "coursera": 478, "cout": [4, 493], "cov": 251, "covari": [251, 339], "cover": 2, "cpp": [2, 4], "cpu": [8, 9, 192, 193, 196, 500], "cpython": 2, "crash": [84, 491], "creat": [0, 2, 6, 9, 84, 124, 144, 173, 290, 334, 464, 466, 482, 491, 493, 495, 497], "create_additive_causal_mask": 6, "criteria": 2, "cross": [7, 102, 433, 435], "cross_entropi": [7, 334], "crowd": 6, "cry": 6, "cubic": 412, "cummax": 0, "cummin": 0, "cumprod": 0, "cumsum": 0, "cumul": [0, 84, 109, 110, 111, 112], "current": [6, 8, 9, 84, 91, 101, 104, 105, 128, 216, 218, 243, 322, 334, 468, 492, 496], "custom": [8, 113, 146, 411], "custom_decod": 411, "custom_encod": 411, "custom_funct": 1, "custom_kernel_myexp_float": 1, "custom_tim": 2, "cvpr": 348, "cxx": 4, "cycl": 490, "d": [0, 1, 2, 6, 101, 105, 118, 119, 149, 176, 195, 210, 214, 237, 295, 302, 304, 305, 306, 323, 343, 346, 349, 354, 360, 395, 467, 470, 472, 500], "d1": 500, "d2": 500, "d2fdx2": 494, "d_i": 363, "dampen": 479, "darwin": 2, "data": [0, 2, 7, 8, 11, 19, 126, 144, 159, 160, 166, 173, 199, 231, 235, 248, 257, 302, 304, 310, 314, 349, 413, 414, 415, 416, 417, 418, 419, 420, 491, 492, 493, 495, 497], "dataset": [5, 492, 496], "datatyp": 52, "dbuild_shared_lib": 9, "dcmake_build_typ": [4, 9], "ddof": [0, 76, 81, 288, 309], "deal": 491, "debug": [1, 3, 492], "debugg": 8, "decai": [468, 471, 473, 479, 480, 481, 484], "decay_r": [468, 481, 484], "decay_step": 480, "decent": 7, "decid": [320, 374], "decim": [0, 67, 266], "declar": 2, "decltyp": 1, "decod": 411, "decomposit": [189, 190, 197], "decor": [1, 113], "decoupl": 471, "deep": [339, 414, 415, 416, 417], "def": [1, 2, 5, 6, 7, 113, 141, 143, 146, 308, 334, 464, 491, 492, 493, 494, 495, 496, 497, 500], "default": [1, 2, 9, 15, 16, 17, 18, 19, 27, 28, 29, 30, 83, 84, 91, 95, 96, 99, 100, 101, 102, 103, 104, 105, 113, 114, 115, 117, 118, 119, 122, 123, 124, 126, 127, 128, 141, 143, 144, 146, 148, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 167, 168, 169, 172, 173, 177, 185, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 209, 211, 213, 214, 222, 223, 224, 227, 231, 235, 238, 239, 241, 243, 244, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 262, 263, 266, 273, 274, 282, 283, 286, 287, 288, 290, 292, 294, 299, 301, 302, 303, 304, 305, 306, 307, 308, 309, 312, 314, 316, 326, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 350, 353, 354, 356, 359, 360, 362, 363, 366, 367, 368, 370, 375, 377, 382, 384, 387, 390, 391, 392, 393, 395, 398, 403, 407, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 464, 467, 468, 469, 470, 471, 472, 473, 478, 479, 480, 488, 490, 491, 493, 494, 497, 499, 501], "default_devic": 501, "default_stream": 501, "defin": [1, 2, 5, 6, 7, 9, 113, 127, 146, 168, 191, 195, 244, 316, 319, 497], "definit": [113, 189, 190, 251], "degre": [0, 245, 446], "delta": [438, 467], "delv": [416, 417], "demonstr": 497, "denomin": [359, 434, 467, 469, 470, 471, 472, 478], "dens": [214, 500], "depend": [0, 2, 3, 4, 5, 9, 79, 195, 354, 360, 395, 491, 492, 495, 499, 500], "depth": [319, 338, 343, 346, 349, 368, 494], "dequant": [0, 243], "deriv": [2, 493, 494, 496], "descend": 372, "descent": [479, 491, 496], "describ": [2, 496], "descript": [2, 4, 6, 326], "design": [1, 5, 8, 488, 500], "destin": [0, 2, 61, 128, 229, 242], "destroi": 491, "detach": 494, "detail": [1, 2, 11, 222, 334, 348, 398, 403, 412, 414, 415, 416, 417, 467, 469, 470, 472, 473, 495, 498], "detect": 491, "determin": [0, 2, 119, 251, 322, 326, 381, 499], "dev": [2, 9], "develop": [2, 4, 9], "developer_dir": 9, "deviat": [0, 252, 288, 414, 416, 419], "deviatoin": 0, "devic": [1, 2, 8, 9, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 114, 115, 116, 117, 118, 119, 122, 123, 126, 127, 128, 129, 130, 131, 134, 135, 136, 138, 139, 140, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 170, 171, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 216, 223, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 309, 310, 313, 314, 315, 324, 500, 501], "device_info": 224, "devicetyp": 10, "df": 497, "dfdx": [493, 494, 495], "dft": [150, 151, 152, 153, 154, 155, 159, 160, 161], "dhwc": 349, "diag": [0, 197], "diagon": [0, 46, 118, 144, 302, 304, 305, 306], "dict": [95, 137, 142, 200, 216, 269, 270, 271, 316, 318, 380, 385, 388, 389, 464, 466, 474, 475, 477, 490, 493, 494, 499], "dict_kei": [320, 475], "dictionari": [6, 95, 141, 175, 200, 216, 269, 270, 318, 319, 322, 334, 374, 383, 388, 389, 476, 490, 499], "did": 6, "diff": 2, "differ": [8, 183, 291, 310, 445, 491, 493, 494], "differenti": [1, 2, 8, 340, 421], "difficult": 494, "difficulti": [414, 415], "dilat": [0, 99, 100, 101, 102, 103, 104, 105, 341, 342, 343, 344, 345, 346], "dim": [1, 6, 148, 149, 351, 355, 359, 361, 390, 392, 394, 398, 403, 411], "dimens": [0, 1, 2, 6, 16, 18, 27, 28, 63, 69, 79, 85, 86, 87, 95, 100, 101, 102, 104, 105, 119, 139, 148, 149, 157, 158, 160, 161, 163, 167, 168, 176, 189, 190, 192, 193, 194, 195, 196, 197, 198, 209, 210, 211, 213, 227, 241, 242, 243, 247, 256, 288, 292, 296, 299, 303, 309, 339, 341, 342, 343, 344, 345, 346, 348, 349, 353, 354, 355, 359, 360, 361, 390, 394, 395, 398, 411, 412, 426, 435, 491, 494], "dimension": [31, 145, 147, 150, 151, 152, 153, 154, 155, 159, 160, 161, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 351, 363, 366, 367, 368, 392, 393, 403, 495, 497], "dir": 4, "direct": [2, 6, 372, 473, 500], "directli": [2, 6, 84], "directori": [2, 4, 6, 9], "disabl": [120, 222, 491], "disable_compil": 491, "disappoint": 6, "discard": [6, 319], "discov": 9, "discoveri": 473, "discret": [106, 150, 151, 152, 153, 154, 155, 159, 160, 161, 351, 392], "discuss": 2, "disk": 6, "dispatch": 2, "dispatch_thread": 2, "dispatchthread": 1, "displai": 334, "distanc": [6, 446], "distribut": [8, 9, 246, 247, 248, 250, 251, 252, 257, 258, 363, 414, 415, 416, 417, 419, 420, 436, 439, 444, 446, 461], "diverg": 439, "divid": [0, 2, 39, 165, 243, 261, 492], "divis": [0, 129, 165, 243, 261], "divisor": [288, 309], "divmod": 0, "dloss_dw": 494, "dloss_dx": 494, "dlpack": 497, "dlvalu": 308, "dmlx_build_cpu": 9, "dmlx_build_gguf": 9, "dmlx_build_safetensor": 9, "dmlx_metal_debug": 3, "dmlx_metal_jit": 9, "do": [0, 2, 6, 9, 310, 334, 376, 387, 461, 464, 491, 492, 493, 494, 496], "doc": [2, 7, 492], "document": [2, 3, 4, 66, 80, 146, 269, 270, 326, 491, 493, 494, 495], "doe": [0, 2, 3, 6, 9, 217, 310, 318, 334, 491, 495, 496, 497], "doesn": [2, 334, 493], "domain": [257, 492], "don": [1, 9, 491, 500], "done": [334, 347, 394, 491, 492, 496, 497], "dot": [142, 194, 198, 299, 319, 379, 390], "doubl": [0, 6], "doubt": 6, "down": [6, 318], "downsampl": [336, 337, 338, 366, 367, 368], "dparam": 308, "draw": 247, "drop": 374, "dropout": [334, 348, 349, 384, 411, 491], "dropout2d": 334, "dropout3d": 334, "dst": 128, "dt": 135, "dtype": [0, 1, 2, 6, 12, 19, 31, 38, 39, 79, 82, 126, 127, 144, 146, 162, 163, 166, 173, 183, 185, 192, 193, 195, 196, 199, 235, 248, 250, 251, 252, 254, 257, 258, 279, 280, 302, 304, 307, 310, 314, 326, 382, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 435, 442, 480, 481, 482, 483, 484, 491, 492, 493, 494, 495, 497, 498, 499], "dtypecategori": [183, 326], "dual": 441, "duchi": 469, "duplic": 493, "dure": [3, 95, 347, 348, 349, 412, 497], "dx": 113, "dy": 113, "dyld": 492, "dyld_library_path": 492, "dylib": 2, "dynam": [0, 493, 496], "e": [2, 7, 9, 113, 135, 146, 167, 168, 184, 275, 339, 341, 342, 343, 344, 345, 346, 348, 349, 355, 359, 361, 375, 394, 431, 432, 454, 455, 460, 466, 469, 491, 493, 496, 501], "e5": 326, "e8": 326, "each": [0, 1, 2, 69, 117, 137, 148, 168, 183, 189, 190, 192, 193, 194, 197, 198, 210, 214, 238, 243, 244, 247, 262, 271, 272, 283, 300, 303, 310, 312, 313, 348, 349, 351, 354, 355, 360, 395, 398, 411, 433, 435, 488, 491, 492, 493, 496], "eager": 496, "earli": 348, "earlier": 2, "eas": 6, "easi": [2, 334, 492], "easier": [1, 142, 496], "edg": [94, 238, 412, 491], "edit": [9, 389], "effect": [348, 491, 496], "effici": [6, 8, 167, 348, 398, 496, 498], "eigenvalu": [192, 193], "eigenvector": 192, "einstein": [131, 132], "einsum": 132, "either": [9, 14, 66, 79, 80, 88, 89, 90, 94, 129, 130, 134, 165, 170, 171, 175, 186, 187, 188, 195, 205, 210, 212, 228, 230, 234, 240, 261, 264, 291, 308, 337, 338, 367, 368, 400, 412, 416, 417, 497], "elem": [1, 146], "elem_to_loc": [1, 2], "element": [0, 1, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 29, 71, 84, 88, 89, 90, 93, 107, 108, 109, 110, 111, 112, 117, 129, 130, 134, 135, 136, 138, 140, 144, 164, 165, 168, 170, 171, 177, 178, 179, 180, 181, 182, 186, 187, 188, 201, 202, 203, 204, 205, 206, 207, 208, 212, 214, 228, 230, 232, 234, 239, 240, 243, 244, 260, 261, 262, 264, 265, 267, 275, 276, 277, 278, 284, 285, 291, 295, 297, 298, 301, 308, 310, 313, 340, 347, 348, 349, 354, 358, 360, 369, 391, 395, 398, 402, 421, 428, 429, 431, 432, 447, 448, 450, 453, 454, 455, 456, 491, 494], "elementwis": [1, 97, 98], "elif": 6, "ellipsi": 495, "elman": 395, "els": [0, 2, 6, 334, 375, 492, 496], "elsewher": [304, 495], "elu": [334, 451], "emb": [6, 351, 392, 403], "embed": [6, 316, 334, 392, 398, 403, 434], "empti": 251, "enabl": [3, 6, 9, 95, 133, 479], "enclos": 493, "encod": [2, 148, 398, 403, 411, 435], "encount": [2, 494], "end": [119, 190, 220, 243, 265, 354, 360, 409, 427, 438, 445, 451, 457, 458, 480, 483, 493], "end_axi": [0, 50, 163], "end_encod": 2, "endif": 2, "endl": [4, 493], "endswith": 375, "enhanc": [6, 398, 496], "enjoi": 2, "enough": [2, 496], "ensur": [0, 1, 2, 9, 146, 318, 441, 492, 493], "ensure_row_contigu": [1, 146], "enter": 6, "entir": [16, 18, 27, 28, 209, 211, 213, 227, 241, 288, 292, 309, 348, 349], "entri": [0, 253, 307, 348, 349], "entropi": [7, 433, 435], "enumer": 334, "environ": [9, 120, 133, 492], "ep": [5, 145, 147, 339, 355, 359, 361, 394, 434, 436, 446, 466, 467, 468, 469, 470, 471, 472, 478], "epoch": 7, "epsilon": [339, 355, 359, 361, 394, 434, 436, 467, 469, 470, 471, 472, 478], "epsilon_1": 468, "epsilon_2": 468, "equal": [0, 1, 17, 29, 83, 144, 171, 177, 188, 234, 239, 254, 283, 359, 363], "equal_nan": [0, 17, 83, 177], "equat": [131, 132], "equival": [0, 2, 32, 66, 80, 127, 130, 165, 168, 172, 295, 340, 350, 352, 356, 357, 358, 364, 365, 389, 391, 393, 396, 397, 399, 401, 404, 405, 406, 407, 408, 410], "erf": [0, 136, 491], "erfinv": 0, "error": [0, 2, 9, 124, 135, 136, 223, 224, 283, 352, 423, 424, 425, 441, 443, 491, 494, 497], "error_norm": 5, "estim": 472, "eta": 473, "etc": [2, 243, 334, 412, 492], "eval": [2, 3, 5, 6, 7, 334, 464, 466, 491, 492, 493, 494, 496, 498], "eval_cpu": 2, "eval_fn": 7, "eval_gpu": 2, "evalu": [2, 6, 7, 8, 128, 137, 184, 311, 334, 373, 384, 464, 466, 491, 493, 498], "even": [1, 2, 6, 95, 491, 493, 496, 497], "evenli": [0, 199], "everi": [243, 320, 466, 484, 494], "everyth": [6, 492], "everywher": 0, "exact": [424, 425], "exactli": [2, 6, 148, 377, 494], "exampl": [0, 3, 4, 5, 6, 7, 9, 19, 39, 113, 124, 141, 142, 143, 146, 149, 163, 175, 183, 185, 192, 193, 195, 196, 279, 280, 290, 295, 307, 318, 321, 322, 334, 336, 337, 338, 339, 359, 366, 367, 368, 375, 377, 384, 387, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 435, 442, 461, 466, 475, 480, 481, 482, 483, 484, 488, 494, 495, 496, 497, 498, 499], "exce": 318, "exceed": 223, "except": [8, 113, 144, 156, 157, 159, 160, 161, 355, 377, 493, 495, 497], "exclud": [242, 296], "exclus": [0, 84, 90], "execut": [2, 4, 9, 85, 86, 87, 185, 219, 497, 500], "execute_process": 4, "exist": [2, 3, 6, 375, 387], "exp": [0, 1, 140, 146, 205, 209, 248, 281, 340, 350, 402, 421, 422, 439, 451, 452, 456, 491, 493, 500], "exp_elementwis": [1, 146], "expand_dim": 0, "expect": [2, 6, 341, 342, 343, 344, 345, 346, 347, 348, 349, 403, 411, 436, 491, 495], "expens": 411, "expensive_fun": 496, "experiment": [141, 143, 175, 497], "explain": 2, "explicit": [2, 475, 488, 497], "explicitli": [167, 334, 488], "explor": 9, "expm1": 0, "exponenti": [0, 138, 140, 340, 350, 399, 421, 422, 451, 481], "exponential_decai": 466, "export": [8, 9, 141, 142, 175], "export_funct": 493, "ext_modul": 2, "extend": [2, 238], "extens": [8, 200, 225, 381, 499], "extern": 497, "extra": [1, 320, 321, 493], "extract": [0, 6, 46, 118, 119, 279, 334, 374, 464, 492], "extras_requir": 2, "extrem": [495, 496], "ey": [0, 6, 194, 198], "f": [0, 2, 5, 7, 113, 195, 334, 360, 471, 491, 497], "f_jvp": 113, "f_t": 360, "f_vjp": 113, "f_vmap": 113, "face": 6, "factor": [2, 15, 172, 189, 190, 196, 412, 435, 481, 484], "fail": 491, "fall": [2, 113], "fallback": 2, "fals": [0, 1, 2, 6, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 76, 77, 81, 83, 95, 102, 109, 110, 111, 112, 124, 141, 143, 146, 177, 183, 189, 190, 195, 198, 200, 209, 211, 213, 214, 223, 227, 241, 288, 292, 309, 313, 316, 319, 320, 321, 322, 326, 355, 359, 361, 363, 375, 377, 387, 390, 393, 398, 403, 411, 412, 433, 436, 468, 470, 471, 479, 493, 497], "famili": 6, "fan": [414, 415, 416, 417], "fan_in": [414, 415, 416, 417], "fan_out": [414, 415, 416, 417], "far": 466, "fast": [1, 8, 352, 425, 492, 500], "faster": [1, 2, 9, 130, 423, 433, 491, 494], "featur": [1, 8, 99, 100, 101, 102, 103, 104, 105, 148, 339, 354, 355, 359, 360, 361, 363, 393, 394, 395, 398, 411, 412, 491, 496], "feed": 6, "feed_forward": 6, "feedforward": [414, 415], "feel": 6, "fetch": 1, "few": [1, 2, 6, 7, 8, 9, 492, 493, 496, 498], "ffn": 6, "ffn_norm": 6, "fft": 8, "figur": 492, "file": [4, 6, 9, 141, 142, 143, 175, 200, 268, 269, 270, 271, 272, 377, 381, 492, 493, 494, 499], "file_or_weight": 377, "fill": [0, 2, 166, 236, 304, 315, 413, 414, 415, 416, 417, 419, 420], "filter": [0, 106, 341, 342, 343, 344, 345, 346, 370, 374], "filter_and_map": 334, "filter_fn": [370, 374], "final": [2, 4, 5, 6, 7, 172, 480, 483], "find": [2, 4, 5, 9, 492], "find_packag": [2, 4], "finder": 9, "fine": [488, 493, 496], "finetun": 334, "finish": 2, "finit": [0, 178, 231], "first": [0, 1, 2, 3, 4, 5, 6, 7, 9, 119, 122, 124, 163, 169, 183, 185, 186, 197, 206, 208, 210, 239, 256, 264, 293, 299, 302, 308, 319, 321, 322, 334, 337, 338, 355, 367, 368, 412, 434, 442, 468, 472, 475, 491, 493, 494, 497, 500], "first_lay": 496, "firt": 491, "fit": [2, 243, 500], "five": 491, "fix": [2, 6, 9, 491, 496], "flag": [2, 4, 9, 491, 497], "flat": [167, 168, 319, 323], "flat_param": 271, "flatten": [0, 29, 30, 109, 110, 111, 112, 195, 237, 239, 242, 262, 265, 282, 295, 296, 301, 319, 491], "flexibl": 8, "flexibli": 389, "flip": [0, 102, 106], "float": [0, 1, 2, 12, 15, 17, 19, 79, 145, 146, 147, 148, 149, 162, 165, 166, 172, 177, 183, 195, 231, 244, 246, 250, 252, 318, 326, 339, 347, 348, 349, 355, 359, 361, 370, 382, 394, 398, 403, 409, 411, 412, 413, 414, 415, 416, 417, 419, 420, 434, 435, 436, 438, 442, 445, 446, 457, 458, 467, 468, 469, 470, 471, 472, 473, 478, 479, 480, 481, 483, 484], "float16": [1, 2, 12, 146, 172, 200, 326, 370, 496, 497], "float16_t": [1, 2], "float32": [0, 1, 2, 12, 19, 144, 146, 149, 172, 173, 183, 192, 193, 195, 196, 199, 235, 248, 250, 251, 252, 257, 258, 280, 304, 314, 326, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 435, 442, 480, 481, 482, 483, 484, 491, 492, 493, 494, 495, 496, 497, 498, 499], "float64": 183, "floor": [0, 1, 165], "floor_divid": 0, "flow": [0, 289, 496], "flush": 2, "fn": [175, 317, 320, 321, 322, 498], "follow": [1, 2, 4, 6, 7, 8, 9, 19, 106, 117, 149, 167, 195, 238, 243, 321, 334, 424, 425, 439, 467, 468, 469, 472, 473, 479, 488, 491, 492, 493, 494, 500], "foo": 492, "food": 6, "forc": [6, 7, 334, 492, 498], "forg": [9, 492], "formal": [117, 243], "format": [6, 142, 200, 268, 269, 270, 271, 272, 497], "formul": [340, 350], "formula": 445, "forth": 412, "forward": [1, 2, 308, 491, 496], "found": [4, 374], "four": 339, "fourier": [150, 151, 152, 153, 154, 155, 159, 160, 161], "fourth": 493, "frac": [135, 243, 275, 339, 347, 348, 349, 355, 359, 361, 363, 394, 402, 414, 415, 416, 417, 434, 436, 438, 441, 452, 454, 455, 467, 469, 470, 471, 472, 478], "fraction": 19, "framework": [2, 8], "free": 222, "freez": [334, 387, 464], "freq": 148, "frequenc": [148, 398, 403], "frequent": [491, 496], "friend": 6, "fro": 195, "frobeniu": 195, "from": [0, 1, 2, 4, 6, 7, 8, 84, 116, 117, 119, 122, 123, 126, 127, 128, 146, 157, 158, 160, 161, 166, 167, 172, 175, 195, 200, 210, 214, 219, 222, 236, 243, 245, 246, 247, 248, 249, 250, 254, 257, 271, 279, 286, 289, 291, 295, 296, 301, 302, 313, 315, 319, 320, 321, 322, 323, 334, 363, 375, 377, 390, 414, 415, 416, 417, 419, 420, 436, 445, 461, 466, 490, 491, 492, 493, 494, 496, 497, 498, 499, 500], "from_embed": 392, "from_linear": 393, "front": [2, 493], "frozen": [334, 375, 385, 387, 393, 464], "fuction": 130, "full": [0, 1, 2, 7, 66, 80, 106, 146, 281, 388, 389, 436, 491, 492, 493, 496], "full_turn": 403, "fulli": [2, 8, 492, 497, 500], "fun": [95, 141, 143, 169, 184, 308, 311, 312, 491, 493, 495, 496, 500], "fun1": 496, "func": 395, "function": [0, 1, 2, 3, 5, 6, 7, 8, 17, 19, 84, 95, 113, 130, 135, 136, 141, 143, 146, 169, 175, 177, 184, 189, 190, 192, 193, 194, 195, 196, 197, 198, 210, 224, 275, 308, 311, 312, 317, 318, 320, 321, 322, 334, 340, 350, 352, 353, 356, 357, 358, 364, 365, 369, 371, 375, 382, 387, 391, 395, 396, 397, 399, 400, 401, 402, 404, 405, 406, 407, 408, 409, 410, 411, 423, 424, 425, 426, 427, 428, 429, 431, 432, 433, 447, 452, 454, 455, 456, 457, 458, 459, 461, 466, 475, 488, 490, 492, 495, 496, 497, 499], "functionexport": 143, "functool": 491, "further": [2, 9, 494], "fuse": [1, 491], "fusibl": 491, "futur": [6, 141, 143, 175, 393, 495, 496], "fx": 113, "g": [3, 9, 113, 146, 195, 243, 360, 460, 478, 479, 493, 496, 501], "g_t": [360, 467, 469, 470, 471, 472, 473, 478, 479], "gain": [414, 415, 416, 417], "gamma": [339, 355, 359, 361, 394, 414, 415, 416, 417], "gap": 1, "gate": [353, 354, 426], "gather": [0, 122, 167, 168], "gather_mm": [0, 168], "gather_qmm": 0, "gaurante": 310, "gaussian": [5, 352, 423, 424, 425, 436], "gaussian_nll_loss": 334, "gelu": [334, 424, 425, 491], "gelu_approx": [334, 352, 423], "gelu_fast_approx": [334, 352, 423], "geluapprox": 352, "gelufast": 352, "gener": [0, 1, 2, 3, 5, 12, 19, 102, 144, 146, 157, 158, 199, 214, 246, 251, 252, 253, 254, 257, 258, 411, 488, 491, 495, 496, 501], "general_": 2, "generate_stub": 9, "geq": [409, 458], "get": [2, 5, 7, 9, 100, 101, 102, 104, 105, 114, 115, 162, 216, 217, 218, 219, 249, 334, 491, 493, 494, 496, 500], "get_cache_memori": 215, "get_command_encod": 2, "get_kernel": 2, "gguf": [9, 200, 269, 499], "gh": 1, "gii": 1, "git": 9, "github": [5, 7, 9, 491], "give": [2, 6, 7, 29, 491], "given": [0, 2, 9, 16, 18, 29, 39, 84, 92, 94, 96, 109, 110, 111, 112, 117, 119, 132, 137, 139, 149, 150, 151, 152, 153, 154, 155, 159, 160, 161, 166, 167, 195, 209, 211, 213, 222, 227, 231, 233, 241, 251, 253, 254, 265, 266, 274, 281, 283, 288, 292, 294, 300, 301, 302, 304, 305, 306, 309, 324, 347, 374, 390, 434, 436, 442], "gix": 1, "gix_mult": 1, "giy_mult": 1, "global": [120, 122, 123, 124, 126, 127, 128, 133, 255, 318, 488, 491], "glorot": [414, 415], "glorot_norm": 334, "glorot_uniform": 334, "glu": [6, 334], "gm": 1, "gn": 1, "go": [2, 6, 494], "golub": 195, "good": [2, 9, 466, 491, 492, 500], "goroshin": 348, "gower": 6, "gpu": [1, 3, 8, 9, 216, 495, 500], "gputrac": [3, 225], "grad": [2, 5, 7, 113, 308, 318, 466, 474, 491, 492, 493, 494, 495, 496, 498], "grad_fn": [5, 491, 494], "gradient": [0, 5, 7, 113, 169, 289, 308, 317, 318, 334, 375, 388, 393, 411, 441, 464, 466, 467, 468, 470, 471, 472, 473, 474, 477, 479, 491, 492, 494, 495, 496, 497, 498], "grain": 488, "graph": [2, 6, 7, 8, 142, 493, 494], "great": 3, "greater": [0, 6, 29, 140, 171, 239, 318, 409, 458], "greater_equ": 0, "grep": 9, "grid": [2, 146, 214], "grid_dim": 2, "grid_grad": 1, "grid_idx": 1, "grid_sampl": 1, "grid_sample_grad": 1, "grid_sample_ref": 1, "grid_sample_vjp": 1, "grid_shap": 1, "grid_siz": 1, "ground": [5, 6, 435, 445], "group": [0, 1, 99, 100, 101, 102, 103, 104, 105, 117, 122, 123, 124, 126, 127, 128, 149, 168, 243, 244, 310, 316, 341, 342, 355, 392, 393, 492], "group_dim": 2, "group_siz": [0, 117, 168, 243, 244, 316, 392, 393], "groupnorm": 334, "grow": 496, "gru": 334, "guid": [2, 4, 8, 493], "gw": 1, "h": [1, 2, 4, 99, 100, 101, 103, 104, 105, 195, 339, 342, 343, 345, 346, 348, 349, 354, 360, 395, 494, 496], "h_": [354, 360, 395], "h_in": 1, "h_stride": 1, "h_t": [354, 360, 395], "ha": [2, 3, 6, 7, 8, 9, 79, 95, 119, 128, 156, 157, 159, 160, 161, 169, 189, 190, 192, 193, 194, 197, 198, 214, 218, 247, 339, 354, 360, 363, 395, 464, 466, 491, 493, 495, 496, 498, 500], "had": 6, "hadamard": [0, 172], "hadamard_transform": 0, "half": [2, 19, 254, 258, 398, 496], "halv": [353, 426], "hand": [6, 494, 496], "handi": 494, "handl": [2, 334, 491], "happen": [2, 6, 145, 411, 466, 491, 496], "happi": 6, "hard": 6, "hard_shrink": [334, 356], "hard_tanh": [334, 357], "hardcod": 491, "hardshrink": [334, 427], "hardswish": 334, "hardtanh": [334, 428], "hat": [117, 243], "have": [0, 1, 2, 6, 9, 17, 83, 85, 86, 87, 91, 113, 122, 149, 157, 158, 160, 161, 168, 177, 210, 225, 247, 310, 319, 360, 390, 400, 473, 475, 490, 491, 492, 493, 495, 496, 500], "haven": 6, "hazan": 469, "he": [6, 416, 417], "he_norm": 334, "he_uniform": 334, "head": [149, 390, 411], "header": [2, 146], "heart": 6, "heavi": 6, "height": [337, 338, 339, 342, 343, 345, 346, 348, 349, 367, 368], "hello": [319, 323], "help": [2, 6, 491, 500], "helper": [6, 146, 491], "henc": [0, 2, 243, 491], "hendryck": 425, "here": [2, 6, 466, 491, 493, 494, 496, 499, 500], "hermitian": [192, 193], "hf": 360, "hg": 360, "hh": 395, "hi": [6, 360], "hidden": [354, 360, 395, 411], "hidden_dim": [7, 464, 466], "hidden_s": [354, 360, 395], "hierarchi": 326, "high": [254, 258, 334, 351, 420, 461], "high_pad_s": 0, "higher": [2, 176, 224, 442, 494], "highli": 9, "him": 6, "hing": 437, "hinge_loss": 334, "hinton": 478, "hit": 2, "hn": 354, "ho": 360, "hold": [2, 6, 11, 12, 195, 491], "homebrew": 492, "hopkin": 195, "host": 2, "host1": 492, "host2": 492, "host_nam": 1, "hostfil": 492, "hostnam": 492, "hot": 435, "hour": 6, "how": [2, 4, 6, 7, 334, 336, 337, 338, 341, 342, 343, 344, 345, 346, 351, 366, 367, 368, 392, 412, 474, 491, 495, 500], "howev": [2, 113, 334, 352, 355, 475, 488, 491, 492, 496, 497], "hr": 354, "http": [355, 359, 361, 369, 394, 425, 447], "huber": 438, "huber_loss": 334, "human": [416, 417], "hundr": 9, "hurri": 6, "hutter": 471, "hyperbol": [0, 21, 23, 26, 108, 278, 298, 410, 459], "hz": 354, "i": [0, 1, 2, 3, 4, 6, 7, 8, 9, 17, 19, 29, 38, 79, 84, 94, 100, 101, 102, 104, 105, 106, 109, 110, 111, 112, 113, 118, 119, 122, 123, 125, 126, 127, 128, 130, 137, 141, 143, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 165, 166, 167, 168, 172, 175, 177, 178, 183, 184, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 200, 205, 209, 210, 214, 220, 223, 224, 238, 239, 242, 243, 244, 251, 252, 253, 263, 265, 268, 269, 270, 275, 281, 283, 288, 289, 294, 295, 296, 299, 302, 303, 307, 308, 309, 310, 311, 312, 313, 316, 318, 319, 320, 321, 322, 326, 328, 334, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 347, 348, 349, 352, 354, 355, 359, 360, 361, 363, 366, 367, 368, 374, 375, 381, 383, 384, 386, 387, 389, 390, 391, 393, 394, 395, 398, 403, 409, 411, 412, 416, 417, 423, 425, 433, 434, 436, 441, 442, 445, 446, 448, 453, 458, 464, 466, 468, 470, 471, 473, 474, 475, 480, 482, 483, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "i386": 9, "i_n": 1, "i_nw": 1, "i_s": 1, "i_sw": 1, "i_t": 360, "iclr": [470, 471, 472], "id": [7, 9], "idea": [494, 496], "idempot": [375, 387], "ident": [0, 113, 128, 144, 289, 334, 384], "identifi": [2, 319, 490], "idim": 7, "idiom": [7, 491], "idx": [39, 495], "ie": [387, 492], "ieee": 326, "ignor": [6, 39, 94, 95, 137, 468], "ih": 395, "ii": 1, "ij": 214, "imag": [0, 342, 343, 345, 346, 348, 349, 412], "imagenet": [416, 417], "imaginari": 174, "immedi": [6, 370], "implement": [0, 1, 5, 7, 148, 149, 195, 351, 374, 390, 398, 400, 403, 409, 411, 412, 458, 467, 468, 469, 472, 473, 474, 486, 491, 494], "impli": 310, "implicit": [488, 491, 494], "implicitli": 496, "import": [2, 3, 5, 6, 7, 9, 113, 124, 172, 175, 195, 271, 308, 319, 320, 321, 322, 323, 334, 336, 337, 338, 339, 359, 366, 367, 368, 377, 412, 433, 435, 442, 461, 464, 466, 491, 492, 494, 495, 496, 497, 498], "import_funct": 493, "imported_ab": 493, "imported_fun": 493, "imported_funct": 493, "improv": [1, 2, 3, 6, 433, 467, 468, 469, 470, 471, 472, 478, 491, 492], "in_ax": [312, 494], "in_channel": [341, 342, 343, 344, 345, 346], "in_dim": [334, 464], "in_proj": 464, "inci": 2, "includ": [1, 2, 4, 109, 110, 111, 112, 142, 146, 217, 218, 223, 361, 371, 383, 393, 436, 466, 491, 493, 494, 495, 498, 499, 501], "include_dir": 2, "inclus": [0, 42, 43, 44, 45, 109, 110, 111, 112, 163], "incom": 2, "inconveni": 491, "incorpor": 497, "incorrect": 497, "increas": 224, "increment": 19, "incur": [6, 9], "incx": 2, "independ": [121, 348, 349], "index": [0, 1, 2, 8, 10, 29, 39, 139, 144, 169, 214, 239, 279, 280, 295, 296, 308], "indic": [0, 2, 17, 27, 28, 29, 30, 39, 167, 168, 169, 177, 178, 179, 180, 181, 182, 183, 197, 242, 279, 280, 283, 295, 296, 308, 384, 386, 435, 442, 482, 495], "indices_or_sect": [72, 283], "indirectli": 497, "individu": [334, 348, 349], "ineffici": [495, 496], "inexact": [12, 183], "inf": [195, 231, 390], "infer": [8, 166, 200, 302, 307, 492, 493], "infin": [0, 179, 181, 182, 231, 366, 367, 368, 472], "infinit": [17, 177, 178], "info": [6, 9], "inform": [3, 4, 6, 7, 9, 132, 162, 216, 269, 270, 326, 334, 339, 352, 390, 493, 494, 500], "inherit": [7, 490], "inifn": 179, "init": [334, 391, 461, 466, 480, 481, 483, 484, 492], "init_fn": [413, 414, 415, 416, 417, 418, 419, 420, 461], "init_valu": 1, "initi": [1, 3, 5, 6, 124, 322, 334, 339, 355, 359, 361, 363, 391, 394, 413, 414, 415, 416, 417, 418, 419, 420, 464, 475, 480, 481, 483, 484, 491, 492, 493, 496], "initializer_list": 0, "inject": 0, "inlin": 0, "inner": [0, 491], "inorm": 359, "inp": [1, 146], "inp_ndim": 1, "inp_shap": 1, "inp_strid": 1, "inplac": [2, 9], "input": [0, 1, 2, 5, 6, 13, 14, 15, 16, 17, 18, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 116, 118, 119, 122, 123, 128, 129, 130, 131, 132, 134, 135, 136, 138, 139, 140, 141, 142, 143, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 167, 168, 169, 170, 171, 172, 174, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 227, 228, 229, 230, 231, 232, 234, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 253, 256, 259, 260, 261, 262, 263, 264, 265, 266, 267, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 295, 296, 297, 298, 299, 300, 301, 302, 303, 305, 306, 307, 308, 309, 310, 312, 313, 315, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 348, 349, 351, 353, 354, 355, 359, 360, 361, 363, 366, 367, 368, 390, 393, 394, 395, 398, 409, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 436, 437, 438, 439, 441, 442, 444, 446, 458, 461, 491, 493, 494, 495, 498, 499], "input_dil": [0, 102], "input_dim": [7, 334, 363, 393], "input_nam": [1, 146], "input_s": [354, 360, 395], "inputs1": 442, "inputs2": 442, "insert": [119, 139, 500], "insid": [491, 493], "inspect": [3, 491, 498], "inspir": 8, "instabl": 446, "instal": [2, 4], "instanc": [6, 39, 113, 243, 323, 334, 359, 370, 371, 372, 375, 377, 378, 379, 384, 387, 388, 389, 400, 464, 497], "instancenorm": 334, "instanti": [1, 2, 7, 496], "instantiate_kernel": 2, "instead": [2, 9, 113, 334, 389, 403, 492, 494, 496], "instruct": [4, 493], "int": [0, 1, 2, 4, 6, 7, 10, 16, 18, 19, 27, 28, 29, 30, 34, 35, 36, 37, 42, 43, 44, 45, 46, 47, 50, 57, 58, 59, 60, 61, 64, 67, 69, 72, 75, 76, 77, 78, 79, 81, 84, 91, 92, 96, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 117, 118, 119, 126, 127, 128, 132, 139, 144, 148, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 166, 168, 169, 173, 183, 191, 195, 199, 209, 211, 213, 216, 217, 218, 219, 222, 223, 224, 227, 229, 235, 238, 239, 241, 242, 243, 244, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 262, 263, 265, 266, 279, 280, 281, 282, 283, 286, 287, 288, 292, 293, 295, 296, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 312, 314, 316, 334, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 351, 353, 354, 355, 359, 360, 361, 363, 366, 367, 368, 390, 392, 393, 394, 395, 398, 403, 411, 426, 434, 435, 439, 444, 446, 464, 480, 482, 483, 484], "int16": 326, "int32": [0, 1, 12, 19, 39, 163, 183, 185, 195, 254, 279, 307, 326, 412, 495, 498], "int64": [12, 326], "int64_t": 2, "int8": [12, 326], "int_0": 135, "integ": [0, 12, 165, 167, 168, 183, 195, 216, 238, 243, 244, 246, 253, 254, 283, 295, 299, 312, 326, 351, 382, 482, 495], "integr": [19, 295, 496], "intend": [0, 491], "interact": 411, "interest": 500, "interfac": 2, "intermedi": 497, "intern": 339, "interpol": 412, "interpret": 4, "interv": [19, 199, 254, 258], "introduc": [0, 265], "intuit": 334, "invalid": [0, 84], "invers": [0, 20, 21, 22, 23, 24, 25, 26, 136, 153, 154, 155, 156, 157, 158, 190, 194, 198], "invert": 0, "involv": [466, 491], "iogpu": 224, "iostream": 4, "ip": 492, "is_avail": 124, "is_equival": 2, "is_floating_point": 2, "is_leaf": [319, 320, 321, 322], "is_leaf_fn": 374, "isclos": 0, "isfinit": 0, "ish": 6, "ishmael": 6, "isinf": 0, "isnan": 0, "isneginf": 0, "isposinf": 0, "issu": [492, 494, 497], "issubdtyp": [12, 326], "item": [0, 2, 5, 6, 7, 320, 466, 493, 496, 497, 498], "iter": [5, 7, 197, 320, 321, 488, 491, 496], "iterm": 9, "itertool": [6, 320], "its": [0, 1, 2, 9, 149, 190, 210, 239, 256, 304, 317, 323, 334, 393, 466, 470, 471, 472, 492, 496, 497, 500], "itself": [2, 316, 475], "ix": 1, "ix_n": 1, "ix_nw": 1, "ix_s": 1, "ix_sw": 1, "iy_n": 1, "iy_nw": 1, "iy_s": 1, "iy_sw": 1, "j": [6, 9, 195, 348, 469, 470, 472], "j8": 2, "jacobian": [2, 184, 311, 498], "jain": 348, "jax": [8, 488], "jit": 146, "jmlr": 469, "jnp": 497, "john": 195, "join": 482, "join_schedul": 466, "jointli": 251, "just": [2, 4, 7, 361, 491, 493, 495], "jvp": [2, 113, 498], "k": [0, 6, 46, 91, 118, 144, 149, 167, 172, 301, 304, 305, 306, 363, 375], "kaim": 417, "keep": [2, 16, 18, 27, 28, 209, 211, 213, 227, 241, 288, 292, 309, 334, 374, 494, 496], "keepdim": [0, 16, 18, 27, 28, 34, 35, 36, 37, 57, 58, 59, 60, 64, 76, 77, 81, 195, 209, 211, 213, 227, 241, 281, 288, 292, 309], "kei": [1, 3, 6, 141, 149, 175, 216, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 319, 320, 374, 375, 387, 390, 475, 488, 490, 493, 494], "kept": 224, "kernel": [2, 8, 9, 99, 100, 101, 102, 103, 104, 105, 146, 336, 366, 491, 495], "kernel_dil": [0, 102], "kernel_s": [336, 337, 338, 341, 342, 343, 344, 345, 346, 366, 367, 368], "key_cach": 6, "key_input_dim": 390, "key_proj": 6, "keyword": [141, 169, 271, 272, 308, 320, 334, 488, 493, 499, 501], "kind": 6, "kingma": [470, 472], "kl_div_loss": 334, "kname": 2, "know": [2, 6], "known": [401, 453], "kron": 0, "kroneck": [0, 185], "kth": [0, 29, 239], "kullback": 439, "kw_onli": 2, "kwarg": [11, 121, 141, 142, 175, 271, 272, 324, 493, 501], "l": [6, 7, 189, 190, 192, 193, 334, 339, 341, 344, 354, 360, 395, 445], "l1": [308, 438, 440, 441, 445], "l1_loss": 334, "l2": [438, 441, 479], "l2_loss": 334, "l_": 438, "la": 195, "label": [3, 5, 435, 442], "label_smooth": 435, "lack": 495, "lambd": [356, 407, 427, 457], "lambda": [320, 321, 322, 334, 356, 370, 375, 382, 407, 427, 451, 457, 467, 468, 469, 470, 471, 472, 473, 478, 479, 491, 492, 493, 494], "languag": [1, 2, 4], "larg": [6, 334, 390, 441, 491, 492, 493, 496], "larger": [1, 148, 224, 398, 473], "largest": [195, 231, 301], "lasso": 308, "last": [0, 1, 6, 30, 79, 145, 147, 152, 155, 157, 158, 160, 161, 163, 167, 168, 176, 189, 190, 192, 193, 194, 196, 197, 198, 210, 219, 247, 282, 299, 310, 341, 342, 343, 344, 345, 346, 348, 349, 355, 412, 497], "latenc": 492, "later": [3, 9, 466], "launch": [1, 2, 124, 492, 495], "layer": [8, 145, 316, 334, 336, 337, 338, 348, 349, 354, 355, 360, 361, 363, 366, 367, 368, 384, 389, 392, 393, 395, 400, 411, 460, 464, 493], "layer_s": 7, "layernorm": 334, "layout": 1, "lazi": [8, 464, 498], "lazili": [6, 334], "lceil": 91, "ld": [354, 360, 395], "lead": [0, 19, 84, 491], "leaf": [95, 316, 319, 320, 321, 322, 374], "leaf_modul": 334, "leaki": [362, 430], "leaky_relu": 334, "leakyrelu": 334, "learn": [5, 7, 8, 339, 355, 359, 361, 391, 394, 466, 467, 468, 469, 470, 471, 472, 473, 478, 479], "learnabl": [341, 342, 343, 344, 345, 346, 400], "learning_r": [7, 466, 467, 468, 469, 470, 471, 472, 473, 475, 478, 479, 480, 481, 482, 483, 484, 491], "least": [6, 85, 86, 87, 94, 189, 190, 192, 193, 194, 196, 197, 198, 243], "leav": [2, 137, 320, 321, 322], "lectur": 478, "lecun": 348, "left": [0, 6, 148, 186, 195, 243, 265, 352, 398, 412, 424, 425, 436, 438, 446], "left_shift": 0, "leibler": 439, "len": [6, 152, 155, 158, 161, 172, 482], "length": [6, 286, 339, 341, 344, 354, 360, 395, 482], "leq": [438, 451], "less": [0, 1, 6, 29, 188, 224, 239, 398, 445], "less_equ": 0, "let": [1, 2, 5, 6, 190, 491, 493, 494, 496, 497], "level": [0, 167, 168, 416, 417], "lh": [354, 360, 395], "lhs_indic": [0, 167, 168], "lhs_mask": 91, "lib": 492, "libmlx": 9, "libmlx_ext": 2, "libmpi": 492, "librari": [2, 4, 9, 328, 334, 493], "like": [2, 6, 8, 127, 141, 143, 175, 183, 236, 315, 349, 441, 475, 477, 491, 492, 493, 494, 496, 497, 498, 500], "likelihood": [436, 444], "limit": [0, 2, 94, 222, 223, 224, 495], "linalg": 172, "line": [6, 492, 493, 496, 497], "linear": [0, 2, 6, 7, 8, 316, 320, 334, 340, 350, 352, 353, 362, 377, 393, 395, 396, 397, 399, 401, 412, 421, 422, 423, 424, 425, 426, 430, 449, 450, 451, 453, 461, 464, 475, 483, 491, 493], "linear1": 6, "linear2": 6, "linear3": 6, "linear_schedul": [466, 482], "linearli": 390, "link": [2, 4, 9], "linspac": 0, "lion": 466, "list": [1, 6, 11, 16, 18, 31, 72, 79, 84, 85, 86, 87, 92, 95, 96, 102, 132, 137, 146, 151, 152, 154, 155, 157, 158, 160, 161, 166, 169, 184, 195, 209, 211, 213, 214, 227, 235, 238, 241, 246, 247, 248, 250, 251, 252, 254, 257, 258, 269, 281, 283, 287, 288, 292, 299, 300, 303, 308, 309, 311, 314, 319, 322, 323, 334, 375, 377, 378, 379, 380, 385, 387, 388, 389, 464, 466, 470, 471, 472, 473, 482, 490, 491, 492, 493, 494, 496], "liter": [2, 238, 412, 416, 417, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446], "littl": 6, "liu": 6, "live": [8, 146, 500], "ll": [1, 5, 7, 438, 491, 494], "llama": 6, "llamaattent": 6, "llamaencoderlay": 6, "llm": 8, "load": [7, 8, 328, 377, 492], "load_weight": [334, 496], "loader": 7, "loader_path": 2, "loan": 195, "loc": [1, 250, 252], "local": [334, 348, 492], "locat": [0, 2, 4, 84, 279, 280, 388, 389, 492, 500], "log": [0, 203, 205, 209, 364, 365, 431, 432, 433, 436, 439, 441, 444, 456], "log10": 0, "log1p": 0, "log2": 0, "log_cosh_loss": 334, "log_sigmoid": [334, 364], "log_softmax": [334, 365], "logaddexp": 0, "logarithm": [0, 201, 202, 203, 204], "logcosh": 441, "logic": [0, 2, 206, 207, 208], "logical_and": 0, "logical_not": 0, "logical_or": 0, "logist": [0, 5, 275, 425, 453], "logit": [6, 247, 433, 435, 491], "logsigmoid": 334, "logsoftmax": 334, "logsumexp": 0, "long": 6, "longer": [6, 106, 494], "look": [2, 6, 492], "lookup": 351, "loop": [6, 7, 491, 492, 494, 496], "loshchilov": 471, "loss": [5, 7, 308, 334, 466, 491, 492, 494, 496], "loss_and_grad": 334, "loss_and_grad_fn": [7, 466, 491, 494], "loss_fn": [5, 7, 466, 491, 494], "loss_grad_fn": 492, "lot": [492, 494], "low": [254, 258, 420, 461], "low_pad_s": 0, "lower": [189, 190, 192, 193, 198, 243, 254, 257, 258, 304, 420], "lr": [5, 473], "lr_schedul": [480, 481, 482, 484], "lstm": 334, "lto": 2, "lu": 6, "luckili": 496, "lvalu": 308, "m": [0, 2, 4, 6, 9, 91, 144, 167, 172, 195, 304, 467, 491], "m1": [1, 6, 491, 494, 500], "m10": 326, "m7": 326, "m_": [470, 471, 472, 473], "m_t": [470, 471, 472, 473], "mac": 492, "machin": [6, 8, 9, 478, 492], "maco": [9, 224], "macosx": 9, "made": [6, 328], "mai": [2, 4, 141, 143, 175, 195, 316, 348, 492, 494, 495], "main": [4, 8, 119, 144, 146, 302, 320, 321, 334, 492], "maintain": [348, 349, 473], "major": [0, 2], "make": [1, 2, 3, 4, 6, 7, 9, 142, 143, 210, 233, 274, 334, 480, 481, 483, 484, 491, 496, 498, 500], "make_shar": 2, "malloc_or_wait": 2, "man": 6, "manag": [290, 488, 492, 493, 500], "mani": [2, 84, 283, 341, 342, 343, 344, 345, 346, 351, 392, 491, 492, 493, 496], "manual": 334, "map": [2, 7, 39, 200, 320, 351, 370, 493], "map_fn": [370, 374], "map_torch_to_mlx": 6, "margin": [442, 446], "margin_ranking_loss": 334, "mask": [0, 6, 91, 149, 384, 390, 495], "mask_lh": [0, 91], "mask_n": 1, "mask_nw": 1, "mask_out": [0, 91], "mask_rh": [0, 91], "mask_s": 1, "mask_sw": 1, "matadata": 200, "match": [9, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 217, 377, 412, 435, 492, 495, 497], "materi": [6, 8], "math": [6, 446, 491], "mathbf": 190, "mathcal": 363, "mathemat": 195, "mathrm": [135, 275, 359], "matmul": [0, 167, 500], "matric": [195, 196, 197], "matrix": [0, 5, 15, 46, 91, 117, 118, 144, 167, 168, 172, 173, 189, 190, 192, 193, 194, 195, 196, 197, 198, 210, 214, 243, 244, 251, 392, 393, 418, 461], "matter": [6, 334, 493], "max": [0, 1, 2, 195, 212, 340, 366, 367, 368, 391, 421, 428, 429, 434, 436, 437, 442, 446, 448, 450, 468, 472, 491, 494, 500], "max_buffer_s": 216, "max_freq": 403, "max_i": 243, "max_norm": 318, "max_recommended_working_set_s": [216, 224], "max_val": 428, "maximum": [0, 7, 27, 39, 94, 109, 219, 223, 318, 334, 362, 366, 367, 368, 396, 403, 424, 425, 430, 449, 464, 496], "maxpool1d": 334, "maxpool2d": 334, "maxpool3d": 334, "maxtotalthreadsperthreadgroup": 2, "mca": 492, "md": 195, "me": 6, "mean": [0, 1, 5, 6, 7, 147, 250, 251, 252, 308, 334, 339, 355, 375, 394, 419, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 491, 494, 497], "meant": 113, "measur": 500, "mechan": 411, "medic": 349, "meet": 9, "member": [2, 334, 380, 385], "memori": [0, 1, 2, 8, 84, 215, 217, 218, 219, 221, 222, 223, 224, 411, 464, 468, 491, 496, 497], "memory_order_relax": 1, "memory_s": [216, 224], "memoryview": [496, 497], "merg": 491, "meshgrid": 0, "metadata": [5, 200, 269, 270], "metal": [2, 4, 8, 146], "metal_captur": 3, "metal_kernel": 1, "metal_path": 9, "metallib": [2, 9], "method": [2, 6, 10, 11, 31, 113, 121, 162, 316, 324, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 381, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 464, 467, 468, 469, 470, 471, 472, 473, 475, 478, 479, 486], "millisecond": [9, 491, 500], "min": [0, 2, 195, 228, 340, 391, 421, 428, 429, 448, 450], "min_freq": 403, "min_i": 243, "min_val": 428, "mind": [2, 6], "mine": 6, "minibatch": 7, "minim": 492, "minimum": [0, 28, 39, 94, 110, 403, 433, 434], "minsizerel": 9, "minu": 140, "minut": 6, "mish": 334, "mismatch": 493, "miss": [377, 493, 499], "mix": 495, "mkdir": [3, 9], "ml": 9, "mlp": [7, 334, 411, 466], "mlp_dim": [6, 411], "mlx": [1, 3, 5, 6, 7, 9, 328, 334, 461, 464, 466, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "mlx_build_acceler": 4, "mlx_build_benchmark": 9, "mlx_build_cpu": 9, "mlx_build_exampl": 9, "mlx_build_gguf": 9, "mlx_build_met": [2, 4, 9], "mlx_build_metallib": 2, "mlx_build_python_bind": 9, "mlx_build_safetensor": 9, "mlx_build_test": 9, "mlx_cxx_flag": 4, "mlx_disable_compil": [120, 133, 491], "mlx_ext": 2, "mlx_ext_metallib": 2, "mlx_found": 4, "mlx_include_dir": [2, 4], "mlx_librari": 4, "mlx_metal_debug": [3, 9], "mlx_metal_jit": 9, "mlx_root": 4, "mlx_sample_extens": 2, "mlx_trace": 3, "mlxfn": [141, 143, 175, 493], "mnist": 7, "mode": [0, 1, 2, 106, 238, 373, 384, 386, 412, 416, 417, 492], "model": [5, 7, 8, 271, 316, 317, 320, 321, 334, 370, 373, 375, 377, 381, 384, 386, 387, 388, 390, 411, 461, 464, 466, 474, 475, 477, 491, 492, 493, 496], "modest": 2, "modif": 497, "modifi": 497, "modul": [2, 4, 6, 7, 316, 317, 400, 411, 461, 477, 490, 491, 496], "moment": [6, 468, 472], "momentum": [339, 473, 475, 479, 491], "monei": 6, "monoton": 447, "more": [1, 2, 3, 4, 7, 11, 79, 119, 141, 167, 189, 190, 192, 193, 194, 197, 198, 210, 222, 223, 269, 270, 326, 334, 339, 348, 398, 403, 411, 412, 414, 415, 416, 417, 433, 488, 491, 492, 494, 495, 498, 500], "most": [2, 149, 247, 307, 334, 477, 491, 492, 494, 495, 496], "move": [0, 2, 229, 500], "moveaxi": 0, "mpi": 328, "mpiexec": 492, "mpirun": 492, "mse": 308, "mse_loss": 334, "mtl": 2, "mtl_capture_en": 3, "mtlcommandbuff": 2, "mu": 479, "much": [1, 2, 6, 336, 337, 338, 366, 367, 368, 491, 496], "multi": [8, 149, 341, 342, 343, 344, 345, 346, 493, 495, 497], "multidimension": 214, "multiheadattent": [6, 334], "multipl": [0, 1, 9, 15, 91, 143, 145, 147, 167, 168, 210, 230, 243, 244, 390, 403, 481, 482, 484, 491, 496, 499], "multipli": [0, 2, 39, 168, 243, 244, 347, 403, 412], "murtadha": 6, "must": [0, 1, 2, 3, 9, 91, 94, 141, 148, 149, 166, 168, 192, 193, 195, 246, 247, 251, 254, 257, 258, 313, 412, 497], "mx": [1, 2, 3, 4, 5, 6, 7, 39, 97, 98, 113, 124, 127, 141, 142, 143, 146, 163, 175, 183, 185, 192, 193, 195, 196, 200, 253, 271, 279, 280, 307, 308, 318, 334, 336, 337, 338, 339, 350, 359, 362, 366, 367, 368, 370, 377, 381, 396, 412, 413, 414, 415, 416, 417, 418, 419, 420, 422, 430, 433, 434, 435, 439, 442, 449, 459, 461, 464, 466, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501], "my": [6, 9], "my_devic": 501, "my_path": 271, "myexp": [1, 146], "myexp_strid": 1, "mymlp": 464, "n": [0, 1, 2, 6, 31, 91, 99, 100, 101, 102, 103, 104, 105, 144, 149, 150, 152, 153, 155, 156, 159, 161, 172, 173, 251, 288, 304, 309, 339, 341, 342, 343, 344, 345, 346, 348, 349, 354, 360, 395, 412, 441, 446, 492], "n_kv": 149, "n_q": 149, "n_t": 354, "naiv": [2, 494], "naive_add": 494, "name": [1, 2, 113, 142, 146, 168, 200, 243, 244, 269, 270, 271, 272, 334, 355, 374, 377, 379, 492, 495, 499], "named_modul": 334, "namespac": 4, "nan": [0, 17, 83, 177, 178, 180, 231], "nan_to_num": 0, "nanobind": [2, 411], "nanobind_add_modul": 2, "nativ": 9, "natur": [0, 201, 203, 496], "nb": 2, "nb_domain": 2, "nb_func": 411, "nb_modul": 2, "nb_static": 2, "nbyte": 2, "nc": 339, "ndarrai": [31, 495, 496, 498], "ndhwc": [343, 346, 349], "ndim": [0, 1, 2, 163, 195, 197, 412], "ne": 1, "nearest": [1, 412], "necessari": 334, "necessarili": 301, "need": [1, 2, 4, 6, 7, 8, 9, 83, 243, 334, 388, 389, 403, 411, 488, 492, 494, 496, 497, 498, 500], "neg": [0, 119, 163, 181, 231, 265, 302, 362, 366, 367, 368, 390, 436, 444, 446, 495], "negat": [0, 232], "negative_slop": [362, 430], "neginf": [0, 231], "neighbor": 412, "neither": [169, 308], "nelem": 2, "nervou": 6, "nest": [79, 95, 322, 334, 464, 490, 494], "nesterov": 479, "network": [6, 8, 339, 348, 351, 414, 415, 461, 464, 478, 492], "neural": [6, 8, 351, 414, 415, 447, 461, 464, 478], "never": [6, 496], "new": [0, 2, 7, 92, 119, 229, 233, 263, 287, 303, 310, 320, 321, 382, 390, 464, 466, 477, 482, 491, 493, 495, 496, 497], "new_tre": 321, "next": [2, 4, 6, 7, 222, 493], "nh": [354, 360, 395], "nhwc": [339, 342, 345], "nice": [494, 496], "nlc": [339, 341, 344], "nld": [354, 360, 395], "nlh": [354, 360, 395], "nll": [436, 444], "nll_loss": 334, "nn": [2, 6, 7, 271, 320, 334, 461, 464, 466, 475, 477, 491, 493, 496], "nobodi": 6, "node": [95, 137, 312, 321, 322], "nois": 5, "noisi": 5, "nomins": 2, "non": [0, 1, 2, 4, 9, 214, 385, 395, 447, 464], "none": [1, 2, 6, 10, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 116, 117, 118, 119, 120, 122, 123, 126, 127, 128, 129, 130, 131, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 221, 225, 226, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 270, 271, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 312, 313, 314, 315, 316, 319, 320, 321, 322, 336, 337, 338, 352, 366, 367, 368, 370, 374, 375, 382, 387, 390, 395, 403, 411, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 468, 486, 493, 495], "nonlinear": [395, 491], "nonzero": 495, "noop": 387, "nor": [2, 169, 308], "norm": [6, 147, 318, 355, 446, 472, 473], "norm1": 6, "norm2": 6, "norm_first": 411, "normal": [1, 2, 5, 6, 145, 146, 147, 192, 251, 257, 334, 336, 337, 338, 339, 355, 359, 361, 366, 367, 368, 394, 411, 414, 416, 497, 500], "not_equ": 0, "notabl": [6, 8], "notat": [117, 319, 379], "note": [0, 1, 2, 4, 6, 9, 17, 19, 84, 91, 95, 101, 104, 105, 113, 149, 157, 158, 168, 177, 195, 217, 243, 247, 310, 316, 334, 394, 412, 466, 497, 499], "noth": [6, 113, 334, 496], "notic": [6, 493, 494, 499], "now": [1, 2, 6, 9, 393, 491, 492, 497], "np": [1, 6, 7, 492, 497, 498], "npy": [200, 268, 499], "npz": [6, 200, 271, 272, 377, 381, 499], "nuclear": 195, "nullopt": 0, "num": [0, 6, 199, 256], "num_class": [7, 466], "num_decoder_lay": 411, "num_embed": [351, 392], "num_encoder_lay": 411, "num_epoch": [7, 466], "num_exampl": 5, "num_featur": [5, 339], "num_group": 355, "num_head": [6, 390, 411], "num_it": 5, "num_lay": [6, 7, 466], "num_param": 334, "num_paramet": 391, "num_sampl": 247, "num_split": 0, "number": [0, 2, 12, 19, 62, 71, 95, 100, 101, 102, 104, 105, 117, 142, 144, 149, 168, 169, 173, 184, 199, 231, 238, 243, 244, 247, 250, 252, 256, 258, 262, 265, 266, 299, 300, 304, 308, 311, 312, 316, 334, 339, 341, 342, 343, 344, 345, 346, 348, 349, 355, 359, 390, 391, 411, 412, 414, 415, 416, 417, 480, 482, 483, 488, 491, 494, 501], "number_of_el": 0, "numer": [6, 145, 147, 195, 205, 209, 281, 339, 355, 359, 361, 394, 433, 434, 436, 446, 467, 468, 469, 470, 471, 472, 478, 491, 496], "numpi": [2, 6, 7, 8, 14, 17, 19, 88, 89, 90, 92, 129, 130, 134, 170, 171, 177, 186, 187, 188, 205, 210, 212, 228, 230, 234, 240, 261, 264, 291, 496, 498, 499], "nw": 1, "nwhc": 348, "o": [2, 9, 149, 360], "o_t": 360, "obj": 269, "object": [3, 11, 31, 51, 79, 95, 142, 143, 146, 183, 271, 312, 319, 320, 321, 322, 326, 348, 411, 490], "observ": 6, "occupi": [117, 168, 243, 244], "occur": 497, "odim": 7, "odot": [354, 360], "off": [6, 9, 496], "offer": 441, "offset": [0, 1, 2, 6, 47, 84, 119, 145, 148, 302], "often": 349, "ok": [377, 491, 493, 494], "okai": [491, 496], "old": 6, "older": [141, 143, 175], "omit": [472, 492], "onc": [2, 9, 491, 493], "one": [0, 2, 4, 6, 9, 39, 79, 85, 94, 100, 101, 102, 104, 105, 124, 139, 141, 144, 145, 147, 148, 195, 203, 210, 244, 247, 286, 291, 307, 326, 387, 412, 435, 492, 493, 500], "ones": [0, 2, 6, 236, 271, 280, 304, 388, 389, 466, 492, 495], "ones_lik": 0, "onli": [1, 2, 6, 8, 9, 83, 91, 100, 101, 102, 104, 105, 113, 192, 193, 195, 224, 243, 251, 310, 334, 374, 375, 377, 382, 384, 387, 388, 389, 464, 491, 492, 493, 494, 499, 500], "onlin": 469, "op": [1, 2, 237, 310, 375, 496], "open": [3, 9, 19, 254, 258], "openmpi": 492, "oper": [3, 6, 8, 10, 38, 85, 86, 87, 102, 149, 167, 168, 240, 242, 281, 289, 296, 324, 334, 411, 473, 491, 492, 494, 495, 496, 497, 498, 500, 501], "operand": [131, 132, 167], "opportun": 491, "opt": [474, 492], "optim": [1, 3, 5, 7, 8, 388, 491, 492, 494, 496], "option": [0, 3, 6, 15, 16, 18, 19, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 84, 85, 86, 87, 91, 95, 96, 99, 100, 101, 102, 103, 104, 105, 106, 109, 110, 111, 112, 113, 117, 118, 119, 122, 123, 124, 126, 127, 128, 141, 143, 144, 145, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 166, 167, 168, 169, 173, 181, 182, 185, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 209, 211, 213, 214, 223, 227, 231, 235, 238, 239, 241, 243, 244, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 262, 263, 265, 281, 282, 283, 286, 287, 288, 292, 294, 295, 299, 301, 302, 303, 304, 305, 306, 307, 308, 309, 312, 314, 316, 319, 320, 321, 322, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 354, 360, 363, 366, 367, 368, 370, 374, 375, 377, 382, 387, 390, 392, 393, 395, 398, 403, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 467, 468, 469, 470, 471, 472, 473, 475, 478, 479, 480, 488, 491, 493, 499, 501], "ord": 195, "order": [0, 1, 29, 84, 102, 132, 192, 193, 195, 239, 243, 301, 334, 355, 388, 400, 475, 491, 494], "ordinari": 176, "org": [355, 359, 361, 369, 394, 425, 447], "origin": [6, 119, 318, 339, 383, 414, 415, 416, 417, 467, 468, 469, 472, 473, 493, 497], "orthonorm": 172, "ostream": 2, "ostringstream": 2, "other": [0, 2, 6, 8, 183, 195, 334, 376, 464, 473, 491, 492, 493, 495, 496, 498], "other_input": 334, "otherwis": [19, 102, 124, 223, 253, 316, 319, 320, 321, 322, 375, 377, 387, 409, 411, 412, 427, 433, 438, 445, 457, 458, 496, 497], "our": [1, 2, 6, 7, 400, 467, 468, 469, 472, 473, 492], "out": [0, 1, 2, 9, 91, 146, 175, 348, 349, 384, 491, 492, 493, 494, 495], "out_ax": [312, 494], "out_channel": [341, 342, 343, 344, 345, 346], "out_dim": [334, 464], "out_dtyp": 2, "out_idx": 2, "out_mask": 91, "out_proj": [6, 464], "out_ptr": 2, "out_shap": [1, 2], "outer": [0, 491, 496], "outlier": 441, "output": [0, 1, 2, 6, 9, 16, 17, 18, 19, 29, 84, 91, 92, 95, 97, 98, 109, 110, 111, 112, 113, 131, 142, 144, 145, 146, 147, 148, 149, 156, 159, 160, 161, 166, 167, 169, 172, 173, 177, 195, 199, 209, 211, 213, 214, 227, 231, 235, 236, 239, 241, 242, 246, 247, 248, 250, 251, 252, 254, 257, 258, 271, 272, 279, 280, 281, 286, 288, 292, 296, 302, 304, 308, 309, 310, 311, 312, 313, 314, 315, 339, 341, 342, 343, 344, 345, 346, 359, 363, 390, 393, 409, 411, 412, 414, 415, 416, 417, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 458, 461, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500], "output_dim": [7, 334, 363, 393], "output_directori": 2, "output_dtyp": [1, 146], "output_fil": 6, "output_nam": [1, 146], "output_shap": [1, 146], "output_strip_trailing_whitespac": 4, "output_vari": 4, "outsid": [146, 163], "over": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 152, 155, 158, 161, 176, 195, 197, 199, 209, 211, 213, 227, 239, 241, 267, 281, 282, 288, 292, 299, 301, 309, 339, 341, 342, 343, 344, 345, 346, 355, 361, 394, 435, 480, 483, 492, 494], "overal": 2, "overhead": [491, 496, 500], "overlap": 1, "overload": 19, "overrid": [2, 133], "overview": 3, "overwrit": 6, "own": [9, 497], "owndata": 497, "p": [9, 246, 334, 347, 348, 349, 446, 470, 472], "pack": [168, 243, 244], "packag": [2, 5, 7, 9, 328, 461, 492], "package_data": 2, "pad": [0, 1, 99, 100, 101, 102, 103, 104, 105, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 336, 337, 338, 341, 342, 343, 344, 345, 346, 366, 367, 368], "pad_valu": 0, "pad_width": [0, 238], "padding_hi": 0, "padding_lo": 0, "page": 498, "pain": 6, "pair": [0, 2, 238, 377, 398], "pairwis": 446, "pan": 6, "paper": [339, 403, 467, 468, 469, 472, 473], "parallel": [492, 500], "param": [308, 316, 334, 461, 493, 494], "paramet": [0, 1, 2, 5, 6, 7, 13, 14, 15, 16, 17, 18, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 116, 117, 118, 119, 122, 123, 124, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 222, 223, 224, 225, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 359, 360, 361, 362, 363, 366, 367, 368, 370, 371, 374, 375, 377, 382, 383, 384, 387, 388, 389, 390, 391, 392, 393, 394, 395, 398, 400, 403, 407, 409, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 426, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 458, 460, 461, 464, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 477, 478, 479, 480, 481, 482, 483, 484, 486, 491, 492, 493, 494, 496], "parameter_scal": 468, "parametr": [391, 448], "pars": [6, 142], "parse_arg": 6, "parser": 6, "part": [1, 2, 141, 143, 174, 175, 259, 494, 495], "partial": [388, 389, 491, 496], "particip": [122, 123, 126, 127, 128], "particular": [243, 355], "particularli": 491, "partit": [0, 29], "pass": [1, 2, 6, 7, 66, 80, 237, 238, 308, 317, 319, 320, 321, 334, 375, 387, 388, 389, 400, 491, 492, 493, 496], "password": 492, "path": [3, 4, 9, 132, 141, 142, 143, 175, 225, 271, 272, 316, 321, 377, 492], "pattern": [334, 496], "peak": [219, 221], "penalti": 479, "pep": 497, "per": [6, 7, 117, 149, 168, 243, 244, 316, 339, 355, 359, 361, 394, 486, 491, 492, 496], "perceptron": [8, 493], "perf_count": 491, "perfectli": 496, "perform": [0, 1, 2, 3, 6, 8, 15, 91, 102, 109, 110, 111, 112, 128, 131, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 167, 168, 172, 192, 193, 210, 244, 266, 281, 295, 334, 355, 411, 416, 417, 466, 491, 492, 495, 496, 500], "perhap": [2, 6], "perm": 7, "permtuat": 253, "permut": [0, 7], "persist": 9, "pg": 195, "phi": [352, 423], "physic": 492, "pi": [135, 352, 403, 424, 494], "pick": 2, "pip": [2, 4, 9], "pipelin": 2, "pixel": 348, "place": [2, 6, 39, 265, 266, 316, 492, 496, 497], "placehold": 491, "plai": [2, 6], "plain": 400, "plan": [2, 491], "platform": 9, "plu": [0, 203], "point": [0, 2, 5, 6, 9, 84, 162, 165, 244, 326], "pointer": 2, "pool": [336, 337, 338, 366, 367, 368, 500], "popul": 2, "portion": 347, "posinf": [0, 231], "posit": [0, 6, 29, 119, 148, 163, 169, 182, 189, 190, 229, 231, 239, 251, 265, 302, 308, 320, 334, 341, 342, 343, 344, 345, 346, 390, 398, 403, 436, 446, 493], "possibl": [283, 351, 392, 491, 492, 495, 500], "possibli": [6, 15, 91, 167, 210, 318], "postur": 6, "potenti": 223, "power": [0, 494, 497], "practic": [2, 491], "pre": [9, 149, 433], "preced": 355, "precis": [0, 2, 6, 140, 149, 334, 352, 394, 433, 474, 491], "preclud": 334, "pred": [437, 441], "predic": [316, 382], "predict": [433, 436, 437, 438, 439, 440, 441, 443, 444, 445], "prefix": [312, 319], "prelu": 334, "prepar": [2, 6], "prepend": [3, 210], "preprint": [6, 467, 473], "preprocessor": 9, "present": 1, "preserv": [263, 494], "press": [6, 195], "pressur": 2, "pretti": [491, 496], "prevent": [289, 446, 497], "previou": [222, 223, 224], "primal": [1, 2, 113, 184, 311], "primit": 494, "print": [1, 2, 5, 6, 7, 9, 113, 185, 318, 319, 320, 321, 323, 334, 488, 491, 492, 493, 494, 495, 496, 497, 498], "prior": [242, 295, 296], "priorit": 494, "privat": [2, 4], "prng": [246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 488], "prob": 433, "probabl": [9, 254, 347, 348, 349, 393, 433, 435, 439, 500], "problem": [5, 7, 334], "process": [6, 102, 106, 121, 122, 123, 124, 126, 127, 128, 320, 321, 349, 351, 411, 490, 492], "processor": 9, "prod": [0, 1], "produc": [0, 2, 9, 95, 390, 461, 493], "product": [0, 2, 15, 84, 111, 176, 184, 185, 191, 210, 237, 241, 299, 311, 390, 498], "profil": 3, "program": [4, 219, 492], "programmat": 389, "project": [3, 4, 6, 390, 493], "project_source_dir": 2, "promot": [2, 149], "promote_typ": 2, "promoted_dtyp": 2, "prompt": 6, "propag": [494, 495], "properti": [32, 39, 48, 52, 62, 63, 69, 71, 383, 386, 476, 494], "proportion": 318, "protocol": 497, "provid": [0, 2, 6, 84, 117, 141, 142, 169, 253, 265, 299, 308, 320, 322, 328, 334, 370, 375, 377, 387, 388, 389, 392, 393, 411, 412, 460, 464, 492, 493, 499, 501], "pseudo": 488, "pth": 6, "public": [2, 334], "pun": 0, "pure": [1, 113, 334, 466], "purpos": [1, 195], "purs": 6, "push": 2, "push_back": 2, "put": [0, 1, 7, 242, 491, 492], "put_along_axi": 0, "py": [2, 6, 9, 492], "pypi": 9, "python": [1, 3, 4, 6, 51, 69, 79, 137, 319, 320, 321, 322, 323, 464, 474, 475, 477, 490, 492, 493, 494, 497], "python_execut": 4, "python_requir": 2, "pytorch": [6, 8, 352, 355, 494], "pytorch_compat": 355, "q": [149, 196], "qualifi": 492, "quantiz": [0, 117, 168, 200, 244, 392, 393], "quantized_matmul": 0, "quantizedembed": 334, "quantizedlinear": 334, "quarter": 6, "queri": [6, 149, 224, 390], "query_input_dim": 390, "query_proj": 6, "question": [6, 496], "queue": 3, "quick": [2, 8], "quit": [494, 497], "quotient": [0, 129, 130, 165], "r": [2, 6, 196, 308, 348, 354], "r_t": 354, "race": 500, "radian": [0, 116], "rag": 6, "rain": 6, "rais": [0, 6, 113, 195, 223, 240, 283, 377, 493], "ram": 6, "random": [1, 2, 3, 5, 6, 7, 8, 146, 336, 337, 338, 339, 359, 366, 367, 368, 377, 384, 491, 493, 494, 500, 501], "randomli": [5, 6, 253, 347, 348, 349], "rang": [0, 2, 3, 5, 6, 7, 9, 19, 163, 167, 199, 415, 417, 424, 425, 466, 480, 481, 482, 483, 484, 488, 491, 494, 496, 500], "rank": [0, 126, 127, 128, 442, 492], "rate": [5, 466, 467, 468, 469, 470, 471, 472, 473, 478, 479], "rather": [2, 494, 500], "ratio": [0, 25], "rceil": 91, "re": [7, 9, 461], "readabl": 3, "readi": 2, "real": [0, 156, 157, 158, 159, 160, 161, 189, 190, 192, 193], "realli": 361, "reason": [1, 6, 495], "reboot": 9, "receiv": [126, 127, 316, 482, 497], "reciproc": [0, 267], "reclaim": 222, "recommend": [9, 223, 473], "recompil": [95, 491], "record": [3, 219, 496], "recreat": [323, 466], "rectifi": [362, 396, 397, 416, 417, 430, 449, 450], "recurr": [354, 360, 395], "recurs": [142, 334, 374, 375, 380, 385, 387, 464], "recv": 127, "redirect": 2, "reduc": [0, 1, 9, 16, 18, 27, 28, 123, 209, 211, 213, 227, 241, 288, 292, 309, 322, 339, 411, 441], "reduct": [16, 18, 123, 209, 211, 227, 241, 322, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 492], "redund": 494, "refer": [195, 359, 369, 383, 414, 415, 416, 417, 425, 447, 495], "reflect": [383, 491, 495, 497], "regard": 352, "regardless": [84, 149], "regist": [2, 7], "register_librari": 2, "regress": [8, 441], "regular": [39, 348, 447, 471, 491, 493, 495], "regularli": 2, "reimplement": 2, "rel": [17, 177, 468, 491], "relative_step": 468, "relax": 223, "releas": 4, "relev": 2, "reli": [1, 2], "relu": [334, 391, 411, 448, 461], "relu6": 334, "remain": [0, 6, 224, 308, 321, 347, 348, 349, 492], "remaind": [0, 130], "remov": [0, 119, 210, 247, 286, 435], "rep": [0, 300], "repeat": [0, 300], "repeatedli": 5, "repetit": 262, "replac": [0, 6, 231, 388, 389, 411, 445], "replai": 3, "repli": 6, "repo": [5, 7, 9, 491], "report": [217, 223], "repres": [2, 6, 121, 124, 168, 442, 446, 497], "represent": [6, 243, 310, 319, 323], "request": 2, "requir": [1, 2, 4, 6, 334, 492, 496, 497], "requires_grad": 494, "rerun": [491, 496], "rescal": 318, "research": 8, "reset": 221, "reset_peak_memori": 219, "reshap": [0, 6, 195, 412, 491, 495], "resid": 224, "resolv": 2, "resourc": 2, "resource_limit": 216, "respect": [2, 5, 7, 113, 145, 147, 167, 168, 169, 243, 308, 320, 334, 339, 352, 355, 359, 361, 464, 494, 498], "respons": 2, "rest": [6, 148, 320, 321, 398], "restart": 9, "restor": 265, "result": [0, 6, 15, 19, 39, 79, 84, 95, 142, 145, 147, 168, 185, 195, 210, 244, 251, 262, 287, 320, 321, 322, 403, 433, 491, 494, 497], "resum": 6, "return": [0, 1, 2, 4, 5, 6, 7, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 51, 69, 79, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 116, 117, 118, 119, 122, 123, 124, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 138, 139, 140, 141, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 218, 222, 223, 224, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 266, 267, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 334, 354, 360, 370, 371, 372, 374, 375, 376, 377, 378, 379, 380, 384, 385, 387, 388, 389, 395, 413, 414, 415, 416, 417, 418, 419, 420, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 461, 464, 474, 490, 491, 492, 493, 494, 495, 496, 497, 499, 500], "return_metadata": 200, "revers": [0, 2, 42, 43, 44, 45, 84, 109, 110, 111, 112, 303, 403], "rf": 9, "rfft": 156, "rfft2": 157, "rfftn": 158, "rho": 467, "rhs_indic": [0, 167, 168], "rhs_mask": 91, "right": [0, 1, 2, 9, 243, 264, 265, 352, 412, 424, 425, 436, 438, 446], "right_shift": 0, "ring": 124, "rm": [6, 9, 147, 468], "rmsnorm": [6, 334], "rmsprop": 466, "rnn": [334, 354], "roadcast": 254, "robust": 441, "roform": [6, 398], "roll": 0, "root": [0, 6, 147, 267, 284, 394], "rope": [6, 334], "rosetta": 9, "rotari": [6, 148, 398], "rotat": [148, 398], "round": [0, 243], "routin": 2, "row": [0, 1, 2, 84, 144, 146, 173, 243, 304], "row_contigu": 2, "rpath": 2, "rsqrt": 0, "rtol": [0, 17, 177], "rule": [2, 466], "run": [1, 2, 3, 4, 6, 7, 8, 9, 10, 146, 237, 324, 339, 370, 467, 468, 470, 471, 472, 491, 492, 493, 496, 500, 501], "runtim": [6, 124, 328, 491, 492], "runtime_error": 2, "safetensor": [9, 200, 270, 377, 381, 466, 496, 499], "sai": [2, 6, 461, 496], "said": 6, "sake": 494, "same": [0, 2, 6, 9, 17, 39, 83, 92, 95, 100, 101, 102, 104, 105, 106, 122, 145, 147, 156, 159, 160, 161, 168, 169, 177, 184, 238, 247, 265, 266, 280, 310, 311, 313, 321, 334, 337, 338, 339, 347, 355, 359, 367, 368, 392, 413, 414, 415, 416, 417, 418, 419, 420, 435, 446, 464, 474, 488, 491, 492, 493, 495, 500], "sampl": [2, 5, 6, 199, 246, 247, 248, 250, 251, 254, 257, 258, 414, 415, 416, 417, 419, 420, 436, 442, 446, 488, 491, 493], "sat": 6, "save": [3, 6, 8, 200, 225, 243, 269, 270, 271, 272, 381, 493, 496], "save_gguf": 499, "save_safetensor": [381, 466, 499], "save_weight": 334, "savez": [6, 381, 499], "savez_compress": 499, "saw": [6, 494], "scalar": [0, 2, 14, 15, 17, 31, 51, 79, 83, 88, 89, 90, 91, 92, 94, 129, 130, 134, 165, 166, 169, 170, 171, 172, 177, 186, 187, 188, 199, 205, 206, 207, 208, 210, 212, 228, 230, 231, 234, 238, 240, 246, 254, 257, 258, 261, 264, 269, 291, 308, 310, 313, 317, 446, 493, 494, 496, 498], "scale": [0, 2, 6, 15, 117, 145, 147, 148, 149, 168, 172, 243, 244, 250, 252, 318, 348, 349, 361, 390, 398, 399, 403, 412, 451, 468], "scale_arr": 2, "scale_factor": 412, "scale_paramet": 468, "scatter": 0, "scatter_add": 0, "scatter_add_axi": 0, "scatter_max": 0, "scatter_min": 0, "scatter_prod": 0, "schedul": [2, 223, 466, 480, 481, 482, 483, 484, 486, 500], "schema": 3, "scipi": 172, "scope": 334, "score": [6, 149, 442], "sdk": 9, "se": 1, "second": [6, 9, 119, 183, 185, 186, 206, 208, 210, 264, 293, 302, 308, 337, 338, 367, 368, 434, 442, 468, 472, 491, 493, 494, 500], "second_layer_a": 496, "second_layer_b": 496, "secret": 6, "section": [1, 6, 9, 283, 446, 491, 492, 494], "see": [1, 2, 4, 6, 7, 9, 11, 12, 33, 34, 35, 36, 37, 40, 41, 42, 43, 44, 45, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 195, 222, 269, 270, 316, 326, 334, 339, 340, 348, 350, 352, 356, 357, 358, 364, 365, 373, 391, 392, 393, 396, 397, 398, 399, 401, 403, 404, 405, 406, 407, 408, 410, 412, 414, 415, 416, 417, 423, 424, 425, 451, 491, 492, 493, 494, 495, 498, 500], "seed": 249, "seen": 497, "select": [0, 3, 9, 124, 192, 193, 301, 313, 370, 374, 382], "self": [6, 7, 10, 31, 32, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 51, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 113, 162, 334, 447, 464], "selu": 334, "semant": [14, 88, 89, 90, 92, 129, 130, 134, 170, 171, 186, 187, 188, 205, 210, 212, 228, 230, 234, 240, 261, 264, 291, 500], "semi": [189, 190, 251], "send": 492, "sennrich": 6, "sensit": 441, "sentencepiec": 6, "separ": [6, 66, 80, 355, 442], "sequenc": [6, 16, 18, 34, 35, 57, 58, 59, 60, 64, 72, 75, 76, 77, 81, 84, 92, 102, 126, 139, 146, 151, 152, 154, 155, 157, 158, 160, 161, 166, 169, 209, 211, 213, 227, 235, 241, 246, 247, 248, 250, 251, 252, 254, 257, 258, 263, 279, 280, 281, 283, 286, 288, 292, 299, 300, 303, 307, 308, 309, 314, 339, 341, 344, 354, 360, 395, 411, 488, 500], "sequenti": [334, 461], "seri": 9, "serial": 466, "set": [2, 4, 6, 7, 9, 95, 113, 120, 122, 123, 124, 126, 127, 128, 133, 145, 148, 216, 222, 223, 224, 273, 274, 290, 352, 361, 363, 373, 375, 382, 383, 384, 387, 388, 393, 398, 409, 434, 446, 458, 464, 466, 468, 470, 471, 475, 488, 493, 494, 496], "set_byt": 2, "set_compute_pipeline_st": 2, "set_data": 2, "set_default_devic": 2, "set_dtyp": 334, "set_input_arrai": 2, "set_memory_limit": 222, "set_output_arrai": 2, "set_vector_byt": 2, "setup": [2, 4, 5, 7, 9, 491, 493], "sever": [6, 9, 99, 100, 101, 102, 103, 104, 105, 271, 272, 491, 499], "sgd": [5, 7, 466, 473, 475, 480, 481, 484, 491], "shade": [1, 2], "shall": 6, "shape": [0, 2, 3, 6, 7, 66, 83, 84, 91, 92, 95, 99, 100, 101, 102, 103, 104, 105, 119, 122, 126, 127, 141, 143, 146, 149, 150, 153, 156, 159, 160, 161, 166, 167, 172, 184, 194, 198, 210, 235, 236, 246, 247, 248, 250, 251, 252, 254, 257, 258, 263, 265, 280, 307, 310, 311, 313, 314, 315, 334, 336, 337, 338, 339, 341, 342, 343, 344, 345, 346, 348, 349, 354, 359, 360, 363, 366, 367, 368, 377, 395, 413, 414, 415, 416, 417, 418, 419, 420, 435, 446, 466, 491, 493, 494, 495, 498, 500], "shapeless": [0, 95, 141, 143], "share": [8, 117, 168, 243, 244, 310, 492], "shazeer": 6, "shift": [0, 186, 264, 265, 339], "shop": 6, "should": [1, 2, 4, 5, 6, 7, 9, 84, 119, 122, 145, 146, 147, 149, 184, 215, 224, 225, 242, 243, 296, 302, 308, 311, 316, 319, 334, 341, 342, 343, 344, 345, 346, 348, 349, 384, 390, 400, 435, 437, 442, 464, 490, 491, 492, 493, 494, 496, 497, 501], "show": [9, 326, 491], "shown": 2, "shuffl": 7, "side": [0, 238, 336, 337, 338, 366, 367, 368, 491], "sigma": [352, 353, 354, 360, 402, 414, 415, 416, 417, 425, 426, 431, 452, 453], "sigmoid": [0, 6, 334, 364, 401, 425, 431, 433, 453], "sign": [0, 17, 177, 326, 473], "signal": [106, 412], "signatur": [1, 146], "signedinteg": [12, 183], "signific": 243, "silent": [159, 160, 161], "silicon": [2, 6, 8, 9, 500], "silu": 334, "simd": 1, "simd_sum": 1, "simdgroup": 1, "simdgroup_s": 1, "similar": [6, 168, 183, 320, 388, 389, 390, 434, 497, 499], "similarli": [2, 9, 210, 494, 496], "simpl": [2, 6, 7, 334, 351, 460, 466, 491, 492, 493, 494, 496], "simple_axpbi": 2, "simple_tim": 2, "simplest": [2, 334, 492], "simpli": [2, 6, 9, 350, 362, 396, 422, 430, 449, 459, 464, 491, 492, 494], "simplic": 0, "simultan": 1, "sin": [0, 113, 403, 493, 494, 498], "sinc": [1, 2, 6, 7, 168, 219, 464, 473, 482, 491, 493, 497, 500], "sine": [0, 22, 23, 277, 278, 493, 494], "sing": 195, "singer": 469, "singl": [2, 7, 137, 184, 200, 214, 238, 311, 337, 338, 367, 368, 491, 493, 495, 499], "singleton": [0, 16, 18, 27, 28, 124, 209, 210, 211, 213, 227, 241, 288, 292, 309], "singular": [195, 197], "sinh": 0, "sinusoid": 403, "sinusoidalpositionalencod": 334, "size": [0, 1, 2, 6, 7, 52, 69, 91, 100, 101, 104, 105, 117, 139, 145, 146, 147, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 166, 168, 172, 173, 183, 191, 195, 218, 223, 224, 243, 244, 247, 263, 279, 283, 286, 307, 310, 316, 334, 336, 337, 338, 341, 342, 343, 344, 345, 346, 351, 359, 366, 367, 368, 392, 393, 412, 468, 492, 496, 497], "size_in_megabyt": 224, "size_t": [0, 2], "skip": [3, 84], "slice": [0, 280, 495], "slice_s": [0, 279], "slice_upd": 0, "slide": [336, 337, 338, 366, 367, 368], "slight": [6, 496], "slightli": [398, 500], "slope": 362, "slot": 492, "slow": 491, "slowli": 6, "small": [6, 140, 145, 147, 339, 355, 361, 394, 436, 441, 446, 491, 492, 500], "smaller": [0, 9, 239, 473, 491], "smallest": 195, "smile": 6, "smooth": [435, 445, 478], "smooth_l1_loss": 334, "sned": 128, "snippet": 492, "so": [1, 2, 6, 9, 169, 172, 308, 347, 412, 466, 491, 492, 496, 500], "softmax": [0, 6, 149, 334, 365, 432, 435], "softmin": 334, "softplu": [334, 369, 447], "softshrink": 334, "softsign": 334, "solv": 334, "some": [0, 2, 5, 6, 7, 142, 375, 387, 466, 475, 491, 493, 494, 496], "someon": 6, "someth": [5, 6, 495], "sometim": 491, "sonoma": 9, "soon": 6, "sort": [0, 29, 30, 239, 301], "sourc": [0, 1, 2, 3, 4, 61, 126, 127, 146, 229, 303, 492], "space": [0, 2, 199, 433, 444], "spars": [0, 214], "spatial": [100, 101, 102, 104, 105, 336, 337, 338, 355, 366, 367, 368, 412], "speak": [6, 195], "special": 2, "specif": [1, 2, 9, 124, 492, 494], "specifi": [0, 2, 19, 38, 100, 101, 102, 104, 105, 119, 157, 158, 166, 169, 191, 195, 199, 229, 235, 242, 247, 262, 293, 295, 296, 299, 302, 303, 308, 312, 314, 339, 409, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 458, 491, 492, 493, 494, 500], "speed": [1, 2], "spent": 6, "split": [0, 353, 355, 426], "splittabl": 488, "sqrt": [0, 6, 135, 149, 172, 339, 352, 355, 359, 361, 363, 394, 403, 414, 415, 416, 417, 424, 467, 469, 470, 471, 478, 491], "squar": [0, 5, 6, 147, 173, 194, 198, 267, 284, 308, 320, 334, 394, 443, 445, 467, 468, 470, 471, 472, 494, 497], "squeez": [0, 412, 491], "src": [0, 126, 127], "ssh": 492, "stabil": [145, 147, 339, 355, 359, 361, 394, 433, 434, 436, 467, 468, 469, 470, 471, 472, 478], "stabl": [205, 209, 281, 441], "stable_abi": 2, "stack": [0, 491], "standard": [0, 1, 4, 7, 51, 79, 210, 248, 252, 288, 411, 414, 416, 419, 492, 498], "starmap": [6, 320], "start": [0, 1, 2, 5, 6, 8, 9, 19, 148, 199, 225, 279, 280, 283, 322, 491, 493, 495, 500], "start_axi": [0, 50, 163], "start_captur": 3, "start_indic": [279, 280], "state": [6, 7, 334, 354, 360, 395, 466, 475, 488, 491], "static": [9, 491], "static_cast": 2, "std": [0, 2, 4, 419, 493], "step": [0, 3, 4, 6, 7, 19, 334, 354, 360, 395, 468, 475, 480, 482, 483, 484, 491, 492], "step_decai": 466, "step_siz": 484, "still": [6, 9, 195, 491, 496], "stochast": [469, 470, 472, 479, 496], "stood": 6, "stop": [0, 2, 6, 19, 199, 226, 289, 494, 495], "stop_captur": 3, "stop_gradi": [0, 494], "storag": 84, "store": 6, "str": [2, 106, 124, 131, 132, 141, 142, 143, 146, 169, 175, 192, 193, 195, 200, 214, 216, 225, 268, 269, 270, 271, 272, 308, 316, 319, 323, 370, 371, 374, 375, 377, 379, 381, 387, 412, 416, 417, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446], "straight": 6, "strang": 6, "stream": [2, 8, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 115, 116, 117, 118, 119, 122, 123, 126, 127, 128, 129, 130, 131, 134, 135, 136, 138, 139, 140, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 164, 165, 166, 167, 168, 170, 171, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 309, 310, 313, 314, 315, 492, 500], "streamcontext": 290, "streamordevic": [0, 2], "street": 6, "strength": [473, 479], "strict": [124, 170, 187, 375, 377, 387], "strictli": [195, 224], "stride": [0, 2, 84, 99, 100, 101, 102, 103, 104, 105, 336, 337, 338, 341, 342, 343, 344, 345, 346, 366, 367, 368, 398, 495], "string": [0, 2, 132, 141, 146, 175, 216, 238, 493, 497, 499], "structur": [2, 474, 494], "stub": 9, "style": [2, 14, 17, 88, 89, 90, 129, 130, 134, 170, 171, 177, 186, 187, 188, 205, 210, 212, 228, 230, 234, 240, 261, 264, 291], "su": 6, "sub": [0, 7, 119, 256, 279, 280, 302, 316], "subarrai": [119, 283], "subclass": 464, "subdivid": 1, "subdtyp": 183, "subgradi": 469, "sublinear": 468, "submodul": [6, 7, 334, 371, 375, 376, 387, 389], "suboptim": 493, "subscript": [131, 132], "subsect": 6, "subsequ": [124, 466], "subset": [334, 374], "substanti": 9, "subtl": 491, "subtract": [0, 39], "subtyp": [183, 326], "succe": 124, "sudo": [9, 224], "sum": [0, 2, 5, 14, 112, 123, 143, 176, 195, 209, 281, 299, 302, 334, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 492, 495, 497], "sum_": [195, 441], "sum_i": 432, "sum_j": [454, 455], "summat": [131, 132], "super": [6, 7, 334, 464], "superset": [320, 474], "support": [1, 2, 6, 8, 9, 17, 91, 101, 104, 105, 149, 163, 172, 177, 189, 190, 192, 193, 194, 196, 197, 198, 200, 210, 243, 251, 492, 494, 495, 497, 499], "suppos": [494, 500], "sure": [2, 3, 6, 9, 334, 491], "surpass": [416, 417], "surpris": 6, "sw": 1, "swap": [0, 106, 223, 293, 389], "swapax": [0, 113], "swiglu": 6, "swish": [401, 453], "switch": 9, "symbol": 473, "symmetr": [100, 101, 104, 105, 189, 190, 192, 193], "symmetri": [192, 193], "synchron": [2, 491], "syntax": [39, 495], "synthet": 5, "sysctl": 224, "system": [4, 6, 9, 216, 217, 218, 224], "t": [0, 1, 2, 4, 6, 9, 135, 146, 149, 168, 189, 190, 244, 308, 334, 354, 360, 395, 467, 468, 469, 470, 471, 472, 473, 478, 479, 491, 493, 494, 500], "t_kv": 149, "t_q": 149, "tabl": [1, 195, 326, 351], "take": [0, 2, 6, 7, 88, 89, 90, 95, 141, 167, 169, 184, 212, 228, 236, 244, 296, 308, 311, 312, 315, 321, 322, 336, 337, 338, 366, 367, 368, 390, 433, 488, 492, 493, 494, 495, 499, 500, 501], "take_along_axi": [0, 495], "taken": [119, 295, 302], "talk": 492, "tan": 0, "tangent": [0, 2, 24, 25, 26, 113, 184, 297, 298, 410, 459], "tangent_i": 2, "tangent_x": 2, "tanh": [0, 334, 352, 354, 360, 369, 395, 424, 447], "target": [2, 308, 433, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 491], "target_include_directori": 2, "target_link_librari": [2, 4], "target_link_opt": 2, "target_sourc": 2, "task": [223, 441], "tau": 479, "tcp": 492, "tell": [4, 6, 491, 497], "temp": 6, "templat": [0, 1, 2, 146], "ten": 496, "tend": 473, "tensor": [200, 299, 446, 497], "tensordot": 0, "term": [2, 436, 467, 468, 469, 470, 471, 472, 478], "termin": 9, "test": [7, 9, 492], "test_imag": 7, "test_label": 7, "text": [6, 352, 354, 360, 369, 395, 402, 409, 414, 415, 416, 417, 424, 427, 428, 429, 436, 437, 438, 441, 442, 445, 447, 448, 451, 452, 457, 458, 468, 473], "textrm": [243, 352, 353, 423, 426], "tf": 497, "tgp_size": 2, "th": [109, 110, 111, 112, 118, 144, 192, 482], "than": [1, 2, 6, 79, 106, 119, 130, 148, 167, 170, 171, 187, 188, 189, 190, 192, 193, 194, 197, 198, 210, 222, 224, 318, 320, 398, 409, 412, 442, 445, 458, 468, 473, 491, 493, 494, 500], "thank": 496, "thei": [1, 2, 5, 6, 9, 17, 106, 168, 177, 400, 437, 464, 473, 490, 491, 492, 493, 496, 498, 499, 500], "them": [0, 2, 6, 122, 334, 375, 387, 492, 493, 500], "themselv": [2, 491], "thi": [0, 1, 2, 4, 6, 7, 9, 16, 17, 18, 19, 27, 28, 29, 30, 84, 113, 133, 141, 143, 146, 167, 168, 172, 175, 177, 184, 189, 190, 192, 193, 194, 195, 196, 197, 198, 205, 209, 210, 211, 213, 215, 217, 224, 227, 239, 241, 247, 274, 281, 282, 283, 288, 292, 295, 301, 309, 318, 321, 322, 334, 347, 348, 349, 353, 354, 360, 371, 372, 374, 375, 378, 379, 380, 385, 387, 388, 389, 390, 393, 395, 409, 414, 415, 416, 417, 424, 425, 426, 433, 441, 458, 464, 475, 490, 491, 492, 493, 494, 496, 497, 499], "thing": [2, 6, 492], "third": [191, 338, 368, 493], "thompson": 348, "those": [2, 6, 334], "though": [2, 6, 491, 493, 496, 497], "thousand": 496, "thread": [1, 2], "thread_index_in_simdgroup": 1, "thread_position_in_grid": [1, 2, 146], "threadgroup": [1, 2, 146], "threads_per_simdgroup": 1, "three": [6, 87, 338, 368, 412], "threefri": 488, "threshold": [409, 438, 445, 458], "through": [1, 2, 289, 411, 473, 491, 493, 494, 497], "throw": [2, 95, 124], "thu": [6, 334], "thumb": 466, "tic": 491, "tieleman": 478, "tile": [0, 149], "time": [2, 6, 9, 223, 300, 334, 354, 360, 395, 491, 494, 496, 500], "timeit": [491, 494], "titl": 2, "tmp": [1, 146], "to_quant": 316, "to_stream": 2, "toc": 491, "togeth": [0, 1, 2, 7, 243, 320, 321, 492], "tok_embed": 6, "token": [6, 351, 392], "told": 6, "toler": [0, 17, 177], "too": [183, 491, 496], "took": 6, "tool": 9, "top": [2, 301, 363, 412], "topk": 0, "torch": [6, 497], "torch_weight": 6, "total": [224, 494], "total_norm": 318, "tpi": 491, "trace": [0, 3, 143, 491], "trace_fil": 3, "tracer": 388, "track": [2, 334, 339], "track_running_stat": 339, "trade": 496, "tradit": [6, 148, 348, 349, 398], "train": [6, 7, 334, 339, 347, 348, 349, 373, 375, 387, 414, 415, 493], "train_imag": [7, 466], "train_label": [7, 466], "trainabl": [7, 317, 334, 464], "trainable_paramet": [334, 374, 475], "transform": [1, 6, 8, 113, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 172, 317, 334, 339, 355, 361, 363, 374, 375, 387, 393, 398, 495], "transformerencod": 271, "transit": 482, "translat": [145, 361], "transpos": [0, 6, 32, 103, 104, 105, 168, 244, 344, 345, 346], "treat": [0, 2, 113, 157, 158, 160, 161, 295, 412, 491], "tree": [8, 95, 137, 169, 308, 312, 319, 320, 321, 322, 323, 474, 475, 477, 486, 494], "tree_flatten": [271, 320, 323, 334, 466, 493], "tree_map": [321, 334, 492], "tree_unflatten": [6, 466, 493], "trembl": 6, "tri": 0, "triangl": [192, 193, 304], "triangular": [189, 190, 198], "trigger": 491, "tril": 0, "trilinear": 412, "triplet": 446, "triplet_loss": 334, "triu": 0, "true": [0, 1, 2, 4, 5, 6, 17, 42, 43, 44, 45, 83, 95, 109, 110, 111, 112, 146, 148, 168, 177, 183, 189, 190, 195, 200, 214, 223, 244, 281, 313, 316, 319, 320, 321, 322, 326, 334, 339, 341, 342, 343, 344, 345, 346, 354, 355, 359, 360, 361, 363, 374, 375, 377, 384, 387, 393, 395, 398, 403, 411, 412, 433, 441, 468, 470, 471, 491, 493], "truncat": [150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 257], "truth": [5, 435, 445], "try": [2, 9, 124], "tupl": [0, 31, 66, 69, 80, 96, 100, 101, 102, 104, 105, 126, 130, 132, 137, 139, 141, 175, 184, 192, 195, 196, 197, 238, 243, 263, 265, 279, 280, 286, 307, 308, 311, 319, 320, 321, 322, 323, 336, 337, 338, 342, 343, 345, 346, 366, 367, 368, 377, 379, 400, 412, 468, 470, 471, 472, 473, 490, 493, 494], "tutori": 2, "twice": 500, "two": [0, 2, 14, 15, 17, 25, 83, 86, 88, 89, 90, 91, 119, 129, 134, 151, 154, 160, 167, 168, 170, 171, 177, 185, 187, 188, 189, 190, 191, 192, 193, 194, 196, 197, 198, 205, 210, 212, 228, 230, 234, 237, 293, 322, 337, 353, 360, 367, 426, 434, 491, 492, 493, 494, 495, 500], "txt": [2, 4], "type": [0, 1, 2, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 69, 79, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 116, 117, 118, 119, 122, 123, 124, 126, 127, 128, 129, 130, 131, 132, 134, 135, 136, 138, 139, 140, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 216, 222, 223, 224, 227, 228, 229, 230, 231, 232, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 266, 267, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 291, 292, 293, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 318, 319, 322, 334, 382, 411, 413, 414, 415, 416, 417, 418, 419, 420, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 491, 493, 495], "type_to_nam": 2, "typenam": [0, 1, 2], "typic": [0, 149, 351, 466, 491, 496], "u": [1, 2, 4, 189, 192, 193, 197, 363, 389, 486, 496], "u_": 467, "u_t": 467, "uint": [1, 2, 146], "uint16": [12, 326], "uint3": 1, "uint32": [12, 27, 28, 29, 30, 247, 326], "uint64": [12, 326], "uint8": [12, 326], "ultra": 6, "unabl": 9, "unam": 9, "unari": 491, "unchang": [148, 289, 398], "uncheck": 9, "uncompress": 271, "undefin": [0, 29, 113, 189, 190, 239, 251, 495], "under": [2, 195], "underli": [2, 310], "understand": [6, 414, 415], "unevalu": 142, "unexpect": [2, 19], "unflatten": 0, "unfreez": [334, 375], "unfrozen": 387, "unifi": 8, "uniform": [3, 334, 363, 377, 415, 417, 461, 488, 491, 494, 500], "uniformli": 258, "unintend": 0, "union": [19, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 85, 86, 87, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 181, 182, 183, 185, 192, 193, 216, 269, 290], "uniqu": [2, 488], "unique_ptr": 2, "unit": [340, 350, 352, 353, 354, 362, 396, 397, 399, 401, 414, 415, 416, 417, 421, 422, 423, 424, 425, 426, 430, 449, 450, 451, 453], "unittest": 9, "univers": 195, "unless": [6, 17, 177, 195, 464], "unlik": [6, 17, 177, 348, 349, 383], "unnecessari": [2, 6], "unnorm": [247, 433, 435], "unscal": 468, "unsign": [168, 243, 244, 326], "unsignedinteg": 12, "unspecifi": [16, 18, 19, 27, 28, 29, 30, 96, 109, 110, 111, 112, 166, 209, 211, 213, 227, 235, 239, 241, 262, 281, 282, 288, 292, 295, 301, 302, 309, 314, 501], "unsqueez": 6, "unsupport": 200, "until": [2, 496, 498], "unus": 2, "up": [1, 2, 6, 113, 491], "upcast": 2, "updat": [0, 1, 2, 5, 6, 7, 39, 95, 280, 316, 320, 322, 339, 370, 371, 377, 382, 383, 384, 389, 466, 468, 471, 473, 474, 475, 479, 480, 481, 482, 483, 484, 491, 492, 493, 496], "update_modul": 334, "uplo": [192, 193], "upon": [6, 320, 321], "upper": [189, 190, 192, 193, 198, 243, 254, 257, 258, 420], "upsampl": 334, "us": [0, 3, 5, 6, 7, 8, 9, 19, 39, 84, 113, 117, 120, 122, 123, 126, 127, 128, 130, 146, 148, 163, 168, 186, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 210, 217, 218, 219, 222, 224, 243, 244, 262, 263, 264, 265, 294, 307, 319, 322, 326, 328, 334, 337, 338, 348, 351, 352, 354, 360, 363, 367, 368, 370, 374, 381, 388, 390, 392, 393, 395, 398, 403, 411, 412, 416, 417, 424, 425, 434, 461, 464, 466, 467, 468, 470, 471, 472, 473, 474, 475, 488, 490, 491, 492, 493, 494, 495, 498, 500], "usag": [411, 491], "user": [2, 6, 334], "usual": [351, 392, 490, 496], "util": [1, 2, 6, 8, 9, 271, 334, 466, 492], "v": [6, 106, 149, 192, 334, 375, 497], "v_": [467, 469, 470, 471, 472, 478, 479], "v_t": [467, 469, 470, 471, 472, 478, 479], "val": [0, 31, 166], "valid": [7, 106, 163, 312, 319, 375, 387, 490], "valid_parameter_filt": 370, "valu": [0, 1, 5, 6, 12, 13, 17, 19, 27, 28, 51, 79, 83, 94, 141, 144, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 166, 175, 177, 191, 195, 197, 199, 216, 224, 231, 238, 242, 246, 247, 248, 250, 251, 252, 254, 257, 258, 265, 269, 295, 296, 308, 312, 317, 319, 320, 321, 322, 326, 337, 338, 340, 347, 348, 349, 350, 356, 359, 363, 367, 368, 374, 390, 391, 407, 409, 411, 413, 433, 434, 435, 436, 437, 438, 440, 441, 442, 443, 444, 445, 458, 464, 468, 471, 480, 481, 483, 484, 494], "value_and_grad": [7, 113, 334, 388, 464, 466, 477, 491, 494, 497, 498], "value_and_grad_fn": 496, "value_cach": 6, "value_dim": 390, "value_input_dim": 390, "value_output_dim": 390, "value_proj": 6, "valueerror": [113, 195, 377, 494], "values_hat": 6, "van": 195, "var": [0, 339, 355, 359, 361, 436], "variabl": [9, 95, 113, 120, 133, 141, 142, 143, 169, 184, 308, 311, 312, 491, 492, 493], "varianc": [0, 288, 309, 339, 355, 436], "variant": [6, 445, 472], "variou": 195, "vector": [0, 2, 5, 8, 176, 184, 195, 295, 311, 312, 351, 435, 493, 498], "verbos": [1, 146], "veri": [6, 390, 492, 496, 500], "verifi": [5, 9], "versa": 265, "version": [2, 4, 9, 117, 141, 143, 175, 205, 209, 243, 281, 312, 488, 494, 495], "versu": 491, "via": [9, 113, 474, 477, 492, 496, 497], "vice": 265, "video": 349, "view": [0, 3, 84, 497], "virtual": 2, "visual": 142, "vjp": [2, 113, 498], "vmap": [2, 113, 493, 494, 496, 498], "vmap_add": 494, "vocab_s": 6, "vocabulari": [351, 392], "void": [1, 2], "vt": 197, "w": [0, 1, 5, 100, 101, 104, 105, 117, 168, 192, 243, 244, 308, 321, 339, 342, 343, 345, 346, 348, 349, 363, 466, 479, 494], "w1": [6, 318], "w2": [6, 318], "w3": 6, "w_": [354, 360, 395, 467, 468, 469, 470, 471, 472, 473, 478, 479], "w_1": 243, "w_g": 243, "w_i": [117, 243], "w_in": 1, "w_q": 243, "w_star": 5, "w_stride": 1, "w_t": [467, 469, 470, 471, 472, 473, 478, 479], "wa": [4, 6, 84, 124, 126, 127, 492, 493, 496], "wai": [2, 6, 9, 334, 412, 491, 492, 493, 494, 495], "wait": [2, 6, 223], "walk": [6, 493], "walkthrough": 2, "walsh": 172, "want": [1, 6, 492, 493, 494, 500], "warm": [2, 491], "warmup": [482, 483], "warmup_init": 468, "watch": [6, 491], "wd": 473, "we": [0, 1, 2, 5, 6, 7, 113, 117, 126, 127, 168, 243, 244, 334, 351, 392, 400, 471, 473, 488, 490, 491, 492, 493, 494, 496, 500], "weight": [0, 5, 99, 100, 101, 102, 103, 104, 105, 145, 147, 320, 334, 377, 381, 392, 393, 433, 435, 464, 468, 471, 473, 475, 479, 494, 496], "weight_decai": [468, 471, 473, 479], "weight_fil": 6, "weights_fp16": 496, "well": [6, 334, 375, 387, 390, 496], "wen": 6, "went": 6, "were": [6, 500], "wet": 6, "what": [2, 6, 320], "whatsoev": 6, "whc": 348, "when": [0, 1, 2, 6, 8, 9, 95, 102, 113, 128, 189, 190, 192, 193, 194, 195, 197, 198, 200, 341, 342, 343, 344, 345, 346, 412, 416, 417, 433, 439, 445, 464, 466, 482, 488, 491, 492, 493, 500], "where": [0, 4, 7, 144, 177, 190, 243, 308, 312, 339, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 352, 354, 355, 359, 360, 361, 363, 374, 391, 394, 395, 409, 416, 417, 422, 423, 425, 436, 442, 448, 451, 453, 458, 475, 492, 494, 495], "wherea": 494, "whether": [141, 143, 146, 168, 192, 193, 198, 244, 354, 360, 374, 390, 395, 433, 436, 442], "which": [0, 1, 2, 6, 7, 8, 9, 19, 38, 84, 95, 102, 119, 122, 123, 126, 127, 128, 137, 141, 143, 148, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 163, 169, 175, 178, 179, 180, 181, 182, 184, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 200, 214, 225, 243, 247, 248, 262, 263, 265, 268, 269, 270, 271, 272, 286, 287, 295, 302, 307, 308, 311, 312, 316, 337, 338, 348, 349, 352, 367, 368, 370, 374, 398, 433, 435, 438, 442, 445, 461, 474, 475, 488, 491, 492, 493, 494, 495, 496, 500, 501], "while": [2, 3, 6, 9, 263, 398, 496, 497], "whistl": 2, "who": 6, "whose": [144, 316, 317], "why": 6, "wide": 496, "width": [337, 338, 339, 342, 343, 345, 346, 348, 349, 367, 368, 392, 393], "window": [9, 336, 337, 338, 366, 367, 368], "wipe": 9, "wire": 224, "wired_limit_mb": 224, "wise": [0, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 88, 89, 90, 93, 107, 108, 129, 130, 134, 135, 136, 138, 140, 164, 165, 170, 171, 177, 186, 187, 188, 201, 202, 203, 204, 205, 206, 207, 208, 212, 228, 230, 232, 234, 240, 260, 261, 264, 267, 275, 276, 277, 278, 284, 285, 291, 297, 298, 340, 348, 349, 358, 369, 391, 402, 421, 428, 429, 431, 432, 447, 448, 450, 453, 454, 455, 456, 491], "wish": 9, "with_logit": 433, "within": [0, 3, 29, 177], "without": [1, 6, 8, 289, 390, 460, 490, 491, 492, 493, 496, 497, 500], "wk": 6, "wl": 2, "wo": 6, "word": 0, "work": [2, 3, 6, 223, 491, 492, 493, 494, 495, 496], "workhors": 334, "world": [323, 492], "worri": [1, 496], "would": [2, 6, 412, 492, 493, 495, 496, 497, 500], "wq": 6, "wrap": [113, 334], "wrapper": 493, "write": [0, 1, 2, 6, 334, 497], "written": [2, 493], "wrong": 493, "wrt": 317, "wv": 6, "x": [0, 1, 2, 4, 5, 6, 7, 39, 91, 113, 122, 123, 127, 128, 135, 140, 141, 142, 145, 146, 147, 168, 172, 173, 175, 195, 244, 248, 253, 266, 271, 275, 305, 306, 313, 320, 322, 334, 336, 337, 338, 339, 340, 350, 352, 353, 355, 359, 361, 362, 363, 366, 367, 368, 369, 370, 391, 394, 396, 402, 403, 409, 412, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 445, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 464, 466, 473, 491, 492, 493, 494, 495, 496, 497, 498, 500], "x1": 434, "x2": 434, "x86_64": 9, "x_1": [434, 442], "x_2": [434, 442], "x_cast": 2, "x_grad": 1, "x_i": [432, 454, 455], "x_j": [454, 455], "x_offset": 2, "x_ptr": 2, "x_shape": 1, "x_stride": 2, "x_t": [354, 360, 395], "x_view": 497, "xcode": 9, "xcodeproj": 3, "xcrun": 9, "xf": 360, "xg": 360, "xi": 360, "xn": 354, "xo": 360, "xor": 90, "xr": 354, "xy": [0, 214], "xz": 354, "x\u00b2": 497, "y": [0, 2, 4, 5, 6, 7, 39, 113, 141, 142, 172, 175, 313, 334, 339, 348, 355, 359, 361, 363, 394, 437, 442, 445, 466, 469, 491, 492, 493, 494, 496, 497], "y_": [437, 441], "y_cast": 2, "y_hat": 334, "y_offset": 2, "y_ptr": 2, "y_stride": 2, "ye": 6, "year": 6, "yet": [6, 195, 334, 464, 475, 494, 495, 496, 498], "yield": [6, 7, 488], "you": [2, 3, 4, 6, 7, 8, 9, 224, 334, 403, 411, 461, 488, 491, 492, 493, 494, 495, 497, 499, 500], "your": [2, 6, 9, 464, 494, 496], "z": [2, 354, 491, 493, 496], "z_t": 354, "zeiler": 467, "zero": [0, 141, 144, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 191, 214, 221, 280, 304, 305, 306, 315, 334, 336, 337, 338, 347, 348, 349, 377, 413, 414, 415, 416, 417, 418, 419, 420, 461, 466, 468, 493, 495], "zero_grad": 494, "zeros_lik": 0, "zhang": 6, "zip": [6, 7], "zip_saf": 2}, "titles": ["Operations", "Custom Metal Kernels", "Custom Extensions in MLX", "Metal Debugger", "Using MLX in C++", "Linear Regression", "LLM inference", "Multi-Layer Perceptron", "MLX", "Build and Install", "mlx.core.Device", "mlx.core.Dtype", "mlx.core.DtypeCategory", "mlx.core.abs", "mlx.core.add", "mlx.core.addmm", "mlx.core.all", "mlx.core.allclose", "mlx.core.any", "mlx.core.arange", "mlx.core.arccos", "mlx.core.arccosh", "mlx.core.arcsin", "mlx.core.arcsinh", "mlx.core.arctan", "mlx.core.arctan2", "mlx.core.arctanh", "mlx.core.argmax", "mlx.core.argmin", "mlx.core.argpartition", "mlx.core.argsort", "mlx.core.array", "mlx.core.array.T", "mlx.core.array.abs", "mlx.core.array.all", "mlx.core.array.any", "mlx.core.array.argmax", "mlx.core.array.argmin", "mlx.core.array.astype", "mlx.core.array.at", "mlx.core.array.conj", "mlx.core.array.cos", "mlx.core.array.cummax", "mlx.core.array.cummin", "mlx.core.array.cumprod", "mlx.core.array.cumsum", "mlx.core.array.diag", "mlx.core.array.diagonal", "mlx.core.array.dtype", "mlx.core.array.exp", "mlx.core.array.flatten", "mlx.core.array.item", "mlx.core.array.itemsize", "mlx.core.array.log", "mlx.core.array.log10", "mlx.core.array.log1p", "mlx.core.array.log2", "mlx.core.array.logsumexp", "mlx.core.array.max", "mlx.core.array.mean", "mlx.core.array.min", "mlx.core.array.moveaxis", "mlx.core.array.nbytes", "mlx.core.array.ndim", "mlx.core.array.prod", "mlx.core.array.reciprocal", "mlx.core.array.reshape", "mlx.core.array.round", "mlx.core.array.rsqrt", "mlx.core.array.shape", "mlx.core.array.sin", "mlx.core.array.size", "mlx.core.array.split", "mlx.core.array.sqrt", "mlx.core.array.square", "mlx.core.array.squeeze", "mlx.core.array.std", "mlx.core.array.sum", "mlx.core.array.swapaxes", "mlx.core.array.tolist", "mlx.core.array.transpose", "mlx.core.array.var", "mlx.core.array.view", "mlx.core.array_equal", "mlx.core.as_strided", "mlx.core.atleast_1d", "mlx.core.atleast_2d", "mlx.core.atleast_3d", "mlx.core.bitwise_and", "mlx.core.bitwise_or", "mlx.core.bitwise_xor", "mlx.core.block_masked_mm", "mlx.core.broadcast_to", "mlx.core.ceil", "mlx.core.clip", "mlx.core.compile", "mlx.core.concatenate", "mlx.core.conj", "mlx.core.conjugate", "mlx.core.conv1d", "mlx.core.conv2d", "mlx.core.conv3d", "mlx.core.conv_general", "mlx.core.conv_transpose1d", "mlx.core.conv_transpose2d", "mlx.core.conv_transpose3d", "mlx.core.convolve", "mlx.core.cos", "mlx.core.cosh", "mlx.core.cummax", "mlx.core.cummin", "mlx.core.cumprod", "mlx.core.cumsum", "mlx.core.custom_function", "mlx.core.default_device", "mlx.core.default_stream", "mlx.core.degrees", "mlx.core.dequantize", "mlx.core.diag", "mlx.core.diagonal", "mlx.core.disable_compile", "mlx.core.distributed.Group", "mlx.core.distributed.all_gather", "mlx.core.distributed.all_sum", "mlx.core.distributed.init", "mlx.core.distributed.is_available", "mlx.core.distributed.recv", "mlx.core.distributed.recv_like", "mlx.core.distributed.send", "mlx.core.divide", "mlx.core.divmod", "mlx.core.einsum", "mlx.core.einsum_path", "mlx.core.enable_compile", "mlx.core.equal", "mlx.core.erf", "mlx.core.erfinv", "mlx.core.eval", "mlx.core.exp", "mlx.core.expand_dims", "mlx.core.expm1", "mlx.core.export_function", "mlx.core.export_to_dot", "mlx.core.exporter", "mlx.core.eye", "mlx.core.fast.layer_norm", "mlx.core.fast.metal_kernel", "mlx.core.fast.rms_norm", "mlx.core.fast.rope", "mlx.core.fast.scaled_dot_product_attention", "mlx.core.fft.fft", "mlx.core.fft.fft2", "mlx.core.fft.fftn", "mlx.core.fft.ifft", "mlx.core.fft.ifft2", "mlx.core.fft.ifftn", "mlx.core.fft.irfft", "mlx.core.fft.irfft2", "mlx.core.fft.irfftn", "mlx.core.fft.rfft", "mlx.core.fft.rfft2", "mlx.core.fft.rfftn", "mlx.core.finfo", "mlx.core.flatten", "mlx.core.floor", "mlx.core.floor_divide", "mlx.core.full", "mlx.core.gather_mm", "mlx.core.gather_qmm", "mlx.core.grad", "mlx.core.greater", "mlx.core.greater_equal", "mlx.core.hadamard_transform", "mlx.core.identity", "mlx.core.imag", "mlx.core.import_function", "mlx.core.inner", "mlx.core.isclose", "mlx.core.isfinite", "mlx.core.isinf", "mlx.core.isnan", "mlx.core.isneginf", "mlx.core.isposinf", "mlx.core.issubdtype", "mlx.core.jvp", "mlx.core.kron", "mlx.core.left_shift", "mlx.core.less", "mlx.core.less_equal", "mlx.core.linalg.cholesky", "mlx.core.linalg.cholesky_inv", "mlx.core.linalg.cross", "mlx.core.linalg.eigh", "mlx.core.linalg.eigvalsh", "mlx.core.linalg.inv", "mlx.core.linalg.norm", "mlx.core.linalg.qr", "mlx.core.linalg.svd", "mlx.core.linalg.tri_inv", "mlx.core.linspace", "mlx.core.load", "mlx.core.log", "mlx.core.log10", "mlx.core.log1p", "mlx.core.log2", "mlx.core.logaddexp", "mlx.core.logical_and", "mlx.core.logical_not", "mlx.core.logical_or", "mlx.core.logsumexp", "mlx.core.matmul", "mlx.core.max", "mlx.core.maximum", "mlx.core.mean", "mlx.core.meshgrid", "mlx.core.metal.clear_cache", "mlx.core.metal.device_info", "mlx.core.metal.get_active_memory", "mlx.core.metal.get_cache_memory", "mlx.core.metal.get_peak_memory", "mlx.core.metal.is_available", "mlx.core.metal.reset_peak_memory", "mlx.core.metal.set_cache_limit", "mlx.core.metal.set_memory_limit", "mlx.core.metal.set_wired_limit", "mlx.core.metal.start_capture", "mlx.core.metal.stop_capture", "mlx.core.min", "mlx.core.minimum", "mlx.core.moveaxis", "mlx.core.multiply", "mlx.core.nan_to_num", "mlx.core.negative", "mlx.core.new_stream", "mlx.core.not_equal", "mlx.core.ones", "mlx.core.ones_like", "mlx.core.outer", "mlx.core.pad", "mlx.core.partition", "mlx.core.power", "mlx.core.prod", "mlx.core.put_along_axis", "mlx.core.quantize", "mlx.core.quantized_matmul", "mlx.core.radians", "mlx.core.random.bernoulli", "mlx.core.random.categorical", "mlx.core.random.gumbel", "mlx.core.random.key", "mlx.core.random.laplace", "mlx.core.random.multivariate_normal", "mlx.core.random.normal", "mlx.core.random.permutation", "mlx.core.random.randint", "mlx.core.random.seed", "mlx.core.random.split", "mlx.core.random.truncated_normal", "mlx.core.random.uniform", "mlx.core.real", "mlx.core.reciprocal", "mlx.core.remainder", "mlx.core.repeat", "mlx.core.reshape", "mlx.core.right_shift", "mlx.core.roll", "mlx.core.round", "mlx.core.rsqrt", "mlx.core.save", "mlx.core.save_gguf", "mlx.core.save_safetensors", "mlx.core.savez", "mlx.core.savez_compressed", "mlx.core.set_default_device", "mlx.core.set_default_stream", "mlx.core.sigmoid", "mlx.core.sign", "mlx.core.sin", "mlx.core.sinh", "mlx.core.slice", "mlx.core.slice_update", "mlx.core.softmax", "mlx.core.sort", "mlx.core.split", "mlx.core.sqrt", "mlx.core.square", "mlx.core.squeeze", "mlx.core.stack", "mlx.core.std", "mlx.core.stop_gradient", "mlx.core.stream", "mlx.core.subtract", "mlx.core.sum", "mlx.core.swapaxes", "mlx.core.synchronize", "mlx.core.take", "mlx.core.take_along_axis", "mlx.core.tan", "mlx.core.tanh", "mlx.core.tensordot", "mlx.core.tile", "mlx.core.topk", "mlx.core.trace", "mlx.core.transpose", "mlx.core.tri", "mlx.core.tril", "mlx.core.triu", "mlx.core.unflatten", "mlx.core.value_and_grad", "mlx.core.var", "mlx.core.view", "mlx.core.vjp", "mlx.core.vmap", "mlx.core.where", "mlx.core.zeros", "mlx.core.zeros_like", "mlx.nn.quantize", "mlx.nn.value_and_grad", "mlx.optimizers.clip_grad_norm", "mlx.utils.tree_flatten", "mlx.utils.tree_map", "mlx.utils.tree_map_with_path", "mlx.utils.tree_reduce", "mlx.utils.tree_unflatten", "mlx.core.Stream", "Array", "Data Types", "Devices and Streams", "Distributed Communication", "Export Functions", "Fast", "FFT", "Linear Algebra", "Metal", "Neural Networks", "mlx.nn.ALiBi", "mlx.nn.AvgPool1d", "mlx.nn.AvgPool2d", "mlx.nn.AvgPool3d", "mlx.nn.BatchNorm", "mlx.nn.CELU", "mlx.nn.Conv1d", "mlx.nn.Conv2d", "mlx.nn.Conv3d", "mlx.nn.ConvTranspose1d", "mlx.nn.ConvTranspose2d", "mlx.nn.ConvTranspose3d", "mlx.nn.Dropout", "mlx.nn.Dropout2d", "mlx.nn.Dropout3d", "mlx.nn.ELU", "mlx.nn.Embedding", "mlx.nn.GELU", "mlx.nn.GLU", "mlx.nn.GRU", "mlx.nn.GroupNorm", "mlx.nn.HardShrink", "mlx.nn.HardTanh", "mlx.nn.Hardswish", "mlx.nn.InstanceNorm", "mlx.nn.LSTM", "mlx.nn.LayerNorm", "mlx.nn.LeakyReLU", "mlx.nn.Linear", "mlx.nn.LogSigmoid", "mlx.nn.LogSoftmax", "mlx.nn.MaxPool1d", "mlx.nn.MaxPool2d", "mlx.nn.MaxPool3d", "mlx.nn.Mish", "mlx.nn.Module.apply", "mlx.nn.Module.apply_to_modules", "mlx.nn.Module.children", "mlx.nn.Module.eval", "mlx.nn.Module.filter_and_map", "mlx.nn.Module.freeze", "mlx.nn.Module.leaf_modules", "mlx.nn.Module.load_weights", "mlx.nn.Module.modules", "mlx.nn.Module.named_modules", "mlx.nn.Module.parameters", "mlx.nn.Module.save_weights", "mlx.nn.Module.set_dtype", "mlx.nn.Module.state", "mlx.nn.Module.train", "mlx.nn.Module.trainable_parameters", "mlx.nn.Module.training", "mlx.nn.Module.unfreeze", "mlx.nn.Module.update", "mlx.nn.Module.update_modules", "mlx.nn.MultiHeadAttention", "mlx.nn.PReLU", "mlx.nn.QuantizedEmbedding", "mlx.nn.QuantizedLinear", "mlx.nn.RMSNorm", "mlx.nn.RNN", "mlx.nn.ReLU", "mlx.nn.ReLU6", "mlx.nn.RoPE", "mlx.nn.SELU", "mlx.nn.Sequential", "mlx.nn.SiLU", "mlx.nn.Sigmoid", "mlx.nn.SinusoidalPositionalEncoding", "mlx.nn.Softmax", "mlx.nn.Softmin", "mlx.nn.Softplus", "mlx.nn.Softshrink", "mlx.nn.Softsign", "mlx.nn.Step", "mlx.nn.Tanh", "mlx.nn.Transformer", "mlx.nn.Upsample", "mlx.nn.init.constant", "mlx.nn.init.glorot_normal", "mlx.nn.init.glorot_uniform", "mlx.nn.init.he_normal", "mlx.nn.init.he_uniform", "mlx.nn.init.identity", "mlx.nn.init.normal", "mlx.nn.init.uniform", "mlx.nn.celu", "mlx.nn.elu", "mlx.nn.gelu", "mlx.nn.gelu_approx", "mlx.nn.gelu_fast_approx", "mlx.nn.glu", "mlx.nn.hard_shrink", "mlx.nn.hard_tanh", "mlx.nn.hardswish", "mlx.nn.leaky_relu", "mlx.nn.log_sigmoid", "mlx.nn.log_softmax", "mlx.nn.losses.binary_cross_entropy", "mlx.nn.losses.cosine_similarity_loss", "mlx.nn.losses.cross_entropy", "mlx.nn.losses.gaussian_nll_loss", "mlx.nn.losses.hinge_loss", "mlx.nn.losses.huber_loss", "mlx.nn.losses.kl_div_loss", "mlx.nn.losses.l1_loss", "mlx.nn.losses.log_cosh_loss", "mlx.nn.losses.margin_ranking_loss", "mlx.nn.losses.mse_loss", "mlx.nn.losses.nll_loss", "mlx.nn.losses.smooth_l1_loss", "mlx.nn.losses.triplet_loss", "mlx.nn.mish", "mlx.nn.prelu", "mlx.nn.relu", "mlx.nn.relu6", "mlx.nn.selu", "mlx.nn.sigmoid", "mlx.nn.silu", "mlx.nn.softmax", "mlx.nn.softmin", "mlx.nn.softplus", "mlx.nn.softshrink", "mlx.nn.step", "mlx.nn.tanh", "Functions", "Initializers", "Layers", "Loss Functions", "Module", "Operations", "Optimizers", "mlx.optimizers.AdaDelta", "mlx.optimizers.Adafactor", "mlx.optimizers.Adagrad", "mlx.optimizers.Adam", "mlx.optimizers.AdamW", "mlx.optimizers.Adamax", "mlx.optimizers.Lion", "mlx.optimizers.Optimizer.apply_gradients", "mlx.optimizers.Optimizer.init", "mlx.optimizers.Optimizer.state", "mlx.optimizers.Optimizer.update", "mlx.optimizers.RMSprop", "mlx.optimizers.SGD", "mlx.optimizers.cosine_decay", "mlx.optimizers.exponential_decay", "mlx.optimizers.join_schedules", "mlx.optimizers.linear_schedule", "mlx.optimizers.step_decay", "Common Optimizers", "Optimizer", "Schedulers", "Random", "Transforms", "Tree Utils", "Compilation", "Distributed Communication", "Exporting Functions", "Function Transforms", "Indexing Arrays", "Lazy Evaluation", "Conversion to NumPy and Other Frameworks", "Quick Start Guide", "Saving and Loading Arrays", "Unified Memory", "Using Streams"], "titleterms": {"A": 500, "In": 495, "The": 334, "ab": [13, 33], "adadelta": 467, "adafactor": 468, "adagrad": 469, "adam": 470, "adamax": 472, "adamw": 471, "add": 14, "addmm": 15, "algebra": 332, "alibi": 335, "all": [6, 16, 34, 492], "all_gath": 122, "all_sum": 123, "allclos": 17, "ani": [18, 35], "api": [8, 9], "appli": 370, "apply_gradi": 474, "apply_to_modul": 371, "arang": 19, "arcco": 20, "arccosh": 21, "arcsin": 22, "arcsinh": 23, "arctan": 24, "arctan2": 25, "arctanh": 26, "argmax": [27, 36], "argmin": [28, 37], "argpartit": 29, "argsort": 30, "arrai": [31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 325, 495, 499], "array_equ": 83, "as_strid": 84, "astyp": 38, "atleast_1d": 85, "atleast_2d": 86, "atleast_3d": 87, "attent": 6, "automat": 494, "avgpool1d": 336, "avgpool2d": 337, "avgpool3d": 338, "back": 2, "basic": [491, 493, 498], "batchnorm": 339, "benchmark": 6, "bernoulli": 246, "binari": 9, "binary_cross_entropi": 433, "bind": 2, "bitwise_and": 88, "bitwise_or": 89, "bitwise_xor": 90, "block_masked_mm": 91, "broadcast_to": 92, "build": [2, 9], "c": [4, 8, 9, 493], "categor": 247, "ceil": 93, "celu": [340, 421], "children": 372, "choleski": 189, "cholesky_inv": 190, "class": 334, "clear_cach": 215, "clip": 94, "clip_grad_norm": 318, "cmake": 2, "co": [41, 107], "code": [2, 6], "common": 485, "commun": [328, 492], "compil": [95, 491], "complex": 1, "comput": 496, "concaten": 96, "conj": [40, 97], "conjug": 98, "constant": 413, "conv1d": [99, 341], "conv2d": [100, 342], "conv3d": [101, 343], "conv_gener": 102, "conv_transpose1d": 103, "conv_transpose2d": 104, "conv_transpose3d": 105, "convers": 497, "convert": 6, "convolv": 106, "convtranspose1d": 344, "convtranspose2d": 345, "convtranspose3d": 346, "core": [10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 324], "cosh": 108, "cosine_decai": 480, "cosine_similarity_loss": 434, "cpu": 2, "cross": 191, "cross_entropi": 435, "cummax": [42, 109], "cummin": [43, 110], "cumprod": [44, 111], "cumsum": [45, 112], "custom": [1, 2], "custom_funct": 113, "data": 326, "debug": 491, "debugg": 3, "default_devic": 114, "default_stream": 115, "degre": 116, "dequant": 117, "devic": [10, 327], "device_info": 216, "diag": [46, 118], "diagon": [47, 119], "differ": 495, "differenti": 494, "disable_compil": 120, "distribut": [121, 122, 123, 124, 125, 126, 127, 128, 328, 492], "divid": 129, "divmod": 130, "download": [2, 6], "dropout": 347, "dropout2d": 348, "dropout3d": 349, "dtype": [11, 48], "dtypecategori": 12, "eigh": 192, "eigvalsh": 193, "einsum": 131, "einsum_path": 132, "elu": [350, 422], "embed": 351, "enable_compil": 133, "encod": 6, "end": 2, "equal": 134, "erf": 135, "erfinv": 136, "eval": [137, 373], "evalu": 496, "exampl": [1, 2, 8, 491, 492, 493, 500], "exp": [49, 138], "expand_dim": 139, "expm1": 140, "exponential_decai": 481, "export": [143, 329, 493], "export_funct": 141, "export_to_dot": 142, "extens": 2, "ey": 144, "fast": [145, 146, 147, 148, 149, 330], "fft": [150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 331], "fft2": 151, "fftn": 152, "filter_and_map": 374, "finfo": 162, "flatten": [50, 163], "floor": 164, "floor_divid": 165, "format": 499, "found": 9, "framework": 497, "freez": 375, "from": [9, 495], "full": [6, 166], "function": [329, 460, 463, 491, 493, 494, 498], "further": 8, "gather_mm": 167, "gather_qmm": 168, "gaussian_nll_loss": 436, "gelu": [352, 423], "gelu_approx": 424, "gelu_fast_approx": 425, "gener": 6, "get": 492, "get_active_memori": 217, "get_cache_memori": 218, "get_peak_memori": 219, "glorot_norm": 414, "glorot_uniform": 415, "glu": [353, 426], "gpu": 2, "grad": [169, 334], "graph": [491, 496, 498], "greater": 170, "greater_equ": 171, "grid": 1, "group": 121, "groupnorm": 355, "gru": 354, "guid": 498, "gumbel": 248, "hadamard_transform": 172, "hard_shrink": 427, "hard_tanh": 428, "hardshrink": 356, "hardswish": [358, 429], "hardtanh": 357, "he_norm": 416, "he_uniform": 417, "hinge_loss": 437, "host": 492, "huber_loss": 438, "ident": [173, 418], "ifft": 153, "ifft2": 154, "ifftn": 155, "imag": 174, "implement": [2, 6], "import": 493, "import_funct": 175, "index": 495, "infer": 6, "init": [124, 413, 414, 415, 416, 417, 418, 419, 420, 475], "initi": 461, "inner": 176, "inspect": 334, "instal": [8, 9, 492], "instancenorm": 359, "introduc": 2, "inv": 194, "irfft": 156, "irfft2": 157, "irfftn": 158, "is_avail": [125, 220], "isclos": 177, "isfinit": 178, "isinf": 179, "isnan": 180, "isneginf": 181, "isposinf": 182, "issubdtyp": 183, "item": 51, "items": 52, "jax": 497, "join_schedul": 482, "jvp": 184, "kei": 249, "kernel": 1, "kl_div_loss": 439, "kron": 185, "l1_loss": 440, "laplac": 250, "layer": [6, 7, 462], "layer_norm": 145, "layernorm": 361, "lazi": 496, "leaf_modul": 376, "leaky_relu": 430, "leakyrelu": 362, "left_shift": 186, "less": 187, "less_equ": 188, "linalg": [189, 190, 191, 192, 193, 194, 195, 196, 197, 198], "linear": [5, 332, 363], "linear_schedul": 483, "linspac": 199, "lion": 473, "llm": 6, "load": [6, 200, 466, 499], "load_weight": 377, "log": [53, 201], "log10": [54, 202], "log1p": [55, 203], "log2": [56, 204], "log_cosh_loss": 441, "log_sigmoid": 431, "log_softmax": 432, "logaddexp": 205, "logical_and": 206, "logical_not": 207, "logical_or": 208, "logsigmoid": 364, "logsoftmax": 365, "logsumexp": [57, 209], "loss": [433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 463], "lstm": 360, "margin_ranking_loss": 442, "matmul": 210, "max": [58, 211], "maximum": 212, "maxpool1d": 366, "maxpool2d": 367, "maxpool3d": 368, "mean": [59, 213], "memori": 500, "meshgrid": 214, "metal": [1, 3, 9, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 333], "metal_kernel": 146, "min": [60, 227], "minim": 9, "minimum": 228, "mish": [369, 447], "mlx": [2, 4, 8, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484], "model": 6, "modul": [334, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 464, 493], "more": 493, "moveaxi": [61, 229], "mpi": 492, "mse_loss": 443, "multi": 7, "multiheadattent": 390, "multipl": 493, "multipli": 230, "multivariate_norm": 251, "named_modul": 379, "nan_to_num": 231, "nbyte": 62, "ndim": 63, "neg": 232, "network": 334, "neural": 334, "new_stream": 233, "nll_loss": 444, "nn": [316, 317, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459], "norm": 195, "normal": [252, 419], "not_equ": 234, "numpi": [495, 497], "ones": 235, "ones_lik": 236, "onli": 496, "oper": [0, 2, 465], "optim": [318, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486], "option": 9, "other": 497, "outer": 237, "packag": 4, "pad": 238, "paramet": [334, 380], "partit": 239, "perceptron": 7, "permut": 253, "place": 495, "power": 240, "prelu": [391, 448], "primit": 2, "prod": [64, 241], "pure": 491, "put": 6, "put_along_axi": 242, "python": [2, 8, 9], "pytorch": 497, "qr": 196, "quantiz": [243, 316], "quantized_matmul": 244, "quantizedembed": 392, "quantizedlinear": 393, "quick": [334, 498], "radian": 245, "randint": 254, "random": [246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 488], "read": 8, "real": 259, "reciproc": [65, 260], "recv": 126, "recv_lik": 127, "reduc": 492, "refer": 8, "regress": 5, "relu": [396, 449], "relu6": [397, 450], "remaind": 261, "remot": 492, "repeat": 262, "requir": 9, "reset_peak_memori": 221, "reshap": [66, 263], "result": 2, "rfft": 159, "rfft2": 160, "rfftn": 161, "right_shift": 264, "rms_norm": 147, "rmsnorm": 394, "rmsprop": 478, "rnn": 395, "roll": 265, "rope": [148, 398], "round": [67, 266], "rsqrt": [68, 267], "sampl": 1, "save": [268, 466, 499], "save_gguf": 269, "save_safetensor": 270, "save_weight": 381, "savez": 271, "savez_compress": 272, "scaled_dot_product_attent": 149, "schedul": 487, "script": [2, 6], "seed": 255, "selu": [399, 451], "send": 128, "sequenti": 400, "serial": 499, "set": 492, "set_cache_limit": 222, "set_default_devic": 273, "set_default_stream": 274, "set_dtyp": 382, "set_memory_limit": 223, "set_wired_limit": 224, "setuptool": 2, "sgd": 479, "shape": [1, 69], "shapeless": [491, 493], "shell": 9, "sigmoid": [275, 402, 452], "sign": 276, "silu": [401, 453], "simpl": [1, 500], "sin": [70, 277], "sinh": 278, "sinusoidalpositionalencod": 403, "size": [9, 71], "slice": 279, "slice_upd": 280, "smooth_l1_loss": 445, "softmax": [281, 404, 454], "softmin": [405, 455], "softplu": [406, 456], "softshrink": [407, 457], "softsign": 408, "sort": 282, "sourc": 9, "specifi": 501, "speedup": 491, "split": [72, 256, 283], "sqrt": [73, 284], "squar": [74, 285], "squeez": [75, 286], "stack": 287, "start": [334, 492, 498], "start_captur": 225, "state": [383, 476], "std": [76, 288], "step": [409, 458], "step_decai": 484, "stop_captur": 226, "stop_gradi": 289, "stream": [290, 324, 327, 501], "stride": 1, "subtract": 291, "sum": [77, 292], "support": 326, "svd": 197, "swapax": [78, 293], "synchron": 294, "t": 32, "take": 295, "take_along_axi": 296, "tan": 297, "tanh": [298, 410, 459], "tensordot": 299, "tensorflow": 497, "tile": 300, "togeth": 6, "tolist": 79, "topk": 301, "trace": [302, 493], "train": [384, 386, 491, 492], "trainable_paramet": 385, "transform": [2, 411, 489, 491, 493, 494, 496, 498], "transpos": [80, 303], "tree": 490, "tree_flatten": 319, "tree_map": 320, "tree_map_with_path": 321, "tree_reduc": 322, "tree_unflatten": 323, "tri": 304, "tri_inv": 198, "tril": 305, "triplet_loss": 446, "triu": 306, "troubleshoot": 9, "truncated_norm": 257, "tune": 492, "type": 326, "unflatten": 307, "unfreez": 387, "unifi": 500, "uniform": [258, 420], "up": 492, "updat": [334, 388, 477, 495], "update_modul": 389, "upsampl": 412, "us": [1, 2, 4, 496, 501], "usag": [2, 8], "util": [319, 320, 321, 322, 323, 490], "valu": 334, "value_and_grad": [308, 317], "var": [81, 309], "variabl": 4, "vector": 494, "view": [82, 310], "vjp": [1, 311], "vmap": 312, "weight": 6, "what": 496, "when": 496, "where": 313, "why": 496, "workflow": 3, "x86": 9, "xcode": 3, "you": 496, "zero": 314, "zeros_lik": 315}}) \ No newline at end of file diff --git a/docs/build/html/simd_8h.html b/docs/build/html/simd_8h.html new file mode 100644 index 000000000..b39f478bd --- /dev/null +++ b/docs/build/html/simd_8h.html @@ -0,0 +1,122 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/simd.h File Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      simd.h File Reference
      +
      +
      +
      #include "mlx/backend/cpu/simd/math.h"
      +#include "mlx/backend/cpu/simd/type.h"
      +
      +

      Go to the source code of this file.

      +
      +
      + + + + diff --git a/docs/build/html/simd_8h_source.html b/docs/build/html/simd_8h_source.html new file mode 100644 index 000000000..862205dba --- /dev/null +++ b/docs/build/html/simd_8h_source.html @@ -0,0 +1,124 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/simd.h Source File + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      simd.h
      +
      +
      +Go to the documentation of this file.
      1#pragma once
      +
      2
      +
      3#include "mlx/backend/cpu/simd/math.h"
      +
      4#include "mlx/backend/cpu/simd/type.h"
      + + +
      +
      + + + + diff --git a/docs/build/html/sort_8h.html b/docs/build/html/sort_8h.html index 5e836ec74..457196cc3 100644 --- a/docs/build/html/sort_8h.html +++ b/docs/build/html/sort_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/sort.h File Reference @@ -52,7 +52,7 @@ - +
      @@ -109,7 +109,7 @@ $(function(){initNavTree('kernels_2steel_2conv_2kernels_2steel__conv_8h.html',''
      #include <metal_stdlib>
      -

      Go to the source code of this file.

      +

      Go to the source code of this file.

      @@ -181,8 +181,8 @@ template<typename T, int BM, int BN, int BK, int WM, int WN, int N_CHANNELS = diff --git a/docs/build/html/steel__conv_8h.js b/docs/build/html/steel__conv_8h.js new file mode 100644 index 000000000..8c3aec1d0 --- /dev/null +++ b/docs/build/html/steel__conv_8h.js @@ -0,0 +1,4 @@ +var steel__conv_8h = +[ + [ "implicit_gemm_conv_2d", "steel__conv_8h.html#a5728711d1c2ee4038457babb7ac12888", null ] +]; \ No newline at end of file diff --git a/docs/build/html/kernels_2steel_2conv_2kernels_2steel__conv_8h_source.html b/docs/build/html/steel__conv_8h_source.html similarity index 95% rename from docs/build/html/kernels_2steel_2conv_2kernels_2steel__conv_8h_source.html rename to docs/build/html/steel__conv_8h_source.html index 54ee4c3d2..a4ad76599 100644 --- a/docs/build/html/kernels_2steel_2conv_2kernels_2steel__conv_8h_source.html +++ b/docs/build/html/steel__conv_8h_source.html @@ -3,7 +3,7 @@ - +MLX: mlx/backend/metal/kernels/steel/conv/kernels/steel_conv.h Source File @@ -52,7 +52,7 @@

      Functions

      - +
      @@ -105,7 +105,7 @@ $(function(){initNavTree('kernels_2steel_2conv_2kernels_2steel__conv_8h_source.h
      steel_conv.h
      -Go to the documentation of this file.
      1// Copyright © 2024 Apple Inc.
      +Go to the documentation of this file.
      1// Copyright © 2024 Apple Inc.
      2
      3#include <metal_stdlib>
      4
      @@ -122,7 +122,7 @@ $(function(){initNavTree('kernels_2steel_2conv_2kernels_2steel__conv_8h_source.h
      15 bool SMALL_FILTER = false>
      16[[kernel, max_total_threads_per_threadgroup(WM* WN * 32)]] void
      -
      17implicit_gemm_conv_2d(
      +
      17implicit_gemm_conv_2d(
      18 const device T* A [[buffer(0)]],
      19 const device T* B [[buffer(1)]],
      20 device T* C [[buffer(2)]],
      @@ -283,12 +283,12 @@ $(function(){initNavTree('kernels_2steel_2conv_2kernels_2steel__conv_8h_source.h
      175 mma_op.store_result_safe(C, ldc, short2(tgp_bn, tgp_bm));
      176}
      -
      implicit_gemm_conv_2d
      void implicit_gemm_conv_2d(const device T *A, const device T *B, device T *C, const constant MLXConvParams< 2 > *params, const constant ImplicitGemmConv2DParams *gemm_params, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)
      Definition steel_conv.h:17
      metal
      Definition bf16_math.h:226
      metal::min
      METAL_FUNC bfloat16_t min(bfloat16_t x, bfloat16_t y)
      Definition bf16_math.h:232
      mlx::steel
      Definition attn.h:19
      +
      implicit_gemm_conv_2d
      void implicit_gemm_conv_2d(const device T *A, const device T *B, device T *C, const constant MLXConvParams< 2 > *params, const constant ImplicitGemmConv2DParams *gemm_params, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)
      Definition steel_conv.h:17
      MLXConvParams
      Definition params.h:6
      -
      mlx::steel::BlockMMA
      Definition mma.h:377
      +
      mlx::steel::BlockMMA
      Definition mma.h:449
      mlx::steel::Conv2DInputBlockLoaderLargeFilter
      Definition loader_channel_l.h:23
      mlx::steel::Conv2DInputBlockLoaderSmallChannels
      Definition loader_channel_n.h:59
      mlx::steel::Conv2DInputBlockLoaderSmallFilter
      Definition loader_channel_l.h:171
      @@ -299,8 +299,8 @@ $(function(){initNavTree('kernels_2steel_2conv_2kernels_2steel__conv_8h_source.h diff --git a/docs/build/html/steel__conv__general_8h.html b/docs/build/html/steel__conv__general_8h.html index ce1f6401b..c65523806 100644 --- a/docs/build/html/steel__conv__general_8h.html +++ b/docs/build/html/steel__conv__general_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/steel/conv/kernels/steel_conv_general.h File Reference @@ -52,7 +52,7 @@
      - + - - - - - - - - - - - - - - -
      -
      - - - - - - - -
      -
      MLX -
      -
      - -   - - - - -
      -
      -
      - - - - -
      -
      - -
      -
      -
      - -
      - -
      -
      - - -
      -
      -
      -
      -
      -
      Loading...
      -
      Searching...
      -
      No Matches
      -
      -
      -
      -
      - -
      -
      -Variables
      -
      steel_gemm.h File Reference
      -
      -
      - -

      Go to the source code of this file.

      - - - - - - - - - - - - -

      -Variables

      constexpr std::string_view steel_gemm_fused_kernels
       
      constexpr std::string_view steel_gemm_masked_kernels
       
      constexpr std::string_view steel_gemm_splitk_kernels
       
      constexpr std::string_view steel_gemm_splitk_accum_kernels
       
      constexpr std::string_view steel_gemm_splitk_accum_axbpy_kernels
       
      -

      Variable Documentation

      - -

      ◆ steel_gemm_fused_kernels

      - -
      -
      - - - - - -
      - - - - -
      std::string_view steel_gemm_fused_kernels
      -       		-constexpr
      -
      -Initial value:
      = R"(
      -
      template [[host_name("{name}")]]
      -
      [[kernel]] void gemm<{itype}, {bm}, {bn}, {bk}, {wm}, {wn}, {trans_a}, {trans_b}, float>(
      -
      const device {itype} *A [[buffer(0)]],
      -
      const device {itype} *B [[buffer(1)]],
      -
      const device {itype} *C [[buffer(2), function_constant(use_out_source)]],
      -
      device {itype} *D [[buffer(3)]],
      -
      const constant GEMMParams* params [[buffer(4)]],
      -
      const constant GEMMAddMMParams* addmm_params [[buffer(5), function_constant(use_out_source)]],
      -
      const constant int* batch_shape [[buffer(6)]],
      -
      const constant int64_t* batch_strides [[buffer(7)]],
      -
      const constant uint32_t* lhs_indices [[buffer(10), function_constant(do_gather)]],
      -
      const constant uint32_t* rhs_indices [[buffer(11), function_constant(do_gather)]],
      -
      const constant uint32_t* C_indices [[buffer(12), function_constant(gather_bias)]],
      -
      const constant int* operand_shape [[buffer(13), function_constant(do_gather)]],
      -
      const constant int64_t* operand_strides [[buffer(14), function_constant(do_gather)]],
      -
      const constant packed_int3& operand_batch_ndim [[buffer(15), function_constant(do_gather)]],
      -
      uint simd_lane_id [[thread_index_in_simdgroup]],
      -
      uint simd_group_id [[simdgroup_index_in_threadgroup]],
      -
      uint3 tid [[threadgroup_position_in_grid]],
      -
      uint3 lid [[thread_position_in_threadgroup]]);
      -
      )"
      -
      -
      -
      - -

      ◆ steel_gemm_masked_kernels

      - -
      -
      - - - - - -
      - - - - -
      std::string_view steel_gemm_masked_kernels
      -       		-constexpr
      -
      -Initial value:
      = R"(
      -
      template [[host_name("{name}")]] [[kernel]] void
      -
      block_masked_gemm<
      -
      {itype},
      -
      {outmasktype},
      -
      {opmasktype},
      -
      {bm},
      -
      {bn},
      -
      {bk},
      -
      {wm},
      -
      {wn},
      -
      {trans_a},
      -
      {trans_b},
      -
      {mn_aligned},
      -
      {k_aligned}>(
      -
      const device {itype}* A [[buffer(0)]],
      -
      const device {itype}* B [[buffer(1)]],
      -
      device {itype}* D [[buffer(3)]],
      -
      const constant GEMMParams* params [[buffer(4)]],
      -
      const constant int* batch_shape [[buffer(6)]],
      -
      const constant int64_t* batch_strides [[buffer(7)]],
      -
      const device {outmasktype}* out_mask [[buffer(10)]],
      -
      const device {opmasktype}* lhs_mask [[buffer(11)]],
      -
      const device {opmasktype}* rhs_mask [[buffer(12)]],
      -
      const constant int* mask_strides [[buffer(13)]],
      -
      uint simd_lane_id [[thread_index_in_simdgroup]],
      -
      uint simd_group_id [[simdgroup_index_in_threadgroup]],
      -
      uint3 tid [[threadgroup_position_in_grid]],
      -
      uint3 lid [[thread_position_in_threadgroup]]);
      -
      )"
      -
      -
      -
      - -

      ◆ steel_gemm_splitk_accum_axbpy_kernels

      - -
      -
      - - - - - -
      - - - - -
      std::string_view steel_gemm_splitk_accum_axbpy_kernels
      -       		-constexpr
      -
      -Initial value:
      = R"(
      -
      template [[host_name("{name}")]] [[kernel]] void
      -
      gemm_splitk_accum_axpby<{atype}, {otype}>(
      -
      const device {atype}* C_split [[buffer(0)]],
      -
      device {otype}* D [[buffer(1)]],
      -
      const constant int& k_partitions [[buffer(2)]],
      -
      const constant int& partition_stride [[buffer(3)]],
      -
      const constant int& ldd [[buffer(4)]],
      -
      const device {otype}* C [[buffer(5)]],
      -
      const constant int& ldc [[buffer(6)]],
      -
      const constant int& fdc [[buffer(7)]],
      -
      const constant float& alpha [[buffer(8)]],
      -
      const constant float& beta [[buffer(9)]],
      -
      uint2 gid [[thread_position_in_grid]]);
      -
      )"
      -
      -
      -
      - -

      ◆ steel_gemm_splitk_accum_kernels

      - -
      -
      - - - - - -
      - - - - -
      std::string_view steel_gemm_splitk_accum_kernels
      -       		-constexpr
      -
      -Initial value:
      = R"(
      -
      template [[host_name("{name}")]] [[kernel]] void
      -
      gemm_splitk_accum<{atype}, {otype}>(
      -
      const device {atype}* C_split [[buffer(0)]],
      -
      device {otype}* D [[buffer(1)]],
      -
      const constant int& k_partitions [[buffer(2)]],
      -
      const constant int& partition_stride [[buffer(3)]],
      -
      const constant int& ldd [[buffer(4)]],
      -
      uint2 gid [[thread_position_in_grid]]);
      -
      )"
      -
      -
      -
      - -

      ◆ steel_gemm_splitk_kernels

      - -
      -
      - - - - - -
      - - - - -
      std::string_view steel_gemm_splitk_kernels
      -       		-constexpr
      -
      -Initial value:
      = R"(
      -
      template [[host_name("{name}")]] [[kernel]] void
      -
      gemm_splitk<
      -
      {itype},
      -
      {otype},
      -
      {bm},
      -
      {bn},
      -
      {bk},
      -
      {wm},
      -
      {wn},
      -
      {trans_a},
      -
      {trans_b},
      -
      {mn_aligned},
      -
      {k_aligned}>(
      -
      const device {itype}* A [[buffer(0)]],
      -
      const device {itype}* B [[buffer(1)]],
      -
      device {otype}* C [[buffer(2)]],
      -
      const constant GEMMSpiltKParams* params [[buffer(3)]],
      -
      uint simd_lane_id [[thread_index_in_simdgroup]],
      -
      uint simd_group_id [[simdgroup_index_in_threadgroup]],
      -
      uint3 tid [[threadgroup_position_in_grid]],
      -
      uint3 lid [[thread_position_in_threadgroup]]);
      -
      )"
      -
      -
      -
      -
      -
      - - - - diff --git a/docs/build/html/steel__gemm_8h.js b/docs/build/html/steel__gemm_8h.js deleted file mode 100644 index b5784bf3c..000000000 --- a/docs/build/html/steel__gemm_8h.js +++ /dev/null @@ -1,8 +0,0 @@ -var steel__gemm_8h = -[ - [ "steel_gemm_fused_kernels", "steel__gemm_8h.html#a4c6009fd5357b730805f2fd4ba6e093e", null ], - [ "steel_gemm_masked_kernels", "steel__gemm_8h.html#a62a358fd3ec5365081920d07aceb581c", null ], - [ "steel_gemm_splitk_accum_axbpy_kernels", "steel__gemm_8h.html#a40a86e1381c241aba8511e51a981a4bf", null ], - [ "steel_gemm_splitk_accum_kernels", "steel__gemm_8h.html#a144a64b8d94f0371fb144e2cc308fcf9", null ], - [ "steel_gemm_splitk_kernels", "steel__gemm_8h.html#a92108ab01d826e38bca83d8569b947d9", null ] -]; \ No newline at end of file diff --git a/docs/build/html/steel__gemm_8h_source.html b/docs/build/html/steel__gemm_8h_source.html deleted file mode 100644 index 98d50d836..000000000 --- a/docs/build/html/steel__gemm_8h_source.html +++ /dev/null @@ -1,229 +0,0 @@ - - - - - - - -MLX: mlx/backend/metal/jit/steel_gemm.h Source File - - - - - - - - - - - - - - - - -
      -
      - - - - - - - -
      -
      MLX -
      -
      - -   - - - - -
      -
      -
      - - - - -
      -
      - -
      -
      -
      - -
      - -
      -
      - - -
      -
      -
      -
      -
      -
      Loading...
      -
      Searching...
      -
      No Matches
      -
      -
      -
      -
      - -
      -
      steel_gemm.h
      -
      -
      -Go to the documentation of this file.
      1// Copyright © 2024 Apple Inc.
      -
      2
      -
      3constexpr std::string_view steel_gemm_fused_kernels = R"(
      -
      4template [[host_name("{name}")]]
      -
      5[[kernel]] void gemm<{itype}, {bm}, {bn}, {bk}, {wm}, {wn}, {trans_a}, {trans_b}, float>(
      -
      6 const device {itype} *A [[buffer(0)]],
      -
      7 const device {itype} *B [[buffer(1)]],
      -
      8 const device {itype} *C [[buffer(2), function_constant(use_out_source)]],
      -
      9 device {itype} *D [[buffer(3)]],
      -
      10 const constant GEMMParams* params [[buffer(4)]],
      -
      11 const constant GEMMAddMMParams* addmm_params [[buffer(5), function_constant(use_out_source)]],
      -
      12 const constant int* batch_shape [[buffer(6)]],
      -
      13 const constant int64_t* batch_strides [[buffer(7)]],
      -
      14 const constant uint32_t* lhs_indices [[buffer(10), function_constant(do_gather)]],
      -
      15 const constant uint32_t* rhs_indices [[buffer(11), function_constant(do_gather)]],
      -
      16 const constant uint32_t* C_indices [[buffer(12), function_constant(gather_bias)]],
      -
      17 const constant int* operand_shape [[buffer(13), function_constant(do_gather)]],
      -
      18 const constant int64_t* operand_strides [[buffer(14), function_constant(do_gather)]],
      -
      19 const constant packed_int3& operand_batch_ndim [[buffer(15), function_constant(do_gather)]],
      -
      20 uint simd_lane_id [[thread_index_in_simdgroup]],
      -
      21 uint simd_group_id [[simdgroup_index_in_threadgroup]],
      -
      22 uint3 tid [[threadgroup_position_in_grid]],
      -
      23 uint3 lid [[thread_position_in_threadgroup]]);
      -
      24)";
      -
      25
      -
      26constexpr std::string_view steel_gemm_masked_kernels = R"(
      -
      27template [[host_name("{name}")]] [[kernel]] void
      -
      28block_masked_gemm<
      -
      29 {itype},
      -
      30 {outmasktype},
      -
      31 {opmasktype},
      -
      32 {bm},
      -
      33 {bn},
      -
      34 {bk},
      -
      35 {wm},
      -
      36 {wn},
      -
      37 {trans_a},
      -
      38 {trans_b},
      -
      39 {mn_aligned},
      -
      40 {k_aligned}>(
      -
      41 const device {itype}* A [[buffer(0)]],
      -
      42 const device {itype}* B [[buffer(1)]],
      -
      43 device {itype}* D [[buffer(3)]],
      -
      44 const constant GEMMParams* params [[buffer(4)]],
      -
      45 const constant int* batch_shape [[buffer(6)]],
      -
      46 const constant int64_t* batch_strides [[buffer(7)]],
      -
      47 const device {outmasktype}* out_mask [[buffer(10)]],
      -
      48 const device {opmasktype}* lhs_mask [[buffer(11)]],
      -
      49 const device {opmasktype}* rhs_mask [[buffer(12)]],
      -
      50 const constant int* mask_strides [[buffer(13)]],
      -
      51 uint simd_lane_id [[thread_index_in_simdgroup]],
      -
      52 uint simd_group_id [[simdgroup_index_in_threadgroup]],
      -
      53 uint3 tid [[threadgroup_position_in_grid]],
      -
      54 uint3 lid [[thread_position_in_threadgroup]]);
      -
      55)";
      -
      56
      -
      57constexpr std::string_view steel_gemm_splitk_kernels = R"(
      -
      58template [[host_name("{name}")]] [[kernel]] void
      -
      59gemm_splitk<
      -
      60 {itype},
      -
      61 {otype},
      -
      62 {bm},
      -
      63 {bn},
      -
      64 {bk},
      -
      65 {wm},
      -
      66 {wn},
      -
      67 {trans_a},
      -
      68 {trans_b},
      -
      69 {mn_aligned},
      -
      70 {k_aligned}>(
      -
      71 const device {itype}* A [[buffer(0)]],
      -
      72 const device {itype}* B [[buffer(1)]],
      -
      73 device {otype}* C [[buffer(2)]],
      -
      74 const constant GEMMSpiltKParams* params [[buffer(3)]],
      -
      75 uint simd_lane_id [[thread_index_in_simdgroup]],
      -
      76 uint simd_group_id [[simdgroup_index_in_threadgroup]],
      -
      77 uint3 tid [[threadgroup_position_in_grid]],
      -
      78 uint3 lid [[thread_position_in_threadgroup]]);
      -
      79)";
      -
      80
      -
      81constexpr std::string_view steel_gemm_splitk_accum_kernels = R"(
      -
      82template [[host_name("{name}")]] [[kernel]] void
      -
      83gemm_splitk_accum<{atype}, {otype}>(
      -
      84 const device {atype}* C_split [[buffer(0)]],
      -
      85 device {otype}* D [[buffer(1)]],
      -
      86 const constant int& k_partitions [[buffer(2)]],
      -
      87 const constant int& partition_stride [[buffer(3)]],
      -
      88 const constant int& ldd [[buffer(4)]],
      -
      89 uint2 gid [[thread_position_in_grid]]);
      -
      90)";
      -
      91
      -
      92constexpr std::string_view steel_gemm_splitk_accum_axbpy_kernels = R"(
      -
      93template [[host_name("{name}")]] [[kernel]] void
      -
      94gemm_splitk_accum_axpby<{atype}, {otype}>(
      -
      95 const device {atype}* C_split [[buffer(0)]],
      -
      96 device {otype}* D [[buffer(1)]],
      -
      97 const constant int& k_partitions [[buffer(2)]],
      -
      98 const constant int& partition_stride [[buffer(3)]],
      -
      99 const constant int& ldd [[buffer(4)]],
      -
      100 const device {otype}* C [[buffer(5)]],
      -
      101 const constant int& ldc [[buffer(6)]],
      -
      102 const constant int& fdc [[buffer(7)]],
      -
      103 const constant float& alpha [[buffer(8)]],
      -
      104 const constant float& beta [[buffer(9)]],
      -
      105 uint2 gid [[thread_position_in_grid]]);
      -
      106)";
      -
      steel_gemm_splitk_accum_kernels
      constexpr std::string_view steel_gemm_splitk_accum_kernels
      Definition steel_gemm.h:81
      -
      steel_gemm_splitk_accum_axbpy_kernels
      constexpr std::string_view steel_gemm_splitk_accum_axbpy_kernels
      Definition steel_gemm.h:92
      -
      steel_gemm_fused_kernels
      constexpr std::string_view steel_gemm_fused_kernels
      Definition steel_gemm.h:3
      -
      steel_gemm_masked_kernels
      constexpr std::string_view steel_gemm_masked_kernels
      Definition steel_gemm.h:26
      -
      steel_gemm_splitk_kernels
      constexpr std::string_view steel_gemm_splitk_kernels
      Definition steel_gemm.h:57
      -
      -
      - - - - diff --git a/docs/build/html/steel__gemm__fused_8h.html b/docs/build/html/steel__gemm__fused_8h.html index 7a762952e..28b39ce8e 100644 --- a/docs/build/html/steel__gemm__fused_8h.html +++ b/docs/build/html/steel__gemm__fused_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused.h File Reference @@ -52,7 +52,7 @@
      - + @@ -52,7 +52,7 @@
      - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      ConditionalType< true, T, U > Member List
      +
      +
      + +

      This is the complete list of members for ConditionalType< true, T, U >, including all inherited members.

      + + + +
      type typedefConditionalType< true, T, U >
      type typedefConditionalType< true, T, U >
      +
      + + + + diff --git a/docs/build/html/struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.html b/docs/build/html/struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.html new file mode 100644 index 000000000..69a5b6ca1 --- /dev/null +++ b/docs/build/html/struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.html @@ -0,0 +1,165 @@ + + + + + + + +MLX: ConditionalType< true, T, U > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      ConditionalType< true, T, U > Struct Template Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + +

      +Public Types

      using type = T
       
      using type
       
      +

      Member Typedef Documentation

      + +

      ◆ type [1/2]

      + +
      +
      + + + + +
      using ConditionalType< condition, T, U >::type
      +
      + +
      +
      + +

      ◆ type [2/2]

      + +
      +
      +
      +template<typename T, typename U>
      + + + + +
      using ConditionalType< true, T, U >::type = T
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.js b/docs/build/html/struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.js new file mode 100644 index 000000000..86e6c19f5 --- /dev/null +++ b/docs/build/html/struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.js @@ -0,0 +1,5 @@ +var struct_conditional_type_3_01true_00_01_t_00_01_u_01_4 = +[ + [ "type", "struct_conditional_type.html#a00bac71c43763817c4422bf0363dc92b", null ], + [ "type", "struct_conditional_type_3_01true_00_01_t_00_01_u_01_4.html#abf6b75b3bfc5e3baf8c37afdb8a7e4fc", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_conjugate-members.html b/docs/build/html/struct_conjugate-members.html index 9fbfa5c80..afa9349bc 100644 --- a/docs/build/html/struct_conjugate-members.html +++ b/docs/build/html/struct_conjugate-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@
      - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      CumProd< bool > Member List
      +
      +
      + +

      This is the complete list of members for CumProd< bool >, including all inherited members.

      + + + + + +
      initCumProd< bool >static
      operator()(bool a, T b)CumProd< bool >inline
      simd_exclusive_scan(bool x)CumProd< bool >inline
      simd_scan(bool x)CumProd< bool >inline
      +
      + + + + diff --git a/docs/build/html/struct_cum_prod_3_01bool_01_4.html b/docs/build/html/struct_cum_prod_3_01bool_01_4.html new file mode 100644 index 000000000..337247fbb --- /dev/null +++ b/docs/build/html/struct_cum_prod_3_01bool_01_4.html @@ -0,0 +1,248 @@ + + + + + + + +MLX: CumProd< bool > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Member Functions | +Static Public Attributes | +List of all members
      +
      CumProd< bool > Struct Reference
      +
      +
      + +

      #include <scan.h>

      + + + + + + + + + +

      +Public Member Functions

      template<typename T>
      bool operator() (bool a, T b)
       
      bool simd_scan (bool x)
       
      bool simd_exclusive_scan (bool x)
       
      + + + +

      +Static Public Attributes

      static constexpr constant bool init = true
       
      +

      Member Function Documentation

      + +

      ◆ operator()()

      + +
      +
      +
      +template<typename T>
      + + + + + +
      + + + + + + + + + + + +
      bool CumProd< bool >::operator() (bool a,
      T b )
      +       		+inline
      +
      + +
      +
      + +

      ◆ simd_exclusive_scan()

      + +
      +
      + + + + + +
      + + + + + + + +
      bool CumProd< bool >::simd_exclusive_scan (bool x)
      +       		+inline
      +
      + +
      +
      + +

      ◆ simd_scan()

      + +
      +
      + + + + + +
      + + + + + + + +
      bool CumProd< bool >::simd_scan (bool x)
      +       		+inline
      +
      + +
      +
      +

      Member Data Documentation

      + +

      ◆ init

      + +
      +
      + + + + + +
      + + + + +
      constant bool CumProd< bool >::init = true
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/scan.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_cum_prod_3_01bool_01_4.js b/docs/build/html/struct_cum_prod_3_01bool_01_4.js new file mode 100644 index 000000000..8d18f83df --- /dev/null +++ b/docs/build/html/struct_cum_prod_3_01bool_01_4.js @@ -0,0 +1,7 @@ +var struct_cum_prod_3_01bool_01_4 = +[ + [ "operator()", "struct_cum_prod_3_01bool_01_4.html#ad634be0b139d10ce6d21332eef0d936b", null ], + [ "simd_exclusive_scan", "struct_cum_prod_3_01bool_01_4.html#a1a86e9398bae24182b7be0a6577bf223", null ], + [ "simd_scan", "struct_cum_prod_3_01bool_01_4.html#abeb5ec4237b330e7219f4e881cf10d7a", null ], + [ "init", "struct_cum_prod_3_01bool_01_4.html#ae7a8b0ba9e6898356b87b18766e76d2c", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_cum_sum.html b/docs/build/html/struct_cum_sum.html index 48e03daea..1600a5165 100644 --- a/docs/build/html/struct_cum_sum.html +++ b/docs/build/html/struct_cum_sum.html @@ -3,7 +3,7 @@ - + MLX: CumSum< U > Struct Template Reference @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< bfloat16_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< bfloat16_t >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< bfloat16_t >static
      finite_maxLimits< bfloat16_t >static
      finite_minLimits< bfloat16_t >static
      finite_minLimits< bfloat16_t >static
      maxLimits< bfloat16_t >static
      maxLimits< bfloat16_t >static
      minLimits< bfloat16_t >static
      minLimits< bfloat16_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01bfloat16__t_01_4.html b/docs/build/html/struct_limits_3_01bfloat16__t_01_4.html new file mode 100644 index 000000000..464e46d98 --- /dev/null +++ b/docs/build/html/struct_limits_3_01bfloat16__t_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< bfloat16_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< bfloat16_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant bfloat16_t max = metal::numeric_limits<bfloat16_t>::infinity()
       
      static constexpr constant bfloat16_t min = -metal::numeric_limits<bfloat16_t>::infinity()
       
      static constexpr constant bfloat16_t finite_max = metal::numeric_limits<bfloat16_t>::max()
       
      static constexpr constant bfloat16_t finite_min = -metal::numeric_limits<bfloat16_t>::max()
       
      static const constant bfloat16_t max
       
      static const constant bfloat16_t min
       
      static const constant bfloat16_t finite_max
       
      static const constant bfloat16_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant bfloat16_t Limits< bfloat16_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant bfloat16_t Limits< bfloat16_t >::finite_max = metal::numeric_limits<bfloat16_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant bfloat16_t Limits< bfloat16_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant bfloat16_t Limits< bfloat16_t >::finite_min = -metal::numeric_limits<bfloat16_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant bfloat16_t Limits< bfloat16_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant bfloat16_t Limits< bfloat16_t >::max = metal::numeric_limits<bfloat16_t>::infinity()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant bfloat16_t Limits< bfloat16_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant bfloat16_t Limits< bfloat16_t >::min = -metal::numeric_limits<bfloat16_t>::infinity()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01bfloat16__t_01_4.js b/docs/build/html/struct_limits_3_01bfloat16__t_01_4.js new file mode 100644 index 000000000..1d15ba214 --- /dev/null +++ b/docs/build/html/struct_limits_3_01bfloat16__t_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01bfloat16__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01bfloat16__t_01_4.html#a6337dc35207b3f6f7185cd73eabac211", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01bfloat16__t_01_4.html#ae4132a37154707cc31bbc1734636cf36", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01bfloat16__t_01_4.html#a0ead3618da6718629ea9fa4670b5005f", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01bfloat16__t_01_4.html#a2fd1811b9f615b2b897904bc27d1cb49", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01bool_01_4-members.html b/docs/build/html/struct_limits_3_01bool_01_4-members.html new file mode 100644 index 000000000..ea50d1a03 --- /dev/null +++ b/docs/build/html/struct_limits_3_01bool_01_4-members.html @@ -0,0 +1,126 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< bool > Member List
      +
      +
      + +

      This is the complete list of members for Limits< bool >, including all inherited members.

      + + + + + + + +
      finite_maxLimits< bool >static
      finite_minLimits< bool >static
      maxLimits< bool >static
      maxLimits< bool >static
      minLimits< bool >static
      minLimits< bool >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01bool_01_4.html b/docs/build/html/struct_limits_3_01bool_01_4.html new file mode 100644 index 000000000..9ccbf74b3 --- /dev/null +++ b/docs/build/html/struct_limits_3_01bool_01_4.html @@ -0,0 +1,275 @@ + + + + + + + +MLX: Limits< bool > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< bool > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant bool max = true
       
      static constexpr constant bool min = false
       
      static const constant bool max
       
      static const constant bool min
       
      static const constant bool finite_max
       
      static const constant bool finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max

      + +
      +
      + + + + + +
      + + + + +
      const constant bool Limits< bool >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min

      + +
      +
      + + + + + +
      + + + + +
      const constant bool Limits< bool >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant bool Limits< bool >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant bool Limits< bool >::max = true
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant bool Limits< bool >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant bool Limits< bool >::min = false
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01bool_01_4.js b/docs/build/html/struct_limits_3_01bool_01_4.js new file mode 100644 index 000000000..02dbd1fcf --- /dev/null +++ b/docs/build/html/struct_limits_3_01bool_01_4.js @@ -0,0 +1,9 @@ +var struct_limits_3_01bool_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01bool_01_4.html#acbd2132145888d51220558a101ffcff4", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01bool_01_4.html#a139f787b57536d455490b8ef801d37cc", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01complex64__t_01_4-members.html b/docs/build/html/struct_limits_3_01complex64__t_01_4-members.html new file mode 100644 index 000000000..56bd4fee2 --- /dev/null +++ b/docs/build/html/struct_limits_3_01complex64__t_01_4-members.html @@ -0,0 +1,126 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< complex64_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< complex64_t >, including all inherited members.

      + + + + + + + +
      finite_maxLimits< complex64_t >static
      finite_minLimits< complex64_t >static
      maxLimits< complex64_t >static
      maxLimits< complex64_t >static
      minLimits< complex64_t >static
      minLimits< complex64_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01complex64__t_01_4.html b/docs/build/html/struct_limits_3_01complex64__t_01_4.html new file mode 100644 index 000000000..a16c9b20c --- /dev/null +++ b/docs/build/html/struct_limits_3_01complex64__t_01_4.html @@ -0,0 +1,282 @@ + + + + + + + +MLX: Limits< complex64_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< complex64_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant complex64_t max
       
      static constexpr constant complex64_t min
       
      static const constant complex64_t max
       
      static const constant complex64_t min
       
      static const constant complex64_t finite_max
       
      static const constant complex64_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max

      + +
      +
      + + + + + +
      + + + + +
      const constant complex64_t Limits< complex64_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min

      + +
      +
      + + + + + +
      + + + + +
      const constant complex64_t Limits< complex64_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant complex64_t Limits< complex64_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant complex64_t Limits< complex64_t >::max
      +       		+staticconstexpr
      +
      +Initial value:
      = complex64_t(
      +
      metal::numeric_limits<float>::infinity(),
      +
      metal::numeric_limits<float>::infinity())
      +
      complex64_t
      Definition complex.h:20
      +
      +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant complex64_t Limits< complex64_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant complex64_t Limits< complex64_t >::min
      +       		+staticconstexpr
      +
      +Initial value:
      = complex64_t(
      +
      -metal::numeric_limits<float>::infinity(),
      +
      -metal::numeric_limits<float>::infinity())
      +
      +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01complex64__t_01_4.js b/docs/build/html/struct_limits_3_01complex64__t_01_4.js new file mode 100644 index 000000000..298c0b964 --- /dev/null +++ b/docs/build/html/struct_limits_3_01complex64__t_01_4.js @@ -0,0 +1,9 @@ +var struct_limits_3_01complex64__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01complex64__t_01_4.html#ac01c274b224b90f5210b675a484f4607", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01complex64__t_01_4.html#aa67b04aa7abcd67f7af0808737ab8e14", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01float_01_4-members.html b/docs/build/html/struct_limits_3_01float_01_4-members.html new file mode 100644 index 000000000..65613a8ca --- /dev/null +++ b/docs/build/html/struct_limits_3_01float_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< float > Member List
      +
      +
      + +

      This is the complete list of members for Limits< float >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< float >static
      finite_maxLimits< float >static
      finite_minLimits< float >static
      finite_minLimits< float >static
      maxLimits< float >static
      maxLimits< float >static
      minLimits< float >static
      minLimits< float >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01float_01_4.html b/docs/build/html/struct_limits_3_01float_01_4.html new file mode 100644 index 000000000..b52473bee --- /dev/null +++ b/docs/build/html/struct_limits_3_01float_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< float > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< float > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant float max = metal::numeric_limits<float>::infinity()
       
      static constexpr constant float min = -metal::numeric_limits<float>::infinity()
       
      static constexpr constant float finite_max = metal::numeric_limits<float>::max()
       
      static constexpr constant float finite_min = -metal::numeric_limits<float>::max()
       
      static const constant float max
       
      static const constant float min
       
      static const constant float finite_max
       
      static const constant float finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant float Limits< float >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant float Limits< float >::finite_max = metal::numeric_limits<float>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant float Limits< float >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant float Limits< float >::finite_min = -metal::numeric_limits<float>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant float Limits< float >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant float Limits< float >::max = metal::numeric_limits<float>::infinity()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant float Limits< float >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant float Limits< float >::min = -metal::numeric_limits<float>::infinity()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01float_01_4.js b/docs/build/html/struct_limits_3_01float_01_4.js new file mode 100644 index 000000000..230f57f2c --- /dev/null +++ b/docs/build/html/struct_limits_3_01float_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01float_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01float_01_4.html#a291eea590113fc1858b7f83f2e0c977d", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01float_01_4.html#afaa5162a47083447c5ac758d6dc02a8b", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01float_01_4.html#aba172b22b388190aa3969ef16885d8a6", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01float_01_4.html#a3225e334d372ee86128c89a440d8648f", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01half_01_4-members.html b/docs/build/html/struct_limits_3_01half_01_4-members.html new file mode 100644 index 000000000..a3f40f8d8 --- /dev/null +++ b/docs/build/html/struct_limits_3_01half_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< half > Member List
      +
      +
      + +

      This is the complete list of members for Limits< half >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< half >static
      finite_maxLimits< half >static
      finite_minLimits< half >static
      finite_minLimits< half >static
      maxLimits< half >static
      maxLimits< half >static
      minLimits< half >static
      minLimits< half >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01half_01_4.html b/docs/build/html/struct_limits_3_01half_01_4.html new file mode 100644 index 000000000..b47788d78 --- /dev/null +++ b/docs/build/html/struct_limits_3_01half_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< half > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< half > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant half max = metal::numeric_limits<half>::infinity()
       
      static constexpr constant half min = -metal::numeric_limits<half>::infinity()
       
      static constexpr constant half finite_max = metal::numeric_limits<half>::max()
       
      static constexpr constant half finite_min = -metal::numeric_limits<half>::max()
       
      static const constant half max
       
      static const constant half min
       
      static const constant half finite_max
       
      static const constant half finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant half Limits< half >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant half Limits< half >::finite_max = metal::numeric_limits<half>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant half Limits< half >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant half Limits< half >::finite_min = -metal::numeric_limits<half>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant half Limits< half >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant half Limits< half >::max = metal::numeric_limits<half>::infinity()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant half Limits< half >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant half Limits< half >::min = -metal::numeric_limits<half>::infinity()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01half_01_4.js b/docs/build/html/struct_limits_3_01half_01_4.js new file mode 100644 index 000000000..fa77ff3c0 --- /dev/null +++ b/docs/build/html/struct_limits_3_01half_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01half_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01half_01_4.html#aedaf0190aabf23da20510e558e2690b4", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01half_01_4.html#a98d153748be68dbb428c50df3c0285ab", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01half_01_4.html#a4f9515dbf2a622074f121bea39a7b175", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01half_01_4.html#aca7b036c257878bf1b80912fb5d4516d", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01int16__t_01_4-members.html b/docs/build/html/struct_limits_3_01int16__t_01_4-members.html new file mode 100644 index 000000000..27943ccd0 --- /dev/null +++ b/docs/build/html/struct_limits_3_01int16__t_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< int16_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< int16_t >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< int16_t >static
      finite_maxLimits< int16_t >static
      finite_minLimits< int16_t >static
      finite_minLimits< int16_t >static
      maxLimits< int16_t >static
      maxLimits< int16_t >static
      minLimits< int16_t >static
      minLimits< int16_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01int16__t_01_4.html b/docs/build/html/struct_limits_3_01int16__t_01_4.html new file mode 100644 index 000000000..777731297 --- /dev/null +++ b/docs/build/html/struct_limits_3_01int16__t_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< int16_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< int16_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant int16_t max = metal::numeric_limits<int16_t>::max()
       
      static constexpr constant int16_t min = metal::numeric_limits<int16_t>::min()
       
      static constexpr constant int16_t finite_max = metal::numeric_limits<int16_t>::max()
       
      static constexpr constant int16_t finite_min = metal::numeric_limits<int16_t>::min()
       
      static const constant int16_t max
       
      static const constant int16_t min
       
      static const constant int16_t finite_max
       
      static const constant int16_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int16_t Limits< int16_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int16_t Limits< int16_t >::finite_max = metal::numeric_limits<int16_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int16_t Limits< int16_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int16_t Limits< int16_t >::finite_min = metal::numeric_limits<int16_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int16_t Limits< int16_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int16_t Limits< int16_t >::max = metal::numeric_limits<int16_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int16_t Limits< int16_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int16_t Limits< int16_t >::min = metal::numeric_limits<int16_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01int16__t_01_4.js b/docs/build/html/struct_limits_3_01int16__t_01_4.js new file mode 100644 index 000000000..60693e492 --- /dev/null +++ b/docs/build/html/struct_limits_3_01int16__t_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01int16__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01int16__t_01_4.html#acb2936d1cdbf347a9a014c8e036a5782", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01int16__t_01_4.html#a158c4dbc9333939691b1637478e28e39", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01int16__t_01_4.html#a12d64c398ca7609b7c906f3cf1a6f678", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01int16__t_01_4.html#adca7139647801e223c35b0abc7da5240", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01int32__t_01_4-members.html b/docs/build/html/struct_limits_3_01int32__t_01_4-members.html new file mode 100644 index 000000000..b121599ea --- /dev/null +++ b/docs/build/html/struct_limits_3_01int32__t_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< int32_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< int32_t >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< int32_t >static
      finite_maxLimits< int32_t >static
      finite_minLimits< int32_t >static
      finite_minLimits< int32_t >static
      maxLimits< int32_t >static
      maxLimits< int32_t >static
      minLimits< int32_t >static
      minLimits< int32_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01int32__t_01_4.html b/docs/build/html/struct_limits_3_01int32__t_01_4.html new file mode 100644 index 000000000..031b3ba6b --- /dev/null +++ b/docs/build/html/struct_limits_3_01int32__t_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< int32_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< int32_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant int32_t max = metal::numeric_limits<int32_t>::max()
       
      static constexpr constant int32_t min = metal::numeric_limits<int32_t>::min()
       
      static constexpr constant int32_t finite_max = metal::numeric_limits<int32_t>::max()
       
      static constexpr constant int32_t finite_min = metal::numeric_limits<int32_t>::min()
       
      static const constant int32_t max
       
      static const constant int32_t min
       
      static const constant int32_t finite_max
       
      static const constant int32_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int32_t Limits< int32_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int32_t Limits< int32_t >::finite_max = metal::numeric_limits<int32_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int32_t Limits< int32_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int32_t Limits< int32_t >::finite_min = metal::numeric_limits<int32_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int32_t Limits< int32_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int32_t Limits< int32_t >::max = metal::numeric_limits<int32_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int32_t Limits< int32_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int32_t Limits< int32_t >::min = metal::numeric_limits<int32_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01int32__t_01_4.js b/docs/build/html/struct_limits_3_01int32__t_01_4.js new file mode 100644 index 000000000..fa9bd6bcc --- /dev/null +++ b/docs/build/html/struct_limits_3_01int32__t_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01int32__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01int32__t_01_4.html#aa9ed9f0e8c7400d8fc92e1cba9588794", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01int32__t_01_4.html#ad9777dc6a84dcb9c63b598189ff0a4ff", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01int32__t_01_4.html#af756344b31e84222dd73d3445dcd5640", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01int32__t_01_4.html#af336a1b22a8ed6a83a4cfb5bf8869771", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01int64__t_01_4-members.html b/docs/build/html/struct_limits_3_01int64__t_01_4-members.html new file mode 100644 index 000000000..8e3abd0bf --- /dev/null +++ b/docs/build/html/struct_limits_3_01int64__t_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< int64_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< int64_t >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< int64_t >static
      finite_maxLimits< int64_t >static
      finite_minLimits< int64_t >static
      finite_minLimits< int64_t >static
      maxLimits< int64_t >static
      maxLimits< int64_t >static
      minLimits< int64_t >static
      minLimits< int64_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01int64__t_01_4.html b/docs/build/html/struct_limits_3_01int64__t_01_4.html new file mode 100644 index 000000000..ae4bb2015 --- /dev/null +++ b/docs/build/html/struct_limits_3_01int64__t_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< int64_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< int64_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant int64_t max = metal::numeric_limits<int64_t>::max()
       
      static constexpr constant int64_t min = metal::numeric_limits<int64_t>::min()
       
      static constexpr constant int64_t finite_max = metal::numeric_limits<int64_t>::max()
       
      static constexpr constant int64_t finite_min = metal::numeric_limits<int64_t>::min()
       
      static const constant int64_t max
       
      static const constant int64_t min
       
      static const constant int64_t finite_max
       
      static const constant int64_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int64_t Limits< int64_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int64_t Limits< int64_t >::finite_max = metal::numeric_limits<int64_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int64_t Limits< int64_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int64_t Limits< int64_t >::finite_min = metal::numeric_limits<int64_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int64_t Limits< int64_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int64_t Limits< int64_t >::max = metal::numeric_limits<int64_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int64_t Limits< int64_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int64_t Limits< int64_t >::min = metal::numeric_limits<int64_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01int64__t_01_4.js b/docs/build/html/struct_limits_3_01int64__t_01_4.js new file mode 100644 index 000000000..5ac715d88 --- /dev/null +++ b/docs/build/html/struct_limits_3_01int64__t_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01int64__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01int64__t_01_4.html#a6c7254b641878fa0fb9538814c45457a", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01int64__t_01_4.html#af80726162b44a741aae679f1fe85142a", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01int64__t_01_4.html#ac9c420604c0f3d237ddfb2b8a2439224", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01int64__t_01_4.html#a1c90fb96af515badaccaa835b08f7428", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01int8__t_01_4-members.html b/docs/build/html/struct_limits_3_01int8__t_01_4-members.html new file mode 100644 index 000000000..950631bf4 --- /dev/null +++ b/docs/build/html/struct_limits_3_01int8__t_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< int8_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< int8_t >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< int8_t >static
      finite_maxLimits< int8_t >static
      finite_minLimits< int8_t >static
      finite_minLimits< int8_t >static
      maxLimits< int8_t >static
      maxLimits< int8_t >static
      minLimits< int8_t >static
      minLimits< int8_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01int8__t_01_4.html b/docs/build/html/struct_limits_3_01int8__t_01_4.html new file mode 100644 index 000000000..baf64ce8a --- /dev/null +++ b/docs/build/html/struct_limits_3_01int8__t_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< int8_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< int8_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant int8_t max = metal::numeric_limits<int8_t>::max()
       
      static constexpr constant int8_t min = metal::numeric_limits<int8_t>::min()
       
      static constexpr constant int8_t finite_max = metal::numeric_limits<int8_t>::max()
       
      static constexpr constant int8_t finite_min = metal::numeric_limits<int8_t>::min()
       
      static const constant int8_t max
       
      static const constant int8_t min
       
      static const constant int8_t finite_max
       
      static const constant int8_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int8_t Limits< int8_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int8_t Limits< int8_t >::finite_max = metal::numeric_limits<int8_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int8_t Limits< int8_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int8_t Limits< int8_t >::finite_min = metal::numeric_limits<int8_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int8_t Limits< int8_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int8_t Limits< int8_t >::max = metal::numeric_limits<int8_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant int8_t Limits< int8_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant int8_t Limits< int8_t >::min = metal::numeric_limits<int8_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01int8__t_01_4.js b/docs/build/html/struct_limits_3_01int8__t_01_4.js new file mode 100644 index 000000000..cbfd3ef08 --- /dev/null +++ b/docs/build/html/struct_limits_3_01int8__t_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01int8__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01int8__t_01_4.html#a24cdab873e0fb778393c69f1dc9ecf73", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01int8__t_01_4.html#a592797ce82cc2f7e27b0c477165b3452", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01int8__t_01_4.html#a96fed01fa9249226be69760652643289", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01int8__t_01_4.html#a7a809307d2bba80382f0645d277eaa4b", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01uint16__t_01_4-members.html b/docs/build/html/struct_limits_3_01uint16__t_01_4-members.html new file mode 100644 index 000000000..6d8d2e89a --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint16__t_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< uint16_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< uint16_t >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< uint16_t >static
      finite_maxLimits< uint16_t >static
      finite_minLimits< uint16_t >static
      finite_minLimits< uint16_t >static
      maxLimits< uint16_t >static
      maxLimits< uint16_t >static
      minLimits< uint16_t >static
      minLimits< uint16_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01uint16__t_01_4.html b/docs/build/html/struct_limits_3_01uint16__t_01_4.html new file mode 100644 index 000000000..c9397e61c --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint16__t_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< uint16_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< uint16_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant uint16_t max = metal::numeric_limits<uint16_t>::max()
       
      static constexpr constant uint16_t min = metal::numeric_limits<uint16_t>::min()
       
      static constexpr constant uint16_t finite_max = metal::numeric_limits<uint16_t>::max()
       
      static constexpr constant uint16_t finite_min = metal::numeric_limits<uint16_t>::min()
       
      static const constant uint16_t max
       
      static const constant uint16_t min
       
      static const constant uint16_t finite_max
       
      static const constant uint16_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint16_t Limits< uint16_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint16_t Limits< uint16_t >::finite_max = metal::numeric_limits<uint16_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint16_t Limits< uint16_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint16_t Limits< uint16_t >::finite_min = metal::numeric_limits<uint16_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint16_t Limits< uint16_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint16_t Limits< uint16_t >::max = metal::numeric_limits<uint16_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint16_t Limits< uint16_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint16_t Limits< uint16_t >::min = metal::numeric_limits<uint16_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01uint16__t_01_4.js b/docs/build/html/struct_limits_3_01uint16__t_01_4.js new file mode 100644 index 000000000..587b5f49a --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint16__t_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01uint16__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01uint16__t_01_4.html#a9d517d8265ea1898b6b16e91b8595146", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01uint16__t_01_4.html#a1a7c029eccba4ab89743abdfaabfa7b4", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01uint16__t_01_4.html#a228b33556ba4cb7e6137ab6258628488", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01uint16__t_01_4.html#ae173984c3be8b6750f27daed581805fe", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01uint32__t_01_4-members.html b/docs/build/html/struct_limits_3_01uint32__t_01_4-members.html new file mode 100644 index 000000000..4480166aa --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint32__t_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< uint32_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< uint32_t >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< uint32_t >static
      finite_maxLimits< uint32_t >static
      finite_minLimits< uint32_t >static
      finite_minLimits< uint32_t >static
      maxLimits< uint32_t >static
      maxLimits< uint32_t >static
      minLimits< uint32_t >static
      minLimits< uint32_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01uint32__t_01_4.html b/docs/build/html/struct_limits_3_01uint32__t_01_4.html new file mode 100644 index 000000000..b2d2fc798 --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint32__t_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< uint32_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< uint32_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant uint32_t max = metal::numeric_limits<uint32_t>::max()
       
      static constexpr constant uint32_t min = metal::numeric_limits<uint32_t>::min()
       
      static constexpr constant uint32_t finite_max = metal::numeric_limits<uint32_t>::max()
       
      static constexpr constant uint32_t finite_min = metal::numeric_limits<uint32_t>::min()
       
      static const constant uint32_t max
       
      static const constant uint32_t min
       
      static const constant uint32_t finite_max
       
      static const constant uint32_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint32_t Limits< uint32_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint32_t Limits< uint32_t >::finite_max = metal::numeric_limits<uint32_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint32_t Limits< uint32_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint32_t Limits< uint32_t >::finite_min = metal::numeric_limits<uint32_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint32_t Limits< uint32_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint32_t Limits< uint32_t >::max = metal::numeric_limits<uint32_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint32_t Limits< uint32_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint32_t Limits< uint32_t >::min = metal::numeric_limits<uint32_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01uint32__t_01_4.js b/docs/build/html/struct_limits_3_01uint32__t_01_4.js new file mode 100644 index 000000000..dbf6bead0 --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint32__t_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01uint32__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01uint32__t_01_4.html#a0698139f3fe440d7aa08ac5029d72235", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01uint32__t_01_4.html#ad5d811fce62f44488190ff01d9e7608b", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01uint32__t_01_4.html#a91fa8f7214ec936976a8324c7431c651", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01uint32__t_01_4.html#ab0c3975e02053b234c7b606ababa66e1", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01uint64__t_01_4-members.html b/docs/build/html/struct_limits_3_01uint64__t_01_4-members.html new file mode 100644 index 000000000..7ad3e5854 --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint64__t_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< uint64_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< uint64_t >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< uint64_t >static
      finite_maxLimits< uint64_t >static
      finite_minLimits< uint64_t >static
      finite_minLimits< uint64_t >static
      maxLimits< uint64_t >static
      maxLimits< uint64_t >static
      minLimits< uint64_t >static
      minLimits< uint64_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01uint64__t_01_4.html b/docs/build/html/struct_limits_3_01uint64__t_01_4.html new file mode 100644 index 000000000..6194687d7 --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint64__t_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< uint64_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< uint64_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant uint64_t max = metal::numeric_limits<uint64_t>::max()
       
      static constexpr constant uint64_t min = metal::numeric_limits<uint64_t>::min()
       
      static constexpr constant uint64_t finite_max = metal::numeric_limits<uint64_t>::max()
       
      static constexpr constant uint64_t finite_min = metal::numeric_limits<uint64_t>::min()
       
      static const constant uint64_t max
       
      static const constant uint64_t min
       
      static const constant uint64_t finite_max
       
      static const constant uint64_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint64_t Limits< uint64_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint64_t Limits< uint64_t >::finite_max = metal::numeric_limits<uint64_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint64_t Limits< uint64_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint64_t Limits< uint64_t >::finite_min = metal::numeric_limits<uint64_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint64_t Limits< uint64_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint64_t Limits< uint64_t >::max = metal::numeric_limits<uint64_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint64_t Limits< uint64_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint64_t Limits< uint64_t >::min = metal::numeric_limits<uint64_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01uint64__t_01_4.js b/docs/build/html/struct_limits_3_01uint64__t_01_4.js new file mode 100644 index 000000000..ba4ac3878 --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint64__t_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01uint64__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01uint64__t_01_4.html#aff101ff38be5ccdbb9790aecb3069071", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01uint64__t_01_4.html#a6556e7de6e0670da8f768bbc4479deae", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01uint64__t_01_4.html#aa8c2257881a4e1fa8596fa07dba5e107", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01uint64__t_01_4.html#a80627f39e951398283942cefa48f4dd0", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_limits_3_01uint8__t_01_4-members.html b/docs/build/html/struct_limits_3_01uint8__t_01_4-members.html new file mode 100644 index 000000000..e71aa2eef --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint8__t_01_4-members.html @@ -0,0 +1,128 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      Limits< uint8_t > Member List
      +
      +
      + +

      This is the complete list of members for Limits< uint8_t >, including all inherited members.

      + + + + + + + + + +
      finite_maxLimits< uint8_t >static
      finite_maxLimits< uint8_t >static
      finite_minLimits< uint8_t >static
      finite_minLimits< uint8_t >static
      maxLimits< uint8_t >static
      maxLimits< uint8_t >static
      minLimits< uint8_t >static
      minLimits< uint8_t >static
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01uint8__t_01_4.html b/docs/build/html/struct_limits_3_01uint8__t_01_4.html new file mode 100644 index 000000000..e266dd686 --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint8__t_01_4.html @@ -0,0 +1,323 @@ + + + + + + + +MLX: Limits< uint8_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Attributes | +List of all members
      +
      Limits< uint8_t > Struct Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Static Public Attributes

      static constexpr constant uint8_t max = metal::numeric_limits<uint8_t>::max()
       
      static constexpr constant uint8_t min = metal::numeric_limits<uint8_t>::min()
       
      static constexpr constant uint8_t finite_max = metal::numeric_limits<uint8_t>::max()
       
      static constexpr constant uint8_t finite_min = metal::numeric_limits<uint8_t>::min()
       
      static const constant uint8_t max
       
      static const constant uint8_t min
       
      static const constant uint8_t finite_max
       
      static const constant uint8_t finite_min
       
      +

      Member Data Documentation

      + +

      ◆ finite_max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint8_t Limits< uint8_t >::finite_max
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint8_t Limits< uint8_t >::finite_max = metal::numeric_limits<uint8_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ finite_min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint8_t Limits< uint8_t >::finite_min
      +       		+static
      +
      + +
      +
      + +

      ◆ finite_min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint8_t Limits< uint8_t >::finite_min = metal::numeric_limits<uint8_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ max [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint8_t Limits< uint8_t >::max
      +       		+static
      +
      + +
      +
      + +

      ◆ max [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint8_t Limits< uint8_t >::max = metal::numeric_limits<uint8_t>::max()
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ min [1/2]

      + +
      +
      + + + + + +
      + + + + +
      const constant uint8_t Limits< uint8_t >::min
      +       		+static
      +
      + +
      +
      + +

      ◆ min [2/2]

      + +
      +
      + + + + + +
      + + + + +
      constant uint8_t Limits< uint8_t >::min = metal::numeric_limits<uint8_t>::min()
      +       		+staticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_limits_3_01uint8__t_01_4.js b/docs/build/html/struct_limits_3_01uint8__t_01_4.js new file mode 100644 index 000000000..bc7128b01 --- /dev/null +++ b/docs/build/html/struct_limits_3_01uint8__t_01_4.js @@ -0,0 +1,11 @@ +var struct_limits_3_01uint8__t_01_4 = +[ + [ "finite_max", "struct_limits.html#a5a3eae6d244fbea2aa7b9200001463e5", null ], + [ "finite_max", "struct_limits_3_01uint8__t_01_4.html#a55f48b89033e8c8683f8540ec6b23f02", null ], + [ "finite_min", "struct_limits.html#ae7469d21f2688797ca3e388d919ef05e", null ], + [ "finite_min", "struct_limits_3_01uint8__t_01_4.html#a60cea662971b09f78ef19f1da4760b73", null ], + [ "max", "struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8", null ], + [ "max", "struct_limits_3_01uint8__t_01_4.html#a1570fb640e2e41f96776db5ca08d500c", null ], + [ "min", "struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e", null ], + [ "min", "struct_limits_3_01uint8__t_01_4.html#a408bd5a337e7292f06e63da81193629a", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_log-members.html b/docs/build/html/struct_log-members.html index 295e348d1..bbc7b0629 100644 --- a/docs/build/html/struct_log-members.html +++ b/docs/build/html/struct_log-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      LoopedElemToLoc< 1, OffsetT, false > Member List
      +
      +
      + +

      This is the complete list of members for LoopedElemToLoc< 1, OffsetT, false >, including all inherited members.

      + + + + + + + + + + + + + + +
      dimLoopedElemToLoc< 1, OffsetT, false >
      indexLoopedElemToLoc< 1, OffsetT, false >
      inner_looperLoopedElemToLoc< 1, OffsetT, false >
      location()LoopedElemToLoc< 1, OffsetT, false >inline
      location()LoopedElemToLoc< 1, OffsetT, false >inline
      LoopedElemToLoc(int)LoopedElemToLoc< 1, OffsetT, false >inline
      LoopedElemToLoc(int dim)LoopedElemToLoc< 1, OffsetT, false >inline
      next(const constant int *, const constant int64_t *strides)LoopedElemToLoc< 1, OffsetT, false >inline
      next(int n, const constant int *, const constant int64_t *strides)LoopedElemToLoc< 1, OffsetT, false >inline
      next(const constant int *shape, const constant int64_t *strides)LoopedElemToLoc< 1, OffsetT, false >inline
      next(int n, const constant int *shape, const constant int64_t *strides)LoopedElemToLoc< 1, OffsetT, false >inline
      offsetLoopedElemToLoc< 1, OffsetT, false >
      offsetLoopedElemToLoc< 1, OffsetT, false >
      +
      + + + + diff --git a/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html new file mode 100644 index 000000000..b067edcaf --- /dev/null +++ b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html @@ -0,0 +1,469 @@ + + + + + + + +MLX: LoopedElemToLoc< 1, OffsetT, false > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Member Functions | +Public Attributes | +List of all members
      +
      LoopedElemToLoc< 1, OffsetT, false > Struct Template Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Public Member Functions

       LoopedElemToLoc (int)
       
      void next (const constant int *, const constant int64_t *strides)
       
      void next (int n, const constant int *, const constant int64_t *strides)
       
      OffsetT location ()
       
       LoopedElemToLoc (int dim)
       
      void next (const constant int *shape, const constant int64_t *strides)
       
      void next (int n, const constant int *shape, const constant int64_t *strides)
       
      OffsetT location ()
       
      + + + + + + + + + + + +

      +Public Attributes

      OffsetT offset {0}
       
      int dim
       
      LoopedElemToLoc< DIM - 1, OffsetT, General > inner_looper
       
      OffsetT offset
       
      int index
       
      +

      Constructor & Destructor Documentation

      + +

      ◆ LoopedElemToLoc() [1/2]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + + +
      + + + + + + + +
      LoopedElemToLoc< 1, OffsetT, false >::LoopedElemToLoc (int )
      +       		+inline
      +
      + +
      +
      + +

      ◆ LoopedElemToLoc() [2/2]

      + +
      +
      + + + + + +
      + + + + + + + +
      LoopedElemToLoc< DIM, OffsetT, General >::LoopedElemToLoc (int dim)
      +       		+inline
      +
      + +
      +
      +

      Member Function Documentation

      + +

      ◆ location() [1/2]

      + +
      +
      + + + + + +
      + + + + + + + +
      OffsetT LoopedElemToLoc< DIM, OffsetT, General >::location ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ location() [2/2]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + + +
      + + + + + + + +
      OffsetT LoopedElemToLoc< 1, OffsetT, false >::location ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ next() [1/4]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + + +
      + + + + + + + + + + + +
      void LoopedElemToLoc< 1, OffsetT, false >::next (const constant int * ,
      const constant int64_t * strides )
      +       		+inline
      +
      + +
      +
      + +

      ◆ next() [2/4]

      + +
      +
      + + + + + +
      + + + + + + + + + + + +
      void LoopedElemToLoc< DIM, OffsetT, General >::next (const constant int * shape,
      const constant int64_t * strides )
      +       		+inline
      +
      + +
      +
      + +

      ◆ next() [3/4]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + + +
      + + + + + + + + + + + + + + + + +
      void LoopedElemToLoc< 1, OffsetT, false >::next (int n,
      const constant int * ,
      const constant int64_t * strides )
      +       		+inline
      +
      + +
      +
      + +

      ◆ next() [4/4]

      + +
      +
      + + + + + +
      + + + + + + + + + + + + + + + + +
      void LoopedElemToLoc< DIM, OffsetT, General >::next (int n,
      const constant int * shape,
      const constant int64_t * strides )
      +       		+inline
      +
      + +
      +
      +

      Member Data Documentation

      + +

      ◆ dim

      + +
      +
      + + + + +
      int LoopedElemToLoc< DIM, OffsetT, General >::dim
      +
      + +
      +
      + +

      ◆ index

      + +
      +
      + + + + +
      int LoopedElemToLoc< DIM, OffsetT, General >::index
      +
      + +
      +
      + +

      ◆ inner_looper

      + +
      +
      + + + + +
      LoopedElemToLoc<DIM - 1, OffsetT, General> LoopedElemToLoc< DIM, OffsetT, General >::inner_looper
      +
      + +
      +
      + +

      ◆ offset [1/2]

      + +
      +
      + + + + +
      OffsetT LoopedElemToLoc< DIM, OffsetT, General >::offset
      +
      + +
      +
      + +

      ◆ offset [2/2]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + +
      OffsetT LoopedElemToLoc< 1, OffsetT, false >::offset {0}
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.js b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.js new file mode 100644 index 000000000..cb9d181a1 --- /dev/null +++ b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.js @@ -0,0 +1,16 @@ +var struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4 = +[ + [ "LoopedElemToLoc", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a0e21977d9f23b6994773e8e4f3ee70de", null ], + [ "LoopedElemToLoc", "struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b", null ], + [ "location", "struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e", null ], + [ "location", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a89d9ec4dc2f2f0d77e27aa0c05f261ef", null ], + [ "next", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a9d85f32e5460707827df3ba74b905c23", null ], + [ "next", "struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531", null ], + [ "next", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#a634f4fa8f4b5f1fdd47a26754ef2eba7", null ], + [ "next", "struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394", null ], + [ "dim", "struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364", null ], + [ "index", "struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333", null ], + [ "inner_looper", "struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40", null ], + [ "offset", "struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791", null ], + [ "offset", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01false_01_4.html#af792b1fd4e8286f97b9b863c127a2d9a", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4-members.html b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4-members.html new file mode 100644 index 000000000..27e9aabc8 --- /dev/null +++ b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4-members.html @@ -0,0 +1,135 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      LoopedElemToLoc< 1, OffsetT, true > Member List
      +
      +
      + +

      This is the complete list of members for LoopedElemToLoc< 1, OffsetT, true >, including all inherited members.

      + + + + + + + + + + + + + + + + +
      dimLoopedElemToLoc< 1, OffsetT, true >
      dimLoopedElemToLoc< 1, OffsetT, true >
      indexLoopedElemToLoc< 1, OffsetT, true >
      indexLoopedElemToLoc< 1, OffsetT, true >
      inner_looperLoopedElemToLoc< 1, OffsetT, true >
      location()LoopedElemToLoc< 1, OffsetT, true >inline
      location()LoopedElemToLoc< 1, OffsetT, true >inline
      LoopedElemToLoc(int dim)LoopedElemToLoc< 1, OffsetT, true >inline
      LoopedElemToLoc(int dim)LoopedElemToLoc< 1, OffsetT, true >inline
      next(const constant int *shape, const constant int64_t *strides)LoopedElemToLoc< 1, OffsetT, true >inline
      next(int n, const constant int *shape, const constant int64_t *strides)LoopedElemToLoc< 1, OffsetT, true >inline
      next(const constant int *shape, const constant int64_t *strides)LoopedElemToLoc< 1, OffsetT, true >inline
      next(int n, const constant int *shape, const constant int64_t *strides)LoopedElemToLoc< 1, OffsetT, true >inline
      offsetLoopedElemToLoc< 1, OffsetT, true >
      offsetLoopedElemToLoc< 1, OffsetT, true >
      +
      + + + + diff --git a/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html new file mode 100644 index 000000000..b91735a96 --- /dev/null +++ b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html @@ -0,0 +1,505 @@ + + + + + + + +MLX: LoopedElemToLoc< 1, OffsetT, true > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Member Functions | +Public Attributes | +List of all members
      +
      LoopedElemToLoc< 1, OffsetT, true > Struct Template Reference
      +
      +
      + +

      #include <utils.h>

      + + + + + + + + + + + + + + + + + + +

      +Public Member Functions

       LoopedElemToLoc (int dim)
       
      void next (const constant int *shape, const constant int64_t *strides)
       
      void next (int n, const constant int *shape, const constant int64_t *strides)
       
      OffsetT location ()
       
       LoopedElemToLoc (int dim)
       
      void next (const constant int *shape, const constant int64_t *strides)
       
      void next (int n, const constant int *shape, const constant int64_t *strides)
       
      OffsetT location ()
       
      + + + + + + + + + + + + + + + +

      +Public Attributes

      int dim
       
      OffsetT offset {0}
       
      uint index {0}
       
      int dim
       
      LoopedElemToLoc< DIM - 1, OffsetT, General > inner_looper
       
      OffsetT offset
       
      int index
       
      +

      Constructor & Destructor Documentation

      + +

      ◆ LoopedElemToLoc() [1/2]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + + +
      + + + + + + + +
      LoopedElemToLoc< 1, OffsetT, true >::LoopedElemToLoc (int dim)
      +       		+inline
      +
      + +
      +
      + +

      ◆ LoopedElemToLoc() [2/2]

      + +
      +
      + + + + + +
      + + + + + + + +
      LoopedElemToLoc< DIM, OffsetT, General >::LoopedElemToLoc (int dim)
      +       		+inline
      +
      + +
      +
      +

      Member Function Documentation

      + +

      ◆ location() [1/2]

      + +
      +
      + + + + + +
      + + + + + + + +
      OffsetT LoopedElemToLoc< DIM, OffsetT, General >::location ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ location() [2/2]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + + +
      + + + + + + + +
      OffsetT LoopedElemToLoc< 1, OffsetT, true >::location ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ next() [1/4]

      + +
      +
      + + + + + +
      + + + + + + + + + + + +
      void LoopedElemToLoc< DIM, OffsetT, General >::next (const constant int * shape,
      const constant int64_t * strides )
      +       		+inline
      +
      + +
      +
      + +

      ◆ next() [2/4]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + + +
      + + + + + + + + + + + +
      void LoopedElemToLoc< 1, OffsetT, true >::next (const constant int * shape,
      const constant int64_t * strides )
      +       		+inline
      +
      + +
      +
      + +

      ◆ next() [3/4]

      + +
      +
      + + + + + +
      + + + + + + + + + + + + + + + + +
      void LoopedElemToLoc< DIM, OffsetT, General >::next (int n,
      const constant int * shape,
      const constant int64_t * strides )
      +       		+inline
      +
      + +
      +
      + +

      ◆ next() [4/4]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + + +
      + + + + + + + + + + + + + + + + +
      void LoopedElemToLoc< 1, OffsetT, true >::next (int n,
      const constant int * shape,
      const constant int64_t * strides )
      +       		+inline
      +
      + +
      +
      +

      Member Data Documentation

      + +

      ◆ dim [1/2]

      + +
      +
      + + + + +
      int LoopedElemToLoc< DIM, OffsetT, General >::dim
      +
      + +
      +
      + +

      ◆ dim [2/2]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + +
      int LoopedElemToLoc< 1, OffsetT, true >::dim
      +
      + +
      +
      + +

      ◆ index [1/2]

      + +
      +
      + + + + +
      int LoopedElemToLoc< DIM, OffsetT, General >::index
      +
      + +
      +
      + +

      ◆ index [2/2]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + +
      uint LoopedElemToLoc< 1, OffsetT, true >::index {0}
      +
      + +
      +
      + +

      ◆ inner_looper

      + +
      +
      + + + + +
      LoopedElemToLoc<DIM - 1, OffsetT, General> LoopedElemToLoc< DIM, OffsetT, General >::inner_looper
      +
      + +
      +
      + +

      ◆ offset [1/2]

      + +
      +
      + + + + +
      OffsetT LoopedElemToLoc< DIM, OffsetT, General >::offset
      +
      + +
      +
      + +

      ◆ offset [2/2]

      + +
      +
      +
      +template<typename OffsetT>
      + + + + +
      OffsetT LoopedElemToLoc< 1, OffsetT, true >::offset {0}
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file:
        +
      • mlx/backend/metal/kernels/utils.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.js b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.js new file mode 100644 index 000000000..171b909c0 --- /dev/null +++ b/docs/build/html/struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.js @@ -0,0 +1,18 @@ +var struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4 = +[ + [ "LoopedElemToLoc", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#abf536c7162d36af7367e390789944c86", null ], + [ "LoopedElemToLoc", "struct_looped_elem_to_loc.html#a5653be1c990722a4a215be27efe5648b", null ], + [ "location", "struct_looped_elem_to_loc.html#aba051a428ad0934a9c6d04d4d3ee6e0e", null ], + [ "location", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a66b84b12f6c1494e5908989ed2849a9f", null ], + [ "next", "struct_looped_elem_to_loc.html#a9b41f17e9e1439a278af75a84a937531", null ], + [ "next", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#adc8113de3152d291331f219bfe5970c2", null ], + [ "next", "struct_looped_elem_to_loc.html#af1eb45e030e46c076c9876ef12c0d394", null ], + [ "next", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a70d92070097fe0622b836afc778bf8d2", null ], + [ "dim", "struct_looped_elem_to_loc.html#af8285112846769aba2c0d8615f6f1364", null ], + [ "dim", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a7be6bf560080472d61e74b522979ef1e", null ], + [ "index", "struct_looped_elem_to_loc.html#acbd070b3193d9e87fb2c2db8db571333", null ], + [ "index", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a2cd3b616739b3d5b41e5b46ae335957d", null ], + [ "inner_looper", "struct_looped_elem_to_loc.html#a8fbe77b4a774a30af5734dd9c5bd1f40", null ], + [ "offset", "struct_looped_elem_to_loc.html#acdffe540c383a67417604b6080704791", null ], + [ "offset", "struct_looped_elem_to_loc_3_011_00_01_offset_t_00_01true_01_4.html#a3a18944c158e2747a6ddebb420299a3b", null ] +]; \ No newline at end of file diff --git a/docs/build/html/struct_m_l_x_conv_params-members.html b/docs/build/html/struct_m_l_x_conv_params-members.html index ecf771584..6bb234349 100644 --- a/docs/build/html/struct_m_l_x_conv_params-members.html +++ b/docs/build/html/struct_m_l_x_conv_params-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - +
      @@ -102,20 +102,19 @@ $(function(){initNavTree('unionmlx_1_1core_1_1detail_1_1_int_or_float.html','');
      -
      mlx::core::detail::IntOrFloat Member List
      +
      metal::pointer_element< constant T * > Member List
      -

      This is the complete list of members for mlx::core::detail::IntOrFloat, including all inherited members.

      +

      This is the complete list of members for metal::pointer_element< constant T * >, including all inherited members.

      - - +
      fmlx::core::detail::IntOrFloat
      imlx::core::detail::IntOrFloat
      type typedefmetal::pointer_element< constant T * >
      diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.html b/docs/build/html/structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.html new file mode 100644 index 000000000..2bcf44548 --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.html @@ -0,0 +1,149 @@ + + + + + + + +MLX: metal::pointer_element< constant T * > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      metal::pointer_element< constant T * > Struct Template Reference
      +
      +
      + +

      #include <type_traits.h>

      + + + + +

      +Public Types

      using type = remove_cv_t<T>
       
      +

      Member Typedef Documentation

      + +

      ◆ type

      + +
      +
      +
      +template<typename T>
      + + + + +
      using metal::pointer_element< constant T * >::type = remove_cv_t<T>
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.js b/docs/build/html/structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.js new file mode 100644 index 000000000..391f06832 --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.js @@ -0,0 +1,4 @@ +var structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4 = +[ + [ "type", "structmetal_1_1pointer__element_3_01constant_01_t_01_5_01_4.html#ad154b55b9e450a6376016488c8e68c53", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4-members.html b/docs/build/html/structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4-members.html new file mode 100644 index 000000000..8a24d6055 --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4-members.html @@ -0,0 +1,121 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      metal::pointer_element< device T * > Member List
      +
      +
      + +

      This is the complete list of members for metal::pointer_element< device T * >, including all inherited members.

      + + +
      type typedefmetal::pointer_element< device T * >
      +
      + + + + diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.html b/docs/build/html/structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.html new file mode 100644 index 000000000..e8a5648de --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.html @@ -0,0 +1,149 @@ + + + + + + + +MLX: metal::pointer_element< device T * > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      metal::pointer_element< device T * > Struct Template Reference
      +
      +
      + +

      #include <type_traits.h>

      + + + + +

      +Public Types

      using type = remove_cv_t<T>
       
      +

      Member Typedef Documentation

      + +

      ◆ type

      + +
      +
      +
      +template<typename T>
      + + + + +
      using metal::pointer_element< device T * >::type = remove_cv_t<T>
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.js b/docs/build/html/structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.js new file mode 100644 index 000000000..79f4caa81 --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.js @@ -0,0 +1,4 @@ +var structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4 = +[ + [ "type", "structmetal_1_1pointer__element_3_01device_01_t_01_5_01_4.html#ab36a7c5a64c0693dd3d8ccb322c163d4", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4-members.html b/docs/build/html/structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4-members.html new file mode 100644 index 000000000..aa4f1818e --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4-members.html @@ -0,0 +1,121 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      metal::pointer_element< thread T * > Member List
      +
      +
      + +

      This is the complete list of members for metal::pointer_element< thread T * >, including all inherited members.

      + + +
      type typedefmetal::pointer_element< thread T * >
      +
      + + + + diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.html b/docs/build/html/structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.html new file mode 100644 index 000000000..75e2e3ae6 --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.html @@ -0,0 +1,149 @@ + + + + + + + +MLX: metal::pointer_element< thread T * > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      metal::pointer_element< thread T * > Struct Template Reference
      +
      +
      + +

      #include <type_traits.h>

      + + + + +

      +Public Types

      using type = remove_cv_t<T>
       
      +

      Member Typedef Documentation

      + +

      ◆ type

      + +
      +
      +
      +template<typename T>
      + + + + +
      using metal::pointer_element< thread T * >::type = remove_cv_t<T>
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.js b/docs/build/html/structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.js new file mode 100644 index 000000000..da09c6b88 --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.js @@ -0,0 +1,4 @@ +var structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4 = +[ + [ "type", "structmetal_1_1pointer__element_3_01thread_01_t_01_5_01_4.html#a98fbc2aa99dd26bb35aa9cd1826318d8", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4-members.html b/docs/build/html/structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4-members.html new file mode 100644 index 000000000..971aefdd0 --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4-members.html @@ -0,0 +1,121 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      metal::pointer_element< threadgroup T * > Member List
      +
      +
      + +

      This is the complete list of members for metal::pointer_element< threadgroup T * >, including all inherited members.

      + + +
      type typedefmetal::pointer_element< threadgroup T * >
      +
      + + + + diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.html b/docs/build/html/structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.html new file mode 100644 index 000000000..a44632e7b --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.html @@ -0,0 +1,149 @@ + + + + + + + +MLX: metal::pointer_element< threadgroup T * > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      metal::pointer_element< threadgroup T * > Struct Template Reference
      +
      +
      + +

      #include <type_traits.h>

      + + + + +

      +Public Types

      using type = remove_cv_t<T>
       
      +

      Member Typedef Documentation

      + +

      ◆ type

      + +
      +
      +
      +template<typename T>
      + + + + +
      using metal::pointer_element< threadgroup T * >::type = remove_cv_t<T>
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.js b/docs/build/html/structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.js new file mode 100644 index 000000000..54d92a961 --- /dev/null +++ b/docs/build/html/structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.js @@ -0,0 +1,4 @@ +var structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4 = +[ + [ "type", "structmetal_1_1pointer__element_3_01threadgroup_01_t_01_5_01_4.html#a78c718d6da9d393c139a385f42472362", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1___m_l_x___b_float16-members.html b/docs/build/html/structmlx_1_1core_1_1___m_l_x___b_float16-members.html index 27e4d482f..c67f254b6 100644 --- a/docs/build/html/structmlx_1_1core_1_1___m_l_x___b_float16-members.html +++ b/docs/build/html/structmlx_1_1core_1_1___m_l_x___b_float16-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - +
      @@ -102,21 +102,21 @@ $(function(){initNavTree('structmlx_1_1core_1_1_default_strided_reduce.html','')
      -
      mlx::core::DefaultStridedReduce< T, U, Op > Member List
      +
      mlx::core::ScalarVector< Op > Member List
      -

      This is the complete list of members for mlx::core::DefaultStridedReduce< T, U, Op >, including all inherited members.

      +

      This is the complete list of members for mlx::core::ScalarVector< Op >, including all inherited members.

      - - - + + +
      DefaultStridedReduce(Op op_)mlx::core::DefaultStridedReduce< T, U, Op >inline
      opmlx::core::DefaultStridedReduce< T, U, Op >
      operator()(const T *x, U *accumulator, int size, size_t stride)mlx::core::DefaultStridedReduce< T, U, Op >inline
      opmlx::core::ScalarVector< Op >
      operator()(const T *a, const T *b, U *dst, int size)mlx::core::ScalarVector< Op >inline
      ScalarVector(Op op_)mlx::core::ScalarVector< Op >inline
      diff --git a/docs/build/html/structmlx_1_1core_1_1_default_contiguous_reduce.html b/docs/build/html/structmlx_1_1core_1_1_scalar_vector.html similarity index 69% rename from docs/build/html/structmlx_1_1core_1_1_default_contiguous_reduce.html rename to docs/build/html/structmlx_1_1core_1_1_scalar_vector.html index e4143773d..e1cefb73d 100644 --- a/docs/build/html/structmlx_1_1core_1_1_default_contiguous_reduce.html +++ b/docs/build/html/structmlx_1_1core_1_1_scalar_vector.html @@ -3,9 +3,9 @@ - + -MLX: mlx::core::DefaultContiguousReduce< T, U, Op > Struct Template Reference +MLX: mlx::core::ScalarVector< Op > Struct Template Reference @@ -52,7 +52,7 @@ - +
      @@ -105,39 +105,40 @@ $(function(){initNavTree('structmlx_1_1core_1_1_default_contiguous_reduce.html',
      Public Member Functions | Public Attributes | -List of all members
      -
      mlx::core::DefaultContiguousReduce< T, U, Op > Struct Template Reference
      +List of all members
      +
      mlx::core::ScalarVector< Op > Struct Template Reference
      -

      #include <reduce.h>

      +

      #include <binary.h>

      - - - - + + + + +

      Public Member Functions

       DefaultContiguousReduce (Op op_)
       
      void operator() (const T *x, U *accumulator, int size)
       
       ScalarVector (Op op_)
       
      template<typename T, typename U>
      void operator() (const T *a, const T *b, U *dst, int size)
       
      - - + +

      Public Attributes

      Op op
       
      Op op
       

      Constructor & Destructor Documentation

      - -

      ◆ DefaultContiguousReduce()

      + +

      ◆ ScalarVector()

      -template<typename T, typename U, typename Op>
      +template<typename Op>
      - + @@ -153,26 +154,33 @@ template<typename T, typename U, typename Op>

      Member Function Documentation

      - -

      ◆ operator()()

      + +

      ◆ operator()()

      -template<typename T, typename U, typename Op>
      +template<typename Op>
      +
      +template<typename T, typename U>
      mlx::core::DefaultContiguousReduce< T, U, Op >::DefaultContiguousReduce mlx::core::ScalarVector< Op >::ScalarVector ( Op op_)
      - + - + - + + + + + + @@ -190,16 +198,16 @@ template<typename T, typename U, typename Op>

      Member Data Documentation

      - -

      ◆ op

      + +

      ◆ op

      -template<typename T, typename U, typename Op>
      +template<typename Op>
      void mlx::core::DefaultContiguousReduce< T, U, Op >::operator() void mlx::core::ScalarVector< Op >::operator() (const T * x, const T * a,
      U * accumulator, const T * b,
      U * dst,
      - +
      Op mlx::core::DefaultContiguousReduce< T, U, Op >::opOp mlx::core::ScalarVector< Op >::op
      @@ -207,15 +215,15 @@ template<typename T, typename U, typename Op>
      The documentation for this struct was generated from the following file: diff --git a/docs/build/html/structmlx_1_1core_1_1_scalar_vector.js b/docs/build/html/structmlx_1_1core_1_1_scalar_vector.js new file mode 100644 index 000000000..bea7d3e61 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1_scalar_vector.js @@ -0,0 +1,6 @@ +var structmlx_1_1core_1_1_scalar_vector = +[ + [ "ScalarVector", "structmlx_1_1core_1_1_scalar_vector.html#a69d6a3ddd7586e8e19a42c5e6f5a287b", null ], + [ "operator()", "structmlx_1_1core_1_1_scalar_vector.html#ab174fe55970fb4ee1c6a2b7628a24df1", null ], + [ "op", "structmlx_1_1core_1_1_scalar_vector.html#ac9c2214744bc972150740e169b603b9b", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1_stream-members.html b/docs/build/html/structmlx_1_1core_1_1_stream-members.html index 267a93b99..a12e9fe4d 100644 --- a/docs/build/html/structmlx_1_1core_1_1_stream-members.html +++ b/docs/build/html/structmlx_1_1core_1_1_stream-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@
      - +
      @@ -102,21 +102,21 @@ $(function(){initNavTree('structmlx_1_1core_1_1_default_contiguous_reduce.html',
      -
      mlx::core::DefaultContiguousReduce< T, U, Op > Member List
      +
      mlx::core::VectorScalar< Op > Member List
      -

      This is the complete list of members for mlx::core::DefaultContiguousReduce< T, U, Op >, including all inherited members.

      +

      This is the complete list of members for mlx::core::VectorScalar< Op >, including all inherited members.

      - - - + + +
      DefaultContiguousReduce(Op op_)mlx::core::DefaultContiguousReduce< T, U, Op >inline
      opmlx::core::DefaultContiguousReduce< T, U, Op >
      operator()(const T *x, U *accumulator, int size)mlx::core::DefaultContiguousReduce< T, U, Op >inline
      opmlx::core::VectorScalar< Op >
      operator()(const T *a, const T *b, U *dst, int size)mlx::core::VectorScalar< Op >inline
      VectorScalar(Op op_)mlx::core::VectorScalar< Op >inline
      diff --git a/docs/build/html/structmlx_1_1core_1_1_default_strided_reduce.html b/docs/build/html/structmlx_1_1core_1_1_vector_scalar.html similarity index 69% rename from docs/build/html/structmlx_1_1core_1_1_default_strided_reduce.html rename to docs/build/html/structmlx_1_1core_1_1_vector_scalar.html index 397af890c..080b339ea 100644 --- a/docs/build/html/structmlx_1_1core_1_1_default_strided_reduce.html +++ b/docs/build/html/structmlx_1_1core_1_1_vector_scalar.html @@ -3,9 +3,9 @@ - + -MLX: mlx::core::DefaultStridedReduce< T, U, Op > Struct Template Reference +MLX: mlx::core::VectorScalar< Op > Struct Template Reference @@ -52,7 +52,7 @@
      - +
      @@ -105,39 +105,40 @@ $(function(){initNavTree('structmlx_1_1core_1_1_default_strided_reduce.html','')
      Public Member Functions | Public Attributes | -List of all members
      -
      mlx::core::DefaultStridedReduce< T, U, Op > Struct Template Reference
      +List of all members
      +
      mlx::core::VectorScalar< Op > Struct Template Reference
      -

      #include <reduce.h>

      +

      #include <binary.h>

      - - - - + + + + +

      Public Member Functions

       DefaultStridedReduce (Op op_)
       
      void operator() (const T *x, U *accumulator, int size, size_t stride)
       
       VectorScalar (Op op_)
       
      template<typename T, typename U>
      void operator() (const T *a, const T *b, U *dst, int size)
       
      - - + +

      Public Attributes

      Op op
       
      Op op
       

      Constructor & Destructor Documentation

      - -

      ◆ DefaultStridedReduce()

      + +

      ◆ VectorScalar()

      -template<typename T, typename U, typename Op>
      +template<typename Op>
      - + @@ -153,36 +154,38 @@ template<typename T, typename U, typename Op>

      Member Function Documentation

      - -

      ◆ operator()()

      + +

      ◆ operator()()

      -template<typename T, typename U, typename Op>
      +template<typename Op>
      +
      +template<typename T, typename U>
      mlx::core::DefaultStridedReduce< T, U, Op >::DefaultStridedReduce mlx::core::VectorScalar< Op >::VectorScalar ( Op op_)
      @@ -195,16 +198,16 @@ template<typename T, typename U, typename Op>

      Member Data Documentation

      - -

      ◆ op

      + +

      ◆ op

      -template<typename T, typename U, typename Op>
      +template<typename Op>
      - + - + - + - + - +
      void mlx::core::DefaultStridedReduce< T, U, Op >::operator() void mlx::core::VectorScalar< Op >::operator() (const T * x, const T * a,
      U * accumulator, const T * b,
      int size, U * dst,
      size_t stride )int size )
      - +
      Op mlx::core::DefaultStridedReduce< T, U, Op >::opOp mlx::core::VectorScalar< Op >::op
      @@ -212,15 +215,15 @@ template<typename T, typename U, typename Op>
      The documentation for this struct was generated from the following file: diff --git a/docs/build/html/structmlx_1_1core_1_1_vector_scalar.js b/docs/build/html/structmlx_1_1core_1_1_vector_scalar.js new file mode 100644 index 000000000..b9712673b --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1_vector_scalar.js @@ -0,0 +1,6 @@ +var structmlx_1_1core_1_1_vector_scalar = +[ + [ "VectorScalar", "structmlx_1_1core_1_1_vector_scalar.html#a97088143e6d301d753dcdd1ccdd82287", null ], + [ "operator()", "structmlx_1_1core_1_1_vector_scalar.html#a1af3ff644ce023a7e4f92a7c3634c44f", null ], + [ "op", "structmlx_1_1core_1_1_vector_scalar.html#a5fe1744adb58aaa845acca1e46725537", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1_vector_vector-members.html b/docs/build/html/structmlx_1_1core_1_1_vector_vector-members.html new file mode 100644 index 000000000..6ea14dd26 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1_vector_vector-members.html @@ -0,0 +1,123 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::VectorVector< Op > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::VectorVector< Op >, including all inherited members.

      + + + + +
      opmlx::core::VectorVector< Op >
      operator()(const T *a, const T *b, U *dst, int size)mlx::core::VectorVector< Op >inline
      VectorVector(Op op_)mlx::core::VectorVector< Op >inline
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1_vector_vector.html b/docs/build/html/structmlx_1_1core_1_1_vector_vector.html new file mode 100644 index 000000000..983ff5b72 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1_vector_vector.html @@ -0,0 +1,230 @@ + + + + + + + +MLX: mlx::core::VectorVector< Op > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Member Functions | +Public Attributes | +List of all members
      +
      mlx::core::VectorVector< Op > Struct Template Reference
      +
      +
      + +

      #include <binary.h>

      + + + + + + + +

      +Public Member Functions

       VectorVector (Op op_)
       
      template<typename T, typename U>
      void operator() (const T *a, const T *b, U *dst, int size)
       
      + + + +

      +Public Attributes

      Op op
       
      +

      Constructor & Destructor Documentation

      + +

      ◆ VectorVector()

      + +
      +
      +
      +template<typename Op>
      + + + + + +
      + + + + + + + +
      mlx::core::VectorVector< Op >::VectorVector (Op op_)
      +       		+inline
      +
      + +
      +
      +

      Member Function Documentation

      + +

      ◆ operator()()

      + +
      +
      +
      +template<typename Op>
      +
      +template<typename T, typename U>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + +
      void mlx::core::VectorVector< Op >::operator() (const T * a,
      const T * b,
      U * dst,
      int size )
      +       		+inline
      +
      + +
      +
      +

      Member Data Documentation

      + +

      ◆ op

      + +
      +
      +
      +template<typename Op>
      + + + + +
      Op mlx::core::VectorVector< Op >::op
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1_vector_vector.js b/docs/build/html/structmlx_1_1core_1_1_vector_vector.js new file mode 100644 index 000000000..f17c57aa2 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1_vector_vector.js @@ -0,0 +1,6 @@ +var structmlx_1_1core_1_1_vector_vector = +[ + [ "VectorVector", "structmlx_1_1core_1_1_vector_vector.html#a4867666c95c597a113afb64f173cc022", null ], + [ "operator()", "structmlx_1_1core_1_1_vector_vector.html#a97a0bed419933d7685238a962f2e4215", null ], + [ "op", "structmlx_1_1core_1_1_vector_vector.html#a6d69d070c75cf0281e11e36e0717ab50", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1array_1_1_array_iterator-members.html b/docs/build/html/structmlx_1_1core_1_1array_1_1_array_iterator-members.html index e8b241c8f..81e516ac1 100644 --- a/docs/build/html/structmlx_1_1core_1_1array_1_1_array_iterator-members.html +++ b/docs/build/html/structmlx_1_1core_1_1array_1_1_array_iterator-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@
      - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::numeric_limits< bfloat16_t > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::numeric_limits< bfloat16_t >, including all inherited members.

      + + + + +
      infinity()mlx::core::numeric_limits< bfloat16_t >inlinestatic
      lowest()mlx::core::numeric_limits< bfloat16_t >inlinestatic
      max()mlx::core::numeric_limits< bfloat16_t >inlinestatic
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html new file mode 100644 index 000000000..240b48879 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html @@ -0,0 +1,212 @@ + + + + + + + +MLX: mlx::core::numeric_limits< bfloat16_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Member Functions | +List of all members
      +
      mlx::core::numeric_limits< bfloat16_t > Struct Reference
      +
      +
      + +

      #include <limits.h>

      + + + + + + + + +

      +Static Public Member Functions

      static constexpr bfloat16_t lowest ()
       
      static constexpr bfloat16_t max ()
       
      static constexpr bfloat16_t infinity ()
       
      +

      Member Function Documentation

      + +

      ◆ infinity()

      + +
      +
      + + + + + +
      + + + + + + + +
      static constexpr bfloat16_t mlx::core::numeric_limits< bfloat16_t >::infinity ()
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ lowest()

      + +
      +
      + + + + + +
      + + + + + + + +
      static constexpr bfloat16_t mlx::core::numeric_limits< bfloat16_t >::lowest ()
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ max()

      + +
      +
      + + + + + +
      + + + + + + + +
      static constexpr bfloat16_t mlx::core::numeric_limits< bfloat16_t >::max ()
      +       		+inlinestaticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.js b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.js new file mode 100644 index 000000000..aff24a201 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.js @@ -0,0 +1,6 @@ +var structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4 = +[ + [ "infinity", "structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html#a974982399d0211786599526abdb843b8", null ], + [ "lowest", "structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html#a3623580fbfd92ceb69cdd8e329e18fa8", null ], + [ "max", "structmlx_1_1core_1_1numeric__limits_3_01bfloat16__t_01_4.html#a01712fcb04266320225c168a0e6f619a", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4-members.html b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4-members.html new file mode 100644 index 000000000..65b15d1c4 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4-members.html @@ -0,0 +1,123 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::numeric_limits< float16_t > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::numeric_limits< float16_t >, including all inherited members.

      + + + + +
      infinity()mlx::core::numeric_limits< float16_t >inlinestatic
      lowest()mlx::core::numeric_limits< float16_t >inlinestatic
      max()mlx::core::numeric_limits< float16_t >inlinestatic
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html new file mode 100644 index 000000000..fd49a4938 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html @@ -0,0 +1,212 @@ + + + + + + + +MLX: mlx::core::numeric_limits< float16_t > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Static Public Member Functions | +List of all members
      +
      mlx::core::numeric_limits< float16_t > Struct Reference
      +
      +
      + +

      #include <limits.h>

      + + + + + + + + +

      +Static Public Member Functions

      static constexpr float16_t lowest ()
       
      static constexpr float16_t max ()
       
      static constexpr float16_t infinity ()
       
      +

      Member Function Documentation

      + +

      ◆ infinity()

      + +
      +
      + + + + + +
      + + + + + + + +
      static constexpr float16_t mlx::core::numeric_limits< float16_t >::infinity ()
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ lowest()

      + +
      +
      + + + + + +
      + + + + + + + +
      static constexpr float16_t mlx::core::numeric_limits< float16_t >::lowest ()
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ max()

      + +
      +
      + + + + + +
      + + + + + + + +
      static constexpr float16_t mlx::core::numeric_limits< float16_t >::max ()
      +       		+inlinestaticconstexpr
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.js b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.js new file mode 100644 index 000000000..e69960a2c --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.js @@ -0,0 +1,6 @@ +var structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4 = +[ + [ "infinity", "structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html#a2a8f3f489b47b7e8398bec9895ae0c27", null ], + [ "lowest", "structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html#abc2d9cd8d0a90219f7eb6fd05b98e4ac", null ], + [ "max", "structmlx_1_1core_1_1numeric__limits_3_01float16__t_01_4.html#a6dd1fadd4cc7c2cec6223977c238c334", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float_01_4.html b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float_01_4.html new file mode 100644 index 000000000..4f69f1945 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float_01_4.html @@ -0,0 +1,129 @@ + + + + + + + +MLX: mlx::core::numeric_limits< float > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::numeric_limits< float > Struct Reference
      +
      +
      + +

      #include <limits.h>

      +
      +Inheritance diagram for mlx::core::numeric_limits< float >:
      +
      +
      + +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float_01_4.png b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float_01_4.png new file mode 100644 index 000000000..dbe3f0c78 Binary files /dev/null and b/docs/build/html/structmlx_1_1core_1_1numeric__limits_3_01float_01_4.png differ diff --git a/docs/build/html/structmlx_1_1core_1_1numeric__limits_dup.js b/docs/build/html/structmlx_1_1core_1_1numeric__limits_dup.js deleted file mode 100644 index 886c28194..000000000 --- a/docs/build/html/structmlx_1_1core_1_1numeric__limits_dup.js +++ /dev/null @@ -1,5 +0,0 @@ -var structmlx_1_1core_1_1numeric__limits_dup = -[ - [ "bits", "structmlx_1_1core_1_1numeric__limits.html#ac1750c0d3daefb90b85088908e7fdb31", null ], - [ "value", "structmlx_1_1core_1_1numeric__limits.html#a8e5bdc5369b95d63fa15250f6997c652", null ] -]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1scheduler_1_1_stream_thread-members.html b/docs/build/html/structmlx_1_1core_1_1scheduler_1_1_stream_thread-members.html index 6bbe917eb..5276a85f6 100644 --- a/docs/build/html/structmlx_1_1core_1_1scheduler_1_1_stream_thread-members.html +++ b/docs/build/html/structmlx_1_1core_1_1scheduler_1_1_stream_thread-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@
      - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::simd::ScalarT< T, N > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::simd::ScalarT< T, N >, including all inherited members.

      + + +
      v typedefmlx::core::simd::ScalarT< T, N >
      +
      + + + + diff --git a/docs/build/html/unionmlx_1_1core_1_1detail_1_1_int_or_float.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t.html similarity index 65% rename from docs/build/html/unionmlx_1_1core_1_1detail_1_1_int_or_float.html rename to docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t.html index 6f772be51..555add971 100644 --- a/docs/build/html/unionmlx_1_1core_1_1detail_1_1_int_or_float.html +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t.html @@ -3,9 +3,9 @@ - + -MLX: mlx::core::detail::IntOrFloat Union Reference +MLX: mlx::core::simd::ScalarT< T, N > Struct Template Reference @@ -52,7 +52,7 @@ - +
      @@ -103,60 +103,46 @@ $(function(){initNavTree('unionmlx_1_1core_1_1detail_1_1_int_or_float.html','');
      -Public Attributes | -List of all members
      -
      mlx::core::detail::IntOrFloat Union Reference
      +Public Types | +List of all members
      +
      mlx::core::simd::ScalarT< T, N > Struct Template Reference
      -

      #include <ops.h>

      +

      #include <accelerate_simd.h>

      - - - - - + + +

      -Public Attributes

      int i
       
      float f
       

      +Public Types

      using v = T
       
      -

      Member Data Documentation

      - -

      ◆ f

      +

      Member Typedef Documentation

      + +

      ◆ v

      +
      +template<typename T, int N>
      - +
      float mlx::core::detail::IntOrFloat::fusing mlx::core::simd::ScalarT< T, N >::v = T
      - -

      ◆ i

      - -
      -
      - - - - -
      int mlx::core::detail::IntOrFloat::i
      -
      - -
      -
      -
      The documentation for this union was generated from the following file:
        -
      • mlx/backend/common/ops.h
        • +
        The documentation for this struct was generated from the following file:
      diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t.js b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t.js new file mode 100644 index 000000000..fcec4e8b4 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t.js @@ -0,0 +1,4 @@ +var structmlx_1_1core_1_1simd_1_1_scalar_t = +[ + [ "v", "structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4-members.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4-members.html new file mode 100644 index 000000000..b1659582e --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4-members.html @@ -0,0 +1,122 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::simd::ScalarT< bool, N > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::simd::ScalarT< bool, N >, including all inherited members.

      + + + +
      v typedefmlx::core::simd::ScalarT< bool, N >
      v typedefmlx::core::simd::ScalarT< bool, N >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html new file mode 100644 index 000000000..a97983e71 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html @@ -0,0 +1,165 @@ + + + + + + + +MLX: mlx::core::simd::ScalarT< bool, N > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      mlx::core::simd::ScalarT< bool, N > Struct Template Reference
      +
      +
      + +

      #include <accelerate_simd.h>

      + + + + + + +

      +Public Types

      using v = char
       
      using v
       
      +

      Member Typedef Documentation

      + +

      ◆ v [1/2]

      + +
      +
      + + + + +
      using mlx::core::simd::ScalarT< bool, N >::v
      +
      + +
      +
      + +

      ◆ v [2/2]

      + +
      +
      +
      +template<int N>
      + + + + +
      using mlx::core::simd::ScalarT< bool, N >::v = char
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.js b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.js new file mode 100644 index 000000000..8b26d10e5 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.js @@ -0,0 +1,5 @@ +var structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4 = +[ + [ "v", "structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49", null ], + [ "v", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01bool_00_01_n_01_4.html#a3d47d5ad1ff8981bd9876a5fc1870174", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4-members.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4-members.html new file mode 100644 index 000000000..2f4a85604 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4-members.html @@ -0,0 +1,122 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::simd::ScalarT< int64_t, N > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::simd::ScalarT< int64_t, N >, including all inherited members.

      + + + +
      v typedefmlx::core::simd::ScalarT< int64_t, N >
      v typedefmlx::core::simd::ScalarT< int64_t, N >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html new file mode 100644 index 000000000..3a8b0ba83 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html @@ -0,0 +1,165 @@ + + + + + + + +MLX: mlx::core::simd::ScalarT< int64_t, N > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      mlx::core::simd::ScalarT< int64_t, N > Struct Template Reference
      +
      +
      + +

      #include <accelerate_simd.h>

      + + + + + + +

      +Public Types

      using v = long
       
      using v
       
      +

      Member Typedef Documentation

      + +

      ◆ v [1/2]

      + +
      +
      + + + + +
      using mlx::core::simd::ScalarT< int64_t, N >::v
      +
      + +
      +
      + +

      ◆ v [2/2]

      + +
      +
      +
      +template<int N>
      + + + + +
      using mlx::core::simd::ScalarT< int64_t, N >::v = long
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.js b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.js new file mode 100644 index 000000000..a2c35e526 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.js @@ -0,0 +1,5 @@ +var structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4 = +[ + [ "v", "structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49", null ], + [ "v", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int64__t_00_01_n_01_4.html#aa36db163e4909aea98b7129764184801", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4-members.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4-members.html new file mode 100644 index 000000000..b454d3f8b --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4-members.html @@ -0,0 +1,122 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::simd::ScalarT< int8_t, N > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::simd::ScalarT< int8_t, N >, including all inherited members.

      + + + +
      v typedefmlx::core::simd::ScalarT< int8_t, N >
      v typedefmlx::core::simd::ScalarT< int8_t, N >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html new file mode 100644 index 000000000..9359275f0 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html @@ -0,0 +1,165 @@ + + + + + + + +MLX: mlx::core::simd::ScalarT< int8_t, N > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      mlx::core::simd::ScalarT< int8_t, N > Struct Template Reference
      +
      +
      + +

      #include <accelerate_simd.h>

      + + + + + + +

      +Public Types

      using v = char
       
      using v
       
      +

      Member Typedef Documentation

      + +

      ◆ v [1/2]

      + +
      +
      + + + + +
      using mlx::core::simd::ScalarT< int8_t, N >::v
      +
      + +
      +
      + +

      ◆ v [2/2]

      + +
      +
      +
      +template<int N>
      + + + + +
      using mlx::core::simd::ScalarT< int8_t, N >::v = char
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.js b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.js new file mode 100644 index 000000000..ca5e7a89c --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.js @@ -0,0 +1,5 @@ +var structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4 = +[ + [ "v", "structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49", null ], + [ "v", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01int8__t_00_01_n_01_4.html#af2775b07509324182bd715aac65b7eb0", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4-members.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4-members.html new file mode 100644 index 000000000..328bb3ed4 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4-members.html @@ -0,0 +1,122 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::simd::ScalarT< uint64_t, N > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::simd::ScalarT< uint64_t, N >, including all inherited members.

      + + + +
      v typedefmlx::core::simd::ScalarT< uint64_t, N >
      v typedefmlx::core::simd::ScalarT< uint64_t, N >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html new file mode 100644 index 000000000..0467fefa0 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html @@ -0,0 +1,165 @@ + + + + + + + +MLX: mlx::core::simd::ScalarT< uint64_t, N > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      mlx::core::simd::ScalarT< uint64_t, N > Struct Template Reference
      +
      +
      + +

      #include <accelerate_simd.h>

      + + + + + + +

      +Public Types

      using v = unsigned long
       
      using v
       
      +

      Member Typedef Documentation

      + +

      ◆ v [1/2]

      + +
      +
      + + + + +
      using mlx::core::simd::ScalarT< uint64_t, N >::v
      +
      + +
      +
      + +

      ◆ v [2/2]

      + +
      +
      +
      +template<int N>
      + + + + +
      using mlx::core::simd::ScalarT< uint64_t, N >::v = unsigned long
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.js b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.js new file mode 100644 index 000000000..9ae18465c --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.js @@ -0,0 +1,5 @@ +var structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4 = +[ + [ "v", "structmlx_1_1core_1_1simd_1_1_scalar_t.html#af165519c33808c4f815143f77739db49", null ], + [ "v", "structmlx_1_1core_1_1simd_1_1_scalar_t_3_01uint64__t_00_01_n_01_4.html#aaf352e77f7ab310c40a31d3dd2bde0eb", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd-members.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd-members.html new file mode 100644 index 000000000..c6b616492 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd-members.html @@ -0,0 +1,129 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::simd::Simd< T, N > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::simd::Simd< T, N >, including all inherited members.

      + + + + + + + + + + +
      operator[](int idx) constmlx::core::simd::Simd< T, N >inline
      operator[](int idx)mlx::core::simd::Simd< T, N >inline
      scalar_t typedefmlx::core::simd::Simd< T, N >
      Simd()mlx::core::simd::Simd< T, N >inline
      Simd(Simd< U, N > other)mlx::core::simd::Simd< T, N >inline
      Simd(U v)mlx::core::simd::Simd< T, N >inline
      Simd(Simd< T, N/2 > x, Simd< T, N/2 > y)mlx::core::simd::Simd< T, N >inline
      sizemlx::core::simd::Simd< T, N >static
      valuemlx::core::simd::Simd< T, N >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd.html new file mode 100644 index 000000000..896ea7717 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd.html @@ -0,0 +1,392 @@ + + + + + + + +MLX: mlx::core::simd::Simd< T, N > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +Public Member Functions | +Public Attributes | +Static Public Attributes | +List of all members
      +
      mlx::core::simd::Simd< T, N > Struct Template Reference
      +
      +
      + +

      #include <accelerate_simd.h>

      + + + + +

      +Public Types

      using scalar_t = typename ScalarT<T, N>::v
       
      + + + + + + + + + + + + + + + +

      +Public Member Functions

       Simd ()
       
      template<typename U>
       Simd (Simd< U, N > other)
       
      template<typename U>
       Simd (U v)
       
       Simd (Simd< T, N/2 > x, Simd< T, N/2 > y)
       
      T operator[] (int idx) const
       
      T & operator[] (int idx)
       
      + + + +

      +Public Attributes

      asd::Vector< scalar_t, N >::packed_t value
       
      + + + +

      +Static Public Attributes

      static constexpr int size = N
       
      +

      Member Typedef Documentation

      + +

      ◆ scalar_t

      + +
      +
      +
      +template<typename T, int N>
      + + + + +
      using mlx::core::simd::Simd< T, N >::scalar_t = typename ScalarT<T, N>::v
      +
      + +
      +
      +

      Constructor & Destructor Documentation

      + +

      ◆ Simd() [1/4]

      + +
      +
      +
      +template<typename T, int N>
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< T, N >::Simd ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [2/4]

      + +
      +
      +
      +template<typename T, int N>
      +
      +template<typename U>
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< T, N >::Simd (Simd< U, N > other)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [3/4]

      + +
      +
      +
      +template<typename T, int N>
      +
      +template<typename U>
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< T, N >::Simd (U v)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [4/4]

      + +
      +
      +
      +template<typename T, int N>
      + + + + + +
      + + + + + + + + + + + +
      mlx::core::simd::Simd< T, N >::Simd (Simd< T, N/2 > x,
      Simd< T, N/2 > y )
      +       		+inline
      +
      + +
      +
      +

      Member Function Documentation

      + +

      ◆ operator[]() [1/2]

      + +
      +
      +
      +template<typename T, int N>
      + + + + + +
      + + + + + + + +
      T & mlx::core::simd::Simd< T, N >::operator[] (int idx)
      +       		+inline
      +
      + +
      +
      + +

      ◆ operator[]() [2/2]

      + +
      +
      +
      +template<typename T, int N>
      + + + + + +
      + + + + + + + +
      T mlx::core::simd::Simd< T, N >::operator[] (int idx) const
      +       		+inline
      +
      + +
      +
      +

      Member Data Documentation

      + +

      ◆ size

      + +
      +
      +
      +template<typename T, int N>
      + + + + + +
      + + + + +
      int mlx::core::simd::Simd< T, N >::size = N
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ value

      + +
      +
      +
      +template<typename T, int N>
      + + + + +
      asd::Vector<scalar_t,N>::packed_t mlx::core::simd::Simd< T, N >::value
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd.js b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd.js new file mode 100644 index 000000000..567f429c7 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd.js @@ -0,0 +1,12 @@ +var structmlx_1_1core_1_1simd_1_1_simd = +[ + [ "scalar_t", "structmlx_1_1core_1_1simd_1_1_simd.html#a26040194a37172b6aed7c5d1685362fb", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#a5c24246e05e833fd81d900226a29e6ab", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#a1693c8e542dddf2ab60d309d64de71b6", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#a74091cd8b7c72014f73ec215b52ea2ce", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#aabe757c8fd93dadd6f8859cda99f4927", null ], + [ "operator[]", "structmlx_1_1core_1_1simd_1_1_simd.html#ae877ce4884241399de8e28090441f557", null ], + [ "operator[]", "structmlx_1_1core_1_1simd_1_1_simd.html#a235268dc56eb1bb5b86cd3aade67b77c", null ], + [ "size", "structmlx_1_1core_1_1simd_1_1_simd.html#aa2b56facc70ba4e8d33a74def204a1fd", null ], + [ "value", "structmlx_1_1core_1_1simd_1_1_simd.html#a36e2b7db5ce6eb4dd456e99a4cd2c2cf", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4-members.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4-members.html new file mode 100644 index 000000000..f7743f08e --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4-members.html @@ -0,0 +1,134 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::simd::Simd< T, 1 > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::simd::Simd< T, 1 >, including all inherited members.

      + + + + + + + + + + + + + + + +
      operator[](int idx) constmlx::core::simd::Simd< T, 1 >inline
      operator[](int idx)mlx::core::simd::Simd< T, 1 >inline
      scalar_t typedefmlx::core::simd::Simd< T, 1 >
      Simd()mlx::core::simd::Simd< T, 1 >inline
      Simd(Simd< U, 1 > v)mlx::core::simd::Simd< T, 1 >inline
      Simd(U v)mlx::core::simd::Simd< T, 1 >inline
      Simd()mlx::core::simd::Simd< T, 1 >inline
      Simd(Simd< U, N > other)mlx::core::simd::Simd< T, 1 >inline
      Simd(U v)mlx::core::simd::Simd< T, 1 >inline
      Simd(Simd< T, N/2 > x, Simd< T, N/2 > y)mlx::core::simd::Simd< T, 1 >inline
      sizemlx::core::simd::Simd< T, 1 >static
      sizemlx::core::simd::Simd< T, 1 >static
      valuemlx::core::simd::Simd< T, 1 >
      valuemlx::core::simd::Simd< T, 1 >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html new file mode 100644 index 000000000..ef7a8f841 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html @@ -0,0 +1,505 @@ + + + + + + + +MLX: mlx::core::simd::Simd< T, 1 > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +Public Member Functions | +Public Attributes | +Static Public Attributes | +List of all members
      +
      mlx::core::simd::Simd< T, 1 > Struct Template Reference
      +
      +
      + +

      #include <base_simd.h>

      + + + + +

      +Public Types

      using scalar_t
       
      + + + + + + + + + + + + + + + + + + + + + +

      +Public Member Functions

       Simd ()
       
      template<typename U>
       Simd (Simd< U, 1 > v)
       
      template<typename U>
       Simd (U v)
       
       Simd ()
       
       Simd (Simd< U, N > other)
       
       Simd (U v)
       
       Simd (Simd< T, N/2 > x, Simd< T, N/2 > y)
       
      T operator[] (int idx) const
       
      T & operator[] (int idx)
       
      + + + + + +

      +Public Attributes

      T value
       
      asd::Vector< scalar_t, N >::packed_t value
       
      + + + + + +

      +Static Public Attributes

      static constexpr int size = 1
       
      static constexpr int size
       
      +

      Member Typedef Documentation

      + +

      ◆ scalar_t

      + +
      +
      + + + + +
      using mlx::core::simd::Simd< T, N >::scalar_t
      +
      + +
      +
      +

      Constructor & Destructor Documentation

      + +

      ◆ Simd() [1/7]

      + +
      +
      +
      +template<typename T>
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< T, 1 >::Simd ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [2/7]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename U>
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< T, 1 >::Simd (Simd< U, 1 > v)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [3/7]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename U>
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< T, 1 >::Simd (U v)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [4/7]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< T, N >::Simd ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [5/7]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< T, N >::Simd (Simd< U, N > other)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [6/7]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< T, N >::Simd (U v)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [7/7]

      + +
      +
      + + + + + +
      + + + + + + + + + + + +
      mlx::core::simd::Simd< T, N >::Simd (Simd< T, N/2 > x,
      Simd< T, N/2 > y )
      +       		+inline
      +
      + +
      +
      +

      Member Function Documentation

      + +

      ◆ operator[]() [1/2]

      + +
      +
      + + + + + +
      + + + + + + + +
      T & mlx::core::simd::Simd< T, N >::operator[] (int idx)
      +       		+inline
      +
      + +
      +
      + +

      ◆ operator[]() [2/2]

      + +
      +
      + + + + + +
      + + + + + + + +
      T mlx::core::simd::Simd< T, N >::operator[] (int idx) const
      +       		+inline
      +
      + +
      +
      +

      Member Data Documentation

      + +

      ◆ size [1/2]

      + +
      +
      + + + + + +
      + + + + +
      int mlx::core::simd::Simd< T, N >::size
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ size [2/2]

      + +
      +
      +
      +template<typename T>
      + + + + + +
      + + + + +
      int mlx::core::simd::Simd< T, 1 >::size = 1
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ value [1/2]

      + +
      +
      + + + + +
      asd::Vector<scalar_t,N>::packed_t mlx::core::simd::Simd< T, N >::value
      +
      + +
      +
      + +

      ◆ value [2/2]

      + +
      +
      +
      +template<typename T>
      + + + + +
      T mlx::core::simd::Simd< T, 1 >::value
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.js b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.js new file mode 100644 index 000000000..f31f5bbbb --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.js @@ -0,0 +1,17 @@ +var structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4 = +[ + [ "scalar_t", "structmlx_1_1core_1_1simd_1_1_simd.html#a26040194a37172b6aed7c5d1685362fb", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#a3f6e4a83ecf897465f44160b6fad5a7a", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#a585bc4768c4f7e1313d7e8756fbb00cc", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#acf948f7c5e8829432c0ac17fc9f911e2", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#a5c24246e05e833fd81d900226a29e6ab", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#a1693c8e542dddf2ab60d309d64de71b6", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#a74091cd8b7c72014f73ec215b52ea2ce", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#aabe757c8fd93dadd6f8859cda99f4927", null ], + [ "operator[]", "structmlx_1_1core_1_1simd_1_1_simd.html#ae877ce4884241399de8e28090441f557", null ], + [ "operator[]", "structmlx_1_1core_1_1simd_1_1_simd.html#a235268dc56eb1bb5b86cd3aade67b77c", null ], + [ "size", "structmlx_1_1core_1_1simd_1_1_simd.html#aa2b56facc70ba4e8d33a74def204a1fd", null ], + [ "size", "structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#a14e16c6e2ef5e89135cf8e85dc9f1f1f", null ], + [ "value", "structmlx_1_1core_1_1simd_1_1_simd.html#a36e2b7db5ce6eb4dd456e99a4cd2c2cf", null ], + [ "value", "structmlx_1_1core_1_1simd_1_1_simd_3_01_t_00_011_01_4.html#ac348445fd44bce2b6ee77adeac7d82df", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4-members.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4-members.html new file mode 100644 index 000000000..314dfd9e8 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4-members.html @@ -0,0 +1,141 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::core::simd::Simd< float16_t, N > Member List
      +
      +
      + +

      This is the complete list of members for mlx::core::simd::Simd< float16_t, N >, including all inherited members.

      + + + + + + + + + + + + + + + + + + + + + + +
      operator Simd< float, N >()mlx::core::simd::Simd< float16_t, N >inline
      operator Simd< int16_t, N >()mlx::core::simd::Simd< float16_t, N >inline
      operator[](int idx) constmlx::core::simd::Simd< float16_t, N >inline
      operator[](int idx)mlx::core::simd::Simd< float16_t, N >inline
      operator[](int idx) constmlx::core::simd::Simd< float16_t, N >inline
      operator[](int idx)mlx::core::simd::Simd< float16_t, N >inline
      scalar_t typedefmlx::core::simd::Simd< float16_t, N >
      scalar_t typedefmlx::core::simd::Simd< float16_t, N >
      Simd()mlx::core::simd::Simd< float16_t, N >inline
      Simd(U v)mlx::core::simd::Simd< float16_t, N >inline
      Simd(float16x8_t v)mlx::core::simd::Simd< float16_t, N >inline
      Simd(Simd< float, N > other)mlx::core::simd::Simd< float16_t, N >inline
      Simd(Simd< uint16_t, N > other)mlx::core::simd::Simd< float16_t, N >inline
      Simd()mlx::core::simd::Simd< float16_t, N >inline
      Simd(Simd< U, N > other)mlx::core::simd::Simd< float16_t, N >inline
      Simd(U v)mlx::core::simd::Simd< float16_t, N >inline
      Simd(Simd< float16_t, N/2 > x, Simd< float16_t, N/2 > y)mlx::core::simd::Simd< float16_t, N >inline
      sizemlx::core::simd::Simd< float16_t, N >static
      sizemlx::core::simd::Simd< float16_t, N >static
      valuemlx::core::simd::Simd< float16_t, N >
      valuemlx::core::simd::Simd< float16_t, N >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html new file mode 100644 index 000000000..5b08cd73a --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html @@ -0,0 +1,670 @@ + + + + + + + +MLX: mlx::core::simd::Simd< float16_t, N > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +Public Member Functions | +Public Attributes | +Static Public Attributes | +List of all members
      +
      mlx::core::simd::Simd< float16_t, N > Struct Reference
      +
      +
      + +

      #include <neon_fp16_simd.h>

      + + + + + + +

      +Public Types

      using scalar_t = float16_t
       
      using scalar_t
       
      + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

      +Public Member Functions

       Simd ()
       
      template<typename U>
       Simd (U v)
       
       Simd (float16x8_t v)
       
       Simd (Simd< float, N > other)
       
       Simd (Simd< uint16_t, N > other)
       
       operator Simd< int16_t, N > ()
       
       operator Simd< float, N > ()
       
      float16_t operator[] (int idx) const
       
      float16_t & operator[] (int idx)
       
       Simd ()
       
       Simd (Simd< U, N > other)
       
       Simd (U v)
       
       Simd (Simd< float16_t, N/2 > x, Simd< float16_t, N/2 > y)
       
      float16_t operator[] (int idx) const
       
      float16_t & operator[] (int idx)
       
      + + + + + +

      +Public Attributes

      float16x8_t value
       
      asd::Vector< scalar_t, N >::packed_t value
       
      + + + + + +

      +Static Public Attributes

      static constexpr int size = N
       
      static constexpr int size
       
      +

      Member Typedef Documentation

      + +

      ◆ scalar_t [1/2]

      + +
      +
      + + + + +
      using mlx::core::simd::Simd< float16_t, N >::scalar_t
      +
      + +
      +
      + +

      ◆ scalar_t [2/2]

      + +
      +
      + + + + +
      using mlx::core::simd::Simd< float16_t, N >::scalar_t = float16_t
      +
      + +
      +
      +

      Constructor & Destructor Documentation

      + +

      ◆ Simd() [1/9]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::Simd ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [2/9]

      + +
      +
      +
      +template<typename U>
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::Simd (U v)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [3/9]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::Simd (float16x8_t v)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [4/9]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::Simd (Simd< float, N > other)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [5/9]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::Simd (Simd< uint16_t, N > other)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [6/9]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::Simd ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [7/9]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::Simd (Simd< U, N > other)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [8/9]

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::Simd (U v)
      +       		+inline
      +
      + +
      +
      + +

      ◆ Simd() [9/9]

      + +
      +
      + + + + + +
      + + + + + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::Simd (Simd< float16_t, N/2 > x,
      Simd< float16_t, N/2 > y )
      +       		+inline
      +
      + +
      +
      +

      Member Function Documentation

      + +

      ◆ operator Simd< float, N >()

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::operator Simd< float, N > ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ operator Simd< int16_t, N >()

      + +
      +
      + + + + + +
      + + + + + + + +
      mlx::core::simd::Simd< float16_t, N >::operator Simd< int16_t, N > ()
      +       		+inline
      +
      + +
      +
      + +

      ◆ operator[]() [1/4]

      + +
      +
      + + + + + +
      + + + + + + + +
      float16_t & mlx::core::simd::Simd< float16_t, N >::operator[] (int idx)
      +       		+inline
      +
      + +
      +
      + +

      ◆ operator[]() [2/4]

      + +
      +
      + + + + + +
      + + + + + + + +
      float16_t & mlx::core::simd::Simd< float16_t, N >::operator[] (int idx)
      +       		+inline
      +
      + +
      +
      + +

      ◆ operator[]() [3/4]

      + +
      +
      + + + + + +
      + + + + + + + +
      float16_t mlx::core::simd::Simd< float16_t, N >::operator[] (int idx) const
      +       		+inline
      +
      + +
      +
      + +

      ◆ operator[]() [4/4]

      + +
      +
      + + + + + +
      + + + + + + + +
      float16_t mlx::core::simd::Simd< float16_t, N >::operator[] (int idx) const
      +       		+inline
      +
      + +
      +
      +

      Member Data Documentation

      + +

      ◆ size [1/2]

      + +
      +
      + + + + + +
      + + + + +
      int mlx::core::simd::Simd< float16_t, N >::size
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ size [2/2]

      + +
      +
      + + + + + +
      + + + + +
      int mlx::core::simd::Simd< float16_t, N >::size = N
      +       		+staticconstexpr
      +
      + +
      +
      + +

      ◆ value [1/2]

      + +
      +
      + + + + +
      asd::Vector<scalar_t,N>::packed_t mlx::core::simd::Simd< float16_t, N >::value
      +
      + +
      +
      + +

      ◆ value [2/2]

      + +
      +
      + + + + +
      float16x8_t mlx::core::simd::Simd< float16_t, N >::value
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.js b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.js new file mode 100644 index 000000000..63e36fea3 --- /dev/null +++ b/docs/build/html/structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.js @@ -0,0 +1,24 @@ +var structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4 = +[ + [ "scalar_t", "structmlx_1_1core_1_1simd_1_1_simd.html#a26040194a37172b6aed7c5d1685362fb", null ], + [ "scalar_t", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a422e15f018cd242dd62617f4213dace0", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a04a3a73f98fa5c9090b6cf6154e99e8d", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#ad8b628f8834e983853d557cc1e4124bb", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a5e76655d70c0e9ae49eea536c0e3b8cf", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#ae1dfcaca51f9a6fcdb757cb8413ac223", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a1f30c088a6828d0673e927ed6c0a4b2b", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#a5c24246e05e833fd81d900226a29e6ab", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#a1693c8e542dddf2ab60d309d64de71b6", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#a74091cd8b7c72014f73ec215b52ea2ce", null ], + [ "Simd", "structmlx_1_1core_1_1simd_1_1_simd.html#aabe757c8fd93dadd6f8859cda99f4927", null ], + [ "operator Simd< float, N >", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a98affc184d83627d8654e3530ab52d75", null ], + [ "operator Simd< int16_t, N >", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a823af21442333505114fd3fdac9f24de", null ], + [ "operator[]", "structmlx_1_1core_1_1simd_1_1_simd.html#ae877ce4884241399de8e28090441f557", null ], + [ "operator[]", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a97043111a44318b5eb68977ecacbb638", null ], + [ "operator[]", "structmlx_1_1core_1_1simd_1_1_simd.html#a235268dc56eb1bb5b86cd3aade67b77c", null ], + [ "operator[]", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a4b24316469cd9ecc88f8c073ab1a862e", null ], + [ "size", "structmlx_1_1core_1_1simd_1_1_simd.html#aa2b56facc70ba4e8d33a74def204a1fd", null ], + [ "size", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#a2629cb8da72b6f922ed14cc7b6c43ce7", null ], + [ "value", "structmlx_1_1core_1_1simd_1_1_simd.html#a36e2b7db5ce6eb4dd456e99a4cd2c2cf", null ], + [ "value", "structmlx_1_1core_1_1simd_1_1_simd_3_01float16__t_00_01_n_01_4.html#aa6042509bb67de25bedbee1ee1d66094", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1steel_1_1_accum_helper-members.html b/docs/build/html/structmlx_1_1steel_1_1_accum_helper-members.html index 661661905..19a171391 100644 --- a/docs/build/html/structmlx_1_1steel_1_1_accum_helper-members.html +++ b/docs/build/html/structmlx_1_1steel_1_1_accum_helper-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::steel::BaseMMAFrag< T, 8, 8 > Member List
      +
      +
      + +

      This is the complete list of members for mlx::steel::BaseMMAFrag< T, 8, 8 >, including all inherited members.

      + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      col_frag_type typedefmlx::steel::BaseMMAFrag< T, 8, 8 >
      frag_type typedefmlx::steel::BaseMMAFrag< T, 8, 8 >
      frag_type typedefmlx::steel::BaseMMAFrag< T, 8, 8 >
      get_coord(ushort simd_lane_id)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      get_coord(ushort simd_lane_id)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      kElemColsmlx::steel::BaseMMAFrag< T, 8, 8 >
      kElemRowsmlx::steel::BaseMMAFrag< T, 8, 8 >
      kElemsPerFragmlx::steel::BaseMMAFrag< T, 8, 8 >
      kFragColsmlx::steel::BaseMMAFrag< T, 8, 8 >
      kFragRowsmlx::steel::BaseMMAFrag< T, 8, 8 >
      load(thread frag_type &dst, SrcPtrType src, StrX str_x, StrY str_y)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      load(thread frag_type &dst, SrcPtrType src, StrX str_x, StrY str_y)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      load_safe(thread frag_type &dst, SrcPtrType src, StrX str_x, StrY str_y, LimX lim_x, LimY lim_y, OffX off_x=Int< 0 >{}, OffY off_y=Int< 0 >{})mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      load_safe(thread frag_type &dst, SrcPtrType src, StrX str_x, StrY str_y, LimX lim_x, LimY lim_y, OffX off_x=Int< 0 >{}, OffY off_y=Int< 0 >{})mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      mat_type typedefmlx::steel::BaseMMAFrag< T, 8, 8 >
      mat_type typedefmlx::steel::BaseMMAFrag< T, 8, 8 >
      mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      mma(thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      mma(thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      row_bin_op(thread frag_type &inp_vals, thread T *row_vals)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      row_frag_type typedefmlx::steel::BaseMMAFrag< T, 8, 8 >
      row_reduce(thread const frag_type &inp_vals, thread T *reduced_vals)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      store(const thread frag_type &src, DstPtrType dst, StrX str_x, StrY str_y)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      store(const thread frag_type &src, DstPtrType dst, StrX str_x, StrY str_y)mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      store_safe(const thread frag_type &src, DstPtrType dst, StrX str_x, StrY str_y, LimX lim_x, LimY lim_y, OffX off_x=Int< 0 >{}, OffY off_y=Int< 0 >{})mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      store_safe(const thread frag_type &src, DstPtrType dst, StrX str_x, StrY str_y, LimX lim_x, LimY lim_y, OffX off_x=Int< 0 >{}, OffY off_y=Int< 0 >{})mlx::steel::BaseMMAFrag< T, 8, 8 >inlinestatic
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html b/docs/build/html/structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html new file mode 100644 index 000000000..a039ec2bb --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html @@ -0,0 +1,1090 @@ + + + + + + + +MLX: mlx::steel::BaseMMAFrag< T, 8, 8 > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +Static Public Member Functions | +Public Attributes | +List of all members
      +
      mlx::steel::BaseMMAFrag< T, 8, 8 > Struct Template Reference
      +
      +
      + +

      #include <mma.h>

      + + + + + + + + + + + + + + +

      +Public Types

      typedef metal::simdgroup_matrix< T, kFragRows, kFragCols > mat_type
       
      typedef metal::vec< T, kElemsPerFrag > frag_type
       
      typedef metal::vec< T, kElemRows > row_frag_type
       
      typedef metal::vec< T, kElemCols > col_frag_type
       
      typedef metal::simdgroup_matrix< T, kFragRows, kFragCols > mat_type
       
      typedef metal::vec< T, kElemsPerFrag > frag_type
       
      + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

      +Static Public Member Functions

      static METAL_FUNC constexpr short2 get_coord (ushort simd_lane_id)
       
      template<typename SrcPtrType, typename StrX, typename StrY>
      static METAL_FUNC constexpr void load (thread frag_type &dst, SrcPtrType src, StrX str_x, StrY str_y)
       
      template<typename SrcPtrType, typename StrX, typename StrY, typename LimX, typename LimY, typename OffX, typename OffY>
      static METAL_FUNC constexpr void load_safe (thread frag_type &dst, SrcPtrType src, StrX str_x, StrY str_y, LimX lim_x, LimY lim_y, OffX off_x=Int< 0 >{}, OffY off_y=Int< 0 >{})
       
      template<typename DstPtrType, typename StrX, typename StrY>
      static METAL_FUNC constexpr void store (const thread frag_type &src, DstPtrType dst, StrX str_x, StrY str_y)
       
      template<typename DstPtrType, typename StrX, typename StrY, typename LimX, typename LimY, typename OffX, typename OffY>
      static METAL_FUNC constexpr void store_safe (const thread frag_type &src, DstPtrType dst, StrX str_x, StrY str_y, LimX lim_x, LimY lim_y, OffX off_x=Int< 0 >{}, OffY off_y=Int< 0 >{})
       
      static METAL_FUNC constexpr void mma (thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)
       
      static METAL_FUNC constexpr void mma (thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)
       
      template<typename Op>
      static METAL_FUNC constexpr void row_reduce (thread const frag_type &inp_vals, thread T *reduced_vals)
       
      template<typename Op>
      static METAL_FUNC constexpr void row_bin_op (thread frag_type &inp_vals, thread T *row_vals)
       
      static METAL_FUNC constexpr short2 get_coord (ushort simd_lane_id)
       
      template<typename SrcPtrType, typename StrX, typename StrY>
      static METAL_FUNC constexpr void load (thread frag_type &dst, SrcPtrType src, StrX str_x, StrY str_y)
       
      template<typename SrcPtrType, typename StrX, typename StrY, typename LimX, typename LimY, typename OffX, typename OffY>
      static METAL_FUNC constexpr void load_safe (thread frag_type &dst, SrcPtrType src, StrX str_x, StrY str_y, LimX lim_x, LimY lim_y, OffX off_x=Int< 0 >{}, OffY off_y=Int< 0 >{})
       
      template<typename DstPtrType, typename StrX, typename StrY>
      static METAL_FUNC constexpr void store (const thread frag_type &src, DstPtrType dst, StrX str_x, StrY str_y)
       
      template<typename DstPtrType, typename StrX, typename StrY, typename LimX, typename LimY, typename OffX, typename OffY>
      static METAL_FUNC constexpr void store_safe (const thread frag_type &src, DstPtrType dst, StrX str_x, StrY str_y, LimX lim_x, LimY lim_y, OffX off_x=Int< 0 >{}, OffY off_y=Int< 0 >{})
       
      static METAL_FUNC constexpr void mma (thread frag_type &D, thread frag_type &A, thread frag_type &B, thread frag_type &C)
       
      static METAL_FUNC constexpr void mma (thread mat_type &D, thread mat_type &A, thread mat_type &B, thread mat_type &C)
       
      + + + + + + + + + + + +

      +Public Attributes

      STEEL_CONST int kFragRows = 8
       
      STEEL_CONST int kFragCols = 8
       
      STEEL_CONST int kElemsPerFrag = (kFragRows * kFragCols) / 32
       
      STEEL_CONST int kElemRows = 1
       
      STEEL_CONST int kElemCols = 2
       
      +

      Member Typedef Documentation

      + +

      ◆ col_frag_type

      + +
      +
      +
      +template<typename T>
      + + + + +
      typedef metal::vec<T, kElemCols> mlx::steel::BaseMMAFrag< T, 8, 8 >::col_frag_type
      +
      + +
      +
      + +

      ◆ frag_type [1/2]

      + +
      +
      +
      +template<typename T>
      + + + + +
      typedef metal::vec<T, kElemsPerFrag> mlx::steel::BaseMMAFrag< T, 8, 8 >::frag_type
      +
      + +
      +
      + +

      ◆ frag_type [2/2]

      + +
      +
      +
      +template<typename T>
      + + + + +
      typedef metal::vec<T, kElemsPerFrag> mlx::steel::BaseMMAFrag< T, 8, 8 >::frag_type
      +
      + +
      +
      + +

      ◆ mat_type [1/2]

      + +
      +
      +
      +template<typename T>
      + + + + +
      typedef metal::simdgroup_matrix<T, kFragRows, kFragCols> mlx::steel::BaseMMAFrag< T, 8, 8 >::mat_type
      +
      + +
      +
      + +

      ◆ mat_type [2/2]

      + +
      +
      +
      +template<typename T>
      + + + + +
      typedef metal::simdgroup_matrix<T, kFragRows, kFragCols> mlx::steel::BaseMMAFrag< T, 8, 8 >::mat_type
      +
      + +
      +
      + +

      ◆ row_frag_type

      + +
      +
      +
      +template<typename T>
      + + + + +
      typedef metal::vec<T, kElemRows> mlx::steel::BaseMMAFrag< T, 8, 8 >::row_frag_type
      +
      + +
      +
      +

      Member Function Documentation

      + +

      ◆ get_coord() [1/2]

      + +
      +
      +
      +template<typename T>
      + + + + + +
      + + + + + + + +
      static METAL_FUNC constexpr short2 mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord (ushort simd_lane_id)
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ get_coord() [2/2]

      + +
      +
      +
      +template<typename T>
      + + + + + +
      + + + + + + + +
      static METAL_FUNC constexpr short2 mlx::steel::BaseMMAFrag< T, 8, 8 >::get_coord (ushort simd_lane_id)
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ load() [1/2]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename SrcPtrType, typename StrX, typename StrY>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::load (thread frag_type & dst,
      SrcPtrType src,
      StrX str_x,
      StrY str_y )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ load() [2/2]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename SrcPtrType, typename StrX, typename StrY>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::load (thread frag_type & dst,
      SrcPtrType src,
      StrX str_x,
      StrY str_y )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ load_safe() [1/2]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename SrcPtrType, typename StrX, typename StrY, typename LimX, typename LimY, typename OffX, typename OffY>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe (thread frag_type & dst,
      SrcPtrType src,
      StrX str_x,
      StrY str_y,
      LimX lim_x,
      LimY lim_y,
      OffX off_x = Int<0>{},
      OffY off_y = Int<0>{} )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ load_safe() [2/2]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename SrcPtrType, typename StrX, typename StrY, typename LimX, typename LimY, typename OffX, typename OffY>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::load_safe (thread frag_type & dst,
      SrcPtrType src,
      StrX str_x,
      StrY str_y,
      LimX lim_x,
      LimY lim_y,
      OffX off_x = Int<0>{},
      OffY off_y = Int<0>{} )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ mma() [1/4]

      + +
      +
      +
      +template<typename T>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::mma (thread frag_type & D,
      thread frag_type & A,
      thread frag_type & B,
      thread frag_type & C )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ mma() [2/4]

      + +
      +
      +
      +template<typename T>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::mma (thread frag_type & D,
      thread frag_type & A,
      thread frag_type & B,
      thread frag_type & C )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ mma() [3/4]

      + +
      +
      +
      +template<typename T>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::mma (thread mat_type & D,
      thread mat_type & A,
      thread mat_type & B,
      thread mat_type & C )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ mma() [4/4]

      + +
      +
      +
      +template<typename T>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::mma (thread mat_type & D,
      thread mat_type & A,
      thread mat_type & B,
      thread mat_type & C )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ row_bin_op()

      + +
      +
      +
      +template<typename T>
      +
      +template<typename Op>
      + + + + + +
      + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::row_bin_op (thread frag_type & inp_vals,
      thread T * row_vals )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ row_reduce()

      + +
      +
      +
      +template<typename T>
      +
      +template<typename Op>
      + + + + + +
      + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::row_reduce (thread const frag_type & inp_vals,
      thread T * reduced_vals )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ store() [1/2]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename DstPtrType, typename StrX, typename StrY>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::store (const thread frag_type & src,
      DstPtrType dst,
      StrX str_x,
      StrY str_y )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ store() [2/2]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename DstPtrType, typename StrX, typename StrY>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::store (const thread frag_type & src,
      DstPtrType dst,
      StrX str_x,
      StrY str_y )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ store_safe() [1/2]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename DstPtrType, typename StrX, typename StrY, typename LimX, typename LimY, typename OffX, typename OffY>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe (const thread frag_type & src,
      DstPtrType dst,
      StrX str_x,
      StrY str_y,
      LimX lim_x,
      LimY lim_y,
      OffX off_x = Int<0>{},
      OffY off_y = Int<0>{} )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      + +

      ◆ store_safe() [2/2]

      + +
      +
      +
      +template<typename T>
      +
      +template<typename DstPtrType, typename StrX, typename StrY, typename LimX, typename LimY, typename OffX, typename OffY>
      + + + + + +
      + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      static METAL_FUNC constexpr void mlx::steel::BaseMMAFrag< T, 8, 8 >::store_safe (const thread frag_type & src,
      DstPtrType dst,
      StrX str_x,
      StrY str_y,
      LimX lim_x,
      LimY lim_y,
      OffX off_x = Int<0>{},
      OffY off_y = Int<0>{} )
      +       		+inlinestaticconstexpr
      +
      + +
      +
      +

      Member Data Documentation

      + +

      ◆ kElemCols

      + +
      +
      +
      +template<typename T>
      + + + + +
      STEEL_CONST int mlx::steel::BaseMMAFrag< T, 8, 8 >::kElemCols = 2
      +
      + +
      +
      + +

      ◆ kElemRows

      + +
      +
      +
      +template<typename T>
      + + + + +
      STEEL_CONST int mlx::steel::BaseMMAFrag< T, 8, 8 >::kElemRows = 1
      +
      + +
      +
      + +

      ◆ kElemsPerFrag

      + +
      +
      +
      +template<typename T>
      + + + + +
      STEEL_CONST int mlx::steel::BaseMMAFrag< T, 8, 8 >::kElemsPerFrag = (kFragRows * kFragCols) / 32
      +
      + +
      +
      + +

      ◆ kFragCols

      + +
      +
      +
      +template<typename T>
      + + + + +
      STEEL_CONST int mlx::steel::BaseMMAFrag< T, 8, 8 >::kFragCols = 8
      +
      + +
      +
      + +

      ◆ kFragRows

      + +
      +
      +
      +template<typename T>
      + + + + +
      STEEL_CONST int mlx::steel::BaseMMAFrag< T, 8, 8 >::kFragRows = 8
      +
      + +
      +
      +
      The documentation for this struct was generated from the following files:
        +
      • mlx/backend/metal/kernels/steel/attn/mma.h
        • +
      • mlx/backend/metal/kernels/steel/gemm/mma.h
        • +
      +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.js b/docs/build/html/structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.js new file mode 100644 index 000000000..c5d926990 --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.js @@ -0,0 +1,30 @@ +var structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4 = +[ + [ "col_frag_type", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#adbb262a3c872e26533b68a39db16459e", null ], + [ "frag_type", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8536bfaa108031c2ea3e9ccdc766ee5b", null ], + [ "frag_type", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8536bfaa108031c2ea3e9ccdc766ee5b", null ], + [ "mat_type", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a25675ae18947a97c6e04157b540103a9", null ], + [ "mat_type", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a25675ae18947a97c6e04157b540103a9", null ], + [ "row_frag_type", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a3dcd4301390937f89ed1dde6d28e341f", null ], + [ "get_coord", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a7331fff1d12f2f8b72b0006a3ad0dd83", null ], + [ "get_coord", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a7331fff1d12f2f8b72b0006a3ad0dd83", null ], + [ "load", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ac73006b36fc710feda3a7c796e21415c", null ], + [ "load", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ac73006b36fc710feda3a7c796e21415c", null ], + [ "load_safe", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ad22aaee4a2938cbdd315b39eda84e07d", null ], + [ "load_safe", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#ad22aaee4a2938cbdd315b39eda84e07d", null ], + [ "mma", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8028512f5a3d2b6acaf966be529627a3", null ], + [ "mma", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a8028512f5a3d2b6acaf966be529627a3", null ], + [ "mma", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1868f57d57c8adedab2c58492ec76946", null ], + [ "mma", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1868f57d57c8adedab2c58492ec76946", null ], + [ "row_bin_op", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a318c4279bdc7b39b7919f108b1cd8010", null ], + [ "row_reduce", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a51d662e4cff88b5ad17d7c44bb6b6970", null ], + [ "store", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#aa8f50ea8961ec5b35c1b81366d64f2cb", null ], + [ "store", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#aa8f50ea8961ec5b35c1b81366d64f2cb", null ], + [ "store_safe", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1f0b00daad8eba2f855bb306e70d2328", null ], + [ "store_safe", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a1f0b00daad8eba2f855bb306e70d2328", null ], + [ "kElemCols", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a7c212200d86b4e93f274d99addf668bd", null ], + [ "kElemRows", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a76aa5aa690dbcc954e957d767fad661f", null ], + [ "kElemsPerFrag", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a3c34dfdc944db110f4735f1b25307cf0", null ], + [ "kFragCols", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a211102315e2afbcfcd2e2c201b638e9f", null ], + [ "kFragRows", "structmlx_1_1steel_1_1_base_m_m_a_frag_3_01_t_00_018_00_018_01_4.html#a2fe53db449c692226f23f6b99fb2c0d4", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1steel_1_1_block_loader-members.html b/docs/build/html/structmlx_1_1steel_1_1_block_loader-members.html index adf84b42d..ab8383621 100644 --- a/docs/build/html/structmlx_1_1steel_1_1_block_loader-members.html +++ b/docs/build/html/structmlx_1_1steel_1_1_block_loader-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::steel::ChannelHelper< 1 > Member List
      +
      +
      + +

      This is the complete list of members for mlx::steel::ChannelHelper< 1 >, including all inherited members.

      + + + + + + + +
      excessmlx::steel::ChannelHelper< 1 >
      excessmlx::steel::ChannelHelper< 1 >
      n_channelsmlx::steel::ChannelHelper< 1 >
      n_channelsmlx::steel::ChannelHelper< 1 >
      vec_sizemlx::steel::ChannelHelper< 1 >
      vec_sizemlx::steel::ChannelHelper< 1 >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html new file mode 100644 index 000000000..6b1dca8c3 --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html @@ -0,0 +1,227 @@ + + + + + + + +MLX: mlx::steel::ChannelHelper< 1 > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Attributes | +List of all members
      +
      mlx::steel::ChannelHelper< 1 > Struct Reference
      +
      +
      + +

      #include <loader_channel_n.h>

      + + + + + + + + + + + + + + +

      +Public Attributes

      STEEL_CONST short n_channels = 1
       
      STEEL_CONST short vec_size = 1
       
      STEEL_CONST short excess = 0
       
      STEEL_CONST short n_channels
       
      STEEL_CONST short vec_size
       
      STEEL_CONST short excess
       
      +

      Member Data Documentation

      + +

      ◆ excess [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::excess
      +
      + +
      +
      + +

      ◆ excess [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 1 >::excess = 0
      +
      + +
      +
      + +

      ◆ n_channels [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::n_channels
      +
      + +
      +
      + +

      ◆ n_channels [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 1 >::n_channels = 1
      +
      + +
      +
      + +

      ◆ vec_size [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::vec_size
      +
      + +
      +
      + +

      ◆ vec_size [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 1 >::vec_size = 1
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_011_01_4.js b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_011_01_4.js new file mode 100644 index 000000000..96bb74685 --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_011_01_4.js @@ -0,0 +1,9 @@ +var structmlx_1_1steel_1_1_channel_helper_3_011_01_4 = +[ + [ "excess", "structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000", null ], + [ "excess", "structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#ada22a8bd8a89078cfa28874055c8e753", null ], + [ "n_channels", "structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a", null ], + [ "n_channels", "structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#a06c2fb9c93660e8f6916228cd77f9494", null ], + [ "vec_size", "structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925", null ], + [ "vec_size", "structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#a71449551bbfe56058440755dfd50fc75", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_012_01_4-members.html b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_012_01_4-members.html new file mode 100644 index 000000000..135808a68 --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_012_01_4-members.html @@ -0,0 +1,126 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::steel::ChannelHelper< 2 > Member List
      +
      +
      + +

      This is the complete list of members for mlx::steel::ChannelHelper< 2 >, including all inherited members.

      + + + + + + + +
      excessmlx::steel::ChannelHelper< 2 >
      excessmlx::steel::ChannelHelper< 2 >
      n_channelsmlx::steel::ChannelHelper< 2 >
      n_channelsmlx::steel::ChannelHelper< 2 >
      vec_sizemlx::steel::ChannelHelper< 2 >
      vec_sizemlx::steel::ChannelHelper< 2 >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html new file mode 100644 index 000000000..880f0079f --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html @@ -0,0 +1,227 @@ + + + + + + + +MLX: mlx::steel::ChannelHelper< 2 > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Attributes | +List of all members
      +
      mlx::steel::ChannelHelper< 2 > Struct Reference
      +
      +
      + +

      #include <loader_channel_n.h>

      + + + + + + + + + + + + + + +

      +Public Attributes

      STEEL_CONST short n_channels = 2
       
      STEEL_CONST short vec_size = 2
       
      STEEL_CONST short excess = 0
       
      STEEL_CONST short n_channels
       
      STEEL_CONST short vec_size
       
      STEEL_CONST short excess
       
      +

      Member Data Documentation

      + +

      ◆ excess [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::excess
      +
      + +
      +
      + +

      ◆ excess [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 2 >::excess = 0
      +
      + +
      +
      + +

      ◆ n_channels [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::n_channels
      +
      + +
      +
      + +

      ◆ n_channels [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 2 >::n_channels = 2
      +
      + +
      +
      + +

      ◆ vec_size [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::vec_size
      +
      + +
      +
      + +

      ◆ vec_size [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 2 >::vec_size = 2
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_012_01_4.js b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_012_01_4.js new file mode 100644 index 000000000..d2aaecc9b --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_012_01_4.js @@ -0,0 +1,9 @@ +var structmlx_1_1steel_1_1_channel_helper_3_012_01_4 = +[ + [ "excess", "structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000", null ], + [ "excess", "structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#acc490f3999230aa592c61bbed7eb7cfe", null ], + [ "n_channels", "structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a", null ], + [ "n_channels", "structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#ac66ff37bc2cf78d96667192a6cca73b5", null ], + [ "vec_size", "structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925", null ], + [ "vec_size", "structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#acfb18991a77a9d1d4a79918ac5f387af", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_013_01_4-members.html b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_013_01_4-members.html new file mode 100644 index 000000000..71a689bf6 --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_013_01_4-members.html @@ -0,0 +1,126 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::steel::ChannelHelper< 3 > Member List
      +
      +
      + +

      This is the complete list of members for mlx::steel::ChannelHelper< 3 >, including all inherited members.

      + + + + + + + +
      excessmlx::steel::ChannelHelper< 3 >
      excessmlx::steel::ChannelHelper< 3 >
      n_channelsmlx::steel::ChannelHelper< 3 >
      n_channelsmlx::steel::ChannelHelper< 3 >
      vec_sizemlx::steel::ChannelHelper< 3 >
      vec_sizemlx::steel::ChannelHelper< 3 >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html new file mode 100644 index 000000000..01e70f0a9 --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html @@ -0,0 +1,227 @@ + + + + + + + +MLX: mlx::steel::ChannelHelper< 3 > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Attributes | +List of all members
      +
      mlx::steel::ChannelHelper< 3 > Struct Reference
      +
      +
      + +

      #include <loader_channel_n.h>

      + + + + + + + + + + + + + + +

      +Public Attributes

      STEEL_CONST short n_channels = 3
       
      STEEL_CONST short vec_size = 4
       
      STEEL_CONST short excess = 1
       
      STEEL_CONST short n_channels
       
      STEEL_CONST short vec_size
       
      STEEL_CONST short excess
       
      +

      Member Data Documentation

      + +

      ◆ excess [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::excess
      +
      + +
      +
      + +

      ◆ excess [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 3 >::excess = 1
      +
      + +
      +
      + +

      ◆ n_channels [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::n_channels
      +
      + +
      +
      + +

      ◆ n_channels [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 3 >::n_channels = 3
      +
      + +
      +
      + +

      ◆ vec_size [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::vec_size
      +
      + +
      +
      + +

      ◆ vec_size [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 3 >::vec_size = 4
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_013_01_4.js b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_013_01_4.js new file mode 100644 index 000000000..e8dc5d0d6 --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_013_01_4.js @@ -0,0 +1,9 @@ +var structmlx_1_1steel_1_1_channel_helper_3_013_01_4 = +[ + [ "excess", "structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000", null ], + [ "excess", "structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#aae404674763f3dc73c5ab29169f8b80f", null ], + [ "n_channels", "structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a", null ], + [ "n_channels", "structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#a071c015713b7bab09930661165517eff", null ], + [ "vec_size", "structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925", null ], + [ "vec_size", "structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#a5cb83774601c29564a6bbc010fc0bf7f", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_014_01_4-members.html b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_014_01_4-members.html new file mode 100644 index 000000000..66ebb7d3b --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_014_01_4-members.html @@ -0,0 +1,126 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::steel::ChannelHelper< 4 > Member List
      +
      +
      + +

      This is the complete list of members for mlx::steel::ChannelHelper< 4 >, including all inherited members.

      + + + + + + + +
      excessmlx::steel::ChannelHelper< 4 >
      excessmlx::steel::ChannelHelper< 4 >
      n_channelsmlx::steel::ChannelHelper< 4 >
      n_channelsmlx::steel::ChannelHelper< 4 >
      vec_sizemlx::steel::ChannelHelper< 4 >
      vec_sizemlx::steel::ChannelHelper< 4 >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html new file mode 100644 index 000000000..9b04628f6 --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html @@ -0,0 +1,227 @@ + + + + + + + +MLX: mlx::steel::ChannelHelper< 4 > Struct Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Attributes | +List of all members
      +
      mlx::steel::ChannelHelper< 4 > Struct Reference
      +
      +
      + +

      #include <loader_channel_n.h>

      + + + + + + + + + + + + + + +

      +Public Attributes

      STEEL_CONST short n_channels = 4
       
      STEEL_CONST short vec_size = 4
       
      STEEL_CONST short excess = 0
       
      STEEL_CONST short n_channels
       
      STEEL_CONST short vec_size
       
      STEEL_CONST short excess
       
      +

      Member Data Documentation

      + +

      ◆ excess [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::excess
      +
      + +
      +
      + +

      ◆ excess [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 4 >::excess = 0
      +
      + +
      +
      + +

      ◆ n_channels [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::n_channels
      +
      + +
      +
      + +

      ◆ n_channels [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 4 >::n_channels = 4
      +
      + +
      +
      + +

      ◆ vec_size [1/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< n_channels_ >::vec_size
      +
      + +
      +
      + +

      ◆ vec_size [2/2]

      + +
      +
      + + + + +
      STEEL_CONST short mlx::steel::ChannelHelper< 4 >::vec_size = 4
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_014_01_4.js b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_014_01_4.js new file mode 100644 index 000000000..be9f9b8a9 --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1_channel_helper_3_014_01_4.js @@ -0,0 +1,9 @@ +var structmlx_1_1steel_1_1_channel_helper_3_014_01_4 = +[ + [ "excess", "structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000", null ], + [ "excess", "structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#aecdd8331fec703d739a6f07b9b901ac8", null ], + [ "n_channels", "structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a", null ], + [ "n_channels", "structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#a167b00a84adf93b60e3d7a943d5eb977", null ], + [ "vec_size", "structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925", null ], + [ "vec_size", "structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#af28cdbe2a3c027d95832de07f60448ca", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structmlx_1_1steel_1_1_conv2_d_general_base_info-members.html b/docs/build/html/structmlx_1_1steel_1_1_conv2_d_general_base_info-members.html index e3106538b..889aea96e 100644 --- a/docs/build/html/structmlx_1_1steel_1_1_conv2_d_general_base_info-members.html +++ b/docs/build/html/structmlx_1_1steel_1_1_conv2_d_general_base_info-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx::steel::is_integral< integral_constant< T, v > > Member List
      +
      +
      + +

      This is the complete list of members for mlx::steel::is_integral< integral_constant< T, v > >, including all inherited members.

      + + + + + + + +
      operator value_type() const noexceptmlx::steel::integral_constant< bool, B >inline
      operator value_type() const noexceptmlx::steel::integral_constant< bool, B >inline
      type typedefmlx::steel::integral_constant< bool, B >
      valuemlx::steel::integral_constant< bool, B >static
      valuemlx::steel::integral_constant< bool, B >static
      value_type typedefmlx::steel::integral_constant< bool, B >
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1is__integral_3_01integral__constant_3_01_t_00_01v_01_4_01_4.html b/docs/build/html/structmlx_1_1steel_1_1is__integral_3_01integral__constant_3_01_t_00_01v_01_4_01_4.html new file mode 100644 index 000000000..5a2b4008a --- /dev/null +++ b/docs/build/html/structmlx_1_1steel_1_1is__integral_3_01integral__constant_3_01_t_00_01v_01_4_01_4.html @@ -0,0 +1,154 @@ + + + + + + + +MLX: mlx::steel::is_integral< integral_constant< T, v > > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +List of all members
      +
      mlx::steel::is_integral< integral_constant< T, v > > Struct Template Reference
      +
      +
      + +

      #include <integral_constant.h>

      +
      +Inheritance diagram for mlx::steel::is_integral< integral_constant< T, v > >:
      +
      +
      + + +mlx::steel::integral_constant< bool, B > +mlx::steel::integral_constant< bool, B > + +
      + + + + + + + + + + + + + + + + + +

      +Additional Inherited Members

      - Public Types inherited from mlx::steel::integral_constant< bool, B >
      using value_type
       
      using type
       
      - Public Member Functions inherited from mlx::steel::integral_constant< bool, B >
      METAL_FUNC constexpr operator value_type () const noexcept
       
      METAL_FUNC constexpr operator value_type () const noexcept
       
      - Static Public Attributes inherited from mlx::steel::integral_constant< bool, B >
      static constexpr constant bool value
       
      static constexpr constant bool value
       
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx_1_1steel_1_1is__integral_3_01integral__constant_3_01_t_00_01v_01_4_01_4.png b/docs/build/html/structmlx_1_1steel_1_1is__integral_3_01integral__constant_3_01_t_00_01v_01_4_01_4.png new file mode 100644 index 000000000..d90952553 Binary files /dev/null and b/docs/build/html/structmlx_1_1steel_1_1is__integral_3_01integral__constant_3_01_t_00_01v_01_4_01_4.png differ diff --git a/docs/build/html/structmlx__atomic-members.html b/docs/build/html/structmlx__atomic-members.html index ea6d63df2..542cd7949 100644 --- a/docs/build/html/structmlx__atomic-members.html +++ b/docs/build/html/structmlx__atomic-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      mlx_atomic< T, enable_if_t< is_metal_atomic< T > > > Member List
      +
      +
      + +

      This is the complete list of members for mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >, including all inherited members.

      + + + +
      valmlx_atomic< T, enable_if_t< is_metal_atomic< T > > >
      valmlx_atomic< T, enable_if_t< is_metal_atomic< T > > >
      +
      + + + + diff --git a/docs/build/html/structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html b/docs/build/html/structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html new file mode 100644 index 000000000..b38e2fbce --- /dev/null +++ b/docs/build/html/structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html @@ -0,0 +1,165 @@ + + + + + + + +MLX: mlx_atomic< T, enable_if_t< is_metal_atomic< T > > > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Attributes | +List of all members
      +
      mlx_atomic< T, enable_if_t< is_metal_atomic< T > > > Struct Template Reference
      +
      +
      + +

      #include <atomic.h>

      + + + + + + +

      +Public Attributes

      atomic< T > val
       
      atomic< uint > val
       
      +

      Member Data Documentation

      + +

      ◆ val [1/2]

      + +
      +
      + + + + +
      atomic<uint> mlx_atomic< T, typename >::val
      +
      + +
      +
      + +

      ◆ val [2/2]

      + +
      +
      +
      +template<typename T>
      + + + + +
      atomic<T> mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >::val
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.js b/docs/build/html/structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.js new file mode 100644 index 000000000..4fc1ed712 --- /dev/null +++ b/docs/build/html/structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.js @@ -0,0 +1,5 @@ +var structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4 = +[ + [ "val", "structmlx__atomic.html#a6f6651b8dd8149917c50cd99b13c6747", null ], + [ "val", "structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html#a8dbf729fcd8c4a16e41b546c7405543d", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structpocketfft_1_1detail_1_1_exec_c2_c-members.html b/docs/build/html/structpocketfft_1_1detail_1_1_exec_c2_c-members.html index 20e23e8fa..b97698a0d 100644 --- a/docs/build/html/structpocketfft_1_1detail_1_1_exec_c2_c-members.html +++ b/docs/build/html/structpocketfft_1_1detail_1_1_exec_c2_c-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      pocketfft::detail::add_vec< cmplx< T > > Member List
      +
      +
      + +

      This is the complete list of members for pocketfft::detail::add_vec< cmplx< T > >, including all inherited members.

      + + + +
      type typedefpocketfft::detail::add_vec< cmplx< T > >
      type typedefpocketfft::detail::add_vec< cmplx< T > >
      +
      + + + + diff --git a/docs/build/html/structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html b/docs/build/html/structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html new file mode 100644 index 000000000..f11235a43 --- /dev/null +++ b/docs/build/html/structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html @@ -0,0 +1,165 @@ + + + + + + + +MLX: pocketfft::detail::add_vec< cmplx< T > > Struct Template Reference + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      +Public Types | +List of all members
      +
      pocketfft::detail::add_vec< cmplx< T > > Struct Template Reference
      +
      +
      + +

      #include <pocketfft.h>

      + + + + + + +

      +Public Types

      using type = cmplx<vtype_t<T>>
       
      using type
       
      +

      Member Typedef Documentation

      + +

      ◆ type [1/2]

      + +
      +
      + + + + +
      using pocketfft::detail::add_vec< T >::type
      +
      + +
      +
      + +

      ◆ type [2/2]

      + +
      +
      +
      +template<typename T>
      + + + + +
      using pocketfft::detail::add_vec< cmplx< T > >::type = cmplx<vtype_t<T>>
      +
      + +
      +
      +
      The documentation for this struct was generated from the following file: +
      +
      + + + + diff --git a/docs/build/html/structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.js b/docs/build/html/structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.js new file mode 100644 index 000000000..fb8e9299d --- /dev/null +++ b/docs/build/html/structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.js @@ -0,0 +1,5 @@ +var structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4 = +[ + [ "type", "structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e", null ], + [ "type", "structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html#a257b1c81fb9f559c48ee90497013494e", null ] +]; \ No newline at end of file diff --git a/docs/build/html/structpocketfft_1_1detail_1_1cmplx-members.html b/docs/build/html/structpocketfft_1_1detail_1_1cmplx-members.html index 081d1b432..1183ce9c0 100644 --- a/docs/build/html/structpocketfft_1_1detail_1_1cmplx-members.html +++ b/docs/build/html/structpocketfft_1_1detail_1_1cmplx-members.html @@ -3,7 +3,7 @@ - + MLX: Member List @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - + @@ -52,7 +52,7 @@ - + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      type.h File Reference
      +
      +
      +
      #include "mlx/backend/cpu/simd/base_simd.h"
      +
      +

      Go to the source code of this file.

      +
      +
      + + + + diff --git a/docs/build/html/type_8h_source.html b/docs/build/html/type_8h_source.html new file mode 100644 index 000000000..ebfeeee5d --- /dev/null +++ b/docs/build/html/type_8h_source.html @@ -0,0 +1,127 @@ + + + + + + + +MLX: mlx/backend/cpu/simd/type.h Source File + + + + + + + + + + + + + + + + +
      +
      + + + + + + + +
      +
      MLX +
      +
      + +   + + + + +
      +
      +
      + + + + +
      +
      + +
      +
      +
      + +
      + +
      +
      + + +
      +
      +
      +
      +
      +
      Loading...
      +
      Searching...
      +
      No Matches
      +
      +
      +
      +
      + +
      +
      type.h
      +
      +
      +Go to the documentation of this file.
      1#pragma once
      +
      2
      +
      3#include "mlx/backend/cpu/simd/base_simd.h"
      +
      4
      +
      5#ifdef MLX_USE_ACCELERATE
      +
      6#include "mlx/backend/cpu/simd/accelerate_simd.h"
      +
      7#endif
      +
      accelerate_simd.h
      +
      base_simd.h
      +
      +
      + + + + diff --git a/docs/build/html/type__traits_8h.html b/docs/build/html/type__traits_8h.html index 9361cfdf9..294064a8b 100644 --- a/docs/build/html/type__traits_8h.html +++ b/docs/build/html/type__traits_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/backend/metal/kernels/steel/utils/type_traits.h File Reference @@ -52,7 +52,7 @@ - + + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/usage/distributed.html b/docs/build/html/usage/distributed.html index 8c6c0d912..4eb525912 100644 --- a/docs/build/html/usage/distributed.html +++ b/docs/build/html/usage/distributed.html @@ -8,7 +8,7 @@ - Distributed Communication — MLX 0.22.0 documentation + Distributed Communication — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home @@ -929,7 +929,7 @@ print 4 etc.

      MPI can be installed with Homebrew, using the Anaconda package manager or compiled from source. Most of our testing is done using openmpi installed with the Anaconda package manager as follows:

      -
      $ conda install openmpi
+
      $ conda install conda-forge::openmpi
 
      
 
      
 
      Installing with Homebrew may require specifying the location of libmpi.dyld
diff --git a/docs/build/html/usage/export.html b/docs/build/html/usage/export.html
index 0e4f3ae59..df07ac838 100644
--- a/docs/build/html/usage/export.html
+++ b/docs/build/html/usage/export.html
@@ -8,7 +8,7 @@
     
     
 
-    Exporting Functions — MLX 0.22.0 documentation
+    Exporting Functions — MLX 0.22.1 documentation
   
   
   
@@ -39,7 +39,7 @@
   
 
 
-    
+    
     
     
     
@@ -51,7 +51,7 @@
     
   
   
-  
+  
   
   
   
@@ -130,8 +130,8 @@
       
     
     
-    MLX 0.22.0 documentation - Home
-    MLX 0.22.0 documentation - Home
+    MLX 0.22.1 documentation - Home
+    MLX 0.22.1 documentation - Home
      diff --git a/docs/build/html/usage/function_transforms.html b/docs/build/html/usage/function_transforms.html index c87c980ec..7d9038f02 100644 --- a/docs/build/html/usage/function_transforms.html +++ b/docs/build/html/usage/function_transforms.html @@ -8,7 +8,7 @@ - Function Transforms — MLX 0.22.0 documentation + Function Transforms — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home
      diff --git a/docs/build/html/usage/indexing.html b/docs/build/html/usage/indexing.html index 058c851df..c1a4a8c08 100644 --- a/docs/build/html/usage/indexing.html +++ b/docs/build/html/usage/indexing.html @@ -8,7 +8,7 @@ - Indexing Arrays — MLX 0.22.0 documentation + Indexing Arrays — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/usage/lazy_evaluation.html b/docs/build/html/usage/lazy_evaluation.html index 4774ceece..2d67decb7 100644 --- a/docs/build/html/usage/lazy_evaluation.html +++ b/docs/build/html/usage/lazy_evaluation.html @@ -8,7 +8,7 @@ - Lazy Evaluation — MLX 0.22.0 documentation + Lazy Evaluation — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/usage/numpy.html b/docs/build/html/usage/numpy.html index 193484df0..2584634e1 100644 --- a/docs/build/html/usage/numpy.html +++ b/docs/build/html/usage/numpy.html @@ -8,7 +8,7 @@ - Conversion to NumPy and Other Frameworks — MLX 0.22.0 documentation + Conversion to NumPy and Other Frameworks — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/usage/quick_start.html b/docs/build/html/usage/quick_start.html index fb4e81fd8..7990f3c79 100644 --- a/docs/build/html/usage/quick_start.html +++ b/docs/build/html/usage/quick_start.html @@ -8,7 +8,7 @@ - Quick Start Guide — MLX 0.22.0 documentation + Quick Start Guide — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/usage/saving_and_loading.html b/docs/build/html/usage/saving_and_loading.html index 8b372a2c0..823d5d444 100644 --- a/docs/build/html/usage/saving_and_loading.html +++ b/docs/build/html/usage/saving_and_loading.html @@ -8,7 +8,7 @@ - Saving and Loading Arrays — MLX 0.22.0 documentation + Saving and Loading Arrays — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/usage/unified_memory.html b/docs/build/html/usage/unified_memory.html index aab86dc50..253943ccd 100644 --- a/docs/build/html/usage/unified_memory.html +++ b/docs/build/html/usage/unified_memory.html @@ -8,7 +8,7 @@ - Unified Memory — MLX 0.22.0 documentation + Unified Memory — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/usage/using_streams.html b/docs/build/html/usage/using_streams.html index fc81f2128..6f6aa0776 100644 --- a/docs/build/html/usage/using_streams.html +++ b/docs/build/html/usage/using_streams.html @@ -8,7 +8,7 @@ - Using Streams — MLX 0.22.0 documentation + Using Streams — MLX 0.22.1 documentation @@ -39,7 +39,7 @@ - + @@ -51,7 +51,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.22.0 documentation - Home - MLX 0.22.0 documentation - Home + MLX 0.22.1 documentation - Home + MLX 0.22.1 documentation - Home diff --git a/docs/build/html/utils_8h.html b/docs/build/html/utils_8h.html index 66f627a03..8bb488222 100644 --- a/docs/build/html/utils_8h.html +++ b/docs/build/html/utils_8h.html @@ -3,7 +3,7 @@ - + MLX: mlx/utils.h File Reference @@ -52,7 +52,7 @@ - +