我设计了一个自定义操作,它在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
没有等效项)。
他们是否缺失或是否有不同的方式将操作集成在那里?