Home
last modified time | relevance | path

Searched defs:op_parameter (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconvolution_int8_creator.cc42 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvDwInt8KernelCreator()
77 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvInt8KernelSelect()
103OpParameter *op_parameter, const lite::InnerContext *ctx, in CpuGroupConvInt8KernelCreator()
117 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvInt8KernelCreator()
Ddeconvolution_int8.cc307 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuDeConvInt8KernelCreator()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Darithmetic_self_grad.h26 OpParameter op_parameter; member
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dcommon.cc31OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain()
41OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain()
205OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain()
217OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain()
369OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dactivation_grad.h25 OpParameter op_parameter; member
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_delegate_fp32.cc268OpParameter *op_parameter, const lite::InnerContext *ctx) { in CpuGroupConvFp32KernelCreator()
281 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvFp32KernelCreator()
Ddeconvolution_fp32.cc289 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuDeConvFp32KernelCreator()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconvolution_delegate_fp16.cc156 …tor<lite::Tensor *> &inputs, const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvFp16KernelSelect()
193OpParameter *op_parameter, const InnerContext *ctx) { in CpuGroupConvFp16KernelCreator()
Ddeconvolution_fp16.cc254 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuDeConvFp16KernelCreator()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dgroup_convolution_creator.h36 …std::vector<lite::Tensor *> inputs, std::vector<lite::Tensor *> outputs, OpParameter *op_parameter, in GroupConvCreator()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dfusion_eltwise.cc51 auto *op_parameter = reinterpret_cast<OpenCLKernel *>(node->kernel())->GetParameter(); in CheckSupport() local
85 auto *op_parameter = reinterpret_cast<OpenCLKernel *>(node->kernel())->GetParameter(); local
/third_party/mindspore/mindspore/lite/src/
Dinner_kernel.h77 OpParameter *op_parameter() const { return op_parameter_; } in op_parameter() function
Dscheduler.cc867OpParameter *op_parameter, const kernel::KernelKey &desc, TypeId kernel_data_type, in FindCpuKernel()
915OpParameter *op_parameter, const kernel::KernelKey &desc, kernel::LiteKernel **kernel, in FindGpuKernel()
1022 OpParameter *op_parameter = op_parameters_[node->output_indices_.at(0)]; in FindBackendKernel() local
1269 OpParameter *op_parameter = op_parameters_[node->output_indices_.at(0)]; in SubGraphPreferDataType() local
Dweight_decoder.h217 inline static bool IsChannelFirst(int index, const OpParameter *op_parameter) { in IsChannelFirst()
Dlite_kernel.h133 OpParameter *op_parameter() const { in op_parameter() function
Dweight_decoder.cc323 int WeightDecoder::DequantNode(OpParameter *op_parameter, const std::vector<Tensor *> &in_tensors, in DequantNode()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dconv2d_delegate_fp32_coder.cc41 OpParameter *op_parameter = parameter_gen(node_->primitive_); in Prepare() local