Searched defs:ApplyGradientDescent (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops.h | 32 struct ApplyGradientDescent { struct
|
| D | training_ops_gpu.cu.cc | 337 struct ApplyGradientDescent<GPUDevice, T> { struct
338 void operator()(const GPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
1086 template struct functor::ApplyGradientDescent<GPUDevice, Eigen::half>; member in tensorflow::functor
1087 template struct functor::ApplyGradientDescent<GPUDevice, float>; member in tensorflow::functor
1088 template struct functor::ApplyGradientDescent<GPUDevice, double>; member in tensorflow::functor
1091 template struct functor::ApplyGradientDescent<GPUDevice, complex64>; member in tensorflow::functor
1092 template struct functor::ApplyGradientDescent<GPUDevice, complex128>; member in tensorflow::functor
|
| D | training_ops.cc | 47 struct ApplyGradientDescent<CPUDevice, T> { struct
48 void operator()(const CPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
|