Searched refs:_mirrored_strategy_with_collective_key_base (Results 1 – 1 of 1) sorted by relevance
122 def _mirrored_strategy_with_collective_key_base(devices): function254 lambda: _mirrored_strategy_with_collective_key_base(["/cpu:0"]))257 lambda: _mirrored_strategy_with_collective_key_base(["/gpu:0"]),261 lambda: _mirrored_strategy_with_collective_key_base(["/gpu:0", "/cpu:0"]),265 lambda: _mirrored_strategy_with_collective_key_base(["/gpu:0", "/gpu:1"]),270 lambda: _mirrored_strategy_with_collective_key_base(["/cpu:1", "/cpu:2"]))