Searched defs:SparseApplyAdagrad (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | training_ops.h | 108 struct SparseApplyAdagrad { struct
|
| H A D | training_ops_gpu.cu.cc | 431 struct SparseApplyAdagrad<GPUDevice, T, Tindex, has_epsilon> { struct
432 Status operator()(const GPUDevice& d, typename TTypes<T>::Matrix var, in operator ()()
|
| H A D | training_ops.cc | 166 struct SparseApplyAdagrad<CPUDevice, T, Tindex, has_epsilon> { struct
167 Status operator()(const CPUDevice& d, typename TTypes<T>::Matrix var, in operator ()()
|