tf2xlav1

Documentation

Index

• Variables
• type DeviceType
  • func (DeviceType) Descriptor() protoreflect.EnumDescriptor
  • func (x DeviceType) Enum() *DeviceType
  • func (DeviceType) EnumDescriptor() ([]byte, []int)
  • func (x DeviceType) Number() protoreflect.EnumNumber
  • func (x DeviceType) String() string
  • func (DeviceType) Type() protoreflect.EnumType
  • func (x *DeviceType) UnmarshalJSON(b []byte) error

Variables

var (
	DeviceType_name = map[int32]string{
		0: "DEVICE_TYPE_UNSPECIFIED",
		1: "XLA_TPU_JIT",
		2: "XLA_CPU_JIT",
		3: "XLA_GPU_JIT",
	}
	DeviceType_value = map[string]int32{
		"DEVICE_TYPE_UNSPECIFIED": 0,
		"XLA_TPU_JIT":             1,
		"XLA_CPU_JIT":             2,
		"XLA_GPU_JIT":             3,
	}
)

Enum value maps for DeviceType.

var File_tensorflow_compiler_mlir_tf2xla_api_v1_device_type_proto protoreflect.FileDescriptor

Types

type DeviceType

type DeviceType int32

The requested device type to compile for.

const (
	DeviceType_DEVICE_TYPE_UNSPECIFIED DeviceType = 0
	DeviceType_XLA_TPU_JIT             DeviceType = 1
	DeviceType_XLA_CPU_JIT             DeviceType = 2
	DeviceType_XLA_GPU_JIT             DeviceType = 3
)

func (DeviceType) Descriptor

func (DeviceType) Descriptor() protoreflect.EnumDescriptor

func (DeviceType) Enum

func (x DeviceType) Enum() *DeviceType

func (DeviceType) EnumDescriptor

func (DeviceType) EnumDescriptor() ([]byte, []int)

Deprecated: Use DeviceType.Descriptor instead.

func (DeviceType) Number

func (x DeviceType) Number() protoreflect.EnumNumber

func (DeviceType) String

func (x DeviceType) String() string

func (DeviceType) Type

func (DeviceType) Type() protoreflect.EnumType

func (*DeviceType) UnmarshalJSON

func (x *DeviceType) UnmarshalJSON(b []byte) error

Deprecated: Do not use.