Searched refs:experimental_split_to_logical_devices (Results 1 – 2 of 2) sorted by relevance
64 name: "experimental_split_to_logical_devices"
460 def experimental_split_to_logical_devices(self, tensor, partition_dimensions): member in TPUStrategyV2