HomeSort by relevance Sort by last modified time
    Searched refs:mode_keys (Results 1 - 13 of 13) sorted by null

  /external/tensorflow/tensorflow/python/saved_model/model_utils/
mode_keys_test.py 22 from tensorflow.python.saved_model.model_utils import mode_keys
28 mode_map = mode_keys.ModeKeyMap(**{
29 mode_keys.KerasModeKeys.PREDICT: 3,
30 mode_keys.KerasModeKeys.TEST: 1
34 self.assertEqual(3, mode_map[mode_keys.KerasModeKeys.PREDICT])
35 self.assertEqual(3, mode_map[mode_keys.EstimatorModeKeys.PREDICT])
36 self.assertEqual(1, mode_map[mode_keys.KerasModeKeys.TEST])
37 self.assertEqual(1, mode_map[mode_keys.EstimatorModeKeys.EVAL])
39 _ = mode_map[mode_keys.KerasModeKeys.TRAIN]
41 _ = mode_map[mode_keys.EstimatorModeKeys.TRAIN
    [all...]
export_utils.py 33 from tensorflow.python.saved_model.model_utils import mode_keys
34 from tensorflow.python.saved_model.model_utils.mode_keys import KerasModeKeys as ModeKeys
39 EXPORT_TAG_MAP = mode_keys.ModeKeyMap(**{
49 SIGNATURE_KEY_MAP = mode_keys.ModeKeyMap(**{
283 if mode_keys.is_predict(mode):
285 elif mode_keys.is_train(mode):
export_test.py 34 from tensorflow.python.saved_model.model_utils.mode_keys import KerasModeKeys
  /external/tensorflow/tensorflow/python/keras/utils/
mode_keys.py 21 from tensorflow.python.saved_model.model_utils.mode_keys import KerasModeKeys as ModeKeys
  /external/tensorflow/tensorflow/python/keras/saving/
saved_model_test.py 38 from tensorflow.python.keras.utils import mode_keys
356 mode_keys.ModeKeys.PREDICT)
366 mode_keys.ModeKeys.TEST)
394 sess, saved_model_dir, mode_keys.ModeKeys.TRAIN)
431 mode_keys.ModeKeys.PREDICT)
508 mode_keys.ModeKeys.PREDICT)
538 mode_keys.ModeKeys.PREDICT)
saved_model.py 31 from tensorflow.python.keras.utils import mode_keys
188 _export_mode(mode_keys.ModeKeys.TRAIN, has_saved_vars, **export_args)
190 _export_mode(mode_keys.ModeKeys.TEST, has_saved_vars, **export_args)
197 _export_mode(mode_keys.ModeKeys.PREDICT, has_saved_vars, **export_args)
234 compile_clone = (mode != mode_keys.ModeKeys.PREDICT)
241 mode == mode_keys.ModeKeys.TRAIN):
263 if mode == mode_keys.ModeKeys.TRAIN:
266 elif mode == mode_keys.ModeKeys.TEST:
340 serving_only=(mode == mode_keys.ModeKeys.PREDICT))
  /external/tensorflow/tensorflow/python/keras/engine/
distributed_training_utils.py 37 from tensorflow.python.keras.utils.mode_keys import ModeKeys
    [all...]
training_arrays.py 36 from tensorflow.python.keras.utils.mode_keys import ModeKeys
training_generator.py 36 from tensorflow.python.keras.utils.mode_keys import ModeKeys
training_distributed.py 37 from tensorflow.python.keras.utils.mode_keys import ModeKeys
training.py 48 from tensorflow.python.keras.utils.mode_keys import ModeKeys
    [all...]
  /external/tensorflow/tensorflow/contrib/distribute/python/
keras_backward_compat_test.py 35 from tensorflow.python.keras.utils.mode_keys import ModeKeys
    [all...]
  /external/tensorflow/tensorflow/python/keras/
callbacks.py 39 from tensorflow.python.keras.utils.mode_keys import ModeKeys
    [all...]

Completed in 3581 milliseconds