Searched defs:dirty_inputs (Results 1 – 2 of 2) sorted by relevance
34 const std::unordered_set<at::TensorImpl*>& dirty_inputs, in _process_forward_mode_AD()259 const std::unordered_set<at::TensorImpl*>& dirty_inputs, in _process_backward_mode_ad()441 const std::unordered_set<at::TensorImpl*>& dirty_inputs, in _wrap_outputs()
558 std::unordered_set<at::TensorImpl*> dirty_inputs; in _mark_dirty() local619 auto dirty_inputs = _mark_dirty(self); in _wrap_outputs() local