pub struct EagerInput<Op: GraphOperation> {
pub key: ValueKey<Op>,
pub trace: Option<Trace<Op>>,
pub requires_grad: bool,
pub data: Arc<Op::Operand>,
}Expand description
Input descriptor for recording one eager graph invocation.
Fields§
§key: ValueKey<Op>User-visible eager value key used for cotangent accumulation.
trace: Option<Trace<Op>>Trace node that produced this value, or None for leaves.
requires_grad: boolWhether this value participates in reverse-mode propagation.
data: Arc<Op::Operand>Concrete primal data for saved forward replay.
Auto Trait Implementations§
impl<Op> Freeze for EagerInput<Op>where
<Op as GraphOperation>::InputKey: Freeze,
impl<Op> RefUnwindSafe for EagerInput<Op>where
<Op as GraphOperation>::InputKey: RefUnwindSafe,
<Op as GraphOperation>::Operand: RefUnwindSafe,
Op: RefUnwindSafe,
impl<Op> Send for EagerInput<Op>
impl<Op> Sync for EagerInput<Op>
impl<Op> Unpin for EagerInput<Op>where
<Op as GraphOperation>::InputKey: Unpin,
impl<Op> UnsafeUnpin for EagerInput<Op>where
<Op as GraphOperation>::InputKey: UnsafeUnpin,
impl<Op> UnwindSafe for EagerInput<Op>where
<Op as GraphOperation>::InputKey: UnwindSafe + RefUnwindSafe,
<Op as GraphOperation>::Operand: RefUnwindSafe,
Op: RefUnwindSafe,
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