/third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
D | apply_adam.py | 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 …ormat(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, 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, 49 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, 50 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 51 DataType.F16_Default, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, [all …]
|
D | apply_ada_max.py | 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, 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, 44 …ormat(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, 45 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 47 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 48 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 49 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
|
D | apply_centered_rms_prop.py | 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, 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, 48 … DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 49 … DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, 51 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 52 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 53 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 54 DataType.F16_Default) \
|
D | apply_power_sign.py | 35 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, DataType.F16_Default, 36 DataType.F16_Default, DataType.F16_Default, DataType.F16_5HD, DataType.F16_5HD, 38 …_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_Default, 39 … DataType.F16_Default, DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, 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) \ 44 … .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_Default, DataType.F16_Default, 45 … DataType.F16_Default, DataType.F16_Default, DataType.F16_FracZ, DataType.F16_FracZ,
|
D | apply_add_sign.py | 35 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, DataType.F16_Default, 36 DataType.F16_Default, DataType.F16_Default, DataType.F16_5HD, DataType.F16_5HD, 38 …_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_Default, 39 … DataType.F16_Default, DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, 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) \ 44 … .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_Default, DataType.F16_Default, 45 … DataType.F16_Default, DataType.F16_Default, DataType.F16_FracZ, DataType.F16_FracZ,
|
D | lamb_next_mv.py | 43 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 44 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 45 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 46 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 47 DataType.F16_Default) \
|
D | lamb_next_mv_with_decay.py | 43 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 44 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 45 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 46 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 47 DataType.F16_Default) \
|
D | apply_ftrl.py | 38 … DataType.F16_5HD, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 41 … DataType.F16_FracZ, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 44 … DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 46 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 47 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 48 DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
|
D | fused_mul_apply_momentum_extern.py | 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, 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, 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) \ 46 .dtype_format(DataType.F32_FracZ, DataType.F16_FracZ, DataType.F16_Default, DataType.F16_FracZ, 47 … DataType.F16_Default, DataType.F16_Default, DataType.F16_FracZ, DataType.F32_FracZ, 56 … DataType.F32_Default, DataType.F32_Default, DataType.F16_Default, DataType.F32_Default, [all …]
|
D | apply_adadelta.py | 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, 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, 42 …ormat(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, 43 … DataType.F16_Default, DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, 45 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 46 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 47 DataType.F16_Default, DataType.F16_Default) \
|
D | lamb_apply_optimizer_assign.py | 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, DataType.F16_Default, DataType.F16_Default) \
|
D | apply_adagrad_d_a.py | 39 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, 42 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, 45 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, 47 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 48 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, 49 DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
|
D | apply_proximal_adagrad.py | 35 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, DataType.F16_Default, 36 DataType.F16_Default, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \ 37 …_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_Default, 38 … DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \ 39 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 40 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \ 41 … .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_Default, DataType.F16_Default, 42 … DataType.F16_Default, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ) \
|
D | confusion_matrix.py | 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 …pe_format(DataType.F32_Default, DataType.F32_Default, DataType.F16_Default, DataType.F16_Default) \ 42 …pe_format(DataType.I32_Default, DataType.I32_Default, DataType.F16_Default, DataType.F16_Default) \ 47 …type_format(DataType.I8_Default, DataType.I8_Default, DataType.F16_Default, DataType.F16_Default) \ 52 …type_format(DataType.U8_Default, DataType.U8_Default, DataType.F16_Default, DataType.F16_Default) \
|
D | adam_apply_one_with_decay.py | 40 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 41 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 42 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 43 DataType.F16_Default, DataType.F16_Default) \
|
D | adam_apply_one_with_decay_assign.py | 40 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 41 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 42 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 43 DataType.F16_Default, DataType.F16_Default) \
|
D | apply_proximal_gradient_descent.py | 32 … .dtype_format(DataType.F16_5HD, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 34 ….dtype_format(DataType.F16_FracZ, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 36 ….dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_Default, DataType.F16_Def… 38 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 39 DataType.F16_Default, DataType.F16_Default) \
|
D | adam_apply_one_assign.py | 39 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 40 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 41 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 42 DataType.F16_Default) \
|
D | apply_momentum.py | 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) \ 50 ….dtype_format(DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0, DataType.F16_Default, DataType.F16_NDC… 51 DataType.F16_Default, DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0) \ [all …]
|
D | fused_mul_apply_momentum.py | 35 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, DataType.F16_5HD, 36 DataType.F16_Default, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \ 37 ….dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, DataType.F16_C… 38 … DataType.F16_Default, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \ 39 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 40 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \ 41 .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_Default, DataType.F16_FracZ, 42 … DataType.F16_Default, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ) \
|
D | fused_mul_add_n_l2loss.py | 31 .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_Default, 32 DataType.F16_5HD, DataType.F16_Default) \ 33 .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_Default, 34 DataType.F16_C1HWNCoC0, DataType.F16_Default) \ 35 .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 36 DataType.F16_Default, DataType.F16_Default) \ 37 .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_Default, 38 DataType.F16_FracZ, DataType.F16_Default) \
|
D | lamb_update_with_lr.py | 36 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 37 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, 38 DataType.F16_Default, DataType.F16_Default) \
|
D | lamb_update_with_lr_v2.py | 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, DataType.F16_Default) \
|
D | lamb_next_right.py | 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, DataType.F16_Default) \
|
/third_party/mindspore/mindspore/ops/_op_impl/cpu/ |
D | layer_norm_grad.py | 28 ….dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Defau… 29 … DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
|