Searched refs:WriteTupleIndexTablesAsync (Results 1 – 7 of 7) sorted by relevance
40 TF_RETURN_IF_ERROR(transfer_manager->WriteTupleIndexTablesAsync( in PopulateResultTupleBuffers()47 return transfer_manager->WriteTupleIndexTablesAsync(stream, result); in PopulateResultTupleBuffers()
106 TF_RETURN_IF_ERROR(WriteTupleIndexTablesAsync(stream, device_buffer)); in TransferLiteralToDeviceAsync()
289 TF_RETURN_IF_ERROR(WriteTupleIndexTablesAsync(stream, device_buffer)); in WriteTupleIndexTables()293 Status TransferManager::WriteTupleIndexTablesAsync( in WriteTupleIndexTablesAsync() function in xla::TransferManager
220 Status WriteTupleIndexTablesAsync(se::Stream* stream,
248 dst_xla_context->transfer_manager()->WriteTupleIndexTablesAsync( in TpuDeviceToDeviceCopy()
261 transfer_manager->WriteTupleIndexTablesAsync(stream, shaped_buffer)); in UpdateDynamicInputs()
435 TF_RETURN_IF_ERROR(transfer_manager->WriteTupleIndexTablesAsync( in AllocateDestinationBuffer()