ai.djl.tensorflow.engine

Class TfNDManager

java.lang.Object

ai.djl.ndarray.BaseNDManager

ai.djl.tensorflow.engine.TfNDManager

All Implemented Interfaces:

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

public class TfNDManager
extends ai.djl.ndarray.BaseNDManager

Field Summary

Fields inherited from class ai.djl.ndarray.BaseNDManager

closed, device, parent, resources, uid

Method Summary

Modifier and Type	Method and Description
java.nio.ByteBuffer	allocateDirect(int capacity)
ai.djl.ndarray.NDArray	arange(float start, float stop, float step, ai.djl.ndarray.types.DataType dataType)
void	close()
ai.djl.ndarray.NDArray	create(boolean[] data)
TfNDArray	create(java.nio.Buffer data, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	create(byte[] data)
TfNDArray	create(java.nio.ByteBuffer data, ai.djl.ndarray.types.Shape shape)
ai.djl.ndarray.NDArray	create(float data)
ai.djl.ndarray.NDArray	create(float[] data)
ai.djl.ndarray.NDArray	create(int data)
ai.djl.ndarray.NDArray	create(int[] data)
ai.djl.ndarray.NDArray	create(ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
TfNDArray	create(org.tensorflow.Tensor<?> tensor)
ai.djl.ndarray.NDArray	createCSR(java.nio.Buffer data, long[] indptr, long[] indices, ai.djl.ndarray.types.Shape shape)
ai.djl.ndarray.NDArray	createRowSparse(java.nio.Buffer data, ai.djl.ndarray.types.Shape dataShape, long[] indices, ai.djl.ndarray.types.Shape shape)
void	detach(java.lang.String resourceId)
ai.djl.ndarray.NDArray	eye(int rows, int cols, int k, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	fill(ai.djl.ndarray.types.Shape shape, java.lang.Number value, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	full(ai.djl.ndarray.types.Shape shape, float value, ai.djl.ndarray.types.DataType dataType)
ai.djl.Device	getDevice()
ai.djl.engine.Engine	getEngine()
ai.djl.ndarray.NDManager	getParentManager()
static java.lang.Integer	getRandomSeed()
void	invoke(java.lang.String operation, ai.djl.ndarray.NDArray[] src, ai.djl.ndarray.NDArray[] dest, ai.djl.util.PairList<java.lang.String,?> params)
ai.djl.ndarray.NDList	invoke(java.lang.String operation, ai.djl.ndarray.NDList src, ai.djl.util.PairList<java.lang.String,?> params)
boolean	isOpen()
ai.djl.ndarray.NDArray	linspace(float start, float stop, int num, boolean endpoint)
ai.djl.ndarray.NDList	load(java.nio.file.Path path)
TfNDManager	newSubManager()
TfNDManager	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	randomMultinomial(int n, ai.djl.ndarray.NDArray pValues)
ai.djl.ndarray.NDArray	randomMultinomial(int n, ai.djl.ndarray.NDArray pValues, ai.djl.ndarray.types.Shape shape)
ai.djl.ndarray.NDArray	randomNormal(float loc, float scale, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
ai.djl.ndarray.NDArray	randomUniform(float low, float high, ai.djl.ndarray.types.Shape shape, ai.djl.ndarray.types.DataType dataType)
static void	setRandomSeed(java.lang.Integer seed)
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

attach, debugDump, 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, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, create, createCSR, createRowSparse, decode, decode, eye, eye, eye, eye, full, full, full, linspace, linspace, linspace, linspace, load, newBaseManager, newBaseManager, newBaseManager, ones, ones, randomNormal, randomNormal, randomNormal, randomUniform, randomUniform, zeros, zeros

Method Detail

allocateDirect

public java.nio.ByteBuffer allocateDirect(int capacity)

setRandomSeed

public static void setRandomSeed(java.lang.Integer seed)

getRandomSeed

public static java.lang.Integer getRandomSeed()

create

public ai.djl.ndarray.NDArray create(byte[] data)

create

public ai.djl.ndarray.NDArray create(float[] data)

create

public ai.djl.ndarray.NDArray create(int[] data)

create

public ai.djl.ndarray.NDArray create(boolean[] data)

create

public ai.djl.ndarray.NDArray create(int data)

create

public ai.djl.ndarray.NDArray create(float data)

create

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

create

public TfNDArray create(org.tensorflow.Tensor<?> tensor)

create

public TfNDArray create(java.nio.ByteBuffer data,
                        ai.djl.ndarray.types.Shape shape)

create

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

createCSR

public ai.djl.ndarray.NDArray createCSR(java.nio.Buffer data,
                                        long[] indptr,
                                        long[] indices,
                                        ai.djl.ndarray.types.Shape shape)

createRowSparse

public ai.djl.ndarray.NDArray createRowSparse(java.nio.Buffer data,
                                              ai.djl.ndarray.types.Shape dataShape,
                                              long[] indices,
                                              ai.djl.ndarray.types.Shape shape)

load

public ai.djl.ndarray.NDList load(java.nio.file.Path path)

invoke

public void invoke(java.lang.String operation,
                   ai.djl.ndarray.NDArray[] src,
                   ai.djl.ndarray.NDArray[] dest,
                   ai.djl.util.PairList<java.lang.String,?> params)

invoke

public ai.djl.ndarray.NDList invoke(java.lang.String operation,
                                    ai.djl.ndarray.NDList src,
                                    ai.djl.util.PairList<java.lang.String,?> params)

getEngine

public ai.djl.engine.Engine getEngine()

zeros

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

ones

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

full

public ai.djl.ndarray.NDArray full(ai.djl.ndarray.types.Shape shape,
                                   float value,
                                   ai.djl.ndarray.types.DataType dataType)

fill

public ai.djl.ndarray.NDArray fill(ai.djl.ndarray.types.Shape shape,
                                   java.lang.Number value,
                                   ai.djl.ndarray.types.DataType dataType)

arange

public ai.djl.ndarray.NDArray arange(float start,
                                     float stop,
                                     float step,
                                     ai.djl.ndarray.types.DataType dataType)

eye

public ai.djl.ndarray.NDArray eye(int rows,
                                  int cols,
                                  int k,
                                  ai.djl.ndarray.types.DataType dataType)

linspace

public ai.djl.ndarray.NDArray linspace(float start,
                                       float stop,
                                       int num,
                                       boolean endpoint)

randomUniform

public ai.djl.ndarray.NDArray randomUniform(float low,
                                            float high,
                                            ai.djl.ndarray.types.Shape shape,
                                            ai.djl.ndarray.types.DataType dataType)

randomNormal

public ai.djl.ndarray.NDArray randomNormal(float loc,
                                           float scale,
                                           ai.djl.ndarray.types.Shape shape,
                                           ai.djl.ndarray.types.DataType dataType)

randomMultinomial

public ai.djl.ndarray.NDArray randomMultinomial(int n,
                                                ai.djl.ndarray.NDArray pValues,
                                                ai.djl.ndarray.types.Shape shape)

randomMultinomial

public ai.djl.ndarray.NDArray randomMultinomial(int n,
                                                ai.djl.ndarray.NDArray pValues)

getParentManager

public ai.djl.ndarray.NDManager getParentManager()

Specified by:

getParentManager in interface ai.djl.ndarray.NDManager

Overrides:

getParentManager in class ai.djl.ndarray.BaseNDManager

getDevice

public ai.djl.Device getDevice()

Specified by:

getDevice in interface ai.djl.ndarray.NDManager

Overrides:

getDevice in class ai.djl.ndarray.BaseNDManager

newSubManager

public TfNDManager newSubManager()

newSubManager

public TfNDManager newSubManager(ai.djl.Device device)

isOpen

public boolean isOpen()

Specified by:

isOpen in interface ai.djl.ndarray.NDManager

Overrides:

isOpen in class ai.djl.ndarray.BaseNDManager

detach

public void detach(java.lang.String resourceId)

Specified by:

detach in interface ai.djl.ndarray.NDManager

Overrides:

detach in class ai.djl.ndarray.BaseNDManager

close

public void close()

Specified by:

close in interface ai.djl.ndarray.NDManager

Specified by:

close in interface java.lang.AutoCloseable

Overrides:

close in class ai.djl.ndarray.BaseNDManager