0
我是tensorflow源代碼修改的新手。tensorflow.nn.conv2d - 輸入/內核matmul
我想嘗試一下conv2d算法的變體，並想知道輸入/內核matmul在源代碼中的具體位置，以及如何使用修改重建tensorflow。
A
回答
0
您可以閱讀this教程，瞭解如何向TensorFlow添加新操作。您將學習如何找到操作的實現（即從那裏的內核）。在其github頁面上有源碼編譯tensorflow有明確的說明。
對於Conv2D，該操作在core/ops/nn_ops.cc中定義。它有許多內核（用於CPU，用於GPU，用於XLA，基於MKL，基於Eigen）。你可以看到他們，如果你這樣做：tensorflow/core/kernels$ ls | grep conv
相關問題
- 1. Tensorflow - 具有批量數據的輸入矩陣的matmul
- 2. 內核中的NETLINK輸入函數
- 3. Tensorflow例外與MATMUL
- 4. MATMUL不同等級
- 5. 內核開發新手：內核中的「虛擬」輸入驅動程序？
- 6. 類型錯誤：的「MATMUL」歐普輸入「B」的類型是FLOAT32不匹配參數的INT32輸入「a」
- 7. Tensorflow matmul尺寸錯誤
- 8. TensorFlow中的MatMul問題
- 9. 如何入睡Linux內核？
- 10. OSX內核編程入門
- 11. C++ - 直接寫入內存（內核）
- 12. 如何在Linux內核捕獲鍵盤輸入
- 13. 從內核模塊獲取對輸入設備的引用
- 14. CUDA內核不能改變輸入數組
- 15. android：通過內核輸入事件/ dev/input/event *到java
- 16. 如何使用pycuda將二維數組輸入到內核中？
- 17. 內核＃嘗試讀取文件而不是標準輸入
- 18. 功能在輸入圖像和不同的內核大小
- 19. 阻止Linux內核中進程的輸入
- 20. 寫入內核模塊時發生內核非法指令
- 21. Linux是否放入所有內核的內核？
- 22. 如何優化在單個處理器內核上快速運行的矩陣乘法（matmul）代碼
- 23. 從內核調用內核
- 24. Linux內核的ftrace輸出格式
- 25. 在內核端加密logcat輸出
- 26. 內核模式輸出緩存限制
- 27. 關於Linux內核中的gcov輸出？
- 28. 發送SKB從內核空間傳輸
- 29. Linux內核調試打印輸出？
- 30. 使用內核模塊進行內存寫入導致Android內核中出現內核oops