unary.h File Reference

#include "mlx/allocator.h"
#include "mlx/array.h"
#include "mlx/backend/common/utils.h"
#include "mlx/backend/cpu/encoder.h"
#include "mlx/backend/cpu/simd/simd.h"
#include "mlx/utils.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core

Functions
void	mlx::core::set_unary_output_data (const array &in, array &out)
template<typename T, typename U = T, typename Op>
void	mlx::core::unary_op (const T *a, U *out, size_t shape, size_t stride)
template<typename T, typename U = T, typename Op>
void	mlx::core::unary_op (const array &a, array &out, Op)
template<typename Op>
void	mlx::core::unary (const array &a, array &out, Op op, Stream stream)
template<typename Op>
void	mlx::core::unary_real_fp (const array &a, array &out, Op op, Stream stream)
template<typename Op>
void	mlx::core::unary_fp (const array &a, array &out, Op op, Stream stream)
template<typename Op>
void	mlx::core::unary_signed (const array &a, array &out, Op op, Stream stream)
template<typename Op>
void	mlx::core::unary_complex (const array &a, array &out, Op op, Stream stream)
template<typename Op>
void	mlx::core::unary_complex_to_float (const array &a, array &out, Op op, Stream stream)
template<typename Op>
void	mlx::core::unary_int (const array &a, array &out, Op op, Stream stream)