Home
last modified time | relevance | path

Searched refs:primitive_ (Results 1 – 25 of 65) sorted by relevance

123

/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dconv2d_delegate_fp32_coder.cc35 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/
Dkernel.cc19 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/
Dactivation_int8_coder.cc31 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()
Dconv2d_int8_coder.cc252 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/
Dkernel.h40 … : 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/
Dcustom_add_kernel.cc66 …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/
Dscheduler.cc542 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 …]
Dlite_model.cc39 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/
Dcast_compat_v0.cc24 …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()
Dgather_compat_v0.cc24 …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()
Dpower_compat_v0.cc24 …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()
Dtopk_compat_v0.cc24 …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()
Dexpand_dims_compat_v0.cc25 …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()
Dreduce_compat_v0.cc24 …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()
Dpad_compat_v0.cc24 …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()
Dtile_compat_v0.cc25 …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()
Dfill_compat_v0.cc24 …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()
Dtranspose_compat_v0.cc24 …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()
Dbroadcast_to_compat_v0.cc24 …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()
Dreshape_compat_v0.cc25 …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()
Dpermute_compat_v0.cc24 …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()
Dslice_compat_v0.cc24 …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()
Dstrided_slice_compat_v0.cc24 …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/
Dpattern_engine.h55 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/
Dop_coder_builder.cc27 MS_CHECK_PTR_RET_NULL(node_->primitive_); in build()
28 int primitive_type = GetPrimitiveType(node_->primitive_, schema_version); in build()

123