mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
77 lines
14 KiB
JavaScript
77 lines
14 KiB
JavaScript
var searchData=
|
|
[
|
|
['r2c_0',['r2c',['../namespacepocketfft_1_1detail.html#a4e46762466d399e35b79c324cfe21616',1,'pocketfft::detail::r2c(const shape_t &shape_in, const stride_t &stride_in, const stride_t &stride_out, size_t axis, bool forward, const T *data_in, std::complex< T > *data_out, T fct, size_t nthreads=1)'],['../namespacepocketfft_1_1detail.html#a454179497c44714d4b7425f116468c17',1,'pocketfft::detail::r2c(const shape_t &shape_in, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, bool forward, const T *data_in, std::complex< T > *data_out, T fct, size_t nthreads=1)']]],
|
|
['r2r_5ffftpack_1',['r2r_fftpack',['../namespacepocketfft_1_1detail.html#a1ccca4cbbc6150d65620e2f9cdff62ac',1,'pocketfft::detail']]],
|
|
['r2r_5fgenuine_5fhartley_2',['r2r_genuine_hartley',['../namespacepocketfft_1_1detail.html#a072a67f2c4b3b3ebd030604f3383e1ed',1,'pocketfft::detail']]],
|
|
['r2r_5fseparable_5fhartley_3',['r2r_separable_hartley',['../namespacepocketfft_1_1detail.html#a40b56ae882eea17e6960618415f8eb42',1,'pocketfft::detail']]],
|
|
['rader_5ffft_4',['rader_fft',['../backend_2metal_2kernels_2fft_8h.html#ad123452303f6415904f3592f660769b8',1,'fft.h']]],
|
|
['radians_5',['radians',['../group__ops.html#ga07aa8059adba5b9a8818027b8aafd31e',1,'mlx::core']]],
|
|
['radix10_6',['radix10',['../radix_8h.html#a27054992e148bfcd9f7a69c96d75d926',1,'radix.h']]],
|
|
['radix11_7',['radix11',['../radix_8h.html#a34f7dcefd29f8a7f6bddf9a17609a8a7',1,'radix.h']]],
|
|
['radix12_8',['radix12',['../radix_8h.html#a3c61aa04a4b38575722d864cc9ffd0b7',1,'radix.h']]],
|
|
['radix13_9',['radix13',['../radix_8h.html#a833b7b31facbd62f6971bb108c54d1db',1,'radix.h']]],
|
|
['radix2_10',['radix2',['../radix_8h.html#a745e50244f1470bba233d8b6b993729c',1,'radix.h']]],
|
|
['radix3_11',['radix3',['../radix_8h.html#a497b31cfa28418e1c2fad53d09f7bc0b',1,'radix.h']]],
|
|
['radix4_12',['radix4',['../radix_8h.html#a12cb26bd3ad635d16a195ccea750256d',1,'radix.h']]],
|
|
['radix5_13',['radix5',['../radix_8h.html#a026e6779e6d2ecdef39ff4aad186091e',1,'radix.h']]],
|
|
['radix6_14',['radix6',['../radix_8h.html#afaaa5de58a97f0a5e6a84fc0d598a884',1,'radix.h']]],
|
|
['radix7_15',['radix7',['../radix_8h.html#a6d37e483d806c550deefabf56caf4ec8',1,'radix.h']]],
|
|
['radix8_16',['radix8',['../radix_8h.html#a6b306e1afe1c5b6af4cd80aa02adb138',1,'radix.h']]],
|
|
['radix_5fbutterfly_17',['radix_butterfly',['../backend_2metal_2kernels_2fft_8h.html#a278d980ed397e6841ce0af44b9aa4396',1,'fft.h']]],
|
|
['radix_5ffunc_18',['radix_func',['../metal_2kernels_2hadamard_8h.html#a590e5366adc78bab4fe44e37885d413f',1,'hadamard.h']]],
|
|
['radix_5fn_5fsteps_19',['radix_n_steps',['../backend_2metal_2kernels_2fft_8h.html#aba8d32e0911499671df93678f39da08b',1,'fft.h']]],
|
|
['randint_20',['randint',['../namespacemlx_1_1core_1_1random.html#a7289d3791aa4e8bfe28fafef4fb40e3b',1,'mlx::core::random::randint(const array &low, const array &high, const std::vector< int > &shape, Dtype dtype=int32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#afe20b1c6188f4c8a83203e60b70b8946',1,'mlx::core::random::randint(T low, U high, const std::vector< int > &shape, Dtype dtype=int32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]],
|
|
['randombits_21',['RandomBits',['../classmlx_1_1core_1_1_random_bits.html#a7dd5ed03f2a4ab45d1d5e8e2b587de6b',1,'mlx::core::RandomBits']]],
|
|
['rank_22',['rank',['../structmlx_1_1core_1_1distributed_1_1_group.html#ad4a5cdab8e2209ad4825f0c63a9cbec7',1,'mlx::core::distributed::Group']]],
|
|
['raw_5fevent_23',['raw_event',['../classmlx_1_1core_1_1_event.html#af408d30df17c4771e9e2aa550cb6e921',1,'mlx::core::Event']]],
|
|
['raw_5fgroup_24',['raw_group',['../structmlx_1_1core_1_1distributed_1_1_group.html#a3285b94851cf145a2a472f60b7f7ca36',1,'mlx::core::distributed::Group']]],
|
|
['raw_5fptr_25',['raw_ptr',['../classmlx_1_1core_1_1allocator_1_1_buffer.html#a2dfe63e0b4bffeb965cdc50ad4228dbc',1,'mlx::core::allocator::Buffer']]],
|
|
['read_26',['read',['../classmlx_1_1core_1_1io_1_1_reader.html#ad8d74e2c62b579511089faa4cc6f50a1',1,'mlx::core::io::Reader::read()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a6a5fa6067fea53ba7dbcddcb7d713a7d',1,'mlx::core::io::FileReader::read()']]],
|
|
['readwriter_27',['ReadWriter',['../struct_read_writer.html#a1aa07e41d7ac286ad79bd26a072dfa0c',1,'ReadWriter']]],
|
|
['reciprocal_28',['reciprocal',['../group__ops.html#ga4d29556bb93e2f66916116cf1f062b36',1,'mlx::core']]],
|
|
['reduce_29',['Reduce',['../classmlx_1_1core_1_1_reduce.html#a055368c1d036fb953a23ef230e33dcbf',1,'mlx::core::Reduce']]],
|
|
['reduce_30',['reduce',['../namespacemlx_1_1core_1_1metal.html#abb997ccbed4c9a9ccd975b1574755fca',1,'mlx::core::metal']]],
|
|
['reduce_5futils_31',['reduce_utils',['../namespacemlx_1_1core_1_1metal.html#a2ec39572806310cf528aea06530e8af8',1,'mlx::core::metal']]],
|
|
['reduction_5fop_32',['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)']]],
|
|
['reductionplan_33',['ReductionPlan',['../structmlx_1_1core_1_1_reduction_plan.html#a3df6acc08ef92647aab19d68b0961610',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_, std::vector< int > shape_, std::vector< size_t > strides_)'],['../structmlx_1_1core_1_1_reduction_plan.html#aec7496f3740a0b0d51aaa606f6fd68f4',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_)']]],
|
|
['register_5flibrary_34',['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#a6993ce23a485b725b687628b77751ad2',1,'mlx::core::metal::Device::register_library(const std::string &lib_name, const std::function< std::string(const std::string &)> &lib_path_func=get_colocated_mtllib_path)']]],
|
|
['remainder_35',['Remainder',['../classmlx_1_1core_1_1_remainder.html#a4f3eada4a21898af4a77d1d27ce14641',1,'mlx::core::Remainder']]],
|
|
['remainder_36',['remainder',['../group__ops.html#ga99f5c904f724156a814d7817188351d2',1,'mlx::core']]],
|
|
['remaining_37',['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_38',['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_5fpeak_5fmemory_39',['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_40',['Reshape',['../classmlx_1_1core_1_1_reshape.html#aa7c5a6e18d0615ad36102de01929eb26',1,'mlx::core::Reshape']]],
|
|
['reshape_41',['reshape',['../group__ops.html#ga7b0e3d46a39d99b1628b422049790082',1,'mlx::core']]],
|
|
['resize_42',['resize',['../classpocketfft_1_1detail_1_1arr.html#a8d73baaefa02dff8714e4398c83917e0',1,'pocketfft::detail::arr']]],
|
|
['restart_43',['restart',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288',1,'pocketfft::detail::threading::thread_pool']]],
|
|
['result_5ftype_44',['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_45',['retain_graph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a12ead93cb70ebab865c5e9ce7718f814',1,'mlx::core::detail::RetainGraph']]],
|
|
['retaingraph_46',['RetainGraph',['../structmlx_1_1core_1_1detail_1_1_retain_graph.html#a7fac0244c14cc9e8f580bc1298ff68da',1,'mlx::core::detail::RetainGraph']]],
|
|
['rev_5fiter_47',['rev_iter',['../classpocketfft_1_1detail_1_1rev__iter.html#af7b8c2f1534d3038ba2a3c6b9919e134',1,'pocketfft::detail::rev_iter']]],
|
|
['rev_5fofs_48',['rev_ofs',['../classpocketfft_1_1detail_1_1rev__iter.html#a7f112afa76cb7a4c29cff217a6f5f5a9',1,'pocketfft::detail::rev_iter']]],
|
|
['rfft_49',['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_50',['rfft2',['../namespacemlx_1_1core_1_1fft.html#af63a8d42005954aa5cba69823e9c9de7',1,'mlx::core::fft::rfft2(const array &a, const std::vector< int > &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_51',['rfftn',['../namespacemlx_1_1core_1_1fft.html#a2e4287e110dfb23aaa7f4d65eb72465a',1,'mlx::core::fft::rfftn(const array &a, const std::vector< int > &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_52',['rfftp',['../classpocketfft_1_1detail_1_1rfftp.html#a0c590f917b8e8afa3ff53ccff52e68c5',1,'pocketfft::detail::rfftp']]],
|
|
['right_5fshift_53',['right_shift',['../group__ops.html#gafa376ad57d38ba87378f0272dc379b23',1,'mlx::core']]],
|
|
['rint_54',['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_55',['rms_norm',['../namespacemlx_1_1core_1_1fast.html#ac7b620275c6386f822b7aacc6b312e62',1,'mlx::core::fast']]],
|
|
['rmsnorm_56',['RMSNorm',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a22adaff0749711263388ec151fcfebe2',1,'mlx::core::fast::RMSNorm']]],
|
|
['rmsnormvjp_57',['RMSNormVJP',['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#aac060129b2e1af79bf388bfe705381ca',1,'mlx::core::fast::RMSNormVJP']]],
|
|
['rope_58',['RoPE',['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a031fa27aeba94cfa5ccc633137a12163',1,'mlx::core::fast::RoPE']]],
|
|
['rope_59',['rope',['../namespacemlx_1_1core_1_1fast.html#a44a77225d1e90fd5bf81caccf8c20061',1,'mlx::core::fast']]],
|
|
['rot90_60',['ROT90',['../namespacepocketfft_1_1detail.html#a928bad5278df636ee47402c0a75f64ef',1,'pocketfft::detail']]],
|
|
['rotx90_61',['ROTX90',['../namespacepocketfft_1_1detail.html#ab6a43dc0cec4291e163e68a0875ac501',1,'pocketfft::detail']]],
|
|
['round_62',['Round',['../classmlx_1_1core_1_1_round.html#a1327a359b2aed91f576145a0e70d1dde',1,'mlx::core::Round']]],
|
|
['round_63',['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_64',['round_error',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#afa223448fa4f04c1113a85345dd720c3',1,'metal::_numeric_limits_impl< bfloat16_t >']]],
|
|
['row_5freduce_5fgeneral_65',['row_reduce_general',['../reduce__row_8h.html#a917b01e3de3372d276f8dd28f0e9b8f1',1,'reduce_row.h']]],
|
|
['row_5freduce_5fgeneral_5fdispatch_66',['row_reduce_general_dispatch',['../namespacemlx_1_1core.html#ab1eeca8ec6fa31819ee108fa6ed2c41b',1,'mlx::core']]],
|
|
['row_5freduce_5fgeneral_5fmed_67',['row_reduce_general_med',['../reduce__row_8h.html#ad77d444bcfa9e6afcc1ffc1dcb919f1c',1,'reduce_row.h']]],
|
|
['row_5freduce_5fgeneral_5fno_5fatomics_68',['row_reduce_general_no_atomics',['../reduce__row_8h.html#a034fdccc83e66871550c2d42399f0ab5',1,'reduce_row.h']]],
|
|
['row_5freduce_5fgeneral_5fsmall_69',['row_reduce_general_small',['../reduce__row_8h.html#aea2b1ba640fffb71c73550ec747b3148',1,'reduce_row.h']]],
|
|
['rows_70',['rows',['../class_m_p_s_1_1_matrix_descriptor.html#a04ce0dd1d4632ff8c4328e4b62d71fc0',1,'MPS::MatrixDescriptor']]],
|
|
['rsqrt_71',['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_72',['run',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel']]]
|
|
];
|