Searched refs:EnqueueD2DTransfersOnSrcStream (Results 1 – 3 of 3) sorted by relevance
85 bool EnqueueD2DTransfersOnSrcStream() const override { return false; } in EnqueueD2DTransfersOnSrcStream() function in xla::__anon240436b40111::PjRtTpuClient
162 virtual bool EnqueueD2DTransfersOnSrcStream() const { return true; } in EnqueueD2DTransfersOnSrcStream() function
1380 ->EnqueueD2DTransfersOnSrcStream() in CopyToDevice()