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)
bool	mlx::core::detail::in_tracing ()
	Return true if we are currently performing a function transformation in order to keep the graph when evaluating tracer arrays.
bool	mlx::core::detail::in_dynamic_tracing ()
	Return true if we are in a dynamic (shapeless) trace used for compiling or exporting graphs with dynamic shapes.
bool	mlx::core::detail::retain_graph ()