Searched defs:SparseApplyFtrl (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | training_ops.h | 192 struct SparseApplyFtrl { struct
|
| H A D | training_ops_gpu.cu.cc | 698 struct SparseApplyFtrl<GPUDevice, T, Tindex, has_l2_shrinkage> { struct
699 Status operator()(const GPUDevice& d, typename TTypes<T>::Matrix var, in operator ()()
|
| H A D | training_ops.cc | 589 struct SparseApplyFtrl<CPUDevice, T, Tindex, has_l2_shrinkage> { struct
590 Status operator()(const CPUDevice& d, typename TTypes<T>::Matrix var_flat, in operator ()()
|