| /external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/ |
| D | resnet50_test.py | 53 def compute_gradients(model, images, labels, num_replicas=1): function
129 compute_gradients(model, images, labels))
154 compute_gradients(model, images, labels))
162 compute_gradients(model, images, labels))
279 compute_gradients(model, images, labels))
289 compute_gradients(model, images, labels))
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | revnet_test.py | 35 grads, loss = model.compute_gradients(
105 grads, loss = self.model.compute_gradients(
132 compute_gradients = tfe.defun(self.model.compute_gradients)
134 grads, _ = compute_gradients(saved_hidden=saved_hidden, labels=self.t)
160 grads, _ = model.compute_gradients(saved_hidden=saved_hidden, labels=t)
|
| D | main.py | 56 model.compute_gradients = tfe.defun(model.compute_gradients)
209 grads, loss = model.compute_gradients(saved_hiddens, labels)
|
| /external/tensorflow/tensorflow/contrib/mixed_precision/python/ |
| D | loss_scale_optimizer.py | 114 def compute_gradients(self, member in LossScaleOptimizer
135 grads_and_vars = self._opt.compute_gradients(
|
| D | loss_scale_optimizer_test.py | 107 grads_and_vars = opt.compute_gradients(loss, var_list=[x])
123 grads_and_vars = opt.compute_gradients(loss, var_list=[x])
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | drop_stale_gradient_optimizer.py | 63 def compute_gradients(self, loss, *args, **kwargs): member in DropStaleGradientOptimizer
70 return self._opt.compute_gradients(loss, *args, **kwargs)
|
| D | moving_average_optimizer.py | 90 def compute_gradients(self, *args, **kwargs): member in MovingAverageOptimizer
91 return self._optimizer.compute_gradients(*args, **kwargs)
|
| D | variable_clipping_optimizer.py | 83 def compute_gradients(self, *args, **kwargs): member in VariableClippingOptimizer
84 return self._opt.compute_gradients(*args, **kwargs)
|
| D | model_average_optimizer.py | 149 def compute_gradients(self, *args, **kwargs): member in ModelAverageOptimizer
161 return self._opt.compute_gradients(*args, **kwargs)
|
| D | moving_average_optimizer_test.py | 237 def compute_gradients(self, *args, **kwargs): member in MovingAverageOptimizerTest.testCorrectOverride.WrapperOptimizer
239 return super(WrapperOptimizer, self).compute_gradients(
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 100 def compute_gradients(self, loss, var_list=None, **kwargs): member in CrossShardOptimizer
140 return self._opt.compute_gradients(loss, var_list=var_list, **kwargs)
|
| D | tpu_embedding_gradient.py | 44 grads_and_vars = optimizer.compute_gradients(loss, activation_list)
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | __init__.py | 194 def compute_gradients(self, *args, **kwargs): member in DistributedOptimizer
203 .compute_gradients(*args, **kwargs))
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | optimizer_v2_test.py | 171 grads_and_vars = sgd_op.compute_gradients(loss, [var0, var1])
205 grads_and_vars = sgd_op.compute_gradients(f, [x])
256 grads_and_vars = sgd_op.compute_gradients(cost, [var0, var1],
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/ |
| D | rnn_ptb_graph_test.py | 47 grads = rnn_ptb.clip_gradients(optimizer.compute_gradients(loss), 0.25)
133 grads = rnn_ptb.clip_gradients(optimizer.compute_gradients(loss), 0.25)
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/ |
| D | densenet_test.py | 140 def compute_gradients(model, images, labels): function
292 compute_gradients(model, images, labels))
302 compute_gradients(model, images, labels))
|
| /external/tensorflow/tensorflow/contrib/constrained_optimization/python/ |
| D | external_regret_optimizer.py | 267 grads_and_vars = self.optimizer.compute_gradients(
282 grads_and_vars = self.optimizer.compute_gradients(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer.py | 208 def compute_gradients(self, *args, **kwargs): member in SyncReplicasOptimizer
224 return self._opt.compute_gradients(*args, **kwargs)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-optimizer.pbtxt | 27 name: "compute_gradients"
|
| D | tensorflow.train.-gradient-descent-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-proximal-adagrad-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-adagrad-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-proximal-gradient-descent-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-momentum-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-adadelta-optimizer.pbtxt | 28 name: "compute_gradients"
|