Module ops 

ROCm-accelerated operations

Functions

batch_norm_rocm

Batch normalization using MIOpen

conv2d_rocm

Convolution using MIOpen

matmul_rocm

Matrix multiplication using rocBLAS

relu_rocm

Element-wise ReLU