ONNXModelRunner.h

Go to the documentation of this file.

//=== MLModelRunner/C/ONNXModelRunner.h - C API for ONNXModelRunner - C++ -===//
//
// Part of the MLCompilerBridge Project, under the Apache License v2.0 with LLVM
// Exceptions. See the LICENSE file for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===---------------------------------------------------------------------===//
//===---------------------------------------------------------------------===//
 
#ifndef ONNX_MODEL_RUNNER_WRAPPER_H
#define ONNX_MODEL_RUNNER_WRAPPER_H
 
typedef struct ONNXModelRunner ONNXModelRunner;
typedef struct Environment Environment;
typedef signed Action;
 
#ifdef __cplusplus
extern "C" {
#endif
 
Environment *createEnvironment();
void env_setDone(Environment *env);
void env_resetDone(Environment *env);
bool env_checkDone(Environment *env);
void env_setNumFeatures(Environment *env, int numFeatures);
void env_setStepFunc(Environment *env, float *(*stepFunc)(Action action));
void env_setResetFunc(Environment *env, float *(*resetFunc)());
void env_setNextAgent(Environment *env, char *agentName);
 
ONNXModelRunner *createONNXModelRunner(Environment *env, int numAgents, ...);
ONNXModelRunner *createSingleAgentOMR(char *agent_path);
void evaluate(ONNXModelRunner *obj);
int singleAgentEvaluate(ONNXModelRunner *obj, float *inp, int inp_size);
void destroyEnvironment(Environment *env);
void destroyONNXModelRunner(ONNXModelRunner *obj);
 
#ifdef __cplusplus
}
#endif
 
#endif // ONNX_MODEL_RUNNER_WRAPPER_H