Skip to content

M
mindspore

magicwindyyd / mindspore
与 Fork 源项目一致

Fork自 MindSpore / mindspore

通知 1
Star 1
Fork 0
M
mindspore
提交 bd426b34 编写于 作者: M mindspore-ci-bot 提交者： Gitee
浏览文件
操作

!1344 fixed BatchNormGrad 

Merge pull request !1344 from jiangjinsheng/issue_batchnorm_grad
上级 c14048d0 d691f1c5
隐藏空白更改
内联 并排
Showing with 4 addition and 5 deletion
mindspore/ops/_op_impl/tbe/batchnorm_grad.py
浏览文件 @ bd426b34
......@@ -31,23 +31,22 @@ batch_norm_grad_op_info = TBERegOp("BatchNormGrad") \
.input(2, "scale", False, "required", "all") \
.input(3, "reserve_space_1", False, "required", "all") \
.input(4, "reserve_space_2", False, "required", "all") \
.input(5, "reserve_space_3", False, "required", "all") \
.output(0, "x_backprop", False, "required", "all") \
.output(1, "scale_backprop", False, "required", "all") \
.output(2, "offset_backprop", False, "required", "all") \
.output(3, "reserve_space_4", False, "optional", "all") \
.output(4, "reserve_space_5", False, "optional", "all") \
.dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F32_Default, DataType.F32_Default,
DataType.F32_Default, DataType.F32_Default, DataType.F16_Default, DataType.F32_Default,
DataType.F32_Default, DataType.F16_Default, DataType.F32_Default,
DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
.dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F32_5HD, DataType.F32_5HD,
DataType.F32_5HD, DataType.F32_5HD, DataType.F16_5HD, DataType.F32_5HD,
DataType.F32_5HD, DataType.F16_5HD, DataType.F32_5HD,
DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
.dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
.dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD,
DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD,
DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD,
DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
.get_op_info()
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或