ai.djl.paddlepaddle.engine

Class PpNDManager

java.lang.Object

ai.djl.ndarray.BaseNDManager

ai.djl.paddlepaddle.engine.PpNDManager

All Implemented Interfaces:

ai.djl.ndarray.NDManager, java.lang.AutoCloseable

public class PpNDManager
extends ai.djl.ndarray.BaseNDManager

PpNDManager is the PaddlePaddle implementation of NDManager.

Nested Class Summary

Nested classes/interfaces inherited from class ai.djl.ndarray.BaseNDManager

ai.djl.ndarray.BaseNDManager.TempResource

Field Summary

Fields inherited from class ai.djl.ndarray.BaseNDManager

closed, device, name, parent, resources, tempResources, uid

Method Summary

Modifier and Type	Method and Description
java.nio.ByteBuffer	allocateDirect(int capacity)
PpNDArray	create(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.engine.Engine	getEngine()
PpNDManager	newSubManager()
PpNDManager	newSubManager(ai.djl.Device device)
ai.djl.ndarray.NDArray	ones(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	zeros(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)

Methods inherited from class ai.djl.ndarray.BaseNDManager

arange, attachInternal, close, create, create, create, createCoo, createCSR, createRowSparse, debugDump, detachInternal, eye, full, getDevice, getName, getParentManager, invoke, invoke, isOpen, linspace, load, randomInteger, randomMultinomial, randomMultinomial, randomNormal, randomUniform, setName, tempAttachInternal, toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface ai.djl.ndarray.NDManager

arange, arange, arange, arange, arange, arange, arange, arange, attachAll, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, createCSR, createCSR, createRowSparse, decode, decode, eye, eye, eye, eye, from, full, full, full, linspace, linspace, linspace, linspace, load, newBaseManager, newBaseManager, newBaseManager, ones, ones, randomNormal, randomNormal, randomNormal, randomUniform, randomUniform, ret, tempAttachAll, zeros, zeros

Method Detail

newSubManager

public PpNDManager newSubManager()

Specified by:

newSubManager in interface ai.djl.ndarray.NDManager

Overrides:

newSubManager in class ai.djl.ndarray.BaseNDManager

newSubManager

public PpNDManager newSubManager(ai.djl.Device device)

getEngine

public ai.djl.engine.Engine getEngine()

allocateDirect

public java.nio.ByteBuffer allocateDirect(int capacity)

create

public PpNDArray create(java.nio.Buffer data,
                        ai.djl.ndarray.types.Shape shape,
                        ai.djl.ndarray.types.DataType dataType)

zeros

public ai.djl.ndarray.NDArray zeros(ai.djl.ndarray.types.Shape shape,
                                    ai.djl.ndarray.types.DataType dataType)

Specified by:

zeros in interface ai.djl.ndarray.NDManager

Overrides:

zeros in class ai.djl.ndarray.BaseNDManager

ones

public ai.djl.ndarray.NDArray ones(ai.djl.ndarray.types.Shape shape,
                                   ai.djl.ndarray.types.DataType dataType)

Specified by:

ones in interface ai.djl.ndarray.NDManager

Overrides:

ones in class ai.djl.ndarray.BaseNDManager