Searched refs:use_v2 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | version_utils.py | 55 use_v2 = should_use_v2()
56 …cls = swap_class(cls, training.Model, training_v1.Model, use_v2) # pylint: disable=self-cls-assig…
64 use_v2 = should_use_v2()
65 …cls = swap_class(cls, base_layer.Layer, base_layer_v1.Layer, use_v2) # pylint: disable=self-cls-a…
73 use_v2 = should_use_v2()
76 use_v2)
100 def swap_class(cls, v2_cls, v1_cls, use_v2): argument
105 return v2_cls if use_v2 else v1_cls
110 if ((use_v2 and issubclass(base, v1_cls)
116 or (not use_v2 and issubclass(base, v2_cls)))):
[all …]
|