Home
last modified time | relevance | path

Searched refs:kIsPartial (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.cc59 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 …]
Dtensor_shape.h226 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 = variable
264 static_assert(kIsPartial || std::is_same<Shape, TensorShape>::value,