#include "tensor/CpuKernels.h"
#include "tensor/KernelGenInternal.h"
#include "tensor/Quant.h"
#include <algorithm>
#include <cmath>
#include <sstream>

浏览源代码.

类
struct	eve::tensor::glsl_detail::ChainContext

命名空间
namespace	eve
	Build metadata (engine git commit, build time, third-party version).

namespace	eve::tensor

namespace	eve::tensor::glsl_detail

函数
std::string	eve::tensor::glsl_detail::header (int localX, int localY=1)
	Header.

std::string	eve::tensor::glsl_detail::bufferDecl (int binding, const char *name)
	Buffer decl.

std::string	eve::tensor::glsl_detail::bufferDeclUint (int binding, const char *name)

std::string	eve::tensor::glsl_detail::emitQuantizedBVal (DType dt, int group, const char *bufName="b")

std::string	eve::tensor::glsl_detail::pushConstant ()
	Pushes constant.

int	eve::tensor::glsl_detail::groupsFor (int count)
	Groups for.

std::string	eve::tensor::glsl_detail::scalarStr (float value)
	Scalar str.

void	eve::tensor::glsl_detail::emitInputIndexExprs (std::ostringstream &os, const Graph &g, const FusedGroup &grp, std::vector< std::string > &indexExprs)

bool	eve::tensor::glsl_detail::genElementwise (const Graph &g, const FusedGroup &grp, KernelSpec &out)

bool	eve::tensor::glsl_detail::genMatMul (const Graph &g, const FusedGroup &grp, bool tiled, KernelSpec &out)

bool	eve::tensor::glsl_detail::genConv (const Graph &g, const FusedGroup &grp, KernelSpec &out)

bool	eve::tensor::glsl_detail::genPool (const Graph &g, const FusedGroup &grp, KernelSpec &out)

bool	eve::tensor::glsl_detail::genSoftmax (const Graph &g, const FusedGroup &grp, KernelSpec &out)

bool	eve::tensor::glsl_detail::genNorm (const Graph &g, const FusedGroup &grp, bool rms, KernelSpec &out)

bool	eve::tensor::glsl_detail::genReduceOrArgmax (const Graph &g, const FusedGroup &grp, bool argmax, KernelSpec &out)

bool	eve::tensor::glsl_detail::genEmbedding (const Graph &g, const FusedGroup &grp, KernelSpec &out)

bool	eve::tensor::glsl_detail::genConcat (const Graph &g, const FusedGroup &grp, KernelSpec &out)

bool	eve::tensor::glsl_detail::genSlice (const Graph &g, const FusedGroup &grp, KernelSpec &out)

bool	eve::tensor::glsl_detail::genPermute (const Graph &g, const FusedGroup &grp, KernelSpec &out)

bool	eve::tensor::glsl_detail::genSdpa (const Graph &g, const FusedGroup &grp, KernelSpec &out)

bool	eve::tensor::generateKernel (const Graph &graph, const FusedGroup &group, KernelSpec &out)
	Generate kernel.

bool	eve::tensor::generateMatMulVariant (const Graph &graph, const FusedGroup &group, bool tiled, KernelSpec &out)
	Generate mat mul variant.

类

命名空间

函数