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

浏览源代码.

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

namespace	eve::tensor

namespace	eve::tensor::wgsl_detail

函数
void	eve::tensor::wgsl_detail::genSoftmax (const Graph &, const FusedGroup &, KernelSpec &)
	Gen softmax.

void	eve::tensor::wgsl_detail::genNorm (const Graph &, const FusedGroup &, bool rms, KernelSpec &)
	Gen norm.

void	eve::tensor::wgsl_detail::genReduceOrArgmax (const Graph &, const FusedGroup &, bool argmax, KernelSpec &)
	Gen reduce or argmax.

命名空间

函数