Home
last modified time | relevance | path

Searched refs:TPUExtended (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py333 super(TPUStrategyV2, self).__init__(TPUExtended(
640 super(TPUStrategy, self).__init__(TPUExtended(
702 super(TPUStrategyV1, self).__init__(TPUExtended(
783 class TPUExtended(distribute_lib.StrategyExtendedV1): class
791 super(TPUExtended, self).__init__(container_strategy)
Dvalues_test.py488 tpu_strategy.TPUExtended) and context.executing_eagerly():
502 if isinstance(distribution.extended, tpu_strategy.TPUExtended):
562 if not isinstance(distribution.extended, tpu_strategy.TPUExtended):
590 if not (isinstance(distribution.extended, tpu_strategy.TPUExtended) and
820 if isinstance(distribution.extended, tpu_strategy.TPUExtended):