MLX
 
Loading...
Searching...
No Matches
binary.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::BinaryOpType {
  mlx::core::ScalarScalar , mlx::core::ScalarVector , mlx::core::VectorScalar , mlx::core::VectorVector ,
  mlx::core::General
}
 

Functions

BinaryOpType mlx::core::get_binary_op_type (const array &a, const array &b)
 
void mlx::core::set_binary_op_output_data (const array &a, const array &b, array &out, BinaryOpType bopt, bool donate_with_move=false)