Searched defs:invstd_ (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cuh | 695 … const Tensor& bias_, const Tensor& mean_, const Tensor& invstd_) { in batch_norm_elemt_cuda_template()
736 …e<Tensor, Tensor> batch_norm_gather_stats_cuda_template(const Tensor& mean_, const Tensor& invstd_, in batch_norm_gather_stats_cuda_template()
779 … const Tensor& mean_, const Tensor& invstd_, const Tensor& weight_, in batch_norm_backward_reduce_cuda_template()
839 const Tensor& mean_, const Tensor& invstd_, in batch_norm_backward_elemt_cuda_template()
890 const Tensor& mean_, const Tensor& invstd_, in batch_norm_backward_elemt_cuda_template()
|
| H A D | Normalization.cu | 99 const std::optional<Tensor>& bias_opt, const Tensor& mean_, const Tensor& invstd_) { in batch_norm_elementwise()
|