Home
last modified time | relevance | path

Searched defs:prox_var (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtraining_ops.cc64 xla::XlaOp prox_var = var - grad * lr; in ProximalGradientDescentUpdate() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops.cc85 auto prox_var = var; in operator ()() local
257 auto prox_var = var; in operator ()() local
H A Dtraining_ops_gpu.cu.cc303 T prox_var = var[i] - grad[i] * lr_scaled; in ApplyProximalAdagradKernel() local