Home
last modified time | relevance | path

Searched refs:CreateHostToDeviceChannelHandle (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_client.h143 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() { in CreateHostToDeviceChannelHandle() function
144 return pjrt_client_->CreateHostToDeviceChannelHandle(); in CreateHostToDeviceChannelHandle()
Dxla.cc222 &PyClient::CreateHostToDeviceChannelHandle) in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h221 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle();
Dclient.cc509 StatusOr<ChannelHandle> Client::CreateHostToDeviceChannelHandle() { in CreateHostToDeviceChannelHandle() function in xla::Client
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h263 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() override { in CreateHostToDeviceChannelHandle() function
264 return client()->CreateHostToDeviceChannelHandle(); in CreateHostToDeviceChannelHandle()
Dpjrt_c_api_client.h257 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() override { in CreateHostToDeviceChannelHandle() function
Dtfrt_cpu_pjrt_client.h218 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() override { in CreateHostToDeviceChannelHandle() function
Dpjrt_client.h772 virtual StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() = 0;
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc1456 client()->CreateHostToDeviceChannelHandle()); in GetHostToDeviceChannelHandle()