Skip to content

Commit 005feb1

Browse files
authored
Update cc codepaths for customops (#2436)
1 parent 7136ceb commit 005feb1

File tree

6 files changed

+6
-6
lines changed

6 files changed

+6
-6
lines changed

MANIFEST.in

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
recursive-include keras_cv/custom_ops *.so
1+
recursive-include keras_cv/src/custom_ops *.so

keras_cv/src/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,6 @@ py_library(
1111
name = "keras_cv",
1212
srcs = glob(["**/*.py"]),
1313
data = [
14-
"//keras_cv/custom_ops:_keras_cv_custom_ops.so",
14+
"//keras_cv/src/custom_ops:_keras_cv_custom_ops.so",
1515
]
1616
)

keras_cv/src/custom_ops/box_util.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
1313
limitations under the License.
1414
==============================================================================*/
1515

16-
#include "keras_cv/custom_ops/box_util.h"
16+
#include "keras_cv/src/custom_ops/box_util.h"
1717

1818
#include <algorithm>
1919
#include <cmath>

keras_cv/src/custom_ops/kernels/pairwise_iou_kernel.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ limitations under the License.
1515

1616
#include <vector>
1717

18-
#include "keras_cv/custom_ops/box_util.h"
18+
#include "keras_cv/src/custom_ops/box_util.h"
1919
#include "tensorflow/core/framework/op_kernel.h"
2020
#include "tensorflow/core/framework/tensor.h"
2121
#include "tensorflow/core/framework/tensor_shape.h"

keras_cv/src/custom_ops/kernels/within_any_box_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ limitations under the License.
1515

1616
#define EIGEN_USE_THREADS
1717

18-
#include "keras_cv/custom_ops/box_util.h"
18+
#include "keras_cv/src/custom_ops/box_util.h"
1919
#include "tensorflow/core/framework/op_kernel.h"
2020
#include "tensorflow/core/framework/tensor.h"
2121
#include "tensorflow/core/framework/tensor_shape.h"

keras_cv/src/custom_ops/kernels/withinbox_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ limitations under the License.
1515

1616
#define EIGEN_USE_THREADS
1717

18-
#include "keras_cv/custom_ops/box_util.h"
18+
#include "keras_cv/src/custom_ops/box_util.h"
1919
#include "tensorflow/core/framework/op_kernel.h"
2020
#include "tensorflow/core/framework/tensor.h"
2121
#include "tensorflow/core/framework/tensor_shape.h"

0 commit comments

Comments
 (0)