Searched refs:experimental_assign_to_logical_device (Results 1 – 2 of 2) sorted by relevance
36 name: "experimental_assign_to_logical_device"
404 def experimental_assign_to_logical_device(self, tensor, logical_device_id): member in TPUStrategyV2