MLXConvParams< NDIM > Struct Template Reference

#include <params.h>

Public Attributes
const int	N
const int	C
const int	O
const int	iS [NDIM]
const int	wS [NDIM]
const int	oS [NDIM]
const int	str [NDIM]
const int	pad [NDIM]
const int	kdil [NDIM]
const int	idil [NDIM]
const int64_t	in_strides [NDIM+2]
const int64_t	wt_strides [NDIM+2]
const int64_t	out_strides [NDIM+2]
const int	groups
const bool	flip

Member Data Documentation

C

template<int NDIM>

const int MLXConvParams< NDIM >::C

flip

template<int NDIM>

const bool MLXConvParams< NDIM >::flip

groups

template<int NDIM>

const int MLXConvParams< NDIM >::groups

idil

template<int NDIM>

const int MLXConvParams< NDIM >::idil[NDIM]

in_strides

template<int NDIM>

const int64_t MLXConvParams< NDIM >::in_strides[NDIM+2]

iS

template<int NDIM>

const int MLXConvParams< NDIM >::iS[NDIM]

kdil

template<int NDIM>

const int MLXConvParams< NDIM >::kdil[NDIM]

N

template<int NDIM>

const int MLXConvParams< NDIM >::N

O

template<int NDIM>

const int MLXConvParams< NDIM >::O

oS

template<int NDIM>

const int MLXConvParams< NDIM >::oS[NDIM]

out_strides

template<int NDIM>

const int64_t MLXConvParams< NDIM >::out_strides[NDIM+2]

pad

template<int NDIM>

const int MLXConvParams< NDIM >::pad[NDIM]

str

template<int NDIM>

const int MLXConvParams< NDIM >::str[NDIM]

wS

template<int NDIM>

const int MLXConvParams< NDIM >::wS[NDIM]

wt_strides

template<int NDIM>

const int64_t MLXConvParams< NDIM >::wt_strides[NDIM+2]

---

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

• mlx/backend/metal/kernels/steel/conv/params.h