Searched refs:_create_device_transfers (Results 1 – 1 of 1) sorted by relevance
1643 def _create_device_transfers(self, tensors): member in BaseStagingArea1674 tensors = self._create_device_transfers(tensors)2251 key = self._create_device_transfers(key)[0]