python

Documentation

Index

• Variables
• type DictKeysProto
  • func (*DictKeysProto) Descriptor() ([]byte, []int)
  • func (x *DictKeysProto) GetStrId() []uint32
  • func (*DictKeysProto) ProtoMessage()
  • func (x *DictKeysProto) ProtoReflect() protoreflect.Message
  • func (x *DictKeysProto) Reset()
  • func (x *DictKeysProto) String() string
• type PyHostCallbackProto
  • func (*PyHostCallbackProto) Descriptor() ([]byte, []int)
  • func (x *PyHostCallbackProto) GetCallable() []byte
  • func (*PyHostCallbackProto) ProtoMessage()
  • func (x *PyHostCallbackProto) ProtoReflect() protoreflect.Message
  • func (x *PyHostCallbackProto) Reset()
  • func (x *PyHostCallbackProto) String() string
• type PyTreeDefProto
  • func (*PyTreeDefProto) Descriptor() ([]byte, []int)
  • func (x *PyTreeDefProto) GetInternedStrings() []string
  • func (x *PyTreeDefProto) GetNodes() []*PyTreeNodeDefProto
  • func (*PyTreeDefProto) ProtoMessage()
  • func (x *PyTreeDefProto) ProtoReflect() protoreflect.Message
  • func (x *PyTreeDefProto) Reset()
  • func (x *PyTreeDefProto) String() string
• type PyTreeNodeDefProto
  • func (*PyTreeNodeDefProto) Descriptor() ([]byte, []int)
  • func (x *PyTreeNodeDefProto) GetArity() uint32
  • func (x *PyTreeNodeDefProto) GetDictKeys() *DictKeysProto
  • func (x *PyTreeNodeDefProto) GetType() PyTreeNodeType
  • func (*PyTreeNodeDefProto) ProtoMessage()
  • func (x *PyTreeNodeDefProto) ProtoReflect() protoreflect.Message
  • func (x *PyTreeNodeDefProto) Reset()
  • func (x *PyTreeNodeDefProto) String() string
• type PyTreeNodeType
  • func (PyTreeNodeType) Descriptor() protoreflect.EnumDescriptor
  • func (x PyTreeNodeType) Enum() *PyTreeNodeType
  • func (PyTreeNodeType) EnumDescriptor() ([]byte, []int)
  • func (x PyTreeNodeType) Number() protoreflect.EnumNumber
  • func (x PyTreeNodeType) String() string
  • func (PyTreeNodeType) Type() protoreflect.EnumType

Variables

var (
	PyTreeNodeType_name = map[int32]string{
		0: "PY_TREE_KIND_INVALID",
		1: "PY_TREE_KIND_LEAF",
		2: "PY_TREE_KIND_LIST",
		3: "PY_TREE_KIND_NONE",
		4: "PY_TREE_KIND_TUPLE",
		5: "PY_TREE_KIND_DICT",
	}
	PyTreeNodeType_value = map[string]int32{
		"PY_TREE_KIND_INVALID": 0,
		"PY_TREE_KIND_LEAF":    1,
		"PY_TREE_KIND_LIST":    2,
		"PY_TREE_KIND_NONE":    3,
		"PY_TREE_KIND_TUPLE":   4,
		"PY_TREE_KIND_DICT":    5,
	}
)

Enum value maps for PyTreeNodeType.

var File_tensorflow_compiler_xla_python_py_host_callback_proto protoreflect.FileDescriptor

var File_tensorflow_compiler_xla_python_pytree_proto protoreflect.FileDescriptor

Types

type DictKeysProto

type DictKeysProto struct {
	StrId []uint32 `protobuf:"varint,1,rep,packed,name=str_id,json=strId,proto3" json:"str_id,omitempty"`
	// contains filtered or unexported fields
}

func (*DictKeysProto) Descriptor

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

Deprecated: Use DictKeysProto.ProtoReflect.Descriptor instead.

func (*DictKeysProto) GetStrId

func (x *DictKeysProto) GetStrId() []uint32

func (*DictKeysProto) ProtoMessage

func (*DictKeysProto) ProtoMessage()

func (*DictKeysProto) ProtoReflect

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

func (*DictKeysProto) Reset

func (x *DictKeysProto) Reset()

func (*DictKeysProto) String

func (x *DictKeysProto) String() string

type PyHostCallbackProto

type PyHostCallbackProto struct {
	Callable []byte `protobuf:"bytes,1,opt,name=callable,proto3" json:"callable,omitempty"`
	// contains filtered or unexported fields
}

Represents a JAX host callback that is serialized using the 'cloudpickle' Python library. Typically used for `xla.ifrt.XlaHostCallbackProto.serialized_callback`.

func (*PyHostCallbackProto) Descriptor

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

Deprecated: Use PyHostCallbackProto.ProtoReflect.Descriptor instead.

func (*PyHostCallbackProto) GetCallable

func (x *PyHostCallbackProto) GetCallable() []byte

func (*PyHostCallbackProto) ProtoMessage

func (*PyHostCallbackProto) ProtoMessage()

func (*PyHostCallbackProto) ProtoReflect

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

func (*PyHostCallbackProto) Reset

func (x *PyHostCallbackProto) Reset()

func (*PyHostCallbackProto) String

func (x *PyHostCallbackProto) String() string

type PyTreeDefProto

type PyTreeDefProto struct {
	Nodes []*PyTreeNodeDefProto `protobuf:"bytes,1,rep,name=nodes,proto3" json:"nodes,omitempty"`
	// Extra strings.
	InternedStrings []string `protobuf:"bytes,2,rep,name=interned_strings,json=internedStrings,proto3" json:"interned_strings,omitempty"`
	// contains filtered or unexported fields
}

A Pytree.

func (*PyTreeDefProto) Descriptor

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

Deprecated: Use PyTreeDefProto.ProtoReflect.Descriptor instead.

func (*PyTreeDefProto) GetInternedStrings

func (x *PyTreeDefProto) GetInternedStrings() []string

func (*PyTreeDefProto) GetNodes

func (x *PyTreeDefProto) GetNodes() []*PyTreeNodeDefProto

func (*PyTreeDefProto) ProtoMessage

func (*PyTreeDefProto) ProtoMessage()

func (*PyTreeDefProto) ProtoReflect

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

func (*PyTreeDefProto) Reset

func (x *PyTreeDefProto) Reset()

func (*PyTreeDefProto) String

func (x *PyTreeDefProto) String() string

type PyTreeNodeDefProto

type PyTreeNodeDefProto struct {

	// Recovers the tree structure.
	Arity uint32 `protobuf:"varint,1,opt,name=arity,proto3" json:"arity,omitempty"`
	// Node type.
	Type PyTreeNodeType `protobuf:"varint,2,opt,name=type,proto3,enum=jax.PyTreeNodeType" json:"type,omitempty"`
	// Only set when type == DICT.
	DictKeys *DictKeysProto `protobuf:"bytes,3,opt,name=dict_keys,json=dictKeys,proto3" json:"dict_keys,omitempty"`
	// contains filtered or unexported fields
}

func (*PyTreeNodeDefProto) Descriptor

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

Deprecated: Use PyTreeNodeDefProto.ProtoReflect.Descriptor instead.

func (*PyTreeNodeDefProto) GetArity

func (x *PyTreeNodeDefProto) GetArity() uint32

func (*PyTreeNodeDefProto) GetDictKeys

func (x *PyTreeNodeDefProto) GetDictKeys() *DictKeysProto

func (*PyTreeNodeDefProto) GetType

func (x *PyTreeNodeDefProto) GetType() PyTreeNodeType

func (*PyTreeNodeDefProto) ProtoMessage

func (*PyTreeNodeDefProto) ProtoMessage()

func (*PyTreeNodeDefProto) ProtoReflect

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

func (*PyTreeNodeDefProto) Reset

func (x *PyTreeNodeDefProto) Reset()

func (*PyTreeNodeDefProto) String

func (x *PyTreeNodeDefProto) String() string

type PyTreeNodeType

type PyTreeNodeType int32

const (
	PyTreeNodeType_PY_TREE_KIND_INVALID PyTreeNodeType = 0
	PyTreeNodeType_PY_TREE_KIND_LEAF    PyTreeNodeType = 1
	PyTreeNodeType_PY_TREE_KIND_LIST    PyTreeNodeType = 2
	PyTreeNodeType_PY_TREE_KIND_NONE    PyTreeNodeType = 3
	PyTreeNodeType_PY_TREE_KIND_TUPLE   PyTreeNodeType = 4
	PyTreeNodeType_PY_TREE_KIND_DICT    PyTreeNodeType = 5
)

func (PyTreeNodeType) Descriptor

func (PyTreeNodeType) Descriptor() protoreflect.EnumDescriptor

func (PyTreeNodeType) Enum

func (x PyTreeNodeType) Enum() *PyTreeNodeType

func (PyTreeNodeType) EnumDescriptor

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

Deprecated: Use PyTreeNodeType.Descriptor instead.

func (PyTreeNodeType) Number

func (x PyTreeNodeType) Number() protoreflect.EnumNumber

func (PyTreeNodeType) String

func (x PyTreeNodeType) String() string

func (PyTreeNodeType) Type

func (PyTreeNodeType) Type() protoreflect.EnumType