Searched refs:pythonTorchFunctionState (Results 1 – 1 of 1) sorted by relevance
6 static thread_local PythonTorchFunctionTLS pythonTorchFunctionState; variable9 pythonTorchFunctionState.stack_.push_back(std::move(mode)); in push_onto_stack()13 TORCH_CHECK(!pythonTorchFunctionState.stack_.empty(), "trying to pop from empty mode stack"); in pop_stack()14 auto out = pythonTorchFunctionState.stack_.back(); in pop_stack()15 pythonTorchFunctionState.stack_.pop_back(); in pop_stack()20 …TORCH_CHECK(idx < static_cast<int64_t>(pythonTorchFunctionState.stack_.size()), "Tried to get stac… in get_stack_at()21 return pythonTorchFunctionState.stack_[idx]; in get_stack_at()25 return static_cast<int64_t>(pythonTorchFunctionState.stack_.size()); in stack_len()29 pythonTorchFunctionState.disabled_state_ = disabled_state; in set_disabled_state()33 return pythonTorchFunctionState.disabled_state_; in get_disabled_state()[all …]