Searched refs:TransferFromOutfeedLocal (Results 1 – 5 of 5) sorted by relevance
156 StatusOr<Literal> TransferFromOutfeedLocal(const Shape& shape,
299 StatusOr<Literal> LocalClient::TransferFromOutfeedLocal(const Shape& shape, in TransferFromOutfeedLocal() function in xla::LocalClient
94 client->TransferFromOutfeedLocal( in TestWithDeviceCount()
890 local_client_->TransferFromOutfeedLocal( in XLA_TEST_F()
254 client->TransferFromOutfeedLocal(*outfeed_shape, /*device_ordinal=*/0) in ReplayComputation()