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()
110 TF_RETURN_IF_ERROR(WriteTupleIndexTablesAsync(stream, device_buffer)); in TransferLiteralToDeviceAsync()
283 TF_RETURN_IF_ERROR(WriteTupleIndexTablesAsync(stream, device_buffer)); in WriteTupleIndexTables()287 Status TransferManager::WriteTupleIndexTablesAsync( in WriteTupleIndexTablesAsync() function in xla::TransferManager
218 Status WriteTupleIndexTablesAsync(se::Stream* stream,
256 dst_xla_context->transfer_manager()->WriteTupleIndexTablesAsync( in TpuDeviceToDeviceCopy()
261 transfer_manager->WriteTupleIndexTablesAsync(stream, shaped_buffer)); in UpdateDynamicInputs()
445 TF_RETURN_IF_ERROR(transfer_manager->WriteTupleIndexTablesAsync( in AllocateDestinationBuffer()