| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | arithmetic_compare_fp32.c | 22 int ElementEqualFp32(const float *input0, const float *input1, uint8_t *output, int element_size) { in ElementEqualFp32()
29 int ElementEqualInt32(const int32_t *input0, const int32_t *input1, uint8_t *output, int element_si… in ElementEqualInt32()
37 int ElementNotEqualFp32(const float *input0, const float *input1, uint8_t *output, int element_size… in ElementNotEqualFp32()
44 int ElementNotEqualInt32(const int32_t *input0, const int32_t *input1, uint8_t *output, int element… in ElementNotEqualInt32()
52 int ElementLessFp32(const float *input0, const float *input1, uint8_t *output, int element_size) { in ElementLessFp32()
59 int ElementLessInt32(const int32_t *input0, const int32_t *input1, uint8_t *output, int element_siz… in ElementLessInt32()
67 int ElementLessEqualFp32(const float *input0, const float *input1, uint8_t *output, int element_siz… in ElementLessEqualFp32()
74 int ElementLessEqualInt32(const int32_t *input0, const int32_t *input1, uint8_t *output, int elemen… in ElementLessEqualInt32()
82 int ElementGreaterFp32(const float *input0, const float *input1, uint8_t *output, int element_size)… in ElementGreaterFp32()
89 int ElementGreaterInt32(const int32_t *input0, const int32_t *input1, uint8_t *output, int element_… in ElementGreaterInt32()
[all …]
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | softmax_grad_ext_fusion.py | 43 def before(input0, input1, input2): argument
53 def after(input0, input1, input2): argument
64 def before(input0, input1, input2): argument
73 def after(input0, input1, input2): argument
84 def before(input0, input1, input2): argument
93 def after(input0, input1, input2): argument
|
| D | unsorted_segment_sum_fission.py | 43 def before1(input0, input1): argument
48 def after1(input0, input1): argument
55 def before2(input0, input1): argument
60 def after2(input0, input1): argument
|
| D | adam_apply_one_fusion_test.py | 49 def before(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
67 … def before_cond1(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
85 … def before_cond2(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
103 … def before_cond3(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
121 … def before_cond4(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
139 def after(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
153 def before(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
177 … def before_cond1(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
201 … def before_cond2(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
225 … def before_cond3(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, add2_y): argument
[all …]
|
| D | layer_norm_beta_gamma_backprop_fusion_test.py | 40 def before(input0, input1, input2, input3): argument
48 def before_unmatched_inputs_size(input0, input1, input2): argument
56 def before_unmatched_outputs_size(input0, input1, input2, input3): argument
62 def after(input0, input1, input2, input3): argument
|
| D | adam_apply_one_with_decay_rule.py | 49 …def before_cond1(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, a… argument
67 …def before_cond2(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, a… argument
85 …def before_cond3(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, a… argument
103 …def before_cond4(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, a… argument
121 …def before_cond5(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, a… argument
139 … def after(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, add2_y): argument
154 …def before_cond1(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, a… argument
178 …def before_cond2(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, a… argument
202 …def before_cond3(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, a… argument
226 …def before_cond4(input0, input1, input2, input3, input4, mul0_x, mul1_x, mul2_x, mul3_x, mul4_x, a… argument
[all …]
|
| D | addn_fission_test.py | 39 def before(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
43 def after_divided_by_2(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
54 def after_divided_by_3(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
61 def after_divided_by_4(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
67 def after_divided_by_8(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
72 def after_divided_by_9(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
|
| D | concat_fission_test.py | 37 def before(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
41 def after_divided_by_2(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
52 def after_divided_by_3(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
59 def after_divided_by_4(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
65 def after_divided_by_8(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
70 def after_divided_by_9(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
|
| D | lamb_next_mv_rule_test.py | 42 …def before(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_… argument
65 …def after(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_s… argument
76 …def before_unmatched_real_div4(input0, input1, input2, input3, input4, input5, input6, constant_mu… argument
100 …def before_unmatched_real_div0(input0, input1, input2, input3, input4, input5, input6, constant_mu… argument
124 …def before_unmatched_real_div1(input0, input1, input2, input3, input4, input5, input6, constant_mu… argument
148 …def before_unmatched_real_div2(input0, input1, input2, input3, input4, input5, input6, constant_mu… argument
177 …def before(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_… argument
200 …def after(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_s… argument
211 …def un_match(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul… argument
240 …def before(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_… argument
[all …]
|
| D | lamb_next_mv_with_decay_rule_test.py | 42 …def before(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_… argument
66 …def after(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_s… argument
78 … def before_unmatched_add3(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, argument
103 … def before_unmatched_mul4(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, argument
128 …def before_unmatched_real_div0(input0, input1, input2, input3, input4, input5, input6, constant_mu… argument
153 …def before_unmatched_real_div1(input0, input1, input2, input3, input4, input5, input6, constant_mu… argument
183 …def before(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_… argument
207 …def after(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_s… argument
219 …def un_match(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul… argument
249 …def before(input0, input1, input2, input3, input4, input5, input6, constant_mul0_x, constant_mul1_… argument
[all …]
|
| D | batch_norm_grad_infer_fission_test.py | 42 def before(input0, input1, input2, input3, input4, input5): argument
49 def before_is_training(input0, input1, input2, input3, input4, input5): argument
56 def before_output3_not_null(input0, input1, input2, input3, input4, input5): argument
63 def after(input0, input1, input2, input3, input4, input5): argument
|
| D | clip_by_value_fusion.py | 42 def before1(input0, input1, input2): argument
48 def before2(input0, input1, input2): argument
54 def after(input0, input1, input2): argument
|
| D | batchnorm_to_bninfer.py | 41 def before(input0, input1, input2, input3, input4): argument
47 def after(input0, input1, input2, input3, input4): argument
52 def no_fusion(input0, input1, input2, input3, input4): argument
|
| D | batchnormgrad_to_bninfergrad.py | 41 def before(input0, input1, input2, input3, input4, input5): argument
47 def after(input0, input1, input2, input3, input4, input5): argument
52 def no_fusion(input0, input1, input2, input3, input4, input5): argument
|
| D | fused_batch_norm_fusion_test.py | 53 def before(input0, input1, input2, input3, input4, var0, var1): argument
70 def before_mix_precision0(input0, input1, input2, input3, input4, var0, var1): argument
87 def before_mix_precision1(input0, input1, input2, input3, input4, var0, var1): argument
104 def after(input0, input1, input2, input3, input4, var0, var1): argument
|
| D | lamb_next_right_rule_test.py | 43 def before(input0, input1, mul2_x, mul3_x, true_div1_recip, add2_y): argument
56 def before_unmatched(input0, input1, mul2_x, mul3_x, true_div1_recip, add2_y): argument
69 def after(input0, input1, mul2_x, mul3_x, true_div1_recip, add2_y): argument
|
| D | reshape_transpose_fusion_test.py | 39 def before(input0): argument
45 def after(input0): argument
|
| D | lamb_next_mv_with_decay_v1_rule.py | 44 …def before(input0, input1, input2, input3, input4, input5, input6, mul0_x, mul1_sub, mul2_x, mul3_… argument
67 …def after(input0, input1, input2, input3, input4, input5, input6, mul0_x, mul1_sub, mul2_x, mul3_s… argument
79 …def no_match1(input0, input1, input2, input3, input4, input5, input6, mul0_x, mul1_sub, mul2_x, mu… argument
103 …def no_match2(input0, input1, input2, input3, input4, input5, input6, mul0_x, mul1_sub, mul2_x, mu… argument
127 …def no_match3(input0, input1, input2, input3, input4, input5, input6, mul0_x, mul1_sub, mul2_x, mu… argument
|
| D | stack_fission_test.py | 40 def before(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
44 def after_divided_by_3(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
51 def after_divided_by_4(input0, input1, input2, input3, input4, input5, input6, input7, input8): argument
|
| D | matmul_biasadd_fusion_test.py | 38 def before(input0, input1, input2): argument
44 def after(input0, input1, input2): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | arithmetic_fp16.c | 55 int ElementMulFp16(const float16_t *input0, const float16_t *input1, float16_t *output, int element… in ElementMulFp16()
71 int ElementOptMulFp16(const float16_t *input0, const float16_t *input1, float16_t *output, int elem… in ElementOptMulFp16()
104 int ElementMulReluFp16(const float16_t *input0, const float16_t *input1, float16_t *output, int ele… in ElementMulReluFp16()
125 int ElementOptMulReluFp16(const float16_t *input0, const float16_t *input1, float16_t *output, int … in ElementOptMulReluFp16()
163 int ElementMulRelu6Fp16(const float16_t *input0, const float16_t *input1, float16_t *output, int el… in ElementMulRelu6Fp16()
182 int ElementOptMulRelu6Fp16(const float16_t *input0, const float16_t *input1, float16_t *output, int… in ElementOptMulRelu6Fp16()
219 int ElementAddFp16(const float16_t *input0, const float16_t *input1, float16_t *output, int element… in ElementAddFp16()
241 int ElementOptAddFp16(const float16_t *input0, const float16_t *input1, float16_t *output, int elem… in ElementOptAddFp16()
274 int ElementAddReluFp16(const float16_t *input0, const float16_t *input1, float16_t *output, int ele… in ElementAddReluFp16()
301 int ElementOptAddReluFp16(const float16_t *input0, const float16_t *input1, float16_t *output, int … in ElementOptAddReluFp16()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | arithmetic_int8.c | 52 int ElementNotEqualInt8(int8_t *input0, int8_t *input1, uint8_t *output, int element_size, in ElementNotEqualInt8()
70 int ElementEqualInt8(int8_t *input0, int8_t *input1, uint8_t *output, int element_size, ArithmeticQ… in ElementEqualInt8()
86 int ElementLessInt8(int8_t *input0, int8_t *input1, uint8_t *output, int element_size, ArithmeticQu… in ElementLessInt8()
98 int ElementLessEqualInt8(int8_t *input0, int8_t *input1, uint8_t *output, int element_size, in ElementLessEqualInt8()
111 int ElementGreaterInt8(int8_t *input0, int8_t *input1, uint8_t *output, int element_size, in ElementGreaterInt8()
124 int ElementGreaterEqualInt8(int8_t *input0, int8_t *input1, uint8_t *output, int element_size, in ElementGreaterEqualInt8()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/control/ |
| D | tensorlist_setitem_infer.c | 20 int PreJudge(const TensorC *get_index, TensorListC *input0, const TensorC *value_tensor) { in PreJudge()
44 TensorListC *input0 = (TensorListC *)(inputs[0]); in TensorListSetItemInferShape() local
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_lsh_projection.cc | 39 …auto input0 = TensorConstructUtils::CreateOnesTensor(kNumberTypeInt32, std::vector<int64_t>{4, 3}); in TEST_F() local
68 …auto input0 = TensorConstructUtils::CreateOnesTensor(kNumberTypeInt32, std::vector<int64_t>{4, 3}); in TEST_F() local
97 …auto input0 = TensorConstructUtils::CreateOnesTensor(kNumberTypeInt32, std::vector<int64_t>{4, 3}); in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | full_connection_infer.c | 20 …lConnectionInferPreJudge(const MatMulParameter *param, size_t inputs_size, const TensorC *input0) { in FullConnectionInferPreJudge()
37 const TensorC *input0 = inputs[0]; in FullConnectionInferShape() local
|