#include "common/Exception.h"
#include "tensor/CpuKernels.h"
#include "tensor/KernelGenWgslInternal.h"
#include "tensor/Quant.h"
#include <algorithm>
#include <cmath>
#include <sstream>

浏览源代码.

类
struct	eve::tensor::wgsl_detail::ChainContext

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

namespace	eve::tensor

namespace	eve::tensor::wgsl_detail

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

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

std::string	eve::tensor::wgsl_detail::bufferDeclUint (int binding, const char *name)
	Buffer decl uint.

std::string	eve::tensor::wgsl_detail::emitQuantizedBVal (DType dtype, int group, const char *bufferName)
	Emit quantized b val.

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

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

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

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

void	eve::tensor::wgsl_detail::genElementwise (const Graph &g, const FusedGroup &grp, KernelSpec &out)

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

void	eve::tensor::wgsl_detail::genConv (const Graph &g, const FusedGroup &grp, KernelSpec &out)

void	eve::tensor::wgsl_detail::genPool (const Graph &g, const FusedGroup &grp, KernelSpec &out)

void	eve::tensor::generateKernelWgslImpl (const Graph &graph, const FusedGroup &group, KernelSpec &out)

Result< KernelSpec >	eve::tensor::generateWgslKernel (const Graph &graph, const FusedGroup &group, KernelVariant variant=KernelVariant::Default)
	Lower an optimizer-produced group directly to owning WGSL source and dispatch metadata.

类

命名空间

函数