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 |
---|