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

  /external/tensorflow/tensorflow/contrib/framework/python/framework/
tensor_util.py 138 def _is_rank(expected_rank, actual_tensor):
139 """Returns whether actual_tensor's rank is expected_rank.
142 expected_rank: Integer defining the expected rank, or tensor of same.
148 expected = ops.convert_to_tensor(expected_rank, name='expected')
  /external/tensorflow/tensorflow/contrib/training/python/training/
sequence_queueing_state_saver.py 139 def _check_rank(value, expected_rank):
144 expected_rank: int32 scalar (optionally a `Tensor`).
148 assertions on its rank. If expected_rank is not a `Tensor`, then
152 ValueError: if `expected_rank` is not a `Tensor` and the rank of `value`
153 is known and is not equal to `expected_rank`.
158 math_ops.equal(expected_rank, array_ops.rank(value)), [
161 string_ops.as_string(expected_rank), ", shape received:"
166 if isinstance(expected_rank, ops.Tensor):
167 expected_rank_value = tensor_util.constant_value(expected_rank)
169 expected_rank = int(expected_rank_value
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
set_kernels.cc 105 const auto expected_rank = sparse_tensor_shape.size(); local
106 OP_REQUIRES(ctx, expected_rank == group_rank,
107 errors::Internal("Rank expected ", expected_rank, ", got ",
109 for (int32 j = 0; j < expected_rank; ++j) {
  /external/tensorflow/tensorflow/python/kernel_tests/
sets_test.py 461 expected_rank = None
463 expected_rank = sparse_shape_dims[0].value
465 expected_rank = input_tensor.get_shape().ndims
466 self.assertAllEqual((None, expected_rank),
470 self.assertAllEqual((expected_rank,),
    [all...]

Completed in 107 milliseconds