| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | momentum.py | 43 use_locking=False, name="Momentum", use_nesterov=False): argument
74 self._use_nesterov = use_nesterov
89 use_nesterov=self._use_nesterov).op
100 use_nesterov=self._use_nesterov)
112 use_nesterov=self._use_nesterov).op
124 use_nesterov=self._use_nesterov)
|
| D | momentum_test.py | 176 learning_rate=2.0, momentum=0.9, use_nesterov=True)
212 learning_rate=2.0, momentum=0.9, use_nesterov=True)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | momentum.py | 47 use_locking=False, name="Momentum", use_nesterov=False): argument
78 self._use_nesterov = use_nesterov
103 use_nesterov=self._use_nesterov).op
113 use_nesterov=self._use_nesterov)
123 use_nesterov=self._use_nesterov).op
133 use_nesterov=self._use_nesterov)
|
| D | momentum_test.py | 177 learning_rate=2.0, momentum=0.9, use_nesterov=True)
214 learning_rate=2.0, momentum=0.9, use_nesterov=True)
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | lars_optimizer.py | 60 use_nesterov=False): argument
96 self._use_nesterov = use_nesterov
128 use_nesterov=self._use_nesterov)
140 use_nesterov=self._use_nesterov)
153 use_nesterov=self._use_nesterov).op
165 use_nesterov=self._use_nesterov)
|
| D | weight_decay_optimizers.py | 305 use_locking=False, name="MomentumW", use_nesterov=False): argument
333 use_locking=use_locking, name=name, use_nesterov=use_nesterov)
|
| D | nadam_optimizer.py | 51 use_nesterov=True).op
69 use_nesterov=True)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ResourceApplyKerasMomentum.pbtxt | 42 name: "use_nesterov"
49 summary: "Update \'*var\' according to the momentum scheme. Set use_nesterov = True if you"
|
| D | api_def_ResourceApplyMomentum.pbtxt | 42 name: "use_nesterov"
49 summary: "Update \'*var\' according to the momentum scheme. Set use_nesterov = True if you"
|
| D | api_def_ApplyMomentum.pbtxt | 48 name: "use_nesterov"
55 summary: "Update \'*var\' according to the momentum scheme. Set use_nesterov = True if you"
|
| D | api_def_SparseApplyMomentum.pbtxt | 54 name: "use_nesterov"
63 Set use_nesterov = True if you want to use Nesterov momentum.
|
| D | api_def_ResourceSparseApplyMomentum.pbtxt | 48 name: "use_nesterov"
57 Set use_nesterov = True if you want to use Nesterov momentum.
|
| D | api_def_ResourceSparseApplyKerasMomentum.pbtxt | 48 name: "use_nesterov"
57 Set use_nesterov = True if you want to use Nesterov momentum.
|
| D | api_def_ResourceApplyAdam.pbtxt | 72 name: "use_nesterov"
|
| D | api_def_ApplyAdam.pbtxt | 78 name: "use_nesterov"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops_gpu.cu.cc | 89 typename TTypes<T>::ConstScalar momentum, bool use_nesterov) { in operator ()()
94 if (use_nesterov) { in operator ()()
110 typename TTypes<T>::ConstScalar momentum, bool use_nesterov) { in operator ()()
116 if (use_nesterov) { in operator ()()
135 typename TTypes<T>::ConstFlat grad, bool use_nesterov) { in operator ()()
147 if (use_nesterov) { in operator ()()
|
| D | training_ops.h | 126 typename TTypes<T>::ConstScalar momentum, bool use_nesterov);
135 typename TTypes<T>::ConstScalar momentum, bool use_nesterov);
148 typename TTypes<T>::ConstFlat grad, bool use_nesterov);
|
| D | training_ops.cc | 276 typename TTypes<T>::ConstScalar momentum, bool use_nesterov) { in operator ()()
278 if (use_nesterov) { in operator ()()
292 typename TTypes<T>::ConstScalar momentum, bool use_nesterov) { in operator ()()
294 if (use_nesterov) { in operator ()()
312 typename TTypes<T>::ConstFlat grad, bool use_nesterov) { in operator ()()
322 if (use_nesterov) { in operator ()()
2453 typename TTypes<T>::ConstScalar momentum, bool use_nesterov); \
2671 typename TTypes<T>::ConstScalar momentum, bool use_nesterov); \
3049 typename TTypes<T>::ConstFlat grad, bool use_nesterov); \
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | gradient_descent.py | 113 use_nesterov=self.nesterov)
141 use_nesterov=self.nesterov)
|
| D | optimizer_v2_test.py | 655 learning_rate=0.001, momentum=0.9, use_nesterov=True)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-momentum-optimizer.pbtxt | 21 …self\', \'learning_rate\', \'momentum\', \'use_locking\', \'name\', \'use_nesterov\'], varargs=Non…
|
| D | tensorflow.raw_ops.pbtxt | 117 …lr\', \'beta1\', \'beta2\', \'epsilon\', \'grad\', \'use_locking\', \'use_nesterov\', \'name\'], v…
141 …'var\', \'accum\', \'lr\', \'grad\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
2881 …lr\', \'beta1\', \'beta2\', \'epsilon\', \'grad\', \'use_locking\', \'use_nesterov\', \'name\'], v…
2909 …'var\', \'accum\', \'lr\', \'grad\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
2913 …'var\', \'accum\', \'lr\', \'grad\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
3005 …um\', \'lr\', \'grad\', \'indices\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
3009 …um\', \'lr\', \'grad\', \'indices\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
3449 …um\', \'lr\', \'grad\', \'indices\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | momentum_test.py | 115 learning_rate=0.1, momentum=0.9, use_nesterov=True)
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | optimization_parameters.proto | 112 bool use_nesterov = 2; field
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 117 …lr\', \'beta1\', \'beta2\', \'epsilon\', \'grad\', \'use_locking\', \'use_nesterov\', \'name\'], v…
141 …'var\', \'accum\', \'lr\', \'grad\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
2881 …lr\', \'beta1\', \'beta2\', \'epsilon\', \'grad\', \'use_locking\', \'use_nesterov\', \'name\'], v…
2909 …'var\', \'accum\', \'lr\', \'grad\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
2913 …'var\', \'accum\', \'lr\', \'grad\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
3005 …um\', \'lr\', \'grad\', \'indices\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
3009 …um\', \'lr\', \'grad\', \'indices\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
3449 …um\', \'lr\', \'grad\', \'indices\', \'momentum\', \'use_locking\', \'use_nesterov\', \'name\'], v…
|