2024-06-07 11:28:06 +08:00
var searchData =
[
[ 'b_0' , [ 'b' , [ '../unionbool4__or__uint.html#a47d77eac47598fe420f8f04a615f76ca' , 1 , 'bool4_or_uint::b' ] , [ '../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a92eb5ffee6ae2fec3ad71c777531578f' , 1 , 'mlx::core::Dtype::b' ] ] ] ,
[ 'backward_1' , [ 'BACKWARD' , [ '../namespacepocketfft_1_1detail.html#a9d1eaa7469c018c39e745733eab9a9c3' , 1 , 'pocketfft::detail' ] ] ] ,
[ 'base_2' , [ 'Base' , [ '../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421' , 1 , 'mlx::core::Log' ] ] ] ,
[ 'base_5fwh_3' , [ '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_4' , [ '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' ] ] ] ,
[ 'batch_5fndim_5' , [ 'batch_ndim' , [ '../struct_m_l_x_fast_attention_params.html#a6f3d94dbe44b32e675558768710bf0a3' , 1 , 'MLXFastAttentionParams::batch_ndim' ] , [ '../structmlx_1_1steel_1_1_g_e_m_m_params.html#a640dc138a8bf7b2b5bed6a436b429c2f' , 1 , 'mlx::steel::GEMMParams::batch_ndim' ] ] ] ,
[ 'batch_5fsize_6' , [ 'batch_size' , [ '../struct_read_writer.html#a689f4890c1d2ce33fc6da7550beec735' , 1 , 'ReadWriter' ] ] ] ,
[ 'batch_5fstride_5fa_7' , [ 'batch_stride_a' , [ '../structmlx_1_1steel_1_1_g_e_m_m_params.html#a76f55783a8e2ee203cf8507eee4b000c' , 1 , 'mlx::steel::GEMMParams' ] ] ] ,
[ 'batch_5fstride_5fb_8' , [ 'batch_stride_b' , [ '../structmlx_1_1steel_1_1_g_e_m_m_params.html#a99b959b12d12da657648fa24d43e49e8' , 1 , 'mlx::steel::GEMMParams' ] ] ] ,
[ 'batch_5fstride_5fc_9' , [ 'batch_stride_c' , [ '../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#a77c065db228e9654a0a75a6ffe47c15a' , 1 , 'mlx::steel::GEMMAddMMParams' ] ] ] ,
[ 'batch_5fstride_5fd_10' , [ 'batch_stride_d' , [ '../structmlx_1_1steel_1_1_g_e_m_m_params.html#ad98006dd509a455864e6aa7c52743a41' , 1 , 'mlx::steel::GEMMParams' ] ] ] ,
[ 'batch_5fstride_5fk_11' , [ 'batch_stride_k' , [ '../struct_m_l_x_fast_attention_params.html#a162826d3f288f64c0aea88a36b34859b' , 1 , 'MLXFastAttentionParams' ] ] ] ,
[ 'batch_5fstride_5fo_12' , [ 'batch_stride_o' , [ '../struct_m_l_x_fast_attention_params.html#a3c5b1170999087f3f3a03830193b55c7' , 1 , 'MLXFastAttentionParams' ] ] ] ,
[ 'batch_5fstride_5fq_13' , [ 'batch_stride_q' , [ '../struct_m_l_x_fast_attention_params.html#a98766fc89f75d5eef65b345f16a782d1' , 1 , 'MLXFastAttentionParams' ] ] ] ,
[ 'batch_5fstride_5fv_14' , [ 'batch_stride_v' , [ '../struct_m_l_x_fast_attention_params.html#a1180e311b95cd4b6d4a336d21b873c21' , 1 , 'MLXFastAttentionParams' ] ] ] ,
[ 'bcols_15' , [ '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' ] ] ] ,
2024-07-12 06:32:08 +08:00
[ 'bcols_5fpacked_16' , [ 'BCOLS_PACKED' , [ '../struct_quantized_block_loader.html#a1392a5278cf6e090ea80ebe7c4ac5fbb' , 1 , 'QuantizedBlockLoader' ] ] ] ,
[ 'begin_17' , [ 'begin' , [ '../classmlx_1_1core_1_1array.html#a76b258b169d7d73419ebbf85340fb914' , 1 , 'mlx::core::array' ] ] ] ,
[ 'bernoulli_18' , [ 'bernoulli' , [ '../namespacemlx_1_1core_1_1random.html#acb3f278fea2c4f06dea947d3bac2e9b7' , 1 , 'mlx::core::random::bernoulli(const array &p, const std::vector< int > &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#a37fcba120a1d246176db5256d3201cd4' , 1 , 'mlx::core::random::bernoulli(T p, const std::vector< int > &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_19' , [ 'beta' , [ '../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#ac0ce4d8a6014f8adb29fd0a0bb23139f' , 1 , 'mlx::steel::GEMMAddMMParams::beta' ] , [ '../structmlx_1_1steel_1_1_transform_axpby.html#a5fc726f085bafd1acbc391886f7fb8b6' , 1 , 'mlx::steel::TransformAxpby::beta' ] ] ] ,
[ 'bf16_2eh_20' , [ 'bf16.h' , [ '../backend_2metal_2kernels_2bf16_8h.html' , 1 , '(Global Namespace)' ] , [ '../types_2bf16_8h.html' , 1 , '(Global Namespace)' ] ] ] ,
[ 'bf16_5fmath_2eh_21' , [ 'bf16_math.h' , [ '../bf16__math_8h.html' , 1 , '' ] ] ] ,
[ 'bfloat16_22' , [ '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_23' , [ 'bfloat16_t' , [ '../backend_2metal_2kernels_2bf16_8h.html#a7782de82393104dd4ad754ce3b316e82' , 1 , 'bfloat16_t: bf16.h' ] , [ '../namespacemlx_1_1core.html#acb5d16c9b83778c7621c38e522e0060b' , 1 , 'mlx::core::bfloat16_t' ] ] ] ,
[ 'bfloat16_5fto_5fuint16_24' , [ 'bfloat16_to_uint16' , [ '../bf16__math_8h.html#a51cfdd4502e755310f6f3456f039bea7' , 1 , 'bf16_math.h' ] ] ] ,
[ 'bfloat_5fbinop_25' , [ 'bfloat_binop' , [ '../backend_2metal_2kernels_2bf16_8h.html#a7694892a131c0e31e5153c088cccb707' , 1 , 'bfloat_binop: bf16.h' ] , [ '../types_2bf16_8h.html#a7694892a131c0e31e5153c088cccb707' , 1 , 'bfloat_binop: bf16.h' ] ] ] ,
[ 'bfloat_5fbinop_5fbase_26' , [ 'bfloat_binop_base' , [ '../backend_2metal_2kernels_2bf16_8h.html#a78c92beda4436da9a2e520fa98c59f70' , 1 , 'bfloat_binop_base: bf16.h' ] , [ '../types_2bf16_8h.html#a78c92beda4436da9a2e520fa98c59f70' , 1 , 'bfloat_binop_base: bf16.h' ] ] ] ,
[ 'bfloat_5fbinop_5fhelper_27' , [ 'bfloat_binop_helper' , [ '../backend_2metal_2kernels_2bf16_8h.html#ac7ff36230dab17e8f17b7a7c80888594' , 1 , 'bfloat_binop_helper: bf16.h' ] , [ '../types_2bf16_8h.html#ac7ff36230dab17e8f17b7a7c80888594' , 1 , 'bfloat_binop_helper: bf16.h' ] ] ] ,
[ 'bfloat_5fbitop_28' , [ 'bfloat_bitop' , [ '../types_2bf16_8h.html#aac9ba86d4bf05bcda1936494f9b9b4d3' , 1 , 'bf16.h' ] ] ] ,
[ 'bfloat_5fbits_5fto_5ffloat_29' , [ 'bfloat_bits_to_float' , [ '../backend_2metal_2kernels_2bf16_8h.html#a3b33ae338dc4f223d0f3c748de07bad1' , 1 , 'bf16.h' ] ] ] ,
[ 'bfloat_5fcompop_30' , [ 'bfloat_compop' , [ '../backend_2metal_2kernels_2bf16_8h.html#a330a0883503cb640f1cf628a7ca50239' , 1 , 'bfloat_compop: bf16.h' ] , [ '../types_2bf16_8h.html#a330a0883503cb640f1cf628a7ca50239' , 1 , 'bfloat_compop: bf16.h' ] ] ] ,
[ 'bfloat_5finplace_5fbitop_31' , [ 'bfloat_inplace_bitop' , [ '../types_2bf16_8h.html#af13b46bc58e6e6f675ae47aabec37711' , 1 , 'bf16.h' ] ] ] ,
[ 'bfloat_5finplace_5fop_32' , [ 'bfloat_inplace_op' , [ '../backend_2metal_2kernels_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_33' , [ 'bfloat_inplace_op_addr_space_helper' , [ '../backend_2metal_2kernels_2bf16_8h.html#af30a2cbd2c3415516203b83bd21872f8' , 1 , 'bfloat_inplace_op_addr_space_helper: bf16.h' ] , [ '../backend_2metal_2kernels_2bf16_8h.html#a1457da931c28fa4e2500daa4e6441e8b' , 1 , 'bfloat_inplace_op_addr_space_helper: bf16.h' ] ] ] ,
[ 'bfloat_5finplace_5fop_5fhelper_34' , [ 'bfloat_inplace_op_helper' , [ '../backend_2metal_2kernels_2bf16_8h.html#a2846fd11b5e19b435e9f7ef0998c9b1d' , 1 , 'bfloat_inplace_op_helper: bf16.h' ] , [ '../backend_2metal_2kernels_2bf16_8h.html#afe5988aa8147be2bafda6a5b7792fe15' , 1 , 'bfloat_inplace_op_helper: bf16.h' ] ] ] ,
[ 'bi_35' , [ 'bi' , [ '../struct_quantized_block_loader.html#a85041d72225a2095659c70509291a906' , 1 , 'QuantizedBlockLoader::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' ] , [ '../structmlx_1_1steel_1_1_block_loader.html#a9ef13742bcdf07532d8f09394928a8af' , 1 , 'mlx::steel::BlockLoader::bi' ] ] ] ,
[ 'biases_36' , [ 'biases' , [ '../struct_quantized_block_loader.html#a17d01a6aba0833b073586ef2c09d0fbd' , 1 , 'QuantizedBlockLoader' ] ] ] ,
[ 'binary_37' , [ 'binary' , [ '../namespacemlx_1_1core_1_1metal.html#a269d591ec02e2f7c0f7a718fbfa37f73' , 1 , 'mlx::core::metal' ] ] ] ,
[ 'binary_2eh_38' , [ '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_39' , [ 'binary_g' , [ '../metal_2kernels_2binary_8h.html#a701e3cbc842992aa74f01b91a733774e' , 1 , 'binary_g(device const T *a, device const T *b, device U *c, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary.h' ] , [ '../metal_2kernels_2binary__two_8h.html#acd870cc9ebe074eb313937e02ea1ed69' , 1 , 'binary_g(device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim): binary_two.h' ] ] ] ,
[ 'binary_5fg_5fnd_40' , [ 'binary_g_nd' , [ '../metal_2kernels_2binary_8h.html#a4f273fb6aad88b74bfc63f0f29499c03' , 1 , 'binary_g_nd(device const T *a, device const T *b, device U *c, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], uint3 index, uint3 grid_dim): binary.h' ] , [ '../metal_2kernels_2binary__two_8h.html#aaec603b0d083afaf9e4710fd1dbdf1dc' , 1 , 'binary_g_nd(device const T *a, device const T *b, device U *c, device U *d, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], uint3 index, uint3 grid_dim): binary_two.h' ] ] ] ,
[ 'binary_5fg_5fnd1_41' , [ 'binary_g_nd1' , [ '../metal_2kernels_2binary_8h.html#a6808bfb006cb5473da087a2758d0d867' , 1 , 'binary_g_nd1(device const T *a, device const T *b, device U *c, constant const size_t &a_stride, constant const size_t &b_stride, uint index): binary.h' ] , [ '../metal_2kernels_2binary__two_8h.html#ad1fad37c168192b212a4294f4cf78133' , 1 , 'binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const size_t &a_stride, constant const size_t &b_stride, uint index): binary_two.h' ] ] ] ,
[ 'binary_5fg_5fnd2_42' , [ 'binary_g_nd2' , [ '../metal_2kernels_2binary_8h.html#a8cd5989852ec704c6fd132ae28f4fc14' , 1 , 'binary_g_nd2(device const T *a, device const T *b, device U *c, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim): binary.h' ] , [ '../metal_2kernels_2binary__two_8h.html#a03f7c15a1607576755abb65c542ae347' , 1 , 'binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim): binary_two.h' ] ] ] ,
[ 'binary_5fg_5fnd3_43' , [ 'binary_g_nd3' , [ '../metal_2kernels_2binary_8h.html#ac4979e60b993f7ffb602bcb91cd68bc9' , 1 , 'binary_g_nd3(device const T *a, device const T *b, device U *c, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim): binary.h' ] , [ '../metal_2kernels_2binary__two_8h.html#a43e5943460996c43060d1f3aa1309ba6' , 1 , 'binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim): binary_two.h' ] ] ] ,
[ 'binary_5fop_5fgpu_44' , [ 'binary_op_gpu' , [ '../namespacemlx_1_1core.html#a63f3c166ea4916faa5d038668cfe16b1' , 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#a6437ced86c227e0807352ae6b16da796' , 1 , 'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)' ] ] ] ,
[ 'binary_5fop_5fgpu_5finplace_45' , [ 'binary_op_gpu_inplace' , [ '../namespacemlx_1_1core.html#afee8898eaceab6dd5e90357f4a0613c4' , 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#ad53410af1488f83d74035e3132fa16af' , 1 , 'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)' ] ] ] ,
[ 'binary_5fops_46' , [ 'binary_ops' , [ '../namespacemlx_1_1core_1_1metal.html#a8db7f9cc781d4bfb08423a401665f322' , 1 , 'mlx::core::metal' ] ] ] ,
[ 'binary_5fops_2eh_47' , [ 'binary_ops.h' , [ '../binary__ops_8h.html' , 1 , '' ] ] ] ,
[ 'binary_5fss_48' , [ '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_49' , [ '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_5ftwo_50' , [ 'binary_two' , [ '../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb' , 1 , 'mlx::core::metal' ] ] ] ,
[ 'binary_5ftwo_2eh_51' , [ 'binary_two.h' , [ '../common_2binary__two_8h.html' , 1 , '(Global Namespace)' ] , [ '../metal_2kernels_2binary__two_8h.html' , 1 , '(Global Namespace)' ] ] ] ,
2024-06-07 11:28:06 +08:00
[ 'binary_5fvs_52' , [ '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_5fvv_53' , [ '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' ] ] ] ,
[ 'bits_54' , [ 'bits' , [ '../namespacemlx_1_1core_1_1random.html#abb895baa477f5a06b5f88e69245f1825' , 1 , 'mlx::core::random::bits(const std::vector< int > &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})' ] , [ '../namespacemlx_1_1core_1_1random.html#a548ffed4ba3107b89885ff850ffce5f4' , 1 , 'mlx::core::random::bits(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})' ] ] ] ,
[ 'bits_5f_55' , [ '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_56' , [ 'bits_to_bfloat' , [ '../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca' , 1 , '_MLX_BFloat16' ] ] ] ,
[ 'bits_5fto_5fbfloat_5fstruct_57' , [ 'bits_to_bfloat_struct' , [ '../struct___m_l_x___b_float16_1_1bits__to__bfloat__struct.html' , 1 , '_MLX_BFloat16' ] ] ] ,
[ 'bitwise_5fand_58' , [ 'bitwise_and' , [ '../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada' , 1 , 'mlx::core' ] ] ] ,
[ 'bitwise_5for_59' , [ 'bitwise_or' , [ '../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6' , 1 , 'mlx::core' ] ] ] ,
[ 'bitwise_5fxor_60' , [ 'bitwise_xor' , [ '../group__ops.html#ga3188638fba3a60e264baf69956a1e08b' , 1 , 'mlx::core' ] ] ] ,
[ 'bitwiseand_61' , [ 'BitwiseAnd' , [ '../struct_bitwise_and.html' , 1 , 'BitwiseAnd' ] , [ '../structmlx_1_1core_1_1detail_1_1_bitwise_and.html' , 1 , 'mlx::core::detail::BitwiseAnd' ] ] ] ,
[ 'bitwisebinary_62' , [ '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_63' , [ 'BitwiseOr' , [ '../struct_bitwise_or.html' , 1 , 'BitwiseOr' ] , [ '../structmlx_1_1core_1_1detail_1_1_bitwise_or.html' , 1 , 'mlx::core::detail::BitwiseOr' ] ] ] ,
[ 'bitwisexor_64' , [ 'BitwiseXor' , [ '../struct_bitwise_xor.html' , 1 , 'BitwiseXor' ] , [ '../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html' , 1 , 'mlx::core::detail::BitwiseXor' ] ] ] ,
2024-07-12 06:32:08 +08:00
[ 'bj_65' , [ 'bj' , [ '../struct_quantized_block_loader.html#ae2add92b2aaf3414e91f0470b9b0cc00' , 1 , 'QuantizedBlockLoader::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' ] , [ '../structmlx_1_1steel_1_1_block_loader.html#a78c326e75ee35a484685771143047cd4' , 1 , 'mlx::steel::BlockLoader::bj' ] ] ] ,
2024-06-07 11:28:06 +08:00
[ 'block_5fmasked_5fgemm_66' , [ 'block_masked_gemm' , [ '../steel__gemm__masked_8h.html#af805e998b2046ee30c2b4be813e3af97' , 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 size_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#a477932e2ae9d49366f7ede6db63f9cac' , 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 size_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_67' , [ 'block_masked_mm' , [ '../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6' , 1 , 'mlx::core' ] ] ] ,
[ 'block_5fmerge_5fsort_5ft_68' , [ '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' ] ] ] ,
2024-07-12 06:32:08 +08:00
[ 'block_5fsort_69' , [ '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_70' , [ 'block_sort_nc' , [ '../sort_8h.html#a29229399f51e5c440ffe5c9b99b27598' , 1 , 'sort.h' ] ] ] ,
[ 'blockloader_71' , [ '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()' ] ] ] ,
[ 'blockmaskedmm_72' , [ '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_73' , [ 'BlockMergeSort' , [ '../struct_block_merge_sort.html' , 1 , '' ] ] ] ,
[ 'blockmma_74' , [ '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()' ] ] ] ,
[ 'blockswizzle_75' , [ 'BlockSwizzle' , [ '../structmlx_1_1steel_1_1_block_swizzle.html' , 1 , 'mlx::steel' ] ] ] ,
[ 'bluestein_5ffft_76' , [ 'bluestein_fft' , [ '../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e' , 1 , 'fft.h' ] ] ] ,
[ 'bool4_5for_5fuint_77' , [ 'bool4_or_uint' , [ '../unionbool4__or__uint.html' , 1 , '' ] ] ] ,
[ 'bool_5f_78' , [ 'bool_' , [ '../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa467afb5838aa377d55cce81f84c5512b' , 1 , 'mlx::core::Dtype::bool_' ] , [ '../namespacemlx_1_1core.html#a113d2bac7e4aa6a4cb4a5c3242527b82' , 1 , 'mlx::core::bool_' ] ] ] ,
[ 'broadcast_79' , [ 'Broadcast' , [ '../classmlx_1_1core_1_1_broadcast.html' , 1 , 'mlx::core::Broadcast' ] , [ '../classmlx_1_1core_1_1_broadcast.html#accbab8433c93e281608a268d11afaefb' , 1 , 'mlx::core::Broadcast::Broadcast()' ] ] ] ,
[ 'broadcast_5farrays_80' , [ 'broadcast_arrays' , [ '../group__ops.html#gab783890428b596f715dc7dd2057eae99' , 1 , 'mlx::core' ] ] ] ,
[ 'broadcast_5fshapes_81' , [ 'broadcast_shapes' , [ '../namespacemlx_1_1core.html#a075e07def338cd9d815182d0e6a656c0' , 1 , 'mlx::core' ] ] ] ,
[ 'broadcast_5fto_82' , [ 'broadcast_to' , [ '../group__ops.html#gad256e86cc1a6e6b3832e392baa90318d' , 1 , 'mlx::core' ] ] ] ,
[ 'brows_83' , [ '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_84' , [ 'Bs_offset' , [ '../structmlx_1_1steel_1_1_block_m_m_a.html#a92f6aeee432f53638447eac842f43eca' , 1 , 'mlx::steel::BlockMMA' ] ] ] ,
[ 'bs_5fqmm_5fn_85' , [ 'bs_qmm_n' , [ '../quantized_8h.html#acfc43edcf0c742ddc090d22cb7229284' , 1 , 'quantized.h' ] ] ] ,
[ 'bs_5fqmm_5ft_86' , [ 'bs_qmm_t' , [ '../quantized_8h.html#a3acc2ace766cb855f13da2d1834e7dc7' , 1 , 'quantized.h' ] ] ] ,
[ 'bs_5fqmv_87' , [ 'bs_qmv' , [ '../quantized_8h.html#a03ba4a4a5fe9955319b0aa477d2d7d98' , 1 , 'quantized.h' ] ] ] ,
[ 'bs_5fqmv_5ffast_88' , [ 'bs_qmv_fast' , [ '../quantized_8h.html#a6f92bb9f3d29d707bfc680bebe1c80f7' , 1 , 'quantized.h' ] ] ] ,
[ 'bs_5fqvm_89' , [ 'bs_qvm' , [ '../quantized_8h.html#a3af1c89416632c9275b8218a33cb8a04' , 1 , 'quantized.h' ] ] ] ,
[ 'bsimd_90' , [ 'Bsimd' , [ '../structmlx_1_1steel_1_1_block_m_m_a.html#a3de506004b428d3487bae05ad9e757e6' , 1 , 'mlx::steel::BlockMMA' ] ] ] ,
[ 'buf_91' , [ 'buf' , [ '../struct_read_writer.html#a23bac3c96dd0265ddbee1f256be45ff5' , 1 , 'ReadWriter::buf' ] , [ '../backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697' , 1 , 'buf: allocator.h' ] ] ] ,
[ 'buffer_92' , [ 'Buffer' , [ '../classmlx_1_1core_1_1allocator_1_1_buffer.html' , 1 , 'mlx::core::allocator::Buffer' ] , [ '../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300' , 1 , 'mlx::core::allocator::Buffer::Buffer()' ] ] ] ,
[ 'buffer_93' , [ 'buffer' , [ '../structmlx_1_1core_1_1array_1_1_data.html#a9a51e2d12ba505027cc0fca86bdd39ad' , 1 , 'mlx::core::array::Data::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' ] ] ] ,
[ 'buffers_94' , [ 'buffers' , [ '../struct_indices.html#ad705070a740579c07d109ae4f3d86e76' , 1 , 'Indices' ] ] ] ,
[ 'build_5flib_5fname_95' , [ 'build_lib_name' , [ '../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b' , 1 , 'mlx::core' ] ] ]
2024-06-07 11:28:06 +08:00
] ;