Package org.platanios.tensorflow.proto
Class KernelDef
- java.lang.Object
-
- com.google.protobuf.AbstractMessageLite
-
- com.google.protobuf.AbstractMessage
-
- com.google.protobuf.GeneratedMessageV3
-
- org.platanios.tensorflow.proto.KernelDef
-
- All Implemented Interfaces:
com.google.protobuf.Message
,com.google.protobuf.MessageLite
,com.google.protobuf.MessageLiteOrBuilder
,com.google.protobuf.MessageOrBuilder
,java.io.Serializable
,KernelDefOrBuilder
public final class KernelDef extends com.google.protobuf.GeneratedMessageV3 implements KernelDefOrBuilder
Protobuf typeorg.platanios.tensorflow.proto.KernelDef
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
KernelDef.AttrConstraint
Protobuf typeorg.platanios.tensorflow.proto.KernelDef.AttrConstraint
static interface
KernelDef.AttrConstraintOrBuilder
static class
KernelDef.Builder
Protobuf typeorg.platanios.tensorflow.proto.KernelDef
-
Nested classes/interfaces inherited from class com.google.protobuf.GeneratedMessageV3
com.google.protobuf.GeneratedMessageV3.BuilderParent, com.google.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageType extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage,BuilderType extends com.google.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageType,BuilderType>>, com.google.protobuf.GeneratedMessageV3.ExtendableMessage<MessageType extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage>, com.google.protobuf.GeneratedMessageV3.ExtendableMessageOrBuilder<MessageType extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage>, com.google.protobuf.GeneratedMessageV3.FieldAccessorTable, com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter
-
-
Field Summary
Fields Modifier and Type Field Description static int
CONSTRAINT_FIELD_NUMBER
static int
DEVICE_TYPE_FIELD_NUMBER
static int
HOST_MEMORY_ARG_FIELD_NUMBER
static int
LABEL_FIELD_NUMBER
static int
OP_FIELD_NUMBER
static int
PRIORITY_FIELD_NUMBER
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description boolean
equals(java.lang.Object obj)
KernelDef.AttrConstraint
getConstraint(int index)
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
int
getConstraintCount()
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
java.util.List<KernelDef.AttrConstraint>
getConstraintList()
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
KernelDef.AttrConstraintOrBuilder
getConstraintOrBuilder(int index)
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
java.util.List<? extends KernelDef.AttrConstraintOrBuilder>
getConstraintOrBuilderList()
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
static KernelDef
getDefaultInstance()
KernelDef
getDefaultInstanceForType()
static com.google.protobuf.Descriptors.Descriptor
getDescriptor()
java.lang.String
getDeviceType()
Type of device this kernel runs on.com.google.protobuf.ByteString
getDeviceTypeBytes()
Type of device this kernel runs on.java.lang.String
getHostMemoryArg(int index)
Names of the Op's input_/output_args that reside in host memory instead of device memory.com.google.protobuf.ByteString
getHostMemoryArgBytes(int index)
Names of the Op's input_/output_args that reside in host memory instead of device memory.int
getHostMemoryArgCount()
Names of the Op's input_/output_args that reside in host memory instead of device memory.com.google.protobuf.ProtocolStringList
getHostMemoryArgList()
Names of the Op's input_/output_args that reside in host memory instead of device memory.java.lang.String
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.com.google.protobuf.ByteString
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.java.lang.String
getOp()
Must match the name of an Op.com.google.protobuf.ByteString
getOpBytes()
Must match the name of an Op.com.google.protobuf.Parser<KernelDef>
getParserForType()
int
getPriority()
Prioritization of kernel amongst different devices.int
getSerializedSize()
com.google.protobuf.UnknownFieldSet
getUnknownFields()
int
hashCode()
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
internalGetFieldAccessorTable()
boolean
isInitialized()
static KernelDef.Builder
newBuilder()
static KernelDef.Builder
newBuilder(KernelDef prototype)
KernelDef.Builder
newBuilderForType()
protected KernelDef.Builder
newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent)
protected java.lang.Object
newInstance(com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)
static KernelDef
parseDelimitedFrom(java.io.InputStream input)
static KernelDef
parseDelimitedFrom(java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(byte[] data)
static KernelDef
parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(com.google.protobuf.ByteString data)
static KernelDef
parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(com.google.protobuf.CodedInputStream input)
static KernelDef
parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(java.io.InputStream input)
static KernelDef
parseFrom(java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(java.nio.ByteBuffer data)
static KernelDef
parseFrom(java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static com.google.protobuf.Parser<KernelDef>
parser()
KernelDef.Builder
toBuilder()
void
writeTo(com.google.protobuf.CodedOutputStream output)
-
Methods inherited from class com.google.protobuf.GeneratedMessageV3
canUseUnsafe, computeStringSize, computeStringSizeNoTag, emptyBooleanList, emptyDoubleList, emptyFloatList, emptyIntList, emptyLongList, getAllFields, getDescriptorForType, getField, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, hasField, hasOneof, internalGetMapField, 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, writeStringNoTag
-
Methods inherited from class com.google.protobuf.AbstractMessage
findInitializationErrors, getInitializationErrorString, hashBoolean, hashEnum, hashEnumList, hashFields, hashLong, toString
-
Methods inherited from class com.google.protobuf.AbstractMessageLite
addAll, addAll, checkByteStringIsUtf8, toByteArray, toByteString, writeDelimitedTo, writeTo
-
Methods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
-
-
-
-
Field Detail
-
OP_FIELD_NUMBER
public static final int OP_FIELD_NUMBER
- See Also:
- Constant Field Values
-
DEVICE_TYPE_FIELD_NUMBER
public static final int DEVICE_TYPE_FIELD_NUMBER
- See Also:
- Constant Field Values
-
CONSTRAINT_FIELD_NUMBER
public static final int CONSTRAINT_FIELD_NUMBER
- See Also:
- Constant Field Values
-
HOST_MEMORY_ARG_FIELD_NUMBER
public static final int HOST_MEMORY_ARG_FIELD_NUMBER
- See Also:
- Constant Field Values
-
LABEL_FIELD_NUMBER
public static final int LABEL_FIELD_NUMBER
- See Also:
- Constant Field Values
-
PRIORITY_FIELD_NUMBER
public static final int PRIORITY_FIELD_NUMBER
- See Also:
- Constant Field Values
-
-
Method Detail
-
newInstance
protected java.lang.Object newInstance(com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)
- Overrides:
newInstance
in classcom.google.protobuf.GeneratedMessageV3
-
getUnknownFields
public final com.google.protobuf.UnknownFieldSet getUnknownFields()
- Specified by:
getUnknownFields
in interfacecom.google.protobuf.MessageOrBuilder
- Overrides:
getUnknownFields
in classcom.google.protobuf.GeneratedMessageV3
-
getDescriptor
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor()
-
internalGetFieldAccessorTable
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
- Specified by:
internalGetFieldAccessorTable
in classcom.google.protobuf.GeneratedMessageV3
-
getOp
public java.lang.String getOp()
Must match the name of an Op.
string op = 1;
- Specified by:
getOp
in interfaceKernelDefOrBuilder
- Returns:
- The op.
-
getOpBytes
public com.google.protobuf.ByteString getOpBytes()
Must match the name of an Op.
string op = 1;
- Specified by:
getOpBytes
in interfaceKernelDefOrBuilder
- Returns:
- The bytes for op.
-
getDeviceType
public java.lang.String getDeviceType()
Type of device this kernel runs on.
string device_type = 2;
- Specified by:
getDeviceType
in interfaceKernelDefOrBuilder
- Returns:
- The deviceType.
-
getDeviceTypeBytes
public com.google.protobuf.ByteString getDeviceTypeBytes()
Type of device this kernel runs on.
string device_type = 2;
- Specified by:
getDeviceTypeBytes
in interfaceKernelDefOrBuilder
- Returns:
- The bytes for deviceType.
-
getConstraintList
public java.util.List<KernelDef.AttrConstraint> getConstraintList()
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraintList
in interfaceKernelDefOrBuilder
-
getConstraintOrBuilderList
public java.util.List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList()
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraintOrBuilderList
in interfaceKernelDefOrBuilder
-
getConstraintCount
public int getConstraintCount()
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraintCount
in interfaceKernelDefOrBuilder
-
getConstraint
public KernelDef.AttrConstraint getConstraint(int index)
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraint
in interfaceKernelDefOrBuilder
-
getConstraintOrBuilder
public KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder(int index)
repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraintOrBuilder
in interfaceKernelDefOrBuilder
-
getHostMemoryArgList
public com.google.protobuf.ProtocolStringList 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:
getHostMemoryArgList
in 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:
getHostMemoryArgCount
in interfaceKernelDefOrBuilder
- Returns:
- The count of hostMemoryArg.
-
getHostMemoryArg
public java.lang.String getHostMemoryArg(int index)
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:
getHostMemoryArg
in interfaceKernelDefOrBuilder
- Parameters:
index
- The index of the element to return.- Returns:
- The hostMemoryArg at the given index.
-
getHostMemoryArgBytes
public com.google.protobuf.ByteString getHostMemoryArgBytes(int index)
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:
getHostMemoryArgBytes
in interfaceKernelDefOrBuilder
- Parameters:
index
- The index of the value to return.- Returns:
- The bytes of the hostMemoryArg at the given index.
-
getLabel
public java.lang.String 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:
getLabel
in interfaceKernelDefOrBuilder
- Returns:
- The label.
-
getLabelBytes
public com.google.protobuf.ByteString 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:
getLabelBytes
in 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:
getPriority
in interfaceKernelDefOrBuilder
- Returns:
- The priority.
-
isInitialized
public final boolean isInitialized()
- Specified by:
isInitialized
in interfacecom.google.protobuf.MessageLiteOrBuilder
- Overrides:
isInitialized
in classcom.google.protobuf.GeneratedMessageV3
-
writeTo
public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException
- Specified by:
writeTo
in interfacecom.google.protobuf.MessageLite
- Overrides:
writeTo
in classcom.google.protobuf.GeneratedMessageV3
- Throws:
java.io.IOException
-
getSerializedSize
public int getSerializedSize()
- Specified by:
getSerializedSize
in interfacecom.google.protobuf.MessageLite
- Overrides:
getSerializedSize
in classcom.google.protobuf.GeneratedMessageV3
-
equals
public boolean equals(java.lang.Object obj)
- Specified by:
equals
in interfacecom.google.protobuf.Message
- Overrides:
equals
in classcom.google.protobuf.AbstractMessage
-
hashCode
public int hashCode()
- Specified by:
hashCode
in interfacecom.google.protobuf.Message
- Overrides:
hashCode
in classcom.google.protobuf.AbstractMessage
-
parseFrom
public static KernelDef parseFrom(java.nio.ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException
- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(java.nio.ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(com.google.protobuf.ByteString data) throws com.google.protobuf.InvalidProtocolBufferException
- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(byte[] data) throws com.google.protobuf.InvalidProtocolBufferException
- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
- Throws:
com.google.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(java.io.InputStream input) throws java.io.IOException
- Throws:
java.io.IOException
-
parseFrom
public static KernelDef parseFrom(java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws java.io.IOException
- Throws:
java.io.IOException
-
parseDelimitedFrom
public static KernelDef parseDelimitedFrom(java.io.InputStream input) throws java.io.IOException
- Throws:
java.io.IOException
-
parseDelimitedFrom
public static KernelDef parseDelimitedFrom(java.io.InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws java.io.IOException
- Throws:
java.io.IOException
-
parseFrom
public static KernelDef parseFrom(com.google.protobuf.CodedInputStream input) throws java.io.IOException
- Throws:
java.io.IOException
-
parseFrom
public static KernelDef parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws java.io.IOException
- Throws:
java.io.IOException
-
newBuilderForType
public KernelDef.Builder newBuilderForType()
- Specified by:
newBuilderForType
in interfacecom.google.protobuf.Message
- Specified by:
newBuilderForType
in interfacecom.google.protobuf.MessageLite
-
newBuilder
public static KernelDef.Builder newBuilder()
-
newBuilder
public static KernelDef.Builder newBuilder(KernelDef prototype)
-
toBuilder
public KernelDef.Builder toBuilder()
- Specified by:
toBuilder
in interfacecom.google.protobuf.Message
- Specified by:
toBuilder
in interfacecom.google.protobuf.MessageLite
-
newBuilderForType
protected KernelDef.Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent)
- Specified by:
newBuilderForType
in classcom.google.protobuf.GeneratedMessageV3
-
getDefaultInstance
public static KernelDef getDefaultInstance()
-
parser
public static com.google.protobuf.Parser<KernelDef> parser()
-
getParserForType
public com.google.protobuf.Parser<KernelDef> getParserForType()
- Specified by:
getParserForType
in interfacecom.google.protobuf.Message
- Specified by:
getParserForType
in interfacecom.google.protobuf.MessageLite
- Overrides:
getParserForType
in classcom.google.protobuf.GeneratedMessageV3
-
getDefaultInstanceForType
public KernelDef getDefaultInstanceForType()
- Specified by:
getDefaultInstanceForType
in interfacecom.google.protobuf.MessageLiteOrBuilder
- Specified by:
getDefaultInstanceForType
in interfacecom.google.protobuf.MessageOrBuilder
-
-