KernelDef

Stay organized with collections Save and categorize content based on your preferences.
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
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 .tensor