| Package | Description |
|---|---|
| org.nd4j.imports.graphmapper | |
| org.nd4j.imports.graphmapper.onnx | |
| org.nd4j.imports.graphmapper.tf | |
| org.nd4j.imports.tensorflow |
| Modifier and Type | Method and Description |
|---|---|
void |
GraphMapper.mapNodeType(NODE_TYPE tfNode,
ImportState<GRAPH_TYPE,TENSOR_TYPE> importState,
OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE> opImportOverride,
OpImportFilter<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE> opFilter)
Map a node in to the import state covering the
SameDiff instance |
| Modifier and Type | Method and Description |
|---|---|
SameDiff |
GraphMapper.importGraph(File graphFile,
Map<String,? extends OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE>> opImportOverrides,
OpImportFilter<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE> opFilter)
Import a graph as SameDiff from the given file, with optional op import overrides.
The OpImportOverride instances allow the operation import to be overridden - useful for importing ops
that have not been mapped for import in SameDiff yet, and also for non-standard/user-defined functions. |
SameDiff |
BaseGraphMapper.importGraph(File graphFile,
Map<String,? extends OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE>> opImportOverrides,
OpImportFilter<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE> opFilter) |
SameDiff |
GraphMapper.importGraph(GRAPH_TYPE graph,
Map<String,? extends OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE>> opImportOverrides,
OpImportFilter<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE> opFilter)
This method converts given graph type (in its native format) to SameDiff
The OpImportOverride instances allow the operation import to be overridden - useful for importing ops
that have not been mapped for import in SameDiff yet, and also for non-standard/user-defined functions. |
SameDiff |
BaseGraphMapper.importGraph(GRAPH_TYPE tfGraph,
Map<String,? extends OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE>> opImportOverrides,
OpImportFilter<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE> opFilter) |
SameDiff |
GraphMapper.importGraph(InputStream graphFile,
Map<String,? extends OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE>> opImportOverrides,
OpImportFilter<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE> opFilter) |
SameDiff |
BaseGraphMapper.importGraph(InputStream inputStream,
Map<String,? extends OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE>> opImportOverrides,
OpImportFilter<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE> opFilter) |
protected GRAPH_TYPE |
BaseGraphMapper.readGraph(InputStream inputStream,
Map<String,? extends OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE>> opImportOverrides) |
| Modifier and Type | Method and Description |
|---|---|
void |
OnnxGraphMapper.mapNodeType(Onnx.NodeProto tfNode,
ImportState<Onnx.GraphProto,Onnx.TypeProto.Tensor> importState,
OpImportOverride<Onnx.GraphProto,Onnx.NodeProto,Onnx.AttributeProto> opImportOverride,
OpImportFilter<Onnx.GraphProto,Onnx.NodeProto,Onnx.AttributeProto> opFilter) |
| Modifier and Type | Method and Description |
|---|---|
void |
TFGraphMapper.mapNodeType(NodeDef tfNode,
ImportState<GraphDef,NodeDef> importState,
OpImportOverride<GraphDef,NodeDef,AttrValue> importOverride,
OpImportFilter<GraphDef,NodeDef,AttrValue> opFilter) |
| Modifier and Type | Interface and Description |
|---|---|
interface |
TFImportOverride |
Copyright © 2019. All rights reserved.