public static final class
CollectionDef.Builder
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)Protobuf type
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)
|