Home
last modified time | relevance | path

Searched refs:TransferToInfeedLocal (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.h149 Status TransferToInfeedLocal(const Literal& literal, int device_ordinal);
Dlocal_client.cc291 Status LocalClient::TransferToInfeedLocal(const Literal& literal, in TransferToInfeedLocal() function in xla::LocalClient
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc87 TF_ASSERT_OK(client->TransferToInfeedLocal( in TestWithDeviceCount()
Dlocal_client_execute_test.cc861 ASSERT_IS_OK(local_client_->TransferToInfeedLocal( in XLA_TEST_F()
885 ASSERT_IS_OK(local_client_->TransferToInfeedLocal( in XLA_TEST_F()