SetOrgInputShape
Function
Sets the original shape of the feature matrix Input.
Prototype
1 | void SetOrgInputShape(int64_t orgCi, int64_t orgDi, int64_t orgHi, int64_t orgWi) |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
orgCi |
Input |
Size of the original input channel. |
orgDi |
Input |
Size of the original Input D dimension. |
orgHi |
Input |
Size of the original Input H dimension. |
orgWi |
Input |
Size of the original Input W dimension. |
Returns
None
Restrictions
None
Example
1 2 3 4 | // Instantiate the Conv3D API. auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform); conv3dApiTiling.SetOrgInputShape(orgCi, orgDi, orgHi, orgWi); |
Parent topic: Conv3D Tiling APIs