graph_utils.h File Reference

#include <unordered_map>
#include "mlx/array.h"

Go to the source code of this file.

Classes
struct	mlx::core::NodeNamer

Namespaces
namespace	mlx
namespace	mlx::core

Functions
void	mlx::core::print_graph (std::ostream &os, NodeNamer namer, const std::vector< array > &outputs)
void	mlx::core::print_graph (std::ostream &os, const std::vector< array > &outputs)
template<typename... Arrays, typename = enable_for_arrays_t<Arrays...>>
void	mlx::core::print_graph (std::ostream &os, Arrays &&... outputs)
template<typename... Arrays, typename = enable_for_arrays_t<Arrays...>>
void	mlx::core::print_graph (std::ostream &os, NodeNamer namer, Arrays &&... outputs)
void	mlx::core::export_to_dot (std::ostream &os, NodeNamer namer, const std::vector< array > &outputs)
void	mlx::core::export_to_dot (std::ostream &os, const std::vector< array > &outputs)
template<typename... Arrays, typename = enable_for_arrays_t<Arrays...>>
void	mlx::core::export_to_dot (std::ostream &os, Arrays &&... outputs)
template<typename... Arrays, typename = enable_for_arrays_t<Arrays...>>
void	mlx::core::export_to_dot (std::ostream &os, NodeNamer namer, Arrays &&... outputs)