org.nd4j.jita.allocator.impl.AllocationShape Maven / Gradle / Ivy

Go to download

package org.nd4j.jita.allocator.impl;

import lombok.Data;
import lombok.NoArgsConstructor;
import org.nd4j.linalg.api.buffer.DataBuffer;

/**
 * @author [email protected]
 */
@Data
@NoArgsConstructor
public class AllocationShape {
    private long offset = 0;
    private long length = 0;
    private int stride = 1;
    private int elementSize = 0;
    private DataBuffer.Type dataType = DataBuffer.Type.FLOAT;

    public AllocationShape(long length, int elementSize) {
        this.length = length;
        this.elementSize = elementSize;
    }
}