Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_gpu_test.py43 def prepare_simple_model(input_tensor, loss_name, target): argument
48 if loss_name == 'sparse_categorical_crossentropy':
53 elif loss_name == 'categorical_crossentropy':
58 elif loss_name == 'binary_crossentropy':
Dcompile_utils.py270 loss_name = get_custom_object_name(loss)
271 if loss_name is None:
273 loss = losses_mod.LossFunctionWrapper(loss, name=loss_name)
Dtraining_v1.py528 e.loss_name()
1593 loss_name = endpoint.loss_name()
1596 with K.name_scope(loss_name):
1888 name=endpoint.loss_name())
3042 def loss_name(self): member in _TrainingEndpoint
Dtraining_utils_v1.py834 loss_name = loss.name
835 if loss_name is None:
837 loss_name = loss_type.__name__
840 ' while using as loss `' + loss_name + '`. '
/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: