forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathops.GraphKeys.cs
More file actions
71 lines (60 loc) · 2.69 KB
/
ops.GraphKeys.cs
File metadata and controls
71 lines (60 loc) · 2.69 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
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";
}
}
}