/third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
D | apply_adam.py | 57 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 58 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 59 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 60 DataType.F32_Default) \ 61 …ormat(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_Default, 62 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 63 … DataType.F32_Default, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, 65 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, 66 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 67 DataType.F32_Default, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, [all …]
|
D | apply_ada_max.py | 50 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, 51 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 53 .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_Default, 54 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 56 …ormat(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_Default, 57 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 59 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 60 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 61 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
|
D | lars_update.py | 36 … .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_Default, DataType.F32_Default, 37 DataType.F32_Default, DataType.F32_Default, DataType.F32_FracZ) \ 38 …_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_Default, DataType.F32_Default, 39 DataType.F32_Default, DataType.F32_Default, DataType.F32_C1HWNCoC0) \ 40 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, DataType.F32_Default, 41 DataType.F32_Default, DataType.F32_Default, DataType.F32_5HD) \ 42 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 43 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
|
D | apply_centered_rms_prop.py | 56 … DataType.F32_5HD, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 57 DataType.F32_Default, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, 60 … DataType.F32_FracZ, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 61 DataType.F32_Default, DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ, 64 … DataType.F32_C1HWNCoC0, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 65 … DataType.F32_Default, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, 67 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 68 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 69 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 70 DataType.F32_Default) \
|
D | sparse_apply_r_m_s_prop.py | 39 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, 42 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 43 … DataType.F32_Default, DataType.I32_Default, DataType.F32_Default, DataType.F32_Default, 44 DataType.F32_Default) \ 45 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, 48 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 49 … DataType.F32_Default, DataType.I64_Default, DataType.F32_Default, DataType.F32_Default, 50 DataType.F32_Default) \
|
D | batchnorm.py | 39 ….dtype_format(DataType.F16_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 40 … DataType.F32_Default, DataType.F16_Default, DataType.F32_Default, DataType.F32_Default, 41 DataType.F32_Default, DataType.F32_Default) \ 45 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 46 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 47 DataType.F32_Default, DataType.F32_Default) \
|
D | apply_power_sign.py | 47 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, DataType.F32_Default, 48 DataType.F32_Default, DataType.F32_Default, DataType.F32_5HD, DataType.F32_5HD, 50 …_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_Default, DataType.F32_Default, 51 … DataType.F32_Default, DataType.F32_Default, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, 53 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 54 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 55 DataType.F32_Default) \ 56 … .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_Default, DataType.F32_Default, 57 … DataType.F32_Default, DataType.F32_Default, DataType.F32_FracZ, DataType.F32_FracZ,
|
D | apply_add_sign.py | 47 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, DataType.F32_Default, 48 DataType.F32_Default, DataType.F32_Default, DataType.F32_5HD, DataType.F32_5HD, 50 …_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_Default, DataType.F32_Default, 51 … DataType.F32_Default, DataType.F32_Default, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, 53 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 54 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 55 DataType.F32_Default) \ 56 … .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_Default, DataType.F32_Default, 57 … DataType.F32_Default, DataType.F32_Default, DataType.F32_FracZ, DataType.F32_FracZ,
|
D | batchnorm_grad.py | 39 …type_format(DataType.F16_Default, DataType.F16_Default, DataType.F32_Default, DataType.F32_Default, 40 DataType.F32_Default, DataType.F16_Default, DataType.F32_Default, 41 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \ 45 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 46 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 47 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
|
D | lamb_next_mv.py | 48 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 49 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 50 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 51 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 52 DataType.F32_Default) \
|
D | lamb_next_mv_with_decay.py | 48 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 49 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 50 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 51 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 52 DataType.F32_Default) \
|
D | sgd.py | 42 ….dtype_format(DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0, DataType.F32_Default, DataType.F16_NDC… 43 DataType.F32_Default, DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0) \ 44 ….dtype_format(DataType.F16_FRACTAL_Z_3D, DataType.F16_FRACTAL_Z_3D, DataType.F32_Default, DataType… 45 DataType.F32_Default, DataType.F16_FRACTAL_Z_3D, DataType.F16_FRACTAL_Z_3D) \ 46 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, DataType.F32_5HD, 47 DataType.F32_Default, DataType.F32_5HD, DataType.F32_5HD) \ 48 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 49 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \ 50 .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_Default, DataType.F32_FracZ, 51 DataType.F32_Default, DataType.F32_FracZ, DataType.F32_FracZ) \ [all …]
|
D | apply_ftrl.py | 50 … DataType.F32_5HD, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 53 … DataType.F32_FracZ, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 56 … DataType.F32_C1HWNCoC0, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 58 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 59 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 60 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
|
D | fused_mul_apply_momentum_extern.py | 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, 49 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, DataType.F32_5HD, 50 DataType.F32_Default, DataType.F32_Default, DataType.F16_5HD, DataType.F32_5HD, 52 ….dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_Default, DataType.F32_C… 53 … DataType.F32_Default, DataType.F32_Default, DataType.F16_C1HWNCoC0, DataType.F32_C1HWNCoC0, 55 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 56 … DataType.F32_Default, DataType.F32_Default, DataType.F16_Default, DataType.F32_Default, 57 DataType.F16_Default, DataType.F32_Default) \ 58 .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_Default, DataType.F32_FracZ, [all …]
|
D | apply_adadelta.py | 48 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, 49 DataType.F32_Default, DataType.F32_Default, DataType.F32_5HD, DataType.F32_5HD, 51 .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_Default, 52 … DataType.F32_Default, DataType.F32_Default, DataType.F32_FracZ, DataType.F32_FracZ, 54 …ormat(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_Default, 55 … DataType.F32_Default, DataType.F32_Default, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, 57 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 58 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 59 DataType.F32_Default, DataType.F32_Default) \
|
D | lamb_apply_optimizer_assign.py | 45 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 46 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 47 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 48 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
|
D | apply_adagrad_d_a.py | 51 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.I32_Default, 54 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.I32_Default, 57 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.I32_Default, 59 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 60 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.I32_Default, 61 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
|
D | apply_proximal_adagrad.py | 43 .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_Default, DataType.F32_Default, 44 DataType.F32_Default, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \ 45 …_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_Default, DataType.F32_Default, 46 … DataType.F32_Default, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0) \ 47 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 48 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \ 49 … .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_Default, DataType.F32_Default, 50 … DataType.F32_Default, DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ) \
|
D | confusion_matrix.py | 33 …pe_format(DataType.F16_Default, DataType.F16_Default, DataType.F32_Default, DataType.F32_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… 41 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.U8_Default, DataType.U8_Default… 43 …pe_format(DataType.I32_Default, DataType.I32_Default, DataType.F32_Default, DataType.F32_Default) \ 48 …type_format(DataType.I8_Default, DataType.I8_Default, DataType.F32_Default, DataType.F32_Default) \ 53 …type_format(DataType.U8_Default, DataType.U8_Default, DataType.F32_Default, DataType.F32_Default) \
|
D | sparse_apply_ftrl_d.py | 41 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 42 … DataType.I32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \ 45 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 46 … DataType.I64_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
|
/third_party/mindspore/mindspore/ops/_op_impl/cpu/ |
D | adam.py | 33 .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 34 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 35 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 36 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 37 DataType.F32_Default) \ 39 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 40 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
|
D | adam_weight_decay.py | 32 .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 33 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 34 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 35 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \ 37 DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 38 DataType.F32_Default, DataType.F32_Default, DataType.F16_Default,
|
/third_party/mindspore/mindspore/ops/_op_impl/aicpu/ |
D | fused_sparse_lazy_adam.py | 37 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 38 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 39 … DataType.F32_Default, DataType.F32_Default, DataType.I32_Default, DataType.F32_Default, 40 DataType.F32_Default, DataType.F32_Default) \
|
D | fused_sparse_adam.py | 37 ….dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Defau… 38 … DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, 39 … DataType.F32_Default, DataType.F32_Default, DataType.I32_Default, DataType.F32_Default, 40 DataType.F32_Default, DataType.F32_Default) \
|
D | crop_and_resize.py | 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… 36 DataType.F32_Default) \ [all …]
|