Home
last modified time | relevance | path

Searched defs:meta_graph (Results 1 – 25 of 43) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/
Dconstant_folding_fusion_test.cc39 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildGraph() local
98 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildGraphForOneInput() local
142 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildMixGraph() local
238 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildSplitGraph() local
369 auto meta_graph = BuildGraph(schema::PrimitiveType_AddFusion, new schema::AddFusionT); in TEST_F() local
382 auto meta_graph = BuildMixGraph(); in TEST_F() local
395 auto meta_graph = BuildGraph(schema::PrimitiveType_SubFusion, new schema::SubFusionT); in TEST_F() local
408 auto meta_graph = BuildGraph(schema::PrimitiveType_MulFusion, new schema::MulFusionT); in TEST_F() local
422 auto meta_graph = BuildGraph(schema::PrimitiveType_Transpose, transposeT); in TEST_F() local
437 auto meta_graph = BuildGraph(schema::PrimitiveType_Stack, stackT); in TEST_F() local
[all …]
Dconv_biasadd_fusion_test.cc72 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildGraph() local
144 auto meta_graph = BuildGraph(schema::PrimitiveType_Conv2DFusion, schema::PrimitiveType_BiasAdd); in TEST_F() local
155 auto meta_graph = BuildGraph(schema::PrimitiveType_Conv2DFusion, schema::PrimitiveType_AddFusion); in TEST_F() local
165 …auto meta_graph = BuildGraph(schema::PrimitiveType_Conv2DFusion, schema::PrimitiveType_MatMulFusio… in TEST_F() local
Dconv_activation_fusion_test.cc72 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildGraph() local
135 auto meta_graph = BuildGraph(schema::PrimitiveType_Conv2DFusion, schema::ActivationType_RELU); in TEST_F() local
148 auto meta_graph = BuildGraph(schema::PrimitiveType_Conv2DFusion, schema::ActivationType_RELU6); in TEST_F() local
161 …auto meta_graph = BuildGraph(schema::PrimitiveType_Conv2DFusion, schema::ActivationType_LEAKY_RELU… in TEST_F() local
Dactivation_fusion_test.cc46 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildGraph() local
111 auto meta_graph = BuildGraph(schema::ActivationType_HARD_TANH, schema::ActivationType_RELU); in TEST_F() local
125 auto meta_graph = BuildGraph(schema::ActivationType_RELU6, schema::ActivationType_HARD_TANH); in TEST_F() local
139 auto meta_graph = BuildGraph(schema::ActivationType_RELU, schema::ActivationType_SIGMOID); in TEST_F() local
Dconv_bn_fusion_test.cc73 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildCaffeGraph() local
154 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildTFGraph() local
261 auto meta_graph = BuildCaffeGraph(schema::PrimitiveType_Conv2DFusion); in TEST_F() local
271 auto meta_graph = BuildTFGraph(schema::PrimitiveType_Conv2DFusion); in TEST_F() local
Dtrans_matmul_fusion_test.cc50 void SetTransTensors(std::shared_ptr<schema::MetaGraphT> meta_graph, const std::vector<int> &perm, in SetTransTensors()
93 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildGraph() local
157 auto meta_graph = BuildGraph(trans_param_a, trans_param_b, output_dims); in TEST_F() local
179 auto meta_graph = BuildGraph(trans_param_a, trans_param_b, output_dims); in TEST_F() local
201 auto meta_graph = BuildGraph(trans_param_a, trans_param_b, output_dims); in TEST_F() local
Dconv_scale_fusion_test.cc84 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildGraph() local
186 auto meta_graph = BuildGraph(schema::PrimitiveType_Conv2DFusion, true); in TEST_F() local
197 auto meta_graph = BuildGraph(schema::PrimitiveType_Conv2DFusion, false); in TEST_F() local
Dmatmul_mul_fusion_test.cc69 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildGraph() local
135 auto meta_graph = BuildGraph(); in TEST_F() local
Dadd_concat_act_fusion_test.cc75 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in BuildGraph() local
156 auto meta_graph = BuildGraph(schema::ActivationType_RELU6); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dlite_model.h79 int SetQuantType(const T &meta_graph, const U *c_node, LiteGraph::Node *node) { in SetQuantType()
106 …bool CheckNeedWeightQuant(const T &meta_graph, const flatbuffers::Vector<uint32_t> *in_tensor_inde… in CheckNeedWeightQuant()
140 bool ConvertNodes(const T &meta_graph) { in ConvertNodes()
192 bool ConvertTensors(const T &meta_graph) { in ConvertTensors()
211 int MetaGraphMappingSubGraph(const T &meta_graph) { in MetaGraphMappingSubGraph()
246 int GenerateModel(const T &meta_graph) { in GenerateModel()
Dcache_session.h45 bool ConvertInputOutputTensors(const T &meta_graph, LiteGraph &graph_) { in ConvertInputOutputTensors()
75 int GenerateModelInputOutput(const T &meta_graph, LiteGraph &graph_) { in GenerateModelInputOutput()
Druntime_convert.cc74 schema::MetaGraphT *meta_graph = nullptr; in RuntimeConvert() local
111 schema::MetaGraphT *meta_graph = nullptr; in RuntimeConvert() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/
Dconverter_metagraph.cc37 STATUS ConverterToMetaGraph::UpdateMetaGraphOutputName(schema::MetaGraphT *meta_graph, in UpdateMetaGraphOutputName()
92 auto meta_graph = Export(func_graph, false, false, param->train_model); in Build() local
120 STATUS ConverterToMetaGraph::Save(schema::MetaGraphT *meta_graph, const std::shared_ptr<ConverterPa… in Save()
Dconverter_packed_node.cc48 … schema::MetaGraphT *meta_graph, const std::unique_ptr<schema::CNodeT> &cnode, in AddWeightSumsToInputs()
72 int ReplaceMatMulFusionToCustom(schema::MetaGraphT *meta_graph, const std::unique_ptr<schema::CNode… in ReplaceMatMulFusionToCustom()
169 int ConverterPackedNode(schema::MetaGraphT *meta_graph, const std::string &cpu_option) { in ConverterPackedNode()
Dgraphdef_transform.cc80 int FillGraphOutputShape(MetaGraphT *meta_graph, const std::vector<std::vector<int64_t>> output_sha… in FillGraphOutputShape()
92 void FillGraphInputAndOutputFormats(MetaGraphT *meta_graph, const ConverterPara &para) { in FillGraphInputAndOutputFormats()
Dconverter.cc333 int CheckExistCustomOps(const schema::MetaGraphT *meta_graph, bool *exist_custom_nodes) { in CheckExistCustomOps()
353 int PreInference(const schema::MetaGraphT &meta_graph, bool train_model) { in PreInference()
1054 int ConverterImpl::ExecuteMicro(const schema::MetaGraphT *meta_graph, const std::shared_ptr<Convert… in ExecuteMicro()
1092 auto meta_graph = ConverterToMetaGraph::Build(param, graph); in SaveGraph() local
1136 schema::MetaGraphT *meta_graph = nullptr; in SaveMindIRModel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/st/
Dmultiple_device_test.cc37 void CreateMultyModel1(mindspore::schema::MetaGraphT *meta_graph) { in CreateMultyModel1()
162 void CreateMultyModel2(mindspore::schema::MetaGraphT *meta_graph) { in CreateMultyModel2()
297 auto meta_graph = std::make_shared<mindspore::schema::MetaGraphT>(); in TEST_F() local
355 auto meta_graph = std::make_shared<mindspore::schema::MetaGraphT>(); in TEST_F() local
400 auto meta_graph = std::make_shared<mindspore::schema::MetaGraphT>(); in TEST_F() local
445 auto meta_graph = std::make_shared<mindspore::schema::MetaGraphT>(); in TEST_F() local
469 auto meta_graph = std::make_shared<mindspore::schema::MetaGraphT>(); in TEST_F() local
Doptimize_allocator_test.cc38 void CreateModel1(mindspore::schema::MetaGraphT *meta_graph) { in CreateModel1()
124 auto meta_graph = std::make_shared<mindspore::schema::MetaGraphT>(); in TEST_F() local
Dmix_data_type_test.cc87 void ConstructModel(schema::MetaGraphT *meta_graph) { in ConstructModel()
187 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/converter/parser/tflite/
Dtflite_parsers_test_utils.h30 schema::MetaGraphT *meta_graph = nullptr; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/obfuscator/include/
Ddeobfuscator.h60 MS_API bool IsMetaGraphObfuscated(const T &meta_graph) { in IsMetaGraphObfuscated()
69 MS_API DeObfuscator *GetModelDeObfuscator(const T &meta_graph, Model *model, size_t buf_size) { in GetModelDeObfuscator()
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/common/
Dimport_from_meta_graphT.h35 explicit AnfImporterFromMetaGraphT(schema::MetaGraphT *meta_graph) : meta_graph_(meta_graph) { in AnfImporterFromMetaGraphT()
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/
Dscheduler_test.cc35 auto meta_graph = std::make_shared<mindspore::schema::MetaGraphT>(); in TEST_F() local
178 auto meta_graph = std::make_shared<mindspore::schema::MetaGraphT>(); in TEST_F() local
362 auto meta_graph = std::make_shared<mindspore::schema::MetaGraphT>(); in TEST_F() local
Dinfer_test.cc33 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in TEST_F() local
145 auto meta_graph = std::make_shared<schema::MetaGraphT>(); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnrt/
Dnnrt_delegate.h52 void SetMetaGraph(const void *meta_graph) { in SetMetaGraph()

12