Searched refs:expected_rank (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_rank_op_test.py | 36 expected_rank=0,
41 expected_rank=1,
45 expected_rank=1,
50 expected_rank=2,
55 expected_rank=3,
61 expected_rank=3,
68 expected_rank=4,
76 expected_rank=4,
81 def testRaggedRank(self, test_input, expected_rank, ragged_rank=None): argument
85 test_input), expected_rank)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_variant_op.cc | 304 int index, int expected_rank, const Tensor** output_indices, in GetAndValidateSparseTensorIndicesAndValues() argument
326 if (rank != expected_rank) { in GetAndValidateSparseTensorIndicesAndValues()
330 "].shape but they do not: ", rank, " vs. ", expected_rank); in GetAndValidateSparseTensorIndicesAndValues()
|
| D | set_kernels.cc | 100 const auto expected_rank = sparse_tensor_shape.size(); in CheckGroup() local
101 OP_REQUIRES(ctx, expected_rank == group_rank, in CheckGroup()
102 errors::Internal("Rank expected ", expected_rank, ", got ", in CheckGroup()
104 for (int32 j = 0; j < expected_rank; ++j) { in CheckGroup()
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | sequence_queueing_state_saver.py | 139 def _check_rank(value, expected_rank): argument
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)
170 if not isinstance(expected_rank, ops.Tensor):
172 new_value.set_shape(new_value.get_shape().with_rank(expected_rank))
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | tensor_util.py | 154 def _is_rank(expected_rank, actual_tensor): argument
164 expected = ops.convert_to_tensor(expected_rank, name='expected')
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sets_test.py | 468 expected_rank = None
470 expected_rank = sparse_shape_dims[0].value
472 expected_rank = input_tensor.get_shape().ndims
473 self.assertAllEqual((None, expected_rank),
477 self.assertAllEqual((expected_rank,),
|