Home | History | Annotate | Download | only in kernels

Lines Matching defs:learning_rate

341     // Read the learning_rate.
343     OP_REQUIRES_OK(context, context->input("learning_rate", &learning_rate_t));
344     float learning_rate = learning_rate_t->scalar<float>()();
396         UpdateAndRetrieveGrowableTree(ensemble_resource, learning_rate,
521       const float learning_rate, const uint64 dropout_seed) {
527           ensemble_resource->AddNewTree(learning_rate);