Home
last modified time | relevance | path

Searched refs:ConvParameter (Results 1 – 25 of 137) sorted by relevance

123456

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_depthwise_fp32.h34 const ConvParameter *conv_param, int task_id);
36 void InitSlidingParam(SlidingWindowParam *sliding, const ConvParameter *conv_param, int block);
38 void InitSlidingParamConv(SlidingWindowParam *sliding, const ConvParameter *conv_param, int input_b…
41 void AppendSlidingParamConv(SlidingWindowParam *sliding, const ConvParameter *conv_param, int in_bl…
44 void InitSlidingParamConvDw(SlidingWindowParam *sliding, const ConvParameter *conv_param, int block…
46 void AppendSlidingParamConvDw(SlidingWindowParam *sliding, const ConvParameter *conv_param, int blo…
49 const ConvParameter *conv_param, const SlidingWindowParam *sliding, int task_id);
51 bool CheckConvDwUse3X3(const ConvParameter *conv_param);
53 bool CheckConvDwUseIndirectBuffer(const ConvParameter *conv_param);
55 …tIndirection(float **indirect_buffer, float *src, float *zero_ptr, const ConvParameter *conv_param,
[all …]
Dwinograd_transform.h32 … int out_tile_index, int out_w_block_num, const ConvParameter *conv_param,
36 … int out_tile_index, int output_unit_num, const ConvParameter *conv_param,
40 … int out_tile_index, int output_unit_num, const ConvParameter *conv_param,
Ddeconv_winograd_fp32.h31 int PackDeConvWgDataFp32(const float *nhwc_weight, DeConvComputeUnit *unit, const ConvParameter *co…
34 const ConvParameter *conv_param, DeConvParam *deconv_param, int task_id);
35 int DeconvWgPost(const float *tile_out, float *nc4hw4_output, const ConvParameter *conv_param,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.h27 void RollingIm2ColPackUnitFp32(const float *input_data, const ConvParameter *conv_param, float *pac…
29 void RollingIm2ColPackDwUnitFp32(const float *input_data, const ConvParameter *conv_param, float *p…
32 void rolling_im2col_hwc(const float *in_data, float *data_col, const ConvParameter *conv_param, int…
33 void rolling_im2row_hwc(const float *in_data, float *data_row, const ConvParameter *conv_param, int…
34 void rolling_col2im_hwc(const float *data_col, float *data_im, const ConvParameter *conv_param, int…
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dconv_depthwise_int8.h28 const int32_t *bias_data, const ConvParameter *conv_param, int task_id);
31 … const int32_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding);
34 … const int32_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding,
38 const int8_t *input_zp, const int32_t *output_zp, const ConvParameter *conv_param,
42 … const int32_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding,
Dpack_int8.h40 const ConvParameter *conv_param);
42 void PackInputToC8Int8(const int8_t *input_data, int16_t *packed_input, const ConvParameter *conv_p…
43 …t8(const int8_t *origin_weight_data, int16_t *packed_weight_data, const ConvParameter *conv_param);
46 const ConvParameter *conv_param, bool per_channel, bool is_optimize);
53 void PackDepthwiseInt8Input(const int8_t *src, int16_t *dst, const ConvParameter *conv_param);
/third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/
Ddepthwise_conv2d_infer_test.cc43 ConvParameter *parameter = new ConvParameter; in TEST_F()
89 ConvParameter *parameter = new ConvParameter; in TEST_F()
135 ConvParameter *parameter = new ConvParameter; in TEST_F()
182 ConvParameter *parameter = new ConvParameter; in TEST_F()
229 ConvParameter *parameter = new ConvParameter; in TEST_F()
276 ConvParameter *parameter = new ConvParameter; in TEST_F()
323 ConvParameter *parameter = new ConvParameter; in TEST_F()
370 ConvParameter *parameter = new ConvParameter; in TEST_F()
417 ConvParameter *parameter = new ConvParameter; in TEST_F()
464 ConvParameter *parameter = new ConvParameter; in TEST_F()
[all …]
Dconv2d_infer_test.cc43 ConvParameter *parameter = new ConvParameter; in TEST_F()
88 ConvParameter *parameter = new ConvParameter; in TEST_F()
133 ConvParameter *parameter = new ConvParameter; in TEST_F()
179 ConvParameter *parameter = new ConvParameter; in TEST_F()
225 ConvParameter *parameter = new ConvParameter; in TEST_F()
271 ConvParameter *parameter = new ConvParameter; in TEST_F()
317 ConvParameter *parameter = new ConvParameter; in TEST_F()
363 ConvParameter *parameter = new ConvParameter; in TEST_F()
409 ConvParameter *parameter = new ConvParameter; in TEST_F()
455 ConvParameter *parameter = new ConvParameter; in TEST_F()
[all …]
Ddeconv2d_infer_test.cc45 ConvParameter *parameter = new ConvParameter; in TEST_F()
93 ConvParameter *parameter = new ConvParameter; in TEST_F()
141 ConvParameter *parameter = new ConvParameter; in TEST_F()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dgroup_convolution_creator.h45 conv_param_ = reinterpret_cast<ConvParameter *>(op_parameter); in GroupConvCreator()
55 int GetSingleConvParam(ConvParameter *conv_param, std::vector<lite::Tensor *> *new_inputs,
76 ConvParameter *conv_param_;
83 ConvParameter *CreateNewConvParameter(ConvParameter *parameter);
Dgroup_convolution_creator.cc26 ConvParameter *CreateNewConvParameter(ConvParameter *parameter) { in CreateNewConvParameter()
27 auto conv_parameter = reinterpret_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in CreateNewConvParameter()
32 memcpy(conv_parameter, parameter, sizeof(ConvParameter)); in CreateNewConvParameter()
36 void FreeCurrentConv(ConvParameter *conv_param, std::vector<lite::Tensor *> *new_inputs, in FreeCurrentConv()
187 int GroupConvCreator::GetSingleConvParam(ConvParameter *conv_param, std::vector<lite::Tensor *> *ne… in GetSingleConvParam()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.h27 void RollingIm2ColPackUnitFp16(const float16_t *input_data, const ConvParameter *conv_param, float1…
29 void RollingIm2ColPackDwUnitFp16(const float16_t *input_data, const ConvParameter *conv_param, floa…
31 void RollingCol2ImPackUnitFp16(const float16_t *data_col, float16_t *data_im, const ConvParameter *…
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dconv1x1_fp32_tests.cc34 auto conv_param = new ConvParameter(); in TEST_F()
62 auto conv_param = new ConvParameter(); in TEST_F()
103 auto conv_param = new ConvParameter(); in TEST_F()
122 auto conv_param = new ConvParameter(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dwinograd_transform_fp16.h33 … int out_tile_index, int out_w_block_num, const ConvParameter *conv_param,
38 const ConvParameter *conv_param, OutputTransFp16Func func);
42 const ConvParameter *conv_param, OutputTransFp16Func func);
Dconv_depthwise_fp16.h44 const float16_t *bias_data, const ConvParameter *conv_param, int task_id);
47 … const float16_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding,
51 … const float16_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding,
58 … const float16_t *bias_data, const ConvParameter *conv_param, int start_oh, int end_oh);
Ddeconv_winograd_fp16.h26 int PackDeConvWgDataFp16(const float16_t *nhwc_weight, DeConvComputeUnit *unit, const ConvParameter…
30 … int calculate_count, const ConvParameter *conv_param, DeConvParam *deconv_param, int task_id);
32 void DeconvWgPostFp16(const float16_t *tile_out, float16_t *nc4hw4_output, const ConvParameter *con…
Dconv_fp16.h35 const ConvParameter *conv_param);
39 const ConvParameter *conv_param);
52 … const ConvParameter *conv_param, InputTransFp16Func in_func, OutputTransFp16Func out_func);
/third_party/mindspore/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()
158 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F()
235 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F()
308 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F()
385 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F()
460 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F()
531 auto conv_param = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in TEST_F()
[all …]
/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Ddeconv2d_populate_v0.cc32 auto *conv_param = reinterpret_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in PopulateDeconvParameter()
37 memset(conv_param, 0, sizeof(ConvParameter)); in PopulateDeconvParameter()
Ddedepthwise_conv2d_populate_v0.cc32 auto *conv_param = reinterpret_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in PopulateDeconvDwParameter()
37 memset(conv_param, 0, sizeof(ConvParameter)); in PopulateDeconvDwParameter()
Ddepthwise_conv2d_populate_v0.cc32 auto *conv_param = reinterpret_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in PopulateConvDwParameter()
37 memset(conv_param, 0, sizeof(ConvParameter)); in PopulateConvDwParameter()
Dconv2d_populate_v0.cc33 auto *conv_param = reinterpret_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in PopulateConvParameter()
38 memset(conv_param, 0, sizeof(ConvParameter)); in PopulateConvParameter()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconvolution_delegate_fp16.cc90 static void SetInputOutputShapeInfo(ConvParameter *conv_param, const lite::Tensor *input, const lit… in SetInputOutputShapeInfo()
107 …kernel::SetInputOutputShapeInfo(reinterpret_cast<ConvParameter *>(op_parameter_), in_tensors_.fron… in ReSize()
129 auto conv_param = reinterpret_cast<ConvParameter *>(opParameter); in CpuConvDwFp16KernelCreator()
158 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvFp16KernelSelect()
202 …op_parameter, inputs, outputs, ctx, group_conv_creator, reinterpret_cast<ConvParameter *>(op_param… in CpuGroupConvFp16KernelCreator()
216 auto conv_param = reinterpret_cast<ConvParameter *>(opParameter); in CpuConvFp16KernelCreator()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dconv_1x1_int8_tests.cc32 auto conv_param = new ConvParameter(); in TEST_F()
50 auto conv_param = new ConvParameter(); in TEST_F()
73 ConvParameter *conv_param, int8_t **correct) { in Conv1x1Int8TestInit1_perchannel()
121 auto conv_param = new ConvParameter(); in TEST_F()
142 ConvParameter *conv_param, int8_t **correct) { in Conv1x1Int8TestInit1()
190 auto conv_param = new ConvParameter(); in TEST_F()
211 ConvParameter *conv_param, int8_t **correct) { in Conv1x1Int8TestInit2()
267 auto conv_param = new ConvParameter(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconv_common_base.h27 bool CheckConvDw1DWinograd(const ConvParameter *conv_param, int thread_num);
32 bool CheckIfUseWinograd(int *output_unit, const ConvParameter *conv_param);

123456