Home
last modified time | relevance | path

Searched refs:src_out (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Ddeconvolution_int8.cc278 auto src_out = reinterpret_cast<int8_t *>(output_tensor->data()); in Run() local
280 CHECK_NULL_RETURN(src_out); in Run()
292 output_ptr_ = src_out + batch_index * matmul_param_->col_; in Run()
Dconvolution_1x1_int8.cc557 auto src_out = reinterpret_cast<int8_t *>(out_tensors_[0]->data()); in Run() local
558 CHECK_NULL_RETURN(src_out); in Run()
562 src_out + batch_index * matmul_param_->row_ * matmul_param_->col_); in Run()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_1x1_fp32.cc226 auto src_out = reinterpret_cast<float *>(out_tensors_[0]->data()); in Run() local
228 CHECK_NULL_RETURN(src_out); in Run()
242 output_ptr_ = src_out + batch_index * matmul_param_->row_ * matmul_param_->col_; in Run()
Ddeconvolution_fp32.cc255 float *src_out = reinterpret_cast<float *>(output_tensor->data()); in Run() local
257 CHECK_NULL_RETURN(src_out); in Run()
268 output_ptr_ = src_out + batch_index * output_plane_ * conv_param_->output_channel_; in Run()
Ddeconvolution_winograd_fp32.cc490 float *src_out = reinterpret_cast<float *>(output_tensor->data()); in Run() local
492 CHECK_NULL_RETURN(src_out); in Run()
496 …nhwc_output_ = src_out + batch_index * deconv_param_->output_plane_ * conv_param_->output_channel_; in Run()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_1x1_int8_coder.cc84 std::string src_out = allocator_->GetRuntimeAddr(output_tensor_) + " + batch_index * " + in DoCode() local
86 code.CodeFunction("Pre1x1Trans", kRunArgsAddr, src_in, src_out); in DoCode()