Current section

Files

Jump to
viva_tensor include viva_tensor@native@cuda_LinearLayer.hrl
Raw

include/viva_tensor@native@cuda_LinearLayer.hrl

-record(linear_layer, {
weight :: viva_tensor@native@cuda:accelerated_tensor(),
bias :: viva_tensor@native@cuda:accelerated_tensor(),
input_features :: integer(),
output_features :: integer(),
backend :: viva_tensor@native@cuda:acceleration_backend()
}).