mlx/docs/build/html/search/functions_2.js

var searchData=
[
  ['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_1_1metal.html#a269d591ec02e2f7c0f7a718fbfa37f73',1,'mlx::core::metal']]],
  ['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#ae0c47c0bd95bb8c44339159e04c0f604',1,'mlx::core::binary_op(const array &a, const array &b, array &out, BinaryOpType bopt)'],['../namespacemlx_1_1core.html#a5160ef5819f58cf040c9613ecce548f1',1,'mlx::core::binary_op(const array &a, const array &b, array &out, BinaryOpType bopt)']]],
  ['binary_5fop_5fdims_12',['binary_op_dims',['../namespacemlx_1_1core.html#aeac6fa9529eedba76b27de9d098de963',1,'mlx::core']]],
  ['binary_5fop_5fdispatch_5fdims_13',['binary_op_dispatch_dims',['../namespacemlx_1_1core.html#ad1229ffbba21bdb81f63482a4651bc5a',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_5finvert_28',['bitwise_invert',['../group__ops.html#gaf1182ae7c049fbc9ee190f3e0fffbf83',1,'mlx::core']]],
  ['bitwise_5for_29',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]],
  ['bitwise_5fxor_30',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]],
  ['bitwisebinary_31',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary']]],
  ['bitwiseinvert_32',['BitwiseInvert',['../classmlx_1_1core_1_1_bitwise_invert.html#aaa0180570a82e93988b982b93cd91623',1,'mlx::core::BitwiseInvert']]],
  ['block_5fmasked_5fgemm_33',['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_34',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]],
  ['block_5fsort_35',['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_36',['block_sort_nc',['../sort_8h.html#ad272ef4422c75565a79b7f5e751bb247',1,'sort.h']]],
  ['blockloader_37',['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_38',['BlockLoaderT',['../structmlx_1_1steel_1_1_block_loader_t.html#a076616a7c67ad1b847e0e6b046077ee2',1,'mlx::steel::BlockLoaderT']]],
  ['blockmaskedmm_39',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM']]],
  ['blockmma_40',['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_41',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]],
  ['broadcast_42',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html#a004cce3029c0427569830016f99648cb',1,'mlx::core::Broadcast']]],
  ['broadcast_5farrays_43',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]],
  ['broadcast_5fshapes_44',['broadcast_shapes',['../namespacemlx_1_1core.html#aefa6a2ec6439e9619cafd227a1dc14ab',1,'mlx::core']]],
  ['broadcast_5fto_45',['broadcast_to',['../group__ops.html#ga2fd5891f11593b7f09550e884f969013',1,'mlx::core']]],
  ['broadcastaxes_46',['BroadcastAxes',['../classmlx_1_1core_1_1_broadcast_axes.html#a5136f33489670cdc0802e46725288195',1,'mlx::core::BroadcastAxes']]],
  ['bs_5fqmm_5fn_47',['bs_qmm_n',['../quantized_8h.html#a323d2d70799c8d9dffe4b64a4285a799',1,'quantized.h']]],
  ['bs_5fqmm_5ft_48',['bs_qmm_t',['../quantized_8h.html#a693057a0c311a73ac0759e005b9806c1',1,'quantized.h']]],
  ['bs_5fqmv_49',['bs_qmv',['../quantized_8h.html#aaf4fb9c4318c5cd27d118004dbdeba61',1,'quantized.h']]],
  ['bs_5fqmv_5ffast_50',['bs_qmv_fast',['../quantized_8h.html#a359282a9f71e487e5d86d246896ab33d',1,'quantized.h']]],
  ['bs_5fqvm_51',['bs_qvm',['../quantized_8h.html#a2c53419ba5019d4722c0f4c2026b1142',1,'quantized.h']]],
  ['buffer_52',['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_53',['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_54',['buffer_size',['../classmlx_1_1core_1_1array.html#a914577c63755b2e862d2da68bbf8e3dd',1,'mlx::core::array']]],
  ['build_5fcommand_55',['build_command',['../classmlx_1_1core_1_1_jit_compiler.html#a10a5cde91ab929ccbdbdf4c4d940f156',1,'mlx::core::JitCompiler']]],
  ['build_5flib_5fname_56',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]]
];