tfp.experimental.auto_batching.stackless.ExecutionQueue
Stay organized with collections
Save and categorize content based on your preferences.
A priority queue of resumption points.
tfp.experimental.auto_batching.stackless.ExecutionQueue(
backend
)
Each resumption point is a pair of program counter to resume, and mask of
threads that are waiting there.
This class is a simple wrapper around Python's standard heapq implementation
of priority queues. There are just two subtleties:
Dequeue gets all the threads that were waiting at that point, by coalescing
multiple entries if needed.
Enqueue drops entries with empty masks, because they need never be resumed.
Methods
dequeue
View source
dequeue()
enqueue
View source
enqueue(
program_counter, mask
)
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2023-11-21 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2023-11-21 UTC."],[],[]]
View source on GitHub