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);