Home
last modified time | relevance | path

Searched refs:output_batch_ (Results 1 – 25 of 51) sorted by relevance

123

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Dpooling_grad_fp32_tests.cc40 pooling_param->output_batch_ = 1; in InitPoolingParamFP32()
72 …pooling_param->output_batch_ * pooling_param->output_channel_ * pooling_param->output_h_ * pooling… in TEST_F()
85 AvgPoolingGrad(input_data, output_data, pooling_param->output_batch_, pooling_param); in TEST_F()
92 AvgPoolingGrad(input_data, output_data, pooling_param->output_batch_, pooling_param); in TEST_F()
127 …pooling_param->output_batch_ * pooling_param->output_channel_ * pooling_param->output_h_ * pooling… in TEST_F()
196 pooling_param->output_batch_ = 3; in TEST_F()
269 pool->output_batch_ = 3; in TEST_F()
280 …std::vector<int> dim_x({pool->output_batch_, pool->input_h_, pool->input_w_, pool->input_channel_}… in TEST_F()
287 …std::vector<int> dim_y({pool->output_batch_, pool->output_h_, pool->output_w_, pool->output_channe… in TEST_F()
333 pool->output_batch_ = 3; in TEST_F()
[all …]
Ddeconvolution_grad_fp32_tests.cc42 conv_param->output_batch_ = 2; in TEST_F()
150 conv_param->output_batch_ = 2; in TEST_F()
252 conv_param->output_batch_ = 2; in TEST_F()
359 conv_param->output_batch_ = 2; in TEST_F()
466 conv_param->output_batch_ = 2; in TEST_F()
573 conv_param->output_batch_ = 2; in TEST_F()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dpooling_fp16_grad.cc68 pool_param->output_batch_ = out_shape[kNHWC_N]; in ReSize()
82 int stride = UP_DIV(pool_param->output_batch_, thread_num_); in DoExecute()
83 int count = MSMIN(stride, pool_param->output_batch_ - stride * task_id); in DoExecute()
Dconvolution_fp16_grad_input.cc41 conv_param->output_batch_ = dx_tensor->shape()[(kNHWC_N)]; in ReSize()
95 int batch = conv_param->output_batch_; in DoExecute()
169 int batch = conv_param->output_batch_; in Run()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dpooling_grad.cc68 pool_param->output_batch_ = out_shape[kNHWC_N]; in ReSize()
83 int stride = UP_DIV(pool_param->output_batch_, thread_num_); in Execute()
84 int count = MSMIN(stride, pool_param->output_batch_ - stride * task_id); in Execute()
Dconvolution_grad_input.cc41 conv_param->output_batch_ = dx_tensor->shape()[(kNHWC_N)]; in ReSize()
96 int batch = conv_param->output_batch_; in Execute()
172 int batch = conv_param->output_batch_; in Run()
Ddeconvolution_grad_filter.cc48 conv_param->output_batch_ = dy_tensor->shape().at(kNHWC_N); in Init()
92 int batch = conv_param->output_batch_; in Execute()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dconvolution_grad_filter.c32 int batch = conv_param->output_batch_; in FilterGrad32Arm()
110 int batch = conv_param->output_batch_; in FilterGrad16Arm()
168 int batch = conv_param->output_batch_; in FilterGrad8Arm()
217 int batch = conv_param->output_batch_; in FilterGrad4Arm()
265 int batch = conv_param->output_batch_; in FilterGradLeftoverArm()
315 int batch = conv_param->output_batch_; in ConvDwFilterFp16Grad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dconvolution_grad_filter.c30 int batch = conv_param->output_batch_; in FilterGrad16Arm()
107 int batch = conv_param->output_batch_; in FilterGrad12Arm()
177 int batch = conv_param->output_batch_; in FilterGrad8Arm()
236 int batch = conv_param->output_batch_; in FilterGrad4Arm()
286 int batch = conv_param->output_batch_; in Filtergrad2Arm()
334 int batch = conv_param->output_batch_; in ConvDwFilterGrad()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_depthwise_fp16.cc40 conv_param_->output_batch_ = in_tensors_.front()->shape().at(kNHWC_N); in InitSlideParam()
59 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… in InitPackedInputOutput()
208 PackNHWC8ToNHWCFp16(packed_output_, output_ptr, conv_param_->output_batch_, in Run()
Dconvolution_depthwise_slidewindow_fp16.cc44 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… in InitPackedInputOutput()
182 PackNHWC8ToNHWCFp16(packed_output_, output_ptr, conv_param_->output_batch_, in Run()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_depthwise_fp32.cc39 conv_param_->output_batch_ = in_tensors_.front()->shape().at(kNHWC_N); in InitSlideParam()
59 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… in InitPackedInputOutput()
182 PackNHWCXToNHWCFp32(packed_output_, output_addr, conv_param_->output_batch_, in Run()
Dconvolution_depthwise_slidewindow_x86_fp32.cc50 …conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w_ * oc_tile_ * oc_algin; in InitPackedInputOutput()
149 PackNHWCXToNHWCFp32(packed_output_, output_ptr, conv_param_->output_batch_, in Run()
Dconvolution_depthwise_slidewindow_fp32.cc44 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… in InitPackedInputOutput()
147 PackNHWCXToNHWCFp32(packed_output_, output_ptr, conv_param_->output_batch_, in Run()
Dconvolution_fp32.cc66 …conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w_ * OC_BLOCK * oc_algin; in InitTmpBuffer()
161 PackNC8HW8AlignedToNC8HW8NotAlignedFp32(tmp_output_, output_addr, conv_param_->output_batch_, in Run()
Dconvolution_slidewindow_fp32.cc145 …ctx_->allocator->Malloc(conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_… in InitTmpBuffer()
181 PackNHWCXToNHWCFp32(output_data_, out_data, conv_param_->output_batch_, in Run()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Ddeconvolution_depthwise_int8.cc83 conv_param_->output_batch_ = in_tensors_.front()->shape().at(kNHWC_N); in InitSlideParam()
109 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… in InitBuffer()
227 PackNHWC4ToNHWCInt8(packed_output_, output_addr, conv_param_->output_batch_, in Run()
Dconvolution_depthwise_slidewindow_int8.cc85 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… in InitPackedInputOutput()
321 PackNHWCToNHWC8Int8(input_addr, packed_input_, conv_param_->output_batch_, in Run()
339 PackNHWC8ToNHWCInt8(packed_output_, output_addr, conv_param_->output_batch_, in Run()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dpooling_parameter.h45 int output_batch_; member
Dconv_parameter.h45 int output_batch_; member
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_3x3_int8_coder.cc85 int output_batch = conv_param_->output_batch_; in InitTmpBuffer()
185 code.CodeFunction("PackNC4HW4ToNHWCInt8", tmp_out_, output_tensor_, conv_param_->output_batch_, in DoCode()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dconv2d_tests.cc31 …m->input_batch_, &param->input_h_, &param->input_w_, &param->input_channel_, &param->output_batch_, in CreateParameter()
46 …std::vector<int> output_shape = {param->output_batch_, param->output_h_, param->output_w_, param->… in TestMain_Conv2D()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dconvolution_depthwise_fp32_tests.cc36 conv_param->output_batch_ = 1; in InitConvDwParam()
93 …{conv_param->output_batch_, conv_param->output_h_, conv_param->output_w_, conv_param->output_chann… in InitConvDwCreator()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c33 int output_batch = pooling_param->output_batch_; in AvgPoolingInt8()
98 for (int batch = 0; batch < pooling_param->output_batch_; batch++) { in AvgPoolingOptInt8()
289 int output_batch = pooling_param->output_batch_; in MaxPoolingInt8()
345 for (int batch = 0; batch < pooling_param->output_batch_; batch++) { in MaxPoolingWithQuantInt8()
433 for (int batch = 0; batch < pooling_param->output_batch_; batch++) { in MaxPoolingOptInt8()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dpooling_fp32_coder.cc37 pooling_parameter->output_batch_ = output_tensor_->Batch(); in DoCode()

123