gpu

Documentation

Index

• Variables
• type GpuTopologyProto
  • func (*GpuTopologyProto) Descriptor() ([]byte, []int)
  • func (x *GpuTopologyProto) GetDeviceIds() []int32
  • func (*GpuTopologyProto) ProtoMessage()
  • func (x *GpuTopologyProto) ProtoReflect() protoreflect.Message
  • func (x *GpuTopologyProto) Reset()
  • func (x *GpuTopologyProto) String() string

Variables

var File_tensorflow_compiler_xla_pjrt_gpu_gpu_topology_proto protoreflect.FileDescriptor

Types

type GpuTopologyProto

type GpuTopologyProto struct {
	DeviceIds []int32 `protobuf:"varint,1,rep,packed,name=device_ids,json=deviceIds,proto3" json:"device_ids,omitempty"`
	// contains filtered or unexported fields
}

A proto used to serialize GpuTopology instances.

func (*GpuTopologyProto) Descriptor

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

Deprecated: Use GpuTopologyProto.ProtoReflect.Descriptor instead.

func (*GpuTopologyProto) GetDeviceIds

func (x *GpuTopologyProto) GetDeviceIds() []int32

func (*GpuTopologyProto) ProtoMessage

func (*GpuTopologyProto) ProtoMessage()

func (*GpuTopologyProto) ProtoReflect

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

func (*GpuTopologyProto) Reset

func (x *GpuTopologyProto) Reset()

func (*GpuTopologyProto) String

func (x *GpuTopologyProto) String() string