std1.0.0-beta-7 SafeC docs

std :: ml::gpu_rocm

Function std::rocm_concat

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

int rocm_concat(
    struct RocmBuf a,
    unsigned long na,
    struct RocmBuf c,
    unsigned long nc,
    struct RocmBuf out,
    unsigned long batch,
)

out[b] = (a[b], c[b]) for 'batch' rows of na and nc elements, one launch (the channel concat of NCHW tensors: na = aChannels * height * width).