Home
last modified time | relevance | path

Searched refs:EnqueueOnTpuDeviceSendRecvLocal (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_stream_interface.h31 virtual Status EnqueueOnTpuDeviceSendRecvLocal(
Dtpu_stream.h65 Status EnqueueOnTpuDeviceSendRecvLocal( in EnqueueOnTpuDeviceSendRecvLocal() function
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtpu_client.cc71 TF_RETURN_IF_ERROR(transfer_tpu_stream->EnqueueOnTpuDeviceSendRecvLocal( in ThenMemcpyDeviceToDevice()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tpu_device.cc241 dst_device_to_device_stream_impl->EnqueueOnTpuDeviceSendRecvLocal( in TpuDeviceToDeviceCopy()