| |
| /** @file load_weights.h |
| * @brief |
| * This file contains the definitions of the weight and biases loading functions |
| */ |
| #ifndef LOAD_WEIGHTS_H |
| #define LOAD_WEIGHTS_H |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| |
| /** @brief Init all weights and biases |
| * |
| * @param pObjectContainer The pointer to object container. |
| * @param pFileDir The path to the binary files location |
| * @return vx_status code. |
| */ |
| vx_status loadTensorFromFile(vx_tensor input, const char* pFileDir, const char* pFileName); |
| /** @brief Load Tensor object from file |
| * |
| * @param input The Tensor reference to load. |
| * @param pFileDir The path to the binary files location |
| * @param pFileName The file name |
| * @return vx_status code. |
| */ |
| |
| vx_status initAllWeightsAlexnet(ObjectRefContainerType* pContainer, const char* pFileDir); |
| vx_status initAllWeightsGooglenet(ObjectRefContainerType* pContainer, const char* pFileDir); |
| |
| |
| #ifdef __cplusplus |
| } |
| #endif |
| |
| #endif /* LOAD_WEIGHTS_H */ |
