SetGradOutputShape

Function

Sets the GradOutput shape: Batch, Channel, Depth, Height, and Width.

Prototype

1

bool SetGradOutputShape(int64_t n, int64_t c, int64_t d, int64_t h, int64_t w)

Parameters

Returns

true indicates that the setting is successful, and false indicates that the setting fails.

Restrictions

None

Example

1
2
3

auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance();
ConvBackpropApi::Conv3DBpInputTiling conv3DBpDxTiling(*ascendcPlatform);
conv3DBpDxTiling.SetGradOutputShape(n, c, d, h, w);