Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device_context.h90 se::Stream* GetDeviceToDeviceStream();
H A Dxla_device_context.cc291 se::Stream* XlaDeviceContext::GetDeviceToDeviceStream() { in GetDeviceToDeviceStream() function in tensorflow::XlaDeviceContext
H A Dxla_tpu_device.cc180 : dst_xla_context->GetDeviceToDeviceStream(); in TpuDeviceToDeviceCopy()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dlocal_device_state.h118 se::Stream* GetDeviceToDeviceStream();
H A Dlocal_device_state.cc139 se::Stream* LocalDeviceState::GetDeviceToDeviceStream() { in GetDeviceToDeviceStream() function in xla::LocalDeviceState
H A Dpjrt_stream_executor_client.cc1515 transfer_local_device->GetDeviceToDeviceStream(); in CopyToDevice()