#include "tensor/CpuKernels.h"
#include "tensor/Tensor.h"
#include "common/Exception.h"
#include <algorithm>
#include <cmath>
#include <limits>
#include <vector>

浏览源代码.

命名空间
namespace	eve

namespace	eve::tensor

namespace	eve::tensor::kernels

函数
float	eve::tensor::kernels::applyUnary (OpType t, float x, float s0, float s1)

bool	eve::tensor::kernels::isElementwiseOp (OpType t)

bool	eve::tensor::kernels::broadcastShape (const int aDims, int aRank, const int bDims, int bRank, int *outDims, int &outRank)

void	eve::tensor::kernels::binaryOp (OpType type, const float a, const int aDims, int aRank, const float b, const int bDims, int bRank, float out, const int outDims, int outRank)

void	eve::tensor::kernels::unaryOp (OpType type, const float in, int count, float out, float s0, float s1)

void	eve::tensor::kernels::softmax (const float in, const int dims, int rank, int axis, bool logMode, float *out)

void	eve::tensor::kernels::layernorm (const float in, int rows, int cols, const float scale, const float bias, float eps, float out)

void	eve::tensor::kernels::rmsnorm (const float in, int rows, int cols, const float scale, float eps, float *out)

void	eve::tensor::kernels::conv1d (const float x, const int xDims, const float w, const int wDims, const float bias, int stride, int pad, float out)

void	eve::tensor::kernels::conv2d (const float x, const int xDims, const float w, const int wDims, const float bias, int stride, int pad, float out)

void	eve::tensor::kernels::maxpool2d (const float in, const int dims, int ksize, int stride, int pad, float *out)

void	eve::tensor::kernels::avgpool2d (const float in, const int dims, int ksize, int stride, int pad, float *out)

void	eve::tensor::kernels::embedding (const float table, int vocab, int dim, const float indices, int count, float *out)

void	eve::tensor::kernels::concat (const float const ins, const int const inDims, const int inRanks, int n, int axis, float out, const int *outDims, int outRank)

void	eve::tensor::kernels::sliceOp (const float in, const int inDims, int inRank, int axis, int begin, int end, float out, const int outDims, int outRank)

void	eve::tensor::kernels::permute (const float in, const int inDims, int rank, const int order, float out, const int *outDims)

void	eve::tensor::kernels::reduceAxis (OpType type, const float in, const int dims, int rank, int axis, float out, const int outDims, int outRank)

void	eve::tensor::kernels::argmax (const float in, const int dims, int rank, int axis, float out, const int outDims, int outRank)

void	eve::tensor::kernels::sdpa (const float q, const float k, const float v, const float mask, int B, int H, int T, int S, int D, float scale, float *out)

void	eve::tensor::kernels::resize2d (const float in, const int inDims, int outW, int outH, int mode, float *out)

命名空间

函数