Home
last modified time | relevance | path

Searched refs:TransferFromOutfeedLocal (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.h157 StatusOr<Literal> TransferFromOutfeedLocal(const Shape& shape,
Dlocal_client.cc403 StatusOr<Literal> LocalClient::TransferFromOutfeedLocal(const Shape& shape, in TransferFromOutfeedLocal() function in xla::LocalClient
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc95 client->TransferFromOutfeedLocal( in TestWithDeviceCount()
Dlocal_client_execute_test.cc918 local_client_->TransferFromOutfeedLocal( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc279 client->TransferFromOutfeedLocal(*outfeed_shape, /*device_ordinal=*/0) in ReplayComputation()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc515 local_device->client()->TransferFromOutfeedLocal( in PYBIND11_MODULE()