载入中...
搜索中...
未找到
OnnxGpuKernels.cpp 文件参考
#include "tensor/OnnxGpuKernels.h"
#include <sstream>
#include "tensor/OnnxInternal.h"

浏览源代码.

命名空间

namespace  eve
 Build metadata (engine git commit, build time, third-party version).
 
namespace  eve::tensor
 
namespace  eve::tensor::onnx_detail
 

函数

std::vector< int32_t > eve::tensor::onnx_detail::gpuMatmul (OnnxCompute &device, affine::ByteView a, affine::ByteView b, size_t m, size_t k, size_t n, int az, std::span< const int32_t > bz)
 Gpu matmul.
 
std::vector< int32_t > eve::tensor::onnx_detail::gpuConv (OnnxCompute &device, affine::ByteView x, affine::ByteView w, const affine::ConvShape &s, int xz, std::span< const int32_t > wz)
 Gpu conv.
 
OnnxBuffer eve::tensor::onnx_detail::enqueueInteger (OnnxCompute &d, const std::string &source, OnnxBuffer a, OnnxBuffer b, std::span< const int32_t > z, size_t size, size_t terms)
 
OnnxBuffer eve::tensor::onnx_detail::gpuMatmulResident (OnnxCompute &, OnnxBuffer, OnnxBuffer, bool, bool, size_t, size_t, size_t, int, std::span< const int32_t >, size_t bOffset=0)
 Gpu matmul resident.
 
OnnxBuffer eve::tensor::onnx_detail::gpuConvResident (OnnxCompute &, OnnxBuffer, OnnxBuffer, bool, bool, const affine::ConvShape &, int, std::span< const int32_t >)
 Gpu conv resident.