| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| auto_parallel_test.cc | 41 Output learning_rate = ops::Const(s.WithOpName("learning_rate"), 0.01f, {1}); local
43 s.WithOpName("apply_gradient"), {var}, {learning_rate}, {add});
94 EXPECT_EQ("AutoParallel-Replica-0/learning_rate", node_learning_rate0.name());
117 EXPECT_EQ("AutoParallel-Replica-1/learning_rate", node_learning_rate1.name());
|
| /external/tensorflow/tensorflow/core/kernels/ |
| word2vec_kernels.cc | 275 const Tensor& learning_rate = ctx->input(4); variable
276 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(learning_rate.shape()),
283 auto lr = learning_rate.scalar<float>()();
|
| training_ops.cc | 171 auto learning_rate = accum.constant(lr()) * accum.rsqrt(); local
174 prox_var.device(d) -= grad * learning_rate;
178 (prox_var.abs() - learning_rate * prox_var.constant(l1()))
180 (var.constant(1.0) + var.constant(l2()) * learning_rate);
183 prox_var / (var.constant(1.0) + var.constant(l2()) * learning_rate);
974 auto learning_rate = v.constant(lr_scalar); variable
1005 auto learning_rate = lr_scalar; variable
1468 auto learning_rate = a.constant(lr_scalar) * a.rsqrt(); variable
1502 auto learning_rate = lr_scalar \/ std::sqrt(a); variable
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| training_ops.cc | 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>()(); variable
396 UpdateAndRetrieveGrowableTree(ensemble_resource, learning_rate,
521 const float learning_rate, const uint64 dropout_seed) {
527 ensemble_resource->AddNewTree(learning_rate);
[all...] |
