HomeSort by relevance Sort by last modified time
    Searched refs:loss_shape (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
model_fn.py 162 loss_shape = loss.get_shape()
163 if loss_shape.num_elements() not in (None, 1):
165 if not loss_shape.is_compatible_with(tensor_shape.scalar()):
  /external/tensorflow/tensorflow/cc/gradients/
nn_grad_test.cc 117 TensorShape loss_shape({5});
126 {logits_shape, loss_shape});

Completed in 413 milliseconds