steel_conv.h File Reference

#include <metal_stdlib>

Go to the source code of this file.

Functions
template<typename T, int BM, int BN, int BK, int WM, int WN, int N_CHANNELS = 0, bool SMALL_FILTER = false>
void	implicit_gemm_conv_2d (const device T *A, const device T *B, device T *C, const constant MLXConvParams< 2 > *params, const constant ImplicitGemmConv2DParams *gemm_params, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)

Function Documentation

implicit_gemm_conv_2d()

template<typename T, int BM, int BN, int BK, int WM, int WN, int N_CHANNELS = 0, bool SMALL_FILTER = false>

void implicit_gemm_conv_2d	(	const device T *	A,
		const device T *	B,
		device T *	C,
		const constant MLXConvParams< 2 > *	params,
		const constant ImplicitGemmConv2DParams *	gemm_params,
		uint3	tid,
		uint3	lid,
		uint	simd_gid,
		uint	simd_lid )