aclopRegisterCompileFunc

Description

Registers an operator selector for a dynamic-shape operator. The selection of the tiling policy depends on the operator's input shape.

Restrictions

An operator can register with only one operator selector. To change the operator selector, call aclopUnregisterCompileFunc to unregister and then call aclopRegisterCompileFunc to register again.

Prototype

aclError aclopRegisterCompileFunc(const char *opType, aclopCompileFunc func)

Parameters

Parameter

Input/Output

Description

opType

Input

Pointer to the operator type.

func

Input

Callback function of the operator selector.

typedef aclError (*aclopCompileFunc)(int numInputs, const aclTensorDesc *const inputDesc[], int numOutputs, const aclTensorDesc *const outputDesc[], const aclopAttr *opAttr, aclopKernelDesc *aclopKernelDesc);

Returns

The value 0 indicates success, and other values indicate failure. For details, see aclError.

See Also

For details about the API call sequence and example, see Sample Code for Executing a Dynamic-Shape Operator (Operator Selector Registered).