Searched refs:buffer_key_set (Results 1 – 1 of 1) sorted by relevance
191 auto buffer_key_set = this->key_set(); in generate_buffer_key_set() local192 const bool Autograd = buffer_key_set.has_any(c10::autograd_dispatch_keyset); in generate_buffer_key_set()194 buffer_key_set = buffer_key_set - in generate_buffer_key_set()200 buffer_key_set = in generate_buffer_key_set()201 buffer_key_set | c10::DispatchKeySet{c10::DispatchKey::Dense}; in generate_buffer_key_set()202 buffer_key_set = Autograd in generate_buffer_key_set()203 ? c10::DispatchKeySet{c10::DispatchKey::Autograd} | buffer_key_set in generate_buffer_key_set()204 : buffer_key_set; in generate_buffer_key_set()206 return buffer_key_set; in generate_buffer_key_set()