loader_channel_n.h File Reference

#include "mlx/backend/metal/kernels/steel/utils.h"
#include "mlx/backend/metal/kernels/steel/conv/params.h"

Go to the source code of this file.

Classes
struct	mlx::steel::ChannelHelper< n_channels_ >
struct	mlx::steel::ChannelHelper< 1 >
struct	mlx::steel::ChannelHelper< 2 >
struct	mlx::steel::ChannelHelper< 3 >
struct	mlx::steel::ChannelHelper< 4 >
struct	mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >
struct	mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >

Namespaces
namespace	mlx
namespace	mlx::steel