Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/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()
102 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuGroupConvInt8KernelCreator()
120 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvInt8KernelCreator()
Ddeconvolution_int8.cc333 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuDeConvInt8KernelCreator()
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dcommon.cc30OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain()
40OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain()
193OpParameter *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()
352OpParameter *op_parameter, bool fp16_enable, float atol, float rtol, bool print_data) { in TestMain()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16_grad/
Darithmetic_self_grad.h26 OpParameter op_parameter; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dactivation_grad_fp32.h25 OpParameter op_parameter; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/bolt/
Dbolt_parameter_manager.cc24 ParameterSpec *PopulateConv2DBoltParameter(const OpParameter *op_parameter) { in PopulateConv2DBoltParameter()
Dbolt_parameter_manager.h37 ParameterSpec *CreateBoltParameter(const OpParameter *op_parameter) { in CreateBoltParameter()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dscatter_nd_parameter.h23 OpParameter op_parameter; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Dconvolution_delegate_fp16.cc169 …tor<lite::Tensor *> &inputs, const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvFp16KernelSelect()
205 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuGroupConvFp16KernelCreator()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/
Dinfershape_helper.cc221 int InferShapeByNNACL(const CompileNodePtr &node, OpParameter *op_parameter, Format format, InferCo… in InferShapeByNNACL()
323 int OpsOrNNACLInferShape(const CompileNodePtr &node, OpParameter *op_parameter, InferContext *conte… in OpsOrNNACLInferShape()
351 auto op_parameter = OperatorPopulateRegistry::GetInstance()->CreatePopulateByOp(base_operator); in NodeFallBackInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dgptq.h31 bool transpose, const MatMulParameter *op_parameter) in Gptq()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dweight_decoder.cc403 int WeightDecoder::GetMatMulPreferredDim(const OpParameter *op_parameter, int input_index, in GetMatMulPreferredDim()
428 int WeightDecoder::GetDeConvPreferredDim(const OpParameter *op_parameter, const std::vector<int> &d… in GetDeConvPreferredDim()
440 bool WeightDecoder::IsChannelFirst(int index, const OpParameter *op_parameter) { in IsChannelFirst()
489 int WeightDecoder::DequantNode(const OpParameter *op_parameter, const std::vector<Tensor *> &in_ten… in DequantNode()
Dweight_decoder.h71 …static int GetPreferredDim(const std::vector<T *> &in_tensors, const OpParameter *op_parameter, in… in GetPreferredDim()
227 …static int GetGatherPreferredDim(const OpParameter *op_parameter, const std::vector<T *> &in_tenso… in GetGatherPreferredDim()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dconvolution_delegate_fp32.cc378 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuGroupConvFp32KernelCreator()
392 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvFp32KernelCreator()
Ddeconvolution_fp32.cc330OpParameter *op_parameter, const InnerContext *ctx) { in CpuNormDeconvFp32KernelCreator()
362 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuDeConvFp32KernelCreator()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/
Dextendrt_kernel_exec.h41 OpParameter *op_parameter() const override { in op_parameter() function
Dbase_kernel.h150 virtual OpParameter *op_parameter() const { return nullptr; } in op_parameter() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_export.h41 OpParameter *op_parameter; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/
Doffline_packing_optimizer.cc68 OpParameter *op_parameter, kernel::KernelKey *desc, in CreateMatmulPackDataIntoTable()
207 OpParameter *op_parameter = GetOpParameter(primT.get()); in MatmulPacking() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/nnacl/
Dnnacl_base_kernel.h55 OpParameter *op_parameter() const override { return lite_kernel_->op_parameter(); } in op_parameter() function
Dnnacl_lib.cc81 …auto op_parameter = lite::OperatorPopulateRegistry::GetInstance()->CreatePopulateByOp(spec.primiti… in CreateKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/
Dgroup_convolution_creator.h39 …std::vector<lite::Tensor *> inputs, std::vector<lite::Tensor *> outputs, OpParameter *op_parameter, in GroupConvCreator()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Dconv2d_delegate_dynamic_fp16_coder.cc46 ConvParameter *op_parameter = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in Prepare() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Dconv2d_delegate_fp32_coder.cc35 ConvParameter *op_parameter = static_cast<ConvParameter *>(malloc(sizeof(ConvParameter))); in Prepare() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dfusion_eltwise.cc53 auto *op_parameter = reinterpret_cast<OpenCLKernel *>(node->kernel())->GetParameter(); in CheckSupport() local
87 auto *op_parameter = reinterpret_cast<OpenCLKernel *>(node->kernel())->GetParameter(); local

12