在不加载外部动态库的情况下添加TensorFlow自定义操作

时间:2019-07-15 22:41:07

标签: tensorflow

我一直在浏览tutorial,以便在TensorFlow中添加自定义操作,在本教程中,看起来自定义操作是单独构建的,然后加载了tf.load_op_library。我的问题是,如果您想直接将自定义应用程序直接添加到TensorFlow源代码中,然后进行构建。除了代码必须放在ops目录中之外,我是否应该假定执行相同的步骤?

0 个答案:

没有答案