Documentation ¶
Index ¶
- Variables
- type KernelDef
- func (*KernelDef) Descriptor() ([]byte, []int)deprecated
- func (x *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
- func (x *KernelDef) GetDeviceType() string
- func (x *KernelDef) GetHostMemoryArg() []string
- func (x *KernelDef) GetLabel() string
- func (x *KernelDef) GetOp() string
- func (x *KernelDef) GetPriority() int32
- func (*KernelDef) ProtoMessage()
- func (x *KernelDef) ProtoReflect() protoreflect.Message
- func (x *KernelDef) Reset()
- func (x *KernelDef) String() string
- type KernelDef_AttrConstraint
- func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)deprecated
- func (x *KernelDef_AttrConstraint) GetAllowedValues() *attr_value_go_proto.AttrValue
- func (x *KernelDef_AttrConstraint) GetName() string
- func (*KernelDef_AttrConstraint) ProtoMessage()
- func (x *KernelDef_AttrConstraint) ProtoReflect() protoreflect.Message
- func (x *KernelDef_AttrConstraint) Reset()
- func (x *KernelDef_AttrConstraint) String() string
- type KernelList
Constants ¶
This section is empty.
Variables ¶
View Source
var File_tensorflow_core_framework_kernel_def_proto protoreflect.FileDescriptor
Functions ¶
This section is empty.
Types ¶
type KernelDef ¶
type KernelDef struct { // Must match the name of an Op. Op string `protobuf:"bytes,1,opt,name=op,proto3" json:"op,omitempty"` // Type of device this kernel runs on. DeviceType string `protobuf:"bytes,2,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"` Constraint []*KernelDef_AttrConstraint `protobuf:"bytes,3,rep,name=constraint,proto3" json:"constraint,omitempty"` // Names of the Op's input_/output_args that reside in host memory // instead of device memory. HostMemoryArg []string `protobuf:"bytes,4,rep,name=host_memory_arg,json=hostMemoryArg,proto3" json:"host_memory_arg,omitempty"` // 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. Label string `protobuf:"bytes,5,opt,name=label,proto3" json:"label,omitempty"` // 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. Priority int32 `protobuf:"varint,6,opt,name=priority,proto3" json:"priority,omitempty"` // contains filtered or unexported fields }
func (*KernelDef) Descriptor
deprecated
func (*KernelDef) GetConstraint ¶
func (x *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
func (*KernelDef) GetDeviceType ¶
func (*KernelDef) GetHostMemoryArg ¶
func (*KernelDef) GetPriority ¶
func (*KernelDef) ProtoMessage ¶
func (*KernelDef) ProtoMessage()
func (*KernelDef) ProtoReflect ¶
func (x *KernelDef) ProtoReflect() protoreflect.Message
type KernelDef_AttrConstraint ¶
type KernelDef_AttrConstraint struct { // Name of an attr from the Op. Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` // A list of values that this kernel supports for this attr. // Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops. AllowedValues *attr_value_go_proto.AttrValue `protobuf:"bytes,2,opt,name=allowed_values,json=allowedValues,proto3" json:"allowed_values,omitempty"` // contains filtered or unexported fields }
func (*KernelDef_AttrConstraint) Descriptor
deprecated
func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)
Deprecated: Use KernelDef_AttrConstraint.ProtoReflect.Descriptor instead.
func (*KernelDef_AttrConstraint) GetAllowedValues ¶
func (x *KernelDef_AttrConstraint) GetAllowedValues() *attr_value_go_proto.AttrValue
func (*KernelDef_AttrConstraint) GetName ¶
func (x *KernelDef_AttrConstraint) GetName() string
func (*KernelDef_AttrConstraint) ProtoMessage ¶
func (*KernelDef_AttrConstraint) ProtoMessage()
func (*KernelDef_AttrConstraint) ProtoReflect ¶
func (x *KernelDef_AttrConstraint) ProtoReflect() protoreflect.Message
func (*KernelDef_AttrConstraint) Reset ¶
func (x *KernelDef_AttrConstraint) Reset()
func (*KernelDef_AttrConstraint) String ¶
func (x *KernelDef_AttrConstraint) String() string
type KernelList ¶
type KernelList struct { Kernel []*KernelDef `protobuf:"bytes,1,rep,name=kernel,proto3" json:"kernel,omitempty"` // contains filtered or unexported fields }
A collection of KernelDefs
func (*KernelList) Descriptor
deprecated
func (*KernelList) Descriptor() ([]byte, []int)
Deprecated: Use KernelList.ProtoReflect.Descriptor instead.
func (*KernelList) GetKernel ¶
func (x *KernelList) GetKernel() []*KernelDef
func (*KernelList) ProtoMessage ¶
func (*KernelList) ProtoMessage()
func (*KernelList) ProtoReflect ¶
func (x *KernelList) ProtoReflect() protoreflect.Message
func (*KernelList) Reset ¶
func (x *KernelList) Reset()
func (*KernelList) String ¶
func (x *KernelList) String() string
Click to show internal directories.
Click to hide internal directories.