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

Enable the generated version of asin and asinh.

Browse Source 
PiperOrigin-RevId: 355637657
Change-Id: I2ff3e4b9396cb0f3a811c01b3c34cdf89519650d
This commit is contained in:
Stephan Herhut 2021-02-04 08:58:30 -08:00 committed by TensorFlower Gardener
parent 15c5c8362e
commit 4b2f11c9f1
5 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,8 +19,7 @@ namespace tensorflow {
REGISTER2(UnaryOp, CPU, "Asin", functor::asin, 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, "Asin", functor::asin, float, double);
#endif
#endif

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

@ -21,8 +21,7 @@ REGISTER4(UnaryOp, CPU, "Asinh", functor::asinh, float, double, complex64,
complex128);
#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, "Asinh", functor::asinh, float, double);
#endif
#endif

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

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

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

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

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

@ -41,6 +41,8 @@ filegroup(
"gpu_op_abs.cc",
"gpu_op_acos.cc",
"gpu_op_acosh.cc",
"gpu_op_asin.cc",
"gpu_op_asinh.cc",
"gpu_op_atan.cc",
"gpu_op_atanh.cc",
"gpu_op_ceil.cc",
@ -75,8 +77,6 @@ filegroup(
name = "experimental_unary_kernel_srcs",
srcs = [
"gpu_op_angle.cc",
"gpu_op_asin.cc",
"gpu_op_asinh.cc",
"gpu_op_digamma.cc",
"gpu_op_exp.cc",
"gpu_op_expm1.cc",