std1.0.0-beta-7 SafeC docs

std :: ml::gpu_rocm

Function std::rocm_mp_adam

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

int rocm_mp_adam(
    struct RocmBuf w,
    struct RocmBuf m,
    struct RocmBuf v,
    struct RocmBuf g,
    struct RocmHalfBuf wHalf,
    float lr,
    float beta1,
    float beta2,
    float eps,
    float weightDecay,
    int decoupled,
    unsigned long step,
    struct RocmBuf skip,
)