GpuBackend.h 文件参考

#include <vector>

浏览源代码.

类
class	eve::tensor::GpuProgram
	GPU execution of a compiled tensor Graph via generated compute shaders. 更多...

命名空间
namespace	eve
namespace	eve::tensor

函数
bool	eve::tensor::gpuReduce (const float *data, int size, int op, float &outResult)
	GPU-accelerated reduction for large eager tensors. op: 0 = sum, 1 = min, 2 = max. Returns false (caller should fall back to CPU) when Vulkan/gpgpu isn't available.