Struct tensorflow_proto::xrt::XlaComputationConfig

pub struct XlaComputationConfig {
    pub num_replicas: i32,
    pub num_cores_per_replica: i32,
    pub host_compute_metadata: Option<HostComputeMetadata>,
    pub program_shape: Option<ProgramShapeProto>,
    pub per_core_program_shape: Vec<ProgramShapeProto>,
    pub device_assignment: Option<DeviceAssignment>,
    pub debug_options: Option<DebugOptions>,
}

Options for an XLA compilation.

Fields

num_replicas: i32

The number of replicas the computation will be run on. If this is default (0) it is interpreted as 1.

num_cores_per_replica: i32

The number of "model-parallel" cores per replica. If this is default (0) it is interpreted as 1.

host_compute_metadata: Option<HostComputeMetadata>

Optional metadata about host sends and recvs.

program_shape: Option<ProgramShapeProto>

The arg/result shapes for the whole computation.

per_core_program_shape: Vec<ProgramShapeProto>

The arg/result shapes for each core of a model-parallel computation. per_core_args_and_result_shapes is optional for a single-core computation.

device_assignment: Option<DeviceAssignment>

Describes how replicated computation instances should be assigned to devices. There are num_cores_per_replica computations, and each one will be sent and executed to the set of replica device numbers described in the DeviceAssignment proto.

debug_options: Option<DebugOptions>

The debugging options to be passed to the XLA compilation process.

Trait Implementations

impl Clone for XlaComputationConfig

pub fn clone(&self) -> XlaComputationConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for XlaComputationConfig

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

Formats the value using the given formatter.

impl Default for XlaComputationConfig

pub fn default() -> Self

Returns the "default value" for a type.

impl Message for XlaComputationConfig

pub fn encode_raw<B>(&self, buf: &mut B) where
    B: BufMut, 

pub fn merge_field<B>(
    &mut self,
    tag: u32,
    wire_type: WireType,
    buf: &mut B,
    ctx: DecodeContext
) -> Result<(), DecodeError> where
    B: Buf, 

pub fn encoded_len(&self) -> usize

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

pub fn clear(&mut self)

Clears the message, resetting all fields to their default.

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

Encodes the message to a buffer.

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

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

pub fn decode<B>(buf: B) -> Result<Self, DecodeError> where
    Self: Default,
    B: Buf, 

Decodes an instance of the message from a buffer.

pub fn decode_length_delimited<B>(buf: B) -> Result<Self, DecodeError> where
    Self: Default,
    B: Buf, 

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

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

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

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

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

impl PartialEq<XlaComputationConfig> for XlaComputationConfig

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

This method tests for self and other values to be equal, and is used by ==.

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

This method tests for !=.

impl StructuralPartialEq for XlaComputationConfig