Home | History | Annotate | Download | only in kernels

Lines Matching defs:train

82   Graph* train;
83 SGD(params, &init, &train);
84 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);
114 Graph* train;
115 Adagrad(params, &init, &train);
116 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);
148 Graph* train;
149 Momentum(params, &init, &train);
150 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);
191 Graph* train;
192 Adam(params, &init, &train);
193 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);
231 Graph* train;
232 RMSProp(params, &init, &train);
233 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);
268 Graph* train;
269 AddSign(params, &init, &train);
270 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);
305 Graph* train;
306 PowerSign(params, &init, &train);
307 test::Benchmark("cpu", train, GetOptions(), init).Run(iters);