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
tensorflow
core/kernels
BUILD
lite
delegates/flex
allowlisted_flex_ops.cc
g3doc/guide
op_select_allowlist.md

1
tensorflow/core/kernels/BUILD
View File

@ -6079,6 +6079,7 @@ filegroup(
"decode_wav_op.cc", "decode_wav_op.cc",
"deep_conv2d.cc", "deep_conv2d.cc",
"deep_conv2d.h", "deep_conv2d.h",
"depthwise_conv_grad_op.cc",
"depthwise_conv_op.cc", "depthwise_conv_op.cc",
"dynamic_partition_op.cc", "dynamic_partition_op.cc",
"eigen_contraction_kernel.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", "DenseToSparseSetOperation",
"DepthToSpace", "DepthToSpace",
"DepthwiseConv2dNative", "DepthwiseConv2dNative",
"DepthwiseConv2dNativeBackpropFilter",
"DepthwiseConv2dNativeBackpropInput",
"Dequantize", "Dequantize",
"DestroyResourceOp", "DestroyResourceOp",
"DestroyTemporaryVariable", "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.DenseToSparseSetOperation`
* `raw_ops.DepthToSpace` * `raw_ops.DepthToSpace`
* `raw_ops.DepthwiseConv2dNative` * `raw_ops.DepthwiseConv2dNative`
* `raw_ops.DepthwiseConv2dNativeBackpropFilter`
* `raw_ops.DepthwiseConv2dNativeBackpropInput`
* `raw_ops.Dequantize` * `raw_ops.Dequantize`
* `raw_ops.DestroyResourceOp` * `raw_ops.DestroyResourceOp`
* `raw_ops.DestroyTemporaryVariable` * `raw_ops.DestroyTemporaryVariable`