| Package | Description |
|---|---|
| org.nd4j.imports.graphmapper | |
| org.nd4j.imports.graphmapper.onnx | |
| org.nd4j.imports.graphmapper.tf |
| 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 |
|---|---|
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 |
|---|---|
protected void |
TFGraphMapper.importCondition(String conditionName,
NodeDef tfNode,
ImportState<GraphDef,NodeDef> importState) |
void |
TFGraphMapper.mapNodeType(NodeDef tfNode,
ImportState<GraphDef,NodeDef> importState,
OpImportOverride<GraphDef,NodeDef,AttrValue> importOverride,
OpImportFilter<GraphDef,NodeDef,AttrValue> opFilter) |
Copyright © 2019. All rights reserved.