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

Cleanup training_ops to reduce code redudancy.

Browse Source 
PiperOrigin-RevId: 168280069
This commit is contained in:
A. Unique TensorFlower 2017-09-11 13:55:37 -07:00 committed by TensorFlower Gardener
parent 123fb01eeb
commit c3d19e40ae
1 changed files with 0 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -600,9 +600,6 @@ class ApplyAdadeltaOp : public OpKernel {
}
};
using CPUDevice = Eigen::ThreadPoolDevice;
using GPUDevice = Eigen::GpuDevice;
#define REGISTER_KERNELS(D, T) \
REGISTER_KERNEL_BUILDER( \
Name("ApplyAdadelta").Device(DEVICE_##D).TypeConstraint<T>("T"), \
@ -1061,9 +1058,6 @@ class ApplyAdagradOp : public OpKernel {
bool use_exclusive_lock_;
};
using CPUDevice = Eigen::ThreadPoolDevice;
using GPUDevice = Eigen::GpuDevice;
#define REGISTER_KERNELS(D, T) \
REGISTER_KERNEL_BUILDER( \
Name("ApplyAdagrad").Device(DEVICE_##D).TypeConstraint<T>("T"), \
@ -1171,9 +1165,6 @@ class ApplyProximalAdagradOp : public OpKernel {
bool use_exclusive_lock_;
};
using CPUDevice = Eigen::ThreadPoolDevice;
using GPUDevice = Eigen::GpuDevice;
#define REGISTER_KERNELS(D, T) \
REGISTER_KERNEL_BUILDER( \
Name("ApplyProximalAdagrad").Device(DEVICE_##D).TypeConstraint<T>("T"), \
@ -1600,9 +1591,6 @@ class ApplyAdagradDAOp : public OpKernel {
bool use_exclusive_lock_;
};
using CPUDevice = Eigen::ThreadPoolDevice;
using GPUDevice = Eigen::GpuDevice;
#define REGISTER_KERNELS(D, T) \
REGISTER_KERNEL_BUILDER( \
Name("ApplyAdagradDA").Device(DEVICE_##D).TypeConstraint<T>("T"), \
@ -1925,9 +1913,6 @@ class ApplyFtrlOp : public OpKernel {
bool use_exclusive_lock_;
};
using CPUDevice = Eigen::ThreadPoolDevice;
using GPUDevice = Eigen::GpuDevice;
#define REGISTER_KERNELS(D, T) \
REGISTER_KERNEL_BUILDER( \
Name("ApplyFtrl").Device(DEVICE_##D).TypeConstraint<T>("T"), \
@ -2301,9 +2286,6 @@ class ApplyMomentumOp : public OpKernel {
bool use_nesterov_;
};
using CPUDevice = Eigen::ThreadPoolDevice;
using GPUDevice = Eigen::GpuDevice;
#define REGISTER_KERNELS(D, T) \
REGISTER_KERNEL_BUILDER( \
Name("ApplyMomentum").Device(DEVICE_##D).TypeConstraint<T>("T"), \
@ -2669,9 +2651,6 @@ class ApplyAdamOp<SYCLDevice, T> : public OpKernel {
};
#endif // TENSORFLOW_USE_SYCL
using CPUDevice = Eigen::ThreadPoolDevice;
using GPUDevice = Eigen::GpuDevice;
#define REGISTER_KERNELS(D, T) \
REGISTER_KERNEL_BUILDER( \
Name("ApplyAdam").Device(DEVICE_##D).TypeConstraint<T>("T"), \
@ -2900,9 +2879,6 @@ class ApplyCenteredRMSPropOp : public OpKernel {
bool use_exclusive_lock_;
};
using CPUDevice = Eigen::ThreadPoolDevice;
using GPUDevice = Eigen::GpuDevice;
#define REGISTER_KERNELS(D, T) \
REGISTER_KERNEL_BUILDER( \
Name("ApplyRMSProp").Device(DEVICE_##D).TypeConstraint<T>("T"), \