SetSingleShape

Applicability

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	x
Atlas inference product's AI Core	x
Atlas inference product's Vector Core	x
Atlas training products	x

Function

Sets the shape of Conv3DBackpropInput computation on a single core. The unit is the number of elements.

Prototype

__aicore__ inline void SetSingleShape(uint64_t singleShapeM, uint64_t singleShapeK, uint32_t singleShapeN)

Parameters

**Table 1** API parameters
Parameter	Input/Output	Description
singleShapeM	Input	Size of M on a single core. The unit is element.
singleShapeK	Input	Size of K on a single core. The unit is element.
singleShapeN	Input	Size of N on a single core. The unit is element.

Returns

None

Restrictions

None

Example

const Conv3DBackpropInputTilingData* tilingData;
//... Initialize tilingData.
ConvBackpropApi::Conv3DBackpropInput<weightDxType, inputSizeDxType, gradOutputDxType, gradInputDxType> gradInput_;
//... Set other parameters.
gradInput_.SetSingleShape(singleShapeM_, singleShapeK_, singleShapeN_); // Set the shape of single-core computation.

Parent topic: Conv3DBackpropInput Kernel API