/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
with_cpu
src/autograd/grad_guard.cpp
15 строк
280 B
kolkir
Add autograd disable and enable functionality
28 янв 2025, 00:19
28 янв 2025, 00:19
92c91bd
Код
Авторство
О чём код?
#include <adept/autograd/grad_guard.hpp> namespace adept { namespace { thread_local bool autograd_enabled = true; } // namespace bool autograd_is_enabled() { return autograd_enabled; } void autograd_set_enabled(bool mode) { autograd_enabled = mode; } } // namespace adept