MLX
Loading...
Searching...
No Matches
Functions
unary.h File Reference

Go to the source code of this file.

Functions

template<typename T , typename Op >
void unary_v (device const T *in, device T *out, uint index)
 
template<typename T , typename Op >
void unary_g (device const T *in, device T *out, device const int *in_shape, device const size_t *in_strides, device const int &ndim, uint index)
 

Function Documentation

◆ unary_g()

template<typename T , typename Op >
void unary_g ( device const T * in,
device T * out,
device const int * in_shape,
device const size_t * in_strides,
device const int & ndim,
uint index )

◆ unary_v()

template<typename T , typename Op >
void unary_v ( device const T * in,
device T * out,
uint index )