Searched defs:learning_rates (Results 1 – 5 of 5) sorted by relevance
11 std::vector<double> learning_rates = get_lrs(); in step() local16 void LRScheduler::set_optimizer_lrs(const std::vector<double>& learning_rates) { in set_optimizer_lrs()
107 std::vector<shape_inference::ShapeHandle> learning_rates; in __anone1b182e00302() local
231 std::vector<xla::XlaOp> learning_rates; in Compile() local
264 learning_rates=None, argument
433 std::vector<shape_inference::ShapeHandle> learning_rates; in __anoncc2b8c080b02() local