Searched refs:train_g (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops_test.cc | 89 static void SGD(int32 n, Graph** init_g, Graph** train_g) { in SGD() argument
102 *train_g = g; in SGD()
121 static void Adagrad(int32 n, Graph** init_g, Graph** train_g) { in Adagrad() argument
138 *train_g = g; in Adagrad()
157 static void SparseAdagrad(int32 m, int32 n, Graph** init_g, Graph** train_g) { in SparseAdagrad() argument
176 *train_g = g; in SparseAdagrad()
201 static void Momentum(int32 n, Graph** init_g, Graph** train_g) { in Momentum() argument
220 *train_g = g; in Momentum()
239 static void Adam(int32 n, Graph** init_g, Graph** train_g) { in Adam() argument
267 *train_g = g; in Adam()
[all …]
|
| D | sdca_ops_test.cc | 126 Graph** train_g) { in GetGraphs() argument
226 *train_g = g; in GetGraphs()
|