Searched refs:prox_w (Results 1 – 1 of 1) sorted by relevance
269 auto prox_w = weights_inputs.at(i, /*lock_held=*/true).flat<float>(); in Compute() local270 prox_w.device(context->eigen_cpu_device()) = in Compute()271 regularizations_.EigenShrinkVector(prox_w); in Compute()