Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.h87 se::Stream* GetDeviceToDeviceStream();
Dxla_device_context.cc289 se::Stream* XlaDeviceContext::GetDeviceToDeviceStream() { in GetDeviceToDeviceStream() function in tensorflow::XlaDeviceContext
Dxla_tpu_device.cc177 : dst_xla_context->GetDeviceToDeviceStream(); in TpuDeviceToDeviceCopy()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dlocal_device_state.h118 se::Stream* GetDeviceToDeviceStream();
Dlocal_device_state.cc114 se::Stream* LocalDeviceState::GetDeviceToDeviceStream() { in GetDeviceToDeviceStream() function in xla::LocalDeviceState
Dpjrt_stream_executor_client.cc1388 transfer_local_device->GetDeviceToDeviceStream(); in CopyToDevice()