Home
last modified time | relevance | path

Searched defs:ApplyGradientDescent (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops_gpu.cu.cc262 struct ApplyGradientDescent<GPUDevice, T> { struct
263 void operator()(const GPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
1079 template struct functor::ApplyGradientDescent<GPUDevice, Eigen::half>; member in tensorflow::functor
1080 template struct functor::ApplyGradientDescent<GPUDevice, float>; member in tensorflow::functor
1081 template struct functor::ApplyGradientDescent<GPUDevice, double>; member in tensorflow::functor
1082 template struct functor::ApplyGradientDescent<GPUDevice, complex64>; member in tensorflow::functor
1083 template struct functor::ApplyGradientDescent<GPUDevice, complex128>; member in tensorflow::functor
H A Dtraining_ops.h32 struct ApplyGradientDescent { struct
H A Dtraining_ops.cc47 struct ApplyGradientDescent<CPUDevice, T> { struct
48 void operator()(const CPUDevice& d, typename TTypes<T>::Flat var, in operator ()()