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
相关文章
相关标签/搜索