* @return NNFW_STATUS_NO_ERROR if successful
*/
-NNFW_STATUS nnfw_set_input(nnfw_session *session, int index, NNFW_TYPE type, const void *buffer,
- size_t length);
+NNFW_STATUS nnfw_set_input(nnfw_session *session, uint32_t index, NNFW_TYPE type,
+ const void *buffer, size_t length);
/*
* Set output
* @return NNFW_STATUS_NO_ERROR if successful
*/
-NNFW_STATUS nnfw_set_output(nnfw_session *session, int index, NNFW_TYPE type, void *buffer,
+NNFW_STATUS nnfw_set_output(nnfw_session *session, uint32_t index, NNFW_TYPE type, void *buffer,
size_t length);
/*
*
* @return NNFW_STATUS_NO_ERROR if successful
*/
-NNFW_STATUS nnfw_input_size(nnfw_session *session, int *number);
+NNFW_STATUS nnfw_input_size(nnfw_session *session, uint32_t *number);
/*
* Get the number of outputs
*
* @return NNFW_STATUS_NO_ERROR if successful
*/
-NNFW_STATUS nnfw_output_size(nnfw_session *session, int *number);
+NNFW_STATUS nnfw_output_size(nnfw_session *session, uint32_t *number);
/*
* Get i-th input tensor info
*
* @return NNFW_STATUS_NO_ERROR if successful
*/
-NNFW_STATUS nnfw_input_tensorinfo(nnfw_session *session, int index, nnfw_tensorinfo *tensor_info);
+NNFW_STATUS nnfw_input_tensorinfo(nnfw_session *session, uint32_t index,
+ nnfw_tensorinfo *tensor_info);
/*
* Get i-th output tensor info
*
* @return NNFW_STATUS_NO_ERROR if successful
*/
-NNFW_STATUS nnfw_output_tensorinfo(nnfw_session *session, int index, nnfw_tensorinfo *tensor_info);
+NNFW_STATUS nnfw_output_tensorinfo(nnfw_session *session, uint32_t index,
+ nnfw_tensorinfo *tensor_info);
#ifdef __cplusplus
}
* @return NNFW_STATUS_NO_ERROR if successful
*/
-NNFW_STATUS nnfw_set_input(nnfw_session *session, int index, NNFW_TYPE type, const void *buffer,
- size_t length)
+NNFW_STATUS nnfw_set_input(nnfw_session *session, uint32_t index, NNFW_TYPE type,
+ const void *buffer, size_t length)
{
return session->set_input(index, type, buffer, length);
}
* @return NNFW_STATUS_NO_ERROR if successful
*/
-NNFW_STATUS nnfw_set_output(nnfw_session *session, int index, NNFW_TYPE type, void *buffer,
+NNFW_STATUS nnfw_set_output(nnfw_session *session, uint32_t index, NNFW_TYPE type, void *buffer,
size_t length)
{
return session->set_output(index, type, buffer, length);
return NNFW_STATUS_NO_ERROR;
}
-NNFW_STATUS nnfw_session::set_input(int index, NNFW_TYPE /*type*/, const void *buffer,
+NNFW_STATUS nnfw_session::set_input(uint32_t index, NNFW_TYPE /*type*/, const void *buffer,
size_t length)
{
try
return NNFW_STATUS_NO_ERROR;
}
-NNFW_STATUS nnfw_session::set_output(int index, NNFW_TYPE /*type*/, void *buffer, size_t length)
+NNFW_STATUS nnfw_session::set_output(uint32_t index, NNFW_TYPE /*type*/, void *buffer,
+ size_t length)
{
try
{
NNFW_STATUS prepare();
NNFW_STATUS run();
- NNFW_STATUS set_input(int index, NNFW_TYPE type, const void *buffer, size_t length);
- NNFW_STATUS set_output(int index, NNFW_TYPE type, void *buffer, size_t length);
+ NNFW_STATUS set_input(uint32_t index, NNFW_TYPE type, const void *buffer, size_t length);
+ NNFW_STATUS set_output(uint32_t index, NNFW_TYPE type, void *buffer, size_t length);
private:
std::shared_ptr<neurun::graph::Graph> _graph;