blob: 0d7f07f3cd4c616736344b92716d2d27c1a98ced [file] [log] [blame]
#ifndef SAMPLES_UTIL_ALLOC_H
#define SAMPLES_UTIL_ALLOC_H
#include "iree/base/api.h"
#include "samples/util/model_api.h"
// Allocate the input buffer w.r.t the model config.
// The buffer must be released by the external caller.
iree_status_t alloc_input_buffer(const MlModel *model,
void **buffer);
#endif // SAMPLES_UTIL_ALLOC_H