Home
last modified time | relevance | path

Searched refs:model_type (Results 1 – 25 of 49) sorted by relevance

12

/third_party/mindspore/include/api/
Dserialization.h43 …inline static Status Load(const void *model_data, size_t data_size, ModelType model_type, Graph *g…
55 …inline static Status Load(const std::string &file, ModelType model_type, Graph *graph, const Key &…
67 …inline static Status Load(const std::vector<std::string> &files, ModelType model_type, std::vector…
70 static Status ExportModel(const Model &model, ModelType model_type, Buffer *model_data);
71 static Status ExportModel(const Model &model, ModelType model_type, const std::string &model_file,
76 …static Status Load(const void *model_data, size_t data_size, ModelType model_type, Graph *graph, c…
78 static Status Load(const std::vector<char> &file, ModelType model_type, Graph *graph);
79 …static Status Load(const std::vector<char> &file, ModelType model_type, Graph *graph, const Key &d…
81 …static Status Load(const std::vector<std::vector<char>> &files, ModelType model_type, std::vector<…
85 Status Serialization::Load(const void *model_data, size_t data_size, ModelType model_type, Graph *g… in Load() argument
[all …]
Dmodel.h174 static bool CheckModelSupport(enum DeviceType device_type, ModelType model_type);
192 Status Build(const void *model_data, size_t data_size, ModelType model_type,
206 Status Build(const std::string &model_path, ModelType model_type,
/third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/include/api/
Dserialization.h32 static Status Load(const void *model_data, size_t data_size, ModelType model_type, Graph *graph);
33 inline static Status Load(const std::string &file, ModelType model_type, Graph *graph);
36 static Status ExportModel(const Model &model, ModelType model_type, Buffer *model_data);
37 …static Status ExportModel(const Model &model, ModelType model_type, const std::string &model_file);
40 static Status Load(const std::vector<char> &file, ModelType model_type, Graph *graph);
43 Status Serialization::Load(const std::string &file, ModelType model_type, Graph *graph) { in Load() argument
44 return Load(StringToChar(file), model_type, graph); in Load()
/third_party/mindspore/mindspore/lite/src/cxx_api/
Dserialization.cc40 Status Serialization::Load(const void *model_data, size_t data_size, ModelType model_type, Graph *g… in Load() argument
55 if (model_type != kMindIR) { in Load()
74 Status Serialization::Load(const std::vector<char> &file, ModelType model_type, Graph *graph, const… in Load() argument
85 if (model_type != kMindIR) { in Load()
109 Status Serialization::Load(const std::vector<std::vector<char>> &files, ModelType model_type, in Load() argument
120 Status Serialization::ExportModel(const Model &model, ModelType model_type, Buffer *model_data) { in ExportModel() argument
125 Status Serialization::ExportModel(const Model &model, ModelType model_type, const std::string &mode… in ExportModel() argument
136 if (model_type != kFlatBuffer) { in ExportModel()
137 MS_LOG(ERROR) << "Unsupported Export Format " << model_type; in ExportModel()
/third_party/mindspore/mindspore/ccsrc/cxx_api/graph/
Dgraph_data.cc23 Graph::GraphData::GraphData(const FuncGraphPtr &func_graph, enum ModelType model_type) in GraphData() argument
25 if (model_type != ModelType::kMindIR) {
26 MS_LOG(EXCEPTION) << "Invalid ModelType " << model_type;
29 model_type_ = model_type;
32 Graph::GraphData::GraphData(const Buffer &om_data, enum ModelType model_type) in GraphData() argument
33 : func_graph_(nullptr), om_data_(om_data), model_type_(model_type), data_graph_({}) { in GraphData()
Dgraph_data.h33 explicit GraphData(const FuncGraphPtr &func_graph, enum ModelType model_type = kMindIR);
35 GraphData(const Buffer &om_data, enum ModelType model_type);
/third_party/mindspore/mindspore/ccsrc/cxx_api/
Dserialization.cc103 Status Serialization::Load(const void *model_data, size_t data_size, ModelType model_type, Graph *g… in Load() argument
112 if (model_type == kMindIR) { in Load()
146 } else if (model_type == kOM) { in Load()
151 err_msg << "Unsupported ModelType " << model_type; in Load()
156 Status Serialization::Load(const std::vector<char> &file, ModelType model_type, Graph *graph) { in Load() argument
157 return Load(file, model_type, graph, Key{}, StringToChar(kDecModeAesGcm)); in Load()
160 Status Serialization::Load(const std::vector<char> &file, ModelType model_type, Graph *graph, const… in Load() argument
175 if (model_type == kMindIR) { in Load()
223 } else if (model_type == kOM) { in Load()
234 err_msg << "Unsupported ModelType " << model_type; in Load()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/
Dsentence_piece_vocab.cc34 … const float character_coverage, const SentencePieceModel model_type, in BuildFromFile() argument
54 if (model_type == SentencePieceModel::kWord) { in BuildFromFile()
56 } else if (model_type == SentencePieceModel::kBpe) { in BuildFromFile()
58 } else if (model_type == SentencePieceModel::kChar) { in BuildFromFile()
/third_party/mindspore/mindspore/dataset/text/
Dutils.py158 def from_dataset(cls, dataset, col_names, vocab_size, character_coverage, model_type, params): argument
193 DE_C_INTER_SENTENCEPIECE_MODE[model_type], params)
197 def from_file(cls, file_path, vocab_size, character_coverage, model_type, params): argument
235 DE_C_INTER_SENTENCEPIECE_MODE[model_type], params)
Dvalidators.py446 …[_, col_names, vocab_size, character_coverage, model_type, params], _ = parse_user_args(method, *a…
459 if model_type is not None:
461 type_check(model_type, (str, SentencePieceModel), "model_type")
476 …[file_path, vocab_size, character_coverage, model_type, params], _ = parse_user_args(method, *args…
487 if model_type is not None:
489 type_check(model_type, (str, SentencePieceModel), "model_type")
/third_party/mindspore/mindspore/lite/test/st/scripts/
Dbase_functions.sh6 …local cfg_file_list model_info model_name extra_info model_type cfg_file_name model_file weight_fi…
17 model_type=${model_name##*.}
20 case $model_type in
34 model_type="caffe"
77 …converter_result='converter '${model_type}''${quant_type}' '${model_name}' pass';echo ${converter_…
79 …converter_result='converter '${model_type}''${quant_type}' '${model_name}' failed';echo ${converte…
/third_party/mindspore/mindspore/lite/src/c_api/
Dmodel_c.cc35 …Status Build(const void *model_data, size_t data_size, ModelType model_type, const ContextC *model…
36 Status Build(const std::string &model_path, ModelType model_type, const ContextC *model_context);
56 Status ModelC::Build(const void *model_data, size_t data_size, ModelType model_type, const ContextC… in Build() argument
77 Status ModelC::Build(const std::string &model_path, ModelType model_type, const ContextC *model_con… in Build() argument
314 OH_AI_ModelType model_type, const OH_AI_ContextHandle model_context) { in OH_AI_ModelBuild() argument
319 if (model_type == OH_AI_MODELTYPE_INVALID) { in OH_AI_ModelBuild()
325 …auto ret = impl->Build(model_data, data_size, static_cast<mindspore::ModelType>(model_type), conte… in OH_AI_ModelBuild()
329 …_AI_ModelBuildFromFile(OH_AI_ModelHandle model, const char *model_path, OH_AI_ModelType model_type, in OH_AI_ModelBuildFromFile() argument
335 if (model_type == OH_AI_MODELTYPE_INVALID) { in OH_AI_ModelBuildFromFile()
341 auto ret = impl->Build(model_path, static_cast<mindspore::ModelType>(model_type), context); in OH_AI_ModelBuildFromFile()
/third_party/mindspore/mindspore/lite/src/cxx_api/model/
Dmodel.cc31 Status Model::Build(const void *model_data, size_t data_size, ModelType model_type, in Build() argument
42 Status ret = impl_->Build(model_data, data_size, model_type, model_context); in Build()
49 Status Model::Build(const std::string &model_path, ModelType model_type, const std::shared_ptr<Cont… in Build() argument
60 Status ret = impl_->Build(model_path, model_type, model_context); in Build()
132 bool Model::CheckModelSupport(enum DeviceType device_type, ModelType model_type) { in CheckModelSupport() argument
Dmodel_impl.h63 Status Build(const void *model_data, size_t data_size, ModelType model_type,
65 …Status Build(const std::string &model_path, ModelType model_type, const std::shared_ptr<Context> &…
85 static bool CheckModelSupport(const std::string &device_type, ModelType model_type);
/third_party/mindspore/include/sdk_api/
Dmodel.h136 … OH_AI_ModelType model_type, const OH_AI_ContextHandle model_context);
152 … OH_AI_ModelType model_type, const OH_AI_ContextHandle model_context);
/third_party/mindspore/include/c_api/
Dmodel_c.h75 … OH_AI_ModelType model_type, const OH_AI_ContextHandle model_context);
86 … OH_AI_ModelType model_type, const OH_AI_ContextHandle model_context);
/third_party/boost/boost/geometry/strategies/geographic/
Dazimuth.hpp40 typedef Spheroid model_type; typedef in boost::geometry::strategy::azimuth::geographic
50 inline model_type const& model() const in model()
/third_party/cef/libcef/browser/extensions/api/storage/
Dsync_value_store_cache.cc88 value_store_util::ModelType model_type = in GetStorage() local
93 model_type, extension->id(), in GetStorage()
/third_party/mindspore/mindspore/lite/src/train/
Dtransfer_session.cc181 int TransferSession::Export(const std::string &filename, ModelType model_type, QuantizationType qua… in Export() argument
188 if (model_type == MT_TRAIN) { in Export()
189 return TrainSession::Export(filename, model_type, quant_type, format); in Export()
209 if (!out_put_tensor_name.empty() && model_type == MT_INFERENCE) { in Export()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dbuild_sentence_piece_vocab_node.cc33 … float character_coverage, SentencePieceModel model_type, in BuildSentenceVocabNode() argument
39 model_type_(model_type), in BuildSentenceVocabNode()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/text/
Dbindings.cc57 const SentencePieceModel model_type, const py::dict &params) { in __anonf58a86e70502() argument
73 … path_list, vocab_size, character_coverage, model_type, param_map, &v)); in __anonf58a86e70502()
/third_party/boost/boost/geometry/srs/projections/impl/
Dpj_ell_set.hpp178 typedef typename traits_type::template model_type<T>::type model_type; in apply() typedef
181 model_type const& model = traits_type::template model<T>(param); in apply()
441 >::template model_type<double>::type // dummy type
460 >::template model_type<double>::type // dummy type
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dbuild_sentence_piece_vocab_op.cc26 … float character_coverage, SentencePieceModel model_type, in BuildSentencePieceVocabOp() argument
34 model_type_(model_type), in BuildSentencePieceVocabOp()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/ms/
Dms_model.cc174 bool MsModel::CheckModelSupport(enum ModelType model_type) { in CheckModelSupport() argument
187 auto iter = kSupportedModelMap.find(model_type); in CheckModelSupport()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/
Dmodel.cc177 bool Model::CheckModelSupport(enum DeviceType device_type, ModelType model_type) { in CheckModelSupport() argument
188 return check_model->CheckModelSupport(model_type); in CheckModelSupport()

12