Searched refs:tpu_strategy_one_step (Results 1 – 5 of 5) sorted by relevance
52 combinations.tpu_strategy_one_step,96 combinations.tpu_strategy_one_step]
361 tpu_strategy_one_step = NamedDistribution( variable
87 return combinations.combine(distribution=[combinations.tpu_strategy_one_step,
302 combinations.tpu_strategy_one_step]669 distribution=[combinations.tpu_strategy_one_step],
274 combinations.tpu_strategy_one_step]1076 distribution=[combinations.tpu_strategy_one_step],