Struct tensorflow::eager::raw_ops::ApplyAdam
source · pub struct ApplyAdam { /* private fields */ }
Expand description
ApplyAdam
See : https://www.tensorflow.org/api_docs/python/tf/raw_ops/ApplyAdam
Implementations§
source§impl ApplyAdam
impl ApplyAdam
sourcepub fn use_locking<ArgType: Into<bool>>(self, value: ArgType) -> Self
pub fn use_locking<ArgType: Into<bool>>(self, value: ArgType) -> Self
Sets the use_locking
attribute.
sourcepub fn use_nesterov<ArgType: Into<bool>>(self, value: ArgType) -> Self
pub fn use_nesterov<ArgType: Into<bool>>(self, value: ArgType) -> Self
Sets the use_nesterov
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>, T1: ToTensorHandle<'a>, T2: ToTensorHandle<'a>, T3: ToTensorHandle<'a>, T4: ToTensorHandle<'a>, T5: ToTensorHandle<'a>, T6: ToTensorHandle<'a>, T7: ToTensorHandle<'a>, T8: ToTensorHandle<'a>, T9: ToTensorHandle<'a>>(
&self,
ctx: &'a Context,
var: &T0,
m: &T1,
v: &T2,
beta1_power: &T3,
beta2_power: &T4,
lr: &T5,
beta1: &T6,
beta2: &T7,
epsilon: &T8,
grad: &T9
) -> Result<TensorHandle<'a>>
pub fn call<'a, T0: ToTensorHandle<'a>, T1: ToTensorHandle<'a>, T2: ToTensorHandle<'a>, T3: ToTensorHandle<'a>, T4: ToTensorHandle<'a>, T5: ToTensorHandle<'a>, T6: ToTensorHandle<'a>, T7: ToTensorHandle<'a>, T8: ToTensorHandle<'a>, T9: ToTensorHandle<'a>>( &self, ctx: &'a Context, var: &T0, m: &T1, v: &T2, beta1_power: &T3, beta2_power: &T4, lr: &T5, beta1: &T6, beta2: &T7, epsilon: &T8, grad: &T9 ) -> Result<TensorHandle<'a>>
Execute apply_adam.
Trait Implementations§
Auto Trait Implementations§
impl RefUnwindSafe for ApplyAdam
impl Send for ApplyAdam
impl Sync for ApplyAdam
impl Unpin for ApplyAdam
impl UnwindSafe for ApplyAdam
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