Home
last modified time | relevance | path

Searched refs:input_channel_ (Results 1 – 25 of 113) sorted by relevance

12345

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconv1x1_base.c28 …const char *src_h_ptr = src + src_h * conv_param->input_w_ * conv_param->input_channel_ * data_siz… in Conv1x1InputPack()
29 char *dst_h_ptr = dst + dst_h * conv_param->output_w_ * conv_param->input_channel_ * data_size; in Conv1x1InputPack()
35 memcpy(dst_h_ptr + dst_w * conv_param->input_channel_ * data_size, in Conv1x1InputPack()
36 …src_h_ptr + src_w * conv_param->input_channel_ * data_size, conv_param->input_channel_ * data_size… in Conv1x1InputPack()
Dconv_common_base.c27 conv_param->input_channel_ == conv_param->output_channel_ && conv_param->output_w_ >= 4 && in CheckConvDw1DWinograd()
48 int in_c = conv_param->input_channel_; in SelectOutputUnit()
87 conv_param->stride_h_ == 1 && conv_param->stride_w_ == 1 && conv_param->input_channel_ != 1) { in CheckIfUseWinograd()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dconv_depthwise_int8.c97 …t *src = input_data + b * conv_param->input_h_ * conv_param->input_w_ * conv_param->input_channel_; in ConvDwInt8()
113 const int8_t *src_kh = src + ih * conv_param->input_w_ * conv_param->input_channel_; in ConvDwInt8()
116 int in_sw_step = conv_param->stride_w_ * conv_param->input_channel_; in ConvDwInt8()
127 const int8_t *src_kw = src_kh + iw_origin * conv_param->input_channel_; in ConvDwInt8()
149 src += conv_param->input_channel_; in ConvDw3x3Int8InitBuffer()
152 input += conv_param->input_w_ * conv_param->input_channel_; in ConvDw3x3Int8InitBuffer()
265 …t8Block(output_ptr, buffer, weight_ptr, bias_ptr, 0, 64, 64, ih_offset, conv_param->input_channel_, in ConvDw3x3Int8Row()
279 ConvDw3x3Int8Block(output_ptr, input_ptr, weight_ptr, bias_ptr, c, conv_param->input_channel_, in ConvDw3x3Int8Row()
280 … conv_param->input_channel_, conv_param->input_w_ * conv_param->input_channel_, in ConvDw3x3Int8Row()
281 … conv_param->input_channel_, block_output_h, block_output_w, in_zp, out_zp, out_multiplier_ptr, in ConvDw3x3Int8Row()
[all …]
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dconv1x1_fp32_tests.cc49 conv_param->input_channel_ = 1; in TEST_F()
90 conv_param->input_channel_ = 1; in TEST_F()
104 conv_param->input_channel_ = 2; in TEST_F()
123 conv_param->input_channel_ = 6; in TEST_F()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconvolution_int8_creator.cc61 if (CheckConvDwUse3X3(conv_param) && conv_param->input_channel_ % C8NUM == 0) { in CpuConvDwInt8KernelCreator()
106 …if (conv_param->group_ > conv_param->input_channel_ || conv_param->input_channel_ % conv_param->gr… in CpuGroupConvInt8KernelCreator()
108 << conv_param->input_channel_; in CpuGroupConvInt8KernelCreator()
126 …} else if (conv_param->group_ == conv_param->input_channel_ && conv_param->group_ == conv_param->o… in CpuConvInt8KernelCreator()
Dconvolution_int8.cc55 conv_param_->input_channel_ = input_channel; in InitWeightBias()
150 tmp_size = UP_ROUND(kernel_plane * conv_param_->input_channel_, C4NUM); in InitTmpBuffer()
152 tmp_size = UP_ROUND(kernel_plane * conv_param_->input_channel_, C16NUM); in InitTmpBuffer()
155 … ctx_->allocator->Malloc(thread_count_ * tile_num_ * kernel_plane * conv_param_->input_channel_)); in InitTmpBuffer()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dconvolution_grad_input.cc48 conv_param->input_channel_ = dx_tensor->shape()[(kNHWC_C)]; in ReSize()
53 …ws_size_ = chunk_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_ / c… in ReSize()
55 …int n = conv_param->kernel_w_ * conv_param->kernel_h_ * conv_param->input_channel_ / conv_param->g… in ReSize()
69 … (conv_param->input_channel_ == conv_param->output_channel_) && (conv_param->dilation_h_ == 1) && in ReSize()
91 int in_ch = conv_param->input_channel_; in Execute()
173 int in_ch = conv_param->input_channel_; in Run()
Dconvolution.cc48 conv_param_->input_channel_ = input_x->shape().at(kNHWC_C); in ReSize()
53 …conv_param_->group_ = (conv_param_->group_ == 0) ? conv_param_->input_channel_ : conv_param_->grou… in ReSize()
55 …const int k = conv_param_->kernel_h_ * conv_param_->kernel_w_ * conv_param_->input_channel_ / conv… in ReSize()
64 …(conv_param_->input_channel_ == conv_param_->output_channel_) && (conv_param_->dilation_h_ == 1) && in ReSize()
69 ws_size_ = chunk_ * conv_param_->kernel_h_ * conv_param_->kernel_w_ * conv_param_->input_channel_; in ReSize()
90 const int in_ch = conv_param_->input_channel_; in Execute()
Ddeconvolution_grad_filter.cc53 conv_param->input_channel_ = x_tensor->shape().at(kNHWC_C); in Init()
62 int m = conv_param->input_channel_ / conv_param->group_; in Init()
64 size_t mat_alloc = MatSizeTotal(n, m, chunk * conv_param->input_w_, conv_param->input_channel_); in Init()
87 int in_ch = conv_param->input_channel_; in Execute()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dconvolution_fp16_grad_input.cc48 conv_param->input_channel_ = dx_tensor->shape()[(kNHWC_C)]; in ReSize()
53 …ws_size_ = chunk_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_ / c… in ReSize()
55 …int n = conv_param->kernel_w_ * conv_param->kernel_h_ * conv_param->input_channel_ / conv_param->g… in ReSize()
69 … (conv_param->input_channel_ == conv_param->output_channel_) && (conv_param->dilation_h_ == 1) && in ReSize()
90 int in_ch = conv_param->input_channel_; in DoExecute()
170 int in_ch = conv_param->input_channel_; in Run()
Dconvolution_fp16_grad_filter.cc50 conv_param->input_channel_ = x_tensor->shape().at(kNHWC_C); in ReSize()
64 … (conv_param->input_channel_ == conv_param->output_channel_) && (conv_param->dilation_h_ == 1) && in ReSize()
69 ws_size_ = chunk_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_; in ReSize()
71 …int n = conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_ / conv_param->g… in ReSize()
94 int in_ch = conv_param->input_channel_; in DoExecute()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc40 tmp_size = MSMAX(UP_ROUND(kernel_plane * conv_param_->input_channel_, C4NUM), in InitTmpBuffer()
41 UP_ROUND(kernel_plane * conv_param_->input_channel_, C16NUM)); in InitTmpBuffer()
44 tmp_size = UP_ROUND(kernel_plane * conv_param_->input_channel_, C4NUM); in InitTmpBuffer()
46 tmp_size = UP_ROUND(kernel_plane * conv_param_->input_channel_, C16NUM); in InitTmpBuffer()
53 …matmul_packed_input_size_ = thread_num_ * tile_num_ * kernel_plane * conv_param_->input_channel_ *… in InitTmpBuffer()
91 conv_param_->input_channel_ = input_channel; in InitWeightBias()
303 …} else if (conv_param->group_ == conv_param->input_channel_ && conv_param->group_ == conv_param->o… in CPUConv2DFusionINT8CoderCreator()
Ddeconvolution_int8_coder.cc54 matmul_param_->deep_ = conv_param_->input_channel_; in InitParam()
77 …size = UP_ROUND(conv_param_->output_channel_, C4NUM) * UP_ROUND(conv_param_->input_channel_, C16NU… in InitBiasWeight()
84 …sInt8(reinterpret_cast<int8_t *>(filter_tensor_->data()), weight_ptr_, conv_param_->input_channel_, in InitBiasWeight()
103 UP_ROUND(conv_param_->input_channel_, C16NUM) * sizeof(int8_t); in InitData()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c40 const int channels = conv_param->input_channel_; in RollingIm2ColPackDwUnitFp32()
90 const int channels = conv_param->input_channel_ / conv_param->group_; in rolling_im2col_hwc()
91 const int tot_channels = conv_param->input_channel_; in rolling_im2col_hwc()
209 const int channels = conv_param->input_channel_ / conv_param->group_; in col2im_hwc()
210 const int tot_channels = conv_param->input_channel_; in col2im_hwc()
256 const int channels = conv_param->input_channel_ / conv_param->group_; in rolling_col2im_hwc()
257 const int tot_channels = conv_param->input_channel_; in rolling_col2im_hwc()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_depthwise_indirect_fp32.cc121 int IC_DIV = UP_DIV(conv_param_->input_channel_, div_flag); in MallocPackedInput()
140 if (conv_param_->input_channel_ % div_flag != 0) { in Run()
148 conv_param_->input_h_ * conv_param_->input_w_, conv_param_->input_channel_); in Run()
151 conv_param_->input_h_ * conv_param_->input_w_, conv_param_->input_channel_); in Run()
170 if (conv_param_->input_channel_ % div_flag != 0) { in Run()
Ddeconvolution_depthwise_fp32.cc38 conv_param_->input_channel_ = out_tensors_.front()->shape().at(kNHWC_C); in InitSlideParam()
48 if (conv_param_->input_channel_ % C4NUM != 0) { in InitPackedInputOutput()
50 int IC4 = UP_DIV(conv_param_->input_channel_, C4NUM); in InitPackedInputOutput()
139 if (conv_param_->input_channel_ != conv_param_->output_channel_) { in Run()
166 conv_param_->input_h_ * conv_param_->input_w_, conv_param_->input_channel_); in Run()
Dconvolution_fp32.cc42 …int unit_size = conv_param_->kernel_h_ * conv_param_->kernel_w_ * conv_param_->input_channel_ * C6… in InitTmpBuffer()
44 …int unit_size = conv_param_->kernel_h_ * conv_param_->kernel_w_ * conv_param_->input_channel_ * C4… in InitTmpBuffer()
47 …conv_param_->kernel_h_ * conv_param_->kernel_w_ * conv_param_->input_channel_ * C12NUM * thread_co… in InitTmpBuffer()
205 conv_param_->input_channel_ = in_channel; in MallocWeightBiasData()
Dconvolution_slidewindow_fp32.cc37 ic_res_ = conv_param_->input_channel_ % in_tile_; in Init()
120 int in_channel = conv_param_->input_channel_; in InitTmpBuffer()
131 conv_param_->input_w_ * conv_param_->input_h_, conv_param_->input_channel_); in InitTmpBuffer()
207 conv_param_->input_channel_ = input_channel; in MallocWeightBiasData()
Dconvolution_winograd_fp32.cc36 … input_unit_, kernel_unit_, conv_param_->input_channel_, conv_param_->output_channel_, in WinogradFilterTransform()
43 …thread_count_ * tile_num_ * input_unit_ * input_unit_ * conv_param_->input_channel_ * sizeof(float… in InitTmpBuffer()
66 … ctx_->allocator->Malloc(thread_count_ * tile_num_ * conv_param_->input_channel_ * sizeof(float))); in InitTmpBuffer()
200 conv_param_->input_channel_ = in_channel; in MallocWeightBiasData()
Dconvolution_delegate_fp32.cc151 conv_param->input_channel_ = input->Channel(); in SetInputOutputShapeInfo()
161 if (conv_param->input_channel_ / op_parameter_->thread_num_ <= 64 && in CheckAvxUseSWConv()
177 … conv_param->stride_h_ == 1 && conv_param->stride_w_ == 1 && conv_param->input_channel_ % 8 == 0 && in CpuConvFp32KernelSelect()
255 if (kernel == nullptr && conv_param->input_channel_ < kMaxDwConvSWSize) { in CpuConvDwFp32KernelCreator()
292 …} else if (conv_param->group_ == conv_param->input_channel_ && conv_param->group_ == conv_param->o… in CpuConvFp32KernelCreator()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c39 const int channels = conv_param->input_channel_; in RollingIm2ColPackDwUnitFp16()
89 const int channels = conv_param->input_channel_ / conv_param->group_; in RollingIm2ColPackUnitFp16()
90 const int tot_channels = conv_param->input_channel_; in RollingIm2ColPackUnitFp16()
155 const int channels = conv_param->input_channel_ / conv_param->group_; in RollingCol2ImPackUnitFp16()
156 const int tot_channels = conv_param->input_channel_; in RollingCol2ImPackUnitFp16()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_depthwise_fp16.cc39 conv_param_->input_channel_ = out_tensors_.front()->shape().at(kNHWC_C); in InitSlideParam()
49 if (conv_param_->input_channel_ % C8NUM != 0) { in InitPackedInputOutput()
51 int C8 = UP_DIV(conv_param_->input_channel_, C8NUM); in InitPackedInputOutput()
168 if (conv_param_->input_channel_ != conv_param_->output_channel_) { in Run()
193 conv_param_->input_h_ * conv_param_->input_w_, conv_param_->input_channel_); in Run()
Dconvolution_winograd_fp16.cc31 … oc_block, input_unit_, kernel_unit_, conv_param_->input_channel_, in WinogradFilterTransformFp16()
39 conv_param_->input_channel_ = in_channel; in MallocWeightBiasData()
91 …thread_count_ * row_tile_ * input_unit_ * input_unit_ * conv_param_->input_channel_ * sizeof(float… in InitTmpBuffer()
113 …ctx_->allocator->Malloc(thread_count_ * row_tile_ * conv_param_->input_channel_ * sizeof(float16_t… in InitTmpBuffer()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Ddeconvolution_grad_fp32_tests.cc40 conv_param->input_channel_ = 3; in TEST_F()
74 …aram->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_; in TEST_F()
148 conv_param->input_channel_ = 3; in TEST_F()
182 …aram->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_; in TEST_F()
250 conv_param->input_channel_ = 3; in TEST_F()
287 …aram->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_; in TEST_F()
357 conv_param->input_channel_ = 3; in TEST_F()
391 …aram->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_; in TEST_F()
464 conv_param->input_channel_ = 4; in TEST_F()
498 …aram->output_channel_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_param->input_channel_; in TEST_F()
[all …]
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dconvolution_winograd_fp32_coder.cc82 conv_param_->input_channel_, conv_param_->output_channel_, true); in WinogradFilterTransform()
89 … = thread_num_ * tile_num * input_unit_ * input_unit_ * conv_param_->input_channel_ * sizeof(float… in InitTmpBuffer()
95 col_buffer_size_ = thread_num_ * tile_num * conv_param_->input_channel_ * sizeof(float); in InitTmpBuffer()
126 conv_param_->input_channel_ = in_channel; in InitWeightBias()

12345