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

Formatting

Browse Source
This commit is contained in:
Eugene Kuznetsov 2020-02-01 01:48:46 -08:00
parent 3693256ba0
commit fb91bf6458
1 changed files with 10 additions and 20 deletions

30
tensorflow/core/kernels/training_ops.cc
View File

@ -670,8 +670,7 @@ namespace functor {
DECLARE_GPU_SPEC(Eigen::half);
DECLARE_GPU_SPEC(float);
DECLARE_GPU_SPEC(double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
DECLARE_GPU_SPEC(complex64);
DECLARE_GPU_SPEC(complex128);
@ -683,8 +682,7 @@ DECLARE_GPU_SPEC(complex128);
REGISTER_KERNELS(GPU, Eigen::half);
REGISTER_KERNELS(GPU, float);
REGISTER_KERNELS(GPU, double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
REGISTER_KERNELS(GPU, complex64);
REGISTER_KERNELS(GPU, complex128);
@ -851,8 +849,7 @@ namespace functor {
DECLARE_GPU_SPEC(Eigen::half);
DECLARE_GPU_SPEC(float);
DECLARE_GPU_SPEC(double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
DECLARE_GPU_SPEC(complex64);
DECLARE_GPU_SPEC(complex128);
@ -864,8 +861,7 @@ DECLARE_GPU_SPEC(complex128);
REGISTER_KERNELS(GPU, Eigen::half);
REGISTER_KERNELS(GPU, float);
REGISTER_KERNELS(GPU, double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
REGISTER_KERNELS(GPU, complex64);
REGISTER_KERNELS(GPU, complex128);
@ -1344,8 +1340,7 @@ namespace functor {
DECLARE_GPU_SPEC(Eigen::half);
DECLARE_GPU_SPEC(float);
DECLARE_GPU_SPEC(double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
DECLARE_GPU_SPEC(complex64);
DECLARE_GPU_SPEC(complex128);
@ -1357,8 +1352,7 @@ DECLARE_GPU_SPEC(complex128);
REGISTER_KERNELS(GPU, Eigen::half);
REGISTER_KERNELS(GPU, float);
REGISTER_KERNELS(GPU, double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
REGISTER_KERNELS(GPU, complex64);
REGISTER_KERNELS(GPU, complex128);
@ -1462,8 +1456,7 @@ namespace functor {
DECLARE_GPU_SPEC(Eigen::half);
DECLARE_GPU_SPEC(float);
DECLARE_GPU_SPEC(double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
DECLARE_GPU_SPEC(complex64);
DECLARE_GPU_SPEC(complex128);
@ -1475,8 +1468,7 @@ DECLARE_GPU_SPEC(complex128);
REGISTER_KERNELS(GPU, Eigen::half);
REGISTER_KERNELS(GPU, float);
REGISTER_KERNELS(GPU, double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
REGISTER_KERNELS(GPU, complex64);
REGISTER_KERNELS(GPU, complex128);
@ -4175,8 +4167,7 @@ namespace functor {
DECLARE_GPU_SPEC(Eigen::half);
DECLARE_GPU_SPEC(float);
DECLARE_GPU_SPEC(double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
DECLARE_GPU_SPEC(complex64);
DECLARE_GPU_SPEC(complex128);
@ -4188,8 +4179,7 @@ DECLARE_GPU_SPEC(complex128);
REGISTER_KERNELS(GPU, Eigen::half);
REGISTER_KERNELS(GPU, float);
REGISTER_KERNELS(GPU, double);
#if !defined( \
TENSORFLOW_USE_NVCC) // TODO(b/143684500): Eigen to support complex sqrt
#ifndef TENSORFLOW_USE_NVCC // TODO(b/143684500): Eigen to support complex sqrt
#ifndef PLATFORM_WINDOWS
REGISTER_KERNELS(GPU, complex64);
REGISTER_KERNELS(GPU, complex128);