tensorflow 源代碼分析(4)-自定義op

添加自定義op 本文介紹如何在tensorflow中增加一個新得OP以及op對應得kernel,還有如何編譯和使用新增加得op。 1.新增加op對應得kernel實現,新的op ZeroOutOp類繼承自OpKernel,重新實現Compute接口,在該接口內實現op對應得kernel(tensorflow/core/user_ops/zero_out.cc cpu設備上的實現): REGISTE
相關文章
相關標籤/搜索