public interface KernelDefOrBuilder
extends com.google.protobuf.MessageOrBuilder
Modifier and Type | Method and 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.
|
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.
|
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.
|
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.
|
findInitializationErrors, getAllFields, getDefaultInstanceForType, getDescriptorForType, getField, getInitializationErrorString, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, getUnknownFields, hasField, hasOneof
String getOp()
Must match the name of an Op.
string op = 1;
com.google.protobuf.ByteString getOpBytes()
Must match the name of an Op.
string op = 1;
String getDeviceType()
Type of device this kernel runs on.
string device_type = 2;
com.google.protobuf.ByteString getDeviceTypeBytes()
Type of device this kernel runs on.
string device_type = 2;
List<KernelDef.AttrConstraint> getConstraintList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
KernelDef.AttrConstraint getConstraint(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
int getConstraintCount()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
List<? extends KernelDef.AttrConstraintOrBuilder> getConstraintOrBuilderList()
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
KernelDef.AttrConstraintOrBuilder getConstraintOrBuilder(int index)
repeated .tensorflow.KernelDef.AttrConstraint constraint = 3;
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;
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;
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;
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;
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;
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;
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;
Copyright © 2015–2019. All rights reserved.