experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Support StringToHashBucketFast, Real, Imag ops in TFLite flex delegate

Browse Source 
PiperOrigin-RevId: 314647136
Change-Id: I3e889044e961e22f26c63d00bd3a90083bb15330
This commit is contained in:
Thai Nguyen 2020-06-03 18:55:32 -07:00 committed by TensorFlower Gardener
parent 10d6931f93
commit 73edd2fb7e
2 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/BUILD
View File

@ -6671,6 +6671,7 @@ filegroup(
"spectrogram.h", "spectrogram.h",
"stateless_random_ops.h", "stateless_random_ops.h",
"string_util.h", "string_util.h",
"string_to_hash_bucket_op.h",
"tensor_array.h", "tensor_array.h",
"tile_functor.h", "tile_functor.h",
"tile_ops_cpu_impl.h", "tile_ops_cpu_impl.h",
@ -6735,6 +6736,7 @@ filegroup(
"cwise_op_floor_mod.cc", "cwise_op_floor_mod.cc",
"cwise_op_greater.cc", "cwise_op_greater.cc",
"cwise_op_greater_equal.cc", "cwise_op_greater_equal.cc",
"cwise_op_imag.cc",
"cwise_op_invert.cc", "cwise_op_invert.cc",
"cwise_op_isfinite.cc", "cwise_op_isfinite.cc",
"cwise_op_isnan.cc", "cwise_op_isnan.cc",
@ -6751,6 +6753,7 @@ filegroup(
"cwise_op_mul_2.cc", "cwise_op_mul_2.cc",
"cwise_op_neg.cc", "cwise_op_neg.cc",
"cwise_op_pow.cc", "cwise_op_pow.cc",
"cwise_op_real.cc",
"cwise_op_reciprocal.cc", "cwise_op_reciprocal.cc",
"cwise_op_right_shift.cc", "cwise_op_right_shift.cc",
"cwise_op_round.cc", "cwise_op_round.cc",
@ -6902,6 +6905,7 @@ filegroup(
"stateless_random_ops.cc", "stateless_random_ops.cc",
"string_join_op.cc", "string_join_op.cc",
"string_util.cc", "string_util.cc",
"string_to_hash_bucket_op.cc",
"summary_op.cc", "summary_op.cc",
"tensor_array.cc", "tensor_array.cc",
"tensor_array_ops.cc", "tensor_array_ops.cc",

3
tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
View File

@ -163,6 +163,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
"IRFFT", "IRFFT",
"IRFFT2D", "IRFFT2D",
"IRFFT3D", "IRFFT3D",
"Imag",
"ImmutableConst", "ImmutableConst",
"InTopK", "InTopK",
"InTopKV2", "InTopKV2",
@ -283,6 +284,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
"RandomUniformInt", "RandomUniformInt",
"Range", "Range",
"Rank", "Rank",
"Real",
"RealDiv", "RealDiv",
"Reciprocal", "Reciprocal",
"ReciprocalGrad", "ReciprocalGrad",
@ -423,6 +425,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
"StridedSliceAssign", "StridedSliceAssign",
"StridedSliceGrad", "StridedSliceGrad",
"StringJoin", "StringJoin",
"StringToHashBucketFast",
"Sub", "Sub",
"Sum", "Sum",
"Switch", "Switch",