/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
with_cpu
src/dispatch/tensor_init.hpp
28 строк
1 KB
kolkir
Batchnorm forward tests
22 фев 2025, 18:31
22 фев 2025, 18:31
bca02bd
Код
Авторство
О чём код?
#pragma once #include <adept/dispatch/dispatcher.hpp> #include <adept/tensor.hpp> namespace adept { inline void fill_zero_tensor(Tensor& tensor) { static auto& func = Dispatcher::instance().find("fill_zero_tensor"); Dispatcher::instance().call<void, Tensor&>(func, tensor); } inline void fill_uniform_tensor(float32_t start, float32_t end, Tensor& tensor) { static auto& func = Dispatcher::instance().find("fill_uniform_tensor"); Dispatcher::instance().call<void, float32_t, float32_t, Tensor&>(func, start, end, tensor); } inline void fill_normal_tensor(float32_t mean, float32_t std, Tensor& tensor) { static auto& func = Dispatcher::instance().find("fill_normal_tensor"); Dispatcher::instance().call<void, float32_t, float32_t, Tensor&>(func, mean, std, tensor); } inline void fill_tensor(Tensor& tensor, const std::any& scalar) { static auto& func = Dispatcher::instance().find("fill_tensor"); Dispatcher::instance().call<void, Tensor&, const std::any&>(func, tensor, scalar); } } // namespace adept