SetSingleShape

Applicability

Product	Supported
Atlas 350 Accelerator Card	x
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	x
Atlas inference product AI Core	x
Atlas inference product Vector Core	x
Atlas training product	x

Function Usage

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

Examples

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 APIs