Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_client.h124 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() { in CreateHostToDeviceChannelHandle() function
125 return pjrt_client_->CreateHostToDeviceChannelHandle(); in CreateHostToDeviceChannelHandle()
Dxla.cc221 &PyClient::CreateHostToDeviceChannelHandle) in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h222 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle();
Dclient.cc508 StatusOr<ChannelHandle> Client::CreateHostToDeviceChannelHandle() { in CreateHostToDeviceChannelHandle() function in xla::Client
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h210 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() override { in CreateHostToDeviceChannelHandle() function
211 return client()->CreateHostToDeviceChannelHandle(); in CreateHostToDeviceChannelHandle()
Dpjrt_client.h264 virtual StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() = 0;
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc1420 client()->CreateHostToDeviceChannelHandle()); in GetHostToDeviceChannelHandle()