Searched refs:kIsPartial (Results 1 – 2 of 2) sorted by relevance
59 if (kIsPartial && proto.unknown_rank()) return proto.dim_size() == 0; in IsValid()63 if (d.size() < (kIsPartial ? -1 : 0)) return false; in IsValid()66 } else if (!kIsPartial || num_elements >= 0) { in IsValid()79 if (kIsPartial && proto.unknown_rank()) { in IsValidShape()92 if (d.size() < (kIsPartial ? -1 : 0)) { in IsValidShape()93 if (kIsPartial) { in IsValidShape()104 } else if (!kIsPartial || num_elements >= 0) { in IsValidShape()123 if (kIsPartial && proto.unknown_rank()) { in TensorShapeBase()182 const bool neg = Set16(kIsPartial, dst, 0, size); in InitDims()190 bool neg = Set16(kIsPartial, dst, 0, size0); in InitDims()[all …]
226 return kIsPartial && ndims_byte() == kUnknownRank; in unknown_rank()233 return kIsPartial && dims == kUnknownRank ? -1 : dims; in dims()248 bool IsFullyDefined() const { return !kIsPartial || num_elements() != -1; } in IsFullyDefined()262 static constexpr bool kIsPartial = variable264 static_assert(kIsPartial || std::is_same<Shape, TensorShape>::value,