From 10ab1efd528bb4f1354061a92a43dbd8e0322d80 Mon Sep 17 00:00:00 2001 From: Fu-Hsiang Chan Date: Thu, 24 Nov 2016 22:27:06 +0800 Subject: [PATCH] Update roi_pooling_op.cc --- lib/roi_pooling_layer/roi_pooling_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/roi_pooling_layer/roi_pooling_op.cc b/lib/roi_pooling_layer/roi_pooling_op.cc index 944ef4cc..79fb842d 100644 --- a/lib/roi_pooling_layer/roi_pooling_op.cc +++ b/lib/roi_pooling_layer/roi_pooling_op.cc @@ -22,7 +22,7 @@ limitations under the License. #include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/framework/tensor_shape.h" -#include "tensorflow/core/util/work_sharder.h" +#include "work_sharder.h" using namespace tensorflow; typedef Eigen::ThreadPoolDevice CPUDevice;