Home
last modified time | relevance | path

Searched defs:GetOutputTransFunc (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Dconvolution_winograd_fp16_coder.cc175 std::string ConvolutionWinogradFP16Coder::GetOutputTransFunc(int input_unit, int output_unit, ActTy… in GetOutputTransFunc() function in mindspore::lite::micro::nnacl::ConvolutionWinogradFP16Coder
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Dconvolution_winograd_fp32_coder.cc192 std::string ConvolutionWinogradFP32Coder::GetOutputTransFunc(int input_unit, int output_unit, ActTy… in GetOutputTransFunc() function in mindspore::lite::micro::nnacl::ConvolutionWinogradFP32Coder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dwinograd_utils.c881 OutputTransFunc GetOutputTransFunc(int input_unit, int output_unit, ActType act_type) { in GetOutputTransFunc() function