SetBiasType
Function
Sets the position, data format, and data type of Bias in the memory.
Prototype
1 | void SetBiasType(const ConvCommonApi::TPosition pos, const ConvCommonApi::ConvFormat format, const ConvCommonApi::ConvDtype dtype) |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
pos |
Input |
Position of Bias in the memory. Curently, only TPosition::GM is supported. |
format |
Input |
Bias data format. Currently, only ConvFormat::ND is supported. |
dtype |
Input |
Bias data type. Currently, only ConvDtype::FLOAT16 and ConvDtype::FLOAT are supported. |
Returns
None
Restrictions
This API is optional before calling the GetTiling API. If this API is not called, the default Bias is pos=TPosition::GM, format=ConvFormat::ND and dtype=ConvDtype::FLOAT16.
Example
1 2 3 4 | // Instantiate the Conv3D API. auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform); conv3dApiTiling.SetBiasType(ConvCommonApi::TPosition::GM, ConvCommonApi::ConvFormat::ND, ConvCommonApi::ConvDtype::FLOAT32); |
Parent topic: Conv3D Tiling APIs