| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | glu_fp32.cc | 99 int GluCPUKernel::Split(int task_id) { in Split()
117 int GluCPUKernel::Sigmoid(int task_id) { in Sigmoid()
131 int GluCPUKernel::Mul(int task_id) { in Mul()
147 static int SplitRun(void *cdata, int task_id, float, float) { in SplitRun()
152 static int SigmoidRun(void *cdata, int task_id, float, float) { in SigmoidRun()
157 static int MulRun(void *cdata, int task_id, float, float) { in MulRun()
|
| D | l2_norm_fp32.cc | 92 int L2NormCPUKernel::CalcSquareSum(int task_id) { in CalcSquareSum()
103 int L2NormCPUKernel::DivSqrtSum(int task_id) { in DivSqrtSum()
114 int L2NormCPUKernel::CalcL2NormTrailingAxis(int task_id) { in CalcL2NormTrailingAxis()
131 int SquareSumRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in SquareSumRun()
141 int L2NormRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in L2NormRun()
152 int L2NormTrailingAxisRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in L2NormTrailingAxisRun()
|
| D | cumsum_fp32.cc | 29 int CumsumLaunch(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in CumsumLaunch()
90 int CumSumCPUKernel::DoCumsum(int task_id) { in DoCumsum()
107 int CumSumCPUKernel::DoCumsumInt(int task_id) { in DoCumsumInt()
|
| D | transpose_fp32.cc | 115 int TransposeCPUKernel::TransposeDimGreaterThan6(int task_id) { in TransposeDimGreaterThan6()
143 int TransposeCPUKernel::RunImpl(int task_id) { in RunImpl()
153 int TransposeImpl(void *kernel, int task_id, float lhs_scale, float rhs_scale) { in TransposeImpl()
|
| D | instance_norm_fp32.cc | 48 int InstanceNormCPUKernel::DoInstanceNorm(int task_id) { in DoInstanceNorm()
66 int InstanceNormRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in InstanceNormRun()
|
| D | exp_fp32.cc | 53 int ExpCPUKernel::DoExcute(int task_id) { in DoExcute()
59 int ExpRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in ExpRun()
|
| D | convolution_1x1_fp32.cc | 145 int Convolution1x1CPUKernel::DoConv1x1(int task_id) { in DoConv1x1()
165 int Convolution1x1Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in Convolution1x1Run()
175 int Convolution1x1CPUKernel::DoConv1x1Hw(int task_id) { in DoConv1x1Hw()
212 int Convolution1x1RunHw(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in Convolution1x1RunHw()
|
| D | roi_pooling_fp32.cc | 84 int ROIPoolingCPUKernel::DoExecute(int task_id) { in DoExecute()
99 int ROIPoolingRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in ROIPoolingRun()
|
| D | activation_fp32.cc | 53 int ActivationCPUKernel::DoActivation(int task_id) { in DoActivation()
105 int ActivationRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in ActivationRun()
|
| /third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/ |
| D | tuner.py | 255 def rl_tune(self, task_id, op_json): argument
270 def ga_tune(self, task_id, op_json): argument
283 def single_rl_tune(self, task_id, json_info): argument
326 def fusion_rl_tune(self, task_id, json_info): argument
362 def fusion_ga_tune(self, task_id, json_info): argument
376 def single_ga_tune(self, task_id, json_info): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | convolution_1x1_int8.cc | 63 int Convolution1x1Int8HwRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in Convolution1x1Int8HwRun()
73 int Convolution1x1Int8OcRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in Convolution1x1Int8OcRun()
83 int Convolution1x1Int8OcOptPre(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in Convolution1x1Int8OcOptPre()
93 int Convolution1x1Int8CPUKernel::OcRun(int task_id) { in OcRun()
101 int Convolution1x1Int8CPUKernel::HwRun(int task_id) { in HwRun()
421 int Convolution1x1Int8CPUKernel::RunArmHw(int task_id) { in RunArmHw()
449 int Convolution1x1Int8CPUKernel::RunArm64OptHw(int task_id) { in RunArm64OptHw()
475 int Convolution1x1Int8CPUKernel::RunArm64OptOc(int task_id) { in RunArm64OptOc()
499 int Convolution1x1Int8CPUKernel::RunArmOc(int task_id) { in RunArmOc()
|
| D | mul_int8.cc | 188 int FastHWBroadcastMulInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in FastHWBroadcastMulInt8Run()
194 int MulInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in MulInt8Run()
200 void MulInt8CPUKernel::FastDoExecute(int task_id) { in FastDoExecute()
216 void MulInt8CPUKernel::DoExecute(int task_id) { in DoExecute()
|
| D | pooling_int8.cc | 65 int PoolingInt8CPUKernel::RunImpl(int task_id) { in RunImpl()
87 int PoolingInt8Impl(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in PoolingInt8Impl()
|
| D | detection_post_process_int8.cc | 29 int DetectionPostProcessInt8CPUKernel::DequantizeInt8ToFp32(const int task_id) { in DequantizeInt8ToFp32()
41 int DequantizeInt8ToFp32Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in DequantizeInt8ToFp32Run()
|
| D | reshape_int8.cc | 66 int ReshapeInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in ReshapeInt8Run()
75 int ReshapeInt8CPUKernel::DoExecute(int task_id) { in DoExecute()
|
| D | slice_int8.cc | 62 int SliceInt8CPUKernel::DoSlice(int task_id) { in DoSlice()
75 int SliceInt8Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in SliceInt8Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 87 auto task_id = ParallelBuildManager::StartCompileOp(kernel_json); in TbeOpParallelBuild() local
91 int task_id = -1; in TbeOpParallelBuild() local
111 void ParallelBuildManager::SavePreBuildTaskInfo(int32_t task_id, const AnfNodePtr &anf_node, in SavePreBuildTaskInfo()
126 void ParallelBuildManager::SaveTaskInfo(int32_t task_id, const mindspore::AnfNodePtr &anf_node, in SaveTaskInfo()
149 void ParallelBuildManager::PreTaskFinishProcess(int32_t task_id, const std::string &pre_build_resul… in PreTaskFinishProcess()
169 std::pair<int32_t, KernelModPtr> ParallelBuildManager::TaskFinishProcess(int32_t task_id, const std… in TaskFinishProcess()
308 bool ParallelBuildManager::WaitOne(int *task_id, std::string *task_result, std::string *pre_build_r… in WaitOne()
319 AnfNodePtr ParallelBuildManager::GetAnfNodeByTaskID(int32_t task_id) { in GetAnfNodeByTaskID()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_1x1_fp16.cc | 199 int Convolution1x1FP16CPUKernel::RunOc(int task_id) { in RunOc()
228 int Convolution1x1FP16CPUKernel::RunHw(int task_id) { in RunHw()
257 static int Convolution1x1Fp16RunOc(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in Convolution1x1Fp16RunOc()
267 static int Convolution1x1Fp16RunHw(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in Convolution1x1Fp16RunHw()
|
| D | crop_fp16.cc | 35 int CropFp16CPUKernel::DoExecute(int task_id) { in DoExecute()
40 static int CropFp16Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in CropFp16Run()
|
| D | instance_norm_fp16.cc | 94 int InstanceNormFp16CPUKernel::DoInstanceNorm(int task_id) { in DoInstanceNorm()
108 int InstanceNormFp16Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in InstanceNormFp16Run()
|
| D | deconvolution_winograd_fp16.cc | 194 int DeConvWinogradFp16CPUKernel::DoDeconv(int task_id) { in DoDeconv()
215 int DeConvWinogradFp16CPUKernel::DeDeconvPost(int task_id) { in DeDeconvPost()
229 int DeConvWgFp16Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in DeConvWgFp16Run()
235 int DeConvWgPostFp16Run(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in DeConvWgPostFp16Run()
|
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/ |
| D | conv1x1_run_int8_wrapper.c | 33 int OcOptPre(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in OcOptPre()
54 int RunArm64OptOc(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in RunArm64OptOc()
80 int RunArmOc(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in RunArmOc()
111 int RunArm64OptHw(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in RunArm64OptHw()
137 int RunArmHw(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in RunArmHw()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | crop_fp16.c | 23 void Fp16Crop(const float16_t *input, float16_t *output, int task_id, const CropParameter *para) { in Fp16Crop()
43 void Fp16Crop1D(const float16_t *input, float16_t *output, int task_id, const CropParameter *para) { in Fp16Crop1D()
60 void Fp16Crop2D(const float16_t *input, float16_t *output, int task_id, const CropParameter *para) { in Fp16Crop2D()
82 void Fp16Crop3D(const float16_t *input, float16_t *output, int task_id, const CropParameter *para) { in Fp16Crop3D()
116 void Fp16Crop4D(const float16_t *input, float16_t *output, int task_id, const CropParameter *para) { in Fp16Crop4D()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | constant_of_shape.cc | 29 int ConstantOfShapeRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in ConstantOfShapeRun()
40 int ConstantOfShapeCPUKernel::DoExecute(int task_id) { in DoExecute()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | neg_fp16_grad.cc | 31 int NegGradRun(void *cdata, int task_id, float lhs_scale, float rhs_scale) { in NegGradRun()
46 int NegGradCPUKernelFp16::DoNegGrad(int task_id) { in DoNegGrad()
|