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.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;
String
getDeviceType()
Type of device this kernel runs on.org.nd4j.shade.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.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.int
getHostMemoryArgCount()
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.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.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
getOp()
Must match the name of an Op.org.nd4j.shade.protobuf.ByteString
getOpBytes()
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.
-
-