Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.h173 Status TransferToInfeedLocal(const LiteralSlice& literal, int device_ordinal);
Dlocal_client.cc428 Status LocalClient::TransferToInfeedLocal(const LiteralSlice& literal, in TransferToInfeedLocal() function in xla::LocalClient
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc89 TF_ASSERT_OK(client->TransferToInfeedLocal( in TestWithDeviceCount()
Dlocal_client_execute_test.cc912 ASSERT_IS_OK(local_client_->TransferToInfeedLocal( in XLA_TEST_F()
936 ASSERT_IS_OK(local_client_->TransferToInfeedLocal( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc975 return local_device->client()->TransferToInfeedLocal( in TransferToInfeed()