Welcome to mirror list, hosted at ThFree Co, Russian Federation.

ops.h « ops « ctranslate2 « include - github.com/OpenNMT/CTranslate2.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 9262118b5015b0efdf1fa1daa6b39e6d2e6998cf (plain)
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
#pragma once

// Operators following ONNX specifications.

#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 "reshape.h"
#include "sin.h"
#include "softmax.h"
#include "split.h"
#include "squeeze.h"
#include "sub.h"
#include "swish.h"
#include "tile.h"
#include "topk.h"
#include "transpose.h"
#include "dequantize.h"
#include "unsqueeze.h"
#include "min_max.h"
#include "log.h"