Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
CANN ops 仓重组为 ops-nn/ops-math 等新仓后,TF 框架侧的算子注册(REGISTER_CUSTOM_OP)需要迁移到新仓工具链。旧注册 API(ParseParamsFn/AutoMappingFn、标量 OriginOpType)以及依赖 TF protobuf NodeDef(node_def.pb.h)的解析逻辑在新工具链中已不存在,需统一迁移为 ParseParamsByOperatorFn(AutoMappingByOpFn)、OriginOpType(std::vectorge::AscendString{...}),复杂算子改为基于 ge::Operator 的子图分解(ParseOpToGraphFn / FusionParseParamsFn)。本需求为 TF 插件迁移系列的第三部分,补齐 common/index/norm/optim/matmul/rnn 各模块剩余算子的 TF 注册与解析。
cann/ops-nn 仓 TF 插件迁移工程(本仓已完成迁移 part 1/2 的同系列工作)。
补齐 TF 框架侧算子在 ops-nn 新工具链的注册与图解析/分解能力,覆盖:Gather/ResourceGather、TensorScatterAdd/TensorScatterSub、FusedLayerNorm/FusedLayerNormGrad、BroadcastGradientArgs、SparseSegmentMean/SparseSegmentMeanGrad、SparseSlice、SparseToDense、Einsum、AscendWeightQuant、BNTrainingUpdate、GatherPoint/GatherPointGrad、BlockLSTM、DynamicRNN/DynamicRnn、DynamicRNNV2/DynamicRnnV2 系列、ApplyAdagrad 等,保证携带上述算子的 TF 模型可正常解析落图运行。
按模块将注册迁移拆分为四组:
关联 PR:https://gitcode.com/cann/ops-nn/pull/9758
/assign @Nice_try
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
Backgroud(背景信息)
CANN ops 仓重组为 ops-nn/ops-math 等新仓后,TF 框架侧的算子注册(REGISTER_CUSTOM_OP)需要迁移到新仓工具链。旧注册 API(ParseParamsFn/AutoMappingFn、标量 OriginOpType)以及依赖 TF protobuf NodeDef(node_def.pb.h)的解析逻辑在新工具链中已不存在,需统一迁移为 ParseParamsByOperatorFn(AutoMappingByOpFn)、OriginOpType(std::vectorge::AscendString{...}),复杂算子改为基于 ge::Operator 的子图分解(ParseOpToGraphFn / FusionParseParamsFn)。本需求为 TF 插件迁移系列的第三部分,补齐 common/index/norm/optim/matmul/rnn 各模块剩余算子的 TF 注册与解析。
Origin(信息来源)
cann/ops-nn 仓 TF 插件迁移工程(本仓已完成迁移 part 1/2 的同系列工作)。
Benefit / Necessity (价值/作用)
补齐 TF 框架侧算子在 ops-nn 新工具链的注册与图解析/分解能力,覆盖:Gather/ResourceGather、TensorScatterAdd/TensorScatterSub、FusedLayerNorm/FusedLayerNormGrad、BroadcastGradientArgs、SparseSegmentMean/SparseSegmentMeanGrad、SparseSlice、SparseToDense、Einsum、AscendWeightQuant、BNTrainingUpdate、GatherPoint/GatherPointGrad、BlockLSTM、DynamicRNN/DynamicRnn、DynamicRNNV2/DynamicRnnV2 系列、ApplyAdagrad 等,保证携带上述算子的 TF 模型可正常解析落图运行。
Design(设计方案)
按模块将注册迁移拆分为四组:
全部为 *_tf_plugin.cpp 注册文件变更(24 个文件,+627/-8),不涉及 kernel/tiling 与测试文件。
关联 PR:https://gitcode.com/cann/ops-nn/pull/9758