SetGroups

Function

Sets the group size for group convolution. A group size of 1 indicates common convolution. Currently, the high-level Conv3D APIs do not support group convolution.

Prototype

1

void SetGroups(int64_t groups)

Parameters

Returns

None

Restrictions

This API is optional before calling the GetTiling API. If this API is not called, the default value of groups is 1. Currently, only the value 1 is supported. The convolution capability with group > 1 is not supported.

Example

1
2
3
4

// Instantiate the Conv3D API.
auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform);
conv3dApiTiling.SetGroups(groups);