Struct HloModuleConfigProto 

pub struct HloModuleConfigProto {

    pub entry_computation_layout: Option<ProgramShapeProto>,
    pub seed: u64,
    pub launch_id: i32,
    pub replica_count: i64,
    pub num_partitions: i64,
    pub param_requires_broadcast_via_collectives: Vec<bool>,
    pub use_spmd_partitioning: bool,
    pub use_auto_spmd_partitioning: bool,
    pub auto_spmd_partitioning_mesh_shape: Vec<i64>,
    pub auto_spmd_partitioning_mesh_ids: Vec<i64>,
    pub deduplicate_hlo: bool,
    pub intra_op_parallelism_threads: i64,
    pub device_type: String,
    pub debug_options: Option<DebugOptions>,
    pub static_device_assignment: Option<DeviceAssignmentProto>,
    pub allow_separate_sharding_programs: bool,
    pub shardable_value_update_pairs: Vec<ShardableValueUpdatePairProto>,
    pub alias_passthrough_params: bool,
    pub content_aware_computation_sorting: bool,
    pub fusion_config_collection: i32,
    pub fusion_config: Vec<BoolList>,
    pub dot_config: HashMap<String, Int64List>,
    pub layout_config: Vec<Int64ListList>,
    pub memory_space_assignment_config: Vec<u64>,
    pub phase_ordering_config: Vec<BoolList>,
    pub phase_index: i32,
    pub allow_spmd_sharding_propagation_to_parameters: Vec<bool>,
    pub allow_spmd_sharding_propagation_to_output: Vec<bool>,
    pub analysis_allowance_map: HashMap<String, i64>,
    pub matrix_unit_operand_precision: i32,
    pub fdo_profile: Vec<u8>,
    pub device_memory_size: i64,
    pub use_shardy_partitioner: bool,
}

Serialization of HloModuleConfig. See the C++ class definition for descriptions of each field. There are no guarantees of backwards or forwards compatibility. Next id: 35.

Fields

entry_computation_layout: Option<ProgramShapeProto>

seed: u64

launch_id: i32

replica_count: i64

num_partitions: i64

param_requires_broadcast_via_collectives: Vec<bool>

use_spmd_partitioning: bool

use_auto_spmd_partitioning: bool

auto_spmd_partitioning_mesh_shape: Vec<i64>

auto_spmd_partitioning_mesh_ids: Vec<i64>

deduplicate_hlo: bool

intra_op_parallelism_threads: i64

device_type: String

debug_options: Option<DebugOptions>

static_device_assignment: Option<DeviceAssignmentProto>

allow_separate_sharding_programs: bool

shardable_value_update_pairs: Vec<ShardableValueUpdatePairProto>

alias_passthrough_params: bool

content_aware_computation_sorting: bool

fusion_config_collection: i32

fusion_config: Vec<BoolList>

dot_config: HashMap<String, Int64List>

layout_config: Vec<Int64ListList>

memory_space_assignment_config: Vec<u64>

phase_ordering_config: Vec<BoolList>

phase_index: i32

allow_spmd_sharding_propagation_to_parameters: Vec<bool>

allow_spmd_sharding_propagation_to_output: Vec<bool>

analysis_allowance_map: HashMap<String, i64>

matrix_unit_operand_precision: i32

fdo_profile: Vec<u8>

device_memory_size: i64

use_shardy_partitioner: bool

Implementations

impl HloModuleConfigProto

pub fn fusion_config_collection(&self) -> FusionConfigCollection

Returns the enum value of fusion_config_collection, or the default if the field is set to an invalid enum value.

pub fn set_fusion_config_collection(&mut self, value: FusionConfigCollection)

Sets fusion_config_collection to the provided enum value.

pub fn matrix_unit_operand_precision(&self) -> Precision

Returns the enum value of matrix_unit_operand_precision, or the default if the field is set to an invalid enum value.

pub fn set_matrix_unit_operand_precision(&mut self, value: Precision)

Sets matrix_unit_operand_precision to the provided enum value.

Trait Implementations

impl Clone for HloModuleConfigProto

fn clone(&self) -> HloModuleConfigProto

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for HloModuleConfigProto

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

Formats the value using the given formatter.

impl Default for HloModuleConfigProto

fn default() -> Self

Returns the “default value” for a type.

impl Message for HloModuleConfigProto

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 HloModuleConfigProto

fn eq(&self, other: &HloModuleConfigProto) -> 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 HloModuleConfigProto