gemm_f16.h 190 Bytes
Newer Older
Zhekai Zhang's avatar
Zhekai Zhang committed
1
2
3
4
5
6
7
8
9
#pragma once

#include "common.h"
#include "Tensor.h"

Tensor gemm_f16(
    Tensor input,  // FP16
    Tensor weight, // FP16
    Tensor out,      // FP16
muyangli's avatar
muyangli committed
10
11
    Tensor bias,
    float alpha
Zhekai Zhang's avatar
Zhekai Zhang committed
12
);