Searched refs:TransferFromOutfeedLocal (Results 1 – 6 of 6) sorted by relevance
157 StatusOr<Literal> TransferFromOutfeedLocal(const Shape& shape,
403 StatusOr<Literal> LocalClient::TransferFromOutfeedLocal(const Shape& shape, in TransferFromOutfeedLocal() function in xla::LocalClient
95 client->TransferFromOutfeedLocal( in TestWithDeviceCount()
918 local_client_->TransferFromOutfeedLocal( in XLA_TEST_F()
279 client->TransferFromOutfeedLocal(*outfeed_shape, /*device_ordinal=*/0) in ReplayComputation()
515 local_device->client()->TransferFromOutfeedLocal( in PYBIND11_MODULE()