ternary.h File Reference

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

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core

Enumerations
enum class	mlx::core::TernaryOpType { mlx::core::ScalarScalarScalar , mlx::core::VectorVectorVector , mlx::core::General }

Functions
TernaryOpType	mlx::core::get_ternary_op_type (const array &a, const array &b, const array &c)
void	mlx::core::set_ternary_op_output_data (const array &a, const array &b, const array &c, array &out, TernaryOpType topt, bool donate_with_move=false)