Searched refs:ndims_byte (Results 1 – 2 of 2) sorted by relevance
138 uint8 ndims_byte() const { return buf()[14]; } in ndims_byte() function287 return kIsPartial && ndims_byte() == kUnknownRank; in unknown_rank()293 uint8 dims = ndims_byte(); in dims()
327 set_ndims_byte(b.ndims_byte()); in SlowCopyFrom()399 CHECK_LT(ndims_byte(), MaxDimensions()) << "Too many dimensions in tensor"; in AddDim()422 if (TF_PREDICT_FALSE(ndims_byte() >= MaxDimensions())) { in AddDimWithStatus()444 const int nd = ndims_byte(); in UnsafeAddDim()