REG_AUTO_MAPPING_OP

Applicability

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

Header File

#include <graph/custom.h>

Function Usage

Developers can register custom operators with the framework. The framework calls REG_AUTO_MAPPING_OP at the beginning of build to register custom operators.

Prototype

REG_AUTO_MAPPING_OP(custom_op_class)

Parameters

Parameter	Input/Output	Description
custom_op_class	Input	Name of a custom operator.

Returns

None

Restrictions

None

Parent topic: Basic Data Structures and APIs of Graphs