Searched refs:model_string (Results 1 – 4 of 4) sorted by relevance
39 std::string ConvertSentencepieceModel(const std::string& model_string);41 const std::string& model_string);45 int GetVocabularySize(const std::string& model_string);
171 int GetVocabularySize(const std::string& model_string) { in GetVocabularySize() argument172 const EncoderConfig* config = GetEncoderConfig(model_string.data()); in GetVocabularySize()176 std::string ConvertSentencepieceModel(const std::string& model_string) { in ConvertSentencepieceModel() argument177 const auto result = ConvertSentencepieceModelToFlatBuffer(model_string); in ConvertSentencepieceModel()185 const std::string& model_string) { in ConvertSentencepieceModelForDecoder() argument187 ConvertSentencepieceModelToFlatBufferForDecoder(model_string); in ConvertSentencepieceModelForDecoder()
377 model_string = values['model'].lower()378 if 'intel' in model_string or 'amd' in model_string:380 elif 'aarch64' in model_string or 'armv8' in model_string:382 elif 'armv7' in model_string:385 logging.warning('unknown family for cpu model \'%s\'', model_string)387 match = model_re.search(model_string)
112 model_string: bytes, graph_module: Optional[torch.fx.GraphModule] = None127 f.write(model_string)