Searched refs:TPUStrategyV1 (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_correctness_test_base.py | 334 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)) and
364 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)))
553 if (isinstance(distribution, tpu_strategy.TPUStrategyV1) and
|
| D | keras_rnn_model_correctness_test.py | 120 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
|
| D | keras_utils_test.py | 104 if (isinstance(distribution, tpu_strategy.TPUStrategyV1) and
555 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)) and
|
| D | distribute_strategy_test.py | 146 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
468 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
517 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | moving_averages_test.py | 179 distribution, (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
237 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
|
| D | strategy_combinations.py | 82 return tpu_lib.TPUStrategyV1(resolver, steps_per_run,
|
| D | tpu_strategy.py | 171 class TPUStrategyV1(distribute_lib.StrategyV1): class
192 super(TPUStrategyV1, self).__init__(TPUExtended(
|
| D | metrics_v1_test.py | 103 tpu_strategy.TPUStrategyV1)):
|
| D | values_test.py | 803 _TPU_STRATEGIES = (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 3 is_instance: "<class \'tensorflow.python.distribute.tpu_strategy.TPUStrategyV1\'>"
|