Home
last modified time | relevance | path

Searched defs:model_type (Results 1 – 25 of 45) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mindir_loader/
Dmodel_loader.h40 …void RegModelLoader(mindspore::ModelType model_type, std::function<std::shared_ptr<ModelLoader>()>… in RegModelLoader()
44 std::shared_ptr<ModelLoader> GetModelLoader(mindspore::ModelType model_type) { in GetModelLoader()
58 …ModelLoaderRegistrar(const mindspore::ModelType &model_type, std::function<std::shared_ptr<ModelLo… in ModelLoaderRegistrar()
64 #define REG_MODEL_LOADER(model_type, model_loader_creator) \ argument
/third_party/mindspore/mindspore-src/source/include/api/
Dserialization.h144 Status Serialization::Load(const void *model_data, size_t data_size, ModelType model_type, Graph *g… in Load()
149 Status Serialization::Load(const std::string &file, ModelType model_type, Graph *graph, const Key &… in Load()
154 Status Serialization::Load(const std::vector<std::string> &files, ModelType model_type, std::vector… in Load()
163 Status Serialization::ExportModel(const Model &model, ModelType model_type, const std::string &mode… in ExportModel()
170 Status Serialization::ExportModel(const Model &model, ModelType model_type, Buffer *model_data, in ExportModel()
177 Status Serialization::ExportWeightsCollaborateWithMicro(const Model &model, ModelType model_type, in ExportWeightsCollaborateWithMicro()
Dmodel.h418 Status Model::Build(const void *model_data, size_t data_size, ModelType model_type, in Build()
425 Status Model::Build(const std::string &model_path, ModelType model_type, const std::shared_ptr<Cont… in Build()
431 Status Model::Build(const std::string &model_path, ModelType model_type, in Build()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/
Dserialization.cc41 Status Serialization::Load(const void *model_data, size_t data_size, ModelType model_type, Graph *g… in Load()
88 Status Serialization::Load(const std::vector<char> &file, ModelType model_type, Graph *graph, const… in Load()
136 Status Serialization::Load(const std::vector<std::vector<char>> &files, ModelType model_type, in Load()
147 Status Serialization::ExportModel(const Model &model, ModelType model_type, Buffer *model_data, in ExportModel()
177 Status Serialization::ExportModel(const Model &model, ModelType model_type, const std::vector<char>… in ExportModel()
203 Status Serialization::ExportWeightsCollaborateWithMicro(const Model &model, ModelType model_type, in ExportWeightsCollaborateWithMicro()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/graph/
Dgraph_data.cc22 Graph::GraphData::GraphData(const FuncGraphPtr &func_graph, enum ModelType model_type) in GraphData()
31 Graph::GraphData::GraphData(const Buffer &om_data, enum ModelType model_type) in GraphData()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/graph/
Dgraph_data.cc23 Graph::GraphData::GraphData(const FuncGraphPtr &func_graph, mindspore::ModelType model_type) in GraphData()
32 Graph::GraphData::GraphData(const Buffer &om_data, mindspore::ModelType model_type) in GraphData()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/
Dserialization.cc102 Status Serialization::Load(const void *model_data, size_t data_size, ModelType model_type, Graph *g… in Load()
155 Status Serialization::Load(const std::vector<char> &file, ModelType model_type, Graph *graph) { in Load()
159 Status Serialization::Load(const std::vector<char> &file, ModelType model_type, Graph *graph, const… in Load()
237 Status Serialization::Load(const std::vector<std::vector<char>> &files, ModelType model_type, in Load()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/
Dserialization.cc127 Status Serialization::Load(const void *model_data, size_t data_size, ModelType model_type, Graph *g… in Load()
179 Status Serialization::Load(const std::vector<char> &file, ModelType model_type, Graph *graph) { in Load()
183 Status Serialization::Load(const std::vector<char> &file, ModelType model_type, Graph *graph, const… in Load()
261 Status Serialization::Load(const std::vector<std::vector<char>> &files, ModelType model_type, in Load()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dcache_session.cc197 …ession::LoadModelAndCompileByPath(const std::string &model_path, mindspore::ModelType model_type) { in LoadModelAndCompileByPath()
240 …InOutFromBuffer(const char *model_buf, size_t size, bool take_buf, mindspore::ModelType model_type, in ImportInOutFromBuffer()
400 const char *CacheSession::LoadModelByPath(const std::string &file, mindspore::ModelType model_type,… in LoadModelByPath()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/model/
Dmodel.cc71 Status Model::Build(const void *model_data, size_t data_size, ModelType model_type, in Build()
89 Status Model::Build(const std::vector<char> &model_path, ModelType model_type, in Build()
107 Status Model::Build(const std::vector<char> &model_path, ModelType model_type, in Build()
163 Status Model::Build(const void *model_data, size_t data_size, ModelType model_type, in Build()
373 bool Model::CheckModelSupport(DeviceType device_type, ModelType model_type) { in CheckModelSupport()
Dmodel_group.cc61 Status ModelGroup::CalMaxSizeOfWorkspace(ModelType model_type, const std::shared_ptr<Context> &ms_c… in CalMaxSizeOfWorkspace()
Dmodel_impl.cc243 …r ModelImpl::LoadGraphByBufferImpl(const void *model_buff, size_t model_size, ModelType model_type, in LoadGraphByBufferImpl()
337 Status ModelImpl::BuildByBufferImpl(const void *model_buff, size_t model_size, ModelType model_type, in BuildByBufferImpl()
472 Status ModelImpl::Build(const void *model_data, size_t data_size, ModelType model_type, in Build()
477 Status ModelImpl::Build(const std::string &model_path, ModelType model_type, in Build()
873 bool ModelImpl::CheckModelSupport(DeviceType device_type, ModelType model_type) { in CheckModelSupport()
Dmodel_group_impl.cc94 Status ModelGroupImpl::CalMaxSizeOfWorkspace(ModelType model_type, const std::shared_ptr<Context> &… in CalMaxSizeOfWorkspace()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtransfer_session.cc187 int TransferSession::ExportInner(DestType destination, ModelType model_type, QuantizationType quant… in ExportInner()
265 int TransferSession::Export(const std::string &filename, ModelType model_type, QuantizationType qua… in Export()
270 int TransferSession::Export(Buffer *model_buffer, ModelType model_type, QuantizationType quant_type… in Export()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/model/
Dmodel.cc69 Status Model::Build(const void *model_data, size_t data_size, ModelType model_type, in Build()
128 Status Model::Build(const void *model_data, size_t data_size, ModelType model_type, in Build()
175 Status Model::Build(const std::vector<char> &model_path, ModelType model_type, in Build()
244 Status Model::Build(const std::vector<char> &model_path, ModelType model_type, in Build()
450 bool Model::CheckModelSupport(DeviceType device_type, ModelType model_type) { in CheckModelSupport()
Dmodel_group.cc54 Status ModelGroup::CalMaxSizeOfWorkspace(ModelType model_type, const std::shared_ptr<Context> &ms_c… in CalMaxSizeOfWorkspace()
Dmodel_group_impl.cc100 Status ModelGroupImpl::CalMaxSizeOfWorkspace(ModelType model_type, const std::shared_ptr<Context> &… in CalMaxSizeOfWorkspace()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/c_api/
Dmodel_c.cc189 …uild(OH_AI_ModelHandle model, const void *model_data, size_t data_size, OH_AI_ModelType model_type, in OH_AI_ModelBuild()
219 …_AI_ModelBuildFromFile(OH_AI_ModelHandle model, const char *model_path, OH_AI_ModelType model_type, in OH_AI_ModelBuildFromFile()
488 …uild(OH_AI_ModelHandle model, const void *model_data, size_t data_size, OH_AI_ModelType model_type, in OH_AI_TrainModelBuild()
525 …rainModelBuildFromFile(OH_AI_ModelHandle model, const char *model_path, OH_AI_ModelType model_type, in OH_AI_TrainModelBuildFromFile()
654 OH_AI_Status OH_AI_ExportModel(OH_AI_ModelHandle model, OH_AI_ModelType model_type, const char *mod… in OH_AI_ExportModel()
675 OH_AI_Status OH_AI_ExportModelBuffer(OH_AI_ModelHandle model, OH_AI_ModelType model_type, char **mo… in OH_AI_ExportModelBuffer()
705 …_AI_ExportWeightsCollaborateWithMicro(OH_AI_ModelHandle model, OH_AI_ModelType model_type, const c… in OH_AI_ExportWeightsCollaborateWithMicro()
/third_party/mindspore/mindspore-src/source/mindspore/lite/python/api/
Dmodel.py137 …def build_from_file(self, model_path, model_type, context=None, config_path="", config_dict: dict … argument
797 def cal_max_size_of_workspace(self, model_type, context): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/model_pool/
Dmodel_worker.cc48 ModelType model_type) { in InitModelWorker()
112 Status ModelWorker::Init(const char *model_buf, size_t size, ModelType model_type) { in Init()
Dresource_manager.cc254 ModelType model_type) { in Launch()
276 bool *create_success, ModelType model_type) { in InitModelWorker()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dbuild_sentence_piece_vocab_node.cc28 … float character_coverage, SentencePieceModel model_type, in BuildSentenceVocabNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/text/
Dsentence_piece_vocab.cc41 … float character_coverage, const SentencePieceModel &model_type, in BuildFromFile()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dserialization.cc94 …atus Serialization::Load(const void *model_data, size_t data_size, mindspore::ModelType model_type, in Load()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dbuild_sentence_piece_vocab_op.cc25 … float character_coverage, SentencePieceModel model_type, in BuildSentencePieceVocabOp()

12