public final class
KernelDef
Protobuf type tensorflow.KernelDef
Nested Classes
class | KernelDef.AttrConstraint | Protobuf type tensorflow.KernelDef.AttrConstraint
|
|
interface | KernelDef.AttrConstraintOrBuilder | ||
class | KernelDef.Builder | Protobuf type tensorflow.KernelDef
|
Constants
int | CONSTRAINT_FIELD_NUMBER | |
int | DEVICE_TYPE_FIELD_NUMBER | |
int | HOST_MEMORY_ARG_FIELD_NUMBER | |
int | LABEL_FIELD_NUMBER | |
int | OP_FIELD_NUMBER | |
int | PRIORITY_FIELD_NUMBER |
Public Methods
boolean |
equals(Object obj)
|
KernelDef.AttrConstraint |
getConstraint(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
int |
getConstraintCount()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
List<KernelDef.AttrConstraint> |
getConstraintList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
KernelDef.AttrConstraintOrBuilder |
getConstraintOrBuilder(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
List<? extends KernelDef.AttrConstraintOrBuilder> |
getConstraintOrBuilderList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
static KernelDef | |
KernelDef | |
final static com.google.protobuf.Descriptors.Descriptor | |
String |
getDeviceType()
Type of device this kernel runs on. |
com.google.protobuf.ByteString |
getDeviceTypeBytes()
Type of device this kernel runs on. |
String |
getHostMemoryArg(int index)
Names of the Op's input_/output_args that reside in host memory instead of device memory. |
com.google.protobuf.ByteString |
getHostMemoryArgBytes(int index)
Names of the Op's input_/output_args that reside in host memory instead of device memory. |
int |
getHostMemoryArgCount()
Names of the Op's input_/output_args that reside in host memory instead of device memory. |
com.google.protobuf.ProtocolStringList |
getHostMemoryArgList()
Names of the Op's input_/output_args that reside in host memory instead of device memory. |
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. |
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 |
getOp()
Must match the name of an Op. |
com.google.protobuf.ByteString |
getOpBytes()
Must match the name of an Op. |
|
|
int |
getPriority()
Prioritization of kernel amongst different devices. |
int | |
final com.google.protobuf.UnknownFieldSet | |
int |
hashCode()
|
final boolean | |
static KernelDef.Builder | |
static KernelDef.Builder |
newBuilder(KernelDef prototype)
|
KernelDef.Builder | |
static KernelDef |
parseDelimitedFrom(InputStream input)
|
static KernelDef |
parseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static KernelDef |
parseFrom(ByteBuffer data)
|
static KernelDef |
parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static KernelDef |
parseFrom(ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static KernelDef |
parseFrom(com.google.protobuf.CodedInputStream input)
|
static KernelDef |
parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static KernelDef |
parseFrom(com.google.protobuf.ByteString data)
|
static KernelDef |
parseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static KernelDef |
parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
static
|
parser()
|
KernelDef.Builder | |
void |
writeTo(com.google.protobuf.CodedOutputStream output)
|
Inherited Methods
Constants
public static final int CONSTRAINT_FIELD_NUMBER
Constant Value:
3
public static final int DEVICE_TYPE_FIELD_NUMBER
Constant Value:
2
public static final int HOST_MEMORY_ARG_FIELD_NUMBER
Constant Value:
4
public static final int LABEL_FIELD_NUMBER
Constant Value:
5
public static final int OP_FIELD_NUMBER
Constant Value:
1
public static final int PRIORITY_FIELD_NUMBER
Constant Value:
6
Public Methods
public boolean equals (Object obj)
public KernelDef.AttrConstraint getConstraint (int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public int getConstraintCount ()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public List<KernelDef.AttrConstraint> getConstraintList ()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder (int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList ()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor ()
public String getDeviceType ()
Type of device this kernel runs on.
string device_type = 2;
public com.google.protobuf.ByteString getDeviceTypeBytes ()
Type of device this kernel runs on.
string device_type = 2;
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;
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;
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;
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;
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;
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;
public String getOp ()
Must match the name of an Op.
string op = 1;
public com.google.protobuf.ByteString getOpBytes ()
Must match the name of an Op.
string op = 1;
public
getParserForType
()
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;
public int getSerializedSize ()
public final com.google.protobuf.UnknownFieldSet getUnknownFields ()
public int hashCode ()
public final boolean isInitialized ()
public static KernelDef parseDelimitedFrom (InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
IOException |
---|
public static KernelDef parseFrom (com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
IOException |
---|
public static KernelDef parseFrom (ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
InvalidProtocolBufferException |
---|
public static KernelDef parseFrom (byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
InvalidProtocolBufferException |
---|
public static KernelDef parseFrom (com.google.protobuf.ByteString data)
Throws
InvalidProtocolBufferException |
---|
public static KernelDef parseFrom (InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
IOException |
---|
public static KernelDef parseFrom (com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
InvalidProtocolBufferException |
---|
public
static
parser
()
public void writeTo (com.google.protobuf.CodedOutputStream output)
Throws
IOException |
---|