| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/ |
| D | conv_common_base.c | 32 bool CheckWinogradInputOutputUnit(int input_unit, int output_unit) { in CheckWinogradInputOutputUnit()
85 bool CheckIfUseWinograd(int *output_unit, const ConvParameter *conv_param) { in CheckIfUseWinograd()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | convolution_winograd_arm32_fp32.h | 28 int output_unit, float *origin_weight, float *origin_bias) in ConvolutionWinogradARM32CPUKernel()
|
| D | convolution_winograd_sse_fp32.h | 28 int output_unit, float *origin_weight, float *origin_bias) in ConvolutionWinogradSSECPUKernel()
|
| D | convolution_winograd_arm64_fp32.h | 28 int output_unit, float *origin_weight, float *origin_bias) in ConvolutionWinogradARM64CPUKernel()
|
| D | convolution_winograd_avx_fp32.h | 28 int output_unit, float *origin_weight, float *origin_bias) in ConvolutionWinogradAVXCPUKernel()
|
| D | convolution_winograd_base_fp32.h | 33 int output_unit, float *origin_weight, float *origin_bias) in ConvolutionWinogradBaseCPUKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_winograd_fp32_coder.h | 31 … const LiteGraph::Node *node, size_t node_index, Target target, int output_unit) in ConvolutionWinogradFP32Coder()
|
| D | convolution_winograd_fp32_coder.cc | 192 std::string ConvolutionWinogradFP32Coder::GetOutputTransFunc(int input_unit, int output_unit, ActTy… in GetOutputTransFunc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/ |
| D | convolution_winograd_fp16_coder.h | 37 … const LiteGraph::Node *node, size_t node_index, Target target, int output_unit) in ConvolutionWinogradFP16Coder()
|
| D | convolution_winograd_fp16_coder.cc | 175 std::string ConvolutionWinogradFP16Coder::GetOutputTransFunc(int input_unit, int output_unit, ActTy… in GetOutputTransFunc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | winograd_transform.c | 79 int output_unit = conv_param->output_unit_; in WinogradInputTransform() local
127 int output_unit = conv_param->output_unit_; in WinogradInputTransformOptStep() local
169 int output_unit = conv_param->output_unit_; in WinogradOutputNHWCTransform() local
228 int output_unit = conv_param->output_unit_; in WinogradOutputNC4HW4Transform() local
|
| D | winograd_utils.c | 881 OutputTransFunc GetOutputTransFunc(int input_unit, int output_unit, ActType act_type) { in GetOutputTransFunc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | winograd_transform_fp16.c | 100 int output_unit = conv_param->output_unit_; in WinogradInputTransformFp16() local
145 int output_unit = conv_param->output_unit_; in WinogradInputTransformOptStepFp16() local
187 int output_unit = conv_param->output_unit_; in WinogradOutputNHWCTransformFp16() local
226 int output_unit = conv_param->output_unit_; in WinogradOutputNC8HW8TransformFp16() local
|
| D | winograd_utils_fp16.c | 774 OutputTransFp16Func GetOutputTransFp16Func(int input_unit, int output_unit, ActType act_type) { in GetOutputTransFp16Func()
3268 void CheckIfUseWinogradFp16(bool *use_winograd, int *output_unit, const ConvParameter *conv_param) { in CheckIfUseWinogradFp16()
|