Browse Source

!2501 Fix some mistakes of TransData and ConfusionTransposeD vm ops

Merge pull request !2501 from liuwenhao/master
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
41785e18c1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/_op_impl/tbe/trans_data.py

+ 1
- 1
mindspore/ops/_op_impl/tbe/trans_data.py View File

@@ -50,7 +50,7 @@ trans_data_op_info = TBERegOp("TransData") \
.dtype_format(DataType.F16_HWCN, DataType.F16_FracZ) \
.dtype_format(DataType.F16_FracZ, DataType.F16_HWCN) \
.dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_HWCN) \
.dtype_format(DataType.F16_HWCN, DataType.F16_5HD) \
.dtype_format(DataType.F16_HWCN, DataType.F16_C1HWNCoC0) \
.dtype_format(DataType.F16_Default, DataType.F16_FracNZ) \
.dtype_format(DataType.F32_Default, DataType.F32_FracNZ) \
.dtype_format(DataType.F16_FracNZ, DataType.F16_Default) \


Loading…
Cancel
Save