Struct tensorflow::eager::raw_ops::StagePeek
source · pub struct StagePeek { /* private fields */ }
Expand description
StagePeek
See : https://www.tensorflow.org/api_docs/python/tf/raw_ops/StagePeek
Implementations§
source§impl StagePeek
impl StagePeek
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, T0: ToTensorHandle<'a>>(
&self,
ctx: &'a Context,
index: &T0
) -> Result<TensorHandle<'a>>
pub fn call<'a, T0: ToTensorHandle<'a>>( &self, ctx: &'a Context, index: &T0 ) -> Result<TensorHandle<'a>>
Execute stage_peek.
Trait Implementations§
Auto Trait Implementations§
impl RefUnwindSafe for StagePeek
impl Send for StagePeek
impl Sync for StagePeek
impl Unpin for StagePeek
impl UnwindSafe for StagePeek
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