std1.0.0-beta-7 SafeC docs

std :: ml::gpu_miopen

Function std::rocm_concat_channels

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

int rocm_concat_channels(
    struct RocmBuf a,
    unsigned long aChannels,
    struct RocmBuf c,
    unsigned long cChannels,
    struct RocmBuf out,
    unsigned long batch,
    unsigned long plane,
)

out[b] = (a[b], c[b]) along channels, for each of 'batch' images whose planes are 'plane' = height*width elements (concat_channels).