Searched defs:input_unit (Results 1 – 8 of 8) sorted by relevance
| /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()
65 int input_unit = i + kernel_w - 1; in SelectOutputUnit() local
|
| D | minimal_filtering_generator.c | 256 … int oc_block, int input_unit, int kernel_unit, int channel, int batch, bool pack) { in WinogradWeightTransform()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_winograd_fp32_coder.cc | 182 std::string ConvolutionWinogradFP32Coder::GetInputTransFunc(int input_unit) { in GetInputTransFunc()
187 std::string ConvolutionWinogradFP32Coder::GetOutputTransFunc(int input_unit, int output_unit, ActTy… in GetOutputTransFunc()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | winograd_transform_fp16.c | 28 int input_unit = conv_param->input_unit_; in WinogradInputTransformFp16() local
136 int input_unit = conv_param->input_unit_; in WinogradOutputNHWCTransformFp16() local
176 int input_unit = conv_param->input_unit_; in WinogradOutputNC8HW8TransformFp16() local
208 … const float *matrix_gt, int oc_block, int input_unit, int kernel_unit, in WinogradWeightTransformFp16()
|
| D | winograd_utils_fp16.c | 82 InputTransFp16Func GetInputTransFp16Func(int input_unit) { return InputTransFp16FuncList[input_unit… in GetInputTransFp16Func()
432 OutputTransFp16Func GetOutputTransFp16Func(int input_unit, int output_unit, ActType act_type) { in GetOutputTransFp16Func()
2906 int input_unit = i + kernel_w - 1; in SelectOutputUnitFp16() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | winograd_transform.c | 24 int input_unit = conv_param->input_unit_; in WinogradInputTransform() local
109 int input_unit = conv_param->input_unit_; in WinogradOutputNHWCTransform() local
169 int input_unit = conv_param->input_unit_; in WinogradOutputNC4HW4Transform() local
|
| D | winograd_utils.c | 53 InputTransFunc GetInputTransFunc(int input_unit) { return InputTransFuncList[input_unit]; } in GetInputTransFunc()
307 OutputTransFunc GetOutputTransFunc(int input_unit, int output_unit, ActType act_type) { in GetOutputTransFunc()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/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
|