public class BasicGraphExecutioner extends Object implements GraphExecutioner
GraphExecutioner.Type
Constructor and Description |
---|
BasicGraphExecutioner() |
Modifier and Type | Method and Description |
---|---|
ByteBuffer |
convertToFlatBuffers(SameDiff diff,
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 graph,
ExecutorConfiguration configuration)
This method executes given graph and returns results
|
GraphExecutioner.Type |
getExecutionerType()
This method returns Type of this executioner
|
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) |
public GraphExecutioner.Type getExecutionerType()
getExecutionerType
in interface GraphExecutioner
public INDArray[] executeGraph(SameDiff graph, ExecutorConfiguration configuration)
executeGraph
in interface GraphExecutioner
graph
- public ByteBuffer convertToFlatBuffers(SameDiff diff, ExecutorConfiguration configuration)
GraphExecutioner
convertToFlatBuffers
in interface GraphExecutioner
diff
- public INDArray[] executeGraph(SameDiff sd)
executeGraph
in interface GraphExecutioner
sd
- public INDArray[] reuseGraph(SameDiff graph, Map<Integer,INDArray> inputs)
reuseGraph
in interface GraphExecutioner
public INDArray[] executeGraph(int id, SDVariable... variables)
executeGraph
in interface GraphExecutioner
id
- variables
- public int registerGraph(SameDiff graph)
registerGraph
in interface GraphExecutioner
graph
- public INDArray[] importProto(File file)
GraphExecutioner
importProto
in interface GraphExecutioner
Copyright © 2019. All rights reserved.