Module ops

Module ops 

Source
Expand description

Implementation of common array operations with mixed precision.

Functionsยง

add
Element-wise addition with specified precision.
matmul
Matrix multiplication with specified precision.
multiply
Element-wise multiplication with specified precision.