PipeModelRunner.h

Go to the documentation of this file.

//===--- MLModelRunner/C/PipeModelRunner.h - C API for PipeModelRunner ---===//
//
// 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 PIPE_MODEL_RUNNER_WRAPPER_H
#define PIPE_MODEL_RUNNER_WRAPPER_H
 
#ifdef __cplusplus
extern "C" {
#endif
 
// Define an opaque pointer type for PipeModelRunnerWrapper
typedef struct PipeModelRunnerWrapper PipeModelRunnerWrapper;
 
PipeModelRunnerWrapper *createPipeModelRunner(const char *outBoundName,
                                              const char *inBoundName,
                                              int serDesType);
 
void populateFloatFeatures(PipeModelRunnerWrapper *obj, const char *name,
                           const float *data, const int size);
void populateIntFeatures(PipeModelRunnerWrapper *obj, const char *name,
                         const int *data, const int size);
 
int evaluateIntFeatures(PipeModelRunnerWrapper *obj);
float evaluateFloatFeatures(PipeModelRunnerWrapper *obj);
 
void destroyPipeModelRunner(PipeModelRunnerWrapper *obj);
 
#ifdef __cplusplus
}
#endif
 
#endif // PIPE_MODEL_RUNNER_WRAPPER_H