CollectionDef should cover most collections.
To add a user-defined collection, do one of the following:
1. For simple data types, such as string, int, float:
tf.add_to_collection("your_collection_name", your_simple_value)
strings will be stored as bytes_list.
2. For Protobuf types, there are three ways to add them:
1) tf.add_to_collection("your_collection_name",
your_proto.SerializeToString())
collection_def {
key: "user_defined_bytes_collection"
value {
bytes_list {
value: "queue_name: \"test_queue\"\n"
}
}
}
or
2) tf.add_to_collection("your_collection_name", str(your_proto))
collection_def {
key: "user_defined_string_collection"
value {
bytes_list {
value: "\n\ntest_queue"
}
}
}
or
3) any_buf = any_pb2.Any()
tf.add_to_collection("your_collection_name",
any_buf.Pack(your_proto))
collection_def {
key: "user_defined_any_collection"
value {
any_list {
value {
type_url: "type.googleapis.com/tensorflow.QueueRunnerDef"
value: "\n\ntest_queue"
}
}
}
}
3. For Python objects, implement to_proto() and from_proto(), and register
them in the following manner:
ops.register_proto_function("your_collection_name",
proto_type,
to_proto=YourPythonObject.to_proto,
from_proto=YourPythonObject.from_proto)
These functions will be invoked to serialize and de-serialize the
collection. For example,
ops.register_proto_function(ops.GraphKeys.GLOBAL_VARIABLES,
proto_type=variable_pb2.VariableDef,
to_proto=Variable.to_proto,
from_proto=Variable.from_proto)
tensorflow.CollectionDef
Nested Classes
| class | CollectionDef.AnyList | AnyList is used for collecting Any protos. |
|
| interface | CollectionDef.AnyListOrBuilder | ||
| class | CollectionDef.Builder | CollectionDef should cover most collections. |
|
| class | CollectionDef.BytesList | BytesList is used for collecting strings and serialized protobufs. |
|
| interface | CollectionDef.BytesListOrBuilder | ||
| class | CollectionDef.FloatList | FloatList is used for collecting float values. |
|
| interface | CollectionDef.FloatListOrBuilder | ||
| class | CollectionDef.Int64List | Int64List is used for collecting int, int64 and long values. |
|
| interface | CollectionDef.Int64ListOrBuilder | ||
| enum | CollectionDef.KindCase | ||
| class | CollectionDef.NodeList | NodeList is used for collecting nodes in graph. |
|
| interface | CollectionDef.NodeListOrBuilder | ||
Constants
| int | ANY_LIST_FIELD_NUMBER | |
| int | BYTES_LIST_FIELD_NUMBER | |
| int | FLOAT_LIST_FIELD_NUMBER | |
| int | INT64_LIST_FIELD_NUMBER | |
| int | NODE_LIST_FIELD_NUMBER |
Public Methods
| boolean |
equals(Object obj)
|
| CollectionDef.AnyList |
getAnyList()
.tensorflow.CollectionDef.AnyList any_list = 5;
|
| CollectionDef.AnyListOrBuilder |
getAnyListOrBuilder()
.tensorflow.CollectionDef.AnyList any_list = 5;
|
| CollectionDef.BytesList |
getBytesList()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
|
| CollectionDef.BytesListOrBuilder |
getBytesListOrBuilder()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
|
| static CollectionDef | |
| CollectionDef | |
| final static com.google.protobuf.Descriptors.Descriptor | |
| CollectionDef.FloatList |
getFloatList()
.tensorflow.CollectionDef.FloatList float_list = 4;
|
| CollectionDef.FloatListOrBuilder |
getFloatListOrBuilder()
.tensorflow.CollectionDef.FloatList float_list = 4;
|
| CollectionDef.Int64List |
getInt64List()
.tensorflow.CollectionDef.Int64List int64_list = 3;
|
| CollectionDef.Int64ListOrBuilder |
getInt64ListOrBuilder()
.tensorflow.CollectionDef.Int64List int64_list = 3;
|
| CollectionDef.KindCase | |
| CollectionDef.NodeList |
getNodeList()
.tensorflow.CollectionDef.NodeList node_list = 1;
|
| CollectionDef.NodeListOrBuilder |
getNodeListOrBuilder()
.tensorflow.CollectionDef.NodeList node_list = 1;
|
|
|
|
| int | |
| final com.google.protobuf.UnknownFieldSet | |
| boolean |
hasAnyList()
.tensorflow.CollectionDef.AnyList any_list = 5;
|
| boolean |
hasBytesList()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
|
| boolean |
hasFloatList()
.tensorflow.CollectionDef.FloatList float_list = 4;
|
| boolean |
hasInt64List()
.tensorflow.CollectionDef.Int64List int64_list = 3;
|
| boolean |
hasNodeList()
.tensorflow.CollectionDef.NodeList node_list = 1;
|
| int |
hashCode()
|
| final boolean | |
| static CollectionDef.Builder |
newBuilder(CollectionDef prototype)
|
| static CollectionDef.Builder | |
| CollectionDef.Builder | |
| static CollectionDef |
parseDelimitedFrom(InputStream input)
|
| static CollectionDef |
parseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
| static CollectionDef |
parseFrom(ByteBuffer data)
|
| static CollectionDef |
parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
| static CollectionDef |
parseFrom(ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
| static CollectionDef |
parseFrom(com.google.protobuf.CodedInputStream input)
|
| static CollectionDef |
parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
| static CollectionDef |
parseFrom(com.google.protobuf.ByteString data)
|
| static CollectionDef |
parseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
| static CollectionDef |
parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
|
static
|
parser()
|
| CollectionDef.Builder | |
| void |
writeTo(com.google.protobuf.CodedOutputStream output)
|
Inherited Methods
Constants
public static final int ANY_LIST_FIELD_NUMBER
public static final int BYTES_LIST_FIELD_NUMBER
public static final int FLOAT_LIST_FIELD_NUMBER
public static final int INT64_LIST_FIELD_NUMBER
public static final int NODE_LIST_FIELD_NUMBER
Public Methods
public boolean equals (Object obj)
public CollectionDef.AnyListOrBuilder getAnyListOrBuilder ()
.tensorflow.CollectionDef.AnyList any_list = 5;
public CollectionDef.BytesListOrBuilder getBytesListOrBuilder ()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor ()
public CollectionDef.FloatListOrBuilder getFloatListOrBuilder ()
.tensorflow.CollectionDef.FloatList float_list = 4;
public CollectionDef.Int64ListOrBuilder getInt64ListOrBuilder ()
.tensorflow.CollectionDef.Int64List int64_list = 3;
public CollectionDef.NodeListOrBuilder getNodeListOrBuilder ()
.tensorflow.CollectionDef.NodeList node_list = 1;
public
getParserForType
()
public int getSerializedSize ()
public final com.google.protobuf.UnknownFieldSet getUnknownFields ()
public boolean hasAnyList ()
.tensorflow.CollectionDef.AnyList any_list = 5;
public boolean hasBytesList ()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
public boolean hasFloatList ()
.tensorflow.CollectionDef.FloatList float_list = 4;
public boolean hasInt64List ()
.tensorflow.CollectionDef.Int64List int64_list = 3;
public boolean hasNodeList ()
.tensorflow.CollectionDef.NodeList node_list = 1;
public int hashCode ()
public final boolean isInitialized ()
public static CollectionDef parseDelimitedFrom (InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
| IOException |
|---|
public static CollectionDef parseFrom (com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
| IOException |
|---|
public static CollectionDef parseFrom (ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
| InvalidProtocolBufferException |
|---|
public static CollectionDef parseFrom (com.google.protobuf.CodedInputStream input)
Throws
| IOException |
|---|
public static CollectionDef parseFrom (byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
| InvalidProtocolBufferException |
|---|
public static CollectionDef parseFrom (com.google.protobuf.ByteString data)
Throws
| InvalidProtocolBufferException |
|---|
public static CollectionDef parseFrom (InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
| IOException |
|---|
public static CollectionDef parseFrom (com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
| InvalidProtocolBufferException |
|---|
public
static
parser
()
public void writeTo (com.google.protobuf.CodedOutputStream output)
Throws
| IOException |
|---|