public interface
KernelDefOrBuilder
Known Indirect Subclasses |
Public Methods
abstract KernelDef.AttrConstraint |
getConstraint(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
abstract int |
getConstraintCount()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
abstract List<KernelDef.AttrConstraint> |
getConstraintList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
abstract KernelDef.AttrConstraintOrBuilder |
getConstraintOrBuilder(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
abstract List<? extends KernelDef.AttrConstraintOrBuilder> |
getConstraintOrBuilderList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
|
abstract String |
getDeviceType()
Type of device this kernel runs on. |
abstract com.google.protobuf.ByteString |
getDeviceTypeBytes()
Type of device this kernel runs on. |
abstract String |
getHostMemoryArg(int index)
Names of the Op's input_/output_args that reside in host memory instead of device memory. |
abstract com.google.protobuf.ByteString |
getHostMemoryArgBytes(int index)
Names of the Op's input_/output_args that reside in host memory instead of device memory. |
abstract int |
getHostMemoryArgCount()
Names of the Op's input_/output_args that reside in host memory instead of device memory. |
abstract List<String> |
getHostMemoryArgList()
Names of the Op's input_/output_args that reside in host memory instead of device memory. |
abstract 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. |
abstract 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. |
abstract String |
getOp()
Must match the name of an Op. |
abstract com.google.protobuf.ByteString |
getOpBytes()
Must match the name of an Op. |
abstract int |
getPriority()
Prioritization of kernel amongst different devices. |
Public Methods
public abstract KernelDef.AttrConstraint getConstraint (int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public abstract int getConstraintCount ()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public abstract List<KernelDef.AttrConstraint> getConstraintList ()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public abstract KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder (int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public abstract List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList ()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
public abstract String getDeviceType ()
Type of device this kernel runs on.
string device_type = 2;
public abstract com.google.protobuf.ByteString getDeviceTypeBytes ()
Type of device this kernel runs on.
string device_type = 2;
public abstract 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 abstract 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 abstract 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 abstract List<String> 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 abstract 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 abstract 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 abstract String getOp ()
Must match the name of an Op.
string op = 1;
public abstract com.google.protobuf.ByteString getOpBytes ()
Must match the name of an Op.
string op = 1;
public abstract 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;