Searched refs:init_g (Results 1 – 3 of 3) 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
94 *init_g = g; in SGD()
121 static void Adagrad(int32 n, Graph** init_g, Graph** train_g) { in Adagrad() argument
129 *init_g = g; in Adagrad()
157 static void SparseAdagrad(int32 m, int32 n, Graph** init_g, Graph** train_g) { in SparseAdagrad() argument
165 *init_g = g; in SparseAdagrad()
201 static void Momentum(int32 n, Graph** init_g, Graph** train_g) { in Momentum() argument
210 *init_g = g; in Momentum()
239 static void Adam(int32 n, Graph** init_g, Graph** train_g) { in Adam() argument
250 *init_g = g; in Adam()
[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()
|