ct2rs 0.9.21

Rust bindings for OpenNMT/CTranslate2
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

#pragma once

#include "op.h"

namespace ctranslate2 {
  namespace ops {

    // This enum order should remain fixed.
    enum class ActivationType {
      ReLU,
      GELUTanh,
      Swish,
      GELU,
      GELUSigmoid,
      Tanh,
      Sigmoid,
    };

    const UnaryOp& get_activation_op(ActivationType type);

  }
}