Searched defs:accum_update (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | training_ops_gpu.cu.cc | 313 GpuLaunchConfig cfg, T* var, T* accum, T* accum_update, const T* plr, in ApplyAdadeltaKernel()
803 T* var, T* accum, T* accum_update, const T* lr, const T* rho, in SparseApplyAdadeltaKernel()
|
| H A D | training_ops.cc | 1071 Tensor accum_update; in DoValidate() local
1127 Tensor accum_update; in DoCompute() local
1210 Tensor accum_update; in DoCompute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | training_ops.cc | 916 xla::XlaOp var, accum, accum_update; in Compile() local
|