Class KernelDef
java.lang.Object
com.google.protobuf.AbstractMessageLite
com.google.protobuf.AbstractMessage
com.google.protobuf.GeneratedMessageV3
org.tensorflow.proto.KernelDef
- All Implemented Interfaces:
Message, MessageLite, MessageLiteOrBuilder, MessageOrBuilder, Serializable, KernelDefOrBuilder
Protobuf type
tensorflow.KernelDef- See Also:
-
Nested Class Summary
Nested ClassesModifier and TypeClassDescriptionstatic final classProtobuf typetensorflow.KernelDef.AttrConstraintstatic interfacestatic final classProtobuf typetensorflow.KernelDefNested classes/interfaces inherited from class GeneratedMessageV3
GeneratedMessageV3.BuilderParent, GeneratedMessageV3.ExtendableBuilder<MessageType, BuilderType>, GeneratedMessageV3.ExtendableMessage<MessageType>, GeneratedMessageV3.ExtendableMessageOrBuilder<MessageType>, GeneratedMessageV3.FieldAccessorTable, GeneratedMessageV3.UnusedPrivateParameterNested classes/interfaces inherited from class AbstractMessageLite
AbstractMessageLite.InternalOneOfEnum -
Field Summary
FieldsModifier and TypeFieldDescriptionstatic final intstatic final intstatic final intstatic final intstatic final intstatic final intFields inherited from class GeneratedMessageV3
alwaysUseFieldBuilders, unknownFieldsFields inherited from class AbstractMessage
memoizedSizeFields inherited from class AbstractMessageLite
memoizedHashCode -
Method Summary
Modifier and TypeMethodDescriptionbooleangetConstraint(int index) repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;intrepeated .tensorflow.KernelDef.AttrConstraint constraint = 3;repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;getConstraintOrBuilder(int index) repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;List<? extends KernelDef.AttrConstraintOrBuilder> repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;static KernelDefstatic final Descriptors.DescriptorType of device this kernel runs on.Type of device this kernel runs on.getHostMemoryArg(int index) Names of the Op's input_/output_args that reside in host memory instead of device memory.getHostMemoryArgBytes(int index) Names of the Op's input_/output_args that reside in host memory instead of device memory.intNames of the Op's input_/output_args that reside in host memory instead of device memory.Names of the Op's input_/output_args that reside in host memory instead of device memory.getLabel()This allows experimental kernels to be registered for an op that won't be used unless the user specifies a "_kernel" attr with value matching this.This allows experimental kernels to be registered for an op that won't be used unless the user specifies a "_kernel" attr with value matching this.getOp()Must match the name of an Op.Must match the name of an Op.intPrioritization of kernel amongst different devices.intfinal UnknownFieldSetinthashCode()protected GeneratedMessageV3.FieldAccessorTablefinal booleanstatic KernelDef.Builderstatic KernelDef.BuildernewBuilder(KernelDef prototype) protected KernelDef.Builderprotected Objectstatic KernelDefparseDelimitedFrom(InputStream input) static KernelDefparseDelimitedFrom(InputStream input, ExtensionRegistryLite extensionRegistry) static KernelDefparseFrom(byte[] data) static KernelDefparseFrom(byte[] data, ExtensionRegistryLite extensionRegistry) static KernelDefparseFrom(ByteString data) static KernelDefparseFrom(ByteString data, ExtensionRegistryLite extensionRegistry) static KernelDefparseFrom(CodedInputStream input) static KernelDefparseFrom(CodedInputStream input, ExtensionRegistryLite extensionRegistry) static KernelDefparseFrom(InputStream input) static KernelDefparseFrom(InputStream input, ExtensionRegistryLite extensionRegistry) static KernelDefparseFrom(ByteBuffer data) static KernelDefparseFrom(ByteBuffer data, ExtensionRegistryLite extensionRegistry) parser()voidwriteTo(CodedOutputStream output) Methods inherited from class GeneratedMessageV3
canUseUnsafe, computeStringSize, computeStringSizeNoTag, emptyBooleanList, emptyDoubleList, emptyFloatList, emptyIntList, emptyLongList, getAllFields, getDescriptorForType, getField, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, hasField, hasOneof, internalGetMapField, isStringEmpty, makeExtensionsImmutable, 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 AbstractMessage
findInitializationErrors, getInitializationErrorString, hashBoolean, hashEnum, hashEnumList, hashFields, hashLong, toStringMethods inherited from class AbstractMessageLite
addAll, addAll, checkByteStringIsUtf8, toByteArray, toByteString, writeDelimitedTo, writeToMethods inherited from interface MessageLite
toByteArray, toByteString, writeDelimitedTo, writeToMethods inherited from interface MessageOrBuilder
findInitializationErrors, getAllFields, getDescriptorForType, getField, getInitializationErrorString, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, hasField, hasOneof
-
Field Details
-
OP_FIELD_NUMBER
public static final int OP_FIELD_NUMBER- See Also:
-
DEVICE_TYPE_FIELD_NUMBER
public static final int DEVICE_TYPE_FIELD_NUMBER- See Also:
-
CONSTRAINT_FIELD_NUMBER
public static final int CONSTRAINT_FIELD_NUMBER- See Also:
-
HOST_MEMORY_ARG_FIELD_NUMBER
public static final int HOST_MEMORY_ARG_FIELD_NUMBER- See Also:
-
LABEL_FIELD_NUMBER
public static final int LABEL_FIELD_NUMBER- See Also:
-
PRIORITY_FIELD_NUMBER
public static final int PRIORITY_FIELD_NUMBER- See Also:
-
-
Method Details
-
newInstance
- Overrides:
newInstancein classGeneratedMessageV3
-
getUnknownFields
- Specified by:
getUnknownFieldsin interfaceMessageOrBuilder- Overrides:
getUnknownFieldsin classGeneratedMessageV3
-
getDescriptor
-
internalGetFieldAccessorTable
- Specified by:
internalGetFieldAccessorTablein classGeneratedMessageV3
-
getOp
Must match the name of an Op.
string op = 1;- Specified by:
getOpin interfaceKernelDefOrBuilder- Returns:
- The op.
-
getOpBytes
Must match the name of an Op.
string op = 1;- Specified by:
getOpBytesin interfaceKernelDefOrBuilder- Returns:
- The bytes for op.
-
getDeviceType
Type of device this kernel runs on.
string device_type = 2;- Specified by:
getDeviceTypein interfaceKernelDefOrBuilder- Returns:
- The deviceType.
-
getDeviceTypeBytes
Type of device this kernel runs on.
string device_type = 2;- Specified by:
getDeviceTypeBytesin interfaceKernelDefOrBuilder- Returns:
- The bytes for deviceType.
-
getConstraintList
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;- Specified by:
getConstraintListin interfaceKernelDefOrBuilder
-
getConstraintOrBuilderList
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;- Specified by:
getConstraintOrBuilderListin interfaceKernelDefOrBuilder
-
getConstraintCount
public int getConstraintCount()repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;- Specified by:
getConstraintCountin interfaceKernelDefOrBuilder
-
getConstraint
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;- Specified by:
getConstraintin interfaceKernelDefOrBuilder
-
getConstraintOrBuilder
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;- Specified by:
getConstraintOrBuilderin interfaceKernelDefOrBuilder
-
getHostMemoryArgList
Names of the Op's input_/output_args that reside in host memory instead of device memory.
repeated string host_memory_arg = 4;- Specified by:
getHostMemoryArgListin interfaceKernelDefOrBuilder- Returns:
- A list containing the hostMemoryArg.
-
getHostMemoryArgCount
public int getHostMemoryArgCount()Names of the Op's input_/output_args that reside in host memory instead of device memory.
repeated string host_memory_arg = 4;- Specified by:
getHostMemoryArgCountin interfaceKernelDefOrBuilder- Returns:
- The count of hostMemoryArg.
-
getHostMemoryArg
Names of the Op's input_/output_args that reside in host memory instead of device memory.
repeated string host_memory_arg = 4;- Specified by:
getHostMemoryArgin interfaceKernelDefOrBuilder- Parameters:
index- The index of the element to return.- Returns:
- The hostMemoryArg at the given index.
-
getHostMemoryArgBytes
Names of the Op's input_/output_args that reside in host memory instead of device memory.
repeated string host_memory_arg = 4;- Specified by:
getHostMemoryArgBytesin interfaceKernelDefOrBuilder- Parameters:
index- The index of the value to return.- Returns:
- The bytes of the hostMemoryArg at the given index.
-
getLabel
This allows experimental kernels to be registered for an op that won't be used unless the user specifies a "_kernel" attr with value matching this.
string label = 5;- Specified by:
getLabelin interfaceKernelDefOrBuilder- Returns:
- The label.
-
getLabelBytes
This allows experimental kernels to be registered for an op that won't be used unless the user specifies a "_kernel" attr with value matching this.
string label = 5;- Specified by:
getLabelBytesin interfaceKernelDefOrBuilder- Returns:
- The bytes for label.
-
getPriority
public int getPriority()Prioritization of kernel amongst different devices. By default we assume priority is 0. The higher the priority the better. By default (i.e. if this is not set), we prefer GPU kernels over CPU.
int32 priority = 6;- Specified by:
getPriorityin interfaceKernelDefOrBuilder- Returns:
- The priority.
-
isInitialized
public final boolean isInitialized()- Specified by:
isInitializedin interfaceMessageLiteOrBuilder- Overrides:
isInitializedin classGeneratedMessageV3
-
writeTo
- Specified by:
writeToin interfaceMessageLite- Overrides:
writeToin classGeneratedMessageV3- Throws:
IOException
-
getSerializedSize
public int getSerializedSize()- Specified by:
getSerializedSizein interfaceMessageLite- Overrides:
getSerializedSizein classGeneratedMessageV3
-
equals
- Specified by:
equalsin interfaceMessage- Overrides:
equalsin classAbstractMessage
-
hashCode
public int hashCode()- Specified by:
hashCodein interfaceMessage- Overrides:
hashCodein classAbstractMessage
-
parseFrom
- Throws:
InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(ByteBuffer data, ExtensionRegistryLite extensionRegistry) throws InvalidProtocolBufferException - Throws:
InvalidProtocolBufferException
-
parseFrom
- Throws:
InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(ByteString data, ExtensionRegistryLite extensionRegistry) throws InvalidProtocolBufferException - Throws:
InvalidProtocolBufferException
-
parseFrom
- Throws:
InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(byte[] data, ExtensionRegistryLite extensionRegistry) throws InvalidProtocolBufferException - Throws:
InvalidProtocolBufferException
-
parseFrom
- Throws:
IOException
-
parseFrom
public static KernelDef parseFrom(InputStream input, ExtensionRegistryLite extensionRegistry) throws IOException - Throws:
IOException
-
parseDelimitedFrom
- Throws:
IOException
-
parseDelimitedFrom
public static KernelDef parseDelimitedFrom(InputStream input, ExtensionRegistryLite extensionRegistry) throws IOException - Throws:
IOException
-
parseFrom
- Throws:
IOException
-
parseFrom
public static KernelDef parseFrom(CodedInputStream input, ExtensionRegistryLite extensionRegistry) throws IOException - Throws:
IOException
-
newBuilderForType
- Specified by:
newBuilderForTypein interfaceMessage- Specified by:
newBuilderForTypein interfaceMessageLite
-
newBuilder
-
newBuilder
-
toBuilder
- Specified by:
toBuilderin interfaceMessage- Specified by:
toBuilderin interfaceMessageLite
-
newBuilderForType
- Specified by:
newBuilderForTypein classGeneratedMessageV3
-
getDefaultInstance
-
parser
-
getParserForType
- Specified by:
getParserForTypein interfaceMessage- Specified by:
getParserForTypein interfaceMessageLite- Overrides:
getParserForTypein classGeneratedMessageV3
-
getDefaultInstanceForType
- Specified by:
getDefaultInstanceForTypein interfaceMessageLiteOrBuilder- Specified by:
getDefaultInstanceForTypein interfaceMessageOrBuilder
-