org.tensorflow.framework.KernelDef Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of proto Show documentation
Show all versions of proto Show documentation
Java API for TensorFlow protocol buffers.
// Generated by the protocol buffer compiler. DO NOT EDIT!
// source: tensorflow/core/framework/kernel_def.proto
package org.tensorflow.framework;
/**
* Protobuf type {@code tensorflow.KernelDef}
*/
public final class KernelDef extends
com.google.protobuf.GeneratedMessageV3 implements
// @@protoc_insertion_point(message_implements:tensorflow.KernelDef)
KernelDefOrBuilder {
private static final long serialVersionUID = 0L;
// Use KernelDef.newBuilder() to construct.
private KernelDef(com.google.protobuf.GeneratedMessageV3.Builder builder) {
super(builder);
}
private KernelDef() {
op_ = "";
deviceType_ = "";
constraint_ = java.util.Collections.emptyList();
hostMemoryArg_ = com.google.protobuf.LazyStringArrayList.EMPTY;
label_ = "";
priority_ = 0;
}
@java.lang.Override
public final com.google.protobuf.UnknownFieldSet
getUnknownFields() {
return this.unknownFields;
}
private KernelDef(
com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
this();
if (extensionRegistry == null) {
throw new java.lang.NullPointerException();
}
int mutable_bitField0_ = 0;
com.google.protobuf.UnknownFieldSet.Builder unknownFields =
com.google.protobuf.UnknownFieldSet.newBuilder();
try {
boolean done = false;
while (!done) {
int tag = input.readTag();
switch (tag) {
case 0:
done = true;
break;
default: {
if (!parseUnknownFieldProto3(
input, unknownFields, extensionRegistry, tag)) {
done = true;
}
break;
}
case 10: {
java.lang.String s = input.readStringRequireUtf8();
op_ = s;
break;
}
case 18: {
java.lang.String s = input.readStringRequireUtf8();
deviceType_ = s;
break;
}
case 26: {
if (!((mutable_bitField0_ & 0x00000004) == 0x00000004)) {
constraint_ = new java.util.ArrayList();
mutable_bitField0_ |= 0x00000004;
}
constraint_.add(
input.readMessage(org.tensorflow.framework.KernelDef.AttrConstraint.parser(), extensionRegistry));
break;
}
case 34: {
java.lang.String s = input.readStringRequireUtf8();
if (!((mutable_bitField0_ & 0x00000008) == 0x00000008)) {
hostMemoryArg_ = new com.google.protobuf.LazyStringArrayList();
mutable_bitField0_ |= 0x00000008;
}
hostMemoryArg_.add(s);
break;
}
case 42: {
java.lang.String s = input.readStringRequireUtf8();
label_ = s;
break;
}
case 48: {
priority_ = input.readInt32();
break;
}
}
}
} catch (com.google.protobuf.InvalidProtocolBufferException e) {
throw e.setUnfinishedMessage(this);
} catch (java.io.IOException e) {
throw new com.google.protobuf.InvalidProtocolBufferException(
e).setUnfinishedMessage(this);
} finally {
if (((mutable_bitField0_ & 0x00000004) == 0x00000004)) {
constraint_ = java.util.Collections.unmodifiableList(constraint_);
}
if (((mutable_bitField0_ & 0x00000008) == 0x00000008)) {
hostMemoryArg_ = hostMemoryArg_.getUnmodifiableView();
}
this.unknownFields = unknownFields.build();
makeExtensionsImmutable();
}
}
public static final com.google.protobuf.Descriptors.Descriptor
getDescriptor() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_descriptor;
}
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
internalGetFieldAccessorTable() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_fieldAccessorTable
.ensureFieldAccessorsInitialized(
org.tensorflow.framework.KernelDef.class, org.tensorflow.framework.KernelDef.Builder.class);
}
public interface AttrConstraintOrBuilder extends
// @@protoc_insertion_point(interface_extends:tensorflow.KernelDef.AttrConstraint)
com.google.protobuf.MessageOrBuilder {
/**
*
* Name of an attr from the Op.
*
*
* string name = 1;
*/
java.lang.String getName();
/**
*
* Name of an attr from the Op.
*
*
* string name = 1;
*/
com.google.protobuf.ByteString
getNameBytes();
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
boolean hasAllowedValues();
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
org.tensorflow.framework.AttrValue getAllowedValues();
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
org.tensorflow.framework.AttrValueOrBuilder getAllowedValuesOrBuilder();
}
/**
* Protobuf type {@code tensorflow.KernelDef.AttrConstraint}
*/
public static final class AttrConstraint extends
com.google.protobuf.GeneratedMessageV3 implements
// @@protoc_insertion_point(message_implements:tensorflow.KernelDef.AttrConstraint)
AttrConstraintOrBuilder {
private static final long serialVersionUID = 0L;
// Use AttrConstraint.newBuilder() to construct.
private AttrConstraint(com.google.protobuf.GeneratedMessageV3.Builder builder) {
super(builder);
}
private AttrConstraint() {
name_ = "";
}
@java.lang.Override
public final com.google.protobuf.UnknownFieldSet
getUnknownFields() {
return this.unknownFields;
}
private AttrConstraint(
com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
this();
if (extensionRegistry == null) {
throw new java.lang.NullPointerException();
}
int mutable_bitField0_ = 0;
com.google.protobuf.UnknownFieldSet.Builder unknownFields =
com.google.protobuf.UnknownFieldSet.newBuilder();
try {
boolean done = false;
while (!done) {
int tag = input.readTag();
switch (tag) {
case 0:
done = true;
break;
default: {
if (!parseUnknownFieldProto3(
input, unknownFields, extensionRegistry, tag)) {
done = true;
}
break;
}
case 10: {
java.lang.String s = input.readStringRequireUtf8();
name_ = s;
break;
}
case 18: {
org.tensorflow.framework.AttrValue.Builder subBuilder = null;
if (allowedValues_ != null) {
subBuilder = allowedValues_.toBuilder();
}
allowedValues_ = input.readMessage(org.tensorflow.framework.AttrValue.parser(), extensionRegistry);
if (subBuilder != null) {
subBuilder.mergeFrom(allowedValues_);
allowedValues_ = subBuilder.buildPartial();
}
break;
}
}
}
} catch (com.google.protobuf.InvalidProtocolBufferException e) {
throw e.setUnfinishedMessage(this);
} catch (java.io.IOException e) {
throw new com.google.protobuf.InvalidProtocolBufferException(
e).setUnfinishedMessage(this);
} finally {
this.unknownFields = unknownFields.build();
makeExtensionsImmutable();
}
}
public static final com.google.protobuf.Descriptors.Descriptor
getDescriptor() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_AttrConstraint_descriptor;
}
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
internalGetFieldAccessorTable() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_AttrConstraint_fieldAccessorTable
.ensureFieldAccessorsInitialized(
org.tensorflow.framework.KernelDef.AttrConstraint.class, org.tensorflow.framework.KernelDef.AttrConstraint.Builder.class);
}
public static final int NAME_FIELD_NUMBER = 1;
private volatile java.lang.Object name_;
/**
*
* Name of an attr from the Op.
*
*
* string name = 1;
*/
public java.lang.String getName() {
java.lang.Object ref = name_;
if (ref instanceof java.lang.String) {
return (java.lang.String) ref;
} else {
com.google.protobuf.ByteString bs =
(com.google.protobuf.ByteString) ref;
java.lang.String s = bs.toStringUtf8();
name_ = s;
return s;
}
}
/**
*
* Name of an attr from the Op.
*
*
* string name = 1;
*/
public com.google.protobuf.ByteString
getNameBytes() {
java.lang.Object ref = name_;
if (ref instanceof java.lang.String) {
com.google.protobuf.ByteString b =
com.google.protobuf.ByteString.copyFromUtf8(
(java.lang.String) ref);
name_ = b;
return b;
} else {
return (com.google.protobuf.ByteString) ref;
}
}
public static final int ALLOWED_VALUES_FIELD_NUMBER = 2;
private org.tensorflow.framework.AttrValue allowedValues_;
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public boolean hasAllowedValues() {
return allowedValues_ != null;
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public org.tensorflow.framework.AttrValue getAllowedValues() {
return allowedValues_ == null ? org.tensorflow.framework.AttrValue.getDefaultInstance() : allowedValues_;
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public org.tensorflow.framework.AttrValueOrBuilder getAllowedValuesOrBuilder() {
return getAllowedValues();
}
private byte memoizedIsInitialized = -1;
public final boolean isInitialized() {
byte isInitialized = memoizedIsInitialized;
if (isInitialized == 1) return true;
if (isInitialized == 0) return false;
memoizedIsInitialized = 1;
return true;
}
public void writeTo(com.google.protobuf.CodedOutputStream output)
throws java.io.IOException {
if (!getNameBytes().isEmpty()) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 1, name_);
}
if (allowedValues_ != null) {
output.writeMessage(2, getAllowedValues());
}
unknownFields.writeTo(output);
}
public int getSerializedSize() {
int size = memoizedSize;
if (size != -1) return size;
size = 0;
if (!getNameBytes().isEmpty()) {
size += com.google.protobuf.GeneratedMessageV3.computeStringSize(1, name_);
}
if (allowedValues_ != null) {
size += com.google.protobuf.CodedOutputStream
.computeMessageSize(2, getAllowedValues());
}
size += unknownFields.getSerializedSize();
memoizedSize = size;
return size;
}
@java.lang.Override
public boolean equals(final java.lang.Object obj) {
if (obj == this) {
return true;
}
if (!(obj instanceof org.tensorflow.framework.KernelDef.AttrConstraint)) {
return super.equals(obj);
}
org.tensorflow.framework.KernelDef.AttrConstraint other = (org.tensorflow.framework.KernelDef.AttrConstraint) obj;
boolean result = true;
result = result && getName()
.equals(other.getName());
result = result && (hasAllowedValues() == other.hasAllowedValues());
if (hasAllowedValues()) {
result = result && getAllowedValues()
.equals(other.getAllowedValues());
}
result = result && unknownFields.equals(other.unknownFields);
return result;
}
@java.lang.Override
public int hashCode() {
if (memoizedHashCode != 0) {
return memoizedHashCode;
}
int hash = 41;
hash = (19 * hash) + getDescriptor().hashCode();
hash = (37 * hash) + NAME_FIELD_NUMBER;
hash = (53 * hash) + getName().hashCode();
if (hasAllowedValues()) {
hash = (37 * hash) + ALLOWED_VALUES_FIELD_NUMBER;
hash = (53 * hash) + getAllowedValues().hashCode();
}
hash = (29 * hash) + unknownFields.hashCode();
memoizedHashCode = hash;
return hash;
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(
java.nio.ByteBuffer data)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(
java.nio.ByteBuffer data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(
com.google.protobuf.ByteString data)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(
com.google.protobuf.ByteString data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(byte[] data)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(
byte[] data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(java.io.InputStream input)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseWithIOException(PARSER, input);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(
java.io.InputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseWithIOException(PARSER, input, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseDelimitedFrom(java.io.InputStream input)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseDelimitedWithIOException(PARSER, input);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseDelimitedFrom(
java.io.InputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseDelimitedWithIOException(PARSER, input, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(
com.google.protobuf.CodedInputStream input)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseWithIOException(PARSER, input);
}
public static org.tensorflow.framework.KernelDef.AttrConstraint parseFrom(
com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseWithIOException(PARSER, input, extensionRegistry);
}
public Builder newBuilderForType() { return newBuilder(); }
public static Builder newBuilder() {
return DEFAULT_INSTANCE.toBuilder();
}
public static Builder newBuilder(org.tensorflow.framework.KernelDef.AttrConstraint prototype) {
return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
}
public Builder toBuilder() {
return this == DEFAULT_INSTANCE
? new Builder() : new Builder().mergeFrom(this);
}
@java.lang.Override
protected Builder newBuilderForType(
com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
Builder builder = new Builder(parent);
return builder;
}
/**
* Protobuf type {@code tensorflow.KernelDef.AttrConstraint}
*/
public static final class Builder extends
com.google.protobuf.GeneratedMessageV3.Builder implements
// @@protoc_insertion_point(builder_implements:tensorflow.KernelDef.AttrConstraint)
org.tensorflow.framework.KernelDef.AttrConstraintOrBuilder {
public static final com.google.protobuf.Descriptors.Descriptor
getDescriptor() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_AttrConstraint_descriptor;
}
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
internalGetFieldAccessorTable() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_AttrConstraint_fieldAccessorTable
.ensureFieldAccessorsInitialized(
org.tensorflow.framework.KernelDef.AttrConstraint.class, org.tensorflow.framework.KernelDef.AttrConstraint.Builder.class);
}
// Construct using org.tensorflow.framework.KernelDef.AttrConstraint.newBuilder()
private Builder() {
maybeForceBuilderInitialization();
}
private Builder(
com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
super(parent);
maybeForceBuilderInitialization();
}
private void maybeForceBuilderInitialization() {
if (com.google.protobuf.GeneratedMessageV3
.alwaysUseFieldBuilders) {
}
}
public Builder clear() {
super.clear();
name_ = "";
if (allowedValuesBuilder_ == null) {
allowedValues_ = null;
} else {
allowedValues_ = null;
allowedValuesBuilder_ = null;
}
return this;
}
public com.google.protobuf.Descriptors.Descriptor
getDescriptorForType() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_AttrConstraint_descriptor;
}
public org.tensorflow.framework.KernelDef.AttrConstraint getDefaultInstanceForType() {
return org.tensorflow.framework.KernelDef.AttrConstraint.getDefaultInstance();
}
public org.tensorflow.framework.KernelDef.AttrConstraint build() {
org.tensorflow.framework.KernelDef.AttrConstraint result = buildPartial();
if (!result.isInitialized()) {
throw newUninitializedMessageException(result);
}
return result;
}
public org.tensorflow.framework.KernelDef.AttrConstraint buildPartial() {
org.tensorflow.framework.KernelDef.AttrConstraint result = new org.tensorflow.framework.KernelDef.AttrConstraint(this);
result.name_ = name_;
if (allowedValuesBuilder_ == null) {
result.allowedValues_ = allowedValues_;
} else {
result.allowedValues_ = allowedValuesBuilder_.build();
}
onBuilt();
return result;
}
public Builder clone() {
return (Builder) super.clone();
}
public Builder setField(
com.google.protobuf.Descriptors.FieldDescriptor field,
java.lang.Object value) {
return (Builder) super.setField(field, value);
}
public Builder clearField(
com.google.protobuf.Descriptors.FieldDescriptor field) {
return (Builder) super.clearField(field);
}
public Builder clearOneof(
com.google.protobuf.Descriptors.OneofDescriptor oneof) {
return (Builder) super.clearOneof(oneof);
}
public Builder setRepeatedField(
com.google.protobuf.Descriptors.FieldDescriptor field,
int index, java.lang.Object value) {
return (Builder) super.setRepeatedField(field, index, value);
}
public Builder addRepeatedField(
com.google.protobuf.Descriptors.FieldDescriptor field,
java.lang.Object value) {
return (Builder) super.addRepeatedField(field, value);
}
public Builder mergeFrom(com.google.protobuf.Message other) {
if (other instanceof org.tensorflow.framework.KernelDef.AttrConstraint) {
return mergeFrom((org.tensorflow.framework.KernelDef.AttrConstraint)other);
} else {
super.mergeFrom(other);
return this;
}
}
public Builder mergeFrom(org.tensorflow.framework.KernelDef.AttrConstraint other) {
if (other == org.tensorflow.framework.KernelDef.AttrConstraint.getDefaultInstance()) return this;
if (!other.getName().isEmpty()) {
name_ = other.name_;
onChanged();
}
if (other.hasAllowedValues()) {
mergeAllowedValues(other.getAllowedValues());
}
this.mergeUnknownFields(other.unknownFields);
onChanged();
return this;
}
public final boolean isInitialized() {
return true;
}
public Builder mergeFrom(
com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws java.io.IOException {
org.tensorflow.framework.KernelDef.AttrConstraint parsedMessage = null;
try {
parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry);
} catch (com.google.protobuf.InvalidProtocolBufferException e) {
parsedMessage = (org.tensorflow.framework.KernelDef.AttrConstraint) e.getUnfinishedMessage();
throw e.unwrapIOException();
} finally {
if (parsedMessage != null) {
mergeFrom(parsedMessage);
}
}
return this;
}
private java.lang.Object name_ = "";
/**
*
* Name of an attr from the Op.
*
*
* string name = 1;
*/
public java.lang.String getName() {
java.lang.Object ref = name_;
if (!(ref instanceof java.lang.String)) {
com.google.protobuf.ByteString bs =
(com.google.protobuf.ByteString) ref;
java.lang.String s = bs.toStringUtf8();
name_ = s;
return s;
} else {
return (java.lang.String) ref;
}
}
/**
*
* Name of an attr from the Op.
*
*
* string name = 1;
*/
public com.google.protobuf.ByteString
getNameBytes() {
java.lang.Object ref = name_;
if (ref instanceof String) {
com.google.protobuf.ByteString b =
com.google.protobuf.ByteString.copyFromUtf8(
(java.lang.String) ref);
name_ = b;
return b;
} else {
return (com.google.protobuf.ByteString) ref;
}
}
/**
*
* Name of an attr from the Op.
*
*
* string name = 1;
*/
public Builder setName(
java.lang.String value) {
if (value == null) {
throw new NullPointerException();
}
name_ = value;
onChanged();
return this;
}
/**
*
* Name of an attr from the Op.
*
*
* string name = 1;
*/
public Builder clearName() {
name_ = getDefaultInstance().getName();
onChanged();
return this;
}
/**
*
* Name of an attr from the Op.
*
*
* string name = 1;
*/
public Builder setNameBytes(
com.google.protobuf.ByteString value) {
if (value == null) {
throw new NullPointerException();
}
checkByteStringIsUtf8(value);
name_ = value;
onChanged();
return this;
}
private org.tensorflow.framework.AttrValue allowedValues_ = null;
private com.google.protobuf.SingleFieldBuilderV3<
org.tensorflow.framework.AttrValue, org.tensorflow.framework.AttrValue.Builder, org.tensorflow.framework.AttrValueOrBuilder> allowedValuesBuilder_;
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public boolean hasAllowedValues() {
return allowedValuesBuilder_ != null || allowedValues_ != null;
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public org.tensorflow.framework.AttrValue getAllowedValues() {
if (allowedValuesBuilder_ == null) {
return allowedValues_ == null ? org.tensorflow.framework.AttrValue.getDefaultInstance() : allowedValues_;
} else {
return allowedValuesBuilder_.getMessage();
}
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public Builder setAllowedValues(org.tensorflow.framework.AttrValue value) {
if (allowedValuesBuilder_ == null) {
if (value == null) {
throw new NullPointerException();
}
allowedValues_ = value;
onChanged();
} else {
allowedValuesBuilder_.setMessage(value);
}
return this;
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public Builder setAllowedValues(
org.tensorflow.framework.AttrValue.Builder builderForValue) {
if (allowedValuesBuilder_ == null) {
allowedValues_ = builderForValue.build();
onChanged();
} else {
allowedValuesBuilder_.setMessage(builderForValue.build());
}
return this;
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public Builder mergeAllowedValues(org.tensorflow.framework.AttrValue value) {
if (allowedValuesBuilder_ == null) {
if (allowedValues_ != null) {
allowedValues_ =
org.tensorflow.framework.AttrValue.newBuilder(allowedValues_).mergeFrom(value).buildPartial();
} else {
allowedValues_ = value;
}
onChanged();
} else {
allowedValuesBuilder_.mergeFrom(value);
}
return this;
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public Builder clearAllowedValues() {
if (allowedValuesBuilder_ == null) {
allowedValues_ = null;
onChanged();
} else {
allowedValues_ = null;
allowedValuesBuilder_ = null;
}
return this;
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public org.tensorflow.framework.AttrValue.Builder getAllowedValuesBuilder() {
onChanged();
return getAllowedValuesFieldBuilder().getBuilder();
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
public org.tensorflow.framework.AttrValueOrBuilder getAllowedValuesOrBuilder() {
if (allowedValuesBuilder_ != null) {
return allowedValuesBuilder_.getMessageOrBuilder();
} else {
return allowedValues_ == null ?
org.tensorflow.framework.AttrValue.getDefaultInstance() : allowedValues_;
}
}
/**
*
* A list of values that this kernel supports for this attr.
* Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
*
*
* .tensorflow.AttrValue allowed_values = 2;
*/
private com.google.protobuf.SingleFieldBuilderV3<
org.tensorflow.framework.AttrValue, org.tensorflow.framework.AttrValue.Builder, org.tensorflow.framework.AttrValueOrBuilder>
getAllowedValuesFieldBuilder() {
if (allowedValuesBuilder_ == null) {
allowedValuesBuilder_ = new com.google.protobuf.SingleFieldBuilderV3<
org.tensorflow.framework.AttrValue, org.tensorflow.framework.AttrValue.Builder, org.tensorflow.framework.AttrValueOrBuilder>(
getAllowedValues(),
getParentForChildren(),
isClean());
allowedValues_ = null;
}
return allowedValuesBuilder_;
}
public final Builder setUnknownFields(
final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.setUnknownFieldsProto3(unknownFields);
}
public final Builder mergeUnknownFields(
final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.mergeUnknownFields(unknownFields);
}
// @@protoc_insertion_point(builder_scope:tensorflow.KernelDef.AttrConstraint)
}
// @@protoc_insertion_point(class_scope:tensorflow.KernelDef.AttrConstraint)
private static final org.tensorflow.framework.KernelDef.AttrConstraint DEFAULT_INSTANCE;
static {
DEFAULT_INSTANCE = new org.tensorflow.framework.KernelDef.AttrConstraint();
}
public static org.tensorflow.framework.KernelDef.AttrConstraint getDefaultInstance() {
return DEFAULT_INSTANCE;
}
private static final com.google.protobuf.Parser
PARSER = new com.google.protobuf.AbstractParser() {
public AttrConstraint parsePartialFrom(
com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
return new AttrConstraint(input, extensionRegistry);
}
};
public static com.google.protobuf.Parser parser() {
return PARSER;
}
@java.lang.Override
public com.google.protobuf.Parser getParserForType() {
return PARSER;
}
public org.tensorflow.framework.KernelDef.AttrConstraint getDefaultInstanceForType() {
return DEFAULT_INSTANCE;
}
}
private int bitField0_;
public static final int OP_FIELD_NUMBER = 1;
private volatile java.lang.Object op_;
/**
*
* Must match the name of an Op.
*
*
* string op = 1;
*/
public java.lang.String getOp() {
java.lang.Object ref = op_;
if (ref instanceof java.lang.String) {
return (java.lang.String) ref;
} else {
com.google.protobuf.ByteString bs =
(com.google.protobuf.ByteString) ref;
java.lang.String s = bs.toStringUtf8();
op_ = s;
return s;
}
}
/**
*
* Must match the name of an Op.
*
*
* string op = 1;
*/
public com.google.protobuf.ByteString
getOpBytes() {
java.lang.Object ref = op_;
if (ref instanceof java.lang.String) {
com.google.protobuf.ByteString b =
com.google.protobuf.ByteString.copyFromUtf8(
(java.lang.String) ref);
op_ = b;
return b;
} else {
return (com.google.protobuf.ByteString) ref;
}
}
public static final int DEVICE_TYPE_FIELD_NUMBER = 2;
private volatile java.lang.Object deviceType_;
/**
*
* Type of device this kernel runs on.
*
*
* string device_type = 2;
*/
public java.lang.String getDeviceType() {
java.lang.Object ref = deviceType_;
if (ref instanceof java.lang.String) {
return (java.lang.String) ref;
} else {
com.google.protobuf.ByteString bs =
(com.google.protobuf.ByteString) ref;
java.lang.String s = bs.toStringUtf8();
deviceType_ = s;
return s;
}
}
/**
*
* Type of device this kernel runs on.
*
*
* string device_type = 2;
*/
public com.google.protobuf.ByteString
getDeviceTypeBytes() {
java.lang.Object ref = deviceType_;
if (ref instanceof java.lang.String) {
com.google.protobuf.ByteString b =
com.google.protobuf.ByteString.copyFromUtf8(
(java.lang.String) ref);
deviceType_ = b;
return b;
} else {
return (com.google.protobuf.ByteString) ref;
}
}
public static final int CONSTRAINT_FIELD_NUMBER = 3;
private java.util.List constraint_;
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public java.util.List getConstraintList() {
return constraint_;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public java.util.List
getConstraintOrBuilderList() {
return constraint_;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public int getConstraintCount() {
return constraint_.size();
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public org.tensorflow.framework.KernelDef.AttrConstraint getConstraint(int index) {
return constraint_.get(index);
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public org.tensorflow.framework.KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder(
int index) {
return constraint_.get(index);
}
public static final int HOST_MEMORY_ARG_FIELD_NUMBER = 4;
private com.google.protobuf.LazyStringList hostMemoryArg_;
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public com.google.protobuf.ProtocolStringList
getHostMemoryArgList() {
return hostMemoryArg_;
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public int getHostMemoryArgCount() {
return hostMemoryArg_.size();
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public java.lang.String getHostMemoryArg(int index) {
return hostMemoryArg_.get(index);
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public com.google.protobuf.ByteString
getHostMemoryArgBytes(int index) {
return hostMemoryArg_.getByteString(index);
}
public static final int LABEL_FIELD_NUMBER = 5;
private volatile java.lang.Object label_;
/**
*
* 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;
*/
public java.lang.String getLabel() {
java.lang.Object ref = label_;
if (ref instanceof java.lang.String) {
return (java.lang.String) ref;
} else {
com.google.protobuf.ByteString bs =
(com.google.protobuf.ByteString) ref;
java.lang.String s = bs.toStringUtf8();
label_ = s;
return s;
}
}
/**
*
* 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;
*/
public com.google.protobuf.ByteString
getLabelBytes() {
java.lang.Object ref = label_;
if (ref instanceof java.lang.String) {
com.google.protobuf.ByteString b =
com.google.protobuf.ByteString.copyFromUtf8(
(java.lang.String) ref);
label_ = b;
return b;
} else {
return (com.google.protobuf.ByteString) ref;
}
}
public static final int PRIORITY_FIELD_NUMBER = 6;
private int priority_;
/**
*
* 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;
*/
public int getPriority() {
return priority_;
}
private byte memoizedIsInitialized = -1;
public final boolean isInitialized() {
byte isInitialized = memoizedIsInitialized;
if (isInitialized == 1) return true;
if (isInitialized == 0) return false;
memoizedIsInitialized = 1;
return true;
}
public void writeTo(com.google.protobuf.CodedOutputStream output)
throws java.io.IOException {
if (!getOpBytes().isEmpty()) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 1, op_);
}
if (!getDeviceTypeBytes().isEmpty()) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 2, deviceType_);
}
for (int i = 0; i < constraint_.size(); i++) {
output.writeMessage(3, constraint_.get(i));
}
for (int i = 0; i < hostMemoryArg_.size(); i++) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 4, hostMemoryArg_.getRaw(i));
}
if (!getLabelBytes().isEmpty()) {
com.google.protobuf.GeneratedMessageV3.writeString(output, 5, label_);
}
if (priority_ != 0) {
output.writeInt32(6, priority_);
}
unknownFields.writeTo(output);
}
public int getSerializedSize() {
int size = memoizedSize;
if (size != -1) return size;
size = 0;
if (!getOpBytes().isEmpty()) {
size += com.google.protobuf.GeneratedMessageV3.computeStringSize(1, op_);
}
if (!getDeviceTypeBytes().isEmpty()) {
size += com.google.protobuf.GeneratedMessageV3.computeStringSize(2, deviceType_);
}
for (int i = 0; i < constraint_.size(); i++) {
size += com.google.protobuf.CodedOutputStream
.computeMessageSize(3, constraint_.get(i));
}
{
int dataSize = 0;
for (int i = 0; i < hostMemoryArg_.size(); i++) {
dataSize += computeStringSizeNoTag(hostMemoryArg_.getRaw(i));
}
size += dataSize;
size += 1 * getHostMemoryArgList().size();
}
if (!getLabelBytes().isEmpty()) {
size += com.google.protobuf.GeneratedMessageV3.computeStringSize(5, label_);
}
if (priority_ != 0) {
size += com.google.protobuf.CodedOutputStream
.computeInt32Size(6, priority_);
}
size += unknownFields.getSerializedSize();
memoizedSize = size;
return size;
}
@java.lang.Override
public boolean equals(final java.lang.Object obj) {
if (obj == this) {
return true;
}
if (!(obj instanceof org.tensorflow.framework.KernelDef)) {
return super.equals(obj);
}
org.tensorflow.framework.KernelDef other = (org.tensorflow.framework.KernelDef) obj;
boolean result = true;
result = result && getOp()
.equals(other.getOp());
result = result && getDeviceType()
.equals(other.getDeviceType());
result = result && getConstraintList()
.equals(other.getConstraintList());
result = result && getHostMemoryArgList()
.equals(other.getHostMemoryArgList());
result = result && getLabel()
.equals(other.getLabel());
result = result && (getPriority()
== other.getPriority());
result = result && unknownFields.equals(other.unknownFields);
return result;
}
@java.lang.Override
public int hashCode() {
if (memoizedHashCode != 0) {
return memoizedHashCode;
}
int hash = 41;
hash = (19 * hash) + getDescriptor().hashCode();
hash = (37 * hash) + OP_FIELD_NUMBER;
hash = (53 * hash) + getOp().hashCode();
hash = (37 * hash) + DEVICE_TYPE_FIELD_NUMBER;
hash = (53 * hash) + getDeviceType().hashCode();
if (getConstraintCount() > 0) {
hash = (37 * hash) + CONSTRAINT_FIELD_NUMBER;
hash = (53 * hash) + getConstraintList().hashCode();
}
if (getHostMemoryArgCount() > 0) {
hash = (37 * hash) + HOST_MEMORY_ARG_FIELD_NUMBER;
hash = (53 * hash) + getHostMemoryArgList().hashCode();
}
hash = (37 * hash) + LABEL_FIELD_NUMBER;
hash = (53 * hash) + getLabel().hashCode();
hash = (37 * hash) + PRIORITY_FIELD_NUMBER;
hash = (53 * hash) + getPriority();
hash = (29 * hash) + unknownFields.hashCode();
memoizedHashCode = hash;
return hash;
}
public static org.tensorflow.framework.KernelDef parseFrom(
java.nio.ByteBuffer data)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data);
}
public static org.tensorflow.framework.KernelDef parseFrom(
java.nio.ByteBuffer data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef parseFrom(
com.google.protobuf.ByteString data)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data);
}
public static org.tensorflow.framework.KernelDef parseFrom(
com.google.protobuf.ByteString data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef parseFrom(byte[] data)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data);
}
public static org.tensorflow.framework.KernelDef parseFrom(
byte[] data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
return PARSER.parseFrom(data, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef parseFrom(java.io.InputStream input)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseWithIOException(PARSER, input);
}
public static org.tensorflow.framework.KernelDef parseFrom(
java.io.InputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseWithIOException(PARSER, input, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef parseDelimitedFrom(java.io.InputStream input)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseDelimitedWithIOException(PARSER, input);
}
public static org.tensorflow.framework.KernelDef parseDelimitedFrom(
java.io.InputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseDelimitedWithIOException(PARSER, input, extensionRegistry);
}
public static org.tensorflow.framework.KernelDef parseFrom(
com.google.protobuf.CodedInputStream input)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseWithIOException(PARSER, input);
}
public static org.tensorflow.framework.KernelDef parseFrom(
com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws java.io.IOException {
return com.google.protobuf.GeneratedMessageV3
.parseWithIOException(PARSER, input, extensionRegistry);
}
public Builder newBuilderForType() { return newBuilder(); }
public static Builder newBuilder() {
return DEFAULT_INSTANCE.toBuilder();
}
public static Builder newBuilder(org.tensorflow.framework.KernelDef prototype) {
return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype);
}
public Builder toBuilder() {
return this == DEFAULT_INSTANCE
? new Builder() : new Builder().mergeFrom(this);
}
@java.lang.Override
protected Builder newBuilderForType(
com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
Builder builder = new Builder(parent);
return builder;
}
/**
* Protobuf type {@code tensorflow.KernelDef}
*/
public static final class Builder extends
com.google.protobuf.GeneratedMessageV3.Builder implements
// @@protoc_insertion_point(builder_implements:tensorflow.KernelDef)
org.tensorflow.framework.KernelDefOrBuilder {
public static final com.google.protobuf.Descriptors.Descriptor
getDescriptor() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_descriptor;
}
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
internalGetFieldAccessorTable() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_fieldAccessorTable
.ensureFieldAccessorsInitialized(
org.tensorflow.framework.KernelDef.class, org.tensorflow.framework.KernelDef.Builder.class);
}
// Construct using org.tensorflow.framework.KernelDef.newBuilder()
private Builder() {
maybeForceBuilderInitialization();
}
private Builder(
com.google.protobuf.GeneratedMessageV3.BuilderParent parent) {
super(parent);
maybeForceBuilderInitialization();
}
private void maybeForceBuilderInitialization() {
if (com.google.protobuf.GeneratedMessageV3
.alwaysUseFieldBuilders) {
getConstraintFieldBuilder();
}
}
public Builder clear() {
super.clear();
op_ = "";
deviceType_ = "";
if (constraintBuilder_ == null) {
constraint_ = java.util.Collections.emptyList();
bitField0_ = (bitField0_ & ~0x00000004);
} else {
constraintBuilder_.clear();
}
hostMemoryArg_ = com.google.protobuf.LazyStringArrayList.EMPTY;
bitField0_ = (bitField0_ & ~0x00000008);
label_ = "";
priority_ = 0;
return this;
}
public com.google.protobuf.Descriptors.Descriptor
getDescriptorForType() {
return org.tensorflow.framework.KernelDefProtos.internal_static_tensorflow_KernelDef_descriptor;
}
public org.tensorflow.framework.KernelDef getDefaultInstanceForType() {
return org.tensorflow.framework.KernelDef.getDefaultInstance();
}
public org.tensorflow.framework.KernelDef build() {
org.tensorflow.framework.KernelDef result = buildPartial();
if (!result.isInitialized()) {
throw newUninitializedMessageException(result);
}
return result;
}
public org.tensorflow.framework.KernelDef buildPartial() {
org.tensorflow.framework.KernelDef result = new org.tensorflow.framework.KernelDef(this);
int from_bitField0_ = bitField0_;
int to_bitField0_ = 0;
result.op_ = op_;
result.deviceType_ = deviceType_;
if (constraintBuilder_ == null) {
if (((bitField0_ & 0x00000004) == 0x00000004)) {
constraint_ = java.util.Collections.unmodifiableList(constraint_);
bitField0_ = (bitField0_ & ~0x00000004);
}
result.constraint_ = constraint_;
} else {
result.constraint_ = constraintBuilder_.build();
}
if (((bitField0_ & 0x00000008) == 0x00000008)) {
hostMemoryArg_ = hostMemoryArg_.getUnmodifiableView();
bitField0_ = (bitField0_ & ~0x00000008);
}
result.hostMemoryArg_ = hostMemoryArg_;
result.label_ = label_;
result.priority_ = priority_;
result.bitField0_ = to_bitField0_;
onBuilt();
return result;
}
public Builder clone() {
return (Builder) super.clone();
}
public Builder setField(
com.google.protobuf.Descriptors.FieldDescriptor field,
java.lang.Object value) {
return (Builder) super.setField(field, value);
}
public Builder clearField(
com.google.protobuf.Descriptors.FieldDescriptor field) {
return (Builder) super.clearField(field);
}
public Builder clearOneof(
com.google.protobuf.Descriptors.OneofDescriptor oneof) {
return (Builder) super.clearOneof(oneof);
}
public Builder setRepeatedField(
com.google.protobuf.Descriptors.FieldDescriptor field,
int index, java.lang.Object value) {
return (Builder) super.setRepeatedField(field, index, value);
}
public Builder addRepeatedField(
com.google.protobuf.Descriptors.FieldDescriptor field,
java.lang.Object value) {
return (Builder) super.addRepeatedField(field, value);
}
public Builder mergeFrom(com.google.protobuf.Message other) {
if (other instanceof org.tensorflow.framework.KernelDef) {
return mergeFrom((org.tensorflow.framework.KernelDef)other);
} else {
super.mergeFrom(other);
return this;
}
}
public Builder mergeFrom(org.tensorflow.framework.KernelDef other) {
if (other == org.tensorflow.framework.KernelDef.getDefaultInstance()) return this;
if (!other.getOp().isEmpty()) {
op_ = other.op_;
onChanged();
}
if (!other.getDeviceType().isEmpty()) {
deviceType_ = other.deviceType_;
onChanged();
}
if (constraintBuilder_ == null) {
if (!other.constraint_.isEmpty()) {
if (constraint_.isEmpty()) {
constraint_ = other.constraint_;
bitField0_ = (bitField0_ & ~0x00000004);
} else {
ensureConstraintIsMutable();
constraint_.addAll(other.constraint_);
}
onChanged();
}
} else {
if (!other.constraint_.isEmpty()) {
if (constraintBuilder_.isEmpty()) {
constraintBuilder_.dispose();
constraintBuilder_ = null;
constraint_ = other.constraint_;
bitField0_ = (bitField0_ & ~0x00000004);
constraintBuilder_ =
com.google.protobuf.GeneratedMessageV3.alwaysUseFieldBuilders ?
getConstraintFieldBuilder() : null;
} else {
constraintBuilder_.addAllMessages(other.constraint_);
}
}
}
if (!other.hostMemoryArg_.isEmpty()) {
if (hostMemoryArg_.isEmpty()) {
hostMemoryArg_ = other.hostMemoryArg_;
bitField0_ = (bitField0_ & ~0x00000008);
} else {
ensureHostMemoryArgIsMutable();
hostMemoryArg_.addAll(other.hostMemoryArg_);
}
onChanged();
}
if (!other.getLabel().isEmpty()) {
label_ = other.label_;
onChanged();
}
if (other.getPriority() != 0) {
setPriority(other.getPriority());
}
this.mergeUnknownFields(other.unknownFields);
onChanged();
return this;
}
public final boolean isInitialized() {
return true;
}
public Builder mergeFrom(
com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws java.io.IOException {
org.tensorflow.framework.KernelDef parsedMessage = null;
try {
parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry);
} catch (com.google.protobuf.InvalidProtocolBufferException e) {
parsedMessage = (org.tensorflow.framework.KernelDef) e.getUnfinishedMessage();
throw e.unwrapIOException();
} finally {
if (parsedMessage != null) {
mergeFrom(parsedMessage);
}
}
return this;
}
private int bitField0_;
private java.lang.Object op_ = "";
/**
*
* Must match the name of an Op.
*
*
* string op = 1;
*/
public java.lang.String getOp() {
java.lang.Object ref = op_;
if (!(ref instanceof java.lang.String)) {
com.google.protobuf.ByteString bs =
(com.google.protobuf.ByteString) ref;
java.lang.String s = bs.toStringUtf8();
op_ = s;
return s;
} else {
return (java.lang.String) ref;
}
}
/**
*
* Must match the name of an Op.
*
*
* string op = 1;
*/
public com.google.protobuf.ByteString
getOpBytes() {
java.lang.Object ref = op_;
if (ref instanceof String) {
com.google.protobuf.ByteString b =
com.google.protobuf.ByteString.copyFromUtf8(
(java.lang.String) ref);
op_ = b;
return b;
} else {
return (com.google.protobuf.ByteString) ref;
}
}
/**
*
* Must match the name of an Op.
*
*
* string op = 1;
*/
public Builder setOp(
java.lang.String value) {
if (value == null) {
throw new NullPointerException();
}
op_ = value;
onChanged();
return this;
}
/**
*
* Must match the name of an Op.
*
*
* string op = 1;
*/
public Builder clearOp() {
op_ = getDefaultInstance().getOp();
onChanged();
return this;
}
/**
*
* Must match the name of an Op.
*
*
* string op = 1;
*/
public Builder setOpBytes(
com.google.protobuf.ByteString value) {
if (value == null) {
throw new NullPointerException();
}
checkByteStringIsUtf8(value);
op_ = value;
onChanged();
return this;
}
private java.lang.Object deviceType_ = "";
/**
*
* Type of device this kernel runs on.
*
*
* string device_type = 2;
*/
public java.lang.String getDeviceType() {
java.lang.Object ref = deviceType_;
if (!(ref instanceof java.lang.String)) {
com.google.protobuf.ByteString bs =
(com.google.protobuf.ByteString) ref;
java.lang.String s = bs.toStringUtf8();
deviceType_ = s;
return s;
} else {
return (java.lang.String) ref;
}
}
/**
*
* Type of device this kernel runs on.
*
*
* string device_type = 2;
*/
public com.google.protobuf.ByteString
getDeviceTypeBytes() {
java.lang.Object ref = deviceType_;
if (ref instanceof String) {
com.google.protobuf.ByteString b =
com.google.protobuf.ByteString.copyFromUtf8(
(java.lang.String) ref);
deviceType_ = b;
return b;
} else {
return (com.google.protobuf.ByteString) ref;
}
}
/**
*
* Type of device this kernel runs on.
*
*
* string device_type = 2;
*/
public Builder setDeviceType(
java.lang.String value) {
if (value == null) {
throw new NullPointerException();
}
deviceType_ = value;
onChanged();
return this;
}
/**
*
* Type of device this kernel runs on.
*
*
* string device_type = 2;
*/
public Builder clearDeviceType() {
deviceType_ = getDefaultInstance().getDeviceType();
onChanged();
return this;
}
/**
*
* Type of device this kernel runs on.
*
*
* string device_type = 2;
*/
public Builder setDeviceTypeBytes(
com.google.protobuf.ByteString value) {
if (value == null) {
throw new NullPointerException();
}
checkByteStringIsUtf8(value);
deviceType_ = value;
onChanged();
return this;
}
private java.util.List constraint_ =
java.util.Collections.emptyList();
private void ensureConstraintIsMutable() {
if (!((bitField0_ & 0x00000004) == 0x00000004)) {
constraint_ = new java.util.ArrayList(constraint_);
bitField0_ |= 0x00000004;
}
}
private com.google.protobuf.RepeatedFieldBuilderV3<
org.tensorflow.framework.KernelDef.AttrConstraint, org.tensorflow.framework.KernelDef.AttrConstraint.Builder, org.tensorflow.framework.KernelDef.AttrConstraintOrBuilder> constraintBuilder_;
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public java.util.List getConstraintList() {
if (constraintBuilder_ == null) {
return java.util.Collections.unmodifiableList(constraint_);
} else {
return constraintBuilder_.getMessageList();
}
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public int getConstraintCount() {
if (constraintBuilder_ == null) {
return constraint_.size();
} else {
return constraintBuilder_.getCount();
}
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public org.tensorflow.framework.KernelDef.AttrConstraint getConstraint(int index) {
if (constraintBuilder_ == null) {
return constraint_.get(index);
} else {
return constraintBuilder_.getMessage(index);
}
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public Builder setConstraint(
int index, org.tensorflow.framework.KernelDef.AttrConstraint value) {
if (constraintBuilder_ == null) {
if (value == null) {
throw new NullPointerException();
}
ensureConstraintIsMutable();
constraint_.set(index, value);
onChanged();
} else {
constraintBuilder_.setMessage(index, value);
}
return this;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public Builder setConstraint(
int index, org.tensorflow.framework.KernelDef.AttrConstraint.Builder builderForValue) {
if (constraintBuilder_ == null) {
ensureConstraintIsMutable();
constraint_.set(index, builderForValue.build());
onChanged();
} else {
constraintBuilder_.setMessage(index, builderForValue.build());
}
return this;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public Builder addConstraint(org.tensorflow.framework.KernelDef.AttrConstraint value) {
if (constraintBuilder_ == null) {
if (value == null) {
throw new NullPointerException();
}
ensureConstraintIsMutable();
constraint_.add(value);
onChanged();
} else {
constraintBuilder_.addMessage(value);
}
return this;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public Builder addConstraint(
int index, org.tensorflow.framework.KernelDef.AttrConstraint value) {
if (constraintBuilder_ == null) {
if (value == null) {
throw new NullPointerException();
}
ensureConstraintIsMutable();
constraint_.add(index, value);
onChanged();
} else {
constraintBuilder_.addMessage(index, value);
}
return this;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public Builder addConstraint(
org.tensorflow.framework.KernelDef.AttrConstraint.Builder builderForValue) {
if (constraintBuilder_ == null) {
ensureConstraintIsMutable();
constraint_.add(builderForValue.build());
onChanged();
} else {
constraintBuilder_.addMessage(builderForValue.build());
}
return this;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public Builder addConstraint(
int index, org.tensorflow.framework.KernelDef.AttrConstraint.Builder builderForValue) {
if (constraintBuilder_ == null) {
ensureConstraintIsMutable();
constraint_.add(index, builderForValue.build());
onChanged();
} else {
constraintBuilder_.addMessage(index, builderForValue.build());
}
return this;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public Builder addAllConstraint(
java.lang.Iterable values) {
if (constraintBuilder_ == null) {
ensureConstraintIsMutable();
com.google.protobuf.AbstractMessageLite.Builder.addAll(
values, constraint_);
onChanged();
} else {
constraintBuilder_.addAllMessages(values);
}
return this;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public Builder clearConstraint() {
if (constraintBuilder_ == null) {
constraint_ = java.util.Collections.emptyList();
bitField0_ = (bitField0_ & ~0x00000004);
onChanged();
} else {
constraintBuilder_.clear();
}
return this;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public Builder removeConstraint(int index) {
if (constraintBuilder_ == null) {
ensureConstraintIsMutable();
constraint_.remove(index);
onChanged();
} else {
constraintBuilder_.remove(index);
}
return this;
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public org.tensorflow.framework.KernelDef.AttrConstraint.Builder getConstraintBuilder(
int index) {
return getConstraintFieldBuilder().getBuilder(index);
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public org.tensorflow.framework.KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder(
int index) {
if (constraintBuilder_ == null) {
return constraint_.get(index); } else {
return constraintBuilder_.getMessageOrBuilder(index);
}
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public java.util.List
getConstraintOrBuilderList() {
if (constraintBuilder_ != null) {
return constraintBuilder_.getMessageOrBuilderList();
} else {
return java.util.Collections.unmodifiableList(constraint_);
}
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public org.tensorflow.framework.KernelDef.AttrConstraint.Builder addConstraintBuilder() {
return getConstraintFieldBuilder().addBuilder(
org.tensorflow.framework.KernelDef.AttrConstraint.getDefaultInstance());
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public org.tensorflow.framework.KernelDef.AttrConstraint.Builder addConstraintBuilder(
int index) {
return getConstraintFieldBuilder().addBuilder(
index, org.tensorflow.framework.KernelDef.AttrConstraint.getDefaultInstance());
}
/**
* repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
*/
public java.util.List
getConstraintBuilderList() {
return getConstraintFieldBuilder().getBuilderList();
}
private com.google.protobuf.RepeatedFieldBuilderV3<
org.tensorflow.framework.KernelDef.AttrConstraint, org.tensorflow.framework.KernelDef.AttrConstraint.Builder, org.tensorflow.framework.KernelDef.AttrConstraintOrBuilder>
getConstraintFieldBuilder() {
if (constraintBuilder_ == null) {
constraintBuilder_ = new com.google.protobuf.RepeatedFieldBuilderV3<
org.tensorflow.framework.KernelDef.AttrConstraint, org.tensorflow.framework.KernelDef.AttrConstraint.Builder, org.tensorflow.framework.KernelDef.AttrConstraintOrBuilder>(
constraint_,
((bitField0_ & 0x00000004) == 0x00000004),
getParentForChildren(),
isClean());
constraint_ = null;
}
return constraintBuilder_;
}
private com.google.protobuf.LazyStringList hostMemoryArg_ = com.google.protobuf.LazyStringArrayList.EMPTY;
private void ensureHostMemoryArgIsMutable() {
if (!((bitField0_ & 0x00000008) == 0x00000008)) {
hostMemoryArg_ = new com.google.protobuf.LazyStringArrayList(hostMemoryArg_);
bitField0_ |= 0x00000008;
}
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public com.google.protobuf.ProtocolStringList
getHostMemoryArgList() {
return hostMemoryArg_.getUnmodifiableView();
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public int getHostMemoryArgCount() {
return hostMemoryArg_.size();
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public java.lang.String getHostMemoryArg(int index) {
return hostMemoryArg_.get(index);
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public com.google.protobuf.ByteString
getHostMemoryArgBytes(int index) {
return hostMemoryArg_.getByteString(index);
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public Builder setHostMemoryArg(
int index, java.lang.String value) {
if (value == null) {
throw new NullPointerException();
}
ensureHostMemoryArgIsMutable();
hostMemoryArg_.set(index, value);
onChanged();
return this;
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public Builder addHostMemoryArg(
java.lang.String value) {
if (value == null) {
throw new NullPointerException();
}
ensureHostMemoryArgIsMutable();
hostMemoryArg_.add(value);
onChanged();
return this;
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public Builder addAllHostMemoryArg(
java.lang.Iterable values) {
ensureHostMemoryArgIsMutable();
com.google.protobuf.AbstractMessageLite.Builder.addAll(
values, hostMemoryArg_);
onChanged();
return this;
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public Builder clearHostMemoryArg() {
hostMemoryArg_ = com.google.protobuf.LazyStringArrayList.EMPTY;
bitField0_ = (bitField0_ & ~0x00000008);
onChanged();
return this;
}
/**
*
* Names of the Op's input_/output_args that reside in host memory
* instead of device memory.
*
*
* repeated string host_memory_arg = 4;
*/
public Builder addHostMemoryArgBytes(
com.google.protobuf.ByteString value) {
if (value == null) {
throw new NullPointerException();
}
checkByteStringIsUtf8(value);
ensureHostMemoryArgIsMutable();
hostMemoryArg_.add(value);
onChanged();
return this;
}
private java.lang.Object label_ = "";
/**
*
* 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;
*/
public java.lang.String getLabel() {
java.lang.Object ref = label_;
if (!(ref instanceof java.lang.String)) {
com.google.protobuf.ByteString bs =
(com.google.protobuf.ByteString) ref;
java.lang.String s = bs.toStringUtf8();
label_ = s;
return s;
} else {
return (java.lang.String) ref;
}
}
/**
*
* 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;
*/
public com.google.protobuf.ByteString
getLabelBytes() {
java.lang.Object ref = label_;
if (ref instanceof String) {
com.google.protobuf.ByteString b =
com.google.protobuf.ByteString.copyFromUtf8(
(java.lang.String) ref);
label_ = b;
return b;
} else {
return (com.google.protobuf.ByteString) ref;
}
}
/**
*
* 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;
*/
public Builder setLabel(
java.lang.String value) {
if (value == null) {
throw new NullPointerException();
}
label_ = value;
onChanged();
return this;
}
/**
*
* 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;
*/
public Builder clearLabel() {
label_ = getDefaultInstance().getLabel();
onChanged();
return this;
}
/**
*
* 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;
*/
public Builder setLabelBytes(
com.google.protobuf.ByteString value) {
if (value == null) {
throw new NullPointerException();
}
checkByteStringIsUtf8(value);
label_ = value;
onChanged();
return this;
}
private int priority_ ;
/**
*
* 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;
*/
public int getPriority() {
return priority_;
}
/**
*
* 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;
*/
public Builder setPriority(int value) {
priority_ = value;
onChanged();
return this;
}
/**
*
* 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;
*/
public Builder clearPriority() {
priority_ = 0;
onChanged();
return this;
}
public final Builder setUnknownFields(
final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.setUnknownFieldsProto3(unknownFields);
}
public final Builder mergeUnknownFields(
final com.google.protobuf.UnknownFieldSet unknownFields) {
return super.mergeUnknownFields(unknownFields);
}
// @@protoc_insertion_point(builder_scope:tensorflow.KernelDef)
}
// @@protoc_insertion_point(class_scope:tensorflow.KernelDef)
private static final org.tensorflow.framework.KernelDef DEFAULT_INSTANCE;
static {
DEFAULT_INSTANCE = new org.tensorflow.framework.KernelDef();
}
public static org.tensorflow.framework.KernelDef getDefaultInstance() {
return DEFAULT_INSTANCE;
}
private static final com.google.protobuf.Parser
PARSER = new com.google.protobuf.AbstractParser() {
public KernelDef parsePartialFrom(
com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry)
throws com.google.protobuf.InvalidProtocolBufferException {
return new KernelDef(input, extensionRegistry);
}
};
public static com.google.protobuf.Parser parser() {
return PARSER;
}
@java.lang.Override
public com.google.protobuf.Parser getParserForType() {
return PARSER;
}
public org.tensorflow.framework.KernelDef getDefaultInstanceForType() {
return DEFAULT_INSTANCE;
}
}