cpu

Documentation

Index

• Constants
• Variables
• type BackendConfig
  • func (*BackendConfig) Descriptor() ([]byte, []int)
  • func (x *BackendConfig) GetOuterDimensionPartitions() []int64
  • func (*BackendConfig) ProtoMessage()
  • func (x *BackendConfig) ProtoReflect() protoreflect.Message
  • func (x *BackendConfig) Reset()
  • func (x *BackendConfig) String() string
• type XlaFrameworkMappingProto
  • func (*XlaFrameworkMappingProto) Descriptor() ([]byte, []int)
  • func (x *XlaFrameworkMappingProto) GetFlattenedOutputs() []int64
  • func (x *XlaFrameworkMappingProto) GetInputs() []int64
  • func (x *XlaFrameworkMappingProto) GetOutputIsTuple() bool
  • func (x *XlaFrameworkMappingProto) GetResult() int64
  • func (*XlaFrameworkMappingProto) ProtoMessage()
  • func (x *XlaFrameworkMappingProto) ProtoReflect() protoreflect.Message
  • func (x *XlaFrameworkMappingProto) Reset()
  • func (x *XlaFrameworkMappingProto) String() string
• type XlaRuntimeCpuExecutableProto
  • func (*XlaRuntimeCpuExecutableProto) Descriptor() ([]byte, []int)
  • func (x *XlaRuntimeCpuExecutableProto) GetBufferAssignment() *service.BufferAssignmentProto
  • func (x *XlaRuntimeCpuExecutableProto) GetXlaFrameworkMapping() *XlaFrameworkMappingProto
  • func (x *XlaRuntimeCpuExecutableProto) GetXlaRuntimeExecutable() *service.XlaRuntimeExecutableProto
  • func (*XlaRuntimeCpuExecutableProto) ProtoMessage()
  • func (x *XlaRuntimeCpuExecutableProto) ProtoReflect() protoreflect.Message
  • func (x *XlaRuntimeCpuExecutableProto) Reset()
  • func (x *XlaRuntimeCpuExecutableProto) String() string

Constants

const (
	Default_XlaFrameworkMappingProto_Result = int64(-1)
)

Default values for XlaFrameworkMappingProto fields.

Variables

var File_tensorflow_compiler_xla_service_cpu_backend_config_proto protoreflect.FileDescriptor

var File_tensorflow_compiler_xla_service_cpu_executable_proto protoreflect.FileDescriptor

var File_tensorflow_compiler_xla_service_cpu_xla_framework_proto protoreflect.FileDescriptor

Types

type BackendConfig

type BackendConfig struct {

	// Number of partitions per outer dimension (in order, starting with
	// outer-most dimension first). Used by the parallel cpu backend to partition
	// HLOs into parallel tasks.
	OuterDimensionPartitions []int64 `` /* 143-byte string literal not displayed */
	// contains filtered or unexported fields
}

Backend config for XLA:CPU.

func (*BackendConfig) Descriptor

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

Deprecated: Use BackendConfig.ProtoReflect.Descriptor instead.

func (*BackendConfig) GetOuterDimensionPartitions

func (x *BackendConfig) GetOuterDimensionPartitions() []int64

func (*BackendConfig) ProtoMessage

func (*BackendConfig) ProtoMessage()

func (*BackendConfig) ProtoReflect

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

func (*BackendConfig) Reset

func (x *BackendConfig) Reset()

func (*BackendConfig) String

func (x *BackendConfig) String() string

type XlaFrameworkMappingProto

type XlaFrameworkMappingProto struct {
	Inputs           []int64 `protobuf:"varint,1,rep,packed,name=inputs" json:"inputs,omitempty"`
	FlattenedOutputs []int64 `protobuf:"varint,2,rep,packed,name=flattened_outputs,json=flattenedOutputs" json:"flattened_outputs,omitempty"`
	Result           *int64  `protobuf:"varint,3,opt,name=result,def=-1" json:"result,omitempty"`
	OutputIsTuple    *bool   `protobuf:"varint,4,opt,name=output_is_tuple,json=outputIsTuple" json:"output_is_tuple,omitempty"`
	// contains filtered or unexported fields
}

func (*XlaFrameworkMappingProto) Descriptor

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

Deprecated: Use XlaFrameworkMappingProto.ProtoReflect.Descriptor instead.

func (*XlaFrameworkMappingProto) GetFlattenedOutputs

func (x *XlaFrameworkMappingProto) GetFlattenedOutputs() []int64

func (*XlaFrameworkMappingProto) GetInputs

func (x *XlaFrameworkMappingProto) GetInputs() []int64

func (*XlaFrameworkMappingProto) GetOutputIsTuple

func (x *XlaFrameworkMappingProto) GetOutputIsTuple() bool

func (*XlaFrameworkMappingProto) GetResult

func (x *XlaFrameworkMappingProto) GetResult() int64

func (*XlaFrameworkMappingProto) ProtoMessage

func (*XlaFrameworkMappingProto) ProtoMessage()

func (*XlaFrameworkMappingProto) ProtoReflect

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

func (*XlaFrameworkMappingProto) Reset

func (x *XlaFrameworkMappingProto) Reset()

func (*XlaFrameworkMappingProto) String

func (x *XlaFrameworkMappingProto) String() string

type XlaRuntimeCpuExecutableProto

type XlaRuntimeCpuExecutableProto struct {
	XlaRuntimeExecutable *service.XlaRuntimeExecutableProto `protobuf:"bytes,1,opt,name=xla_runtime_executable,json=xlaRuntimeExecutable" json:"xla_runtime_executable,omitempty"`
	XlaFrameworkMapping  *XlaFrameworkMappingProto          `protobuf:"bytes,2,opt,name=xla_framework_mapping,json=xlaFrameworkMapping" json:"xla_framework_mapping,omitempty"`
	BufferAssignment     *service.BufferAssignmentProto     `protobuf:"bytes,3,opt,name=buffer_assignment,json=bufferAssignment" json:"buffer_assignment,omitempty"`
	// contains filtered or unexported fields
}

func (*XlaRuntimeCpuExecutableProto) Descriptor

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

Deprecated: Use XlaRuntimeCpuExecutableProto.ProtoReflect.Descriptor instead.

func (*XlaRuntimeCpuExecutableProto) GetBufferAssignment

func (x *XlaRuntimeCpuExecutableProto) GetBufferAssignment() *service.BufferAssignmentProto

func (*XlaRuntimeCpuExecutableProto) GetXlaFrameworkMapping

func (x *XlaRuntimeCpuExecutableProto) GetXlaFrameworkMapping() *XlaFrameworkMappingProto

func (*XlaRuntimeCpuExecutableProto) GetXlaRuntimeExecutable

func (x *XlaRuntimeCpuExecutableProto) GetXlaRuntimeExecutable() *service.XlaRuntimeExecutableProto

func (*XlaRuntimeCpuExecutableProto) ProtoMessage

func (*XlaRuntimeCpuExecutableProto) ProtoMessage()

func (*XlaRuntimeCpuExecutableProto) ProtoReflect

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

func (*XlaRuntimeCpuExecutableProto) Reset

func (x *XlaRuntimeCpuExecutableProto) Reset()

func (*XlaRuntimeCpuExecutableProto) String

func (x *XlaRuntimeCpuExecutableProto) String() string