ct2rs 0.9.19

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

#include "ctranslate2/ops/min_max.h"

#include "dispatch.h"

namespace ctranslate2 {
  namespace ops {

    void Min::operator()(const StorageView& a, const StorageView& b, StorageView& c) const {
      PROFILE("Min");
      DEVICE_AND_TYPE_DISPATCH(a.device(), a.dtype(), (compute<D, T>(a, b, c)));
    }

    void Max::operator()(const StorageView& a, const StorageView& b, StorageView& c) const {
      PROFILE("Max");
      DEVICE_AND_TYPE_DISPATCH(a.device(), a.dtype(), (compute<D, T>(a, b, c)));
    }
  }
}