Searched defs:save_var (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cu | 306 void batch_norm_mean_var(const Tensor& self, Tensor& save_mean, Tensor& save_var) { in batch_norm_mean_var()
347 const Tensor& save_mean, const Tensor& save_var, in batch_norm_update_stats()
381 const Tensor& save_mean, const Tensor& save_var, in batch_norm_update_stats_and_invert()
491 Tensor output, save_mean, save_var, reserve; in _batch_norm_with_update_cuda() local
512 Tensor& out, Tensor& save_mean, Tensor& save_var, Tensor& reserve) { in _batch_norm_with_update_cuda_out()
557 const Tensor& save_var = c10::value_or_else(save_var_opt, [] {return Tensor();}); in _new_batch_norm_backward_cuda() local
808 auto save_var = at::empty({n_input}, options); in batch_norm_update_stats_cuda() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 762 … bool train, double momentum, double eps, Tensor& out, Tensor& save_mean, Tensor& save_var) { in batch_norm_cpu_out()
824 Tensor save_var; in batch_norm_cpu() local
865 Tensor& out, Tensor& save_mean, Tensor& save_var, Tensor& reserve) { in _batch_norm_with_update_cpu_out()
902 …g_var, bool train, double momentum, double eps, Tensor& out, Tensor& save_mean, Tensor& save_var) { in _batch_norm_legit_cpu_out()
907 …s_opt, bool train, double momentum, double eps, Tensor& out, Tensor& save_mean, Tensor& save_var) { in _batch_norm_legit_no_stats_cpu_out()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | BatchNorm_miopen.cpp | 113 Tensor save_mean, save_var; in miopen_batch_norm() local
184 save_var{ save_var_t, "save_var", 5 }; in miopen_batch_norm_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | BatchNorm.cpp | 185 Tensor save_mean, save_var; in cudnn_batch_norm() local
297 save_var{save_var_t, "save_var", 5}, in cudnn_batch_norm_backward() local
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | gen_autograd_functions.py | 563 def save_var(var: SavedAttribute, is_output: bool) -> None: function
|