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;