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
@ -798,36 +798,7 @@ tf_gen_op_libs(
|
|||||||
"ragged_conversion_ops",
|
"ragged_conversion_ops",
|
||||||
"ragged_math_ops",
|
"ragged_math_ops",
|
||||||
],
|
],
|
||||||
deps = [":ragged_to_dense_util"],
|
deps = ["//tensorflow/core/util: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",
|
|
||||||
],
|
|
||||||
)
|
)
|
||||||
|
|
||||||
cc_library(
|
cc_library(
|
||||||
|
@ -1508,7 +1508,7 @@ tf_kernel_library(
|
|||||||
"//tensorflow/core:framework_lite",
|
"//tensorflow/core:framework_lite",
|
||||||
"//tensorflow/core:lib",
|
"//tensorflow/core:lib",
|
||||||
"//tensorflow/core:protos_all_cc",
|
"//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_op.cc",
|
||||||
"queue_ops.cc",
|
"queue_ops.cc",
|
||||||
"ragged_range_op.cc",
|
"ragged_range_op.cc",
|
||||||
|
"ragged_tensor_to_tensor_op.cc",
|
||||||
"random_op.cc",
|
"random_op.cc",
|
||||||
"random_op_cpu.h",
|
"random_op_cpu.h",
|
||||||
"random_poisson_op.cc",
|
"random_poisson_op.cc",
|
||||||
|
@ -37,9 +37,9 @@ limitations under the License.
|
|||||||
#include "tensorflow/core/lib/bfloat16/bfloat16.h"
|
#include "tensorflow/core/lib/bfloat16/bfloat16.h"
|
||||||
#include "tensorflow/core/lib/core/errors.h"
|
#include "tensorflow/core/lib/core/errors.h"
|
||||||
#include "tensorflow/core/lib/core/status.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/platform/types.h"
|
||||||
#include "tensorflow/core/util/bcast.h"
|
#include "tensorflow/core/util/bcast.h"
|
||||||
|
#include "tensorflow/core/util/ragged_to_dense_util.h"
|
||||||
|
|
||||||
namespace tensorflow {
|
namespace tensorflow {
|
||||||
|
|
||||||
|
@ -15,7 +15,7 @@ limitations under the License.
|
|||||||
#include "tensorflow/core/framework/common_shape_fns.h"
|
#include "tensorflow/core/framework/common_shape_fns.h"
|
||||||
#include "tensorflow/core/framework/op.h"
|
#include "tensorflow/core/framework/op.h"
|
||||||
#include "tensorflow/core/framework/shape_inference.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 {
|
namespace tensorflow {
|
||||||
|
|
||||||
|
@ -93,6 +93,8 @@ filegroup(
|
|||||||
"port.h",
|
"port.h",
|
||||||
"presized_cuckoo_map.h",
|
"presized_cuckoo_map.h",
|
||||||
"ptr_util.h",
|
"ptr_util.h",
|
||||||
|
"ragged_to_dense_util.cc",
|
||||||
|
"ragged_to_dense_util.h",
|
||||||
"reffed_status_callback.h",
|
"reffed_status_callback.h",
|
||||||
"saved_tensor_slice_util.cc",
|
"saved_tensor_slice_util.cc",
|
||||||
"saved_tensor_slice_util.h",
|
"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(
|
tf_cc_test(
|
||||||
name = "stats_calculator_test",
|
name = "stats_calculator_test",
|
||||||
srcs = ["stats_calculator_test.cc"],
|
srcs = ["stats_calculator_test.cc"],
|
||||||
|
@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
|
|||||||
limitations under the License.
|
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/op.h"
|
||||||
#include "tensorflow/core/framework/shape_inference.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.
|
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 <gmock/gmock.h>
|
||||||
#include <gtest/gtest.h>
|
|
||||||
#include "tensorflow/core/framework/op.h"
|
#include "tensorflow/core/framework/op.h"
|
||||||
#include "tensorflow/core/framework/shape_inference.h"
|
#include "tensorflow/core/framework/shape_inference.h"
|
||||||
#include "tensorflow/core/framework/tensor_shape.h"
|
#include "tensorflow/core/framework/tensor_shape.h"
|
@ -299,6 +299,7 @@ bool IsAllowlistedFlexOp(const std::string& tensorflow_op_name) {
|
|||||||
"RFFT2D",
|
"RFFT2D",
|
||||||
"RFFT3D",
|
"RFFT3D",
|
||||||
"RaggedRange",
|
"RaggedRange",
|
||||||
|
"RaggedTensorToTensor",
|
||||||
"RandomGamma",
|
"RandomGamma",
|
||||||
"RandomStandardNormal",
|
"RandomStandardNormal",
|
||||||
"RandomUniform",
|
"RandomUniform",
|
||||||
|
Loading…
x
Reference in New Issue
Block a user