Replace sparse_fill_empty_rows_op with the tf native interface

tensorflow · Nov 15, 2023 · 5ab4cec · 5ab4cec
1 parent de53ad0
commit 5ab4cec
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 431 deletions.
diff --git a/tensorflow_recommenders_addons/dynamic_embedding/core/BUILD b/tensorflow_recommenders_addons/dynamic_embedding/core/BUILD
@@ -76,8 +76,6 @@ custom_op_library(
         "kernels/segment_reduction_ops.h",
         "kernels/segment_reduction_ops_impl.cc",
         "kernels/segment_reduction_ops_impl.h",
-        "kernels/sparse_fill_empty_rows_op.cc",
-        "kernels/sparse_fill_empty_rows_op.h",
         "ops/math_ops.cc",
         "utils/utils.h",
     ],
@@ -89,8 +87,6 @@ custom_op_library(
     cuda_srcs = [
         "kernels/segment_reduction_ops.h",
         "kernels/segment_reduction_ops_gpu.cu.cc",
-        "kernels/sparse_fill_empty_rows_op.h",
-        "kernels/sparse_fill_empty_rows_op.cu.cc",
     ],
 )
 

diff --git a/tensorflow_recommenders_addons/dynamic_embedding/core/kernels/sparse_fill_empty_rows_op.cc b/tensorflow_recommenders_addons/dynamic_embedding/core/kernels/sparse_fill_empty_rows_op.cc
diff --git a/...orflow_recommenders_addons/dynamic_embedding/core/kernels/sparse_fill_empty_rows_op.cu.cc b/...orflow_recommenders_addons/dynamic_embedding/core/kernels/sparse_fill_empty_rows_op.cu.cc
diff --git a/tensorflow_recommenders_addons/dynamic_embedding/core/kernels/sparse_fill_empty_rows_op.h b/tensorflow_recommenders_addons/dynamic_embedding/core/kernels/sparse_fill_empty_rows_op.h