Searched refs:sparse_indices_size (Results 1 – 2 of 2) sorted by relevance
700 int64 sparse_indices_size = in ValidateShapeSize() local702 if (sparse_indices_size < 0) { in ValidateShapeSize()703 return sparse_indices_size; in ValidateShapeSize()705 sparse_indices_size = in ValidateShapeSize()706 MultiplyWithoutOverflow(sparse_indices_size, sizeof(int64)); in ValidateShapeSize()707 if (sparse_indices_size < 0) { in ValidateShapeSize()708 return sparse_indices_size; in ValidateShapeSize()713 static_cast<uint64>(sparse_indices_size) > in ValidateShapeSize()
1977 TF_RET_CHECK(proto.sparse_indices_size() % subshape().rank() == 0) in CopyFromProto()1979 << proto.sparse_indices_size() << ") for shape of rank " in CopyFromProto()1981 const int64 index_count = proto.sparse_indices_size() / subshape().rank(); in CopyFromProto()