Searched refs:weights_rank (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | weights_broadcast_ops.py | 47 weights_rank, weights_shape, values_rank, values_shape): argument
50 (weights_rank, weights_shape, values_rank, values_shape)) as scope:
52 values_rank, weights_rank, name="is_same_rank")
85 weights_rank = array_ops.rank(weights, name="rank")
86 weights_rank_static = tensor_util.constant_value(weights_rank)
120 is_scalar = math_ops.equal(0, weights_rank, name="is_scalar")
131 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/keras/utils/ |
| D | losses_utils.py | 114 weights_rank = weights_shape.ndims
115 if weights_rank == 0: # If weights is scalar, do nothing.
118 if (y_pred_rank is not None) and (weights_rank is not None):
120 if weights_rank - y_pred_rank == 1:
122 elif y_pred_rank - weights_rank == 1:
|
| /external/tensorflow/tensorflow/contrib/kernel_methods/python/ |
| D | losses.py | 103 weights_rank = weights.get_shape().ndims
104 if weights_rank not in [0, 1, 2]:
109 if weights_rank > 0:
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | losses_impl.py | 819 weights_rank = weights_shape.ndims
821 if (labels_rank is not None) and (weights_rank is not None):
823 rank_diff = weights_rank - labels_rank
830 if (weights_rank is None) or (
831 weights_rank > 0 and weights_shape.dims[-1].is_compatible_with(1)):
|