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
Public Methods
| CollectionDef.Builder |
addRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
|
| CollectionDef |
build()
|
| CollectionDef | |
| CollectionDef.Builder |
clear()
|
| CollectionDef.Builder |
clearAnyList()
.tensorflow.CollectionDef.AnyList any_list = 5;
|
| CollectionDef.Builder |
clearBytesList()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
|
| CollectionDef.Builder |
clearField(com.google.protobuf.Descriptors.FieldDescriptor field)
|
| CollectionDef.Builder |
clearFloatList()
.tensorflow.CollectionDef.FloatList float_list = 4;
|
| CollectionDef.Builder |
clearInt64List()
.tensorflow.CollectionDef.Int64List int64_list = 3;
|
| CollectionDef.Builder | |
| CollectionDef.Builder |
clearNodeList()
.tensorflow.CollectionDef.NodeList node_list = 1;
|
| CollectionDef.Builder |
clearOneof(com.google.protobuf.Descriptors.OneofDescriptor oneof)
|
| CollectionDef.Builder |
clone()
|
| CollectionDef.AnyList |
getAnyList()
.tensorflow.CollectionDef.AnyList any_list = 5;
|
| CollectionDef.AnyList.Builder |
getAnyListBuilder()
.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.BytesList.Builder |
getBytesListBuilder()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
|
| CollectionDef.BytesListOrBuilder |
getBytesListOrBuilder()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
|
| CollectionDef | |
| final static com.google.protobuf.Descriptors.Descriptor | |
| com.google.protobuf.Descriptors.Descriptor | |
| CollectionDef.FloatList |
getFloatList()
.tensorflow.CollectionDef.FloatList float_list = 4;
|
| CollectionDef.FloatList.Builder |
getFloatListBuilder()
.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.Int64List.Builder |
getInt64ListBuilder()
.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.NodeList.Builder |
getNodeListBuilder()
.tensorflow.CollectionDef.NodeList node_list = 1;
|
| CollectionDef.NodeListOrBuilder |
getNodeListOrBuilder()
.tensorflow.CollectionDef.NodeList node_list = 1;
|
| 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;
|
| final boolean | |
| CollectionDef.Builder | |
| CollectionDef.Builder | |
| CollectionDef.Builder | |
| CollectionDef.Builder |
mergeFrom(com.google.protobuf.Message other)
|
| CollectionDef.Builder |
mergeFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
| CollectionDef.Builder | |
| CollectionDef.Builder | |
| final CollectionDef.Builder |
mergeUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)
|
| CollectionDef.Builder | |
| CollectionDef.Builder |
setAnyList(CollectionDef.AnyList.Builder builderForValue)
.tensorflow.CollectionDef.AnyList any_list = 5;
|
| CollectionDef.Builder |
setBytesList(CollectionDef.BytesList.Builder builderForValue)
.tensorflow.CollectionDef.BytesList bytes_list = 2;
|
| CollectionDef.Builder | |
| CollectionDef.Builder |
setField(com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
|
| CollectionDef.Builder | |
| CollectionDef.Builder |
setFloatList(CollectionDef.FloatList.Builder builderForValue)
.tensorflow.CollectionDef.FloatList float_list = 4;
|
| CollectionDef.Builder | |
| CollectionDef.Builder |
setInt64List(CollectionDef.Int64List.Builder builderForValue)
.tensorflow.CollectionDef.Int64List int64_list = 3;
|
| CollectionDef.Builder |
setNodeList(CollectionDef.NodeList.Builder builderForValue)
.tensorflow.CollectionDef.NodeList node_list = 1;
|
| CollectionDef.Builder | |
| CollectionDef.Builder |
setRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, int index, Object value)
|
| final CollectionDef.Builder |
setUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)
|
Inherited Methods
Public Methods
public CollectionDef.Builder addRepeatedField (com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
public CollectionDef.AnyList.Builder getAnyListBuilder ()
.tensorflow.CollectionDef.AnyList any_list = 5;
public CollectionDef.AnyListOrBuilder getAnyListOrBuilder ()
.tensorflow.CollectionDef.AnyList any_list = 5;
public CollectionDef.BytesList.Builder getBytesListBuilder ()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
public CollectionDef.BytesListOrBuilder getBytesListOrBuilder ()
.tensorflow.CollectionDef.BytesList bytes_list = 2;
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor ()
public com.google.protobuf.Descriptors.Descriptor getDescriptorForType ()
public CollectionDef.FloatList.Builder getFloatListBuilder ()
.tensorflow.CollectionDef.FloatList float_list = 4;
public CollectionDef.FloatListOrBuilder getFloatListOrBuilder ()
.tensorflow.CollectionDef.FloatList float_list = 4;
public CollectionDef.Int64List.Builder getInt64ListBuilder ()
.tensorflow.CollectionDef.Int64List int64_list = 3;
public CollectionDef.Int64ListOrBuilder getInt64ListOrBuilder ()
.tensorflow.CollectionDef.Int64List int64_list = 3;
public CollectionDef.NodeList.Builder getNodeListBuilder ()
.tensorflow.CollectionDef.NodeList node_list = 1;
public CollectionDef.NodeListOrBuilder getNodeListOrBuilder ()
.tensorflow.CollectionDef.NodeList node_list = 1;
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 final boolean isInitialized ()
public CollectionDef.Builder mergeAnyList (CollectionDef.AnyList value)
.tensorflow.CollectionDef.AnyList any_list = 5;
public CollectionDef.Builder mergeBytesList (CollectionDef.BytesList value)
.tensorflow.CollectionDef.BytesList bytes_list = 2;
public CollectionDef.Builder mergeFloatList (CollectionDef.FloatList value)
.tensorflow.CollectionDef.FloatList float_list = 4;
public CollectionDef.Builder mergeFrom (com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
| IOException |
|---|
public CollectionDef.Builder mergeInt64List (CollectionDef.Int64List value)
.tensorflow.CollectionDef.Int64List int64_list = 3;
public CollectionDef.Builder mergeNodeList (CollectionDef.NodeList value)
.tensorflow.CollectionDef.NodeList node_list = 1;
public final CollectionDef.Builder mergeUnknownFields (com.google.protobuf.UnknownFieldSet unknownFields)
public CollectionDef.Builder setAnyList (CollectionDef.AnyList value)
.tensorflow.CollectionDef.AnyList any_list = 5;
public CollectionDef.Builder setAnyList (CollectionDef.AnyList.Builder builderForValue)
.tensorflow.CollectionDef.AnyList any_list = 5;
public CollectionDef.Builder setBytesList (CollectionDef.BytesList.Builder builderForValue)
.tensorflow.CollectionDef.BytesList bytes_list = 2;
public CollectionDef.Builder setBytesList (CollectionDef.BytesList value)
.tensorflow.CollectionDef.BytesList bytes_list = 2;
public CollectionDef.Builder setField (com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
public CollectionDef.Builder setFloatList (CollectionDef.FloatList value)
.tensorflow.CollectionDef.FloatList float_list = 4;
public CollectionDef.Builder setFloatList (CollectionDef.FloatList.Builder builderForValue)
.tensorflow.CollectionDef.FloatList float_list = 4;
public CollectionDef.Builder setInt64List (CollectionDef.Int64List value)
.tensorflow.CollectionDef.Int64List int64_list = 3;
public CollectionDef.Builder setInt64List (CollectionDef.Int64List.Builder builderForValue)
.tensorflow.CollectionDef.Int64List int64_list = 3;
public CollectionDef.Builder setNodeList (CollectionDef.NodeList.Builder builderForValue)
.tensorflow.CollectionDef.NodeList node_list = 1;
public CollectionDef.Builder setNodeList (CollectionDef.NodeList value)
.tensorflow.CollectionDef.NodeList node_list = 1;