| /external/tensorflow/tensorflow/contrib/gan/python/losses/python/ |
| D | losses_impl.py | 84 add_summaries=False): argument
112 if add_summaries:
126 add_summaries=False): argument
166 if add_summaries:
186 add_summaries=False): argument
239 if add_summaries:
254 add_summaries=False): argument
294 if add_summaries:
317 add_summaries=False): argument
395 if add_summaries:
[all …]
|
| D | tuple_losses_impl.py | 256 def cycle_consistency_loss(cyclegan_model, scope=None, add_summaries=False): argument
282 scope, add_summaries)
|
| D | losses_impl_test.py | 169 self._g_loss_fn(self._discriminator_gen_outputs, add_summaries=True)
176 add_summaries=True)
392 self._g_loss_fn(add_summaries=True, **self._generator_kwargs)
397 self._d_loss_fn(add_summaries=True, **self._discriminator_kwargs)
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | gan_estimator_impl.py | 113 add_summaries=None, argument
186 add_summaries)
202 add_summaries, generator_scope='Generator'): argument
213 generator_scope, add_summaries, mode)
229 gan_model, add_summaries=use_loss_summaries),
231 gan_model, add_summaries=use_loss_summaries))
249 generator_inputs, generator_scope, add_summaries, mode): argument
263 if add_summaries:
264 if not isinstance(add_summaries, (tuple, list)):
265 add_summaries = [add_summaries]
[all …]
|
| D | stargan_estimator_impl.py | 106 add_summaries=None, argument
176 add_summaries)
193 add_summaries, argument
202 add_summaries, mode)
238 input_data_domain_label, generator_scope, add_summaries, argument
252 if add_summaries:
253 if not isinstance(add_summaries, (tuple, list)):
254 add_summaries = [add_summaries]
256 for summary_type in add_summaries:
|
| D | tpu_gan_estimator_impl.py | 93 add_summaries=None, argument
238 add_summaries=None if is_on_tpu else add_summaries)
284 gan_model, add_summaries=not is_on_tpu),
286 gan_model, add_summaries=not is_on_tpu))
290 gan_model, add_summaries=False, reduction=losses.Reduction.NONE),
292 gan_model, add_summaries=False, reduction=losses.Reduction.NONE))
298 gan_model, add_summaries=not is_on_tpu),
300 gan_model, add_summaries=not is_on_tpu))
|
| D | latent_gan_estimator_impl.py | 100 add_summaries = params['add_summaries']
115 loss = loss_fn(gan_model, features, labels, add_summaries)
126 if add_summaries:
|
| D | head_impl.py | 131 generator_loss_fn, add_summaries=use_loss_summaries)
133 discriminator_loss_fn, add_summaries=use_loss_summaries)
|
| D | gan_estimator_test.py | 81 add_summaries=False)
123 def dummy_loss_fn(gan_model, add_summaries=True): argument
124 del add_summaries
|
| D | head_test.py | 35 def dummy_loss(gan_model, add_summaries=True): # pylint:disable=unused-argument argument
|
| D | stargan_estimator_test.py | 77 add_summaries=False)
|
| /external/tensorflow/tensorflow/contrib/gan/python/ |
| D | train.py | 577 add_summaries=True): argument
654 gen_loss = generator_loss_fn(model, add_summaries=add_summaries)
655 dis_loss = discriminator_loss_fn(pooled_model, add_summaries=add_summaries)
664 add_summaries=add_summaries)
668 model, add_summaries=add_summaries)
673 pooled_model, add_summaries=add_summaries)
678 model, add_summaries=add_summaries)
682 pooled_model, add_summaries=add_summaries)
738 model, add_summaries=kwargs.get('add_summaries', True))
778 add_summaries=True): argument
[all …]
|
| D | train_test.py | 520 loss = train.gan_loss(get_gan_model_fn(), add_summaries=True)
529 loss = train.cyclegan_loss(get_gan_model_fn(), add_summaries=True)
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | quantile_ops_test.py | 158 summaries = accumulator.add_summaries([self._feature_0, self._feature_1],
187 summaries = accumulator.add_summaries([self._feature_0, self._feature_1],
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | boosted_trees_ops.py | 146 def add_summaries(self, float_columns, example_weights): member in QuantileAccumulator
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 496 CopySummaryToProto(summary, stream_proto->add_summaries()); in Compute()
|