| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/ |
| D | quant_dtype_cast_coder.cc | 47 int num_unit_thread = input_tensor_->ElementsNum(); in DoCode() local
60 num_unit_thread); in DoCode()
67 num_unit_thread, from_uint8_src); in DoCode()
69 code.CodeFunction("Int8ToUInt8", input_tensor_, output_tensor_, num_unit_thread); in DoCode()
72 num_unit_thread); in DoCode()
75 num_unit_thread); in DoCode()
77 code.CodeFunction("UInt8ToInt8", input_tensor_, output_tensor_, num_unit_thread); in DoCode()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | quant_dtype_cast.cc | 74 int num_unit_thread = MSMIN(thread_n_stride_, num_unit_ - task_id * thread_n_stride_); in QuantDTypeCast() local
75 if (num_unit_thread <= 0) { in QuantDTypeCast()
90 quant_arg.zeroPoint, num_unit_thread); in QuantDTypeCast()
97 quant_arg.zeroPoint, num_unit_thread, from_uint8_src); in QuantDTypeCast()
99 ret = Int8ToUInt8(int8_ptr_ + thread_offset, uint8_ptr_ + thread_offset, num_unit_thread); in QuantDTypeCast()
102 quant_arg.zeroPoint, num_unit_thread); in QuantDTypeCast()
105 quant_arg.zeroPoint, num_unit_thread); in QuantDTypeCast()
107 ret = UInt8ToInt8(uint8_ptr_ + thread_offset, int8_ptr_ + thread_offset, num_unit_thread); in QuantDTypeCast()
111 input_quant_arg.zeroPoint, num_unit_thread); in QuantDTypeCast()
119 output_quant_arg.zeroPoint, num_unit_thread, from_uint8_src); in QuantDTypeCast()
|
| D | split_base.cc | 115 int num_unit_thread = MSMIN(thread_n_stride_, num_unit_ - task_id * thread_n_stride_); in Split() local
116 if (num_unit_thread <= 0) { in Split()
122 num_unit_thread, param, lite::DataTypeSize(in_tensors_.front()->data_type())); in Split()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | quant_dtype_cast_fp16.cc | 90 int num_unit_thread = MSMIN(thread_n_stride_, num_unit_ - task_id * thread_n_stride_); in QuantDTypeCast() local
91 if (num_unit_thread <= 0) { in QuantDTypeCast()
107 quant_arg.zeroPoint, num_unit_thread); in QuantDTypeCast()
110 quant_arg.zeroPoint, num_unit_thread); in QuantDTypeCast()
117 quant_arg.zeroPoint, num_unit_thread); in QuantDTypeCast()
120 quant_arg.zeroPoint, num_unit_thread); in QuantDTypeCast()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | space_to_depth_fp32.cc | 67 int num_unit_thread = MSMIN(thread_h_stride_, num_unit_ - task_id * thread_h_stride_); in SpaceToDepth() local
68 if (num_unit_thread <= 0) { in SpaceToDepth()
79 … param->block_size_, thread_offset, thread_offset + num_unit_thread, sizeof(float)); in SpaceToDepth()
|
| D | scatter_nd_fp32.cc | 133 int num_unit_thread = MSMIN(thread_n_stride_, num_unit_ - task_id * thread_n_stride_); in ScatterND() local
134 if (num_unit_thread <= 0) { in ScatterND()
140 unit_size_, num_unit_thread); in ScatterND()
|
| D | scatter_nd_update_fp32.cc | 112 int num_unit_thread = MSMIN(thread_n_stride_, num_unit_ - task_id * thread_n_stride_); in ScatterNdUpdate() local
113 if (num_unit_thread <= 0) { in ScatterNdUpdate()
118 unit_size_, num_unit_thread); in ScatterNdUpdate()
|
| D | glu_fp32.cc | 102 int num_unit_thread = MSMIN(thread_n_stride_, num_unit_ - task_id * thread_n_stride_); in Split() local
103 if (num_unit_thread <= 0) { in Split()
109 …thread_offset, num_unit_thread, &split_param_, lite::DataTypeSize(in_tensors_.front()->data_type()… in Split()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | split_int8.cc | 70 int num_unit_thread = MSMIN(thread_n_stride_, num_unit_ - task_id * thread_n_stride_); in Split() local
71 if (num_unit_thread <= 0) { in Split()
79 num_unit_thread, param); in Split()
|
| D | detection_post_process_int8.cc | 30 int num_unit_thread = MSMIN(thread_n_stride_, quant_size_ - task_id * thread_n_stride_); in DequantizeInt8ToFp32() local
33 quant_param_.zeroPoint, num_unit_thread); in DequantizeInt8ToFp32()
|