Searched defs:bias_g (Results 1 – 3 of 3) sorted by relevance
/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 …ensor& invstd, const std::optional<Tensor>& weight_opt, bool input_g, bool weight_g, bool bias_g) { in batch_norm_backward_reduce_cuda()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | Convolution.cpp | 1658 auto bias_g = subtensor(bias, 0, params.groups, g); in _convolution() local
|