m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!1344 fixed BatchNormGrad

Browse Source 
Merge pull request !1344 from jiangjinsheng/issue_batchnorm_grad
pull/1344/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
parent c14048d05a d691f1c5d7
commit bd426b34f6
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File

9
mindspore/ops/_op_impl/tbe/batchnorm_grad.py
Unescape View File

@ -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()

Write Preview
Loading…
Cancel
Save