Searched refs:experimental_split_to_logical_devices (Results 1 – 3 of 3) sorted by relevance
64 name: "experimental_split_to_logical_devices"
340 x = strategy.experimental_split_to_logical_devices(x, [1, 2])
504 def experimental_split_to_logical_devices(self, tensor, partition_dimensions): member in TPUStrategyV2