Home
last modified time | relevance | path

Searched defs:bias_g (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cuh780 … 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 DNormalization.cu725 …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 DConvolution.cpp1658 auto bias_g = subtensor(bias, 0, params.groups, g); in _convolution() local