Searched refs:train_g (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops_test.cc | 60 static void SGD(int32 n, Graph** init_g, Graph** train_g) { in SGD() argument
73 *train_g = g; in SGD()
88 static void Adagrad(int32 n, Graph** init_g, Graph** train_g) { in Adagrad() argument
105 *train_g = g; in Adagrad()
120 static void Momentum(int32 n, Graph** init_g, Graph** train_g) { in Momentum() argument
139 *train_g = g; in Momentum()
154 static void Adam(int32 n, Graph** init_g, Graph** train_g) { in Adam() argument
182 *train_g = g; in Adam()
197 static void RMSProp(int32 n, Graph** init_g, Graph** train_g) { in RMSProp() argument
222 *train_g = g; in RMSProp()
[all …]
|
| D | sdca_ops_test.cc | 126 Graph** train_g) { in GetGraphs() argument
226 *train_g = g; in GetGraphs()
|