Struct tensorflow_proto::tensorflow::tpu::TopologyProto

pub struct TopologyProto {
    pub mesh_shape: Vec<i32>,
    pub num_tasks: i32,
    pub num_tpu_devices_per_task: i32,
    pub device_coordinates: Vec<i32>,
}

Describes the geometry of a TPU mesh.

Fields

mesh_shape: Vec<i32>

The dimensions of the TPU topology, in cores. Typically, this is a 3D topology [x, y, core], where the major dimensions correspond to TPU chips, and the minor dimension describes the number of cores on a multicore chip.

num_tasks: i32

Number of TensorFlow tasks in the cluster.

num_tpu_devices_per_task: i32

Number of TPU devices per task.

device_coordinates: Vec<i32>

A flattened rank 3 int32 array with shape [num_tasks, num_tpu_devices_per_task, len(mesh_shape)]. tasks is the number of tasks in the TPU cluster, devices is the number of TPU devices per task, and the minor dimension corresponds to a position in the TPU mesh topology. Each entry [task, device, axis] gives the axis-th coordinate in the topology of a task/device pair.

Trait Implementations

impl Clone for TopologyProto

pub fn clone(&self) -> TopologyProto

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for TopologyProto

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

Formats the value using the given formatter.

impl Default for TopologyProto

pub fn default() -> Self

Returns the "default value" for a type.

impl Message for TopologyProto

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<TopologyProto> for TopologyProto

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

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

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

This method tests for !=.

impl StructuralPartialEq for TopologyProto