Home
last modified time | relevance | path

Searched refs:loss_name (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_gpu_test.py41 def prepare_simple_model(input_tensor, loss_name, target): argument
46 if loss_name == 'sparse_categorical_crossentropy':
51 elif loss_name == 'categorical_crossentropy':
56 elif loss_name == 'binary_crossentropy':
Dtraining_utils.py504 loss_name = loss.name
505 if loss_name is None:
507 loss_name = loss_type.__name__
510 ' while using as loss `' + loss_name + '`. '
Dtraining.py1656 loss_name = self.output_names[i] + '_loss'
1657 with K.name_scope(loss_name):
1704 self._compile_metrics_tensors[loss_name] = aggregated_output_loss
/external/tensorflow/tensorflow/python/keras/wrappers/
Dscikit_learn.py296 loss_name = self.model.loss
297 if hasattr(loss_name, '__name__'):
298 loss_name = loss_name.__name__
299 if loss_name == 'categorical_crossentropy' and len(y.shape) != 2:
/external/tensorflow/tensorflow/contrib/gan/python/losses/python/
Dtuple_losses_test.py293 for loss_name in tfgan_losses.__all__:
294 if loss_name in manual_tests: continue
295 keyword_args = (generator_keyword_args if 'generator' in loss_name else
297 add_loss_consistency_test(ConsistentLossesTest, loss_name, keyword_args)
/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py959 loss_name = "loss_%s" % type(loss)
962 loss_name = loss.name
964 "to REGULARIZATION_LOSSES.", v_name, loss_name)