/third_party/mindspore/mindspore/ops/_op_impl/aicpu/ |
D | cast.py | 17 from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType 23 .dtype_format(DataType.U8_Default, DataType.U8_Default) \ 24 .dtype_format(DataType.U8_Default, DataType.U16_Default) \ 25 .dtype_format(DataType.U8_Default, DataType.U32_Default) \ 26 .dtype_format(DataType.U8_Default, DataType.U64_Default) \ 27 .dtype_format(DataType.U8_Default, DataType.I8_Default) \ 28 .dtype_format(DataType.U8_Default, DataType.I16_Default) \ 29 .dtype_format(DataType.U8_Default, DataType.I32_Default) \ 30 .dtype_format(DataType.U8_Default, DataType.I64_Default) \ 31 .dtype_format(DataType.U8_Default, DataType.F16_Default) \ [all …]
|
D | cache_swap_table.py | 17 from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType 25 .dtype_format(DataType.I8_Default, DataType.I32_Default, \ 26 DataType.I8_Default, DataType.I8_Default) \ 27 .dtype_format(DataType.I16_Default, DataType.I32_Default, \ 28 DataType.I16_Default, DataType.I16_Default) \ 29 .dtype_format(DataType.I32_Default, DataType.I32_Default, \ 30 DataType.I32_Default, DataType.I32_Default) \ 31 .dtype_format(DataType.I64_Default, DataType.I32_Default, \ 32 DataType.I64_Default, DataType.I64_Default) \ 33 .dtype_format(DataType.U8_Default, DataType.I32_Default, \ [all …]
|
D | embedding_lookup.py | 17 from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType 25 .dtype_format(DataType.I8_Default, DataType.I32_Default, \ 26 DataType.I32_Default, DataType.I8_Default) \ 27 .dtype_format(DataType.I16_Default, DataType.I32_Default, \ 28 DataType.I32_Default, DataType.I16_Default) \ 29 .dtype_format(DataType.I32_Default, DataType.I32_Default, \ 30 DataType.I32_Default, DataType.I32_Default) \ 31 .dtype_format(DataType.I64_Default, DataType.I32_Default, \ 32 DataType.I32_Default, DataType.I64_Default) \ 33 .dtype_format(DataType.U8_Default, DataType.I32_Default, \ [all …]
|
D | scatter.py | 17 from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType 26 .dtype_format(DataType.I8_Default, DataType.I32_Default, \ 27 DataType.I32_Default, DataType.I8_Default, DataType.I8_Default) \ 28 .dtype_format(DataType.I16_Default, DataType.I32_Default, \ 29 DataType.I32_Default, DataType.I16_Default, DataType.I16_Default) \ 30 .dtype_format(DataType.I32_Default, DataType.I32_Default, \ 31 DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \ 32 .dtype_format(DataType.I64_Default, DataType.I32_Default, \ 33 DataType.I32_Default, DataType.I64_Default, DataType.I64_Default) \ 34 .dtype_format(DataType.U8_Default, DataType.I32_Default, \ [all …]
|
D | gather.py | 17 from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType 25 .dtype_format(DataType.I8_Default, DataType.I32_Default, \ 26 DataType.I32_Default, DataType.I8_Default) \ 27 .dtype_format(DataType.I16_Default, DataType.I32_Default, \ 28 DataType.I32_Default, DataType.I16_Default) \ 29 .dtype_format(DataType.I32_Default, DataType.I32_Default, \ 30 DataType.I32_Default, DataType.I32_Default) \ 31 .dtype_format(DataType.I64_Default, DataType.I32_Default, \ 32 DataType.I32_Default, DataType.I64_Default) \ 33 .dtype_format(DataType.U8_Default, DataType.I32_Default, \ [all …]
|
D | crop_and_resize.py | 17 from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType 27 ….dtype_format(DataType.I8_Default, DataType.F32_Default, DataType.I32_Default, DataType.I32_Defaul… 28 DataType.F32_Default) \ 29 ….dtype_format(DataType.I16_Default, DataType.F32_Default, DataType.I32_Default, DataType.I32_Defau… 30 DataType.F32_Default) \ 31 ….dtype_format(DataType.I32_Default, DataType.F32_Default, DataType.I32_Default, DataType.I32_Defau… 32 DataType.F32_Default) \ 33 ….dtype_format(DataType.I64_Default, DataType.F32_Default, DataType.I32_Default, DataType.I32_Defau… 34 DataType.F32_Default) \ 35 ….dtype_format(DataType.F16_Default, DataType.F32_Default, DataType.I32_Default, DataType.I32_Defau… [all …]
|
D | edit_distance.py | 17 from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType 29 .dtype_format(DataType.I64_Default, DataType.I8_Default, DataType.I64_Default, \ 30 DataType.I64_Default, DataType.I8_Default, DataType.I64_Default, DataType.F32_Default,) \ 31 .dtype_format(DataType.I64_Default, DataType.I16_Default, DataType.I64_Default, \ 32 DataType.I64_Default, DataType.I16_Default, DataType.I64_Default, DataType.F32_Default,) \ 33 .dtype_format(DataType.I64_Default, DataType.I32_Default, DataType.I64_Default, \ 34 DataType.I64_Default, DataType.I32_Default, DataType.I64_Default, DataType.F32_Default,) \ 35 .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.I64_Default, \ 36 DataType.I64_Default, DataType.I64_Default, DataType.I64_Default, DataType.F32_Default,) \ 37 .dtype_format(DataType.I64_Default, DataType.U8_Default, DataType.I64_Default, \ [all …]
|
D | reverse_sequence.py | 17 from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType 25 .dtype_format(DataType.BOOL_Default, DataType.I32_Default, DataType.BOOL_Default) \ 26 .dtype_format(DataType.I8_Default, DataType.I32_Default, DataType.I8_Default) \ 27 .dtype_format(DataType.I16_Default, DataType.I32_Default, DataType.I16_Default) \ 28 .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \ 29 .dtype_format(DataType.I32_Default, DataType.I64_Default, DataType.I32_Default) \ 30 .dtype_format(DataType.I64_Default, DataType.I32_Default, DataType.I64_Default) \ 31 .dtype_format(DataType.U8_Default, DataType.I32_Default, DataType.U8_Default) \ 32 .dtype_format(DataType.U16_Default, DataType.I32_Default, DataType.U16_Default) \ 33 .dtype_format(DataType.U32_Default, DataType.I32_Default, DataType.U32_Default) \ [all …]
|
/third_party/mindspore/mindspore/ops/_op_impl/cpu/ |
D | cast.py | 17 from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType 22 .dtype_format(DataType.U8_Default, DataType.U8_Default) \ 23 .dtype_format(DataType.U8_Default, DataType.U16_Default) \ 24 .dtype_format(DataType.U8_Default, DataType.U32_Default) \ 25 .dtype_format(DataType.U8_Default, DataType.U64_Default) \ 26 .dtype_format(DataType.U8_Default, DataType.I8_Default) \ 27 .dtype_format(DataType.U8_Default, DataType.I16_Default) \ 28 .dtype_format(DataType.U8_Default, DataType.I32_Default) \ 29 .dtype_format(DataType.U8_Default, DataType.I64_Default) \ 30 .dtype_format(DataType.U8_Default, DataType.F16_Default) \ [all …]
|
/third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
D | trans_data.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 31 .dtype_format(DataType.F32_NHWC, DataType.F32_5HD) \ 32 .dtype_format(DataType.F32_Default, DataType.F32_5HD) \ 33 .dtype_format(DataType.F32_5HD, DataType.F32_NHWC) \ 34 .dtype_format(DataType.F32_5HD, DataType.F32_Default) \ 35 .dtype_format(DataType.F32_FracZ, DataType.F32_Default) \ 36 .dtype_format(DataType.F32_Default, DataType.F32_FracZ) \ 37 .dtype_format(DataType.F32_HWCN, DataType.F32_FracZ) \ 38 .dtype_format(DataType.F32_FracZ, DataType.F32_HWCN) \ 39 .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_HWCN) \ [all …]
|
D | trans_data_ds.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 32 .dtype_format(DataType.F32_NHWC, DataType.F32_5HD) \ 33 .dtype_format(DataType.F32_Default, DataType.F32_5HD) \ 34 .dtype_format(DataType.F32_5HD, DataType.F32_NHWC) \ 35 .dtype_format(DataType.F32_5HD, DataType.F32_Default) \ 36 .dtype_format(DataType.F32_FracZ, DataType.F32_Default) \ 37 .dtype_format(DataType.F32_Default, DataType.F32_FracZ) \ 38 .dtype_format(DataType.F32_HWCN, DataType.F32_FracZ) \ 39 .dtype_format(DataType.F32_FracZ, DataType.F32_HWCN) \ 40 .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_HWCN) \ [all …]
|
D | assign.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 29 .dtype_format(DataType.BOOL_Default, DataType.BOOL_Default, DataType.BOOL_Default) \ 30 .dtype_format(DataType.BOOL_5HD, DataType.BOOL_5HD, DataType.BOOL_5HD) \ 31 .dtype_format(DataType.BOOL_C1HWNCoC0, DataType.BOOL_C1HWNCoC0, DataType.BOOL_C1HWNCoC0) \ 32 .dtype_format(DataType.BOOL_FracZ, DataType.BOOL_FracZ, DataType.BOOL_FracZ) \ 33 .dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.I8_Default) \ 34 .dtype_format(DataType.I8_5HD, DataType.I8_5HD, DataType.I8_5HD) \ 35 .dtype_format(DataType.I8_C1HWNCoC0, DataType.I8_C1HWNCoC0, DataType.I8_C1HWNCoC0) \ 36 .dtype_format(DataType.I8_FracZ, DataType.I8_FracZ, DataType.I8_FracZ) \ 37 .dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.U8_Default) \ [all …]
|
D | dynamic_gru_v2.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 49 .dtype_format(DataType.F16_FracNZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F32_Default, 50 … DataType.F32_Default, DataType.None_Default, DataType.F32_FracNZ, DataType.F32_FracNZ, 51 … DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ, 52 DataType.F32_FracNZ) \ 53 .dtype_format(DataType.F16_FracNZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F32_Default, 54 … DataType.None_Default, DataType.None_Default, DataType.F32_FracNZ, DataType.F32_FracNZ, 55 … DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ, 56 DataType.F32_FracNZ) \ 57 … .dtype_format(DataType.F16_FracNZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.None_Default, [all …]
|
D | gather_v2_ds.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 32 ….dtype_format(DataType.I8_Default, DataType.I32_Default, DataType.I32_Default, DataType.I8_Default… 33 ….dtype_format(DataType.I8_Default, DataType.I64_Default, DataType.I32_Default, DataType.I8_Default… 34 .dtype_format(DataType.I8_5HD, DataType.I32_5HD, DataType.I32_5HD, DataType.I8_5HD) \ 35 .dtype_format(DataType.I8_5HD, DataType.I64_5HD, DataType.I32_5HD, DataType.I8_5HD) \ 36 .dtype_format(DataType.I8_FracZ, DataType.I32_FracZ, DataType.I32_FracZ, DataType.I8_FracZ) \ 37 .dtype_format(DataType.I8_FracZ, DataType.I64_FracZ, DataType.I32_FracZ, DataType.I8_FracZ) \ 38 ….dtype_format(DataType.U8_Default, DataType.I32_Default, DataType.I32_Default, DataType.U8_Default… 39 ….dtype_format(DataType.U8_Default, DataType.I64_Default, DataType.I32_Default, DataType.U8_Default… 40 .dtype_format(DataType.U8_5HD, DataType.I32_5HD, DataType.I32_5HD, DataType.U8_5HD) \ [all …]
|
D | apply_centered_rms_prop.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 39 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, 40 … DataType.F16_5HD, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 41 DataType.F16_Default, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, 42 DataType.F16_5HD) \ 43 .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, 44 … DataType.F16_FracZ, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 45 DataType.F16_Default, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, 46 DataType.F16_FracZ) \ 47 ….dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16… [all …]
|
D | apply_adam.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 41 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 42 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 43 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 44 DataType.F16_Default) \ 45 ….dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16… 46 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 47 … DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, 48 DataType.F16_C1HWNCoC0) \ 49 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, [all …]
|
D | confusion_matrix.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 32 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 33 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F32_Default, DataType.F32_Defau… 34 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, DataType.I32_Defau… 35 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.I8_Default, DataType.I8_Default… 36 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.U8_Default, DataType.U8_Default… 37 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F16_Default, DataType.F16_Defau… 38 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 39 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.I32_Default, DataType.I32_Defau… 40 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.I8_Default, DataType.I8_Default… [all …]
|
D | apply_ada_max.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 38 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, 39 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 40 DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \ 41 .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_Default, 42 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 43 DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ) \ 44 ….dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16… 45 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 46 … DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \ [all …]
|
D | apply_adagrad_d_a.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 38 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, 39 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, 40 DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \ 41 .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, 42 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, 43 DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ) \ 44 ….dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16… 45 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, 46 DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \ [all …]
|
D | apply_ftrl.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 37 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, 38 … DataType.F16_5HD, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 39 DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \ 40 .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, 41 … DataType.F16_FracZ, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 42 DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ) \ 43 ….dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16… 44 … DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 45 DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \ [all …]
|
D | apply_momentum.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 34 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 35 DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \ 36 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, DataType.F16_5HD, 37 DataType.F16_Default, DataType.F16_5HD, DataType.F16_5HD) \ 38 ….dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_C… 39 DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \ 40 .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_Default, DataType.F16_FracZ, 41 DataType.F16_Default, DataType.F16_FracZ, DataType.F16_FracZ) \ 42 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… [all …]
|
D | fused_mul_apply_momentum_extern.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 37 .dtype_format(DataType.F32_5HD, DataType.F16_5HD, DataType.F16_Default, DataType.F16_5HD, 38 DataType.F16_Default, DataType.F16_Default, DataType.F16_5HD, DataType.F32_5HD, 39 DataType.F16_5HD, DataType.F16_5HD) \ 40 ….dtype_format(DataType.F32_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_C… 41 … DataType.F16_Default, DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F32_C1HWNCoC0, 42 DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \ 43 ….dtype_format(DataType.F32_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 44 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F32_Default, 45 DataType.F16_Default, DataType.F16_Default) \ [all …]
|
D | sparse_gather_v2_ds.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 32 ….dtype_format(DataType.F16_Default, DataType.I32_Default, DataType.I32_Default, DataType.F16_Defau… 33 ….dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.I32_Default, DataType.F32_Defau… 34 ….dtype_format(DataType.I8_Default, DataType.I32_Default, DataType.I32_Default, DataType.I8_Default… 35 ….dtype_format(DataType.U8_Default, DataType.I32_Default, DataType.I32_Default, DataType.U8_Default… 36 ….dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Defau… 37 ….dtype_format(DataType.U32_Default, DataType.I32_Default, DataType.I32_Default, DataType.U32_Defau… 38 ….dtype_format(DataType.I16_Default, DataType.I32_Default, DataType.I32_Default, DataType.I16_Defau… 39 ….dtype_format(DataType.U16_Default, DataType.I32_Default, DataType.I32_Default, DataType.U16_Defau… 40 ….dtype_format(DataType.I64_Default, DataType.I32_Default, DataType.I32_Default, DataType.I64_Defau… [all …]
|
D | apply_adadelta.py | 17 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType 36 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, 37 DataType.F16_Default, DataType.F16_Default, DataType.F16_5HD, DataType.F16_5HD, 38 DataType.F16_5HD, DataType.F16_5HD) \ 39 .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_Default, 40 … DataType.F16_Default, DataType.F16_Default, DataType.F16_FracZ, DataType.F16_FracZ, 41 DataType.F16_FracZ, DataType.F16_FracZ) \ 42 ….dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16… 43 … DataType.F16_Default, DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, 44 DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \ [all …]
|
/third_party/mindspore/tests/ut/cpp/dataset/ |
D | datatype_test.cc | 33 uint8_t x = DataType::kTypeInfo[DataType::DE_BOOL].sizeInBytes_; in TEST_F() 34 DataType d = DataType(DataType::DE_BOOL); in TEST_F() 37 x = DataType::kTypeInfo[DataType::DE_INT8].sizeInBytes_; in TEST_F() 38 d = DataType(DataType::DE_INT8); in TEST_F() 41 x = DataType::kTypeInfo[DataType::DE_UINT8].sizeInBytes_; in TEST_F() 42 d = DataType(DataType::DE_UINT8); in TEST_F() 45 x = DataType::kTypeInfo[DataType::DE_INT16].sizeInBytes_; in TEST_F() 46 d = DataType(DataType::DE_INT16); in TEST_F() 49 x = DataType::kTypeInfo[DataType::DE_UINT16].sizeInBytes_; in TEST_F() 50 d = DataType(DataType::DE_UINT16); in TEST_F() [all …]
|