binary.h File Reference

#include <cassert>
#include "mlx/allocator.h"
#include "mlx/array.h"
#include "mlx/backend/common/binary.h"
#include "mlx/backend/common/utils.h"
#include "mlx/backend/cpu/simd/simd.h"

Go to the source code of this file.

Classes
struct	mlx::core::VectorScalar< Op >
struct	mlx::core::ScalarVector< Op >
struct	mlx::core::VectorVector< Op >

Namespaces
namespace	mlx
namespace	mlx::core

Functions
template<typename T, typename U, typename Op, int D, bool Strided>
void	mlx::core::binary_op_dims (const T *a, const T *b, U *out, Op op, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &out_strides, int axis)
template<typename T, typename U, bool Strided, typename Op>
void	mlx::core::binary_op_dispatch_dims (const array &a, const array &b, array &out, Op op, int dim, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &out_strides)
template<typename T, typename U, typename Op>
void	mlx::core::binary_op (const array &a, const array &b, array &out, Op op)
template<typename T, typename Op>
void	mlx::core::binary_op (const array &a, const array &b, array &out, Op op)
template<typename Op>
void	mlx::core::binary (const array &a, const array &b, array &out, Op op)