Home
last modified time | relevance | path

Searched defs:input1 (Results 1 – 25 of 179) sorted by relevance

12345678

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Darithmetic_compare_fp32.c22 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/mindspore/ccsrc/backend/kernel_compiler/cpu/
Deltwise_grad_cpu_kernel.cc29 void EltWiseGradCPUKernel<T>::ReluGrad(const T *input1, const T *input2, T *out, size_t start, size… in ReluGrad() argument
41 void EltWiseGradCPUKernel<T>::ReLU6Grad(const T *input1, const T *input2, T *out, size_t start, siz… in ReLU6Grad() argument
53 void EltWiseGradCPUKernel<T>::AbsGrad(const T *input1, const T *input2, T *out, size_t start, size_… in AbsGrad() argument
65 void EltWiseGradCPUKernel<T>::SigmoidGrad(const T *input1, const T *input2, T *out, size_t start, s… in SigmoidGrad() argument
77 void EltWiseGradCPUKernel<T>::SqrtGrad(const T *input1, const T *input2, T *out, size_t start, size… in SqrtGrad() argument
84 void EltWiseGradCPUKernel<T>::TanhGrad(const T *input1, const T *input2, T *out, size_t start, size… in TanhGrad() argument
96 void EltWiseGradCPUKernel<T>::GeluGrad(const T *input1, const T *input2, T *out, size_t start, size… in GeluGrad() argument
108 void EltWiseGradCPUKernel<T>::AsinGrad(const T *input1, const T *input2, T *out, size_t start, size… in AsinGrad() argument
129 void EltWiseGradCPUKernel<T>::ACosGrad(const T *input1, const T *input2, T *out, size_t start, size… in ACosGrad() argument
150 void EltWiseGradCPUKernel<T>::AtanGrad(const T *input1, const T *input2, T *out, size_t start, size… in AtanGrad() argument
[all …]
Darithmetic_logic_cpu_kernel.cc35 void ArithmeticLogicCPUKernel<T>::Less(const T *input1, const T *input2, bool *out) const { in Less() argument
61 void ArithmeticLogicCPUKernel<T>::Equal(const T *input1, const T *input2, bool *out) const { in Equal() argument
77 void ArithmeticLogicCPUKernel<T>::NotEqual(const T *input1, const T *input2, bool *out) const { in NotEqual() argument
93 void ArithmeticLogicCPUKernel<T>::LogicalAnd(const T *input1, const T *input2, bool *out) const { in LogicalAnd() argument
107 void ArithmeticLogicCPUKernel<T>::LogicalOr(const T *input1, const T *input2, bool *out) const { in LogicalOr() argument
121 void ArithmeticLogicCPUKernel<T>::Greater(const T *input1, const T *input2, bool *out) const { in Greater() argument
137 void ArithmeticLogicCPUKernel<T>::GreaterEqual(const T *input1, const T *input2, bool *out) const { in GreaterEqual() argument
153 void ArithmeticLogicCPUKernel<T>::LessEqual(const T *input1, const T *input2, bool *out) const { in LessEqual() argument
225 const auto *input1 = reinterpret_cast<T *>(inputs[0]->addr); in Launch() local
Darithmetic_cpu_kernel.cc38 void ElementRealDiv(const T *input1, const T *input2, T *out, size_t size, size_t delta_1, size_t d… in ElementRealDiv()
65 void ArithmeticCPUKernel<T>::AssignAdd(T *input1, const T *input2, T *out) const { in AssignAdd()
76 void ArithmeticCPUKernel<T>::Add(const T *input1, const T *input2, T *out) const { in Add() argument
90 void ArithmeticCPUKernel<T>::Sub(const T *input1, const T *input2, T *out) { in Sub()
125 void ArithmeticCPUKernel<T>::Mul(const T *input1, const T *input2, T *out) { in Mul()
159 void ArithmeticCPUKernel<T>::RealDiv(const T *input1, const T *input2, T *out) { in RealDiv()
210 void ArithmeticCPUKernel<T>::Div(const T *input1, const T *input2, T *out) const { in Div() argument
239 void ArithmeticCPUKernel<T>::FloorDiv(const T *input1, const T *input2, T *out) const { in FloorDiv() argument
268 void ArithmeticCPUKernel<T>::Mod(const T *input1, const T *input2, T *out) const { in Mod() argument
290 void ArithmeticCPUKernel<T>::FloorMod(const T *input1, const T *input2, T *out) const { in FloorMod() argument
[all …]
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dsoftmax_grad_ext_fusion.py43 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
Dunsorted_segment_sum_fission.py43 def before1(input0, input1): argument
48 def after1(input0, input1): argument
55 def before2(input0, input1): argument
60 def after2(input0, input1): argument
Dadam_apply_one_fusion_test.py49 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 …]
Dlayer_norm_beta_gamma_backprop_fusion_test.py40 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
Dadam_apply_one_with_decay_rule.py49 …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 …]
Daddn_fission_test.py39 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
Dconcat_fission_test.py37 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
Dlamb_next_mv_rule_test.py42 …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 …]
Dlamb_next_mv_with_decay_rule_test.py42 …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 …]
Dbatch_norm_grad_infer_fission_test.py42 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
Dclip_by_value_fusion.py42 def before1(input0, input1, input2): argument
48 def before2(input0, input1, input2): argument
54 def after(input0, input1, input2): argument
Dbatchnorm_to_bninfer.py41 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
Dbatchnormgrad_to_bninfergrad.py41 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
Dfused_batch_norm_fusion_test.py53 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
Dlamb_next_right_rule_test.py43 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
Dlamb_next_mv_with_decay_v1_rule.py44 …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
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Darithmetic_fp16.c55 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/tests/st/ops/custom_ops_tbe/
Dadd3_impl.py24 def add3_compute(input1, input2, const_bias): argument
47 def CusAdd3Impl(input1, inptu2, sum1, const_bias, kernel_name="CusAdd3Impl"): argument
Dcus_add3.py27 def infer_shape(self, input1, input2): argument
30 def infer_dtype(self, input1, input2): argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Darithmetic_int8.c52 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/gpu/cuda_impl/
Dequalcount_impl.cu20 __global__ void EqualCount(const int size, const T* input1, const T* input2, T* output) { in EqualCount()
33 void CalEqualCount(const int size, const T* input1, const T* input2, T* output, cudaStream_t cuda_s… in CalEqualCount()

12345678