mnn-sys 0.1.2

Rust bindings for MNN, a lightweight deep neural network inference engine.
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

#include "schedule_c.h"
#include <MNN/Interpreter.hpp>
#include <MNN/MNNForwardType.h>

MNNScheduleConfig *mnnsc_create() {
  auto mnnsc = new MNN::ScheduleConfig();
  mnnsc->saveTensors = std::vector<std::string>();
  return reinterpret_cast<MNNScheduleConfig *>(mnnsc);
}

MNNScheduleConfig *mnnsc_clone(const MNNScheduleConfig *from) {
  auto mnn_from = reinterpret_cast<const MNN::ScheduleConfig *>(from);
  auto mnn_to = new MNN::ScheduleConfig(*mnn_from);
  return reinterpret_cast<MNNScheduleConfig *>(mnn_to);
}

void mnnsc_destroy(MNNScheduleConfig *config) {
  auto mnn_config = reinterpret_cast<MNN::ScheduleConfig *>(config);
  delete mnn_config;
}

void mnnsc_set_save_tensors(MNNScheduleConfig *config,
                            const char *const *saveTensors,
                            size_t saveTensorsSize) {
  auto mnn_config = reinterpret_cast<MNN::ScheduleConfig *>(config);
  auto mnn_saveTensors =
      std::vector<std::string>(saveTensors, saveTensors + saveTensorsSize);
  mnn_config->saveTensors = std::move(mnn_saveTensors);
}

void mnnsc_set_type(MNNScheduleConfig *config, MNNForwardType type) {
  auto mnn_config = reinterpret_cast<MNN::ScheduleConfig *>(config);
  mnn_config->type = type;
}

void mnnsc_set_num_threads(MNNScheduleConfig *config, int numThread) {
  auto mnn_config = reinterpret_cast<MNN::ScheduleConfig *>(config);
  mnn_config->numThread = numThread;
}

void mnnsc_set_mode(MNNScheduleConfig *config, int mode) {
  auto mnn_config = reinterpret_cast<MNN::ScheduleConfig *>(config);
  mnn_config->mode = mode;
}

void mnnsc_set_backup_type(MNNScheduleConfig *config,
                           MNNForwardType backupType) {
  auto mnn_config = reinterpret_cast<MNN::ScheduleConfig *>(config);
  mnn_config->backupType = backupType;
}
void mnnsc_set_backend_config(MNNScheduleConfig *config,
                              MNNBackendConfig *backendConfig) {
  auto mnn_config = reinterpret_cast<MNN::ScheduleConfig *>(config);
  mnn_config->backendConfig =
      reinterpret_cast<MNN::BackendConfig *>(backendConfig);
}

MNNForwardType mnnsc_get_type(MNNScheduleConfig *config) {
  return reinterpret_cast<MNN::ScheduleConfig *>(config)->type;
}
MNNForwardType mnnsc_get_backup_type(MNNScheduleConfig *config) {
  return reinterpret_cast<MNN::ScheduleConfig *>(config)->backupType;
}