Home
last modified time | relevance | path

Searched refs:primitive (Results 1 – 25 of 287) sorted by relevance

12345678910>>...12

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dlite_graph_to_hdi_model_v2_1.cpp34 if (primitive == nullptr) { in ConvertActivation()
56 if (primitive == nullptr) { in ConvertAddFusion()
74 if (primitive == nullptr) { in ConvertAll()
91 if (primitive == nullptr) { in ConvertArgMaxFusion()
111 if (primitive == nullptr) { in ConvertAssert()
128 if (primitive == nullptr) { in ConvertAvgPoolFusion()
153 if (primitive == nullptr) { in ConvertBatchToSpaceND()
171 if (primitive == nullptr) { in ConvertBiasAdd()
186 if (primitive == nullptr) { in ConvertBroadcastTo()
203 if (primitive == nullptr) { in ConvertCast()
[all …]
H A Dlite_graph_to_hdi_model_v2_0.cpp34 if (primitive == nullptr) { in ConvertActivation()
56 if (primitive == nullptr) { in ConvertAddFusion()
74 if (primitive == nullptr) { in ConvertArgMaxFusion()
94 if (primitive == nullptr) { in ConvertAvgPoolFusion()
119 if (primitive == nullptr) { in ConvertBatchToSpaceND()
137 if (primitive == nullptr) { in ConvertBiasAdd()
152 if (primitive == nullptr) { in ConvertCast()
167 if (primitive == nullptr) { in ConvertConcat()
183 if (primitive == nullptr) { in ConvertConv2DFusion()
209 if (primitive == nullptr) { in ConvertConv2dTransposeFusion()
[all …]
H A Dlite_graph_to_hdi_model_v1_0.cpp34 if (primitive == nullptr) { in ConvertActivation()
56 if (primitive == nullptr) { in ConvertAddFusion()
74 if (primitive == nullptr) { in ConvertArgMaxFusion()
94 if (primitive == nullptr) { in ConvertAvgPoolFusion()
119 if (primitive == nullptr) { in ConvertBatchToSpaceND()
137 if (primitive == nullptr) { in ConvertBiasAdd()
152 if (primitive == nullptr) { in ConvertCast()
167 if (primitive == nullptr) { in ConvertConcat()
183 if (primitive == nullptr) { in ConvertConv2DFusion()
209 if (primitive == nullptr) { in ConvertConv2dTransposeFusion()
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/example/drivers/nnrt/v1_0/hdi_cpu_service/include/
H A Dnode_functions.h31 int32_t ParsePrimitive(const std::vector<int8_t>& primitive, T& attr, in ParsePrimitive() argument
34 if (primitive.empty()) { in ParsePrimitive()
40 bool ret = parcelData.WriteBuffer(primitive.data(), primitive.size()); in ParsePrimitive()
54 PrimUniquePtr GetAddPrimitive(const std::vector<int8_t>& primitive);
55 PrimUniquePtr GetAvgPoolPrimitive(const std::vector<int8_t>& primitive);
56 PrimUniquePtr GetConcatPrimitive(const std::vector<int8_t>& primitive);
57 PrimUniquePtr GetConv2dPrimitive(const std::vector<int8_t>& primitive);
61 PrimUniquePtr GetSoftmaxPrimitive(const std::vector<int8_t>& primitive);
62 PrimUniquePtr GetReshapePrimitive(const std::vector<int8_t>& primitive);
64 PrimUniquePtr GetActivationPrimitive(const std::vector<int8_t>& primitive);
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/example/drivers/nnrt/v2_0/hdi_cpu_service/include/
H A Dnode_functions.h31 int32_t ParsePrimitive(const std::vector<int8_t>& primitive, T& attr, in ParsePrimitive() argument
34 if (primitive.empty()) { in ParsePrimitive()
40 bool ret = parcelData.WriteBuffer(primitive.data(), primitive.size()); in ParsePrimitive()
54 PrimUniquePtr GetAddPrimitive(const std::vector<int8_t>& primitive);
55 PrimUniquePtr GetAvgPoolPrimitive(const std::vector<int8_t>& primitive);
56 PrimUniquePtr GetConcatPrimitive(const std::vector<int8_t>& primitive);
57 PrimUniquePtr GetConv2dPrimitive(const std::vector<int8_t>& primitive);
61 PrimUniquePtr GetSoftmaxPrimitive(const std::vector<int8_t>& primitive);
62 PrimUniquePtr GetReshapePrimitive(const std::vector<int8_t>& primitive);
64 PrimUniquePtr GetActivationPrimitive(const std::vector<int8_t>& primitive);
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/device_manager/
H A Ddevice_manager_test.cpp54 MSLITE::LiteGraph::Node* getNode(void* primitive) in getNode() argument
59 node->primitive_ = primitive; in getNode()
176 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
197 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
216 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
238 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
257 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
284 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
304 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
323 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dinner_model_test.cpp972 MSLITE::LiteGraph::Node* getNode(void* primitive) in getNode() argument
977 node->primitive_ = primitive; in getNode()
1094 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1115 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1134 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1156 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1175 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1202 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1222 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1241 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dinner_model_test.cpp911 MSLITE::LiteGraph::Node* getNode(void* primitive) in getNode() argument
916 node->primitive_ = primitive; in getNode()
1033 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1054 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1073 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1095 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1114 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1141 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1161 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
1180 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/example/drivers/nnrt/v1_0/hdi_cpu_service/src/
H A Dnode_functions.cpp26 PrimUniquePtr GetAddPrimitive(const std::vector<int8_t>& primitive) in GetAddPrimitive() argument
47 PrimUniquePtr GetAvgPoolPrimitive(const std::vector<int8_t>& primitive) in GetAvgPoolPrimitive() argument
76 PrimUniquePtr GetConcatPrimitive(const std::vector<int8_t>& primitive) in GetConcatPrimitive() argument
98 PrimUniquePtr GetConv2dPrimitive(const std::vector<int8_t>& primitive) in GetConv2dPrimitive() argument
157 PrimUniquePtr GetMaxPoolFusionPrimitive(const std::vector<int8_t>& primitive) in GetMaxPoolFusionPrimitive() argument
187 PrimUniquePtr GetMatMulFusionPrimitive(const std::vector<int8_t>& primitive) in GetMatMulFusionPrimitive() argument
213 PrimUniquePtr GetSoftmaxPrimitive(const std::vector<int8_t>& primitive) in GetSoftmaxPrimitive() argument
236 PrimUniquePtr GetReshapePrimitive(const std::vector<int8_t>& primitive) in GetReshapePrimitive() argument
258 PrimUniquePtr GetScaleFusionPrimitive(const std::vector<int8_t>& primitive) in GetScaleFusionPrimitive() argument
282 PrimUniquePtr GetActivationPrimitive(const std::vector<int8_t>& primitive) in GetActivationPrimitive() argument
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/example/drivers/nnrt/v2_0/hdi_cpu_service/src/
H A Dnode_functions.cpp26 PrimUniquePtr GetAddPrimitive(const std::vector<int8_t>& primitive) in GetAddPrimitive() argument
47 PrimUniquePtr GetAvgPoolPrimitive(const std::vector<int8_t>& primitive) in GetAvgPoolPrimitive() argument
76 PrimUniquePtr GetConcatPrimitive(const std::vector<int8_t>& primitive) in GetConcatPrimitive() argument
98 PrimUniquePtr GetConv2dPrimitive(const std::vector<int8_t>& primitive) in GetConv2dPrimitive() argument
157 PrimUniquePtr GetMaxPoolFusionPrimitive(const std::vector<int8_t>& primitive) in GetMaxPoolFusionPrimitive() argument
187 PrimUniquePtr GetMatMulFusionPrimitive(const std::vector<int8_t>& primitive) in GetMatMulFusionPrimitive() argument
213 PrimUniquePtr GetSoftmaxPrimitive(const std::vector<int8_t>& primitive) in GetSoftmaxPrimitive() argument
236 PrimUniquePtr GetReshapePrimitive(const std::vector<int8_t>& primitive) in GetReshapePrimitive() argument
258 PrimUniquePtr GetScaleFusionPrimitive(const std::vector<int8_t>& primitive) in GetScaleFusionPrimitive() argument
282 PrimUniquePtr GetActivationPrimitive(const std::vector<int8_t>& primitive) in GetActivationPrimitive() argument
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dhard_sigmoid_test.cpp162 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
164 EXPECT_NE(expectPrimitive, primitive);
167 auto returnValue = mindspore::lite::MindIR_Activation_GetActivationType(primitive.get());
178 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
180 EXPECT_EQ(expectPrimitive, primitive);
H A Dhswish_builder_test.cpp162 LiteGraphPrimitvePtr primitive = m_hswish.GetPrimitive(); variable
164 EXPECT_NE(expectPrimitive, primitive);
167 auto returnValue = mindspore::lite::MindIR_Activation_GetActivationType(primitive.get());
179 LiteGraphPrimitvePtr primitive = m_hswish.GetPrimitive(); variable
181 EXPECT_EQ(expectPrimitive, primitive);
H A Dabs_test.cpp140 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
142 EXPECT_NE(expectPrimitive, primitive);
152 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
154 EXPECT_EQ(expectPrimitive, primitive);
H A Dceil_test.cpp140 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
142 EXPECT_NE(expectPrimitive, primitive);
152 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
154 EXPECT_EQ(expectPrimitive, primitive);
H A Dcos_test.cpp140 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
142 EXPECT_NE(expectPrimitive, primitive);
152 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
154 EXPECT_EQ(expectPrimitive, primitive);
H A Derf_test.cpp140 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
142 EXPECT_NE(expectPrimitive, primitive);
152 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
154 EXPECT_EQ(expectPrimitive, primitive);
H A Dfloor_test.cpp140 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
142 EXPECT_NE(expectPrimitive, primitive);
152 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
154 EXPECT_EQ(expectPrimitive, primitive);
H A Dless_test.cpp140 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
142 EXPECT_NE(expectPrimitive, primitive);
152 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
154 EXPECT_EQ(expectPrimitive, primitive);
H A Dlog_test.cpp140 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
142 EXPECT_NE(expectPrimitive, primitive);
152 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
154 EXPECT_EQ(expectPrimitive, primitive);
H A Dlogical_and_test.cpp140 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
142 EXPECT_NE(expectPrimitive, primitive);
152 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
154 EXPECT_EQ(expectPrimitive, primitive);
H A Dlogical_not_test.cpp140 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
142 EXPECT_NE(expectPrimitive, primitive);
152 LiteGraphPrimitvePtr primitive = m_builder.GetPrimitive(); variable
154 EXPECT_EQ(expectPrimitive, primitive);
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dsqrt_builder.cpp77 auto primitive = mindspore::lite::MindIR_Sqrt_CreatePrimitive(); in GetPrimitive() local
78 if (primitive == nullptr) { in GetPrimitive()
83 LiteGraphTensorPtr graphPrimitivePtr(primitive, DestroyLiteGraphPrimitive); in GetPrimitive()
H A Dsquared_difference_builder.cpp75 auto primitive = mindspore::lite::MindIR_SquaredDifference_CreatePrimitive(); in GetPrimitive() local
76 if (primitive == nullptr) { in GetPrimitive()
81 LiteGraphTensorPtr graphPrimitivePtr(primitive, DestroyLiteGraphPrimitive); in GetPrimitive()
H A Dtranspose_builder.cpp75 auto primitive = mindspore::lite::MindIR_Transpose_CreatePrimitive(); in GetPrimitive() local
76 if (primitive == nullptr) { in GetPrimitive()
81 LiteGraphPrimitvePtr graphPrimitivePtr(primitive, DestroyLiteGraphPrimitive); in GetPrimitive()
H A Dtanh_builder.cpp80 auto primitive = in GetPrimitive() local
82 if (primitive == nullptr) { in GetPrimitive()
87 LiteGraphPrimitvePtr graphPrimitivePtr(primitive, DestroyLiteGraphPrimitive); in GetPrimitive()

12345678910>>...12