finish bitnet i2 e2e
This commit is contained in:
parent
2a01a7ce0d
commit
4e1ab50628
6 changed files with 16 additions and 637 deletions
7
ggml.h
7
ggml.h
|
@ -507,8 +507,6 @@ extern "C" {
|
|||
GGML_OP_CROSS_ENTROPY_LOSS,
|
||||
GGML_OP_CROSS_ENTROPY_LOSS_BACK,
|
||||
|
||||
GGML_OP_BITLINEAR_QUANT,
|
||||
|
||||
GGML_OP_COUNT,
|
||||
};
|
||||
|
||||
|
@ -996,11 +994,6 @@ extern "C" {
|
|||
struct ggml_context * ctx,
|
||||
struct ggml_tensor * a);
|
||||
|
||||
// for bitnet
|
||||
GGML_API struct ggml_tensor * ggml_bitlinear_quant(
|
||||
struct ggml_context * ctx,
|
||||
struct ggml_tensor * a);
|
||||
|
||||
// argmax along rows
|
||||
GGML_API struct ggml_tensor * ggml_argmax(
|
||||
struct ggml_context * ctx,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue