TensorFlow.NET/src/TensorFlowNET.Core/ops.GraphKeys.cs at master · lokinfey/TensorFlow.NET · GitHub

History

71 lines (60 loc) · 2.69 KB

Raw

using System;

using System.Collections.Generic;

using System.Text;

namespace Tensorflow

{

public partial class ops

{

/// <summary>

/// Standard names to use for graph collections.

/// The standard library uses various well-known names to collect and

/// retrieve values associated with a graph. For example, the

/// `tf.Optimizer` subclasses default to optimizing the variables

/// collected under `tf.GraphKeys.TRAINABLE_VARIABLES` if none is

/// specified, but it is also possible to pass an explicit list of

/// variables.

/// </summary>

public static class GraphKeys

{

/// <summary>

/// the subset of `Variable` objects that will be trained by an optimizer.

/// </summary>

public static string TRAINABLE_VARIABLES = "trainable_variables";

/// <summary>

/// Trainable resource-style variables.

/// </summary>

public static string TRAINABLE_RESOURCE_VARIABLES = "trainable_resource_variables";

/// <summary>

/// Key for streaming model ports.

/// </summary>

public static string _STREAMING_MODEL_PORTS = "streaming_model_ports";

/// <summary>

/// Key to collect losses

/// </summary>

public const string LOSSES = "losses";

/// <summary>

/// Key to collect Variable objects that are global (shared across machines).

/// Default collection for all variables, except local ones.

/// </summary>

public static string GLOBAL_VARIABLES = "variables";

public static string TRAIN_OP = "train_op";

public static string GLOBAL_STEP = GLOBAL_STEP = "global_step";

public static string[] _VARIABLE_COLLECTIONS = new string[] { "variables", "trainable_variables", "model_variables" };

/// <summary>

/// Key to collect BaseSaverBuilder.SaveableObject instances for checkpointing.

/// </summary>

public static string SAVEABLE_OBJECTS = "saveable_objects";

/// <summary>

/// Key to collect update_ops

/// </summary>

public static string UPDATE_OPS = "update_ops";

// Key to collect summaries.

public const string SUMMARIES = "summaries";

// Used to store v2 summary names.

public static string _SUMMARY_COLLECTION = "_SUMMARY_V2";

// Key for control flow context.

public static string COND_CONTEXT = "cond_context";

public static string WHILE_CONTEXT = "while_context";

}

TensorFlow.NET/src/TensorFlowNET.Core/ops.GraphKeys.cs at master · lokinfey/TensorFlow.NET · GitHub

Expand file tree

Breadcrumbs

ops.GraphKeys.cs

Latest commit

History

Breadcrumbs

ops.GraphKeys.cs

File metadata and controls