Star0

Star0

cann-robotfeat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient)

a131d0e5创建于 9月28日历史提交

文件	最后提交记录	最后更新时间
examples	feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !5245 merge feature/migrate-conversion-ops into master feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: 迁移5个conversion算子，包含op_graph、op_host、tests、CMakeLists和README See merge request: cann/ops-math!5245	1 个月前
op_graph	feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !5245 merge feature/migrate-conversion-ops into master feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: 迁移5个conversion算子，包含op_graph、op_host、tests、CMakeLists和README See merge request: cann/ops-math!5245	1 个月前
op_host	feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !5245 merge feature/migrate-conversion-ops into master feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: 迁移5个conversion算子，包含op_graph、op_host、tests、CMakeLists和README See merge request: cann/ops-math!5245	1 个月前
tests	feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !5245 merge feature/migrate-conversion-ops into master feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: 迁移5个conversion算子，包含op_graph、op_host、tests、CMakeLists和README See merge request: cann/ops-math!5245	1 个月前
CMakeLists.txt	feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !5245 merge feature/migrate-conversion-ops into master feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: 迁移5个conversion算子，包含op_graph、op_host、tests、CMakeLists和README See merge request: cann/ops-math!5245	1 个月前
README.md	feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Co-authored-by: Dazhou_Cao<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !5245 merge feature/migrate-conversion-ops into master feat: 迁移5个conversion算子(identity/identity_n/snapshot/stop_gradient/prevent_gradient) Created-by: Dazhou_Cao Commit-by: Dazhou_Cao Merged-by: cann-robot Description: 迁移5个conversion算子，包含op_graph、op_host、tests、CMakeLists和README See merge request: cann/ops-math!5245	1 个月前

StopGradient

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
Atlas A2系列产品	√
Atlas 200I/500 A2推理产品	√
Atlas推理系列产品	√
Atlas训练系列产品	√

功能说明

算子功能：将输入张量原样传递到输出，同时阻止反向传播计算经过该节点。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入张量。	ALL	ND
y	输出	与输入 `x` 形状、数据类型和内容相同的张量。	与 `x` 相同	ND

约束说明

x 和 y 的形状及数据类型保持一致。
该算子不执行数值变换，仅改变梯度传播行为。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_stop_gradient	通过算子IR构图方式调用StopGradient算子。