Searched refs:TransferFromOutfeedLocal (Results 1 – 6 of 6) sorted by relevance
180 Status TransferFromOutfeedLocal(int device_ordinal,
436 Status LocalClient::TransferFromOutfeedLocal(int device_ordinal, in TransferFromOutfeedLocal() function in xla::LocalClient
96 TF_ASSERT_OK(client->TransferFromOutfeedLocal(device_ordinal, &outfeed)); in TestWithDeviceCount()
941 ASSERT_IS_OK(local_client_->TransferFromOutfeedLocal( in XLA_TEST_F()
287 client->TransferFromOutfeedLocal(/*device_ordinal=*/0, &outfeed)); in ReplayComputation()
982 return local_device->client()->TransferFromOutfeedLocal( in TransferFromOutfeed()