Home
last modified time | relevance | path

Searched defs:InitParam (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Ddeconvolution_int8_coder.cc52 int DeconvolutionInt8Coder::InitParam() { in InitParam() function in mindspore::lite::micro::nnacl::DeconvolutionInt8Coder
Dconv2d_1x1_int8_coder.cc253 int Conv2D1x1Int8Coder::InitParam() { in InitParam() function in mindspore::lite::micro::nnacl::Conv2D1x1Int8Coder
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Ddeconv2d_fp32_coder.cc50 int DeConvolutionFP32Coder::InitParam() { in InitParam() function in mindspore::lite::micro::nnacl::DeConvolutionFP32Coder
Dlstm_fp32_coder.cc104 int LstmFP32Coder::InitParam() { in InitParam() function in mindspore::lite::micro::nnacl::LstmFP32Coder
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Dgru_fp16.cc61 int GruFp16CPUKernel::InitParam() { in InitParam() function in mindspore::kernel::GruFp16CPUKernel
Dlstm_fp16_base.cc110 int LstmFp16BaseCPUKernel::InitParam() { in InitParam() function in mindspore::kernel::LstmFp16BaseCPUKernel
Ddeconvolution_fp16.cc96 int DeConvolutionFp16CPUKernel::InitParam() { in InitParam() function in mindspore::kernel::DeConvolutionFp16CPUKernel
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dgru_fp32.cc61 int GruCPUKernel::InitParam() { in InitParam() function in mindspore::kernel::GruCPUKernel
Ddeconvolution_fp32.cc111 int DeConvolutionCPUKernel::InitParam() { in InitParam() function in mindspore::kernel::DeConvolutionCPUKernel
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/kernel_builder/ascend/ascendc/op_kernel/
Dprompt_kv_cache.cpp93 __aicore__ inline void InitParam(GM_ADDR tiling) { in InitParam() function in KernelPromptKvCache
181 __aicore__ inline void InitParam(GM_ADDR tiling) { in InitParam() function in KernelPromptKvCacheCopyAll
Ddecoder_kv_cache.cpp94 __aicore__ inline void InitParam(GM_ADDR tiling) { in InitParam() function in KernelDecoderKvCache
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/ascendc/op_kernel/
Dprompt_kv_cache.cpp93 __aicore__ inline void InitParam(GM_ADDR tiling) { in InitParam() function in KernelPromptKvCache
181 __aicore__ inline void InitParam(GM_ADDR tiling) { in InitParam() function in KernelPromptKvCacheCopyAll
Ddecoder_kv_cache.cpp94 __aicore__ inline void InitParam(GM_ADDR tiling) { in InitParam() function in KernelDecoderKvCache
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Ddeconvolution_int8.cc142 int DeConvInt8CPUKernel::InitParam() { in InitParam() function in mindspore::kernel::DeConvInt8CPUKernel
Dconvolution_1x1_int8.cc359 int Convolution1x1Int8CPUKernel::InitParam() { in InitParam() function in mindspore::kernel::Convolution1x1Int8CPUKernel
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/
Dlstm_grad_weight_fp32.cc165 int LSTMGradWeightCPUKernel::InitParam() { in InitParam() function in mindspore::kernel::LSTMGradWeightCPUKernel
Dlstm_grad_data_fp32.cc184 int LSTMGradDataCPUKernel::InitParam() { in InitParam() function in mindspore::kernel::LSTMGradDataCPUKernel
Dlstm_grad_fp32.cc197 int LSTMGradCPUKernel::InitParam() { in InitParam() function in mindspore::kernel::LSTMGradCPUKernel
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Dlstm_mindir_dynamic_fp16_coder.cc97 int LstmMindirDynamicFP16Coder::InitParam() { in InitParam() function in mindspore::lite::micro::nnacl::LstmMindirDynamicFP16Coder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc980 DfGraphConvertor &DfGraphConvertor::InitParam(const TensorOrderMap &tensors) { in InitParam() function in mindspore::transform::DfGraphConvertor