CTranslate2/include/ctranslate2/ops/ops.h at b6fa7649dee1bf67163edcf582bd0ca2631e1abe · PyRin-c/CTranslate2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
#pragma once

#include "add.h"
#include "bias_add.h"
#include "concat.h"
#include "conv1d.h"
#include "cos.h"
#include "gather.h"
#include "gelu.h"
#include "gemm.h"
#include "gumbel_max.h"
#include "identity.h"
#include "layer_norm.h"
#include "matmul.h"
#include "mean.h"
#include "mul.h"
#include "multinomial.h"
#include "quantize.h"
#include "relu.h"
#include "sin.h"
#include "softmax.h"
#include "split.h"
#include "squeeze.h"
#include "sub.h"
#include "sigmoid.h"
#include "swish.h"
#include "tile.h"
#include "topk.h"
#include "topp_mask.h"
#include "transpose.h"
#include "dequantize.h"
#include "unsqueeze.h"
#include "min_max.h"
#include "log.h"
#include "rms_norm.h"
#include "tanh.h"
#include "median_filter.h"
#include "rotary.h"
#include "alibi_add.h"
#include "slide.h"
#include "nccl_ops.h"
#include "flash_attention.h"
#ifndef CT2_USE_HIP
#include "awq/gemm.h"
#include "awq/gemv.h"
#include "awq/dequantize_awq.h"
#endif
#include "sum.h"
#include "rotor_quant_kv.h"