Add RaggedTensorToTensor to flex delegate
The ragged_to_dense_util should be put in core/util instead of core/ops. Then it can be added to mobile_srcs_only_runtime. PiperOrigin-RevId: 320885158 Change-Id: I1083c2ac57e87b4e29a6e5fef2998d7b7a951133
This commit is contained in:
parent
7ae0db7a05
commit
525bfce301
tensorflow
@ -798,36 +798,7 @@ tf_gen_op_libs(
|
||||
"ragged_conversion_ops",
|
||||
"ragged_math_ops",
|
||||
],
|
||||
deps = [":ragged_to_dense_util"],
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "ragged_to_dense_util",
|
||||
srcs = [
|
||||
"ops/ragged_to_dense_util.cc",
|
||||
],
|
||||
hdrs = [
|
||||
"ops/ragged_to_dense_util.h",
|
||||
],
|
||||
deps = [
|
||||
":framework",
|
||||
":protos_all_cc",
|
||||
],
|
||||
)
|
||||
|
||||
tf_cc_test(
|
||||
name = "ragged_to_dense_util_test",
|
||||
srcs = [
|
||||
"ops/ragged_to_dense_util_test.cc",
|
||||
],
|
||||
deps = [
|
||||
":framework",
|
||||
":protos_all_cc",
|
||||
":ragged_to_dense_util",
|
||||
":test",
|
||||
":testlib",
|
||||
"@com_google_googletest//:gtest_main",
|
||||
],
|
||||
deps = ["//tensorflow/core/util:ragged_to_dense_util"],
|
||||
)
|
||||
|
||||
cc_library(
|
||||
|
@ -1508,7 +1508,7 @@ tf_kernel_library(
|
||||
"//tensorflow/core:framework_lite",
|
||||
"//tensorflow/core:lib",
|
||||
"//tensorflow/core:protos_all_cc",
|
||||
"//tensorflow/core:ragged_to_dense_util",
|
||||
"//tensorflow/core/util:ragged_to_dense_util",
|
||||
],
|
||||
)
|
||||
|
||||
@ -6909,6 +6909,7 @@ filegroup(
|
||||
"queue_op.cc",
|
||||
"queue_ops.cc",
|
||||
"ragged_range_op.cc",
|
||||
"ragged_tensor_to_tensor_op.cc",
|
||||
"random_op.cc",
|
||||
"random_op_cpu.h",
|
||||
"random_poisson_op.cc",
|
||||
|
@ -37,9 +37,9 @@ limitations under the License.
|
||||
#include "tensorflow/core/lib/bfloat16/bfloat16.h"
|
||||
#include "tensorflow/core/lib/core/errors.h"
|
||||
#include "tensorflow/core/lib/core/status.h"
|
||||
#include "tensorflow/core/ops/ragged_to_dense_util.h"
|
||||
#include "tensorflow/core/platform/types.h"
|
||||
#include "tensorflow/core/util/bcast.h"
|
||||
#include "tensorflow/core/util/ragged_to_dense_util.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
@ -15,7 +15,7 @@ limitations under the License.
|
||||
#include "tensorflow/core/framework/common_shape_fns.h"
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/framework/shape_inference.h"
|
||||
#include "tensorflow/core/ops/ragged_to_dense_util.h"
|
||||
#include "tensorflow/core/util/ragged_to_dense_util.h"
|
||||
|
||||
namespace tensorflow {
|
||||
|
||||
|
@ -93,6 +93,8 @@ filegroup(
|
||||
"port.h",
|
||||
"presized_cuckoo_map.h",
|
||||
"ptr_util.h",
|
||||
"ragged_to_dense_util.cc",
|
||||
"ragged_to_dense_util.h",
|
||||
"reffed_status_callback.h",
|
||||
"saved_tensor_slice_util.cc",
|
||||
"saved_tensor_slice_util.h",
|
||||
@ -368,6 +370,35 @@ cc_library(
|
||||
],
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "ragged_to_dense_util",
|
||||
srcs = [
|
||||
"ragged_to_dense_util.cc",
|
||||
],
|
||||
hdrs = [
|
||||
"ragged_to_dense_util.h",
|
||||
],
|
||||
deps = [
|
||||
"//tensorflow/core:framework",
|
||||
"//tensorflow/core:protos_all_cc",
|
||||
],
|
||||
)
|
||||
|
||||
tf_cc_test(
|
||||
name = "ragged_to_dense_util_test",
|
||||
srcs = [
|
||||
"ragged_to_dense_util_test.cc",
|
||||
],
|
||||
deps = [
|
||||
":ragged_to_dense_util",
|
||||
"//tensorflow/core:framework",
|
||||
"//tensorflow/core:protos_all_cc",
|
||||
"//tensorflow/core:test",
|
||||
"//tensorflow/core:testlib",
|
||||
"@com_google_googletest//:gtest_main",
|
||||
],
|
||||
)
|
||||
|
||||
tf_cc_test(
|
||||
name = "stats_calculator_test",
|
||||
srcs = ["stats_calculator_test.cc"],
|
||||
|
@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#include "tensorflow/core/ops/ragged_to_dense_util.h"
|
||||
#include "tensorflow/core/util/ragged_to_dense_util.h"
|
||||
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/framework/shape_inference.h"
|
@ -13,10 +13,9 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#include "tensorflow/core/ops/ragged_to_dense_util.h"
|
||||
#include "tensorflow/core/util/ragged_to_dense_util.h"
|
||||
|
||||
#include <gmock/gmock.h>
|
||||
#include <gtest/gtest.h>
|
||||
#include "tensorflow/core/framework/op.h"
|
||||
#include "tensorflow/core/framework/shape_inference.h"
|
||||
#include "tensorflow/core/framework/tensor_shape.h"
|
@ -299,6 +299,7 @@ bool IsAllowlistedFlexOp(const std::string& tensorflow_op_name) {
|
||||
"RFFT2D",
|
||||
"RFFT3D",
|
||||
"RaggedRange",
|
||||
"RaggedTensorToTensor",
|
||||
"RandomGamma",
|
||||
"RandomStandardNormal",
|
||||
"RandomUniform",
|
||||
|
Loading…
Reference in New Issue
Block a user