如何在TensorFlow C ++中加载自定义操作

时间:2018-04-04 08:14:53

标签: python c++ tensorflow customization libraries

我设计了一个自定义操作,它在python中按预期工作。我用通常的代码行来称呼它:

import tensorflow as tf
custom_mod = tf.load_op_library('/path/to/.so')

with tf.device("/gpu:0"):
  with tf.Session as sess:
    #(note: function name .exec() depends on kernel code)
    custom_mod_in_graph = custom_mod.exec() 
    sess.run(custom_mod_in_graph)

但是我需要使用TF C ++ API运行相同的操作,遗憾的是我找不到任何相应的调用过程(即:tf.load_op_library没有等效项)。

他们是否缺失或是否有不同的方式将操作集成在那里?

0 个答案:

没有答案