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

Go to the source code of this file.

Functions

template<typename T , typename Op >
void ternary_v (device const bool *a, device const T *b, device const T *c, device T *d, uint index)
 
template<typename T , typename Op >
void ternary_g_nd1 (device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t &a_strides, constant const size_t &b_strides, constant const size_t &c_strides, uint index)
 
template<typename T , typename Op >
void ternary_g_nd2 (device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t a_strides[2], constant const size_t b_strides[2], constant const size_t c_strides[2], uint2 index, uint2 grid_dim)
 
template<typename T , typename Op >
void ternary_g_nd3 (device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t a_strides[3], constant const size_t b_strides[3], constant const size_t c_strides[3], uint3 index, uint3 grid_dim)
 
template<typename T , typename Op , int DIM>
void ternary_g_nd (device const bool *a, device const T *b, device const T *c, device T *d, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], constant const size_t c_strides[DIM], uint3 index, uint3 grid_dim)
 
template<typename T , typename Op >
void ternary_g (device const bool *a, device const T *b, device const T *c, device T *d, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const size_t *c_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
 

Function Documentation

◆ ternary_g()

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

◆ ternary_g_nd()

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

◆ ternary_g_nd1()

template<typename T , typename Op >
void ternary_g_nd1 ( device const bool * a,
device const T * b,
device const T * c,
device T * d,
constant const size_t & a_strides,
constant const size_t & b_strides,
constant const size_t & c_strides,
uint index )

◆ ternary_g_nd2()

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

◆ ternary_g_nd3()

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

◆ ternary_v()

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