Allow ComplexAbs Op on mobile platforms (#18113)
authorIlya Polenov <daioptych@gmail.com>
Mon, 9 Apr 2018 15:45:48 +0000 (15:45 +0000)
committerRasmus Munk Larsen <rmlarsen@google.com>
Mon, 9 Apr 2018 15:45:48 +0000 (08:45 -0700)
Seems like it was disabled long time ago before open-sourcing Tensorflow.
I think disabling it is no longer necessary.
Works now on Android. Could anyone check on iOS?

Somewhat related issue: #11804

tensorflow/core/kernels/cwise_op_abs.cc

index 1466f24..1920c54 100644 (file)
@@ -18,9 +18,7 @@ limitations under the License.
 namespace tensorflow {
 REGISTER5(UnaryOp, CPU, "Abs", functor::abs, float, Eigen::half, double, int32,
           int64);
-#if !defined(IS_MOBILE_PLATFORM)
 REGISTER2(UnaryOp, CPU, "ComplexAbs", functor::abs, complex64, complex128);
-#endif
 
 #if GOOGLE_CUDA
 REGISTER4(UnaryOp, GPU, "Abs", functor::abs, float, Eigen::half, double, int64);