Packages

viva_tensor

2.2.103
2.2.104 2.2.103 2.2.102 2.2.100 2.1.0 retired 1.3.0 retired 1.2.1 retired 1.2.0 retired 1.1.0 retired 1.0.0 retired

Tensor library for Gleam/BEAM with a pure Gleam API, zero-copy views, and optional native acceleration

HexDocs Report package

Current section

Files

Jump to
Readme
10 Versions
7 Dependencies
Dependants
Files
Activity
viva_tensor include viva_tensor@models@t5_T5Block.hrl
Raw

include/viva_tensor@models@t5_T5Block.hrl 

-record(t5_block, {

    self_attention :: viva_tensor@nn@attention:multi_head_attention(),

    cross_attention :: gleam@option:option(viva_tensor@nn@attention:multi_head_attention()),

    ffn_w1 :: viva_tensor@tensor:tensor(),

    ffn_w_gate :: viva_tensor@tensor:tensor(),

    ffn_w2 :: viva_tensor@tensor:tensor(),

    norm1 :: viva_tensor@nn@norm:rms_norm(),

    norm2 :: viva_tensor@nn@norm:rms_norm(),

    norm3 :: gleam@option:option(viva_tensor@nn@norm:rms_norm())

}).