Searched refs:loss_shape (Results 1 – 2 of 2) sorted by relevance
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()):
117 TensorShape loss_shape({5}); in TEST_F() local126 {logits_shape, loss_shape}); in TEST_F()