Searched defs:mode_key (Results 1 – 3 of 3) sorted by relevance
125 TorchDispatchModeTLS::get_mode(TorchDispatchModeKey mode_key) { in get_mode()131 TorchDispatchModeKey mode_key) { in set_mode()149 TorchDispatchModeTLS::unset_mode(TorchDispatchModeKey mode_key) { in unset_mode()185 std::string to_string(TorchDispatchModeKey mode_key) { in to_string()
1138 std::optional<c10::impl::TorchDispatchModeKey> mode_key = std::nullopt; in push_on_torch_dispatch_stack() local1162 std::optional<c10::impl::TorchDispatchModeKey> mode_key = std::nullopt; in pop_torch_dispatch_stack() local1210 auto mode_key = py::cast<c10::impl::TorchDispatchModeKey>(maybe_mode_key_obj); in set_dispatch_mode() local1225 auto mode_key = py::cast<c10::impl::TorchDispatchModeKey>(arg); in get_dispatch_mode() local1241 auto mode_key = py::cast<c10::impl::TorchDispatchModeKey>(arg); in unset_dispatch_mode() local
503 def unset_mode_pre_dispatch(mode_key, schema_check=False): argument597 def _get_dispatch_mode_pre_dispatch(mode_key): argument