org.nd4j.autodiff.execution

Class NativeGraphExecutioner

java.lang.Object

org.nd4j.autodiff.execution.NativeGraphExecutioner

All Implemented Interfaces:

GraphExecutioner

public class NativeGraphExecutioner
extends Object
implements GraphExecutioner

Nested Class Summary

Nested classes/interfaces inherited from interface org.nd4j.autodiff.execution.GraphExecutioner

GraphExecutioner.Type

Constructor Summary

Constructors

Constructor and Description
NativeGraphExecutioner()

Method Summary

Modifier and Type	Method and Description
ByteBuffer	convertToFlatBuffers(SameDiff sd, ExecutorConfiguration configuration) This method converts given SameDiff instance to FlatBuffers representation
INDArray[]	executeGraph(int id, SDVariable... variables) This method executes
INDArray[]	executeGraph(SameDiff sd) This method executes given graph and returns results PLEASE NOTE: Default configuration is used
INDArray[]	executeGraph(SameDiff sd, ExecutorConfiguration configuration) This method executes given graph and returns results
GraphExecutioner.Type	getExecutionerType() This method returns Type of this executioner
static byte	getFlatOpType(Op.Type type)
static long	getOpNum(String name, Op.Type type)
INDArray[]	importProto(File file) This method executes TF graph PLEASE NOTE: This feature is under development yet
int	registerGraph(SameDiff graph) This method stores given graph for future execution
INDArray[]	reuseGraph(SameDiff graph, Map<Integer,INDArray> inputs)

Methods inherited from class java.lang.Object

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

Constructor Detail

NativeGraphExecutioner

public NativeGraphExecutioner()

Method Detail

getExecutionerType

public GraphExecutioner.Type getExecutionerType()

This method returns Type of this executioner

Specified by:

getExecutionerType in interface GraphExecutioner

Returns:

executeGraph

public INDArray[] executeGraph(SameDiff sd)

This method executes given graph and returns results PLEASE NOTE: Default configuration is used

Specified by:

executeGraph in interface GraphExecutioner

Parameters:

sd -

Returns:

reuseGraph

public INDArray[] reuseGraph(SameDiff graph,
                             Map<Integer,INDArray> inputs)

Specified by:

reuseGraph in interface GraphExecutioner

convertToFlatBuffers

public ByteBuffer convertToFlatBuffers(SameDiff sd,
                                       ExecutorConfiguration configuration)

Description copied from interface: GraphExecutioner

This method converts given SameDiff instance to FlatBuffers representation

Specified by:

convertToFlatBuffers in interface GraphExecutioner

Returns:

executeGraph

public INDArray[] executeGraph(SameDiff sd,
                               ExecutorConfiguration configuration)

This method executes given graph and returns results

Specified by:

executeGraph in interface GraphExecutioner

Parameters:

sd -

Returns:

getOpNum

public static long getOpNum(String name,
                            Op.Type type)

getFlatOpType

public static byte getFlatOpType(Op.Type type)

executeGraph

public INDArray[] executeGraph(int id,
                               SDVariable... variables)

This method executes

Specified by:

executeGraph in interface GraphExecutioner

Parameters:

id -

variables -

Returns:

registerGraph

public int registerGraph(SameDiff graph)

This method stores given graph for future execution

Specified by:

registerGraph in interface GraphExecutioner

Parameters:

graph -

Returns:

importProto

public INDArray[] importProto(File file)

Description copied from interface: GraphExecutioner

This method executes TF graph PLEASE NOTE: This feature is under development yet

Specified by:

importProto in interface GraphExecutioner

Returns: