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

Add several trigonometric kernels to TFMobile

Browse Source 
This includes:
 * cos
 * cosh
 * sin
 * sinh
 * tan

Note that tanh was already included, and TFLite already supports sin.

PiperOrigin-RevId: 219204972
This commit is contained in:
Jared Duke 2018-10-29 15:32:46 -07:00 committed by TensorFlower Gardener
parent d617d45f31
commit 72c4105a28
3 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/contrib/lite/build_def.bzl
View File

@ -280,7 +280,7 @@ def generated_test_models():
"rsqrt",
"shape",
"sigmoid",
#"sin", # b/118502200
"sin",
"slice",
"softmax",
"space_to_batch_nd",

5
tensorflow/contrib/makefile/tf_op_files.txt
View File

@ -52,6 +52,8 @@ tensorflow/core/kernels/cwise_op_add_2.cc
tensorflow/core/kernels/cwise_op_bitwise_and.cc
tensorflow/core/kernels/cwise_op_bitwise_or.cc
tensorflow/core/kernels/cwise_op_bitwise_xor.cc
tensorflow/core/kernels/cwise_op_cos.cc
tensorflow/core/kernels/cwise_op_cosh.cc
tensorflow/core/kernels/cwise_op_div.cc
tensorflow/core/kernels/cwise_op_equal_to_1.cc
tensorflow/core/kernels/cwise_op_equal_to_2.cc
@ -86,10 +88,13 @@ tensorflow/core/kernels/cwise_op_rsqrt.cc
tensorflow/core/kernels/cwise_op_select.cc
tensorflow/core/kernels/cwise_op_sigmoid.cc
tensorflow/core/kernels/cwise_op_sign.cc
tensorflow/core/kernels/cwise_op_sin.cc
tensorflow/core/kernels/cwise_op_sinh.cc
tensorflow/core/kernels/cwise_op_sqrt.cc
tensorflow/core/kernels/cwise_op_square.cc
tensorflow/core/kernels/cwise_op_squared_difference.cc
tensorflow/core/kernels/cwise_op_sub.cc
tensorflow/core/kernels/cwise_op_tan.cc
tensorflow/core/kernels/cwise_op_tanh.cc
tensorflow/core/kernels/cwise_op_xdivy.cc
tensorflow/core/kernels/cwise_op_xlogy.cc

5
tensorflow/core/kernels/BUILD
View File

@ -5356,6 +5356,8 @@ filegroup(
"cwise_op_bitwise_and.cc",
"cwise_op_bitwise_or.cc",
"cwise_op_bitwise_xor.cc",
"cwise_op_cos.cc",
"cwise_op_cosh.cc",
"cwise_op_div.cc",
"cwise_op_equal_to_1.cc",
"cwise_op_equal_to_2.cc",
@ -5390,10 +5392,13 @@ filegroup(
"cwise_op_select.cc",
"cwise_op_sigmoid.cc",
"cwise_op_sign.cc",
"cwise_op_sin.cc",
"cwise_op_sinh.cc",
"cwise_op_sqrt.cc",
"cwise_op_square.cc",
"cwise_op_squared_difference.cc",
"cwise_op_sub.cc",
"cwise_op_tan.cc",
"cwise_op_tanh.cc",
"cwise_op_xlogy.cc",
"cwise_op_xdivy.cc",