/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
D | convolution_winograd_fp16.cc | 64 …matrix_a, matrix_at, matrix_b, matrix_bt, matrix_g_, matrix_gt_, coef, output_unit_, kernel_unit_); in MallocWeightBiasData() 132 out_func_ = GetOutputTransFp16Func(input_unit_, output_unit_, conv_param_->act_type_); in ConfigInputOutput() 160 input_unit_ = output_unit_ + kernel_unit_ - 1; in Init() 162 conv_param_->output_unit_ = output_unit_; in Init() 174 …int cal_plane = UP_DIV(out_tensor->Height(), output_unit_) * UP_DIV(out_tensor->Width(), output_un… in AdjustNumberOfThread()
|
D | convolution_winograd_fp16.h | 35 …seCPUKernel(parameter, inputs, outputs, ctx, origin_weight, origin_bias), output_unit_(out_unit) {} in ConvolutionWinogradFP16CPUKernel() 72 int output_unit_; variable
|
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
D | convolution_winograd_fp32.cc | 85 out_func_ = GetOutputTransFunc(input_unit_, output_unit_, conv_param_->act_type_); in ConfigInputOutput() 103 input_unit_ = output_unit_ + kernel_unit_ - 1; in Init() 105 conv_param_->output_unit_ = output_unit_; in Init() 226 …matrix_a, matrix_at, matrix_b, matrix_bt, matrix_g_, matrix_gt_, coef, output_unit_, kernel_unit_); in MallocWeightBiasData()
|
D | convolution_winograd_fp32.h | 35 output_unit_(output_unit) {} in ConvolutionWinogradCPUKernel() 68 int output_unit_{0};
|
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
D | convolution_winograd_fp32_coder.cc | 113 input_unit_ = output_unit_ + kernel_unit_ - 1; in Prepare() 115 conv_param_->output_unit_ = output_unit_; in Prepare() 149 …r(matrix_a, matrix_at, matrix_b, matrix_bt, matrix_g, matrix_gt, coef, output_unit_, kernel_unit_); in InitWeightBias() 177 out_func_ = GetOutputTransFunc(input_unit_, output_unit_, conv_param_->act_type_); in ConfigInputOutput()
|
D | convolution_winograd_fp32_coder.h | 31 … : Conv2DBaseCoder(in_tensors, out_tensors, node, node_index, target), output_unit_(output_unit) {} in ConvolutionWinogradFP32Coder() 59 int output_unit_{0};
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
D | conv_winograd_fp32.c | 27 if (conv_param->output_unit_ == 0) { in ConvWinogardFp32() 31 int out_w_block = UP_DIV(conv_param->output_w_, conv_param->output_unit_); in ConvWinogardFp32() 32 int out_h_block = UP_DIV(conv_param->output_h_, conv_param->output_unit_); in ConvWinogardFp32()
|
D | winograd_transform.c | 25 int output_unit = conv_param->output_unit_; in WinogradInputTransform() 101 int output_unit = conv_param->output_unit_; in WinogradOutputNHWCTransform() 160 int output_unit = conv_param->output_unit_; in WinogradOutputNC4HW4Transform()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
D | winograd_transform_fp16.c | 29 int output_unit = conv_param->output_unit_; in WinogradInputTransformFp16() 131 int output_unit = conv_param->output_unit_; in WinogradOutputNHWCTransformFp16() 170 int output_unit = conv_param->output_unit_; in WinogradOutputNC8HW8TransformFp16()
|
D | conv_fp16.c | 196 NNACL_CHECK_ZERO_RETURN(conv_param->output_unit_); in ConvWinogardFp16() 199 int out_w_block = UP_DIV(conv_param->output_w_, conv_param->output_unit_); in ConvWinogardFp16() 200 int out_h_block = UP_DIV(conv_param->output_h_, conv_param->output_unit_); in ConvWinogardFp16()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
D | conv_parameter.h | 51 int output_unit_; member
|
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/ |
D | nnacl_fp32_serializer.cc | 70 …v_parameter.output_channel_, "thread_num", conv_parameter.input_unit_, conv_parameter.output_unit_, in CodeStruct()
|
D | nnacl_int8_serializer.cc | 60 …conv_parameter.input_unit_, conv_parameter.output_unit_, conv_parameter.pad_mode_, conv_parameter.… in CodeStruct()
|