Searched refs:transfer_tpu_stream (Results 1 – 1 of 1) sorted by relevance
69 auto* transfer_tpu_stream = tensorflow::down_cast<tf_tpu::TpuStream*>( in ThenMemcpyDeviceToDevice() local71 TF_RETURN_IF_ERROR(transfer_tpu_stream->EnqueueOnTpuDeviceSendRecvLocal( in ThenMemcpyDeviceToDevice()