Home
last modified time | relevance | path

Searched refs:rank_diff (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dconfusion_matrix.py69 rank_diff = predictions_rank - labels_rank
70 if rank_diff == expected_rank_diff + 1:
72 elif rank_diff == expected_rank_diff - 1:
77 rank_diff = array_ops.rank(predictions) - array_ops.rank(labels)
81 math_ops.equal(expected_rank_diff + 1, rank_diff),
87 math_ops.equal(expected_rank_diff - 1, rank_diff),
Dmetrics_impl.py137 rank_diff = weights_rank_tensor - array_ops.rank(predictions)
141 math_ops.equal(rank_diff, -1),
153 math_ops.equal(rank_diff, 1), maybe_squeeze_weights,
/external/tensorflow/tensorflow/python/keras/utils/
Dlosses_utils.py100 rank_diff = array_ops.rank(y_pred) - array_ops.rank(y_true)
107 math_ops.equal(1, rank_diff), maybe_squeeze_dims, squeeze_dims)
128 rank_diff = weights_rank_tensor - array_ops.rank(y_pred)
133 math_ops.equal(rank_diff,
139 math_ops.equal(rank_diff, 1), maybe_squeeze_weights,
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dtensor_util.py109 rank_diff = predictions_rank - labels_rank
110 if rank_diff == -1:
112 elif rank_diff == 1:
117 rank_diff = array_ops.rank(predictions) - array_ops.rank(labels)
121 math_ops.equal(1, rank_diff),
127 math_ops.equal(-1, rank_diff),
/external/tensorflow/tensorflow/python/ops/losses/
Dlosses_impl.py823 rank_diff = weights_rank - labels_rank
824 if rank_diff == 1:
829 rank_diff = array_ops.rank(weights) - array_ops.rank(labels)
833 math_ops.equal(1, rank_diff),
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dcwise_op_vectorizer.cc79 Output rank_diff = ops::Sub(scope, max_rank, rank); in ExpandDimsForBroadcast() local
83 scope, ops::ExpandDims(scope, rank_diff, ops::Const(scope, 0)), in ExpandDimsForBroadcast()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py716 rank_diff = array_ops.reshape(max_rank - ranks[i], [1])
717 ones = array_ops.tile([1], rank_diff)
/external/tensorflow/tensorflow/python/keras/
Dbackend.py3401 rank_diff = len(input_t.shape) - len(mask_t.shape)
3402 for _ in range(rank_diff):