/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/fx/tracer/meta_patch/patched_function/normalization.py
15 строк
517 B
Hongxin Liu
[misc] update pre-commit and run all files (#4752)
19 сен 2023, 09:20
Не верифицирован
19 сен 2023, 09:20
079bf3c
Код
Авторство
О чём код?
import torch from ...registry import meta_patched_function @meta_patched_function.register(torch.nn.functional.layer_norm) def torch_nn_func_layernorm(input, normalized_shape, weight=None, bias=None, eps=1e-05): return torch.empty(input.shape, device="meta") @meta_patched_function.register(torch.nn.functional.batch_norm) def torch_nn_func_batchnorm( input, running_mean, running_var, weight=None, bias=None, training=False, momentum=0.1, eps=1e-05 ): return torch.empty(input.shape, device="meta")