std1.0.0-beta-7 SafeC docs

std :: ml::gpu_cudnn

Function std::conv2d_forward_cuda

#include <std/ml/gpu_cudnn.h> source · definition

struct FeatureMap conv2d_forward_cuda(const &Conv2D layer, const &FeatureMap input)

Same signatures and results as the CPU functions; each falls back to the CPU implementation if there is no usable GPU. Every call uploads the weights — inside a loop, upload once and use the queued ops above.