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:
Thai Nguyen 2020-07-12 20:28:00 -07:00 committed by TensorFlower Gardener
parent 7ae0db7a05
commit 525bfce301
9 changed files with 39 additions and 36 deletions

View File

@ -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(

View File

@ -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",

View File

@ -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 {

View File

@ -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 {

View File

@ -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"],

View File

@ -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"

View File

@ -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"

View File

@ -299,6 +299,7 @@ bool IsAllowlistedFlexOp(const std::string& tensorflow_op_name) {
"RFFT2D",
"RFFT3D",
"RaggedRange",
"RaggedTensorToTensor",
"RandomGamma",
"RandomStandardNormal",
"RandomUniform",