Home
last modified time | relevance | path

Searched refs:EnqueueD2DTransfersOnSrcStream (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtpu_client.h84 bool EnqueueD2DTransfersOnSrcStream() const override { return false; } in EnqueueD2DTransfersOnSrcStream() function
H A Dpjrt_stream_executor_client.h188 virtual bool EnqueueD2DTransfersOnSrcStream() const { return true; } in EnqueueD2DTransfersOnSrcStream() function
H A Dpjrt_stream_executor_client.cc1509 client_->EnqueueD2DTransfersOnSrcStream() ? device_->local_device_state() in CopyToDevice()