载入中...
搜索中...
未找到
命名空间 | |
| 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. | |