| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50_test.py | 37 def compute_gradients(model, images, labels, num_replicas=1): function
212 compute_gradients(model, images, labels))
240 compute_gradients(model, images, labels))
248 compute_gradients(model, images, labels))
371 compute_gradients(model, images, labels))
381 compute_gradients(model, images, labels))
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 111 def compute_gradients(self, loss, var_list=None, **kwargs): member in CrossShardOptimizer
161 return self._opt.compute_gradients(loss, var_list=var_list, **kwargs)
|
| D | tpu_embedding_gradient.py | 46 grads_and_vars = optimizer.compute_gradients(loss, activation_list)
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | auto_mixed_precision_test.py | 505 g = optimizer.compute_gradients(y, [x, f])
534 g = optimizer.compute_gradients(y, [x])
598 g = optimizer.compute_gradients(y, [x, f])
625 g = optimizer.compute_gradients(y, [x])
647 g = optimizer.compute_gradients(k, [x])
675 g = optimizer.compute_gradients(y, [x])
701 g = optimizer.compute_gradients(y, [x])
733 g = optimizer.compute_gradients(h, [init_c, init_h])
815 g = optimizer.compute_gradients(y, [x])
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 84 def compute_gradients(self, member in MixedPrecisionLossScaleOptimizer
119 grads_and_vars = self._optimizer.compute_gradients(
|
| D | loss_scale_optimizer_test.py | 64 grads_and_vars = opt.compute_gradients(loss, params)
|
| /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-gradient-descent-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-adadelta-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-adagrad-d-a-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.tpu.-cross-shard-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-adam-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-momentum-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.mixed_precision.-mixed-precision-loss-scale-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.-r-m-s-prop-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.experimental.-mixed-precision-loss-scale-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-ftrl-optimizer.pbtxt | 28 name: "compute_gradients"
|
| D | tensorflow.train.-sync-replicas-optimizer.pbtxt | 28 name: "compute_gradients"
|
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer.py | 206 def compute_gradients(self, *args, **kwargs): member in SyncReplicasOptimizer
222 return self._opt.compute_gradients(*args, **kwargs)
|
| D | optimizer_test.py | 196 grads_and_vars = sgd_op.compute_gradients(loss, [var0, var1])
230 grads_and_vars = sgd_op.compute_gradients(f, [x])
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizer_v1.py | 790 return self.optimizer.compute_gradients(loss, params)
800 grads = self.optimizer.compute_gradients(loss)
802 grads = self.optimizer.compute_gradients(loss, params)
813 grads = self.optimizer.compute_gradients(loss, params)
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_Numpy_Distributed_Image_Classification.ipynb | 279 "def compute_gradients(model, inputs, labels):\n",
325 " gradients = compute_gradients(model, input, labels)\n",
494 " gradients = compute_gradients(model, inputs, labels)\n",
|