org.nd4j.imports.tensorflow

Class TensorFlowImportValidator

java.lang.Object

org.nd4j.imports.tensorflow.TensorFlowImportValidator

public class TensorFlowImportValidator
extends Object

Constructor Summary

Constructors

Constructor and Description
TensorFlowImportValidator()

Method Summary

Modifier and Type	Method and Description
static TFImportStatus	checkAllModelsForImport(@NonNull File directory) Recursively scan the specified directory for .pb files, and evaluate which operations/graphs can/can't be imported
static TFImportStatus	checkAllModelsForImport(@NonNull File directory, boolean includeArchives)
static TFImportStatus	checkAllModelsForImport(File directory, String[] fileExtensions)
static TFImportStatus	checkModelForImport(@NonNull File file) See checkModelForImport(File).
static TFImportStatus	checkModelForImport(@NonNull File file, boolean exceptionOnRead) Check whether the TensorFlow frozen model (protobuf format) can be imported into SameDiff or not
static TFImportStatus	checkModelForImport(String path, InputStream is, boolean exceptionOnRead)
static boolean	isArchiveFile(File f)

Methods inherited from class java.lang.Object

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

Constructor Detail

TensorFlowImportValidator

public TensorFlowImportValidator()

Method Detail

checkAllModelsForImport

public static TFImportStatus checkAllModelsForImport(@NonNull
                                                     @NonNull File directory)
                                              throws IOException

Recursively scan the specified directory for .pb files, and evaluate which operations/graphs can/can't be imported

Parameters:

directory - Directory to scan

Returns:

Status for TensorFlow import for all models in

Throws:

IOException

checkAllModelsForImport

public static TFImportStatus checkAllModelsForImport(@NonNull
                                                     @NonNull File directory,
                                                     boolean includeArchives)
                                              throws IOException

Throws:

IOException

checkAllModelsForImport

public static TFImportStatus checkAllModelsForImport(File directory,
                                                     String[] fileExtensions)
                                              throws IOException

Throws:

IOException

isArchiveFile

public static boolean isArchiveFile(File f)

checkModelForImport

public static TFImportStatus checkModelForImport(@NonNull
                                                 @NonNull File file)
                                          throws IOException

See checkModelForImport(File). Defaults to exceptionOnRead = false

Throws:

IOException

checkModelForImport

public static TFImportStatus checkModelForImport(@NonNull
                                                 @NonNull File file,
                                                 boolean exceptionOnRead)
                                          throws IOException

Check whether the TensorFlow frozen model (protobuf format) can be imported into SameDiff or not

Parameters:

file - Protobuf file

exceptionOnRead - If true, and the file can't be read, throw an exception. If false, return an "empty" TFImportStatus

Returns:

Status for importing the file

Throws:

IOException - If error

checkModelForImport

public static TFImportStatus checkModelForImport(String path,
                                                 InputStream is,
                                                 boolean exceptionOnRead)
                                          throws IOException

Throws:

IOException