Searched refs:nested_key_set (Results 1 – 1 of 1) sorted by relevance
42 auto nested_key_set = buffer.key_set(); in generate_nested_key_set_from_buffer() local43 const bool has_autograd = nested_key_set.has_any(c10::autograd_dispatch_keyset); in generate_nested_key_set_from_buffer()45 nested_key_set = nested_key_set - in generate_nested_key_set_from_buffer()49 nested_key_set = in generate_nested_key_set_from_buffer()50 nested_key_set | c10::DispatchKeySet{c10::DispatchKey::NestedTensor}; in generate_nested_key_set_from_buffer()51 nested_key_set = in generate_nested_key_set_from_buffer()52 has_autograd ? nested_key_set | c10::autograd_nested : nested_key_set; in generate_nested_key_set_from_buffer()53 return nested_key_set; in generate_nested_key_set_from_buffer()