Package org.platanios.tensorflow.proto

Class KernelDef.Builder

  • All Implemented Interfaces:
    com.google.protobuf.Message.Builder, com.google.protobuf.MessageLite.Builder, com.google.protobuf.MessageLiteOrBuilder, com.google.protobuf.MessageOrBuilder, java.lang.Cloneable, KernelDefOrBuilder
    Enclosing class:
    KernelDef
    public static final class KernelDef.Builder
extends com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>
implements KernelDefOrBuilder
    Protobuf type org.platanios.tensorflow.proto.KernelDef

    • Method Detail

      • getDescriptor

        public static final com.google.protobuf.Descriptors.Descriptor getDescriptor()

      • internalGetFieldAccessorTable

        protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
        Specified by:
        internalGetFieldAccessorTable in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • clear

        public KernelDef.Builder clear()
        Specified by:
        clear in interface com.google.protobuf.Message.Builder
        Specified by:
        clear in interface com.google.protobuf.MessageLite.Builder
        Overrides:
        clear in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • getDescriptorForType

        public com.google.protobuf.Descriptors.Descriptor getDescriptorForType()
        Specified by:
        getDescriptorForType in interface com.google.protobuf.Message.Builder
        Specified by:
        getDescriptorForType in interface com.google.protobuf.MessageOrBuilder
        Overrides:
        getDescriptorForType in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • getDefaultInstanceForType

        public KernelDef getDefaultInstanceForType()
        Specified by:
        getDefaultInstanceForType in interface com.google.protobuf.MessageLiteOrBuilder
        Specified by:
        getDefaultInstanceForType in interface com.google.protobuf.MessageOrBuilder

      • build

        public KernelDef build()
        Specified by:
        build in interface com.google.protobuf.Message.Builder
        Specified by:
        build in interface com.google.protobuf.MessageLite.Builder

      • buildPartial

        public KernelDef buildPartial()
        Specified by:
        buildPartial in interface com.google.protobuf.Message.Builder
        Specified by:
        buildPartial in interface com.google.protobuf.MessageLite.Builder

      • clone

        public KernelDef.Builder clone()
        Specified by:
        clone in interface com.google.protobuf.Message.Builder
        Specified by:
        clone in interface com.google.protobuf.MessageLite.Builder
        Overrides:
        clone in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • setField

        public KernelDef.Builder setField​(com.google.protobuf.Descriptors.FieldDescriptor field,
                                  java.lang.Object value)
        Specified by:
        setField in interface com.google.protobuf.Message.Builder
        Overrides:
        setField in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • clearField

        public KernelDef.Builder clearField​(com.google.protobuf.Descriptors.FieldDescriptor field)
        Specified by:
        clearField in interface com.google.protobuf.Message.Builder
        Overrides:
        clearField in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • clearOneof

        public KernelDef.Builder clearOneof​(com.google.protobuf.Descriptors.OneofDescriptor oneof)
        Specified by:
        clearOneof in interface com.google.protobuf.Message.Builder
        Overrides:
        clearOneof in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • setRepeatedField

        public KernelDef.Builder setRepeatedField​(com.google.protobuf.Descriptors.FieldDescriptor field,
                                          int index,
                                          java.lang.Object value)
        Specified by:
        setRepeatedField in interface com.google.protobuf.Message.Builder
        Overrides:
        setRepeatedField in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • addRepeatedField

        public KernelDef.Builder addRepeatedField​(com.google.protobuf.Descriptors.FieldDescriptor field,
                                          java.lang.Object value)
        Specified by:
        addRepeatedField in interface com.google.protobuf.Message.Builder
        Overrides:
        addRepeatedField in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • mergeFrom

        public KernelDef.Builder mergeFrom​(com.google.protobuf.Message other)
        Specified by:
        mergeFrom in interface com.google.protobuf.Message.Builder
        Overrides:
        mergeFrom in class com.google.protobuf.AbstractMessage.Builder<KernelDef.Builder>

      • isInitialized

        public final boolean isInitialized()
        Specified by:
        isInitialized in interface com.google.protobuf.MessageLiteOrBuilder
        Overrides:
        isInitialized in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • mergeFrom

        public KernelDef.Builder mergeFrom​(com.google.protobuf.CodedInputStream input,
                                   com.google.protobuf.ExtensionRegistryLite extensionRegistry)
                            throws java.io.IOException
        Specified by:
        mergeFrom in interface com.google.protobuf.Message.Builder
        Specified by:
        mergeFrom in interface com.google.protobuf.MessageLite.Builder
        Overrides:
        mergeFrom in class com.google.protobuf.AbstractMessage.Builder<KernelDef.Builder>
        Throws:
        java.io.IOException

      • getOp

        public java.lang.String getOp()
         Must match the name of an Op.
        string op = 1;
        Specified by:
        getOp in interface KernelDefOrBuilder
        Returns:
        The op.

      • getOpBytes

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

      • setDeviceType

        public KernelDef.Builder setDeviceType​(java.lang.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​(com.google.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 .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;
        Specified by:
        getConstraintCount in interface KernelDefOrBuilder

      • addAllConstraint

        public KernelDef.Builder addAllConstraint​(java.lang.Iterable<? extends KernelDef.AttrConstraint> values)
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • clearConstraint

        public KernelDef.Builder clearConstraint()
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • removeConstraint

        public KernelDef.Builder removeConstraint​(int index)
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • getConstraintBuilder

        public KernelDef.AttrConstraint.Builder getConstraintBuilder​(int index)
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • addConstraintBuilder

        public KernelDef.AttrConstraint.Builder addConstraintBuilder()
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • addConstraintBuilder

        public KernelDef.AttrConstraint.Builder addConstraintBuilder​(int index)
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • getConstraintBuilderList

        public java.util.List<KernelDef.AttrConstraint.Builder> getConstraintBuilderList()
        repeated .org.platanios.tensorflow.proto.KernelDef.AttrConstraint constraint = 3;

      • 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 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 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 interface KernelDefOrBuilder
        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 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,
                                          java.lang.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​(java.lang.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​(java.lang.Iterable<java.lang.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​(com.google.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 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 interface KernelDefOrBuilder
        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 interface KernelDefOrBuilder
        Returns:
        The bytes for label.

      • setLabel

        public KernelDef.Builder setLabel​(java.lang.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​(com.google.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.

      • 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 interface KernelDefOrBuilder
        Returns:
        The priority.

      • setPriority

        public KernelDef.Builder setPriority​(int value)
         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;
        Parameters:
        value - The priority to set.
        Returns:
        This builder for chaining.

      • clearPriority

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

      • setUnknownFields

        public final KernelDef.Builder setUnknownFields​(com.google.protobuf.UnknownFieldSet unknownFields)
        Specified by:
        setUnknownFields in interface com.google.protobuf.Message.Builder
        Overrides:
        setUnknownFields in class com.google.protobuf.GeneratedMessageV3.Builder<KernelDef.Builder>

      • mergeUnknownFields

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