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

Add stateless random ops v2 to Flex delegate

Browse Source 
PiperOrigin-RevId: 342631399
Change-Id: I5dbff425fcfd6d8500f60c87836081ab0c74c3f8
This commit is contained in:
Jaesung Chung 2020-11-16 07:29:59 -08:00 committed by TensorFlower Gardener
parent 229aeadda8
commit b63cb8466b
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/BUILD
View File

@ -5974,6 +5974,7 @@ filegroup(
"spacetodepth_op.h",
"spectrogram.h",
"stateless_random_ops.h",
"stateless_random_ops_v2.h",
"string_util.h",
"string_to_hash_bucket_op.h",
"tensor_array.h",
@ -6231,6 +6232,7 @@ filegroup(
"stack.h",
"stack_ops.cc",
"stateless_random_ops.cc",
"stateless_random_ops_v2.cc",
"string_join_op.cc",
"string_lower_op.cc",
"string_util.cc",

6
tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
View File

@ -525,13 +525,19 @@ const std::set<std::string>& GetFlexAllowlist() {
"StackV2",
"StatelessMultinomial",
"StatelessRandomGammaV2",
"StatelessRandomGetKeyCounterAlg",
"StatelessRandomNormal",
"StatelessRandomNormalV2",
"StatelessRandomPoisson",
"StatelessRandomUniform",
"StatelessRandomUniformFullInt",
"StatelessRandomUniformFullIntV2",
"StatelessRandomUniformInt",
"StatelessRandomUniformIntV2",
"StatelessRandomUniformV2",
"StatelessSampleDistortedBoundingBox",
"StatelessTruncatedNormal",
"StatelessTruncatedNormalV2",
"StaticRegexReplace",
"StopGradient",
"StridedSlice",