Searched refs:AutogradNestedTensor (Results 1 – 9 of 9) sorted by relevance
61 {DispatchKey::AutogradNestedTensor, DispatchKey::NestedTensor}) |140 case DispatchKey::AutogradNestedTensor: in getBackendKeySetFromAutograd()
132 case DispatchKey::AutogradNestedTensor: in toString()293 {"AutogradNestedTensor", c10::DispatchKey::AutogradNestedTensor}, in parseDispatchKey()
653 DispatchKey::AutogradNestedTensor,771 DispatchKeySet(DispatchKey::AutogradNestedTensor);
348 AutogradNestedTensor, enumerator
197 c10::DispatchKey::AutogradNestedTensor}; in generate_buffer_key_set()
82 ks = ks.add(DispatchKey.AutogradNestedTensor)
709 DEF_ONE(AutogradNestedTensor) in initDispatchBindings()
110 AutogradNestedTensor = auto() variable in DispatchKey
1324 torch._C._dispatch_keys(x).has(DispatchKey.AutogradNestedTensor)