PermuteLayer functions
Functions to acceleratŠµ PermuteLayer in Synet Framework. More...
Functions | |
SIMD_API void * | SimdSynetPermuteInit (const size_t *shape, const size_t *order, size_t count, SimdTensorDataType type) |
Initilizes permute algorithm. More... | |
SIMD_API size_t | SimdSynetPermuteInternalBufferSize (const void *context) |
Gets size of internal buffer used inside permute algorithm. More... | |
SIMD_API void | SimdSynetPermuteForward (void *context, const uint8_t *src, uint8_t *dst) |
Performs forward propagation of permute algorithm. More... | |
Detailed Description
Functions to acceleratŠµ PermuteLayer in Synet Framework.
Function Documentation
◆ SimdSynetPermuteInit()
void * SimdSynetPermuteInit | ( | const size_t * | shape, |
const size_t * | order, | ||
size_t | count, | ||
SimdTensorDataType | type | ||
) |
Initilizes permute algorithm.
- Parameters
-
[in] shape - a pointer to shape of input tensor. [in] order - a pointer to order of dimensions in output tensor. [in] count - a count of dimensions of input / output tensor. [in] type - an input / output tensor type.
- Returns
- a pointer to permute context. On error it returns NULL. It must be released with using of function SimdRelease. This pointer is used in functions SimdSynetPermuteInternalBufferSize, and SimdSynetPermuteForward.
◆ SimdSynetPermuteInternalBufferSize()
size_t SimdSynetPermuteInternalBufferSize | ( | const void * | context | ) |
Gets size of internal buffer used inside permute algorithm.
- Parameters
-
[in] context - a pointer to permute context. It must be created by function SimdSynetPermuteInit and released by function SimdRelease.
- Returns
- size of internal buffer used inside permute algorithm.
◆ SimdSynetPermuteForward()
void SimdSynetPermuteForward | ( | void * | context, |
const uint8_t * | src, | ||
uint8_t * | dst | ||
) |
Performs forward propagation of permute algorithm.
- Parameters
-
[in] context - a pointer to permute context. It must be created by function SimdSynetPermuteInit and released by function SimdRelease. [in] src - a pointer to input image. [out] dst - a pointer to output image.