Struct tensorflow_proto::xla::ShapeProto

pub struct ShapeProto {
    pub element_type: i32,
    pub dimensions: Vec<i64>,
    pub tuple_shapes: Vec<ShapeProto>,
    pub layout: Option<LayoutProto>,
    pub is_dynamic_dimension: Vec<bool>,
}

A shape describes the number of dimensions in the array, the size of each dimension, and the primitive component type.

Tuples are a special case in that they have rank zero and have tuple_shapes defined.

See the XLA documentation for more information on shapes and layouts.

LINT.IfChange

Fields

element_type: i32

The element type for this shape.

dimensions: Vec<i64>

The size (number of elements) for each dimension, or an upper bound on the size if the dimension is dynamic. In XLA, dimensions are numbered from 0 to N-1 for an N-dimensional array. The first element of 'dimensions' is the size of dimension 0, the second element is the size of dimension 1, and so forth. Empty list indicates a scalar.

If the respective element in 'is_dimension_dynamic' is true then the value in this field represents an upper bound on the size of the dimension.

tuple_shapes: Vec<ShapeProto>

For tuples only, the shapes of constituent shapes in the tuple sequence.

layout: Option<LayoutProto>

The layout used to back this shape.

is_dynamic_dimension: Vec<bool>

For arrays, this indicates whether or not each dimension is dynamically-sized. The number of elements in this repeated field should be zero (indicating that no dimensions are dynamic) or equal to the number of elements in the 'dimensions' field.

Implementations

impl ShapeProto

pub fn element_type(&self) -> PrimitiveType

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

pub fn set_element_type(&mut self, value: PrimitiveType)

Sets element_type to the provided enum value.

Trait Implementations

impl Clone for ShapeProto

fn clone(&self) -> ShapeProto

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ShapeProto

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

Formats the value using the given formatter.

impl Default for ShapeProto

fn default() -> ShapeProto

Returns the "default value" for a type.

impl Message for ShapeProto

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

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

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<B>(&self, buf: &mut B) -> Result<(), EncodeError> where
    B: BufMut, 

Encodes the message to a buffer.

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

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

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

Decodes an instance of the message from a buffer.

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

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

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.

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

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

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

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

This method tests for !=.

impl StructuralPartialEq for ShapeProto