Template Function marian::gpu::gSanitizeGradient¶
Defined in File tensor_operators.cu
Function Documentation¶
-
template<typename T>__global__ void marian::gpu::gSanitizeGradient(T * in, int length, bool * isNaN, bool * isInf, bool pruneNaN, bool clipInf, float forNaN = 0.f, float forInf = 65504.f, float forInfNeg = -65504.f)