remove duplicate code in createBufferDetached. (#83)
Signed-off-by: Robert Altena <Rob@Ra-ai.com>master
parent
526b782e51
commit
fa98b83295
|
@ -1389,51 +1389,6 @@ public class Nd4j {
|
||||||
return Nd4j.getMemoryManager().getCurrentWorkspace() == null ? DATA_BUFFER_FACTORY_INSTANCE.createFloat(length, true) : DATA_BUFFER_FACTORY_INSTANCE.createFloat(length, true, Nd4j.getMemoryManager().getCurrentWorkspace());
|
return Nd4j.getMemoryManager().getCurrentWorkspace() == null ? DATA_BUFFER_FACTORY_INSTANCE.createFloat(length, true) : DATA_BUFFER_FACTORY_INSTANCE.createFloat(length, true, Nd4j.getMemoryManager().getCurrentWorkspace());
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Create a buffer equal of length prod(shape). The buffer is 'detached': Not in any memory workspace even if a
|
|
||||||
* workspace is currently open.
|
|
||||||
*
|
|
||||||
* @param shape the shape of the buffer to create
|
|
||||||
* @param type the opType to create
|
|
||||||
* @return
|
|
||||||
*/
|
|
||||||
public static DataBuffer createBufferDetached(int[] shape, DataType type) {
|
|
||||||
long length = ArrayUtil.prodLong(shape);
|
|
||||||
switch (type){
|
|
||||||
case DOUBLE:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createDouble(length);
|
|
||||||
case FLOAT:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createFloat(length);
|
|
||||||
case HALF:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createHalf(length);
|
|
||||||
case BFLOAT16:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createBFloat16(length);
|
|
||||||
case UINT64:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createULong(length);
|
|
||||||
case LONG:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createLong(length);
|
|
||||||
case UINT32:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createUInt(length);
|
|
||||||
case INT:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createInt(length);
|
|
||||||
case UINT16:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createUShort(length);
|
|
||||||
case SHORT:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createShort(length);
|
|
||||||
case UBYTE:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createUByte(length);
|
|
||||||
case BYTE:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createByte(length);
|
|
||||||
case BOOL:
|
|
||||||
return DATA_BUFFER_FACTORY_INSTANCE.createBool(length);
|
|
||||||
case UTF8:
|
|
||||||
case COMPRESSED:
|
|
||||||
case UNKNOWN:
|
|
||||||
default:
|
|
||||||
throw new UnsupportedOperationException("Cannot create type: " + type);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* See {@link #createBuffer(int[], DataType)}
|
* See {@link #createBuffer(int[], DataType)}
|
||||||
*/
|
*/
|
||||||
|
@ -1475,12 +1430,27 @@ public class Nd4j {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create a buffer equal of length prod(shape). The buffer is 'detached': Not in any memory workspace even if a
|
||||||
|
* workspace is currently open.
|
||||||
|
*
|
||||||
|
* @param shape the shape of the buffer to create
|
||||||
|
* @param type the opType to create
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
public static DataBuffer createBufferDetached(int[] shape, DataType type) {
|
||||||
|
return createBufferDetachedImpl( ArrayUtil.prodLong(shape), type);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* See {@link #createBufferDetached(int[], DataType)}
|
* See {@link #createBufferDetached(int[], DataType)}
|
||||||
*/
|
*/
|
||||||
public static DataBuffer createBufferDetached(long[] shape, DataType type) {
|
public static DataBuffer createBufferDetached(long[] shape, DataType type) {
|
||||||
long length = ArrayUtil.prodLong(shape);
|
return createBufferDetachedImpl( ArrayUtil.prodLong(shape), type);
|
||||||
|
}
|
||||||
|
|
||||||
|
// used by createBufferDetached(long[] DataType) and createBufferDetached(int[] , DataType)
|
||||||
|
private static DataBuffer createBufferDetachedImpl(long length, DataType type){
|
||||||
switch (type){
|
switch (type){
|
||||||
|
|
||||||
case DOUBLE:
|
case DOUBLE:
|
||||||
|
|
Loading…
Reference in New Issue