SetInputType

Function

Sets the position, data format, and data type of the feature matrix input. These settings must be consistent with those in the kernel.

Prototype

void SetInputType(ConvCommonApi::TPosition pos, ConvCommonApi::ConvFormat format, ConvCommonApi::ConvDtype dtype)

Parameters

**Table 1** Parameters
Parameter	Input/Output	Description
pos	Input	Position of the input in the memory. For details, see Position. Currently, only TPosition::GM is supported.
format	Input	Input data format. Currently, only ConvFormat::NDC1HWC0 is supported.
dtype	Input	Input data type. Currently, only ConvDtype::FLOAT16 and ConvDtype::BF16 are supported.

Returns

None

Restrictions

None

Example

optiling::Conv3DBackpropFilterTilingData tilingData;
auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance();
ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform);
conv3dBpDwTiling.SetInputType(ConvCommonApi::TPosition::GM,
                                 ConvCommonApi::ConvFormat::NDC1HWC0,
                                 ConvCommonApi::ConvDtype::FLOAT16);

Parent topic: Conv3DBackpropFilter Tiling API