Searched defs:ele_num (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | activation_fp16.c | 22 int ReluFp16(const float16_t *src, float16_t *dst, int ele_num) { in ReluFp16()
38 int Relu6Fp16(const float16_t *data, float16_t *dst, int ele_num) { in Relu6Fp16()
57 int LReluFp16(const float16_t *src, float16_t *dst, int ele_num, float16_t alpha) { in LReluFp16()
74 int SigmoidFp16(const float16_t *src, float16_t *dst, int ele_num) { in SigmoidFp16()
105 int TanhFp16(const float16_t *src, float16_t *dst, int ele_num) { in TanhFp16()
141 int HSwishFp16(const float16_t *src, float16_t *dst, int ele_num) { in HSwishFp16()
162 int SwishFp16(const float16_t *src, float16_t *dst, int ele_num) { in SwishFp16()
174 int HSigmoidFp16(const float16_t *src, float16_t *dst, int ele_num) { in HSigmoidFp16()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | common_fp16.cc | 28 auto ele_num = input->ElementsNum(); in ConvertInputFp32toFp16() local
46 auto ele_num = output->ElementsNum(); in MallocOutputFp16() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | convert_fp32_to_fp16_pass.cc | 45 auto ele_num = lite::GetShapeSize(tensor->dims); in Run() local
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | tensorlist.h | 74 void set_max_elements_num(int ele_num) { max_elements_num_ = ele_num; } in set_max_elements_num()
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/ |
| D | acl_model_multi.cc | 217 int64_t ele_num = std::accumulate(shape.begin(), shape.end(), 1, std::multiplies<int64_t>()); in SetOutput() local
|