def fill_bf16_mat

bend-ml-llama@0.2.1.0 · main.bend:1493

import bend-ml-llama@0.2.1.0/main.bend as BendMlLlama

import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

1493def fill_bf16_mat(-r: Nat, -c: Nat, +i: U32, bs: List<&2, U32>, m: TA.Mat<r, c>) -> TA.Mat<r, c>:
1494  match m:
1495    case TA.Mat{d}:
1496      TA.Mat{fill_bf16(bs, i, d)}

Uses (5)

Used by (1)

bend-ml-llama@0.2.1.0/main:stream_exact