tensorflow에서 conv_ops와 같은 연산을 만드는 방법은 무엇입니까?

Question

내가 bazel C에 새로 온 ++ 및

을 할 노력하고있어 나는 tensorflow의 컨볼 루션 연산에 대한 몇 가지 변화를 만들고 싶어 무엇 , 그래서 난 내 첫 번째 단계는 단지 작전을 만들 것을 결정 그것처럼. 나는 내가 conv_ops.cc를 복사 // tensorflow/코어/커널에서 내 new_conv_ops.cc에 내 등록 작전의 이름을 변경

했을 무엇

. 또한 중복을 피하기 위해 파일의 일부 함수 이름을 변경했습니다. 그리고 여기에 내 BUILD가 있습니다 file.

당신이 볼 수 있듯이, 나는 conv_ops의 deps 속성을 // tensorflow/core/kernels/BUILD에서 복사합니다. 그런 다음 "bazel build -c opt //tensorflow/core/user_ops:new_conv_ops.so"를 사용하여 새 op를 빌드합니다.

내 문제가

그럼 내가 bounds_check을 삭제하려고이 error.

을 가지고, 다음 deps에 대한 동일한 오류가 발생했습니다 무엇입니까. 그렇다면 // tensorflow/core/user_ops에있는 // tensorflow/core/kernels에 h 파일을 포함하는 것에 약간의 문제가 있다는 것을 알고 있습니다. 그렇다면 어떻게 conv_ops와 같은 새로운 op를 완벽하게 만들 수 있습니까?

Answer 1

TensorFlow에 사용자 지정 작업을 추가하는 방법은 here 자습서에서 다룹니다. 실제 코드 examples을 볼 수도 있습니다.

특정 문제를 해결하려면 tf_custom_op_library 매크로가 필요한 종속성의 대부분을 대상에 추가합니다. 다음과 같이 간단히 작성할 수 있습니다.

tf_custom_op_library(
    name="new_conv_ops.so", 
    srcs=["new_conv_ops.cc"] 
)