Searched refs:TPUStrategyV1 (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | category_encoding_distribution_test.py | 43 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
|
| D | category_crossing_distribution_test.py | 43 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_correctness_test_base.py | 335 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)) and
365 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)))
547 if (isinstance(distribution, tpu_strategy.TPUStrategyV1) and
|
| D | keras_rnn_model_correctness_test.py | 126 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
|
| D | keras_utils_test.py | 106 if (isinstance(distribution, tpu_strategy.TPUStrategyV1) and
477 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)) and
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | moving_averages_test.py | 194 distribution, (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
256 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
|
| D | strategy_combinations_test.py | 138 self.assertIsInstance(strategy, tpu_strategy.TPUStrategyV1)
|
| D | strategy_combinations.py | 114 strategy = tpu_lib.TPUStrategyV1(resolver, steps_per_run,
|
| D | strategy_common_test.py | 184 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1,
|
| D | metrics_v1_test.py | 103 tpu_strategy.TPUStrategyV1)):
|
| D | tpu_strategy.py | 681 class TPUStrategyV1(distribute_lib.StrategyV1): class
702 super(TPUStrategyV1, self).__init__(TPUExtended(
|
| D | strategy_gather_test.py | 676 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1,
|
| D | vars_test.py | 50 _TPU_STRATEGIES = (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)
|
| D | values_test.py | 1100 _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\'>"
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 800 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1,
|