| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | winograd_transform.c | 25 int output_unit = conv_param->output_unit_; in WinogradInputTransform() local
35 int src_x_s = (out_tile_index % out_w_block_num) * output_unit - pad_w; in WinogradInputTransform()
36 int src_y_s = (out_tile_index / out_w_block_num) * output_unit - pad_h; in WinogradInputTransform()
101 int output_unit = conv_param->output_unit_; in WinogradOutputNHWCTransform() local
115 int r_w = output_w - dst_x_s * output_unit; in WinogradOutputNHWCTransform()
116 r_w = r_w > output_unit ? output_unit : r_w; in WinogradOutputNHWCTransform()
117 int r_h = output_h - dst_y_s * output_unit; in WinogradOutputNHWCTransform()
118 r_h = r_h > output_unit ? output_unit : r_h; in WinogradOutputNHWCTransform()
119 int tmp_ix = dst_x_s * output_unit; in WinogradOutputNHWCTransform()
121 int tmp_iy = dst_y_s * output_unit; in WinogradOutputNHWCTransform()
[all …]
|
| D | winograd_utils.h | 164 OutputTransFunc GetOutputTransFunc(int input_unit, int output_unit, ActType act_type);
|
| D | winograd_utils.c | 307 OutputTransFunc GetOutputTransFunc(int input_unit, int output_unit, ActType act_type) { in GetOutputTransFunc() argument
308 if (!CheckWinogradInputOutputUnit(input_unit, output_unit)) { in GetOutputTransFunc()
324 return OutputTransFuncList[index + (input_unit - 2) * act_index + output_unit - 2]; in GetOutputTransFunc()
|
| /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() local
40 int src_x_s = (out_tile_index % out_w_block_num) * output_unit - pad_w; in WinogradInputTransformFp16()
41 int src_y_s = (out_tile_index / out_w_block_num) * output_unit - pad_h; in WinogradInputTransformFp16()
131 int output_unit = conv_param->output_unit_; in WinogradOutputNHWCTransformFp16() local
141 int r_w = output_w - dst_x_s * output_unit; in WinogradOutputNHWCTransformFp16()
142 r_w = r_w > output_unit ? output_unit : r_w; in WinogradOutputNHWCTransformFp16()
143 int r_h = output_h - dst_y_s * output_unit; in WinogradOutputNHWCTransformFp16()
144 r_h = r_h > output_unit ? output_unit : r_h; in WinogradOutputNHWCTransformFp16()
145 int tmp_ix = dst_x_s * output_unit; in WinogradOutputNHWCTransformFp16()
147 int tmp_iy = dst_y_s * output_unit; in WinogradOutputNHWCTransformFp16()
[all …]
|
| D | winograd_utils_fp16.c | 432 OutputTransFp16Func GetOutputTransFp16Func(int input_unit, int output_unit, ActType act_type) { in GetOutputTransFp16Func() argument
433 if (input_unit == 4 && output_unit < 4) { in GetOutputTransFp16Func()
435 return OutputTransFp16FuncReluList4[output_unit]; in GetOutputTransFp16Func()
437 return OutputTransFp16FuncRelu6List4[output_unit]; in GetOutputTransFp16Func()
439 return OutputTransFp16FuncList4[output_unit]; in GetOutputTransFp16Func()
441 } else if (input_unit == 6 && output_unit < 6) { in GetOutputTransFp16Func()
443 return OutputTransFp16FuncReluList6[output_unit]; in GetOutputTransFp16Func()
445 return OutputTransFp16FuncRelu6List6[output_unit]; in GetOutputTransFp16Func()
447 return OutputTransFp16FuncList6[output_unit]; in GetOutputTransFp16Func()
449 } else if (input_unit == 8 && output_unit < 8) { in GetOutputTransFp16Func()
[all …]
|
| D | winograd_utils_fp16.h | 287 OutputTransFp16Func GetOutputTransFp16Func(int input_unit, int output_unit, ActType act_type);
490 void CheckIfUseWinogradFp16(bool *use_winograd, int *output_unit, const ConvParameter *conv_param);
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_winograd_fp32_coder.cc | 187 std::string ConvolutionWinogradFP32Coder::GetOutputTransFunc(int input_unit, int output_unit, ActTy… in GetOutputTransFunc() argument
189 if (input_unit == DIMENSION_4D && output_unit < DIMENSION_4D) { in GetOutputTransFunc()
191 return OutputTransFuncReluList4.at(output_unit); in GetOutputTransFunc()
193 return OutputTransFuncRelu6List4.at(output_unit); in GetOutputTransFunc()
195 return OutputTransFuncList4.at(output_unit); in GetOutputTransFunc()
197 } else if (input_unit == DIMENSION_6D && output_unit < DIMENSION_6D) { in GetOutputTransFunc()
199 return OutputTransFuncReluList6.at(output_unit); in GetOutputTransFunc()
201 return OutputTransFuncRelu6List6.at(output_unit); in GetOutputTransFunc()
203 return OutputTransFuncList6.at(output_unit); in GetOutputTransFunc()
205 } else if (input_unit == DIMENSION_8D && output_unit < DIMENSION_8D) { in GetOutputTransFunc()
[all …]
|
| D | convolution_winograd_fp32_coder.h | 30 … const Model::Node *node, size_t node_index, Target target, int output_unit) in ConvolutionWinogradFP32Coder() argument
31 … : Conv2DBaseCoder(in_tensors, out_tensors, node, node_index, target), output_unit_(output_unit) {} in ConvolutionWinogradFP32Coder()
52 std::string GetOutputTransFunc(int input_unit, int output_unit, ActType act_type);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | conv_common_base.c | 32 bool CheckWinogradInputOutputUnit(int input_unit, int output_unit) { in CheckWinogradInputOutputUnit() argument
36 if ((output_unit >= input_unit) || (output_unit < 2)) { in CheckWinogradInputOutputUnit()
85 bool CheckIfUseWinograd(int *output_unit, const ConvParameter *conv_param) { in CheckIfUseWinograd() argument
88 *output_unit = SelectOutputUnit(conv_param); in CheckIfUseWinograd()
89 if (*output_unit > 1) { in CheckIfUseWinograd()
|
| D | conv_common_base.h | 30 bool CheckWinogradInputOutputUnit(int input_unit, int output_unit);
32 bool CheckIfUseWinograd(int *output_unit, const ConvParameter *conv_param);
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_winograd_fp32.h | 33 int output_unit, float *origin_weight, float *origin_bias) in ConvolutionWinogradCPUKernel() argument
35 output_unit_(output_unit) {} in ConvolutionWinogradCPUKernel()
|