Files
calculet-npu-research-archive/source/baseline-expanded/calrt-sdk-0.7.6/usr/local/include/calrt/calrt_infer.h
T

19 lines
959 B
C++

#pragma once
#include "calrt_utils.h"
#include "calrt_buffer.h"
#include "calrt_device.h"
#include "calrt_vdevice.h"
namespace calrt
{
CALRT_API std::unique_ptr<CalrtInputBuf> createInputBuf(const CalbinModel &model);
CALRT_API std::unique_ptr<CalrtOutputBuf> createOutputBuf(const CalbinModel &model);
CALRT_API CalrtError_e infer(std::unique_ptr<VirtualDevice> &vDev, CalbinModel *model, CalrtInputBuf* inputBuffer, CalrtOutputBuf* outputBuffer);
CALRT_API CalrtError_e infer(VirtualDevice *vDev, CalbinModel *model, CalrtInputBuf* inputBuffer, CalrtOutputBuf* outputBuffer);
CALRT_API CalrtError_e infer_with_fixed_task_type(std::unique_ptr<VirtualDevice> &vDev, CalbinModel *model, CalrtInputBuf* inputBuffer, CalrtOutputBuf* outputBuffer, TaskType_e type);
CALRT_API CalrtError_e infer_with_fixed_task_type(VirtualDevice *vDev, CalbinModel *model, CalrtInputBuf* inputBuffer, CalrtOutputBuf* outputBuffer, TaskType_e type);
}