载入中...
搜索中...
未找到
命名空间 | 函数
CpuKernels.h 文件参考
#include "tensor/Graph.h"
#include <cstddef>

浏览源代码.

命名空间

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)