Current section

Files

Jump to
viva_tensor include viva_tensor@quant@qat_QatLinear.hrl
Raw

include/viva_tensor@quant@qat_QatLinear.hrl

-record(qat_linear, {
weight :: viva_tensor@tensor:tensor(),
bias :: gleam@option:option(viva_tensor@tensor:tensor()),
weight_stats :: viva_tensor@quant@qat:quant_stats(),
weight_config :: viva_tensor@quant@qat:quant_config(),
input_stats :: gleam@option:option(viva_tensor@quant@qat:quant_stats()),
input_config :: viva_tensor@quant@qat:quant_config()
}).