Searched refs:use_iterative_root (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo_test.py | 154 def _testBasicTensor(self, use_iterative_root, use_resource_var): argument
175 use_iterative_root=use_iterative_root)
244 def testBasicTensor(self, use_iterative_root, use_resource_var): argument
245 self._testBasicTensor(use_iterative_root, use_resource_var)
448 def _testSparseUpdateSmall(self, use_iterative_root, use_resource_var): argument
476 use_iterative_root=use_iterative_root)
518 def testSparseUpdateSmall(self, use_iterative_root, use_resource_var): argument
519 self._testSparseUpdateSmall(use_iterative_root, use_resource_var)
521 def _testBasicTensorWithMomentum(self, use_iterative_root, use_resource_var): argument
545 use_iterative_root=use_iterative_root)
[all …]
|
| D | weight_decay_optimizers.py | 405 use_iterative_root=False, argument
450 use_iterative_root=use_iterative_root,
|
| D | shampoo.py | 83 use_iterative_root=False, argument
135 self._use_iterative_root = use_iterative_root
|