21 lines
719 B
C
21 lines
719 B
C
/*
|
|
* Header to call Dynasty inference
|
|
*
|
|
* Copyright (C) 2020 Kneron, Inc. All rights reserved.
|
|
*
|
|
*/
|
|
#ifndef MAIN_H
|
|
#define MAIN_H
|
|
|
|
extern "C" void inference_wrapper(
|
|
const char *platform, const char *encrypt, const char *shape_order, int dump,
|
|
const char *model, const char *radix_file, int num_inputs, const char **input_files,
|
|
const char **input_names, const char *output_folder, const char *debug, const char *cuda,
|
|
const char *ort);
|
|
|
|
extern "C" int check_input_dims(const char *platform, const char *encrypt, const char *model,
|
|
const char *radix, int *dims);
|
|
extern "C" void dump_radix_json(int platform, const char *model, const char *out_json_folder);
|
|
|
|
#endif
|