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

Add set kernels to Flex

Browse Source 
closes #45199

PiperOrigin-RevId: 351490161
Change-Id: I6ce7e52a22c16d34ae007e448f5ebd6482e3556a
This commit is contained in:
Jaesung Chung 2021-01-12 18:27:36 -08:00 committed by TensorFlower Gardener
parent 3654281b02
commit 103ff003da
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/kernels/BUILD
View File

@ -6195,6 +6195,7 @@ filegroup(
"segment_reduction_ops_impl_4.cc",
"segment_reduction_ops_impl_5.cc",
"session_ops.cc",
"set_kernels.cc",
"softplus_op.cc",
"softsign_op.cc",
"spacetobatch_functor.cc",

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

@ -146,6 +146,8 @@ const std::set<std::string>& GetFlexAllowlist() {
"DeepCopy",
"DeleteSessionTensor",
"DenseBincount",
"DenseToDenseSetOperation",
"DenseToSparseSetOperation",
"DepthToSpace",
"DepthwiseConv2dNative",
"Dequantize",
@ -512,6 +514,7 @@ const std::set<std::string>& GetFlexAllowlist() {
"SparseSegmentSum",
"SparseSegmentSumWithNumSegments",
"SparseToDense",
"SparseToSparseSetOperation",
"Split",
"SplitV",
"Sqrt",