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
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#pragma once

#include <cassert>
#include <vector>

#include "ctranslate2/storage_view.h"
#include "ctranslate2/primitives.h"
#include "ctranslate2/profiler.h"

namespace ctranslate2 {
  namespace ops {

    // Base class for operators.
    class Op {
    public:
      virtual ~Op() = default;
    };


    // Base classes for common N-ary operators.
    class UnaryOp : public Op {
    public:
      virtual void operator()(const StorageView&, StorageView&) const = 0;
    };

    class BinaryOp : public Op {
    public:
      virtual void operator()(const StorageView&, const StorageView&, StorageView&) const = 0;
    };

    class TernaryOp : public Op {
    public:
      virtual void operator()(const StorageView&,
                              const StorageView&,
                              const StorageView&,
                              StorageView&) const = 0;
    };

  }
}