Home
last modified time | relevance | path

Searched defs:conv_param (Results 1 – 25 of 95) sorted by relevance

1234

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/
Dconv_common_base.c23 bool CheckConvDw1DWinograd(const ConvParameter *conv_param, int thread_num) { in CheckConvDw1DWinograd()
45 int SelectOutputUnit(const ConvParameter *conv_param) { in SelectOutputUnit()
85 bool CheckIfUseWinograd(int *output_unit, const ConvParameter *conv_param) { in CheckIfUseWinograd()
99 bool CheckAvxUseSW1x1Conv(const ConvParameter *conv_param) { in CheckAvxUseSW1x1Conv()
109 bool CheckAvxUseSWConv(const ConvParameter *conv_param, int thread_nr_) { in CheckAvxUseSWConv()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dconvolution_delegate_fp32.cc165 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter_); in SetInputOutputShapeInfo() local
189 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter_); in CpuConvFp32NC4KernelSelect() local
209 bool ConvolutionDelegateCPUKernel::CheckAvxUseSW1x1Conv(const ConvParameter *conv_param) { in CheckAvxUseSW1x1Conv()
219 bool ConvolutionDelegateCPUKernel::CheckAvxUseSWConv(const ConvParameter *conv_param) { in CheckAvxUseSWConv()
241 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter_); in CreateConv1x1MatmulKernel() local
268 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter_); in CpuConvFp32NHWCKernelSelect() local
398 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvFp32KernelCreator() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dconvolution_sw_arm64.c25 ConvParameter *conv_param = (ConvParameter *)conv->base_.param_; in ConvSWARM64InitGlobalVariable() local
35 ConvParameter *conv_param = (ConvParameter *)conv->base_.param_; in ConvSWARM64RunImpl() local
42 ConvolutionBaseStruct *CreateConvolutionSWARM64(ConvParameter *conv_param) { in CreateConvolutionSWARM64()
Dconvolution_sw_avx.c26 ConvParameter *conv_param = (ConvParameter *)conv->base_.param_; in ConvSWAVXInitGlobalVariable() local
41 ConvParameter *conv_param = (ConvParameter *)conv->base_.param_; in ConvSWAVXRunImpl() local
54 ConvolutionBaseStruct *CreateConvolutionSWAVX(ConvParameter *conv_param) { in CreateConvolutionSWAVX()
Dconvolution_delegate.c108 ConvParameter *conv_param = (ConvParameter *)convolution_delegate->conv_.base_.param_; in ConvolutionDelegateConvNC4KernelSelect() local
127 ConvParameter *conv_param = (ConvParameter *)convolution_delegate->conv_.base_.param_; in ConvolutionDelegateConvNHWCKernelSelect() local
277 KernelBase *CreateConvlutionDelegate(ConvParameter *conv_param) { in CreateConvlutionDelegate()
288 KernelBase *CreateConvolutionDepthwise(ConvParameter *conv_param) { in CreateConvolutionDepthwise()
336 ConvParameter *conv_param = (ConvParameter *)param; in CreateConv2DFusion() local
Dconvolution_winograd.c32 ConvolutionWinogradBaseStruct *SelectConvolutionWinograd(ConvParameter *conv_param) { in SelectConvolutionWinograd()
67 ConvolutionBaseStruct *CreateConvolutionWinograd(ConvParameter *conv_param, int out_unit) { in CreateConvolutionWinograd()
Dconvolution_depthwise_sw.c74 ConvParameter *conv_param = (ConvParameter *)conv_dw->conv_.base_.param_; in ConvDwSWRun() local
105 ConvParameter *conv_param = (ConvParameter *)self->param_; in ConvolutionDepthwiseSWResize() local
166 ConvParameter *conv_param = (ConvParameter *)self->param_; in ConvolutionDdepthwiseSWPrepare() local
188 KernelBase *CreateConvDwSW(ConvParameter *conv_param) { in CreateConvDwSW()
Dconvolution_im2col_arm64.c34 ConvParameter *conv_param = (ConvParameter *)conv->base_.param_; in ConvIm2ColARM64RunImpl() local
54 ConvolutionBaseStruct *CreateConvIm2ColARM64(ConvParameter *conv_param) { in CreateConvIm2ColARM64()
Dconvolution_depthwise_sw_avx.c91 ConvParameter *conv_param = (ConvParameter *)conv_dw->conv_.base_.param_; in ConvDwSWAvxRun() local
166 ConvParameter *conv_param = (ConvParameter *)self->param_; in ConvolutionDepthwiseSWAVXPrepare() local
185 ConvParameter *conv_param = (ConvParameter *)self->param_; in ConvolutionDepthwiseSWAVXResize() local
201 KernelBase *CreateConvDwSWAVX(ConvParameter *conv_param) { in CreateConvDwSWAVX()
Dconvolution_slidewindow.c66 ConvParameter *conv_param = (ConvParameter *)conv_sw->conv_.base_.param_; in ConvSWFreeTmpBuffer() local
101 ConvParameter *conv_param = (ConvParameter *)conv->base_.param_; in ConvSWMallocWeightBiasData() local
159 ConvParameter *conv_param = (ConvParameter *)self->param_; in ConvolutionSWCompute() local
181 ConvParameter *conv_param = (ConvParameter *)self->param_; in ConvolutionSWResize() local
Dconvolution_depthwise_indirect.c41 ConvParameter *conv_param = (ConvParameter *)conv_dw->conv_.base_.param_; in ConvDwIndirectRun() local
136 ConvParameter *conv_param = (ConvParameter *)self->param_; in ConvolutionDepthwiseIndirectCompute() local
207 KernelBase *CreateConvDwIndirect(ConvParameter *conv_param) { in CreateConvDwIndirect()
Dconvolution_im2col_avx.c74 ConvParameter *conv_param = (ConvParameter *)conv->base_.param_; in ConvIm2ColAVXRunImpl() local
132 ConvolutionBaseStruct *CreateConvIm2ColAVX(ConvParameter *conv_param) { in CreateConvIm2ColAVX()
Dconvolution_im2col_avx512.c79 ConvParameter *conv_param = (ConvParameter *)conv->base_.param_; in ConvIm2ColAVX512RunImpl() local
128 ConvolutionBaseStruct *CreateConvIm2ColAVX512(ConvParameter *conv_param) { in CreateConvIm2ColAVX512()
Dconvolution_1x1.c163 ConvParameter *conv_param = (ConvParameter *)conv_1x1->conv_.base_.param_; in InitConv1x1MatmulParam() local
199 ConvParameter *conv_param = (ConvParameter *)conv_1x1->conv_.base_.param_; in InitConv1x1Param() local
283 ConvParameter *conv_param = (ConvParameter *)self->param_; in Convolution1x1Compute() local
350 ConvolutionBaseStruct *CreateConvolution1x1(ConvParameter *conv_param) { in CreateConvolution1x1()
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dconv1x1_fp32_tests.cc30 auto conv_param = new ConvParameter(); in TEST_F() local
58 auto conv_param = new ConvParameter(); in TEST_F() local
99 auto conv_param = new ConvParameter(); in TEST_F() local
118 auto conv_param = new ConvParameter(); in TEST_F() local
Dconvolution_depthwise_fp32_tests.cc30 void InitConvDwParam(ConvParameter *conv_param) { in InitConvDwParam()
57 const ConvParameter *conv_param, const InnerContext *ctx) { in InitConvDwCreator()
105 auto conv_param = new ConvParameter(); in TEST_F() local
Dconvolution_fp32_tests.cc30 void InitConvParam(ConvParameter *conv_param) { in InitConvParam()
95 auto conv_param = new ConvParameter(); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Dconvolution_int8_creator.cc44 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvDwInt8KernelCreator() local
79 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvInt8KernelSelect() local
104 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuGroupConvInt8KernelCreator() local
124 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvInt8KernelCreator() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dconv_1x1_int8_tests.cc33 auto conv_param = new ConvParameter(); in TEST_F() local
51 auto conv_param = new ConvParameter(); in TEST_F() local
74 ConvParameter *conv_param, int8_t **correct) { in Conv1x1Int8TestInit1_perchannel()
122 auto conv_param = new ConvParameter(); in TEST_F() local
143 ConvParameter *conv_param, int8_t **correct) { in Conv1x1Int8TestInit1()
191 auto conv_param = new ConvParameter(); in TEST_F() local
212 ConvParameter *conv_param, int8_t **correct) { in Conv1x1Int8TestInit2()
268 auto conv_param = new ConvParameter(); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Dconvolution_delegate_fp16.cc93 static void SetInputOutputShapeInfo(ConvParameter *conv_param, const lite::Tensor *input, const lit… in SetInputOutputShapeInfo()
142 auto conv_param = reinterpret_cast<ConvParameter *>(opParameter); in CpuConvDwFp16KernelCreator() local
171 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvFp16KernelSelect() local
229 auto conv_param = reinterpret_cast<ConvParameter *>(opParameter); in CpuConvFp16KernelCreator() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Dconv2d_delegate_dynamic_fp16_coder.cc67 void ConvDelegateDynamicFP16Coder::SetInputOutputShapeInfo(ConvParameter *conv_param, const lite::T… in SetInputOutputShapeInfo()
90 auto conv_param = reinterpret_cast<ConvParameter *>(paramGen(node->primitive_)); in CPUConvFP16DynamicCoderSelect() local
134 auto conv_param = reinterpret_cast<ConvParameter *>(param_gen(node->primitive_)); in CPUConv2DFusionDynamicFP16CoderCreator() local
Dconv2d_delegate_fp16_coder.cc64 int SelectOutUnit(const ConvParameter *conv_param) { in SelectOutUnit()
119 auto conv_param = reinterpret_cast<ConvParameter *>(paramGen(node->primitive_)); in CPUConvolutionFP16CoderSelect() local
173 auto conv_param = reinterpret_cast<ConvParameter *>(paramGen(node->primitive_)); in CPUConvDwFp16CoderCreator() local
235 auto conv_param = reinterpret_cast<ConvParameter *>(param_gen(node->primitive_)); in CPUConv2DFusionFP16CoderCreator() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Dconvolution_grad_fp32_tests.cc35 void InitConvParamGroup1FP32(ConvParameter *conv_param) { in InitConvParamGroup1FP32()
63 void InitConvParamGroup3FP32(ConvParameter *conv_param) { in InitConvParamGroup3FP32()
69 void InitConvParamGroup3Dilation2FP32(ConvParameter *conv_param) { in InitConvParamGroup3Dilation2FP32()
79 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F() local
159 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F() local
237 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F() local
311 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F() local
389 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F() local
465 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F() local
537 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Dconv2d_delegate_fp32_coder.cc54 void SetInputOutputShapeInfo(ConvParameter *conv_param, const lite::Tensor *input, const lite::Tens… in SetInputOutputShapeInfo()
76 auto conv_param = reinterpret_cast<ConvParameter *>(param_gen(node->primitive_)); in CPUConvolutionFP32CoderSelect() local
132 auto conv_param = reinterpret_cast<ConvParameter *>(param_gen(node->primitive_)); in CPUConv2DFusionFP32CoderCreator() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/
Dconv1x1_int8.c21 int32_t *multiplier, ConvParameter *conv_param, MATMUL_OPT_DP_FUNC matmul_func, in Conv1x1Int8Opt()
33 int32_t *multiplier, ConvParameter *conv_param, const int32_t *filter_zp) { in Conv1x1Int8()

1234