Searched defs:param_rows (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | training_ops_gpu.cu.cc | 38 const Tindex* indices, Tindex param_rows, Tindex updates_size, in SparseApplyAdagradKernel()
77 const Tindex* indices, Tindex param_rows, Tindex updates_size, in SparseApplyProximalAdagradKernel()
119 Tindex param_rows, Tindex updates_size, in SparseApplyFtrlKernel()
224 const T* momentum, bool use_nesterov, Tindex param_rows, in SparseApplyKerasMomentumKernel()
804 const T* epsilon, const T* grad, const Tindex* indices, Tindex param_rows, in SparseApplyAdadeltaKernel()
|