CustomPassFn

产品	是否支持
Atlas A3 训练系列产品 / Atlas A3 推理系列产品	√
Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

注册自定义Pass的执行函数。

关于接口的详细使用方法请参见使用自定义Pass修改Graph。

      
           PassRegistrationData &CustomPassFn(const CustomPassFunc &custom_pass_fn)

参数名	输入/输出	说明
custom_pass_fn	输入	自定义Pass的执行函数。

返回自身对象的引用。

无

用户自定义并实现CustomPassFunc类函数，即自定义的改图函数。

      
           Status CustomPassFunc(GraphPtr &graph, CustomPassContext &custom_context)

父主题： PassRegistrationData