Home
last modified time | relevance | path

Searched refs:model_config (Results 1 – 23 of 23) sorted by relevance

/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Dmodel_converter.cc52 ::sentencepiece::ModelProto model_config; in ConvertSentencepieceModelToFlatBuffer()
53 if (!model_config.ParseFromString(model_config_str)) { in ConvertSentencepieceModelToFlatBuffer()
56 model_config.InitializationErrorString()); in ConvertSentencepieceModelToFlatBuffer()
60 pieces.reserve(model_config.pieces_size()); in ConvertSentencepieceModelToFlatBuffer()
62 scores.reserve(model_config.pieces_size()); in ConvertSentencepieceModelToFlatBuffer()
64 ids.reserve(model_config.pieces_size()); in ConvertSentencepieceModelToFlatBuffer()
67 for (const auto& piece : model_config.pieces()) { in ConvertSentencepieceModelToFlatBuffer()
97 DecodePrecompiledCharsmap(model_config.normalizer_spec()); in ConvertSentencepieceModelToFlatBuffer()
108 ecb.add_start_code(model_config.trainer_spec().bos_id()); in ConvertSentencepieceModelToFlatBuffer()
109 ecb.add_end_code(model_config.trainer_spec().eos_id()); in ConvertSentencepieceModelToFlatBuffer()
[all …]
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Dmodel_handler.py173 model_config: ModelConfig,
180 cls).__new__(cls, model_config, enable_gpu, output_names,
193 model_config: ModelConfig,
198 cls).__new__(cls, test_name, model_config, cpu_base_result,
212 def __init__(self, model_config: ModelConfig):
213 self._model_config = model_config
222 def model_config(self) -> ModelConfig: member in _ModelHandlerBase
267 saved_model_dir=self.model_config.saved_model_dir,
268 saved_model_tags=self.model_config.saved_model_tags,
269 saved_model_signature_key=self.model_config.saved_model_signature_key)
[all …]
Drun_models.py120 for model_config in self._configs:
124 model_config=model_config,
137 model_name = os.path.split(manager.model_config.saved_model_dir)[-1]
Dresult_analyzer.py113 r = list(result.model_config) + [result.enable_gpu]
/external/tensorflow/tensorflow/python/keras/engine/
Dragged_keras_tensor_test.py111 model_config = model.get_config()
112 model2 = training.Model.from_config(model_config)
169 model_config = model.get_config()
170 model2 = training.Model.from_config(model_config)
190 model_config = model.get_config()
191 model2 = training.Model.from_config(model_config)
206 model_config = model.get_config()
207 model2 = training.Model.from_config(model_config)
222 model_config = model.get_config()
223 model2 = training.Model.from_config(model_config)
[all …]
Dinput_layer_test.py34 from tensorflow.python.keras.saving import model_config
278 model = model_config.model_from_json(model.to_json())
321 model = model_config.model_from_json(model.to_json())
370 model = model_config.model_from_json(model.to_json())
Dtraining.py2302 model_config = {
2308 return model_config
2334 model_config = self._updated_config()
2336 model_config, default=json_utils.get_json_type, **kwargs)
Dfunctional_test.py557 model_config = final_model.get_config()
558 recreated_model = models.Model.from_config(model_config)
617 model_config = multi_io_model.get_config()
618 recreated_model = models.Model.from_config(model_config)
/external/libtextclassifier/native/actions/
Dtflite-sensitive-model.cc30 const TFLiteSensitiveClassifierConfig* model_config) { in Create() argument
32 new TFLiteSensitiveModel(model_config)); in Create()
124 const TFLiteSensitiveClassifierConfig* model_config) in TFLiteSensitiveModel() argument
125 : model_config_(model_config), in TFLiteSensitiveModel()
127 model_config->model_spec()->tflite_model())) {} in TFLiteSensitiveModel()
Dtflite-sensitive-model.h31 const TFLiteSensitiveClassifierConfig* model_config);
39 const TFLiteSensitiveClassifierConfig* model_config);
/external/autotest/server/cros/faft/utils/
Dconfig.py71 model_config = model_configs.get(model, None)
72 if model_config is not None:
74 'MODEL:%s' % model, model_config, prepend=True)
/external/tensorflow/tensorflow/python/keras/saving/
Dhdf5_format.py30 from tensorflow.python.keras.saving import model_config as model_config_lib
179 model_config = f.attrs.get('model_config')
180 if model_config is None:
182 if hasattr(model_config, 'decode'):
183 model_config = model_config.decode('utf-8')
184 model_config = json_utils.decode(model_config)
185 model = model_config_lib.model_from_config(model_config,
Dsaving_utils.py152 model_config = {'class_name': model.__class__.__name__}
154 model_config['config'] = model.get_config()
162 model_config=model_config)
Dsaved_model_experimental.py30 from tensorflow.python.keras.saving import model_config
422 model = model_config.model_from_json(
DBUILD31 "model_config.py",
Dsave_test.py49 from tensorflow.python.keras.saving import model_config
181 loaded_model = model_config.model_from_json(config)
224 loaded_model = model_config.model_from_json(config)
/external/tensorflow/tensorflow/python/keras/
Dmodels.py34 from tensorflow.python.keras.saving import model_config
50 model_from_config = model_config.model_from_config
51 model_from_yaml = model_config.model_from_yaml
52 model_from_json = model_config.model_from_json
Dtesting_utils.py244 model_config = model.get_config()
245 recovered_model = models.Model.from_config(model_config, custom_objects)
301 model_config = model.get_config()
302 recovered_model = models.Sequential.from_config(model_config, custom_objects)
/external/tensorflow/tensorflow/python/keras/layers/
Dkernelized_test.py254 model_config = {
261 rff_layer = kernel_layers.RandomFourierFeatures.from_config(model_config)
Dtensorflow_op_layer_test.py35 from tensorflow.python.keras.saving import model_config
727 model = model_config.model_from_json(keras.Model(inputs, outputs).to_json())
/external/tensorflow/tensorflow/python/keras/feature_column/
Dsequence_feature_column_test.py36 from tensorflow.python.keras.saving import model_config
647 loaded_model = model_config.model_from_json(config)
/external/tensorflow/tensorflow/python/keras/api/
DBUILD97 "tensorflow.python.keras.saving.model_config",
/external/llvm-project/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/
Dsaved_model.pbtxt8465 …alse, \"keras_version\": \"2.2.4-tf\", \"backend\": \"tensorflow\", \"model_config\": {\"class_nam…