Class KernelDef.Builder

  • All Implemented Interfaces:
    Cloneable, org.nd4j.shade.protobuf.Message.Builder, org.nd4j.shade.protobuf.MessageLite.Builder, org.nd4j.shade.protobuf.MessageLiteOrBuilder, org.nd4j.shade.protobuf.MessageOrBuilder, KernelDefOrBuilder
    Enclosing class:
    KernelDef

    public static final class KernelDef.Builder
    extends org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
    implements KernelDefOrBuilder
    Protobuf type tensorflow.KernelDef
    • Method Detail

      • 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.Builder<KernelDef.Builder>
      • clear

        public KernelDef.Builder clear()
        Specified by:
        clear in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        clear in interface org.nd4j.shade.protobuf.MessageLite.Builder
        Overrides:
        clear in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • getDescriptorForType

        public org.nd4j.shade.protobuf.Descriptors.Descriptor getDescriptorForType()
        Specified by:
        getDescriptorForType in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        getDescriptorForType in interface org.nd4j.shade.protobuf.MessageOrBuilder
        Overrides:
        getDescriptorForType in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • getDefaultInstanceForType

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

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

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

        public KernelDef.Builder clone()
        Specified by:
        clone in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        clone in interface org.nd4j.shade.protobuf.MessageLite.Builder
        Overrides:
        clone in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • setField

        public KernelDef.Builder setField​(org.nd4j.shade.protobuf.Descriptors.FieldDescriptor field,
                                          Object value)
        Specified by:
        setField in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        setField in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • clearField

        public KernelDef.Builder clearField​(org.nd4j.shade.protobuf.Descriptors.FieldDescriptor field)
        Specified by:
        clearField in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        clearField in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • clearOneof

        public KernelDef.Builder clearOneof​(org.nd4j.shade.protobuf.Descriptors.OneofDescriptor oneof)
        Specified by:
        clearOneof in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        clearOneof in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • setRepeatedField

        public KernelDef.Builder setRepeatedField​(org.nd4j.shade.protobuf.Descriptors.FieldDescriptor field,
                                                  int index,
                                                  Object value)
        Specified by:
        setRepeatedField in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        setRepeatedField in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • addRepeatedField

        public KernelDef.Builder addRepeatedField​(org.nd4j.shade.protobuf.Descriptors.FieldDescriptor field,
                                                  Object value)
        Specified by:
        addRepeatedField in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        addRepeatedField in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • mergeFrom

        public KernelDef.Builder mergeFrom​(org.nd4j.shade.protobuf.Message other)
        Specified by:
        mergeFrom in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        mergeFrom in class org.nd4j.shade.protobuf.AbstractMessage.Builder<KernelDef.Builder>
      • isInitialized

        public final boolean isInitialized()
        Specified by:
        isInitialized in interface org.nd4j.shade.protobuf.MessageLiteOrBuilder
        Overrides:
        isInitialized in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • mergeFrom

        public KernelDef.Builder mergeFrom​(org.nd4j.shade.protobuf.CodedInputStream input,
                                           org.nd4j.shade.protobuf.ExtensionRegistryLite extensionRegistry)
                                    throws IOException
        Specified by:
        mergeFrom in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        mergeFrom in interface org.nd4j.shade.protobuf.MessageLite.Builder
        Overrides:
        mergeFrom in class org.nd4j.shade.protobuf.AbstractMessage.Builder<KernelDef.Builder>
        Throws:
        IOException
      • 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.
      • setOp

        public KernelDef.Builder setOp​(String value)
         Must match the name of an Op.
         
        string op = 1;
        Parameters:
        value - The op to set.
        Returns:
        This builder for chaining.
      • clearOp

        public KernelDef.Builder clearOp()
         Must match the name of an Op.
         
        string op = 1;
        Returns:
        This builder for chaining.
      • setOpBytes

        public KernelDef.Builder setOpBytes​(org.nd4j.shade.protobuf.ByteString value)
         Must match the name of an Op.
         
        string op = 1;
        Parameters:
        value - The bytes for op to set.
        Returns:
        This builder for chaining.
      • 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.
      • setDeviceType

        public KernelDef.Builder setDeviceType​(String value)
         Type of device this kernel runs on.
         
        string device_type = 2;
        Parameters:
        value - The deviceType to set.
        Returns:
        This builder for chaining.
      • clearDeviceType

        public KernelDef.Builder clearDeviceType()
         Type of device this kernel runs on.
         
        string device_type = 2;
        Returns:
        This builder for chaining.
      • setDeviceTypeBytes

        public KernelDef.Builder setDeviceTypeBytes​(org.nd4j.shade.protobuf.ByteString value)
         Type of device this kernel runs on.
         
        string device_type = 2;
        Parameters:
        value - The bytes for deviceType to set.
        Returns:
        This builder for chaining.
      • getConstraintCount

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

        public KernelDef.Builder clearConstraint()
        repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      • removeConstraint

        public KernelDef.Builder removeConstraint​(int index)
        repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      • getConstraintBuilder

        public KernelDef.AttrConstraint.Builder getConstraintBuilder​(int index)
        repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      • addConstraintBuilder

        public KernelDef.AttrConstraint.Builder addConstraintBuilder​(int index)
        repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
      • 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.
      • setHostMemoryArg

        public KernelDef.Builder setHostMemoryArg​(int index,
                                                  String value)
         Names of the Op's input_/output_args that reside in host memory
         instead of device memory.
         
        repeated string host_memory_arg = 4;
        Parameters:
        index - The index to set the value at.
        value - The hostMemoryArg to set.
        Returns:
        This builder for chaining.
      • addHostMemoryArg

        public KernelDef.Builder addHostMemoryArg​(String value)
         Names of the Op's input_/output_args that reside in host memory
         instead of device memory.
         
        repeated string host_memory_arg = 4;
        Parameters:
        value - The hostMemoryArg to add.
        Returns:
        This builder for chaining.
      • addAllHostMemoryArg

        public KernelDef.Builder addAllHostMemoryArg​(Iterable<String> values)
         Names of the Op's input_/output_args that reside in host memory
         instead of device memory.
         
        repeated string host_memory_arg = 4;
        Parameters:
        values - The hostMemoryArg to add.
        Returns:
        This builder for chaining.
      • clearHostMemoryArg

        public KernelDef.Builder clearHostMemoryArg()
         Names of the Op's input_/output_args that reside in host memory
         instead of device memory.
         
        repeated string host_memory_arg = 4;
        Returns:
        This builder for chaining.
      • addHostMemoryArgBytes

        public KernelDef.Builder addHostMemoryArgBytes​(org.nd4j.shade.protobuf.ByteString value)
         Names of the Op's input_/output_args that reside in host memory
         instead of device memory.
         
        repeated string host_memory_arg = 4;
        Parameters:
        value - The bytes of the hostMemoryArg to add.
        Returns:
        This builder for chaining.
      • 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.
      • setLabel

        public KernelDef.Builder setLabel​(String value)
         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;
        Parameters:
        value - The label to set.
        Returns:
        This builder for chaining.
      • clearLabel

        public KernelDef.Builder clearLabel()
         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;
        Returns:
        This builder for chaining.
      • setLabelBytes

        public KernelDef.Builder setLabelBytes​(org.nd4j.shade.protobuf.ByteString value)
         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;
        Parameters:
        value - The bytes for label to set.
        Returns:
        This builder for chaining.
      • setUnknownFields

        public final KernelDef.Builder setUnknownFields​(org.nd4j.shade.protobuf.UnknownFieldSet unknownFields)
        Specified by:
        setUnknownFields in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        setUnknownFields in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
      • mergeUnknownFields

        public final KernelDef.Builder mergeUnknownFields​(org.nd4j.shade.protobuf.UnknownFieldSet unknownFields)
        Specified by:
        mergeUnknownFields in interface org.nd4j.shade.protobuf.Message.Builder
        Overrides:
        mergeUnknownFields in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>