Searched defs:updates_size (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| 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()
225 Tindex updates_size, Tindex indices_size) { in SparseApplyKerasMomentumKernel()
805 Tindex updates_size, Tindex indices_size) { in SparseApplyAdadeltaKernel()
|