looped_elem_to_loc< dim, offset_t > Struct Template Reference

#include <utils.h>

Public Member Functions
void	next (const constant int *shape, const constant size_t *strides)
void	next (int n, const constant int *shape, const constant size_t *strides)
offset_t	location (offset_t, const constant int *, const constant size_t *, int)

Public Attributes
looped_elem_to_loc< dim - 1, offset_t >	inner_looper
offset_t	offset {0}
int	index {0}

Member Function Documentation

location()

template<int dim, typename offset_t = size_t>

offset_t looped_elem_to_loc< dim, offset_t >::location ( offset_t , const constant int * , const constant size_t * , int  )

inline

next() [1/2]

template<int dim, typename offset_t = size_t>

void looped_elem_to_loc< dim, offset_t >::next ( const constant int * shape, const constant size_t * strides )

inline

next() [2/2]

template<int dim, typename offset_t = size_t>

void looped_elem_to_loc< dim, offset_t >::next ( int n, const constant int * shape, const constant size_t * strides )

inline

Member Data Documentation

index

template<int dim, typename offset_t = size_t>

int looped_elem_to_loc< dim, offset_t >::index {0}

inner_looper

template<int dim, typename offset_t = size_t>

looped_elem_to_loc<dim - 1, offset_t> looped_elem_to_loc< dim, offset_t >::inner_looper

offset

template<int dim, typename offset_t = size_t>

offset_t looped_elem_to_loc< dim, offset_t >::offset {0}

---

The documentation for this struct was generated from the following file:

• mlx/backend/metal/kernels/utils.h