KGraph

A wrapper object to extend functionality and behaviour of static TensorFlow graph.

It tracks all model variables (used in optimizers or layers) and its initializers.

Parameters

graphDef

A serialized representation of the graph.

prefix

A prefix that will be prepended to names in graphDef.

KGraph

fun KGraph(graphDef: ByteArray)

KGraph

fun KGraph(graphDef: ByteArray, prefix: String)

Creates KGraph by serialized representation of the graph.

addOptimizerVariable

fun addOptimizerVariable(variable: Variable<Float>)

Adds a variable used in optimizer to the pool of tracked variables.

addOptimizerVariableAssignAddInitializer

Adds an optimizer initializer of special 'AssignAdd' type for optimizer variable tracked in KGraph.

addOptimizerVariableInitializer

Adds an optimizer initializer for optimizer variable tracked in KGraph.

open override fun close()

Closes internal TensorFlow graph.

copy

Makes a graph copy.

initializeOptimizerVariables

Initializes TensorFlow graph variables used in optimizer.

optimizerVariables

Returns all variables used in optimizer and initialized by Assign TensorFlow operand.

toString

open override fun toString(): String

variableNames

Returns list of variable names in TensorFlow graph.

isClosed

True if the graph object is closed and the occupied resources are freed.