REG_DECOMPOSE_PASS

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 <ge/fusion/pass/decompose_pass.h>

Function Usage

Developers can register the graph modification function in the 1-to-N scenario with the framework. The framework calls the custom graph modification pass at the beginning of build and calls REG_DECOMPOSE_PASS to register the custom pass.

Prototype

REG_DECOMPOSE_PASS(pass_class, decompose_op_types)

Parameters

Parameter	Input/Output	Description
pass_class	Input	Name of a custom pass.
decompose_op_types	Input	Type of the operator to be replaced.

Returns

None

Restrictions

None

Parent topic: Basic Data Structures and APIs of Graphs