Searched refs:crop_para_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | crop_int8.cc | 35 crop_para_->quant_arg.in_args_.scale_ = in_quant_args.front().scale; in Init()
36 crop_para_->quant_arg.in_args_.zp_ = in_quant_args.front().zeroPoint; in Init()
40 crop_para_->quant_arg.out_args_.scale_ = out_quant_args.front().scale; in Init()
41 crop_para_->quant_arg.out_args_.zp_ = out_quant_args.front().zeroPoint; in Init()
43 crop_para_->quant_arg.output_activation_max_ = std::numeric_limits<int8_t>::max(); in Init()
44 crop_para_->quant_arg.output_activation_min_ = std::numeric_limits<int8_t>::min(); in Init()
54 auto ret = ParallelLaunch(this->ms_context_, CropInt8Run, this, crop_para_->thread_count_); in Run()
69 Int8Crop(input_data, output_data, task_id, crop_para_); in DoExecute()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | crop_base.cc | 38 crop_para_->in_shape_ = input_shape_.data(); in ReSize()
39 crop_para_->out_shape_ = output_shape_.data(); in ReSize()
41 crop_para_->input_dim_ = input_dim; in ReSize()
42 PadOffset(input_dim, crop_para_); in ReSize()
|
| D | crop_base.h | 30 crop_para_ = reinterpret_cast<CropParameter *>(op_parameter_); in CropBaseCPUKernel()
31 crop_para_->thread_count_ = op_parameter_->thread_num_; in CropBaseCPUKernel()
42 CropParameter *crop_para_; variable
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | crop_fp32.cc | 54 …Crop4D(input_data, output_data, input->shape().data(), output->shape().data(), crop_para_, thread_… in CropParallelRun()
63 if (output->shape()[1] < crop_para_->thread_count_) { in Run()
66 …op4DNoParallel(input_data, output_data, input->shape().data(), output->shape().data(), crop_para_); in Run()
70 auto ret = ParallelLaunch(this->ms_context_, CropLaunch, this, crop_para_->thread_count_); in Run()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | crop_fp16.cc | 36 Fp16Crop(input_ptr_, output_ptr_, task_id, crop_para_); in DoExecute()
59 auto ret = ParallelLaunch(this->ms_context_, CropFp16Run, this, crop_para_->thread_count_); in Run()
|