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

Add Conj op to flex's whitelist

Browse Source 
PiperOrigin-RevId: 306943019
Change-Id: I24c1dd2ac20824506dcbe241ac71cdc9e552b78e
This commit is contained in:
Jaesung Chung 2020-04-16 16:22:06 -07:00 committed by TensorFlower Gardener
parent d5743e269c
commit 1bbc26d375
2 changed files with 2 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

@ -6654,6 +6654,7 @@ filegroup(
"cwise_op_bitwise_and.cc", "cwise_op_bitwise_and.cc",
"cwise_op_bitwise_or.cc", "cwise_op_bitwise_or.cc",
"cwise_op_bitwise_xor.cc", "cwise_op_bitwise_xor.cc",
"cwise_op_conj.cc",
"cwise_op_cos.cc", "cwise_op_cos.cc",
"cwise_op_cosh.cc", "cwise_op_cosh.cc",
"cwise_op_div.cc", "cwise_op_div.cc",

1
tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
View File

@ -76,6 +76,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
"Concat", "Concat",
"ConcatOffset", "ConcatOffset",
"ConcatV2", "ConcatV2",
"Conj",
"ConjugateTranspose", "ConjugateTranspose",
"Const", "Const",
"ControlTrigger", "ControlTrigger",