Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!29524 fix tensor move data type 

Merge pull request !29524 from laiyongqiang/tensor_move
This commit is contained in:
i-robot 2022-02-07 08:33:55 +00:00 committed by Gitee
parent ae66e00686 8b40f246f0
commit 0d55c15a42
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mindspore/python/mindspore/ops/_op_impl/tbe/tensor_move.py
View File

@ -26,12 +26,20 @@ tensor_move_op_info = TBERegOp("TensorMove") \
.input(0, "x", False, "required", "all") \
.output(0, "y", False, "required", "all") \
.op_pattern("formatAgnostic") \
.dtype_format(DataType.BOOL_None, DataType.BOOL_None) \
.dtype_format(DataType.I8_None, DataType.I8_None) \
.dtype_format(DataType.I16_None, DataType.I16_None) \
.dtype_format(DataType.I32_None, DataType.I32_None) \
.dtype_format(DataType.I64_None, DataType.I64_None) \
.dtype_format(DataType.F16_None, DataType.F16_None) \
.dtype_format(DataType.F32_None, DataType.F32_None) \
.dtype_format(DataType.I8_None, DataType.I8_None) \
.dtype_format(DataType.F64_None, DataType.F32_None) \
.dtype_format(DataType.U8_None, DataType.U8_None) \
.dtype_format(DataType.BOOL_None, DataType.BOOL_None) \
.dtype_format(DataType.U16_None, DataType.U16_None) \
.dtype_format(DataType.U32_None, DataType.U32_None) \
.dtype_format(DataType.U64_None, DataType.U64_None) \
.dtype_format(DataType.C64_Default, DataType.C64_Default) \
.dtype_format(DataType.C128_Default, DataType.C128_Default) \
.get_op_info()