rockx_tensor_util.h

Go to the documentation of this file.

/****************************************************************************
*
*    Copyright (c) 2017 - 2019 by Rockchip Corp.  All rights reserved.
*
*    The material in this file is confidential and contains trade secrets
*    of Rockchip Corporation. This is proprietary information owned by
*    Rockchip Corporation. No part of this work may be disclosed,
*    reproduced, copied, transmitted, or used in any way for any purpose,
*    without the express written permission of Rockchip Corporation.
*
*****************************************************************************/
 
#ifndef _TENSOR_UTIL_H
#define _TENSOR_UTIL_H
 
#include "rockx_type.h"
 
#ifdef __cplusplus
extern "C" {
#endif
 
int rockx_tensor_dtype_size(rockx_data_type type);
 
int rockx_tensor_elems_num(rockx_tensor_t *tensor);
 
int rockx_tensor_size(rockx_tensor_t *tensor);
 
void rockx_tensor_dump(rockx_tensor_t *tensor);
 
void write_tensor(const char *path, rockx_tensor_t *tensor);
 
rockx_ret_t rockx_tensor_init_from_image(rockx_image_t *img, rockx_tensor_t *tensor);
 
rockx_ret_t rockx_tensor_clone(rockx_tensor_t *tensor, rockx_tensor_t *dst_tensor);
 
rockx_ret_t rockx_tensor_release(rockx_tensor_t *tensor);
 
#ifdef __cplusplus
} //extern "C"
#endif
 
#endif //_TENSOR_UTIL_H