Searched defs:input_g (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Convolution.cpp | 1656 auto input_g = subtensor(input, 1, params.groups, g); in _convolution() local
2232 auto input_g = subtensor(input, 1, params.groups, g); in convolution_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cuh | 780 … const bool input_g, const bool weight_g, const bool bias_g) { in batch_norm_backward_reduce_cuda_template()
1533 const bool input_g, const bool weight_g, const bool bias_g) { in batch_norm_backward_reduce_cuda_channels_last_template()
|
| H A D | Normalization.cu | 725 … mean, const Tensor& invstd, const std::optional<Tensor>& weight_opt, bool input_g, bool weight_g,… in batch_norm_backward_reduce_cuda()
|