Searched refs:PyObject_TorchDispatchMode (Results 1 – 4 of 4) sorted by relevance
15 using PyObject_TorchDispatchMode = SafePyObjectT<TorchDispatchModeKey>; variable23 std::shared_ptr<PyObject_TorchDispatchMode> mode);27 static const std::shared_ptr<PyObject_TorchDispatchMode> pop_stack();31 tuple<std::shared_ptr<PyObject_TorchDispatchMode>, TorchDispatchModeKey>34 static const std::shared_ptr<PyObject_TorchDispatchMode>& get_stack_at(38 static const std::optional<std::shared_ptr<PyObject_TorchDispatchMode>>40 static const std::optional<std::shared_ptr<PyObject_TorchDispatchMode>>43 const std::shared_ptr<PyObject_TorchDispatchMode>& mode,52 std::vector<std::shared_ptr<PyObject_TorchDispatchMode>> stack_;58 std::optional<std::shared_ptr<PyObject_TorchDispatchMode>>,
28 std::shared_ptr<PyObject_TorchDispatchMode> mode) { in push_non_infra_mode_onto_stack()37 const std::shared_ptr<PyObject_TorchDispatchMode> TorchDispatchModeTLS::39 std::shared_ptr<PyObject_TorchDispatchMode> out; in pop_stack()65 tuple<std::shared_ptr<PyObject_TorchDispatchMode>, TorchDispatchModeKey>87 const std::shared_ptr<PyObject_TorchDispatchMode>& TorchDispatchModeTLS::124 const std::optional<std::shared_ptr<PyObject_TorchDispatchMode>>130 const std::shared_ptr<PyObject_TorchDispatchMode>& mode, in set_mode()148 const std::optional<std::shared_ptr<PyObject_TorchDispatchMode>>
31 const std::shared_ptr<c10::impl::PyObject_TorchDispatchMode>& get_cur_mode() { in get_cur_mode()36 std::shared_ptr<c10::impl::PyObject_TorchDispatchMode> saved_mode_;
1144 std::make_shared<c10::impl::PyObject_TorchDispatchMode>( in push_on_torch_dispatch_stack()1149 std::make_shared<c10::impl::PyObject_TorchDispatchMode>( in push_on_torch_dispatch_stack()1214 std::make_shared<c10::impl::PyObject_TorchDispatchMode>( in set_dispatch_mode()