export.h File Reference

#include <map>
#include <set>
#include "mlx/array.h"
#include "mlx/export_impl.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core

Typedefs
using	mlx::core::Args = std::vector<array>
using	mlx::core::Kwargs = std::map<std::string, array>

Functions
FunctionExporter	mlx::core::exporter (const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, bool shapeless=false)
	Make an exporter to save multiple traces of a given function to the same file.
FunctionExporter	mlx::core::exporter (const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, bool shapeless=false)
FunctionExporter	mlx::core::exporter (const std::string &path, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, bool shapeless=false)
void	mlx::core::export_function (const std::string &file, const std::function< std::vector< array >(const Args &)> &fun, const Args &args, bool shapeless=false)
	Export a function to a file.
void	mlx::core::export_function (const std::string &file, const std::function< std::vector< array >(const Kwargs &)> &fun, const Kwargs &kwargs, bool shapeless=false)
void	mlx::core::export_function (const std::string &file, const std::function< std::vector< array >(const Args &, const Kwargs &)> &fun, const Args &args, const Kwargs &kwargs, bool shapeless=false)
ImportedFunction	mlx::core::import_function (const std::string &file)
	Import a function from a file.