/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
D | conv2d_delegate_fp32_coder.cc | 35 const void *primitive = node_->primitive_; in Prepare() 37 int primitive_type = GetPrimitiveType(node_->primitive_, schema_version_); in Prepare() 39 GetPrimitiveType(node_->primitive_, schema_version_), schema_version_); in Prepare() 41 OpParameter *op_parameter = parameter_gen(node_->primitive_); in Prepare() 68 const void *primitive = node->primitive_; in CPUConvolutionFP32CoderSelect() 73 GetPrimitiveType(node->primitive_, schema_version), schema_version); in CPUConvolutionFP32CoderSelect() 75 auto conv_param = reinterpret_cast<ConvParameter *>(paramGen(node->primitive_)); in CPUConvolutionFP32CoderSelect() 121 const void *primitive = node->primitive_; in CPUConv2DFusionFP32CoderCreator() 126 GetPrimitiveType(node->primitive_, schema_version), schema_version); in CPUConv2DFusionFP32CoderCreator() 131 auto conv_param = reinterpret_cast<ConvParameter *>(paramGen(node->primitive_)); in CPUConv2DFusionFP32CoderCreator()
|
/third_party/mindspore/mindspore/lite/src/cxx_api/ |
D | kernel.cc | 19 if (primitive_ == nullptr) { in Initialize() 22 type_ = primitive_->value_type(); in Initialize() 24 auto param = primitive_->value_as_Custom(); in Initialize()
|
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
D | activation_int8_coder.cc | 31 const void *primitive_c = node->primitive_; in CPUActivationINT8CoderCreator() 36 GetPrimitiveType(node->primitive_, schema_version), schema_version); in CPUActivationINT8CoderCreator() 41 OpParameter *parameter = parameter_gen(node->primitive_); in CPUActivationINT8CoderCreator() 45 (schema::PrimitiveType)GetPrimitiveType(node->primitive_, schema_version)); in CPUActivationINT8CoderCreator()
|
D | conv2d_int8_coder.cc | 252 const void *primitive = node->primitive_; in CPUConv2DINT8CoderCreator() 257 GetPrimitiveType(node->primitive_, schema_version), schema_version); in CPUConv2DINT8CoderCreator() 262 auto conv_param = reinterpret_cast<ConvParameter *>(paramGen(node->primitive_)); in CPUConv2DINT8CoderCreator() 289 const void *primitive = node->primitive_; in CPUConv2DFusionINT8CoderCreator() 294 GetPrimitiveType(node->primitive_, schema_version), schema_version); in CPUConv2DFusionINT8CoderCreator() 299 auto conv_param = reinterpret_cast<ConvParameter *>(paramGen(node->primitive_)); in CPUConv2DFusionINT8CoderCreator()
|
/third_party/mindspore/include/api/ |
D | kernel.h | 40 … : context_(ctx), inputs_(std::move(inputs)), outputs_(std::move(outputs)), primitive_(primitive) { in Kernel() 102 const schema::Primitive *primitive() const { return this->primitive_; } in primitive() 127 const schema::Primitive *primitive_ = nullptr; variable
|
/third_party/mindspore/mindspore/lite/examples/runtime_extend/src/ |
D | custom_add_kernel.cc | 66 …y::RegisterKernelInterface::GetKernelInterface({}, primitive_)->Infer(&inputs_, &outputs_, primiti… in PreProcess() 90 auto prim = primitive_->value_as_Custom(); in ParseAttrData()
|
/third_party/mindspore/mindspore/lite/src/ |
D | scheduler.cc | 542 auto primitive = node->primitive_; in InferNodeShape() 549 …ret = KernelInferShape(inputs, outputs, node->primitive_, context_->GetProviders(), schema_version… in InferNodeShape() 556 GetPrimitiveType(node->primitive_, schema_version_), schema_version_); in InferNodeShape() 628 auto subgraph_index = GetPartialGraphIndex(partial_node->primitive_, schema_version_); in RestoreSubGraphInput() 645 auto subgraph_index = GetPartialGraphIndex(partial_node->primitive_, schema_version_); in CopyPartialShapeToSubGraph() 674 if (!IsPartialNode(node->primitive_, schema_version_)) { in InferPartialShape() 679 int subgraph_index = GetPartialGraphIndex(node->primitive_, schema_version_); in InferPartialShape() 693 if (IsPartialNode(iter->primitive_, schema_version_)) { in NodeInputIsPartial() 706 if (!IsCallNode(node->primitive_, schema_version_)) { in InferCallShape() 735 auto *primitive = node->primitive_; in InferSubGraphShape() [all …]
|
D | lite_model.cc | 39 auto primitive = node->primitive_; in ConvertAttrs() 109 node->primitive_ = nullptr; in Free() 204 if (node == nullptr || node->primitive_ == nullptr) { in NodeVerify() 219 if (IsPartialNode(node->primitive_, schema_version_)) { in NodeVerify() 220 auto subgraph_index = GetPartialGraphIndex(node->primitive_, schema_version_); in NodeVerify()
|
/third_party/mindspore/mindspore/lite/src/ops/compat/v0/ |
D | cast_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferCastAttr() 29 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferCastAttr()
|
D | gather_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferGatherAttr() 29 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferGatherAttr()
|
D | power_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferPowerToAttr() 33 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferPowerToAttr()
|
D | topk_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferTopkAttr() 33 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferTopkAttr()
|
D | expand_dims_compat_v0.cc | 25 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferExpandDimsAttr() 32 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferExpandDimsAttr()
|
D | reduce_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferReduceToAttr() 33 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferReduceToAttr()
|
D | pad_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferPadAttr() 33 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferPadAttr()
|
D | tile_compat_v0.cc | 25 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferTileToAttr() 34 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferTileToAttr()
|
D | fill_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferFillToAttr() 33 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferFillToAttr()
|
D | transpose_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferTransposeAttr() 33 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferTransposeAttr()
|
D | broadcast_to_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferBroadcastToAttr() 33 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferBroadcastToAttr()
|
D | reshape_compat_v0.cc | 25 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferReshapeAttr() 34 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferReshapeAttr()
|
D | permute_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferPermuteAttr() 33 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferPermuteAttr()
|
D | slice_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferSliceAttr() 33 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferSliceAttr()
|
D | strided_slice_compat_v0.cc | 24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferStridedSliceAttr() 29 auto prim = reinterpret_cast<const schema::v0::Primitive *>(node->primitive_); in TransferStridedSliceAttr()
|
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
D | pattern_engine.h | 55 explicit Var(std::string tag = "") : tag_(std::move(tag)), primitive_(nullptr) { EnsureTag(); } in tag_() 56 …onst PrimitivePtr &primitive, std::string tag = "") : tag_(std::move(tag)), primitive_(primitive) { in tag_() 76 PrimitivePtr primitive() const { return primitive_; } in primitive() 88 PrimitivePtr primitive_; variable
|
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/ |
D | op_coder_builder.cc | 27 MS_CHECK_PTR_RET_NULL(node_->primitive_); in build() 28 int primitive_type = GetPrimitiveType(node_->primitive_, schema_version); in build()
|