Searched refs:model_proto (Results 1 – 6 of 6) sorted by relevance
78 std::string model_proto; in BuildFromFile() local79 …::Status s_status = sentencepiece::SentencePieceTrainer::Train(unorder_map, nullptr, &model_proto); in BuildFromFile()84 vocab->get()->set_model_proto(model_proto); in BuildFromFile()104 (void)os_file.write(vocab->get()->model_proto().data(), vocab->get()->model_proto().size()); in SaveModel()112 const std::string &SentencePieceVocab::model_proto() { return model_proto_; } in model_proto() function in mindspore::dataset::SentencePieceVocab114 …d SentencePieceVocab::set_model_proto(const std::string model_proto) { model_proto_ = model_proto;… in set_model_proto() argument
41 const std::string &model_proto();43 void set_model_proto(const std::string model_proto);
84 std::string model_proto; in SentenceThread() local86 sentencepiece::SentencePieceTrainer::Train(BuildParams(), sentence_iter.get(), &model_proto); in SentenceThread()94 vocab_->set_model_proto(model_proto); in SentenceThread()
1063 bool MSANFModelParser::MSANFParseModelConfigureInfo(const mind_ir::ModelProto &model_proto) { in MSANFParseModelConfigureInfo() argument1064 if (!model_proto.has_producer_name()) { in MSANFParseModelConfigureInfo()1068 producer_name_ = model_proto.producer_name(); in MSANFParseModelConfigureInfo()1071 if (!model_proto.has_model_version()) { in MSANFParseModelConfigureInfo()1075 model_version_ = model_proto.model_version(); in MSANFParseModelConfigureInfo()1078 if (!model_proto.has_ir_version()) { in MSANFParseModelConfigureInfo()1082 ir_version_ = model_proto.ir_version(); in MSANFParseModelConfigureInfo()1087 FuncGraphPtr MSANFModelParser::Parse(const mind_ir::ModelProto &model_proto) { in Parse() argument1090 if (!MSANFParseModelConfigureInfo(model_proto)) { in Parse()1093 const mind_ir::GraphProto &graphBuild = model_proto.graph(); in Parse()[all …]
37 FuncGraphPtr Parse(const mind_ir::ModelProto &model_proto);38 bool MSANFParseModelConfigureInfo(const mind_ir::ModelProto &model_proto);
31 auto status = processor_.LoadFromSerializedProto(vocab_.get()->model_proto()); in SentencePieceTokenizerOp()