std1.0.0-beta-7 SafeC docs

std :: ml::gpu_cuda

Function std::cuda_half_bias

#include <std/ml/gpu_cuda.h> source ยท definition

int cuda_half_bias(
    struct CudaHalfBuf y,
    struct CudaHalfBuf bias,
    unsigned long rows,
    unsigned long channels,
    unsigned long plane,
    int relu,
)

y[r][c][p] += bias[c] (rows x channels x plane, row-major), then ReLU when relu: a linear layer's bias (plane 1) or an NCHW convolution's.