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

[lite] Add DepthwiseConv2dNativeBackpropInput and DepthwiseConv2dNativeBackpropFilter to SELECT_TF allow list.

Browse Source 
PiperOrigin-RevId: 355648712
Change-Id: Id8a9765eca42b3beb3ef58139b789b08fe26ba7c
This commit is contained in:
Karim Nosir 2021-02-04 09:52:35 -08:00 committed by TensorFlower Gardener
parent 492b5d5987
commit ad1df4bc4f
3 changed files with 5 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

@ -6079,6 +6079,7 @@ filegroup(
"decode_wav_op.cc",
"deep_conv2d.cc",
"deep_conv2d.h",
"depthwise_conv_grad_op.cc",
"depthwise_conv_op.cc",
"dynamic_partition_op.cc",
"eigen_contraction_kernel.cc",

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

@ -159,6 +159,8 @@ const std::set<std::string>& GetFlexAllowlist() {
"DenseToSparseSetOperation",
"DepthToSpace",
"DepthwiseConv2dNative",
"DepthwiseConv2dNativeBackpropFilter",
"DepthwiseConv2dNativeBackpropInput",
"Dequantize",
"DestroyResourceOp",
"DestroyTemporaryVariable",

2
tensorflow/lite/g3doc/guide/op_select_allowlist.md
View File

@ -138,6 +138,8 @@ supported by TensorFlow Lite runtime with the Select TensorFlow Ops feature.
* `raw_ops.DenseToSparseSetOperation`
* `raw_ops.DepthToSpace`
* `raw_ops.DepthwiseConv2dNative`
* `raw_ops.DepthwiseConv2dNativeBackpropFilter`
* `raw_ops.DepthwiseConv2dNativeBackpropInput`
* `raw_ops.Dequantize`
* `raw_ops.DestroyResourceOp`
* `raw_ops.DestroyTemporaryVariable`