Struct tensorflow_proto::xrt::XrtExecutionConfig

pub struct XrtExecutionConfig {
    pub device_ordinal: i32,
    pub core_index_in_replica: i32,
    pub execution_instance_key: String,
    pub rng_seed: u32,
    pub release_input_handles: bool,
    pub release_compilation_handle: bool,
    pub return_exploded_tuple: bool,
}

Options for an XLA execution.

Fields

device_ordinal: i32

Local device to run on. This is present because the execute Op may be placed on a device such as CPU or TPU_SYSTEM that logically manages multiple cores.

core_index_in_replica: i32

Which model-parallel computation to run from the compiled bundle.

execution_instance_key: String

Optional key to disambiguate between executions. This is only needed if multiple host send/recvs may be outstanding concurrently with executions.

rng_seed: u32

If non-zero, rng_seed to reset the core with.

release_input_handles: bool

If true, release allocation handles on the inputs after running.

release_compilation_handle: bool

If true, release the handle to the computation after running.

return_exploded_tuple: bool

If set to true, and the result shape is a tuple, then instead of returning a single tuple allocation the execution will return a vector of allocations, one for each of the first-level elements of the result tuple.

Trait Implementations

impl Clone for XrtExecutionConfig

pub fn clone(&self) -> XrtExecutionConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for XrtExecutionConfig

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

Formats the value using the given formatter.

impl Default for XrtExecutionConfig

pub fn default() -> Self

Returns the "default value" for a type.

impl Message for XrtExecutionConfig

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

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

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

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

This method tests for !=.

impl StructuralPartialEq for XrtExecutionConfig