Searched defs:to_save_if_setup_context (Results 1 – 2 of 2) sorted by relevance
262 const std::unordered_set<at::TensorImpl*>& to_save_if_setup_context, in _process_backward_mode_ad()445 const std::unordered_set<at::TensorImpl*>& to_save_if_setup_context, in _wrap_outputs()
610 const std::unordered_set<at::TensorImpl*>& to_save_if_setup_context) { in _wrap_outputs()747 std::unordered_set<at::TensorImpl*>& to_save_if_setup_context, in _get_tensors_to_save()1107 std::unordered_set<at::TensorImpl*> to_save_if_setup_context{}; in process_outputs() local