Struct pjrt_sys::protos::xla::HloModuleProto

pub struct HloModuleProto {

    pub name: String,
    pub entry_computation_name: String,
    pub entry_computation_id: i64,
    pub computations: Vec<HloComputationProto>,
    pub host_program_shape: Option<ProgramShapeProto>,
    pub id: i64,
    pub schedule: Option<HloScheduleProto>,
    pub input_output_alias: Option<HloInputOutputAliasProto>,
    pub buffer_donor: Option<HloBufferDonorProto>,
    pub cross_program_prefetches: Vec<CrossProgramPrefetch>,
    pub is_dynamic: bool,
    pub spmd_output_sharding: Option<OpSharding>,
    pub spmd_parameters_shardings: Vec<OpSharding>,
    pub use_auto_spmd_partitioning: bool,
    pub profile_info: Vec<ProfileInfo>,
    pub device_assignment: Option<DeviceAssignmentProto>,
    pub stack_frame_index: Option<StackFrameIndexProto>,
    pub frontend_attributes: Option<FrontendAttributes>,
}

Serialization of HloModule.

Fields

name: String

entry_computation_name: String

entry_computation_id: i64

computations: Vec<HloComputationProto>

The array of computations is always in a valid dependency order, where callees appear before their callers.

host_program_shape: Option<ProgramShapeProto>

The host program shape (with layout) of the entry computation.

id: i64

The id of this module.

schedule: Option<HloScheduleProto>

The schedule for this module.

input_output_alias: Option<HloInputOutputAliasProto>

Describes alias information between inputs and outputs.

buffer_donor: Option<HloBufferDonorProto>

Describes the information of input buffer donors.

cross_program_prefetches: Vec<CrossProgramPrefetch>

is_dynamic: bool

True if the module contains dynamic computation.

spmd_output_sharding: Option<OpSharding>

spmd_parameters_shardings: Vec<OpSharding>

use_auto_spmd_partitioning: bool

Uses AutoSharding pass or not.

profile_info: Vec<ProfileInfo>

Profile information for the HLO module.

device_assignment: Option<DeviceAssignmentProto>

DeviceAssignment object information.

stack_frame_index: Option<StackFrameIndexProto>

Stack frames index.

frontend_attributes: Option<FrontendAttributes>

Frontend attributes to pass to the XLA backend.

Trait Implementations

impl Clone for HloModuleProto

fn clone(&self) -> HloModuleProto

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for HloModuleProto

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for HloModuleProto

fn default() -> Self

Returns the “default value” for a type.

impl Message for HloModuleProto

fn encoded_len(&self) -> usize

Returns the encoded length of the message without a length delimiter.

fn clear(&mut self)

Clears the message, resetting all fields to their default.

fn encode(&self, buf: &mut impl BufMut) -> Result<(), EncodeError>

where Self: Sized,

Encodes the message to a buffer.

fn encode_to_vec(&self) -> Vec<u8>

where Self: Sized,

Encodes the message to a newly allocated buffer.

fn encode_length_delimited( &self, buf: &mut impl BufMut, ) -> Result<(), EncodeError>

where Self: Sized,

Encodes the message with a length-delimiter to a buffer.

fn encode_length_delimited_to_vec(&self) -> Vec<u8>

where Self: Sized,

Encodes the message with a length-delimiter to a newly allocated buffer.

fn decode(buf: impl Buf) -> Result<Self, DecodeError>

where Self: Default,

Decodes an instance of the message from a buffer.

fn decode_length_delimited(buf: impl Buf) -> Result<Self, DecodeError>

where Self: Default,

Decodes a length-delimited instance of the message from the buffer.

fn merge(&mut self, buf: impl Buf) -> Result<(), DecodeError>

where Self: Sized,

Decodes an instance of the message from a buffer, and merges it into self.

fn merge_length_delimited(&mut self, buf: impl Buf) -> Result<(), DecodeError>

where Self: Sized,

Decodes a length-delimited instance of the message from buffer, and merges it into self.

impl PartialEq for HloModuleProto

fn eq(&self, other: &HloModuleProto) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for HloModuleProto