| Package | Description |
|---|---|
| org.nd4j.imports.graphmapper.tf |
| Modifier and Type | Method and Description |
|---|---|
static SameDiff |
TFGraphMapper.importGraph(@NonNull File f,
Map<String,TFImportOverride> importOverride,
TFOpImportFilter opFilter)
Import a frozen TensorFlow protobuf (.pb) file from the specified file, with optional overrides
|
static SameDiff |
TFGraphMapper.importGraph(@NonNull GraphDef tfGraph,
Map<String,TFImportOverride> importOverride,
TFOpImportFilter opFilter)
Import a TensorFlow model from a GraphDef, with optional import overrides
|
static SameDiff |
TFGraphMapper.importGraph(@NonNull InputStream is,
Map<String,TFImportOverride> importOverride,
TFOpImportFilter opFilter)
Import a frozen TensorFlow protobuf (.pb) file via an input stream, with optional overrides
|
static SameDiff |
TFGraphMapper.importGraphTxt(@NonNull InputStream is,
Map<String,TFImportOverride> importOverride,
TFOpImportFilter opFilter)
Import a frozen TensorFlow protobuf file in text format (.pb.txt) file via an input stream, with optional overrides
|
Copyright © 2021. All rights reserved.