Searched refs:kUnknownRank (Results 1 – 12 of 12) sorted by relevance
159 static constexpr int32 kUnknownRank = -1; variable330 if (!s.Handle() || s->rank_ == kUnknownRank) { in Dim()337 CHECK_NE(s->rank_, kUnknownRank); in DimKnownRank()345 return s.IsSet() ? s->rank_ : kUnknownRank; in Rank()348 return (s.IsSet() && (Rank(s) != kUnknownRank)); in RankKnown()787 inline Shape::Shape() : rank_(InferenceContext::kUnknownRank) {} in Shape()
137 static constexpr uint8 kUnknownRank = 255; variable287 return kIsPartial && ndims_byte() == kUnknownRank; in unknown_rank()294 return kIsPartial && dims == kUnknownRank ? -1 : dims; in dims()
29 constexpr int32 InferenceContext::kUnknownRank; member in tensorflow::shape_inference::InferenceContext256 if (rank == kUnknownRank) return UnknownDim(); in NumElements()318 if (existing == kUnknownRank) { in WithRank()339 if (existing >= rank || existing == kUnknownRank) { in WithRankAtLeast()354 if (existing <= rank || existing == kUnknownRank) { in WithRankAtMost()695 if (rank == kUnknownRank) { in UnknownShapeOfRank()
147 set_ndims_byte(kUnknownRank); in TensorShapeBase()167 out->set_ndims_byte(kUnknownRank); in BuildTensorShapeBase()303 set_ndims_byte(kUnknownRank); in TensorShapeBase()
880 if (c->Rank(specified_input_grad_shape) == InferenceContext::kUnknownRank) { in Conv2DBackpropInputShape()1895 int32 rank = InferenceContext::kUnknownRank; in ConcatShapeHelper()1897 if (rank == InferenceContext::kUnknownRank) rank = c->Rank(c->input(i)); in ConcatShapeHelper()1898 if (rank != InferenceContext::kUnknownRank) { in ConcatShapeHelper()1902 if (rank == InferenceContext::kUnknownRank) { in ConcatShapeHelper()
282 EXPECT_EQ(InferenceContext::kUnknownRank, c.Rank(in0)); in TEST_F()
49 auto& kUnknownRank = shape_inference::InferenceContext::kUnknownRank; variable69 if (num_dims == kUnknownRank) { in Shape()
54 if (rank != c->kUnknownRank) { in RaggedCountSparseOutputShapeFn()
618 if (rank == InferenceContext::kUnknownRank) { in __anon42d741190b02()
657 if (input_shape.num_dims == InferenceContext::kUnknownRank) { in TFE_InferShapes()691 if (shape.num_dims == InferenceContext::kUnknownRank) { in TFE_InferShapes()
579 if (num_dims == shape_inference::InferenceContext::kUnknownRank) { in SetInferenceShape()
70 constexpr int kUnknownRank = -1; variable568 return kUnknownRank; in GetFanoutPortRank()