Package org.tensorflow.framework
Class KernelDef
- java.lang.Object
-
- org.nd4j.shade.protobuf.AbstractMessageLite
-
- org.nd4j.shade.protobuf.AbstractMessage
-
- org.nd4j.shade.protobuf.GeneratedMessageV3
-
- org.tensorflow.framework.KernelDef
-
- All Implemented Interfaces:
Serializable
,org.nd4j.shade.protobuf.Message
,org.nd4j.shade.protobuf.MessageLite
,org.nd4j.shade.protobuf.MessageLiteOrBuilder
,org.nd4j.shade.protobuf.MessageOrBuilder
,KernelDefOrBuilder
public final class KernelDef extends org.nd4j.shade.protobuf.GeneratedMessageV3 implements KernelDefOrBuilder
Protobuf typetensorflow.KernelDef
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
KernelDef.AttrConstraint
Protobuf typetensorflow.KernelDef.AttrConstraint
static interface
KernelDef.AttrConstraintOrBuilder
static class
KernelDef.Builder
Protobuf typetensorflow.KernelDef
-
Nested classes/interfaces inherited from class org.nd4j.shade.protobuf.GeneratedMessageV3
org.nd4j.shade.protobuf.GeneratedMessageV3.BuilderParent, org.nd4j.shade.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageType extends org.nd4j.shade.protobuf.GeneratedMessageV3.ExtendableMessage,BuilderType extends org.nd4j.shade.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageType,BuilderType>>, org.nd4j.shade.protobuf.GeneratedMessageV3.ExtendableMessage<MessageType extends org.nd4j.shade.protobuf.GeneratedMessageV3.ExtendableMessage>, org.nd4j.shade.protobuf.GeneratedMessageV3.ExtendableMessageOrBuilder<MessageType extends org.nd4j.shade.protobuf.GeneratedMessageV3.ExtendableMessage>, org.nd4j.shade.protobuf.GeneratedMessageV3.FieldAccessorTable, org.nd4j.shade.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
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description boolean
equals(Object obj)
KernelDef.AttrConstraint
getConstraint(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
int
getConstraintCount()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
List<KernelDef.AttrConstraint>
getConstraintList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
KernelDef.AttrConstraintOrBuilder
getConstraintOrBuilder(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
List<? extends KernelDef.AttrConstraintOrBuilder>
getConstraintOrBuilderList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
static KernelDef
getDefaultInstance()
KernelDef
getDefaultInstanceForType()
static org.nd4j.shade.protobuf.Descriptors.Descriptor
getDescriptor()
String
getDeviceType()
Type of device this kernel runs on.org.nd4j.shade.protobuf.ByteString
getDeviceTypeBytes()
Type of device this kernel runs on.String
getHostMemoryArg(int index)
Names of the Op's input_/output_args that reside in host memory instead of device memory.org.nd4j.shade.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.org.nd4j.shade.protobuf.ProtocolStringList
getHostMemoryArgList()
Names of the Op's input_/output_args that reside in host memory instead of device memory.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.org.nd4j.shade.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
getOp()
Must match the name of an Op.org.nd4j.shade.protobuf.ByteString
getOpBytes()
Must match the name of an Op.org.nd4j.shade.protobuf.Parser<KernelDef>
getParserForType()
int
getSerializedSize()
org.nd4j.shade.protobuf.UnknownFieldSet
getUnknownFields()
int
hashCode()
protected org.nd4j.shade.protobuf.GeneratedMessageV3.FieldAccessorTable
internalGetFieldAccessorTable()
boolean
isInitialized()
static KernelDef.Builder
newBuilder()
static KernelDef.Builder
newBuilder(KernelDef prototype)
KernelDef.Builder
newBuilderForType()
protected KernelDef.Builder
newBuilderForType(org.nd4j.shade.protobuf.GeneratedMessageV3.BuilderParent parent)
protected Object
newInstance(org.nd4j.shade.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)
static KernelDef
parseDelimitedFrom(InputStream input)
static KernelDef
parseDelimitedFrom(InputStream input, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(byte[] data)
static KernelDef
parseFrom(byte[] data, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(InputStream input)
static KernelDef
parseFrom(InputStream input, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(ByteBuffer data)
static KernelDef
parseFrom(ByteBuffer data, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(org.nd4j.shade.protobuf.ByteString data)
static KernelDef
parseFrom(org.nd4j.shade.protobuf.ByteString data, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
static KernelDef
parseFrom(org.nd4j.shade.protobuf.CodedInputStream input)
static KernelDef
parseFrom(org.nd4j.shade.protobuf.CodedInputStream input, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
static org.nd4j.shade.protobuf.Parser<KernelDef>
parser()
KernelDef.Builder
toBuilder()
void
writeTo(org.nd4j.shade.protobuf.CodedOutputStream output)
-
Methods inherited from class org.nd4j.shade.protobuf.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, writeStringNoTag
-
Methods inherited from class org.nd4j.shade.protobuf.AbstractMessage
findInitializationErrors, getInitializationErrorString, hashBoolean, hashEnum, hashEnumList, hashFields, hashLong, toString
-
Methods inherited from class org.nd4j.shade.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
-
-
Method Detail
-
newInstance
protected Object newInstance(org.nd4j.shade.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)
- Overrides:
newInstance
in classorg.nd4j.shade.protobuf.GeneratedMessageV3
-
getUnknownFields
public final org.nd4j.shade.protobuf.UnknownFieldSet getUnknownFields()
- Specified by:
getUnknownFields
in interfaceorg.nd4j.shade.protobuf.MessageOrBuilder
- Overrides:
getUnknownFields
in classorg.nd4j.shade.protobuf.GeneratedMessageV3
-
getDescriptor
public static final org.nd4j.shade.protobuf.Descriptors.Descriptor getDescriptor()
-
internalGetFieldAccessorTable
protected org.nd4j.shade.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
- Specified by:
internalGetFieldAccessorTable
in classorg.nd4j.shade.protobuf.GeneratedMessageV3
-
getOp
public String getOp()
Must match the name of an Op.
string op = 1;
- Specified by:
getOp
in interfaceKernelDefOrBuilder
- Returns:
- The op.
-
getOpBytes
public org.nd4j.shade.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 String getDeviceType()
Type of device this kernel runs on.
string device_type = 2;
- Specified by:
getDeviceType
in interfaceKernelDefOrBuilder
- Returns:
- The deviceType.
-
getDeviceTypeBytes
public org.nd4j.shade.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 List<KernelDef.AttrConstraint> getConstraintList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraintList
in interfaceKernelDefOrBuilder
-
getConstraintOrBuilderList
public List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraintOrBuilderList
in interfaceKernelDefOrBuilder
-
getConstraintCount
public int getConstraintCount()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraintCount
in interfaceKernelDefOrBuilder
-
getConstraint
public KernelDef.AttrConstraint getConstraint(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraint
in interfaceKernelDefOrBuilder
-
getConstraintOrBuilder
public KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
- Specified by:
getConstraintOrBuilder
in interfaceKernelDefOrBuilder
-
getHostMemoryArgList
public org.nd4j.shade.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 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 org.nd4j.shade.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 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 org.nd4j.shade.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.
-
isInitialized
public final boolean isInitialized()
- Specified by:
isInitialized
in interfaceorg.nd4j.shade.protobuf.MessageLiteOrBuilder
- Overrides:
isInitialized
in classorg.nd4j.shade.protobuf.GeneratedMessageV3
-
writeTo
public void writeTo(org.nd4j.shade.protobuf.CodedOutputStream output) throws IOException
- Specified by:
writeTo
in interfaceorg.nd4j.shade.protobuf.MessageLite
- Overrides:
writeTo
in classorg.nd4j.shade.protobuf.GeneratedMessageV3
- Throws:
IOException
-
getSerializedSize
public int getSerializedSize()
- Specified by:
getSerializedSize
in interfaceorg.nd4j.shade.protobuf.MessageLite
- Overrides:
getSerializedSize
in classorg.nd4j.shade.protobuf.GeneratedMessageV3
-
equals
public boolean equals(Object obj)
- Specified by:
equals
in interfaceorg.nd4j.shade.protobuf.Message
- Overrides:
equals
in classorg.nd4j.shade.protobuf.AbstractMessage
-
hashCode
public int hashCode()
- Specified by:
hashCode
in interfaceorg.nd4j.shade.protobuf.Message
- Overrides:
hashCode
in classorg.nd4j.shade.protobuf.AbstractMessage
-
parseFrom
public static KernelDef parseFrom(ByteBuffer data) throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
- Throws:
org.nd4j.shade.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(ByteBuffer data, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry) throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
- Throws:
org.nd4j.shade.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(org.nd4j.shade.protobuf.ByteString data) throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
- Throws:
org.nd4j.shade.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(org.nd4j.shade.protobuf.ByteString data, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry) throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
- Throws:
org.nd4j.shade.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(byte[] data) throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
- Throws:
org.nd4j.shade.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(byte[] data, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry) throws org.nd4j.shade.protobuf.InvalidProtocolBufferException
- Throws:
org.nd4j.shade.protobuf.InvalidProtocolBufferException
-
parseFrom
public static KernelDef parseFrom(InputStream input) throws IOException
- Throws:
IOException
-
parseFrom
public static KernelDef parseFrom(InputStream input, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
- Throws:
IOException
-
parseDelimitedFrom
public static KernelDef parseDelimitedFrom(InputStream input) throws IOException
- Throws:
IOException
-
parseDelimitedFrom
public static KernelDef parseDelimitedFrom(InputStream input, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
- Throws:
IOException
-
parseFrom
public static KernelDef parseFrom(org.nd4j.shade.protobuf.CodedInputStream input) throws IOException
- Throws:
IOException
-
parseFrom
public static KernelDef parseFrom(org.nd4j.shade.protobuf.CodedInputStream input, org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
- Throws:
IOException
-
newBuilderForType
public KernelDef.Builder newBuilderForType()
- Specified by:
newBuilderForType
in interfaceorg.nd4j.shade.protobuf.Message
- Specified by:
newBuilderForType
in interfaceorg.nd4j.shade.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 interfaceorg.nd4j.shade.protobuf.Message
- Specified by:
toBuilder
in interfaceorg.nd4j.shade.protobuf.MessageLite
-
newBuilderForType
protected KernelDef.Builder newBuilderForType(org.nd4j.shade.protobuf.GeneratedMessageV3.BuilderParent parent)
- Specified by:
newBuilderForType
in classorg.nd4j.shade.protobuf.GeneratedMessageV3
-
getDefaultInstance
public static KernelDef getDefaultInstance()
-
parser
public static org.nd4j.shade.protobuf.Parser<KernelDef> parser()
-
getParserForType
public org.nd4j.shade.protobuf.Parser<KernelDef> getParserForType()
- Specified by:
getParserForType
in interfaceorg.nd4j.shade.protobuf.Message
- Specified by:
getParserForType
in interfaceorg.nd4j.shade.protobuf.MessageLite
- Overrides:
getParserForType
in classorg.nd4j.shade.protobuf.GeneratedMessageV3
-
getDefaultInstanceForType
public KernelDef getDefaultInstanceForType()
- Specified by:
getDefaultInstanceForType
in interfaceorg.nd4j.shade.protobuf.MessageLiteOrBuilder
- Specified by:
getDefaultInstanceForType
in interfaceorg.nd4j.shade.protobuf.MessageOrBuilder
-
-