Searched refs:out_ptr_ (Results 1 – 18 of 18) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | transpose_int8.cc | 76 CHECK_NULL_RETURN(out_ptr_); in DoTranspose()
80 …TransposeDimsInt8(in_ptr_, out_ptr_, out_shape_, transpose_param_, task_id, op_parameter_->thread_… in DoTranspose()
112 out_ptr_ = reinterpret_cast<int8_t *>(out_tensor->data()); in Run()
113 CHECK_NULL_RETURN(out_ptr_); in Run()
116 NHNCTransposeFunc_(in_ptr_, out_ptr_, nhnc_param_[0], nhnc_param_[1], nhnc_param_[2]); in Run()
133 return DoTransposeInt8(in_ptr_, out_ptr_, out_shape_, transpose_param_); in Run()
|
| D | gatherNd_int8.cc | 125 …auto ret = GatherNdInt8(in_ptr_, out_ptr_ + offset * area_, in_offset_ + offset, area_, count, par… in DoGatherNd()
145 out_ptr_ = reinterpret_cast<int8_t *>(out_tensors_.front()->data()); in Run()
147 CHECK_NULL_RETURN(out_ptr_); in Run()
|
| D | tanh_int8.cc | 53 int8_t *cur_output = out_ptr_ + task_id * thread_stride_; in DoActivation()
71 out_ptr_ = reinterpret_cast<int8_t *>(out_tensors_.at(0)->data()); in Run()
|
| D | unsqueeze_int8.cc | 92 out_ptr_ = reinterpret_cast<float *>(out_tensors_.at(0)->data()); in Run()
93 CHECK_NULL_RETURN(out_ptr_); in Run()
|
| D | arithmetic_self_int8.cc | 93 auto ret = arithmeticSelf_run_(in_ptr_ + offset, out_ptr_ + offset, size, para_->quant_arg_); in DoArithmeticSelf()
109 out_ptr_ = reinterpret_cast<int8_t *>(out_tensor->MutableData()); in Run()
|
| D | tanh_int8.h | 45 int8_t *out_ptr_{nullptr};
|
| D | gatherNd_int8.h | 46 int8_t *out_ptr_ = nullptr; variable
|
| D | unsqueeze_int8.h | 49 float *out_ptr_{nullptr};
|
| D | transpose_int8.h | 52 int8_t *out_ptr_ = nullptr; variable
|
| D | arithmetic_self_int8.h | 105 int8_t *out_ptr_{nullptr};
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | fill_fp32.cc | 60 ret = FillFp32(out_ptr_ + offset, size, src_data_); in DoFill()
93 out_ptr_ = reinterpret_cast<float *>(output->MutableData()); in Run()
94 CHECK_NULL_RETURN(out_ptr_); in Run()
|
| D | roi_pooling_fp32.cc | 86 CHECK_NULL_RETURN(out_ptr_); in DoExecute()
91 auto ret = ROIPooling(in_ptr_, out_ptr_, roi_ptr_, max_c_, task_id, param_); in DoExecute()
111 out_ptr_ = reinterpret_cast<float *>(out_tensors_.front()->MutableData()); in Run()
|
| D | gatherNd_fp32.cc | 111 auto ret = GatherNd(in_ptr_, out_ptr_ + offset * area_, in_offset_ + offset, area_, count); in DoGatherNd()
131 out_ptr_ = reinterpret_cast<float *>(out_tensors_.front()->data()); in Run()
133 CHECK_NULL_RETURN(out_ptr_); in Run()
|
| D | reverse_fp32.cc | 126 auto ret = Reverse(in_ptr_ + offset, out_ptr_, thread_sz_stride_, tmp_ + offset); in DoReverse()
139 out_ptr_ = reinterpret_cast<float *>(out_tensors_[0]->MutableData()); in Run()
140 CHECK_NULL_RETURN(out_ptr_); in Run()
|
| D | fill_fp32.h | 44 float *out_ptr_ = nullptr; variable
|
| D | roi_pooling_fp32.h | 45 float *out_ptr_ = nullptr; variable
|
| D | gatherNd_fp32.h | 50 float *out_ptr_ = nullptr; variable
|
| D | reverse_fp32.h | 54 float *out_ptr_ = nullptr; variable
|