Package org.tensorflow.framework
Interface KernelDefOrBuilder
-
- All Superinterfaces:
org.nd4j.shade.protobuf.MessageLiteOrBuilder,org.nd4j.shade.protobuf.MessageOrBuilder
- All Known Implementing Classes:
KernelDef,KernelDef.Builder
public interface KernelDefOrBuilder extends org.nd4j.shade.protobuf.MessageOrBuilder
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description KernelDef.AttrConstraintgetConstraint(int index)repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;intgetConstraintCount()repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;List<KernelDef.AttrConstraint>getConstraintList()repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;KernelDef.AttrConstraintOrBuildergetConstraintOrBuilder(int index)repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;List<? extends KernelDef.AttrConstraintOrBuilder>getConstraintOrBuilderList()repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;StringgetDeviceType()Type of device this kernel runs on.org.nd4j.shade.protobuf.ByteStringgetDeviceTypeBytes()Type of device this kernel runs on.StringgetHostMemoryArg(int index)Names of the Op's input_/output_args that reside in host memory instead of device memory.org.nd4j.shade.protobuf.ByteStringgetHostMemoryArgBytes(int index)Names of the Op's input_/output_args that reside in host memory instead of device memory.intgetHostMemoryArgCount()Names of the Op's input_/output_args that reside in host memory instead of device memory.List<String>getHostMemoryArgList()Names of the Op's input_/output_args that reside in host memory instead of device memory.StringgetLabel()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.org.nd4j.shade.protobuf.ByteStringgetLabelBytes()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.StringgetOp()Must match the name of an Op.org.nd4j.shade.protobuf.ByteStringgetOpBytes()Must match the name of an Op.-
Methods inherited from interface org.nd4j.shade.protobuf.MessageOrBuilder
findInitializationErrors, getAllFields, getDefaultInstanceForType, getDescriptorForType, getField, getInitializationErrorString, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, getUnknownFields, hasField, hasOneof
-
-
-
-
Method Detail
-
getOp
String getOp()
Must match the name of an Op.
string op = 1;- Returns:
- The op.
-
getOpBytes
org.nd4j.shade.protobuf.ByteString getOpBytes()
Must match the name of an Op.
string op = 1;- Returns:
- The bytes for op.
-
getDeviceType
String getDeviceType()
Type of device this kernel runs on.
string device_type = 2;- Returns:
- The deviceType.
-
getDeviceTypeBytes
org.nd4j.shade.protobuf.ByteString getDeviceTypeBytes()
Type of device this kernel runs on.
string device_type = 2;- Returns:
- The bytes for deviceType.
-
getConstraintList
List<KernelDef.AttrConstraint> getConstraintList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
-
getConstraint
KernelDef.AttrConstraint getConstraint(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
-
getConstraintCount
int getConstraintCount()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
-
getConstraintOrBuilderList
List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
-
getConstraintOrBuilder
KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
-
getHostMemoryArgList
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;- Returns:
- A list containing the hostMemoryArg.
-
getHostMemoryArgCount
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;- Returns:
- The count of hostMemoryArg.
-
getHostMemoryArg
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;- Parameters:
index- The index of the element to return.- Returns:
- The hostMemoryArg at the given index.
-
getHostMemoryArgBytes
org.nd4j.shade.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;- Parameters:
index- The index of the value to return.- Returns:
- The bytes of the hostMemoryArg at the given index.
-
getLabel
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;- Returns:
- The label.
-
getLabelBytes
org.nd4j.shade.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;- Returns:
- The bytes for label.
-
-