Tensorflow設備vs DeviceContext

Question

我正在實施Tensorflow的新設備。我想在設備和DeviceContext之間進行一些說明。我已閱讀this question，但我認爲我需要更多信息。

是否應該讓我的系統中的每個設備都有一個設備實例，並且設備實例保留有關該物理設備的信息？然後DeviceContext應該維護關於這個設備的運行時信息。

在另一個問題中，答案表明GPU設備維護幾個設備上下文，每個設備上下文對應一個流，給予特定作業的流（複製與執行）。這聽起來像內核操作綁定到特定的設備上下文，如果是的話，何時/何地發生？

Answer 1

由於GPUDevice有multiple contexts per device，我認爲你不需要每個設備都有一個上下文。因此，我同意設備類將包含關於實際硬件的最小數據，設備上下文將表現爲更多的設備運行時控制（處理內存分配，數據傳輸，執行等），由names of the functions在上下文中

內核操作符與GPU中上下文的綁定發生在FillContextMap中，其中計算圖節點附加到設備上下文。