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

Go to the source code of this file.

Functions

template<typename T , typename U , typename Op >
void binary_ss (device const T *a, device const T *b, device U *c, device U *d, uint index)
 
template<typename T , typename U , typename Op >
void binary_sv (device const T *a, device const T *b, device U *c, device U *d, uint index)
 
template<typename T , typename U , typename Op >
void binary_vs (device const T *a, device const T *b, device U *c, device U *d, uint index)
 
template<typename T , typename U , typename Op >
void binary_vv (device const T *a, device const T *b, device U *c, device U *d, uint index)
 
template<typename T , typename U , typename Op >
void binary_g_nd1 (device const T *a, device const T *b, device U *c, device U *d, constant const size_t &a_stride, constant const size_t &b_stride, uint index)
 
template<typename T , typename U , typename Op >
void binary_g_nd2 (device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim)
 
template<typename T , typename U , typename Op >
void binary_g_nd3 (device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim)
 
template<typename T , typename U , typename Op , int DIM>
void binary_g_nd (device const T *a, device const T *b, device U *c, device U *d, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], uint3 index, uint3 grid_dim)
 
template<typename T , typename U , typename Op >
void binary_g (device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
 

Function Documentation

◆ binary_g()

template<typename T , typename U , typename Op >
void binary_g ( device const T * a,
device const T * b,
device U * c,
device U * d,
constant const int * shape,
constant const size_t * a_strides,
constant const size_t * b_strides,
constant const int & ndim,
uint3 index,
uint3 grid_dim )

◆ binary_g_nd()

template<typename T , typename U , typename Op , int DIM>
void binary_g_nd ( device const T * a,
device const T * b,
device U * c,
device U * d,
constant const int shape[DIM],
constant const size_t a_strides[DIM],
constant const size_t b_strides[DIM],
uint3 index,
uint3 grid_dim )

◆ binary_g_nd1()

template<typename T , typename U , typename Op >
void binary_g_nd1 ( device const T * a,
device const T * b,
device U * c,
device U * d,
constant const size_t & a_stride,
constant const size_t & b_stride,
uint index )

◆ binary_g_nd2()

template<typename T , typename U , typename Op >
void binary_g_nd2 ( device const T * a,
device const T * b,
device U * c,
device U * d,
constant const size_t a_strides[2],
constant const size_t b_strides[2],
uint2 index,
uint2 grid_dim )

◆ binary_g_nd3()

template<typename T , typename U , typename Op >
void binary_g_nd3 ( device const T * a,
device const T * b,
device U * c,
device U * d,
constant const size_t a_strides[3],
constant const size_t b_strides[3],
uint3 index,
uint3 grid_dim )

◆ binary_ss()

template<typename T , typename U , typename Op >
void binary_ss ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint index )

◆ binary_sv()

template<typename T , typename U , typename Op >
void binary_sv ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint index )

◆ binary_vs()

template<typename T , typename U , typename Op >
void binary_vs ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint index )

◆ binary_vv()

template<typename T , typename U , typename Op >
void binary_vv ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint index )