LoopedElemToLoc< DIM, OffsetT, General > Struct Template Reference

#include <utils.h>

Public Member Functions
	LoopedElemToLoc (int dim)
void	next (const constant int *shape, const constant int64_t *strides)
void	next (int n, const constant int *shape, const constant int64_t *strides)
OffsetT	location ()

Public Attributes
int	dim
LoopedElemToLoc< DIM - 1, OffsetT, General >	inner_looper
OffsetT	offset {0}
int	index {0}

Constructor & Destructor Documentation

LoopedElemToLoc()

template<int DIM, typename OffsetT = size_t, bool General = true>

LoopedElemToLoc< DIM, OffsetT, General >::LoopedElemToLoc ( int dim )

inline

Member Function Documentation

location()

template<int DIM, typename OffsetT = size_t, bool General = true>

OffsetT LoopedElemToLoc< DIM, OffsetT, General >::location ( )

inline

next() [1/2]

template<int DIM, typename OffsetT = size_t, bool General = true>

void LoopedElemToLoc< DIM, OffsetT, General >::next ( const constant int * shape, const constant int64_t * strides )

inline

next() [2/2]

template<int DIM, typename OffsetT = size_t, bool General = true>

void LoopedElemToLoc< DIM, OffsetT, General >::next ( int n, const constant int * shape, const constant int64_t * strides )

inline

Member Data Documentation

dim

template<int DIM, typename OffsetT = size_t, bool General = true>

int LoopedElemToLoc< DIM, OffsetT, General >::dim

index

template<int DIM, typename OffsetT = size_t, bool General = true>

int LoopedElemToLoc< DIM, OffsetT, General >::index {0}

inner_looper

template<int DIM, typename OffsetT = size_t, bool General = true>

LoopedElemToLoc<DIM - 1, OffsetT, General> LoopedElemToLoc< DIM, OffsetT, General >::inner_looper

offset

template<int DIM, typename OffsetT = size_t, bool General = true>

OffsetT LoopedElemToLoc< DIM, OffsetT, General >::offset {0}

---

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

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