Home
last modified time | relevance | path

Searched refs:TPUStrategyV2 (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_test.py84 strategy = tpu_lib.TPUStrategyV2(resolver)
203 strategy = tpu_lib.TPUStrategyV2(
210 strategy2 = tpu_lib.TPUStrategyV2(
311 all_core_strategy = tpu_lib.TPUStrategyV2(resolver)
321 first_core_strategy = tpu_lib.TPUStrategyV2(
329 second_core_strategy = tpu_lib.TPUStrategyV2(
356 first_core_strategy = tpu_lib.TPUStrategyV2(
364 second_core_strategy = tpu_lib.TPUStrategyV2(
976 strategy = tpu_lib.TPUStrategyV2(resolver)
1106 strategy = tpu_lib.TPUStrategyV2(
[all …]
Dtpu_strategy_compilation_test.py49 strategy = tpu_lib.TPUStrategyV2(resolver)
Dinput_lib_type_spec_test.py424 (tpu_strategy.TPUStrategyV2, tpu_strategy.TPUStrategy)):
486 (tpu_strategy.TPUStrategyV2, tpu_strategy.TPUStrategy)):
Dstrategy_common_test.py185 tpu_strategy.TPUStrategyV2))
Dtpu_strategy.py225 class TPUStrategyV2(distribute_lib.Strategy): class
333 super(TPUStrategyV2, self).__init__(TPUExtended(
Dstrategy_gather_test.py677 tpu_strategy.TPUStrategyV2))
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-t-p-u-strategy.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.tpu_strategy.TPUStrategyV2\'>"
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py801 tpu_strategy.TPUStrategyV2)):
/external/tensorflow/tensorflow/python/tpu/
Dtpu_outside_compilation_test.py76 return tpu_lib.TPUStrategyV2(resolver)
Dtpu_embedding_v2.py280 tpu_strategy.TPUStrategyV2))
/external/tensorflow/tensorflow/python/keras/
Dcallbacks.py1614 tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV2)