Searched refs:learning_rate_t (Results 1 – 3 of 3) sorted by relevance
88 const Tensor* learning_rate_t; in Compute() local89 OP_REQUIRES_OK(context, context->input("learning_rate", &learning_rate_t)); in Compute()90 const auto learning_rate = learning_rate_t->scalar<float>()(); in Compute()
421 learning_rate_t = GetParam(self._learning_rate, global_step)424 var, indices, -learning_rate_t * preconditioned_grad)427 learning_rate_t * preconditioned_grad)
349 const Tensor* learning_rate_t; in Compute() local350 OP_REQUIRES_OK(context, context->input("learning_rate", &learning_rate_t)); in Compute()351 float learning_rate = learning_rate_t->scalar<float>()(); in Compute()