Home
last modified time | relevance | path

Searched refs:input1_data_ (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dscale_int8.cc38 if (input1_data_ != nullptr && malloced_scale_) { in ~ScaleInt8CPUKernel()
39 free(input1_data_); in ~ScaleInt8CPUKernel()
40 input1_data_ = nullptr; in ~ScaleInt8CPUKernel()
55 input1_data_ = scale_ptr; in InitScaleOffset()
58 input1_data_ = reinterpret_cast<int8_t *>(malloc(out_tensors_.at(0)->Size())); in InitScaleOffset()
59 if (input1_data_ == nullptr) { in InitScaleOffset()
65 … reinterpret_cast<int8_t *>(input1_data_), 0, tile_para->ndim_, tile_para->in_shape1_, in InitScaleOffset()
85 free(input1_data_); in InitScaleOffset()
86 input1_data_ = nullptr; in InitScaleOffset()
267 int8_t *cur_input1_data = input1_data_ + task_id * count_unit_; in Scale()
[all …]
Dmul_int8.cc148 input1_data_ = static_cast<int8_t *>(in_tensors_.at(1)->MutableData()); in Run()
149 MS_ASSERT(input1_data_); in Run()
170 input1_data_ = static_cast<int8_t *>(ctx_->allocator->Malloc(out_tensors_.at(0)->Size())); in Run()
171 if (input1_data_ == nullptr) { in Run()
177 … static_cast<int8_t *>(in_tensors_.at(1)->MutableData()), input0_data_, input1_data_, tile_para); in Run()
180 ctx_->allocator->Free(input1_data_); in Run()
207 int8_t *cur_input1_data = input1_data_ + task_id * count_unit_ * depth; in FastDoExecute()
210 cur_input0_data = input1_data_; in FastDoExecute()
222 int8_t *cur_input1_data = input1_data_ + task_id * count_unit_; in DoExecute()
Dadd_int8.cc181 cur_in1 = input1_data_; in BroadcastRun()
185 cur_in1 = input1_data_ + task_id * stride * in_size_ + i * in_size_; in BroadcastRun()
215 int8_t *cur_in1 = input1_data_ + stride * task_id; in DoExecute()
219 int8_t element_in = arith_para_->in_elements_num0_ == 1 ? input0_data_[0] : input1_data_[0]; in DoExecute()
240 input1_data_ = static_cast<int8_t *>(in_tensors_.at(1)->data()); in Run()
Ddiv_int8.cc76 auto input1_data_ = static_cast<int8_t *>(in_tensors_.at(1)->MutableData()); in DoExecute() local
89 …ret = DivInt8(input0_data_ + task_id * count, input1_data_ + task_id * count, output_data_ + task_… in DoExecute()
Dadd_int8.h53 int8_t *input1_data_ = nullptr; variable
Dsub_int8.cc97 auto input1_data_ = static_cast<int8_t *>(in_tensors_.at(1)->MutableData()); in DoExecute() local
113 …ret = SubInt8(input0_data_ + task_id * stride, input1_data_ + task_id * stride, output_data_ + tas… in DoExecute()
Dscale_int8.h47 int8_t *input1_data_ = nullptr; variable
Dmul_int8.h55 int8_t *input1_data_ = nullptr; variable
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dadd_int8_wrapper.c33 cur_in1 = args->input1_data_; in AddBroadcastInt8Run()
37 cur_in1 = args->input1_data_ + task_id * stride * args->in_size_ + i * args->in_size_; in AddBroadcastInt8Run()
55 int8_t *cur_in1 = args->input1_data_ + stride * task_id; in AddInt8Run()
59 …ent_in = args->arith_para_->in_elements_num0_ == 1 ? args->input0_data_[0] : args->input1_data_[0]; in AddInt8Run()
Dadd_int8_wrapper.h37 int8_t *input1_data_; member