Home
last modified time | relevance | path

Searched refs:rhs_scale (Results 1 – 25 of 209) sorted by relevance

123456789

/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconv1x1_run_int8_wrapper.h51 int OcOptPre(void *cdata, int task_id, float lhs_scale, float rhs_scale);
52 int RunArm64OptOc(void *cdata, int task_id, float lhs_scale, float rhs_scale);
53 int RunArmOc(void *cdata, int task_id, float lhs_scale, float rhs_scale);
54 int RunArm64OptHw(void *cdata, int task_id, float lhs_scale, float rhs_scale);
55 int RunArmHw(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dadd_int8_wrapper.h41 int AddBroadcastInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
43 int AddInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dconv1x1_run_int8_wrapper.c33 int OcOptPre(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in OcOptPre() argument
54 int RunArm64OptOc(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in RunArm64OptOc() argument
80 int RunArmOc(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in RunArmOc() argument
111 int RunArm64OptHw(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in RunArm64OptHw() argument
137 int RunArmHw(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in RunArmHw() argument
Dadd_int8_wrapper.c20 int AddBroadcastInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in AddBroadcastInt8Run() argument
45 int AddInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in AddInt8Run() argument
Dslice_int8_wrapper.h29 int SliceInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dbatchnorm_int8_wrapper.h30 int BatchNormInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dresize_int8_wrapper.h35 int ResizeInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dconcat_int8_wrapper.h33 int ConcatInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dslice_int8_wrapper.c20 int SliceInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in SliceInt8Run() argument
Dconvolution_depthwise_int8_wrapper.h33 int ConvDepthwiseInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dbatchnorm_int8_wrapper.c21 int BatchNormInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in BatchNormInt8Run() argument
Dresize_int8_wrapper.c20 int ResizeInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in ResizeInt8Run() argument
Dconvolution_int8_wrapper.h39 int ConvolutionInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dconcat_int8_wrapper.c19 int ConcatInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in ConcatInt8Run() argument
Dconvolution_int8_wrapper.c19 int ConvolutionInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in ConvolutionInt8Run() argument
Dconvolution_depthwise_int8_wrapper.c19 int ConvDepthwiseInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in ConvDepthwiseInt8Run() argument
/third_party/mindspore/mindspore/core/mindrt/src/thread/
Dthreadpool.cc103 void Worker::set_scale(float lhs_scale, float rhs_scale) { in set_scale() argument
105 rhs_scale_ = rhs_scale; in set_scale()
206 float rhs_scale = (i + 1) * per_scale; in SyncRunTask() local
207 rhs_scale = i == task_num - 1 ? kMaxScale : rhs_scale; in SyncRunTask()
208 task->status |= task->func(task->content, i, lhs_scale, rhs_scale); in SyncRunTask()
253 float rhs_scale = 0; in CalculateScales() local
259 rhs_scale += worker->frequency() * 1.0 / sum_frequency; in CalculateScales()
260 rhs_scale = rhs_scale < 1 ? rhs_scale : 1; in CalculateScales()
261 worker->set_scale(lhs_scale, rhs_scale); in CalculateScales()
262 lhs_scale = rhs_scale; in CalculateScales()
Dthreadpool.h77 void set_scale(float lhs_scale, float rhs_scale);
79 float rhs_scale() const { return rhs_scale_; } in rhs_scale() function
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Daddn_fp16.cc30 int AddNLaunch(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in AddNLaunch() argument
36 return kernel->AddNParallelRun(task_id, lhs_scale, rhs_scale); in AddNLaunch()
44 int AddNFp16CPUKernel::AddNParallelRun(int thread_id, float lhs_scale, float rhs_scale) { in AddNParallelRun() argument
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Daddn_fp32.cc30 int AddNLaunch(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in AddNLaunch() argument
36 return kernel->AddNParallelRun(task_id, lhs_scale, rhs_scale); in AddNLaunch()
48 int AddNCPUKernel::AddNParallelRun(int thread_id, float lhs_scale, float rhs_scale) { in AddNParallelRun() argument
Dpad_fp32.h63 int PadImpl(void *cdata, int task_id, float lhs_scale, float rhs_scale);
64 int MirrorPadImpl(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dl2_norm_fp32.cc131 int SquareSumRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in SquareSumRun() argument
141 int L2NormRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in L2NormRun() argument
152 int L2NormTrailingAxisRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in L2NormTrailingAxisRun() argument
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dmul_int8.h59 int MulInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
60 int FastHWBroadcastMulInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
Dcrop_int8.h42 int CropInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/
Ddeconvolution_fp32_wrapper.h43 int DeConvFp32Run(void *cdata, int task_id, float lhs_scale, float rhs_scale);

123456789