Home
last modified time | relevance | path

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

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