Searched refs:init_g (Results 1 – 3 of 3) 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
65 *init_g = g; in SGD()
88 static void Adagrad(int32 n, Graph** init_g, Graph** train_g) { in Adagrad() argument
96 *init_g = g; in Adagrad()
120 static void Momentum(int32 n, Graph** init_g, Graph** train_g) { in Momentum() argument
129 *init_g = g; in Momentum()
154 static void Adam(int32 n, Graph** init_g, Graph** train_g) { in Adam() argument
165 *init_g = g; in Adam()
197 static void RMSProp(int32 n, Graph** init_g, Graph** train_g) { in RMSProp() argument
208 *init_g = g; in RMSProp()
[all …]
|
| D | basic_ops_benchmark_test.cc | 48 static void MulChain(int chain_length, Graph** init_g, Graph** run_g) { in MulChain() argument
53 *init_g = g; in MulChain()
|
| D | sdca_ops_test.cc | 125 const int32 dense_features_per_group, Graph** const init_g, in GetGraphs() argument
146 *init_g = g; in GetGraphs()
|