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)