Searched refs:thread_offset (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | quant_dtype_cast.cc | 78 int thread_offset = task_id * thread_n_stride_; in QuantDTypeCast() local
89 …ret = DoDequantizeInt8ToFp32(int8_ptr_ + thread_offset, float32_ptr_ + thread_offset, quant_arg.sc… in QuantDTypeCast()
96 …ret = DoQuantizeFp32ToInt8(float32_ptr_ + thread_offset, int8_ptr_ + thread_offset, quant_arg.scal… in QuantDTypeCast()
99 ret = Int8ToUInt8(int8_ptr_ + thread_offset, uint8_ptr_ + thread_offset, num_unit_thread); in QuantDTypeCast()
101 …ret = DoDequantizeUInt8ToFp32(uint8_ptr_ + thread_offset, float32_ptr_ + thread_offset, quant_arg.… in QuantDTypeCast()
104 …ret = DoQuantizeFp32ToUInt8(float32_ptr_ + thread_offset, uint8_ptr_ + thread_offset, quant_arg.sc… in QuantDTypeCast()
107 ret = UInt8ToInt8(uint8_ptr_ + thread_offset, int8_ptr_ + thread_offset, num_unit_thread); in QuantDTypeCast()
110 …ret = DoDequantizeInt8ToFp32(int8_ptr_ + thread_offset, float32_ptr_ + thread_offset, input_quant_… in QuantDTypeCast()
118 …ret = DoQuantizeFp32ToInt8(float32_ptr_ + thread_offset, int8_out_ptr_ + thread_offset, output_qua… in QuantDTypeCast()
|
| D | split_base.cc | 120 int thread_offset = task_id * thread_n_stride_; in Split() local
121 …o ret = DoSplit(input_ptr_, output_ptr_.data(), in_tensors_.front()->shape().data(), thread_offset, in Split()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | quant_dtype_cast_fp16.cc | 94 int thread_offset = task_id * thread_n_stride_; in QuantDTypeCast() local
106 …ret = DoDequantizeInt8ToFp16(int8_ptr_ + thread_offset, float16_ptr_ + thread_offset, quant_arg.sc… in QuantDTypeCast()
109 …ret = DoQuantizeFp16ToInt8(float16_ptr_ + thread_offset, int8_ptr_ + thread_offset, quant_arg.scal… in QuantDTypeCast()
116 …ret = DoDequantizeUInt8ToFp16(uint8_ptr_ + thread_offset, float16_ptr_ + thread_offset, quant_arg.… in QuantDTypeCast()
119 …ret = DoQuantizeFp16ToUInt8(float16_ptr_ + thread_offset, uint8_ptr_ + thread_offset, quant_arg.sc… in QuantDTypeCast()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | space_to_depth_fp32.cc | 71 int thread_offset = task_id * thread_h_stride_; in SpaceToDepth() local
79 … param->block_size_, thread_offset, thread_offset + num_unit_thread, sizeof(float)); in SpaceToDepth()
|
| D | glu_fp32.cc | 106 int thread_offset = task_id * thread_n_stride_; in Split() local
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 | detection_post_process_int8.cc | 31 int thread_offset = task_id * thread_n_stride_; in DequantizeInt8ToFp32() local
32 …int ret = DoDequantizeInt8ToFp32(data_int8_ + thread_offset, data_fp32_ + thread_offset, quant_par… in DequantizeInt8ToFp32()
|
| D | split_int8.cc | 75 int thread_offset = task_id * thread_n_stride_; in Split() local
78 …t = Int8DoSplit(input_ptr_, output_ptr_.data(), in_tensors_.front()->shape().data(), thread_offset, in Split()
|
| /third_party/mesa3d/src/broadcom/compiler/ |
| D | vir_register_allocate.c | 195 struct qreg thread_offset = in v3d_setup_spill_base() local
205 vir_ADD(c, thread_offset, element_offset), in v3d_setup_spill_base()
|