在pywrap_tensorflow.py中找不到Tensorflow TFE_Py_FastPathExecute变量

时间:2019-03-16 19:19:32

标签: python tensorflow

我正在阅读gen_nn_ops.py包中tensorflow.python.ops中的 LeakyRelu 的源代码。其中的函数使用八个参数调用_pywrap_tensorflow.TFE_Py_FastPathExecute

但是,TFE_Py_FastPathExecute中不存在pywrap_tensorflow.py成员。它是从哪里引用的?

1 个答案:

答案 0 :(得分:0)

pywrap_tfe.i行286-290中,显示为:

// Since we wanted to have a function with a variable length of arguments, we
// used the native Python/C interface directly (which by default supports
// passing all arguments as a tuple).
%native(TFE_Py_FastPathExecute) TFE_Py_FastPathExecute_C;

这很可能引用pywrap_tfe_src.cc,在这里我们在行1832-2132中找到TFE_Py_FastPathExecute_C,从 PyObject* TFE_Py_FastPathExecute_C(PyObject*, PyObject* args)