Struct tensorflow_proto::xla::WhileLoopBackendConfig

pub struct WhileLoopBackendConfig {
    pub known_trip_count: Option<KnownTripCount>,
}

A backend-config for kWhile loops that stores the loop's trip count, if it is known.

This is useful for backends that can implement a for i in 0..N loop more efficiently than a while loop. For example, on GPUs, we can implement a for i in 0..N loop by enqueueing the kernels for the loop body N times, whereas implementing a while loop requires a host-device sync on each iteration.

Fields

known_trip_count: Option<KnownTripCount>

This indirection lets us distinguish between known-trip-count == 0 and unknown-trip-count.

Trait Implementations

impl Clone for WhileLoopBackendConfig

fn clone(&self) -> WhileLoopBackendConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for WhileLoopBackendConfig

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

Formats the value using the given formatter.

impl Default for WhileLoopBackendConfig

fn default() -> WhileLoopBackendConfig

Returns the "default value" for a type.

impl Message for WhileLoopBackendConfig

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

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

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

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

This method tests for !=.

impl StructuralPartialEq for WhileLoopBackendConfig