Struct tensorflow::eager::raw_ops::StageSize
source · pub struct StageSize { /* private fields */ }
Expand description
StageSize
See : https://www.tensorflow.org/api_docs/python/tf/raw_ops/StageSize
Implementations§
source§impl StageSize
impl StageSize
sourcepub fn memory_limit<ArgType: Into<i64>>(self, value: ArgType) -> Self
pub fn memory_limit<ArgType: Into<i64>>(self, value: ArgType) -> Self
Sets the memory_limit
attribute.
sourcepub fn dtypes<ArgType: Into<Vec<DataType>>>(self, value: ArgType) -> Self
pub fn dtypes<ArgType: Into<Vec<DataType>>>(self, value: ArgType) -> Self
Sets the dtypes
attribute.
sourcepub fn container<ArgType: Into<String>>(self, value: ArgType) -> Self
pub fn container<ArgType: Into<String>>(self, value: ArgType) -> Self
Sets the container
attribute.
Sets the shared_name
attribute.
sourcepub fn target_device_name<ArgType: Into<String>>(self, value: ArgType) -> Self
pub fn target_device_name<ArgType: Into<String>>(self, value: ArgType) -> Self
Sets the `` attribute.
sourcepub fn call<'a>(&self, ctx: &'a Context) -> Result<TensorHandle<'a>>
pub fn call<'a>(&self, ctx: &'a Context) -> Result<TensorHandle<'a>>
Execute stage_size.
Trait Implementations§
Auto Trait Implementations§
impl RefUnwindSafe for StageSize
impl Send for StageSize
impl Sync for StageSize
impl Unpin for StageSize
impl UnwindSafe for StageSize
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more