Documentation ¶
Index ¶
- type KernelDef
- func (*KernelDef) Descriptor() ([]byte, []int)
- func (m *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
- func (m *KernelDef) GetDeviceType() string
- func (m *KernelDef) GetHostMemoryArg() []string
- func (m *KernelDef) GetLabel() string
- func (m *KernelDef) GetOp() string
- func (m *KernelDef) GetPriority() int32
- func (*KernelDef) ProtoMessage()
- func (m *KernelDef) Reset()
- func (m *KernelDef) String() string
- func (m *KernelDef) XXX_DiscardUnknown()
- func (m *KernelDef) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)
- func (m *KernelDef) XXX_Merge(src proto.Message)
- func (m *KernelDef) XXX_Size() int
- func (m *KernelDef) XXX_Unmarshal(b []byte) error
- type KernelDef_AttrConstraint
- func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)
- func (m *KernelDef_AttrConstraint) GetAllowedValues() *attr_value_go_proto.AttrValue
- func (m *KernelDef_AttrConstraint) GetName() string
- func (*KernelDef_AttrConstraint) ProtoMessage()
- func (m *KernelDef_AttrConstraint) Reset()
- func (m *KernelDef_AttrConstraint) String() string
- func (m *KernelDef_AttrConstraint) XXX_DiscardUnknown()
- func (m *KernelDef_AttrConstraint) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)
- func (m *KernelDef_AttrConstraint) XXX_Merge(src proto.Message)
- func (m *KernelDef_AttrConstraint) XXX_Size() int
- func (m *KernelDef_AttrConstraint) XXX_Unmarshal(b []byte) error
- type KernelList
- func (*KernelList) Descriptor() ([]byte, []int)
- func (m *KernelList) GetKernel() []*KernelDef
- func (*KernelList) ProtoMessage()
- func (m *KernelList) Reset()
- func (m *KernelList) String() string
- func (m *KernelList) XXX_DiscardUnknown()
- func (m *KernelList) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)
- func (m *KernelList) XXX_Merge(src proto.Message)
- func (m *KernelList) XXX_Size() int
- func (m *KernelList) XXX_Unmarshal(b []byte) error
Constants ¶
This section is empty.
Variables ¶
This section is empty.
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"` XXX_NoUnkeyedLiteral struct{} `json:"-"` XXX_unrecognized []byte `json:"-"` XXX_sizecache int32 `json:"-"` }
func (*KernelDef) Descriptor ¶
func (*KernelDef) GetConstraint ¶
func (m *KernelDef) GetConstraint() []*KernelDef_AttrConstraint
func (*KernelDef) GetDeviceType ¶
func (*KernelDef) GetHostMemoryArg ¶
func (*KernelDef) GetPriority ¶
func (*KernelDef) ProtoMessage ¶
func (*KernelDef) ProtoMessage()
func (*KernelDef) XXX_DiscardUnknown ¶
func (m *KernelDef) XXX_DiscardUnknown()
func (*KernelDef) XXX_Marshal ¶
func (*KernelDef) XXX_Unmarshal ¶
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"` XXX_NoUnkeyedLiteral struct{} `json:"-"` XXX_unrecognized []byte `json:"-"` XXX_sizecache int32 `json:"-"` }
func (*KernelDef_AttrConstraint) Descriptor ¶
func (*KernelDef_AttrConstraint) Descriptor() ([]byte, []int)
func (*KernelDef_AttrConstraint) GetAllowedValues ¶
func (m *KernelDef_AttrConstraint) GetAllowedValues() *attr_value_go_proto.AttrValue
func (*KernelDef_AttrConstraint) GetName ¶
func (m *KernelDef_AttrConstraint) GetName() string
func (*KernelDef_AttrConstraint) ProtoMessage ¶
func (*KernelDef_AttrConstraint) ProtoMessage()
func (*KernelDef_AttrConstraint) Reset ¶
func (m *KernelDef_AttrConstraint) Reset()
func (*KernelDef_AttrConstraint) String ¶
func (m *KernelDef_AttrConstraint) String() string
func (*KernelDef_AttrConstraint) XXX_DiscardUnknown ¶
func (m *KernelDef_AttrConstraint) XXX_DiscardUnknown()
func (*KernelDef_AttrConstraint) XXX_Marshal ¶
func (m *KernelDef_AttrConstraint) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)
func (*KernelDef_AttrConstraint) XXX_Merge ¶
func (m *KernelDef_AttrConstraint) XXX_Merge(src proto.Message)
func (*KernelDef_AttrConstraint) XXX_Size ¶
func (m *KernelDef_AttrConstraint) XXX_Size() int
func (*KernelDef_AttrConstraint) XXX_Unmarshal ¶
func (m *KernelDef_AttrConstraint) XXX_Unmarshal(b []byte) error
type KernelList ¶
type KernelList struct { Kernel []*KernelDef `protobuf:"bytes,1,rep,name=kernel,proto3" json:"kernel,omitempty"` XXX_NoUnkeyedLiteral struct{} `json:"-"` XXX_unrecognized []byte `json:"-"` XXX_sizecache int32 `json:"-"` }
A collection of KernelDefs
func (*KernelList) Descriptor ¶
func (*KernelList) Descriptor() ([]byte, []int)
func (*KernelList) GetKernel ¶
func (m *KernelList) GetKernel() []*KernelDef
func (*KernelList) ProtoMessage ¶
func (*KernelList) ProtoMessage()
func (*KernelList) Reset ¶
func (m *KernelList) Reset()
func (*KernelList) String ¶
func (m *KernelList) String() string
func (*KernelList) XXX_DiscardUnknown ¶
func (m *KernelList) XXX_DiscardUnknown()
func (*KernelList) XXX_Marshal ¶
func (m *KernelList) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)
func (*KernelList) XXX_Merge ¶
func (m *KernelList) XXX_Merge(src proto.Message)
func (*KernelList) XXX_Size ¶
func (m *KernelList) XXX_Size() int
func (*KernelList) XXX_Unmarshal ¶
func (m *KernelList) XXX_Unmarshal(b []byte) error
Click to show internal directories.
Click to hide internal directories.