Package org.tensorflow.framework

Class 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 type tensorflow.KernelDef
    See Also:
    Serialized Form

    • Nested Class Summary

      Nested Classes 
      Modifier and Type Class Description
      static class  KernelDef.AttrConstraint
      Protobuf type tensorflow.KernelDef.AttrConstraint
      static interface  KernelDef.AttrConstraintOrBuilder  
      static class  KernelDef.Builder
      Protobuf type tensorflow.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

      • Nested classes/interfaces inherited from class org.nd4j.shade.protobuf.AbstractMessageLite

        org.nd4j.shade.protobuf.AbstractMessageLite.InternalOneOfEnum

    • Method Detail

      • newInstance

        protected Object newInstance​(org.nd4j.shade.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)
        Overrides:
        newInstance in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • getUnknownFields

        public final org.nd4j.shade.protobuf.UnknownFieldSet getUnknownFields()
        Specified by:
        getUnknownFields in interface org.nd4j.shade.protobuf.MessageOrBuilder
        Overrides:
        getUnknownFields in class org.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 class org.nd4j.shade.protobuf.GeneratedMessageV3

      • getOp

        public String getOp()
         Must match the name of an Op.
        string op = 1;
        Specified by:
        getOp in interface KernelDefOrBuilder
        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 interface KernelDefOrBuilder
        Returns:
        The bytes for op.

      • getDeviceType

        public String getDeviceType()
         Type of device this kernel runs on.
        string device_type = 2;
        Specified by:
        getDeviceType in interface KernelDefOrBuilder
        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 interface KernelDefOrBuilder
        Returns:
        The bytes for deviceType.

      • getConstraintCount

        public int getConstraintCount()
        repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
        Specified by:
        getConstraintCount in interface KernelDefOrBuilder

      • 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 interface KernelDefOrBuilder
        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 interface KernelDefOrBuilder
        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 interface KernelDefOrBuilder
        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 interface KernelDefOrBuilder
        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 interface KernelDefOrBuilder
        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 interface KernelDefOrBuilder
        Returns:
        The bytes for label.

      • isInitialized

        public final boolean isInitialized()
        Specified by:
        isInitialized in interface org.nd4j.shade.protobuf.MessageLiteOrBuilder
        Overrides:
        isInitialized in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • writeTo

        public void writeTo​(org.nd4j.shade.protobuf.CodedOutputStream output)
             throws IOException
        Specified by:
        writeTo in interface org.nd4j.shade.protobuf.MessageLite
        Overrides:
        writeTo in class org.nd4j.shade.protobuf.GeneratedMessageV3
        Throws:
        IOException

      • getSerializedSize

        public int getSerializedSize()
        Specified by:
        getSerializedSize in interface org.nd4j.shade.protobuf.MessageLite
        Overrides:
        getSerializedSize in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • equals

        public boolean equals​(Object obj)
        Specified by:
        equals in interface org.nd4j.shade.protobuf.Message
        Overrides:
        equals in class org.nd4j.shade.protobuf.AbstractMessage

      • hashCode

        public int hashCode()
        Specified by:
        hashCode in interface org.nd4j.shade.protobuf.Message
        Overrides:
        hashCode in class org.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​(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 interface org.nd4j.shade.protobuf.Message
        Specified by:
        newBuilderForType in interface org.nd4j.shade.protobuf.MessageLite

      • toBuilder

        public KernelDef.Builder toBuilder()
        Specified by:
        toBuilder in interface org.nd4j.shade.protobuf.Message
        Specified by:
        toBuilder in interface org.nd4j.shade.protobuf.MessageLite

      • newBuilderForType

        protected KernelDef.Builder newBuilderForType​(org.nd4j.shade.protobuf.GeneratedMessageV3.BuilderParent parent)
        Specified by:
        newBuilderForType in class org.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 interface org.nd4j.shade.protobuf.Message
        Specified by:
        getParserForType in interface org.nd4j.shade.protobuf.MessageLite
        Overrides:
        getParserForType in class org.nd4j.shade.protobuf.GeneratedMessageV3

      • getDefaultInstanceForType

        public KernelDef getDefaultInstanceForType()
        Specified by:
        getDefaultInstanceForType in interface org.nd4j.shade.protobuf.MessageLiteOrBuilder
        Specified by:
        getDefaultInstanceForType in interface org.nd4j.shade.protobuf.MessageOrBuilder