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

Enable the generated version of acos and acosh.

Browse Source 
PiperOrigin-RevId: 355633110
Change-Id: Ib08bd2d465000cb314a51718106886e79d187d6e
This commit is contained in:
Stephan Herhut 2021-02-04 08:33:42 -08:00 committed by TensorFlower Gardener
parent 8771e5061f
commit 15c5c8362e
5 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,8 +19,7 @@ namespace tensorflow {
REGISTER2(UnaryOp, CPU, "Acos", functor::acos, float, double);
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
!defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
REGISTER2(UnaryOp, GPU, "Acos", functor::acos, float, double);
#endif
#endif

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

@ -22,8 +22,7 @@ REGISTER4(UnaryOp, CPU, "Acosh", functor::acosh, float, double, complex64,
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
!defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
REGISTER2(UnaryOp, GPU, "Acosh", functor::acosh, float, double);
#endif
#endif

3
tensorflow/core/kernels/cwise_op_gpu_acos.cu.cc
View File

@ -19,8 +19,7 @@ limitations under the License.
namespace tensorflow {
namespace functor {
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
!defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
DEFINE_UNARY2(acos, float, double);
#endif
} // namespace functor

3
tensorflow/core/kernels/cwise_op_gpu_acosh.cu.cc
View File

@ -20,8 +20,7 @@ limitations under the License.
namespace tensorflow {
namespace functor {
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED) || \
!defined(MLIR_GENERATED_EXPERIMENTAL_KERNELS_ENABLED)
#if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
DEFINE_UNARY2(acosh, float, double);
#endif
} // namespace functor

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

@ -39,6 +39,8 @@ filegroup(
name = "enabled_unary_kernel_srcs",
srcs = [
"gpu_op_abs.cc",
"gpu_op_acos.cc",
"gpu_op_acosh.cc",
"gpu_op_atan.cc",
"gpu_op_atanh.cc",
"gpu_op_ceil.cc",
@ -72,8 +74,6 @@ filegroup(
filegroup(
name = "experimental_unary_kernel_srcs",
srcs = [
"gpu_op_acos.cc",
"gpu_op_acosh.cc",
"gpu_op_angle.cc",
"gpu_op_asin.cc",
"gpu_op_asinh.cc",