samples/util/alloc.h

#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