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", "spacetodepth_op.h",
"spectrogram.h", "spectrogram.h",
"stateless_random_ops.h", "stateless_random_ops.h",
"stateless_random_ops_v2.h",
"string_util.h", "string_util.h",
"string_to_hash_bucket_op.h", "string_to_hash_bucket_op.h",
"tensor_array.h", "tensor_array.h",
@ -6231,6 +6232,7 @@ filegroup(
"stack.h", "stack.h",
"stack_ops.cc", "stack_ops.cc",
"stateless_random_ops.cc", "stateless_random_ops.cc",
"stateless_random_ops_v2.cc",
"string_join_op.cc", "string_join_op.cc",
"string_lower_op.cc", "string_lower_op.cc",
"string_util.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", "StackV2",
"StatelessMultinomial", "StatelessMultinomial",
"StatelessRandomGammaV2", "StatelessRandomGammaV2",
"StatelessRandomGetKeyCounterAlg",
"StatelessRandomNormal", "StatelessRandomNormal",
"StatelessRandomNormalV2",
"StatelessRandomPoisson", "StatelessRandomPoisson",
"StatelessRandomUniform", "StatelessRandomUniform",
"StatelessRandomUniformFullInt", "StatelessRandomUniformFullInt",
"StatelessRandomUniformFullIntV2",
"StatelessRandomUniformInt", "StatelessRandomUniformInt",
"StatelessRandomUniformIntV2",
"StatelessRandomUniformV2",
"StatelessSampleDistortedBoundingBox", "StatelessSampleDistortedBoundingBox",
"StatelessTruncatedNormal", "StatelessTruncatedNormal",
"StatelessTruncatedNormalV2",
"StaticRegexReplace", "StaticRegexReplace",
"StopGradient", "StopGradient",
"StridedSlice", "StridedSlice",