OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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