BF16 deconvolution framework
A framework to accelerate BF16 deconvolution in Synet Framework. More...
Functions | |
SIMD_API void * | SimdSynetDeconvolution16bInit (size_t batch, const SimdConvolutionParameters *conv, SimdSynetCompatibilityType compatibility) |
Initilizes BF16 deconvolution algorithm. More... | |
SIMD_API size_t | SimdSynetDeconvolution16bExternalBufferSize (const void *context) |
Gets size in bytes of external temporary buffer required for BF16 deconvolution algorithm. More... | |
SIMD_API size_t | SimdSynetDeconvolution16bInternalBufferSize (const void *context) |
Gets size (in bytes) of internal buffer used inside BF16 deconvolution algorithm. More... | |
SIMD_API const char * | SimdSynetDeconvolution16bInfo (const void *context) |
Gets description of internal implementation of BF16 deconvolution algorithm. More... | |
SIMD_API void | SimdSynetDeconvolution16bForward (void *context, const uint8_t *src, uint8_t *buf, uint8_t *dst) |
Performs forward propagation of BF16 deconvolution algorithm. More... | |
Detailed Description
A framework to accelerate BF16 deconvolution in Synet Framework.
Function Documentation
◆ SimdSynetDeconvolution16bInit()
void * SimdSynetDeconvolution16bInit | ( | size_t | batch, |
const SimdConvolutionParameters * | conv, | ||
SimdSynetCompatibilityType | compatibility | ||
) |
Initilizes BF16 deconvolution algorithm.
- Parameters
-
[in] batch - a batch size. [in] conv - a pointer to deconvolution parameters. [in] compatibility - a flags of calculation compatibility.
- Returns
- a pointer to BF16 convolution context. On error it returns NULL. It must be released with using of function SimdRelease. This pointer is used in functions SimdSynetDeconvolution16bExternalBufferSize, SimdSynetDeconvolution16bInternalBufferSize, SimdSynetDeconvolution16bInfo, SimdSynetDeconvolution16bSetParams and SimdSynetDeconvolution16bForward.
◆ SimdSynetDeconvolution16bExternalBufferSize()
size_t SimdSynetDeconvolution16bExternalBufferSize | ( | const void * | context | ) |
Gets size in bytes of external temporary buffer required for BF16 deconvolution algorithm.
- Parameters
-
[in] context - a pointer to BF16 deconvolution context. It must be created by function SimdSynetDeconvolution16bInit and released by function SimdRelease.
- Returns
- size of external temporary buffer required for BF16 convolution algorithm.
◆ SimdSynetDeconvolution16bInternalBufferSize()
size_t SimdSynetDeconvolution16bInternalBufferSize | ( | const void * | context | ) |
Gets size (in bytes) of internal buffer used inside BF16 deconvolution algorithm.
- Parameters
-
[in] context - a pointer to BF16 deconvolution context. It must be created by function SimdSynetDeconvolution16bInit and released by function SimdRelease.
- Returns
- size of internal buffer used inside BF16 convolution algorithm.
◆ SimdSynetDeconvolution16bInfo()
const char * SimdSynetDeconvolution16bInfo | ( | const void * | context | ) |
Gets description of internal implementation of BF16 deconvolution algorithm.
- Parameters
-
[in] context - a pointer to BF16 deconvolution context. It must be created by function SimdSynetDeconvolution16bInit and released by function SimdRelease.
- Returns
- string with description of internal implementation of BF16 convolution algorithm.
◆ SimdSynetDeconvolution16bForward()
void SimdSynetDeconvolution16bForward | ( | void * | context, |
const uint8_t * | src, | ||
uint8_t * | buf, | ||
uint8_t * | dst | ||
) |
Performs forward propagation of BF16 deconvolution algorithm.
- Parameters
-
[in] context - a pointer to BF16 deconvolution context. It must be created by function SimdSynetDeconvolution16bInit and released by function SimdRelease. [in] src - a pointer to input tensor. [out] buf - a pointer to external temporary buffer. The size of the external temporary buffer is determined by function SimdSynetDeconvolution16bExternalBufferSize. Can be NULL (it causes usage of internal buffer). [out] dst - a pointer to output tensor.