Package org.tensorflow.framework

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>