Package org.tensorflow.framework
Class CollectionDef
java.lang.Object
com.google.protobuf.AbstractMessageLite
com.google.protobuf.AbstractMessage
com.google.protobuf.GeneratedMessageV3
org.tensorflow.framework.CollectionDef
- All Implemented Interfaces:
com.google.protobuf.Message,com.google.protobuf.MessageLite,com.google.protobuf.MessageLiteOrBuilder,com.google.protobuf.MessageOrBuilder,Serializable,CollectionDefOrBuilder
public final class CollectionDef
extends com.google.protobuf.GeneratedMessageV3
implements CollectionDefOrBuilder
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- See Also:
-
Nested Class Summary
Nested ClassesModifier and TypeClassDescriptionstatic final classAnyList is used for collecting Any protos.static interfacestatic final classCollectionDef should cover most collections.static final classBytesList is used for collecting strings and serialized protobufs.static interfacestatic final classFloatList is used for collecting float values.static interfacestatic final classInt64List is used for collecting int, int64 and long values.static interfacestatic enumstatic final classNodeList is used for collecting nodes in graph.static interfaceNested classes/interfaces inherited from class com.google.protobuf.GeneratedMessageV3
com.google.protobuf.GeneratedMessageV3.BuilderParent, com.google.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageT extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage<MessageT>,BuilderT extends com.google.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageT, BuilderT>>, com.google.protobuf.GeneratedMessageV3.ExtendableMessage<MessageT extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage<MessageT>>, com.google.protobuf.GeneratedMessageV3.ExtendableMessageOrBuilder<MessageT extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage<MessageT>>, com.google.protobuf.GeneratedMessageV3.FieldAccessorTable, com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter Nested classes/interfaces inherited from class com.google.protobuf.AbstractMessageLite
com.google.protobuf.AbstractMessageLite.InternalOneOfEnum -
Field Summary
FieldsModifier and TypeFieldDescriptionstatic final intstatic final intstatic final intstatic final intstatic final intFields inherited from class com.google.protobuf.GeneratedMessageV3
alwaysUseFieldBuilders, unknownFieldsFields inherited from class com.google.protobuf.AbstractMessage
memoizedSizeFields inherited from class com.google.protobuf.AbstractMessageLite
memoizedHashCode -
Method Summary
Modifier and TypeMethodDescriptionboolean.tensorflow.CollectionDef.AnyList any_list = 5;.tensorflow.CollectionDef.AnyList any_list = 5;.tensorflow.CollectionDef.BytesList bytes_list = 2;.tensorflow.CollectionDef.BytesList bytes_list = 2;static CollectionDefstatic final com.google.protobuf.Descriptors.Descriptor.tensorflow.CollectionDef.FloatList float_list = 4;.tensorflow.CollectionDef.FloatList float_list = 4;.tensorflow.CollectionDef.Int64List int64_list = 3;.tensorflow.CollectionDef.Int64List int64_list = 3;.tensorflow.CollectionDef.NodeList node_list = 1;.tensorflow.CollectionDef.NodeList node_list = 1;com.google.protobuf.Parser<CollectionDef> intboolean.tensorflow.CollectionDef.AnyList any_list = 5;boolean.tensorflow.CollectionDef.BytesList bytes_list = 2;boolean.tensorflow.CollectionDef.FloatList float_list = 4;inthashCode()boolean.tensorflow.CollectionDef.Int64List int64_list = 3;boolean.tensorflow.CollectionDef.NodeList node_list = 1;protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTablefinal booleanstatic CollectionDef.Builderstatic CollectionDef.BuildernewBuilder(CollectionDef prototype) protected CollectionDef.BuildernewBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent) protected ObjectnewInstance(com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused) static CollectionDefparseDelimitedFrom(InputStream input) static CollectionDefparseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) static CollectionDefparseFrom(byte[] data) static CollectionDefparseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) static CollectionDefparseFrom(com.google.protobuf.ByteString data) static CollectionDefparseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) static CollectionDefparseFrom(com.google.protobuf.CodedInputStream input) static CollectionDefparseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) static CollectionDefparseFrom(InputStream input) static CollectionDefparseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) static CollectionDefparseFrom(ByteBuffer data) static CollectionDefparseFrom(ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) static com.google.protobuf.Parser<CollectionDef> parser()voidwriteTo(com.google.protobuf.CodedOutputStream output) Methods inherited from class com.google.protobuf.GeneratedMessageV3
canUseUnsafe, computeStringSize, computeStringSizeNoTag, emptyBooleanList, emptyDoubleList, emptyFloatList, emptyIntList, emptyList, emptyLongList, getAllFields, getDescriptorForType, getField, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, getUnknownFields, hasField, hasOneof, internalGetMapField, internalGetMapFieldReflection, isStringEmpty, makeExtensionsImmutable, makeMutableCopy, makeMutableCopy, mergeFromAndMakeImmutableInternal, mutableCopy, mutableCopy, mutableCopy, mutableCopy, mutableCopy, newBooleanList, newBuilderForType, newDoubleList, newFloatList, newIntList, newLongList, parseDelimitedWithIOException, parseDelimitedWithIOException, parseUnknownField, parseUnknownFieldProto3, parseWithIOException, parseWithIOException, parseWithIOException, parseWithIOException, serializeBooleanMapTo, serializeIntegerMapTo, serializeLongMapTo, serializeStringMapTo, writeReplace, writeString, writeStringNoTagMethods inherited from class com.google.protobuf.AbstractMessage
findInitializationErrors, getInitializationErrorString, hashBoolean, hashEnum, hashEnumList, hashFields, hashLong, toStringMethods inherited from class com.google.protobuf.AbstractMessageLite
addAll, addAll, checkByteStringIsUtf8, toByteArray, toByteString, writeDelimitedTo, writeToMethods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, wait, wait, waitMethods inherited from interface com.google.protobuf.MessageLite
toByteArray, toByteString, writeDelimitedTo, writeToMethods inherited from interface com.google.protobuf.MessageOrBuilder
findInitializationErrors, getAllFields, getDescriptorForType, getField, getInitializationErrorString, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, getUnknownFields, hasField, hasOneof
-
Field Details
-
NODE_LIST_FIELD_NUMBER
public static final int NODE_LIST_FIELD_NUMBER- See Also:
-
BYTES_LIST_FIELD_NUMBER
public static final int BYTES_LIST_FIELD_NUMBER- See Also:
-
INT64_LIST_FIELD_NUMBER
public static final int INT64_LIST_FIELD_NUMBER- See Also:
-
FLOAT_LIST_FIELD_NUMBER
public static final int FLOAT_LIST_FIELD_NUMBER- See Also:
-
ANY_LIST_FIELD_NUMBER
public static final int ANY_LIST_FIELD_NUMBER- See Also:
-
-
Method Details
-
newInstance
- Overrides:
newInstancein classcom.google.protobuf.GeneratedMessageV3
-
getDescriptor
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() -
internalGetFieldAccessorTable
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()- Specified by:
internalGetFieldAccessorTablein classcom.google.protobuf.GeneratedMessageV3
-
getKindCase
- Specified by:
getKindCasein interfaceCollectionDefOrBuilder
-
hasNodeList
public boolean hasNodeList().tensorflow.CollectionDef.NodeList node_list = 1;- Specified by:
hasNodeListin interfaceCollectionDefOrBuilder- Returns:
- Whether the nodeList field is set.
-
getNodeList
.tensorflow.CollectionDef.NodeList node_list = 1;- Specified by:
getNodeListin interfaceCollectionDefOrBuilder- Returns:
- The nodeList.
-
getNodeListOrBuilder
.tensorflow.CollectionDef.NodeList node_list = 1;- Specified by:
getNodeListOrBuilderin interfaceCollectionDefOrBuilder
-
hasBytesList
public boolean hasBytesList().tensorflow.CollectionDef.BytesList bytes_list = 2;- Specified by:
hasBytesListin interfaceCollectionDefOrBuilder- Returns:
- Whether the bytesList field is set.
-
getBytesList
.tensorflow.CollectionDef.BytesList bytes_list = 2;- Specified by:
getBytesListin interfaceCollectionDefOrBuilder- Returns:
- The bytesList.
-
getBytesListOrBuilder
.tensorflow.CollectionDef.BytesList bytes_list = 2;- Specified by:
getBytesListOrBuilderin interfaceCollectionDefOrBuilder
-
hasInt64List
public boolean hasInt64List().tensorflow.CollectionDef.Int64List int64_list = 3;- Specified by:
hasInt64Listin interfaceCollectionDefOrBuilder- Returns:
- Whether the int64List field is set.
-
getInt64List
.tensorflow.CollectionDef.Int64List int64_list = 3;- Specified by:
getInt64Listin interfaceCollectionDefOrBuilder- Returns:
- The int64List.
-
getInt64ListOrBuilder
.tensorflow.CollectionDef.Int64List int64_list = 3;- Specified by:
getInt64ListOrBuilderin interfaceCollectionDefOrBuilder
-
hasFloatList
public boolean hasFloatList().tensorflow.CollectionDef.FloatList float_list = 4;- Specified by:
hasFloatListin interfaceCollectionDefOrBuilder- Returns:
- Whether the floatList field is set.
-
getFloatList
.tensorflow.CollectionDef.FloatList float_list = 4;- Specified by:
getFloatListin interfaceCollectionDefOrBuilder- Returns:
- The floatList.
-
getFloatListOrBuilder
.tensorflow.CollectionDef.FloatList float_list = 4;- Specified by:
getFloatListOrBuilderin interfaceCollectionDefOrBuilder
-
hasAnyList
public boolean hasAnyList().tensorflow.CollectionDef.AnyList any_list = 5;- Specified by:
hasAnyListin interfaceCollectionDefOrBuilder- Returns:
- Whether the anyList field is set.
-
getAnyList
.tensorflow.CollectionDef.AnyList any_list = 5;- Specified by:
getAnyListin interfaceCollectionDefOrBuilder- Returns:
- The anyList.
-
getAnyListOrBuilder
.tensorflow.CollectionDef.AnyList any_list = 5;- Specified by:
getAnyListOrBuilderin interfaceCollectionDefOrBuilder
-
isInitialized
public final boolean isInitialized()- Specified by:
isInitializedin interfacecom.google.protobuf.MessageLiteOrBuilder- Overrides:
isInitializedin classcom.google.protobuf.GeneratedMessageV3
-
writeTo
- Specified by:
writeToin interfacecom.google.protobuf.MessageLite- Overrides:
writeToin classcom.google.protobuf.GeneratedMessageV3- Throws:
IOException
-
getSerializedSize
public int getSerializedSize()- Specified by:
getSerializedSizein interfacecom.google.protobuf.MessageLite- Overrides:
getSerializedSizein classcom.google.protobuf.GeneratedMessageV3
-
equals
- Specified by:
equalsin interfacecom.google.protobuf.Message- Overrides:
equalsin classcom.google.protobuf.AbstractMessage
-
hashCode
public int hashCode()- Specified by:
hashCodein interfacecom.google.protobuf.Message- Overrides:
hashCodein classcom.google.protobuf.AbstractMessage
-
parseFrom
public static CollectionDef parseFrom(ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException - Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static CollectionDef parseFrom(ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException - Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static CollectionDef parseFrom(com.google.protobuf.ByteString data) throws com.google.protobuf.InvalidProtocolBufferException - Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static CollectionDef parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException - Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static CollectionDef parseFrom(byte[] data) throws com.google.protobuf.InvalidProtocolBufferException - Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static CollectionDef parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException - Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
- Throws:
IOException
-
parseFrom
public static CollectionDef parseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException - Throws:
IOException
-
parseDelimitedFrom
- Throws:
IOException
-
parseDelimitedFrom
public static CollectionDef parseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException - Throws:
IOException
-
parseFrom
public static CollectionDef parseFrom(com.google.protobuf.CodedInputStream input) throws IOException - Throws:
IOException
-
parseFrom
public static CollectionDef parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException - Throws:
IOException
-
newBuilderForType
- Specified by:
newBuilderForTypein interfacecom.google.protobuf.Message- Specified by:
newBuilderForTypein interfacecom.google.protobuf.MessageLite
-
newBuilder
-
newBuilder
-
toBuilder
- Specified by:
toBuilderin interfacecom.google.protobuf.Message- Specified by:
toBuilderin interfacecom.google.protobuf.MessageLite
-
newBuilderForType
protected CollectionDef.Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent) - Specified by:
newBuilderForTypein classcom.google.protobuf.GeneratedMessageV3
-
getDefaultInstance
-
parser
-
getParserForType
- Specified by:
getParserForTypein interfacecom.google.protobuf.Message- Specified by:
getParserForTypein interfacecom.google.protobuf.MessageLite- Overrides:
getParserForTypein classcom.google.protobuf.GeneratedMessageV3
-
getDefaultInstanceForType
- Specified by:
getDefaultInstanceForTypein interfacecom.google.protobuf.MessageLiteOrBuilder- Specified by:
getDefaultInstanceForTypein interfacecom.google.protobuf.MessageOrBuilder
-