Home
last modified time | relevance | path

Searched refs:op_parameter (Results 1 – 25 of 29) sorted by relevance

12

/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() argument
44 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvDwInt8KernelCreator()
63 …kernel = new (std::nothrow) kernel::ConvolutionDepthwise3x3Int8CPUKernel(op_parameter, inputs, out… in CpuConvDwInt8KernelCreator()
67 …kernel = new (std::nothrow) kernel::ConvolutionDepthwiseInt8CPUKernel(op_parameter, inputs, output… in CpuConvDwInt8KernelCreator()
70 …kernel = new (std::nothrow) kernel::ConvolutionDepthwiseSWInt8CPUKernel(op_parameter, inputs, outp… in CpuConvDwInt8KernelCreator()
77 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvInt8KernelSelect() argument
79 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvInt8KernelSelect()
86 kernel = new (std::nothrow) ConvolutionInt8CPUKernel(op_parameter, inputs, outputs, ctx); in CpuConvInt8KernelSelect()
88 kernel = new (std::nothrow) Convolution3x3Int8CPUKernel(op_parameter, inputs, outputs, ctx); in CpuConvInt8KernelSelect()
91 …kernel = new (std::nothrow) kernel::Convolution3x3Int8CPUKernel(op_parameter, inputs, outputs, ctx… in CpuConvInt8KernelSelect()
[all …]
Ddeconvolution_int8.cc307 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuDeConvInt8KernelCreator() argument
309 MS_CHECK_TRUE_RET(op_parameter != nullptr, nullptr); in CpuDeConvInt8KernelCreator()
314 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuDeConvInt8KernelCreator()
319 …kernel::DeConvInt8CPUKernel(op_parameter, inputs, outputs, static_cast<const lite::InnerContext *>… in CpuDeConvInt8KernelCreator()
322 op_parameter, inputs, outputs, static_cast<const lite::InnerContext *>(ctx)); in CpuDeConvInt8KernelCreator()
330 free(op_parameter); in CpuDeConvInt8KernelCreator()
Dconvolution_int8_creator.h26 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter,
/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() argument
158 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvFp16KernelSelect()
166 …kernel::Convolution1x1FP16CPUKernel(op_parameter, inputs, outputs, ctx, origin_weight, origin_bias… in CpuConvFp16KernelSelect()
168 …kernel = new (std::nothrow) kernel::ConvolutionWinogradFP16CPUKernel(op_parameter, inputs, outputs… in CpuConvFp16KernelSelect()
172 … kernel::ConvolutionFP16CPUKernel(op_parameter, inputs, outputs, ctx, origin_weight, origin_bias); in CpuConvFp16KernelSelect()
176 free(op_parameter); in CpuConvFp16KernelSelect()
193 … OpParameter *op_parameter, const InnerContext *ctx) { in CpuGroupConvFp16KernelCreator() argument
195 …new (std::nothrow) GroupConvCreator(inputs, outputs, op_parameter, ctx, false, kNumberTypeFloat16); in CpuGroupConvFp16KernelCreator()
198 free(op_parameter); in CpuGroupConvFp16KernelCreator()
202 …op_parameter, inputs, outputs, ctx, group_conv_creator, reinterpret_cast<ConvParameter *>(op_param… in CpuGroupConvFp16KernelCreator()
[all …]
Ddeconvolution_fp16.cc254 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuDeConvFp16KernelCreator() argument
256 MS_CHECK_TRUE_RET(op_parameter != nullptr, nullptr); in CpuDeConvFp16KernelCreator()
261 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuDeConvFp16KernelCreator()
265 kernel = new (std::nothrow) kernel::DeConvWinogradFp16CPUKernel(op_parameter, inputs, outputs, in CpuDeConvFp16KernelCreator()
269 …kernel::DeConvolutionFp16CPUKernel(op_parameter, inputs, outputs, static_cast<const lite::InnerCon… in CpuDeConvFp16KernelCreator()
273 …DeconvolutionDepthwiseFp16CPUKernel(op_parameter, inputs, outputs, static_cast<const lite::InnerCo… in CpuDeConvFp16KernelCreator()
278 free(op_parameter); in CpuDeConvFp16KernelCreator()
Dconvolution_delegate_fp16.h81 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter,
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dcommon.cc31 … OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain() argument
37 TestMain(input_infos_new, output_info, op_parameter, fp16_enable, atol, rtol, print_data); in TestMain()
41 … OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain() argument
42 auto primitive_type = static_cast<schema::PrimitiveType>(op_parameter->type_); in TestMain()
112 free(op_parameter); in TestMain()
115 auto *inner_kernel = creator(kernel_inputs, outputs, op_parameter, nullptr, key); in TestMain()
118 free(op_parameter); in TestMain()
124 free(op_parameter); in TestMain()
131 free(op_parameter); in TestMain()
205 … OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain() argument
[all …]
Dcommon.h97 … OpParameter *op_parameter, bool fp16_enable = false, float atol = 1e-9, float rtol = 1e-9,
101 … OpParameter *op_parameter, bool fp16_enable = false, float atol = 1e-9, float rtol = 1e-9,
105 … OpParameter *op_parameter, bool fp16_enable = false, float atol = 1e-9, float rtol = 1e-9,
109 … OpParameter *op_parameter, bool fp16_enable = false, float atol = 1e-9, float rtol = 1e-9,
112 … OpParameter *op_parameter, bool fp16_enable = false, float atol = 1e-9, float rtol = 1e-9,
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_delegate_fp32.cc268 … OpParameter *op_parameter, const lite::InnerContext *ctx) { in CpuGroupConvFp32KernelCreator() argument
269 …auto *group_conv_creator = new GroupConvCreator(inputs, outputs, op_parameter, ctx, false, kNumber… in CpuGroupConvFp32KernelCreator()
271 …op_parameter, inputs, outputs, ctx, group_conv_creator, reinterpret_cast<ConvParameter *>(op_param… in CpuGroupConvFp32KernelCreator()
281 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvFp32KernelCreator() argument
283 MS_ASSERT(op_parameter != nullptr); in CpuConvFp32KernelCreator()
287 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuConvFp32KernelCreator()
291 …kernel::ConvolutionDelegateCPUKernel(op_parameter, inputs, outputs, static_cast<const lite::InnerC… in CpuConvFp32KernelCreator()
293 …kernel = CpuConvDwFp32KernelCreator(inputs, outputs, op_parameter, static_cast<const lite::InnerCo… in CpuConvFp32KernelCreator()
295 …kernel = CpuGroupConvFp32KernelCreator(inputs, outputs, op_parameter, static_cast<const lite::Inne… in CpuConvFp32KernelCreator()
300 free(op_parameter); in CpuConvFp32KernelCreator()
Ddeconvolution_fp32.cc289 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuDeConvFp32KernelCreator() argument
291 MS_CHECK_TRUE_RET(op_parameter != nullptr, nullptr); in CpuDeConvFp32KernelCreator()
296 auto conv_param = reinterpret_cast<ConvParameter *>(op_parameter); in CpuDeConvFp32KernelCreator()
306 … kernel = new (std::nothrow) kernel::DeConvolutionWinogradCPUKernel(op_parameter, inputs, outputs, in CpuDeConvFp32KernelCreator()
310 …kernel::DeConvolutionCPUKernel(op_parameter, inputs, outputs, static_cast<const lite::InnerContext… in CpuDeConvFp32KernelCreator()
315 … kernel = new (std::nothrow) kernel::DeConvolutionWinogradCPUKernel(op_parameter, inputs, outputs, in CpuDeConvFp32KernelCreator()
319 …kernel::DeConvolutionCPUKernel(op_parameter, inputs, outputs, static_cast<const lite::InnerContext… in CpuDeConvFp32KernelCreator()
323 … kernel = new (std::nothrow) kernel::DeconvolutionDepthwiseCPUKernel(op_parameter, inputs, outputs, in CpuDeConvFp32KernelCreator()
332 free(op_parameter); in CpuDeConvFp32KernelCreator()
/third_party/mindspore/mindspore/lite/src/
Dscheduler.cc205 ret = CopyConstTensorData(kernel->in_tensors(), kernel->op_parameter()->type_); in HandleBuildinCpuKernelWeight()
867 … OpParameter *op_parameter, const kernel::KernelKey &desc, TypeId kernel_data_type, in FindCpuKernel() argument
869 MS_ASSERT(op_parameter != nullptr); in FindCpuKernel()
870 auto op_type = op_parameter->type_; in FindCpuKernel()
884 ret = WeightDecoder::DequantNode(op_parameter, in_tensors, kernel_data_type); in FindCpuKernel()
900 …y::GetInstance()->GetKernel(in_tensors, out_tensors, context_, ms_context_, cpu_desc, op_parameter, in FindCpuKernel()
915 … OpParameter *op_parameter, const kernel::KernelKey &desc, kernel::LiteKernel **kernel, in FindGpuKernel() argument
917 MS_ASSERT(op_parameter != nullptr); in FindGpuKernel()
934 ret = WeightDecoder::DequantNode(op_parameter, in_tensors, kNumberTypeFloat32); in FindGpuKernel()
941 ret = CopyConstTensorData(in_tensors, op_parameter->type_); in FindGpuKernel()
[all …]
Dweight_decoder.h131 …static int DequantNode(OpParameter *op_parameter, const std::vector<Tensor *> &in_tensors, TypeId …
217 inline static bool IsChannelFirst(int index, const OpParameter *op_parameter) { in IsChannelFirst() argument
218 MS_ASSERT(op_parameter != nullptr); in IsChannelFirst()
219 if (op_parameter->type_ == schema::PrimitiveType_MatMulFusion) { in IsChannelFirst()
220 const auto *param = reinterpret_cast<const MatMulParameter *>(op_parameter); in IsChannelFirst()
Dweight_decoder.cc323 int WeightDecoder::DequantNode(OpParameter *op_parameter, const std::vector<Tensor *> &in_tensors, in DequantNode() argument
325 if (op_parameter->quant_type_ != schema::QuantType_QUANT_WEIGHT) { in DequantNode()
331 auto channel_first = IsChannelFirst(index++, op_parameter); in DequantNode()
Dscheduler.h77 … OpParameter *op_parameter, const kernel::KernelKey &desc, TypeId kernel_data_type,
84 … OpParameter *op_parameter, const kernel::KernelKey &desc, kernel::LiteKernel **kernel,
Dkernel_registry.h47 … OpParameter *op_parameter, kernel::LiteKernel **kernel, const void *primitive = nullptr);
Dlite_kernel.h133 OpParameter *op_parameter() const { in op_parameter() function
136 return std::static_pointer_cast<InnerKernel>(kernel_)->op_parameter(); in op_parameter()
/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
57 auto *arith_param = reinterpret_cast<ArithmeticParameter *>(op_parameter); in CheckSupport()
58 auto *scale_param = reinterpret_cast<ScaleParameter *>(op_parameter); in CheckSupport()
73 …t_type = static_cast<ActivationType>(reinterpret_cast<ActivationParameter *>(op_parameter)->type_); in CheckSupport()
85 auto *op_parameter = reinterpret_cast<OpenCLKernel *>(node->kernel())->GetParameter(); local
94 auto *arith_param = reinterpret_cast<ArithmeticParameter *>(op_parameter);
95 auto *scale_param = reinterpret_cast<ScaleParameter *>(op_parameter);
114 …t_type = static_cast<ActivationType>(reinterpret_cast<ActivationParameter *>(op_parameter)->type_);
495 …int type = in_kernel->op_parameter() == nullptr ? in_kernel->type() : in_kernel->op_parameter()->t… in GetTensorIdx()
/third_party/mindspore/mindspore/lite/src/runtime/
Druntime_pass.cc95 if (reinterpret_cast<ConvParameter *>(start_kernel->op_parameter())->group_ != 1) { in Nc4hw4PassMatch()
146 if (kernel->op_parameter() != nullptr) { in RuntimePassValid()
147 if (kernel->op_parameter()->quant_type_ == schema::QuantType_AwareTraining || in RuntimePassValid()
148 kernel->op_parameter()->quant_type_ == schema::QuantType_PostTraining) { in RuntimePassValid()
195 if (reinterpret_cast<ConvParameter *>(start_kernel->op_parameter())->group_ != 1) { in ConvNormC4PassActIndex()
/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
42 MS_CHECK_PTR(op_parameter); in Prepare()
43 op_parameter->thread_num_ = thread_num_; in Prepare()
46 conv_coder_->set_parameter(op_parameter); in Prepare()
/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() argument
45 conv_param_ = reinterpret_cast<ConvParameter *>(op_parameter); in GroupConvCreator()
/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/src/runtime/kernel/opencl/
Dopencl_subgraph.cc112 parameter->op_parameter.is_zero_shape_ = false; in GenToFormatOp()
113 parameter->op_parameter.type_ = PRIM_TO_FORMAT; in GenToFormatOp()
237 int type = k->op_parameter() == nullptr ? k->type() : k->op_parameter()->type_; in UpdateTensorDataTypePass()
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dactivation_grad_populate.cc38 param->op_parameter.type_ = primitive->value_type(); in PopulateActivationGradParameter()
/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Dactivation_grad_populate_v0.cc38 act_param->op_parameter.type_ = schema::PrimitiveType_ActivationGrad; in PopulateActivationGradParameter()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dactivation_grad.h25 OpParameter op_parameter; member

12