Searched refs:kUnknownRank (Results 1 – 17 of 17) sorted by relevance
60 const int Rank() const { return has_value_ ? value_.size() : kUnknownRank; } in Rank()85 static constexpr int kUnknownRank = -1; variable
127 static_assert(::tensorflow::shape_inference::InferenceContext::kUnknownRank ==128 Shape::kUnknownRank,
35 if (scales_rank == shape_inference::InferenceContext::kUnknownRank || in ScalesZeroPointsShapeValid()36 zero_points_rank == shape_inference::InferenceContext::kUnknownRank) { in ScalesZeroPointsShapeValid()
56 if (rank != c->kUnknownRank) { in RaggedCountSparseOutputShapeFn()
110 if (rank == InferenceContext::kUnknownRank) { in __anon8359b0a00102()
619 if (rank == InferenceContext::kUnknownRank) { in __anon091a71b70b02()
235 static constexpr int32_t kUnknownRank = -1; variable414 if (!s.Handle() || s->rank_ == kUnknownRank) { in Dim()421 CHECK_NE(s->rank_, kUnknownRank); in DimKnownRank()429 return s.IsSet() ? s->rank_ : kUnknownRank; in Rank()432 return (s.IsSet() && (Rank(s) != kUnknownRank)); in RankKnown()878 inline Shape::Shape() : rank_(InferenceContext::kUnknownRank) {} in Shape()
135 static constexpr uint8 kUnknownRank = 255; variable286 return kIsPartial && ndims_byte() == kUnknownRank; in unknown_rank()293 return kIsPartial && dims == kUnknownRank ? -1 : dims; in dims()
30 constexpr int32_t InferenceContext::kUnknownRank; member in tensorflow::shape_inference::InferenceContext263 if (rank == kUnknownRank) return UnknownDim(); in NumElements()325 if (existing == kUnknownRank) { in WithRank()346 if (existing >= rank || existing == kUnknownRank) { in WithRankAtLeast()361 if (existing <= rank || existing == kUnknownRank) { in WithRankAtMost()702 if (rank == kUnknownRank) { in UnknownShapeOfRank()
146 set_ndims_byte(kUnknownRank); in TensorShapeBase()166 out->set_ndims_byte(kUnknownRank); in BuildTensorShapeBase()302 set_ndims_byte(kUnknownRank); in TensorShapeBase()
916 if (c->Rank(specified_input_grad_shape) == InferenceContext::kUnknownRank) { in Conv2DBackpropInputShape()1986 int32_t rank = InferenceContext::kUnknownRank; in ConcatShapeHelper()1988 if (rank == InferenceContext::kUnknownRank) rank = c->Rank(c->input(i)); in ConcatShapeHelper()1989 if (rank != InferenceContext::kUnknownRank) { in ConcatShapeHelper()1993 if (rank == InferenceContext::kUnknownRank) { in ConcatShapeHelper()
290 EXPECT_EQ(InferenceContext::kUnknownRank, c.Rank(in0)); in TEST_F()
41 if (!s.Handle() || s->rank_ == kUnknownRank) {48 CHECK_NE(s->rank_, kUnknownRank);
50 auto& kUnknownRank = shape_inference::InferenceContext::kUnknownRank; variable70 if (num_dims == kUnknownRank) { in Shape()
664 if (input_shape.num_dims == InferenceContext::kUnknownRank) { in TFE_InferShapes()698 if (shape.num_dims == InferenceContext::kUnknownRank) { in TFE_InferShapes()
579 if (num_dims == shape_inference::InferenceContext::kUnknownRank) { in SetInferenceShape()
70 constexpr int kUnknownRank = -1; variable563 return kUnknownRank; in GetFanoutPortRank()