Tensorflow添加引用某些函数

Question

我正在尝试在tensorflow中添加一些操作。但是，当我创建函数时会出现一些问题。

在我的函数中，存在core / kernels / training_op_helpers.h中存在的函数。

Bazel说它需要依赖，所以我添加到bazel BUILD文件，但是有错误，这是它有非法依赖。

我发现tensorflow.bzl中的tf_custom_op_library中存在一些不允许的依赖。这有什么特别的原因吗？

但是有没有办法在tensorflow函数中使用一些函数？我认为当我在张量流中添加op时，使用该函数存在一些限制。

由于

Answer 1

自定义操作的bazel项目必须与从属文件位于同一工作区中。并将tensorflow .bzl加载到您的BUILD文件中。