| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
【feature】: 为Data节点添加_host_tensor标记 Co-authored-by: SeasonChu<zhuxincheng2@huawei.com> # message auto-generated for no-merge-commit merge: !28 merge hostcpu into master 【feature】: 为Data节点添加_host_tensor标记 Created-by: SeasonChu Commit-by: SeasonChu Merged-by: cann-robot Description: # Pull Request ## 描述 为支持 Host CPU 场景下的输入放置决策，TFA 在图加入 GE Session 前，为 GE 图中的 Data 节点设置布尔属性 _host_tensor=true，供 GE 侧识别 Host Tensor 输入。 ## 主要变更 - TF1 GeOp::AddGraph 在调用 AddGraph 前遍历图节点并标记 Data 节点。 - TF2 NpuDevice::AddGeGraphInner 同步增加相同处理。 - 非 Data 节点和节点类型获取失败场景不设置该属性。 - 补充 TF1/TF2 的 GE 测试桩及 UT/ST 覆盖。 ## 变更类型 - [ ] 🐛 Bug 修复 - [x] ✨ 新功能 - [ ] 💄 代码风格更新（格式化，局部变量） - [ ] ♻️ 重构（既不修复错误也不增加功能的代码变动） - [ ] 📦 构建过程或辅助工具的变动 - [ ] 📝 文档内容更新 ## 关联的 Issue 无。 ## 如何测试 新增并覆盖以下场景： 1. Data 节点成功设置 _host_tensor=true。 2. 非 Data 节点不设置 _host_tensor。 3. 节点类型获取失败时跳过标记。 4. TF1 UT/ST 与 TF2 UT 使用对应 GE stub 验证属性设置行为。 ## 核对清单 - [x] 我的代码遵循了项目的代码风格 - [x] 我已补充相关测试 - [ ] 我已更新了相关文档（本次无需更新文档） - [x] 我在标题中使用了合适的类型标签 - [x] 我已经阅读贡献指南并按要求提交 ## 其他信息 该修改仅设置 Data 节点的 _host_tensor 属性，不改变其他节点属性和图执行流程。 See merge request: cann/tensorflow!28 | 1 天前 |