ternary.h File Reference

#include "mlx/allocator.h"
#include "mlx/array.h"
#include "mlx/backend/common/ternary.h"
#include "mlx/backend/common/utils.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core

Functions
template<typename T1, typename T2, typename T3, typename U, typename Op, int D>
void	mlx::core::ternary_op_dims (const T1 *a, const T2 *b, const T3 *c, U *out, Op op, const Shape &shape, const Strides &a_strides, const Strides &b_strides, const Strides &c_strides, const Strides &out_strides, int axis)
template<typename T1, typename T2, typename T3, typename U, typename Op>
void	mlx::core::ternary_op_dispatch_dims (const array &a, const array &b, const array &c, array &out, Op op)
template<typename T1, typename T2, typename T3, typename U, typename Op>
void	mlx::core::ternary_op (const array &a, const array &b, const array &c, array &out, Op op)