mirror of https://github.com/open-mmlab/mmcv.git
17 lines
381 B
C
17 lines
381 B
C
// Copyright (c) OpenMMLab. All rights reserved
|
|
#ifndef ONNXRUNTIME_REGISTER_H
|
|
#define ONNXRUNTIME_REGISTER_H
|
|
#include <onnxruntime_c_api.h>
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
OrtStatus *ORT_API_CALL RegisterCustomOps(OrtSessionOptions *options,
|
|
const OrtApiBase *api);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
#endif // ONNXRUNTIME_REGISTER_H
|