in ndarray/src/main/java/org/tensorflow/ndarray/impl/buffer/raw/ByteRawDataBuffer.java [144:175]
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if (!(obj instanceof ByteDataBuffer)) {
return super.equals(obj);
}
ByteDataBuffer other = (ByteDataBuffer)obj;
if (size() != other.size()) {
return false;
}
return other.accept(new DataStorageVisitor<Boolean>() {
@Override
public Boolean visit(ByteBuffer buffer) {
if (memory.isArray()) {
return buffer.equals(memory.toArrayByteBuffer());
}
return fallback();
}
@Override
public Boolean fallback() {
for (long idx = 0L; idx < size(); ++idx) {
if (other.getByte(idx) != getByte(idx)) {
return false;
}
}
return true;
}
});
}