Home
last modified time | relevance | path

Searched refs:tpu_strategies (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/keras/distribute/
Dstrategy_combinations.py55 tpu_strategies = [ variable
59 all_strategies = strategies_minus_tpu + tpu_strategies
Dkeras_correctness_test_base.py107 tpu_strategies = [
112 combinations.combine(distribution=tpu_strategies),
Ddistribute_strategy_test.py59 from tensorflow.python.keras.distribute.strategy_combinations import tpu_strategies
246 distribution=tpu_strategies, mode=['graph', 'eager'])
250 return combinations.combine(distribution=tpu_strategies, mode=['graph'])
299 distribution=tpu_strategies,
312 distribution=tpu_strategies,
2362 distribution=strategies_minus_default_minus_tpu + tpu_strategies +
Dcustom_training_loop_models_test.py455 distribution=strategy_combinations.tpu_strategies, mode=["eager"]))
/external/tensorflow/tensorflow/python/distribute/
Dstrategy_combinations.py348 tpu_strategies = [ variable
355 all_strategies_minus_default = strategies_minus_default_and_tpu + tpu_strategies
357 all_strategies = strategies_minus_tpu + tpu_strategies
395 return combinations.combine(distribution=tpu_strategies, mode=["graph"])
Dstrategy_combinations_test.py136 combinations.combine(strategy=strategy_combinations.tpu_strategies))
189 combinations.combine(strategy=strategy_combinations.tpu_strategies))
Dcustom_training_loop_input_test.py187 distribution=strategy_combinations.tpu_strategies, mode=["eager"]))
454 distribution=strategy_combinations.tpu_strategies,
474 distribution=strategy_combinations.tpu_strategies,