Function std::cuda_relu_grad

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

int cuda_relu_grad(struct CudaBuf z, struct CudaBuf g, struct CudaBuf out)

out = z > 0 ? g : 0. 'z' may be the ReLU input or its output (same sign test).