Searched refs:get_tpu_strategy (Results 1 – 7 of 7) sorted by relevance
80 def get_tpu_strategy(enable_packed_var=False): function172 strategy = get_tpu_strategy(enable_packed_var)185 strategy = get_tpu_strategy(enable_packed_var)227 strategy = get_tpu_strategy(enable_packed_var)261 strategy = get_tpu_strategy(enable_packed_var)288 strategy = get_tpu_strategy(enable_packed_var)378 strategy = get_tpu_strategy(enable_packed_var)397 strategy = get_tpu_strategy(enable_packed_var)434 strategy = get_tpu_strategy(enable_packed_var)455 strategy = get_tpu_strategy(enable_packed_var)[all …]
45 def get_tpu_strategy(): function59 strategy = get_tpu_strategy()
72 def get_tpu_strategy(): function135 strategy = get_tpu_strategy()155 strategy = get_tpu_strategy()175 strategy = get_tpu_strategy()196 strategy = get_tpu_strategy()224 strategy = get_tpu_strategy()252 strategy = get_tpu_strategy()281 strategy = get_tpu_strategy()304 strategy = get_tpu_strategy()328 strategy = get_tpu_strategy()[all …]
71 def get_tpu_strategy(): function186 strategy = get_tpu_strategy()216 strategy = get_tpu_strategy()248 strategy = get_tpu_strategy()
50 def get_tpu_strategy(): function101 strategy = get_tpu_strategy()119 strategy = get_tpu_strategy()
45 def get_tpu_strategy(): function
115 strategy = tpu_strategy_test_utils.get_tpu_strategy()