| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | arithmetic_compare_fp32.c | 59 #define ELEMENT_COMPARE_OPT(input0, input1, output, element_size, first_scalar, compare_func) \ argument
84 bool first_scalar) { in ElementOptEqualFp32()
93 bool first_scalar) { in ElementOptEqualInt32()
103 bool first_scalar) { in ElementOptNotEqualFp32()
112 bool first_scalar) { in ElementOptNotEqualInt32()
121 bool first_scalar) { in ElementOptNotEqualInt64()
130 …2(const float *input0, const float *input1, uint8_t *output, int element_size, bool first_scalar) { in ElementOptLessFp32()
139 bool first_scalar) { in ElementOptLessInt32()
149 bool first_scalar) { in ElementOptLessEqualFp32()
158 bool first_scalar) { in ElementOptLessEqualInt32()
[all …]
|
| D | arithmetic_fp32.c | 34 int ElementOptFloorMod(const float *in0, const float *in1, float *out, int size, bool first_scalar)… in ElementOptFloorMod()
61 …OptFloorModInt(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptFloorModInt()
87 int ElementOptMod(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptMod()
109 …ementOptModInt(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptModInt()
135 int ElementOptFloorDiv(const float *in0, const float *in1, float *out, int size, bool first_scalar)… in ElementOptFloorDiv()
167 …OptFloorDivInt(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptFloorDivInt()
199 …ElementOptLogicalAnd(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptLogicalAnd()
223 …tLogicalAndInt(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptLogicalAndInt()
247 …lementOptLogicalAndBool(const bool *in0, const bool *in1, bool *out, int size, bool first_scalar) { in ElementOptLogicalAndBool()
282 int ElementOptLogicalOr(const float *in0, const float *in1, float *out, int size, bool first_scalar… in ElementOptLogicalOr()
[all …]
|
| D | sub_fp32.c | 20 int ElementOptSub(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptSub()
36 …t(const float *in0, const float *in1, const float alpha, float *out, int size, bool first_scalar) { in ElementOptSubExt()
62 …ementOptSubInt(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptSubInt()
78 int ElementOptSubRelu(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptSubRelu()
94 int ElementOptSubRelu6(const float *in0, const float *in1, float *out, int size, bool first_scalar)… in ElementOptSubRelu6()
|
| D | mul_fp32.c | 89 int ElementOptMul(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptMul()
107 int ElementOptMulRelu(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptMulRelu()
123 int ElementOptMulRelu6(const float *in0, const float *in1, float *out, int size, bool first_scalar)… in ElementOptMulRelu6()
141 …ementOptMulInt(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptMulInt()
157 …tOptMulReluInt(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptMulReluInt()
173 …OptMulRelu6Int(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptMulRelu6Int()
|
| D | div_fp32.c | 22 int ElementOptDiv(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptDiv()
40 int ElementOptDivRelu(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptDivRelu()
60 int ElementOptDivRelu6(const float *in0, const float *in1, float *out, int size, bool first_scalar)… in ElementOptDivRelu6()
79 …ementOptDivInt(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptDivInt()
|
| D | add_fp32.c | 21 int ElementOptAdd(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptAdd()
37 …t(const float *in0, const float *in1, const float alpha, float *out, int size, bool first_scalar) { in ElementOptAddExt()
53 …ementOptAddInt(const int32_t *in0, const int32_t *in1, int32_t *out, int size, bool first_scalar) { in ElementOptAddInt()
69 int ElementOptAddRelu(const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptAddRelu()
85 int ElementOptAddRelu6(const float *in0, const float *in1, float *out, int size, bool first_scalar)… in ElementOptAddRelu6()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | arithmetic_fp16.c | 75 bool first_scalar) { in ElementOptMulFp16()
129 bool first_scalar) { in ElementOptMulReluFp16()
186 bool first_scalar) { in ElementOptMulRelu6Fp16()
245 bool first_scalar) { in ElementOptAddFp16()
305 bool first_scalar) { in ElementOptAddReluFp16()
371 bool first_scalar) { in ElementOptAddRelu6Fp16()
424 bool first_scalar) { in ElementOptSubFp16()
476 bool first_scalar) { in ElementOptSubReluFp16()
533 bool first_scalar) { in ElementOptSubRelu6Fp16()
586 bool first_scalar) { in ElementOptDivFp16()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | addn.c | 44 void AddNCompute(AddNStruct *addn, bool same_shape, bool first_scalar) { in AddNCompute()
|
| D | arithmetic_compare.c | 71 bool first_scalar = arithmetic_compare->arithmetic_.in_elements_num0_ == 1; in ArithmeticCompareExecute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/f16/ |
| D | arithmetic_compare_f16.c | 62 bool first_scalar = arithmetic_compare_f16->arithmetic_f16_.arithmetic_.in_elements_num0_ == 1; in ArithmeticCompareF16DoExecute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/sse/ |
| D | arithmetic_fp32_sse.h | 267 …calAndSSE(int index, const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptLogicalAndSSE()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx/ |
| D | arithmetic_fp32_avx.h | 267 …calAndAVX(int index, const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptLogicalAndAVX()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/neon/ |
| D | arithmetic_fp32_neon.h | 266 …alAndNEON(int index, const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptLogicalAndNEON()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx512/ |
| D | arithmetic_fp32_avx512.h | 267 …AndAVX512(int index, const float *in0, const float *in1, float *out, int size, bool first_scalar) { in ElementOptLogicalAndAVX512()
|