Searched refs:kUnknownRank (Results 1 – 7 of 7) sorted by relevance
161 static constexpr int32 kUnknownRank = -1; variable343 if (s->rank_ == kUnknownRank) { in Dim()350 CHECK_NE(s->rank_, kUnknownRank); in DimKnownRank()359 return s.IsSet() ? s->rank_ : kUnknownRank; in Rank()362 return (s.IsSet() && (Rank(s) != kUnknownRank)); in RankKnown()802 inline Shape::Shape() : rank_(InferenceContext::kUnknownRank) {} in Shape()
137 static const uint8 kUnknownRank = 255; variable226 return kIsPartial && ndims_byte() == kUnknownRank; in unknown_rank()233 return kIsPartial && dims == kUnknownRank ? -1 : dims; in dims()
29 constexpr int32 InferenceContext::kUnknownRank; member in tensorflow::shape_inference::InferenceContext313 if (rank == kUnknownRank) return UnknownDim(); in NumElements()376 if (existing == kUnknownRank) { in WithRank()397 if (existing >= rank || existing == kUnknownRank) { in WithRankAtLeast()412 if (existing <= rank || existing == kUnknownRank) { in WithRankAtMost()753 if (rank == kUnknownRank) { in UnknownShapeOfRank()
124 set_ndims_byte(kUnknownRank); in TensorShapeBase()234 set_ndims_byte(kUnknownRank); in TensorShapeBase()
1221 int32 rank = InferenceContext::kUnknownRank; in ConcatShapeHelper()1223 if (rank == InferenceContext::kUnknownRank) rank = c->Rank(c->input(i)); in ConcatShapeHelper()1224 if (rank != InferenceContext::kUnknownRank) { in ConcatShapeHelper()1228 if (rank == InferenceContext::kUnknownRank) { in ConcatShapeHelper()
265 EXPECT_EQ(InferenceContext::kUnknownRank, c.Rank(in0)); in TEST_F()
612 if (rank == InferenceContext::kUnknownRank) { in __anondb9326b20b02()