Versions in this module Expand all Collapse all v2 v2.14.1 Dec 8, 2023 v2.14.0 Dec 8, 2023 Changes in this version + var File_tensorflow_compiler_xla_python_pjrt_ifrt_xla_compiler_proto protoreflect.FileDescriptor + var File_tensorflow_compiler_xla_python_pjrt_ifrt_xla_host_callback_proto protoreflect.FileDescriptor + var File_tensorflow_compiler_xla_python_pjrt_ifrt_xla_sharding_proto protoreflect.FileDescriptor + type HloShardingProto struct + Devices *ifrt.DeviceListProto + MemoryKind *string + XlaOpSharding *data.OpSharding + func (*HloShardingProto) Descriptor() ([]byte, []int) + func (*HloShardingProto) ProtoMessage() + func (x *HloShardingProto) GetDevices() *ifrt.DeviceListProto + func (x *HloShardingProto) GetMemoryKind() string + func (x *HloShardingProto) GetXlaOpSharding() *data.OpSharding + func (x *HloShardingProto) ProtoReflect() protoreflect.Message + func (x *HloShardingProto) Reset() + func (x *HloShardingProto) String() string + type XlaCompileOptionsProto struct + CompileOptions *pjrt.CompileOptionsProto + func (*XlaCompileOptionsProto) Descriptor() ([]byte, []int) + func (*XlaCompileOptionsProto) ProtoMessage() + func (x *XlaCompileOptionsProto) GetCompileOptions() *pjrt.CompileOptionsProto + func (x *XlaCompileOptionsProto) ProtoReflect() protoreflect.Message + func (x *XlaCompileOptionsProto) Reset() + func (x *XlaCompileOptionsProto) String() string + type XlaHostCallbackProto struct + Operands []*XlaHostCallbackProto_ArgInfo + Results []*XlaHostCallbackProto_ArgInfo + SerializedCallback *anypb.Any + UseMajorToMinorDataLayoutForCallbacks bool + func (*XlaHostCallbackProto) Descriptor() ([]byte, []int) + func (*XlaHostCallbackProto) ProtoMessage() + func (x *XlaHostCallbackProto) GetOperands() []*XlaHostCallbackProto_ArgInfo + func (x *XlaHostCallbackProto) GetResults() []*XlaHostCallbackProto_ArgInfo + func (x *XlaHostCallbackProto) GetSerializedCallback() *anypb.Any + func (x *XlaHostCallbackProto) GetUseMajorToMinorDataLayoutForCallbacks() bool + func (x *XlaHostCallbackProto) ProtoReflect() protoreflect.Message + func (x *XlaHostCallbackProto) Reset() + func (x *XlaHostCallbackProto) String() string + type XlaHostCallbackProto_ArgInfo struct + ChannelId uint32 + Shape *data.ShapeProto + func (*XlaHostCallbackProto_ArgInfo) Descriptor() ([]byte, []int) + func (*XlaHostCallbackProto_ArgInfo) ProtoMessage() + func (x *XlaHostCallbackProto_ArgInfo) GetChannelId() uint32 + func (x *XlaHostCallbackProto_ArgInfo) GetShape() *data.ShapeProto + func (x *XlaHostCallbackProto_ArgInfo) ProtoReflect() protoreflect.Message + func (x *XlaHostCallbackProto_ArgInfo) Reset() + func (x *XlaHostCallbackProto_ArgInfo) String() string
Go to main page