Struct tensorflow::eager::raw_ops::ConfigureDistributedTPU

pub struct ConfigureDistributedTPU { /* private fields */ }

ConfigureDistributedTPU

See : https://www.tensorflow.org/api_docs/python/tf/raw_ops/ConfigureDistributedTPU

Implementations

impl ConfigureDistributedTPU

pub fn new() -> Self

Creates a new ConfigureDistributedTPU.

pub fn embedding_config<ArgType: Into<String>>(self, value: ArgType) -> Self

Sets the embedding_config attribute.

pub fn tpu_embedding_config<ArgType: Into<String>>(self, value: ArgType) -> Self

Sets the tpu_embedding_config attribute.

pub fn is_global_init<ArgType: Into<bool>>(self, value: ArgType) -> Self

Sets the is_global_init attribute.

pub fn enable_whole_mesh_compilations<ArgType: Into<bool>>( self, value: ArgType ) -> Self

Sets the enable_whole_mesh_compilations attribute.

pub fn compilation_failure_closes_chips<ArgType: Into<bool>>( self, value: ArgType ) -> Self

Sets the compilation_failure_closes_chips attribute.

pub fn tpu_cancellation_closes_chips<ArgType: Into<i64>>( self, value: ArgType ) -> Self

Sets the tpu_cancellation_closes_chips attribute.

pub fn target_device_name<ArgType: Into<String>>(self, value: ArgType) -> Self

Sets the `` attribute.

pub fn call<'a>(&self, ctx: &'a Context) -> Result<TensorHandle<'a>>

Execute configure_distributed_tpu.

Trait Implementations

impl Clone for ConfigureDistributedTPU

fn clone(&self) -> ConfigureDistributedTPU

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ConfigureDistributedTPU

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

Formats the value using the given formatter.

impl Default for ConfigureDistributedTPU

fn default() -> Self

Returns the “default value” for a type.