Searched refs:_optimizer (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | __init__.py | 190 self._optimizer = optimizer
208 return self._optimizer._apply_dense(*args, **kwargs)
212 return self._optimizer._apply_sparse(*args, **kwargs)
216 return self._optimizer._apply_sparse_duplicate_indices(*args,
221 return self._optimizer._prepare(*args, **kwargs)
225 return self._optimizer._create_slots(*args, **kwargs)
229 return self._optimizer._valid_dtypes(*args, **kwargs)
233 return self._optimizer._finish(*args, **kwargs)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | loss_scale_optimizer.py | 69 self._optimizer = opt
74 …grads_and_vars = self._optimizer._compute_gradients(loss, var_list, # pylint: disable=protected-a…
83 grads = self._optimizer.get_gradients(loss, params)
98 return self._optimizer.apply_gradients(grads_and_vars, name)
102 return self._optimizer.learning_rate
106 self._optimizer.learning_rate = lr
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | composable_model.py | 79 self._optimizer = optimizer
135 if (self._optimizer is None or isinstance(self._optimizer,
137 optimizer = self._get_default_optimizer(self._optimizer)
138 elif callable(self._optimizer):
139 optimizer = self._optimizer()
141 optimizer = self._optimizer
|
| D | linear_test.py | 336 def _optimizer(): function
340 n_classes=3, optimizer=_optimizer, feature_columns=[feature_column])
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | moving_average_optimizer.py | 84 self._optimizer = opt
91 return self._optimizer.compute_gradients(*args, **kwargs)
94 train_op = self._optimizer.apply_gradients(
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | step_fn.py | 92 self._optimizer = optimizer
108 return self._optimizer._distributed_apply( # pylint: disable=protected-access
|
| /external/tensorflow/tensorflow/contrib/constrained_optimization/python/ |
| D | constrained_optimizer.py | 51 self._optimizer = optimizer
56 return self._optimizer
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 2022 self._optimizer = optimizer
|