| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | tpu_gan_estimator_impl.py | 90 generator_optimizer=None, argument
218 generator_optimizer = _maybe_make_cross_shard_optimizer(
219 generator_optimizer)
244 get_eval_metric_ops_fn, generator_optimizer, discriminator_optimizer,
275 get_eval_metric_ops_fn, generator_optimizer, discriminator_optimizer, argument
304 g_callable = callable(generator_optimizer)
305 gopt = generator_optimizer() if g_callable else generator_optimizer
365 gan_model, gan_loss, generator_optimizer, discriminator_optimizer, argument
381 optimizer=generator_optimizer,
|
| D | gan_estimator_impl.py | 109 generator_optimizer=None, argument
192 get_eval_metric_ops_fn, generator_optimizer, discriminator_optimizer,
220 get_eval_metric_ops_fn, generator_optimizer, discriminator_optimizer, argument
236 if callable(generator_optimizer):
237 generator_optimizer = generator_optimizer()
242 gan_model, gan_loss, generator_optimizer, discriminator_optimizer,
327 gan_model, gan_loss, generator_optimizer, discriminator_optimizer, argument
331 train_ops = train_op_fn(gan_model, gan_loss, generator_optimizer,
|
| D | stargan_estimator_impl.py | 102 generator_optimizer=None, argument
181 get_eval_metric_ops_fn, generator_optimizer,
211 generator_optimizer, argument
225 generator_optimizer()
226 if callable(generator_optimizer) else generator_optimizer)
329 generator_optimizer, argument
335 train_ops = train_op_fn(gan_model, gan_loss, generator_optimizer,
|
| D | head_impl.py | 45 def gan_head(generator_loss_fn, discriminator_loss_fn, generator_optimizer, argument
74 generator_optimizer=generator_optimizer,
89 generator_optimizer, discriminator_optimizer, argument
136 self._generator_optimizer = generator_optimizer
|
| D | gan_estimator_test.py | 158 generator_optimizer=self._generator_optimizer,
192 generator_optimizer=g_opt,
228 generator_optimizer=gopt,
381 generator_optimizer=training.GradientDescentOptimizer(1.0),
392 generator_optimizer=training.GradientDescentOptimizer(1.0),
|
| D | tpu_gan_estimator_test.py | 120 generator_optimizer=self._generator_optimizer,
169 generator_optimizer=gopt,
270 generator_optimizer=training.GradientDescentOptimizer(1.0),
287 generator_optimizer=training.GradientDescentOptimizer(1.0),
|
| D | stargan_estimator_test.py | 180 generator_optimizer=self._generator_optimizer,
225 generator_optimizer=gopt,
|
| D | head_test.py | 67 generator_optimizer=training.GradientDescentOptimizer(1.0),
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/gan/ |
| D | mnist_test.py | 71 generator_optimizer = tf.train.AdamOptimizer(0.001)
79 mnist.train_one_epoch(generator, discriminator, generator_optimizer,
86 mnist.train_one_epoch(generator, discriminator, generator_optimizer,
|
| D | mnist_graph_test.py | 52 generator_optimizer = tf.train.AdamOptimizer(0.001)
70 generator_train = generator_optimizer.minimize(
|
| D | mnist.py | 197 def train_one_epoch(generator, discriminator, generator_optimizer, argument
253 generator_optimizer.apply_gradients(
|
| /external/tensorflow/tensorflow/contrib/gan/python/ |
| D | train.py | 926 generator_optimizer, argument
992 if isinstance(generator_optimizer,
1006 sync_hooks.append(generator_optimizer.make_session_run_hook(is_chief))
1010 optimizer=generator_optimizer,
|
| D | train_test.py | 926 model, loss, generator_optimizer=g_opt, discriminator_optimizer=d_opt)
|