Searched refs:KerasModeKeys (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/saved_model/model_utils/ |
| D | mode_keys_test.py | 29 mode_keys.KerasModeKeys.PREDICT: 3,
30 mode_keys.KerasModeKeys.TEST: 1
34 self.assertEqual(3, mode_map[mode_keys.KerasModeKeys.PREDICT])
36 self.assertEqual(1, mode_map[mode_keys.KerasModeKeys.TEST])
39 _ = mode_map[mode_keys.KerasModeKeys.TRAIN]
49 {mode_keys.KerasModeKeys.TEST, mode_keys.KerasModeKeys.PREDICT},
54 mode_map[mode_keys.KerasModeKeys.TEST] = 1
59 mode_keys.KerasModeKeys.PREDICT: 3,
|
| D | mode_keys.py | 25 class KerasModeKeys(object): class
57 return mode in [KerasModeKeys.PREDICT, EstimatorModeKeys.PREDICT]
61 return mode in [KerasModeKeys.TEST, EstimatorModeKeys.EVAL]
65 return mode in [KerasModeKeys.TRAIN, EstimatorModeKeys.TRAIN]
95 return KerasModeKeys.TRAIN
97 return KerasModeKeys.TEST
99 return KerasModeKeys.PREDICT
|
| D | export_test.py | 35 from tensorflow.python.saved_model.model_utils.mode_keys import KerasModeKeys
275 ret = _build_export_output(KerasModeKeys.TRAIN)
283 ret = _build_export_output(KerasModeKeys.TEST)
291 ret = _build_export_output(KerasModeKeys.PREDICT)
299 KerasModeKeys.PREDICT,
|
| D | export_utils.py | 34 from tensorflow.python.saved_model.model_utils.mode_keys import KerasModeKeys as ModeKeys
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | mode_keys.py | 21 from tensorflow.python.saved_model.model_utils.mode_keys import KerasModeKeys as ModeKeys
|