Update roi_pooling_op.cc

Arthur-Shi · Nov 24, 2016 · 10ab1ef · 10ab1ef
1 parent d72886d
commit 10ab1ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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;