MLX
Loading...
Searching...
No Matches
Namespaces | Functions
ternary.h File Reference
#include "mlx/array.h"

Go to the source code of this file.

Namespaces

namespace  mlx
 
namespace  mlx::core
 

Functions

void mlx::core::ternary_op_gpu (const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)
 
void mlx::core::ternary_op_gpu_inplace (const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)