kernel_def_go_proto

Documentation

Index

• Variables
• type KernelDef
  • func (*KernelDef) Descriptor() ([]byte, []int)
  • 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)
  • 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
  • func (*KernelList) Descriptor() ([]byte, []int)
  • func (x *KernelList) GetKernel() []*KernelDef
  • func (*KernelList) ProtoMessage()
  • func (x *KernelList) ProtoReflect() protoreflect.Message
  • func (x *KernelList) Reset()
  • func (x *KernelList) String() string

Variables

var File_tensorflow_core_framework_kernel_def_proto protoreflect.FileDescriptor

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

func (*KernelDef) Descriptor() ([]byte, []int)

Deprecated: Use KernelDef.ProtoReflect.Descriptor instead.

func (*KernelDef) GetConstraint

func (x *KernelDef) GetConstraint() []*KernelDef_AttrConstraint

func (*KernelDef) GetDeviceType

func (x *KernelDef) GetDeviceType() string

func (*KernelDef) GetHostMemoryArg

func (x *KernelDef) GetHostMemoryArg() []string

func (*KernelDef) GetLabel

func (x *KernelDef) GetLabel() string

func (*KernelDef) GetOp

func (x *KernelDef) GetOp() string

func (*KernelDef) GetPriority

func (x *KernelDef) GetPriority() int32

func (*KernelDef) ProtoMessage

func (*KernelDef) ProtoMessage()

func (*KernelDef) ProtoReflect

func (x *KernelDef) ProtoReflect() protoreflect.Message

func (*KernelDef) Reset

func (x *KernelDef) Reset()

func (*KernelDef) String

func (x *KernelDef) String() string

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

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

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