Searched refs:weights_rank (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | weights_broadcast_ops.py | 48 weights_rank, weights_shape, values_rank, values_shape): argument
51 (weights_rank, weights_shape, values_rank, values_shape)) as scope:
53 values_rank, weights_rank, name="is_same_rank")
86 weights_rank = array_ops.rank(weights, name="rank")
87 weights_rank_static = tensor_util.constant_value(weights_rank)
121 is_scalar = math_ops.equal(0, weights_rank, name="is_scalar")
132 weights_rank, weights_shape, values_rank, values_shape),
|
| D | metrics_impl.py | 122 weights_rank = weights_shape.ndims
123 if weights_rank == 0:
128 if (predictions_rank is not None) and (weights_rank is not None):
130 if weights_rank - predictions_rank == 1:
132 elif predictions_rank - weights_rank == 1:
145 if ((weights_rank is not None) and
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | util.py | 88 weights_rank = weights_shape.ndims
89 if weights_rank == 0: # If weights is scalar, do nothing.
92 if (y_pred_rank is not None) and (weights_rank is not None):
94 if weights_rank - y_pred_rank == 1:
96 elif y_pred_rank - weights_rank == 1:
|
| D | losses_impl.py | 824 weights_rank = weights_shape.ndims
826 if (labels_rank is not None) and (weights_rank is not None):
828 rank_diff = weights_rank - labels_rank
835 if (weights_rank is None) or (
836 weights_rank > 0 and weights_shape.dims[-1].is_compatible_with(1)):
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | losses_utils.py | 200 weights_rank = weights_shape.ndims
201 if weights_rank == 0: # If weights is scalar, do nothing.
204 if (y_pred_rank is not None) and (weights_rank is not None):
206 if weights_rank - y_pred_rank == 1:
208 elif y_pred_rank - weights_rank == 1:
|