| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/ |
| D | convolution_winograd_fp16_coder.cc | 158 std::string ConvolutionWinogradFP16Coder::GetInputTransFunc(int input_unit) { in GetInputTransFunc()
163 std::string ConvolutionWinogradFP16Coder::GetInputTransStepFunc(int input_unit) { in GetInputTransStepFunc()
169 std::string ConvolutionWinogradFP16Coder::GetInputTransPackFunc(int input_unit) { in GetInputTransPackFunc()
175 std::string ConvolutionWinogradFP16Coder::GetOutputTransFunc(int input_unit, int output_unit, ActTy… in GetOutputTransFunc()
|
| D | conv2d_delegate_fp16_coder.cc | 84 int input_unit = i + kernel_w - 1; in SelectOutUnit() local
|
| /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()
65 int input_unit = i + kernel_w - 1; in SelectOutputUnit() local
|
| D | minimal_filtering_generator.c | 257 … int oc_block, int input_unit, int kernel_unit, int channel, int batch, bool pack) { in WinogradWeightTransform()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | winograd_transform_fp16.c | 21 int input_unit = conv_param->input_unit_; in PrepareTransInputFp16() local
99 int input_unit = conv_param->input_unit_; in WinogradInputTransformFp16() local
144 int input_unit = conv_param->input_unit_; in WinogradInputTransformOptStepFp16() local
192 int input_unit = conv_param->input_unit_; in WinogradOutputNHWCTransformFp16() local
232 int input_unit = conv_param->input_unit_; in WinogradOutputNC8HW8TransformFp16() local
264 … const float *matrix_gt, int oc_block, int input_unit, int kernel_unit, in WinogradWeightTransformFp16()
|
| D | conv_fp16.c | 246 int input_unit = conv_param->input_unit_; in ConvWinogardFp16() local
|
| D | winograd_utils_fp16.c | 114 InputTransFp16Func GetInputTransFp16Func(int input_unit) { return InputTransFp16FuncList[input_unit… in GetInputTransFp16Func()
130 InputTransStepFp16Func GetInputTransStepFp16Func(int input_unit) { return InputTransStepFp16FuncLis… in GetInputTransStepFp16Func()
132 InputTransPackFp16Func GetInputTransPackFp16Func(int input_unit) { return InputTransPackFp16FuncLis… in GetInputTransPackFp16Func()
774 OutputTransFp16Func GetOutputTransFp16Func(int input_unit, int output_unit, ActType act_type) { in GetOutputTransFp16Func()
3248 int input_unit = i + kernel_w - 1; in SelectOutputUnitFp16() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_winograd_fp32_coder.cc | 187 std::string ConvolutionWinogradFP32Coder::GetInputTransFunc(int input_unit) { in GetInputTransFunc()
192 std::string ConvolutionWinogradFP32Coder::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 | 23 int input_unit = conv_param->input_unit_; in PrepareTransInput() local
78 int input_unit = conv_param->input_unit_; in WinogradInputTransform() local
126 int input_unit = conv_param->input_unit_; in WinogradInputTransformOptStep() local
177 int input_unit = conv_param->input_unit_; in WinogradOutputNHWCTransform() local
237 int input_unit = conv_param->input_unit_; in WinogradOutputNC4HW4Transform() local
|
| D | conv_winograd_fp32.c | 31 int input_unit = conv_param->input_unit_; in ConvWinogardFp32() local
155 int input_unit = conv_param->input_unit_; in ConvWinogardFp32CutByBatch() local
|
| D | winograd_utils.c | 69 InputTransFunc GetInputTransFunc(int input_unit) { return InputTransFuncList[input_unit]; } in GetInputTransFunc()
78 InputTransStepFunc GetInputTransStepFunc(int input_unit) { return InputTransStepFuncList[input_unit… in GetInputTransStepFunc()
80 InputTransPackFunc GetInputTransPackFunc(int input_unit) { return InputTransPackFuncList[input_unit… in GetInputTransPackFunc()
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/int8/ |
| D | conv3x3_int8.c | 190 const int input_unit = 4; in Conv3x3Int8FilterTransform() local
754 const int input_unit = 4; in Conv3x3Int8OutputTransform() local
792 const int input_unit = 4; in Conv3x3Int8InputTransform() local
|