transforms_impl.h File Reference

Go to the source code of this file.

Classes
struct	mlx::core::detail::InTracing
struct	mlx::core::detail::RetainGraph

Namespaces
namespace	mlx
namespace	mlx::core
namespace	mlx::core::detail

Functions
std::pair< std::vector< array >, std::vector< array > >	mlx::core::detail::vmap_trace (const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &inputs, const std::vector< int > &in_axes)
std::vector< array >	mlx::core::detail::vmap_replace (const std::vector< array > &inputs, const std::vector< array > &s_inputs, const std::vector< array > &s_outputs, const std::vector< int > &in_axes, const std::vector< int > &out_axes)
void	mlx::core::detail::compile_erase (std::uintptr_t fun_id)
void	mlx::core::detail::compile_clear_cache ()

Variables
std::function< std::vector< array >(const std::vector< array > &)	mlx::core::detail::compile )(const std::function< std::vector< array >(const std::vector< array > &)> &fun, std::uintptr_t fun_id, bool shapeless=false, std::vector< uint64_t > constants={})