Home
last modified time | relevance | path

Searched refs:buffer_key_set (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DNestedTensorImpl.h191 auto buffer_key_set = this->key_set(); in generate_buffer_key_set() local
192 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()