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

Enable the generated version of tan kernel.

Browse Source 
PiperOrigin-RevId: 354337603
Change-Id: I50461369c85276ccf176f9fd0ff5a7597a1aae31
This commit is contained in:
Stephan Herhut 2021-01-28 10:13:42 -08:00 committed by TensorFlower Gardener
parent 9d7088c351
commit 85cc0cc7f4
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/cwise_op_gpu_tan.cu.cc
View File

@ -19,7 +19,9 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
namespace functor { namespace functor {
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
DEFINE_UNARY3(tan, Eigen::half, float, double); DEFINE_UNARY3(tan, Eigen::half, float, double);
#endif
} // namespace functor } // namespace functor
} // namespace tensorflow } // namespace tensorflow

3
tensorflow/core/kernels/cwise_op_tan.cc
View File

@ -20,8 +20,7 @@ REGISTER6(UnaryOp, CPU, "Tan", functor::tan, Eigen::half, bfloat16, float,
double, complex64, complex128); double, complex64, complex128);
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \ #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
!defined(MLIR_GENERATED_EXPERIMENTAL_GPU_KERNELS_ENABLED)
REGISTER3(UnaryOp, GPU, "Tan", functor::tan, Eigen::half, float, double); REGISTER3(UnaryOp, GPU, "Tan", functor::tan, Eigen::half, float, double);
#endif #endif
#endif #endif

2
tensorflow/core/kernels/mlir_generated/BUILD
View File

@ -41,6 +41,7 @@ filegroup(
"gpu_op_abs.cc", "gpu_op_abs.cc",
"gpu_op_atan.cc", "gpu_op_atan.cc",
"gpu_op_atanh.cc", "gpu_op_atanh.cc",
"gpu_op_tan.cc",
"gpu_op_tanh.cc", "gpu_op_tanh.cc",
], ],
compatible_with = get_compatible_with_cloud(), compatible_with = get_compatible_with_cloud(),
@ -80,7 +81,6 @@ filegroup(
"gpu_op_sinh.cc", "gpu_op_sinh.cc",
"gpu_op_sqrt.cc", "gpu_op_sqrt.cc",
"gpu_op_square.cc", "gpu_op_square.cc",
"gpu_op_tan.cc",
], ],
compatible_with = get_compatible_with_cloud(), compatible_with = get_compatible_with_cloud(),
) )