Function std::rocm_relu_grad

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

int rocm_relu_grad(struct RocmBuf z, struct RocmBuf g, struct RocmBuf out)

out = z > 0 ? g : 0